This commit is contained in:
RePixelatedMC 2024-09-06 14:35:42 +02:00
parent 613575c0c5
commit 9f392654d3

View File

@ -26,17 +26,11 @@ import java.util.function.Consumer;
import java.util.function.Function; import java.util.function.Function;
import java.util.function.Predicate; import java.util.function.Predicate;
import java.util.stream.Collectors; import java.util.stream.Collectors;
import java.util.stream.IntStream;
import java.util.stream.Stream;
public class EngineMobHandlerSVC extends IrisEngineService implements IrisMobDataHandler { public class EngineMobHandlerSVC extends IrisEngineService implements IrisMobDataHandler {
private HashSet<HistoryData> history; private HashSet<HistoryData> history;
private Function<Map<IrisMobPiece, List<Integer>>, LinkedHashMap<IrisMobPiece, List<Integer>>> sortMapFunction;
private Function<Map<IrisMobPiece, Integer>, LinkedHashMap<IrisMobPiece, Integer>> sortMapSimpleFunction;
private int id; private int id;
public int energyMax; public int energyMax;
public int energy; public int energy;
@ -58,52 +52,6 @@ public class EngineMobHandlerSVC extends IrisEngineService implements IrisMobDat
this.entityType = (entityType) -> Types.valueOf(INMS.get().getMobCategory(entityType)); this.entityType = (entityType) -> Types.valueOf(INMS.get().getMobCategory(entityType));
this.loadedChunks = new HashSet<>(); this.loadedChunks = new HashSet<>();
this.bukkitLimits = getBukkitLimits(); this.bukkitLimits = getBukkitLimits();
this.sortMapFunction =
new Function<>() {
private Map<IrisMobPiece, List<Integer>> lastMap;
private LinkedHashMap<IrisMobPiece, List<Integer>> cachedSortedMap;
@Override
public LinkedHashMap<IrisMobPiece, List<Integer>> apply(Map<IrisMobPiece, List<Integer>> inputMap) {
if (cachedSortedMap == null || !inputMap.equals(lastMap)) {
cachedSortedMap = inputMap.entrySet()
.stream()
.sorted(Map.Entry.<IrisMobPiece, List<Integer>>comparingByValue(
Comparator.comparingInt(list -> list.stream().mapToInt(Integer::intValue).sum())
).reversed())
.collect(Collectors.toMap(
Map.Entry::getKey,
Map.Entry::getValue,
(e1, e2) -> e1,
LinkedHashMap::new
));
lastMap = new HashMap<>(inputMap);
}
return cachedSortedMap;
}
};
this.sortMapSimpleFunction = new Function<>() {
private Map<IrisMobPiece, Integer> lastMap;
private LinkedHashMap<IrisMobPiece, Integer> cachedSortedMap;
@Override
public LinkedHashMap<IrisMobPiece, Integer> apply(Map<IrisMobPiece, Integer> inputMap) {
if (cachedSortedMap == null || !inputMap.equals(lastMap)) {
cachedSortedMap = inputMap.entrySet()
.stream()
.sorted(Map.Entry.<IrisMobPiece, Integer>comparingByValue(Comparator.reverseOrder()))
.collect(Collectors.toMap(
Map.Entry::getKey,
Map.Entry::getValue,
(e1, e2) -> e1,
LinkedHashMap::new
));
lastMap = new HashMap<>(inputMap);
}
return cachedSortedMap;
}
};
new Ticker(); new Ticker();
} }
@ -111,8 +59,6 @@ public class EngineMobHandlerSVC extends IrisEngineService implements IrisMobDat
@Override @Override
public void onDisable(boolean hotload) { public void onDisable(boolean hotload) {
} }
private class Ticker extends Looper { private class Ticker extends Looper {
@ -132,7 +78,6 @@ public class EngineMobHandlerSVC extends IrisEngineService implements IrisMobDat
if (engine.isClosed() || engine.getCacheID() != id) { if (engine.isClosed() || engine.getCacheID() != id) {
interrupt(); interrupt();
} }
PrecisionStopwatch stopwatch = new PrecisionStopwatch(); PrecisionStopwatch stopwatch = new PrecisionStopwatch();
stopwatch.begin(); stopwatch.begin();
@ -140,18 +85,13 @@ public class EngineMobHandlerSVC extends IrisEngineService implements IrisMobDat
.collect(Collectors.toCollection(HashSet::new)); .collect(Collectors.toCollection(HashSet::new));
updateMaxEnergy(); updateMaxEnergy();
Predicate<IrisMobPiece> shouldTick = IrisMobPiece::shouldTick; Predicate<IrisMobPiece> shouldTick = IrisMobPiece::shouldTick;
Function<IrisMobPiece, List<Integer>> tickCosts = piece -> piece.getTickCosts(1); Map<IrisMobPiece, Integer> costs = assignEnergyToPieces(pieces.stream()
Map<IrisMobPiece, List<Integer>> pieceTickCostsMap = pieces.stream()
.collect(Collectors.toMap( .collect(Collectors.toMap(
Function.identity(), piece -> piece,
tickCosts piece -> piece.getTickCosts(60)
)); )));
Consumer<IrisMobPiece> tick = piece -> piece.tick(costs.get(piece));
Consumer<IrisMobPiece> tick = piece -> piece.tick(42);
pieces.stream() pieces.stream()
.filter(shouldTick) .filter(shouldTick)
@ -172,72 +112,65 @@ public class EngineMobHandlerSVC extends IrisEngineService implements IrisMobDat
} }
/** /**
* @param req Data to do calculations with. List<Integer>> = Requested energy + future * Assigns energy to pieces based on current and future energy requirements.
* @return returns the energy distribution for each piece * @param req Data to do calculations with. Map of IrisMobPiece to List<Integer> representing requested energy + future
* @return Map of IrisMobPiece to Integer representing the energy distribution for each piece
*/ */
private LinkedHashMap<IrisMobPiece, Integer> assignEnergyToPieces(LinkedHashMap<IrisMobPiece, List<Integer>> req) { private Map<IrisMobPiece, Integer> assignEnergyToPieces(Map<IrisMobPiece, List<Integer>> req) {
final int FUTURE_VIEW = 60;
int currentEnergy = calculateNewEnergy();
// Might need caching? Map<IrisMobPiece, Integer> finalMap = new LinkedHashMap<>();
int futureView = 60; req.forEach((piece, value) -> finalMap.put(piece, null));
int energy = calculateNewEnergy();
LinkedHashMap<IrisMobPiece, Integer> finalMap = req.keySet().stream() double currentEnergyRatio = (double) currentEnergy / sumCurrentRequests(req);
.collect(Collectors.toMap( double futureEnergyRatio = (double) predictEnergy(FUTURE_VIEW) / calculateFutureMedian(req, FUTURE_VIEW);
piece -> piece,
piece -> null,
(existing, replacement) -> existing,
LinkedHashMap::new
));
Function<Integer, Stream<Integer>> viewSingleFuture = (future) -> if (futureEnergyRatio > 1.25 && currentEnergyRatio > 1) {
req.values().stream() req.forEach((piece, energyList) -> finalMap.put(piece, energyList.isEmpty() ? 0 : energyList.get(0)));
.mapToInt(list -> list.isEmpty() ? 0 : list.get(future)).boxed(); } else if (currentEnergyRatio > 1) {
Function<Integer, Function<Integer, Stream<Integer>>> viewFuture = (rangeMin) -> (rangeMax) -> // TODO: Implement hard part
IntStream.range(rangeMin, rangeMax)
.boxed()
.flatMap(viewSingleFuture);
Function<Integer, Double> viewFutureMedian = (value) -> viewFuture.apply(value).apply(futureView) } else if (currentEnergyRatio < 1) {
.sorted() double scale = calculateScale(currentEnergy, req);
.collect(Collectors.collectingAndThen(Collectors.toList(), req.forEach((piece, energyList) -> finalMap.put(piece, (int) (energyList.get(0) * scale)));
list -> {
int size = list.size();
if (size % 2 == 0) {
return (list.get(size / 2 - 1) + list.get(size / 2)) / 2.0;
} else {
return list.get(size / 2).doubleValue();
}
}));
// Logic
if ((predictEnergy(futureView) / viewFutureMedian.apply(0)) > 1.25 && (energy / viewSingleFuture.apply(0).mapToInt(Integer::intValue).sum() > 1)) {
finalMap = req.entrySet().stream()
.collect(Collectors.toMap(
Map.Entry::getKey,
e -> e.getValue().isEmpty() ? 0 : e.getValue().get(0),
(v1, v2) -> v1,
LinkedHashMap::new
));
} else if ((energy / viewSingleFuture.apply(0).mapToInt(Integer::intValue).sum() > 1)) {
// hard part
} else if ((energy / viewSingleFuture.apply(0).mapToInt(Integer::intValue).sum() < 1)) {
double scale = 1;
while ((double) energy / viewSingleFuture.apply(0).mapToInt(Integer::intValue).sum() >= scale) {
scale -= 0.1;
}
double finalScale = scale + 0.1;
LinkedHashMap<IrisMobPiece, Integer> finalMap1 = finalMap;
req.forEach((key, value) -> finalMap1.put(key, (int) (value.get(0) * finalScale)));
} }
return finalMap; return finalMap;
}
private int sumCurrentRequests(Map<IrisMobPiece, List<Integer>> req) {
return req.values().stream()
.mapToInt(list -> list.isEmpty() ? 0 : list.get(0))
.sum();
}
private double calculateFutureMedian(Map<IrisMobPiece, List<Integer>> req, int futureView) {
List<Integer> allFutureValues = new ArrayList<>();
for (int i = 0; i < futureView; i++) {
for (List<Integer> energyList : req.values()) {
if (i < energyList.size()) {
allFutureValues.add(energyList.get(i));
}
}
}
Collections.sort(allFutureValues);
int size = allFutureValues.size();
if (size % 2 == 0) {
return (allFutureValues.get(size / 2 - 1) + allFutureValues.get(size / 2)) / 2.0;
} else {
return allFutureValues.get(size / 2);
}
}
private double calculateScale(int currentEnergy, Map<IrisMobPiece, List<Integer>> req) {
double scale = 1.0;
double ratio = (double) currentEnergy / sumCurrentRequests(req);
while (ratio < scale) {
scale -= 0.1;
}
return Math.min(scale + 0.1, 1.0);
} }