60 lines
2.3 KiB
Java
60 lines
2.3 KiB
Java
package net.minecraft.world.entity.schedule;
|
|
|
|
import com.google.common.collect.Maps;
|
|
import java.util.Comparator;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
import java.util.Map.Entry;
|
|
import java.util.stream.Collectors;
|
|
import net.minecraft.core.Registry;
|
|
import net.minecraft.core.registries.BuiltInRegistries;
|
|
|
|
public class Schedule {
|
|
public static final int WORK_START_TIME = 2000;
|
|
public static final int TOTAL_WORK_TIME = 7000;
|
|
public static final Schedule EMPTY = register("empty").changeActivityAt(0, Activity.IDLE).build();
|
|
public static final Schedule SIMPLE = register("simple").changeActivityAt(5000, Activity.WORK).changeActivityAt(11000, Activity.REST).build();
|
|
public static final Schedule VILLAGER_BABY = register("villager_baby")
|
|
.changeActivityAt(10, Activity.IDLE)
|
|
.changeActivityAt(3000, Activity.PLAY)
|
|
.changeActivityAt(6000, Activity.IDLE)
|
|
.changeActivityAt(10000, Activity.PLAY)
|
|
.changeActivityAt(12000, Activity.REST)
|
|
.build();
|
|
public static final Schedule VILLAGER_DEFAULT = register("villager_default")
|
|
.changeActivityAt(10, Activity.IDLE)
|
|
.changeActivityAt(2000, Activity.WORK)
|
|
.changeActivityAt(9000, Activity.MEET)
|
|
.changeActivityAt(11000, Activity.IDLE)
|
|
.changeActivityAt(12000, Activity.REST)
|
|
.build();
|
|
private final Map<Activity, Timeline> timelines = Maps.<Activity, Timeline>newHashMap();
|
|
|
|
protected static ScheduleBuilder register(String key) {
|
|
Schedule schedule = Registry.register(BuiltInRegistries.SCHEDULE, key, new Schedule());
|
|
return new ScheduleBuilder(schedule);
|
|
}
|
|
|
|
protected void ensureTimelineExistsFor(Activity activity) {
|
|
if (!this.timelines.containsKey(activity)) {
|
|
this.timelines.put(activity, new Timeline());
|
|
}
|
|
}
|
|
|
|
protected Timeline getTimelineFor(Activity activity) {
|
|
return (Timeline)this.timelines.get(activity);
|
|
}
|
|
|
|
protected List<Timeline> getAllTimelinesExceptFor(Activity activity) {
|
|
return (List<Timeline>)this.timelines.entrySet().stream().filter(entry -> entry.getKey() != activity).map(Entry::getValue).collect(Collectors.toList());
|
|
}
|
|
|
|
public Activity getActivityAt(int dayTime) {
|
|
return (Activity)this.timelines
|
|
.entrySet()
|
|
.stream()
|
|
.max(Comparator.comparingDouble(entry -> ((Timeline)entry.getValue()).getValueAt(dayTime)))
|
|
.map(Entry::getKey)
|
|
.orElse(Activity.IDLE);
|
|
}
|
|
}
|