mirror of
https://github.com/Grasscutters/Grasscutter.git
synced 2025-01-24 03:43:01 +08:00
TSJ and TSV parsing (#1962)
* Deserialization support for tsv files * Benchmarking * Apparently moving the setter out of the lambda fixed the setAccessible issue * Thread it * Use AllArgsConstructor instead of field reflection * Clean up AllArgsConstructor TSV deserialization * Refactor TsvUtils * Remove AllArgsConstructors from Excels * Set field accessible * [WIP] TSJ improvements * [WIP] More TSV stuff * [WIP] More TSV stuff * Working TSV parser (slow) * Load Excels in TSJ > JSON > TSV priority
This commit is contained in:
parent
46b0c7cf93
commit
0b5329514b
@ -11,7 +11,10 @@ import emu.grasscutter.game.world.SpawnDataEntry;
|
|||||||
import emu.grasscutter.game.world.SpawnDataEntry.GridBlockId;
|
import emu.grasscutter.game.world.SpawnDataEntry.GridBlockId;
|
||||||
import emu.grasscutter.game.world.SpawnDataEntry.SpawnGroupEntry;
|
import emu.grasscutter.game.world.SpawnDataEntry.SpawnGroupEntry;
|
||||||
import emu.grasscutter.scripts.SceneIndexManager;
|
import emu.grasscutter.scripts.SceneIndexManager;
|
||||||
|
import emu.grasscutter.utils.FileUtils;
|
||||||
import emu.grasscutter.utils.JsonUtils;
|
import emu.grasscutter.utils.JsonUtils;
|
||||||
|
import emu.grasscutter.utils.TsvUtils;
|
||||||
|
import it.unimi.dsi.fastutil.Pair;
|
||||||
import it.unimi.dsi.fastutil.ints.Int2ObjectMap;
|
import it.unimi.dsi.fastutil.ints.Int2ObjectMap;
|
||||||
import it.unimi.dsi.fastutil.ints.IntArrayList;
|
import it.unimi.dsi.fastutil.ints.IntArrayList;
|
||||||
import it.unimi.dsi.fastutil.ints.IntArraySet;
|
import it.unimi.dsi.fastutil.ints.IntArraySet;
|
||||||
@ -23,6 +26,8 @@ import java.io.*;
|
|||||||
import java.nio.file.Files;
|
import java.nio.file.Files;
|
||||||
import java.nio.file.Path;
|
import java.nio.file.Path;
|
||||||
import java.util.*;
|
import java.util.*;
|
||||||
|
import java.util.concurrent.ConcurrentLinkedQueue;
|
||||||
|
import java.util.concurrent.CopyOnWriteArraySet;
|
||||||
import java.util.regex.Pattern;
|
import java.util.regex.Pattern;
|
||||||
import java.util.stream.Stream;
|
import java.util.stream.Stream;
|
||||||
|
|
||||||
@ -32,8 +37,9 @@ import static emu.grasscutter.utils.Language.translate;
|
|||||||
|
|
||||||
public class ResourceLoader {
|
public class ResourceLoader {
|
||||||
|
|
||||||
private static final List<String> loadedResources = new ArrayList<>();
|
private static final Set<String> loadedResources = new CopyOnWriteArraySet<>();
|
||||||
|
|
||||||
|
// Get a list of all resource classes, sorted by loadPriority
|
||||||
public static List<Class<?>> getResourceDefClasses() {
|
public static List<Class<?>> getResourceDefClasses() {
|
||||||
Reflections reflections = new Reflections(ResourceLoader.class.getPackage().getName());
|
Reflections reflections = new Reflections(ResourceLoader.class.getPackage().getName());
|
||||||
Set<?> classes = reflections.getSubTypesOf(GameResource.class);
|
Set<?> classes = reflections.getSubTypesOf(GameResource.class);
|
||||||
@ -51,6 +57,25 @@ public class ResourceLoader {
|
|||||||
return classList;
|
return classList;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Get a list containing sets of all resource classes, sorted by loadPriority
|
||||||
|
protected static List<Set<Class<?>>> getResourceDefClassesPrioritySets() {
|
||||||
|
val reflections = new Reflections(ResourceLoader.class.getPackage().getName());
|
||||||
|
val classes = reflections.getSubTypesOf(GameResource.class);
|
||||||
|
val priorities = ResourceType.LoadPriority.getInOrder();
|
||||||
|
Grasscutter.getLogger().debug("Priorities are "+priorities);
|
||||||
|
val map = new LinkedHashMap<ResourceType.LoadPriority, Set<Class<?>>>(priorities.size());
|
||||||
|
priorities.forEach(p -> map.put(p, new HashSet<>()));
|
||||||
|
|
||||||
|
classes.forEach(c -> {
|
||||||
|
// val c = (Class<?>) o;
|
||||||
|
val annotation = c.getAnnotation(ResourceType.class);
|
||||||
|
if (annotation != null) {
|
||||||
|
map.get(annotation.loadPriority()).add(c);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
return List.copyOf(map.values());
|
||||||
|
}
|
||||||
|
|
||||||
private static boolean loadedAll = false;
|
private static boolean loadedAll = false;
|
||||||
public static void loadAll() {
|
public static void loadAll() {
|
||||||
if (loadedAll) return;
|
if (loadedAll) return;
|
||||||
@ -86,48 +111,66 @@ public class ResourceLoader {
|
|||||||
}
|
}
|
||||||
|
|
||||||
public static void loadResources(boolean doReload) {
|
public static void loadResources(boolean doReload) {
|
||||||
for (Class<?> resourceDefinition : getResourceDefClasses()) {
|
long startTime = System.nanoTime();
|
||||||
ResourceType type = resourceDefinition.getAnnotation(ResourceType.class);
|
val errors = new ConcurrentLinkedQueue<Pair<String, Exception>>(); // Logger in a parallel stream will deadlock
|
||||||
|
|
||||||
if (type == null) {
|
getResourceDefClassesPrioritySets().forEach(classes -> {
|
||||||
continue;
|
classes.stream()
|
||||||
}
|
.parallel().unordered()
|
||||||
|
.forEach(c -> {
|
||||||
|
val type = c.getAnnotation(ResourceType.class);
|
||||||
|
if (type == null) return;
|
||||||
|
|
||||||
@SuppressWarnings("rawtypes")
|
val map = GameData.getMapByResourceDef(c);
|
||||||
Int2ObjectMap map = GameData.getMapByResourceDef(resourceDefinition);
|
if (map == null) return;
|
||||||
|
|
||||||
if (map == null) {
|
try {
|
||||||
continue;
|
loadFromResource(c, type, map, doReload);
|
||||||
}
|
} catch (Exception e) {
|
||||||
|
errors.add(Pair.of(Arrays.toString(type.name()), e));
|
||||||
try {
|
}
|
||||||
loadFromResource(resourceDefinition, type, map, doReload);
|
});
|
||||||
} catch (Exception e) {
|
});
|
||||||
Grasscutter.getLogger().error("Error loading resource file: " + Arrays.toString(type.name()), e.getLocalizedMessage());
|
errors.forEach(pair -> Grasscutter.getLogger().error("Error loading resource file: " + pair.left(), pair.right()));
|
||||||
}
|
long endTime = System.nanoTime();
|
||||||
}
|
long ns = (endTime - startTime); //divide by 1000000 to get milliseconds.
|
||||||
|
Grasscutter.getLogger().debug("Loading resources took "+ns+"ns == "+ns/1000000+"ms");
|
||||||
}
|
}
|
||||||
|
|
||||||
@SuppressWarnings("rawtypes")
|
@SuppressWarnings("rawtypes")
|
||||||
protected static void loadFromResource(Class<?> c, ResourceType type, Int2ObjectMap map, boolean doReload) throws Exception {
|
protected static void loadFromResource(Class<?> c, ResourceType type, Int2ObjectMap map, boolean doReload) throws Exception {
|
||||||
if (!loadedResources.contains(c.getSimpleName()) || doReload) {
|
val simpleName = c.getSimpleName();
|
||||||
|
if (doReload || !loadedResources.contains(simpleName)) {
|
||||||
for (String name : type.name()) {
|
for (String name : type.name()) {
|
||||||
loadFromResource(c, name, map);
|
loadFromResource(c, FileUtils.getExcelPath(name), map);
|
||||||
}
|
}
|
||||||
loadedResources.add(c.getSimpleName());
|
loadedResources.add(simpleName);
|
||||||
Grasscutter.getLogger().debug("Loaded " + map.size() + " " + c.getSimpleName() + "s.");
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@SuppressWarnings({"rawtypes", "unchecked"})
|
@SuppressWarnings({"rawtypes", "unchecked"})
|
||||||
protected static <T> void loadFromResource(Class<T> c, String fileName, Int2ObjectMap map) throws Exception {
|
protected static <T> void loadFromResource(Class<T> c, Path filename, Int2ObjectMap map) throws Exception {
|
||||||
List<T> list = JsonUtils.loadToList(getResourcePath("ExcelBinOutput/" + fileName), c);
|
val results = switch (FileUtils.getFileExtension(filename)) {
|
||||||
|
case "json" -> JsonUtils.loadToList(filename, c);
|
||||||
for (T o : list) {
|
case "tsj" -> TsvUtils.loadTsjToListSetField(c, filename);
|
||||||
|
case "tsv" -> TsvUtils.loadTsvToListSetField(c, filename);
|
||||||
|
default -> null;
|
||||||
|
};
|
||||||
|
if (results == null) return;
|
||||||
|
results.forEach(o -> {
|
||||||
GameResource res = (GameResource) o;
|
GameResource res = (GameResource) o;
|
||||||
res.onLoad();
|
res.onLoad();
|
||||||
map.put(res.getId(), res);
|
map.put(res.getId(), res);
|
||||||
}
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
@SuppressWarnings({"rawtypes", "unchecked"})
|
||||||
|
protected static <T> void loadFromResource(Class<T> c, String fileName, Int2ObjectMap map) throws Exception {
|
||||||
|
JsonUtils.loadToList(getResourcePath("ExcelBinOutput/" + fileName), c).forEach(o -> {
|
||||||
|
GameResource res = (GameResource) o;
|
||||||
|
res.onLoad();
|
||||||
|
map.put(res.getId(), res);
|
||||||
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
public class ScenePointConfig { // Sadly this doesn't work as a local class in loadScenePoints()
|
public class ScenePointConfig { // Sadly this doesn't work as a local class in loadScenePoints()
|
||||||
|
@ -2,6 +2,8 @@ package emu.grasscutter.data;
|
|||||||
|
|
||||||
import java.lang.annotation.Retention;
|
import java.lang.annotation.Retention;
|
||||||
import java.lang.annotation.RetentionPolicy;
|
import java.lang.annotation.RetentionPolicy;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.stream.Stream;
|
||||||
|
|
||||||
@Retention(RetentionPolicy.RUNTIME)
|
@Retention(RetentionPolicy.RUNTIME)
|
||||||
public @interface ResourceType {
|
public @interface ResourceType {
|
||||||
@ -28,5 +30,9 @@ public @interface ResourceType {
|
|||||||
public int value() {
|
public int value() {
|
||||||
return value;
|
return value;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public static List<LoadPriority> getInOrder() {
|
||||||
|
return Stream.of(LoadPriority.values()).sorted((x, y) -> y.value() - x.value()).toList();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -21,7 +21,7 @@ public class ActivityWatcherData extends GameResource {
|
|||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void onLoad() {
|
public void onLoad() {
|
||||||
triggerConfig.paramList = triggerConfig.paramList.stream().filter(x -> !x.isBlank()).toList();
|
triggerConfig.paramList = triggerConfig.paramList.stream().filter(x -> (x != null) && !x.isBlank()).toList();
|
||||||
triggerConfig.watcherTriggerType = WatcherTriggerType.getTypeByName(triggerConfig.triggerType);
|
triggerConfig.watcherTriggerType = WatcherTriggerType.getTypeByName(triggerConfig.triggerType);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -40,8 +40,11 @@ public class BattlePassMissionData extends GameResource {
|
|||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void onLoad() {
|
public void onLoad() {
|
||||||
if (this.getTriggerConfig() != null && getTriggerConfig().getParamList()[0].length() > 0) {
|
if (this.getTriggerConfig() != null) {
|
||||||
this.mainParams = Arrays.stream(getTriggerConfig().getParamList()[0].split("[:;,]")).map(Integer::parseInt).collect(Collectors.toSet());
|
var params = getTriggerConfig().getParamList()[0];
|
||||||
|
if ((params != null) && !params.isEmpty()) {
|
||||||
|
this.mainParams = Arrays.stream(params.split("[:;,]")).map(Integer::parseInt).collect(Collectors.toSet());
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1,19 +1,18 @@
|
|||||||
package emu.grasscutter.game.props.ItemUseAction;
|
package emu.grasscutter.game.props.ItemUseAction;
|
||||||
|
|
||||||
import emu.grasscutter.game.props.ItemUseOp;
|
import emu.grasscutter.game.props.ItemUseOp;
|
||||||
import lombok.Getter;
|
|
||||||
|
|
||||||
public class ItemUseAddExp extends ItemUseAction {
|
|
||||||
@Getter private int exp = 0;
|
|
||||||
|
|
||||||
|
public class ItemUseAddExp extends ItemUseInt {
|
||||||
@Override
|
@Override
|
||||||
public ItemUseOp getItemUseOp() {
|
public ItemUseOp getItemUseOp() {
|
||||||
return ItemUseOp.ITEM_USE_ADD_EXP;
|
return ItemUseOp.ITEM_USE_ADD_EXP;
|
||||||
}
|
}
|
||||||
|
|
||||||
public ItemUseAddExp(String[] useParam) {
|
public ItemUseAddExp(String[] useParam) {
|
||||||
try {
|
super(useParam);
|
||||||
this.exp = Integer.parseInt(useParam[0]);
|
}
|
||||||
} catch (NumberFormatException ignored) {}
|
|
||||||
|
public int getExp() {
|
||||||
|
return this.i;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -14,7 +14,7 @@ public class ItemUseAddItem extends ItemUseInt {
|
|||||||
super(useParam);
|
super(useParam);
|
||||||
try {
|
try {
|
||||||
this.count = Integer.parseInt(useParam[1]);
|
this.count = Integer.parseInt(useParam[1]);
|
||||||
} catch (NumberFormatException ignored) {}
|
} catch (NumberFormatException | ArrayIndexOutOfBoundsException ignored) {}
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
@ -1,19 +1,14 @@
|
|||||||
package emu.grasscutter.game.props.ItemUseAction;
|
package emu.grasscutter.game.props.ItemUseAction;
|
||||||
|
|
||||||
import emu.grasscutter.game.props.ItemUseOp;
|
import emu.grasscutter.game.props.ItemUseOp;
|
||||||
import lombok.Getter;
|
|
||||||
|
|
||||||
public class ItemUseAddReliquaryExp extends ItemUseAction {
|
|
||||||
@Getter private int exp = 0;
|
|
||||||
|
|
||||||
|
public class ItemUseAddReliquaryExp extends ItemUseAddExp {
|
||||||
@Override
|
@Override
|
||||||
public ItemUseOp getItemUseOp() {
|
public ItemUseOp getItemUseOp() {
|
||||||
return ItemUseOp.ITEM_USE_ADD_RELIQUARY_EXP;
|
return ItemUseOp.ITEM_USE_ADD_RELIQUARY_EXP;
|
||||||
}
|
}
|
||||||
|
|
||||||
public ItemUseAddReliquaryExp(String[] useParam) {
|
public ItemUseAddReliquaryExp(String[] useParam) {
|
||||||
try {
|
super(useParam);
|
||||||
this.exp = Integer.parseInt(useParam[0]);
|
|
||||||
} catch (NumberFormatException ignored) {}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -14,7 +14,7 @@ public class ItemUseAddServerBuff extends ItemUseInt {
|
|||||||
super(useParam);
|
super(useParam);
|
||||||
try {
|
try {
|
||||||
this.duration = Integer.parseInt(useParam[1]);
|
this.duration = Integer.parseInt(useParam[1]);
|
||||||
} catch (NumberFormatException ignored) {}
|
} catch (NumberFormatException | ArrayIndexOutOfBoundsException ignored) {}
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
@ -1,19 +1,14 @@
|
|||||||
package emu.grasscutter.game.props.ItemUseAction;
|
package emu.grasscutter.game.props.ItemUseAction;
|
||||||
|
|
||||||
import emu.grasscutter.game.props.ItemUseOp;
|
import emu.grasscutter.game.props.ItemUseOp;
|
||||||
import lombok.Getter;
|
|
||||||
|
|
||||||
public class ItemUseAddWeaponExp extends ItemUseAction {
|
|
||||||
@Getter private int exp = 0;
|
|
||||||
|
|
||||||
|
public class ItemUseAddWeaponExp extends ItemUseAddExp {
|
||||||
@Override
|
@Override
|
||||||
public ItemUseOp getItemUseOp() {
|
public ItemUseOp getItemUseOp() {
|
||||||
return ItemUseOp.ITEM_USE_ADD_WEAPON_EXP;
|
return ItemUseOp.ITEM_USE_ADD_WEAPON_EXP;
|
||||||
}
|
}
|
||||||
|
|
||||||
public ItemUseAddWeaponExp(String[] useParam) {
|
public ItemUseAddWeaponExp(String[] useParam) {
|
||||||
try {
|
super(useParam);
|
||||||
this.exp = Integer.parseInt(useParam[0]);
|
|
||||||
} catch (NumberFormatException ignored) {}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -15,10 +15,10 @@ public class ItemUseCombineItem extends ItemUseInt {
|
|||||||
super(useParam);
|
super(useParam);
|
||||||
try {
|
try {
|
||||||
this.resultId = Integer.parseInt(useParam[1]);
|
this.resultId = Integer.parseInt(useParam[1]);
|
||||||
} catch (NumberFormatException ignored) {}
|
} catch (NumberFormatException | ArrayIndexOutOfBoundsException ignored) {}
|
||||||
try {
|
try {
|
||||||
this.resultCount = Integer.parseInt(useParam[2]);
|
this.resultCount = Integer.parseInt(useParam[2]);
|
||||||
} catch (NumberFormatException ignored) {}
|
} catch (NumberFormatException | ArrayIndexOutOfBoundsException ignored) {}
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
@ -19,10 +19,10 @@ public class ItemUseGainAvatar extends ItemUseInt {
|
|||||||
super(useParam);
|
super(useParam);
|
||||||
try {
|
try {
|
||||||
this.level = Integer.parseInt(useParam[1]);
|
this.level = Integer.parseInt(useParam[1]);
|
||||||
} catch (NumberFormatException ignored) {}
|
} catch (NumberFormatException | ArrayIndexOutOfBoundsException ignored) {}
|
||||||
try {
|
try {
|
||||||
this.constellation = Integer.parseInt(useParam[2]);
|
this.constellation = Integer.parseInt(useParam[2]);
|
||||||
} catch (NumberFormatException ignored) {}
|
} catch (NumberFormatException | ArrayIndexOutOfBoundsException ignored) {}
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
@ -8,6 +8,6 @@ public abstract class ItemUseInt extends ItemUseAction {
|
|||||||
public ItemUseInt(String[] useParam) {
|
public ItemUseInt(String[] useParam) {
|
||||||
try {
|
try {
|
||||||
this.i = Integer.parseInt(useParam[0]);
|
this.i = Integer.parseInt(useParam[0]);
|
||||||
} catch (NumberFormatException ignored) {}
|
} catch (NumberFormatException | ArrayIndexOutOfBoundsException ignored) {}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -1,6 +1,7 @@
|
|||||||
package emu.grasscutter.utils;
|
package emu.grasscutter.utils;
|
||||||
|
|
||||||
import emu.grasscutter.Grasscutter;
|
import emu.grasscutter.Grasscutter;
|
||||||
|
import lombok.val;
|
||||||
|
|
||||||
import java.io.File;
|
import java.io.File;
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
@ -111,6 +112,24 @@ public final class FileUtils {
|
|||||||
return RESOURCES_PATH.resolve(path);
|
return RESOURCES_PATH.resolve(path);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public static Path getExcelPath(String filename) {
|
||||||
|
return getTsjJsonTsv(RESOURCES_PATH.resolve("ExcelBinOutput"), filename);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Gets path of a resource.
|
||||||
|
// If multiple formats of it exist, priority is TSJ > JSON > TSV
|
||||||
|
// If none exist, return the TSJ path, in case it wants to create a file
|
||||||
|
public static Path getTsjJsonTsv(Path root, String filename) {
|
||||||
|
val name = getFilenameWithoutExtension(filename);
|
||||||
|
val tsj = root.resolve(name + ".tsj");
|
||||||
|
if (Files.exists(tsj)) return tsj;
|
||||||
|
val json = root.resolve(name + ".json");
|
||||||
|
if (Files.exists(json)) return json;
|
||||||
|
val tsv = root.resolve(name + ".tsv");
|
||||||
|
if (Files.exists(tsv)) return tsv;
|
||||||
|
return tsj;
|
||||||
|
}
|
||||||
|
|
||||||
public static Path getScriptPath(String path) {
|
public static Path getScriptPath(String path) {
|
||||||
return SCRIPTS_PATH.resolve(path);
|
return SCRIPTS_PATH.resolve(path);
|
||||||
}
|
}
|
||||||
@ -167,14 +186,19 @@ public final class FileUtils {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@Deprecated // No current uses of this anyway
|
@Deprecated // Misnamed legacy function
|
||||||
public static String getFilenameWithoutPath(String fileName) {
|
public static String getFilenameWithoutPath(String filename) {
|
||||||
int i = fileName.lastIndexOf(".");
|
return getFilenameWithoutExtension(filename);
|
||||||
if (i > 0) {
|
}
|
||||||
return fileName.substring(0, i);
|
public static String getFilenameWithoutExtension(String filename) {
|
||||||
} else {
|
int i = filename.lastIndexOf(".");
|
||||||
return fileName;
|
return (i < 0) ? filename : filename.substring(0, i);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public static String getFileExtension(Path path) {
|
||||||
|
val filename = path.toString();
|
||||||
|
int i = filename.lastIndexOf(".");
|
||||||
|
return (i < 0) ? "" : filename.substring(i+1);
|
||||||
}
|
}
|
||||||
|
|
||||||
public static List<Path> getPathsFromResource(String folder) throws URISyntaxException {
|
public static List<Path> getPathsFromResource(String folder) throws URISyntaxException {
|
||||||
|
@ -4,6 +4,7 @@ import java.io.FileInputStream;
|
|||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
import java.io.InputStreamReader;
|
import java.io.InputStreamReader;
|
||||||
import java.io.Reader;
|
import java.io.Reader;
|
||||||
|
import java.lang.reflect.Type;
|
||||||
import java.nio.charset.StandardCharsets;
|
import java.nio.charset.StandardCharsets;
|
||||||
import java.nio.file.Files;
|
import java.nio.file.Files;
|
||||||
import java.nio.file.Path;
|
import java.nio.file.Path;
|
||||||
@ -18,6 +19,7 @@ import com.google.gson.reflect.TypeToken;
|
|||||||
|
|
||||||
import emu.grasscutter.data.common.DynamicFloat;
|
import emu.grasscutter.data.common.DynamicFloat;
|
||||||
import emu.grasscutter.utils.JsonAdapters.*;
|
import emu.grasscutter.utils.JsonAdapters.*;
|
||||||
|
|
||||||
import it.unimi.dsi.fastutil.ints.IntList;
|
import it.unimi.dsi.fastutil.ints.IntList;
|
||||||
|
|
||||||
public final class JsonUtils {
|
public final class JsonUtils {
|
||||||
@ -102,4 +104,12 @@ public final class JsonUtils {
|
|||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public static <T> T decode(String jsonData, Type type) {
|
||||||
|
try {
|
||||||
|
return gson.fromJson(jsonData, type);
|
||||||
|
} catch (Exception ignored) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
596
src/main/java/emu/grasscutter/utils/TsvUtils.java
Normal file
596
src/main/java/emu/grasscutter/utils/TsvUtils.java
Normal file
@ -0,0 +1,596 @@
|
|||||||
|
package emu.grasscutter.utils;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.lang.reflect.Array;
|
||||||
|
import java.lang.reflect.Constructor;
|
||||||
|
import java.lang.reflect.Field;
|
||||||
|
import java.lang.reflect.ParameterizedType;
|
||||||
|
import java.lang.reflect.Type;
|
||||||
|
import java.nio.charset.StandardCharsets;
|
||||||
|
import java.nio.file.Files;
|
||||||
|
import java.nio.file.Path;
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.Arrays;
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
|
import java.util.Set;
|
||||||
|
import java.util.SortedMap;
|
||||||
|
import java.util.TreeMap;
|
||||||
|
import java.util.function.Function;
|
||||||
|
import java.util.stream.IntStream;
|
||||||
|
import java.util.stream.Stream;
|
||||||
|
|
||||||
|
import com.google.gson.JsonArray;
|
||||||
|
import com.google.gson.JsonElement;
|
||||||
|
import com.google.gson.JsonNull;
|
||||||
|
import com.google.gson.JsonObject;
|
||||||
|
import com.google.gson.JsonPrimitive;
|
||||||
|
import com.google.gson.annotations.SerializedName;
|
||||||
|
|
||||||
|
import emu.grasscutter.Grasscutter;
|
||||||
|
import it.unimi.dsi.fastutil.Pair;
|
||||||
|
import it.unimi.dsi.fastutil.ints.Int2ObjectRBTreeMap;
|
||||||
|
import it.unimi.dsi.fastutil.ints.Int2ObjectSortedMap;
|
||||||
|
import it.unimi.dsi.fastutil.objects.Object2IntArrayMap;
|
||||||
|
import lombok.val;
|
||||||
|
|
||||||
|
import static emu.grasscutter.utils.Utils.nonRegexSplit;
|
||||||
|
|
||||||
|
// Throughout this file, commented System.out.println debug log calls are left in.
|
||||||
|
// This is because the default logger will deadlock when operating on parallel streams.
|
||||||
|
public class TsvUtils {
|
||||||
|
private static final Map<Type, Object> defaultValues = Map.ofEntries(
|
||||||
|
// Map.entry(String.class, null), // builder hates null values
|
||||||
|
Map.entry(Integer.class, 0),
|
||||||
|
Map.entry(int.class, 0),
|
||||||
|
Map.entry(Long.class, 0L),
|
||||||
|
Map.entry(long.class, 0L),
|
||||||
|
Map.entry(Float.class, 0f),
|
||||||
|
Map.entry(float.class, 0f),
|
||||||
|
Map.entry(Double.class, 0d),
|
||||||
|
Map.entry(double.class, 0d),
|
||||||
|
Map.entry(Boolean.class, false),
|
||||||
|
Map.entry(boolean.class, false)
|
||||||
|
);
|
||||||
|
private static final Set<Type> primitiveTypes = Set.of(String.class, Integer.class, int.class, Long.class, long.class, Float.class, float.class, Double.class, double.class, Boolean.class, boolean.class);
|
||||||
|
|
||||||
|
private static final Function<String, Object> parseString = value -> value;
|
||||||
|
private static final Function<String, Object> parseInt = value -> (int) Double.parseDouble(value); //Integer::parseInt;
|
||||||
|
private static final Function<String, Object> parseLong = value -> (long) Double.parseDouble(value); //Long::parseLong;
|
||||||
|
private static Map<Type, Function<String, Object>> primitiveTypeParsers = Map.ofEntries(
|
||||||
|
Map.entry(String.class, parseString),
|
||||||
|
Map.entry(Integer.class, parseInt),
|
||||||
|
Map.entry(int.class, parseInt),
|
||||||
|
Map.entry(Long.class, parseLong),
|
||||||
|
Map.entry(long.class, parseLong),
|
||||||
|
Map.entry(Float.class, Float::parseFloat),
|
||||||
|
Map.entry(float.class, Float::parseFloat),
|
||||||
|
Map.entry(Double.class, Double::parseDouble),
|
||||||
|
Map.entry(double.class, Double::parseDouble),
|
||||||
|
Map.entry(Boolean.class, Boolean::parseBoolean),
|
||||||
|
Map.entry(boolean.class, Boolean::parseBoolean)
|
||||||
|
);
|
||||||
|
private static final Map<Type, Function<String, Object>> typeParsers = new HashMap<>(primitiveTypeParsers);
|
||||||
|
|
||||||
|
@SuppressWarnings("unchecked")
|
||||||
|
private static <T> T parsePrimitive(Class<T> type, String string) {
|
||||||
|
if (string == null || string.isEmpty()) return (T) defaultValues.get(type);
|
||||||
|
return (T) primitiveTypeParsers.get(type).apply(string);
|
||||||
|
}
|
||||||
|
// This is more expensive than parsing as the correct types, but it is more tolerant of mismatched data like ints with .0
|
||||||
|
private static double parseNumber(String string) {
|
||||||
|
if (string == null || string.isEmpty()) return 0d;
|
||||||
|
return Double.parseDouble(string);
|
||||||
|
}
|
||||||
|
@SuppressWarnings("unchecked")
|
||||||
|
private static <T> T parseEnum(Class<T> enumType, String string) {
|
||||||
|
if (string == null || string.isEmpty()) return null;
|
||||||
|
return (T) getEnumTypeParser(enumType).apply(string);
|
||||||
|
}
|
||||||
|
|
||||||
|
// This is idiotic. I hate it. I'll have to look into how Gson beats the JVM into submission over classes where reflection magically fails to find the NoArgsConstructor later.
|
||||||
|
public static <T> T newObj(Class<T> objClass) {
|
||||||
|
try {
|
||||||
|
return objClass.getDeclaredConstructor().newInstance();
|
||||||
|
} catch (Exception ignored) {
|
||||||
|
return JsonUtils.decode("{}", objClass);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private static final Map<Class<?>, Function<String, Object>> enumTypeParsers = new HashMap<>();
|
||||||
|
@SuppressWarnings("deprecated") // Field::isAccessible is deprecated because it doesn't do what people think it does. It does what we want it to, however.
|
||||||
|
private static Function<String, Object> makeEnumTypeParser(Class<?> enumClass) {
|
||||||
|
if (!enumClass.isEnum()) {
|
||||||
|
// System.out.println("Called makeEnumTypeParser with non-enum enumClass "+enumClass);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Make mappings of (string) names to enum constants
|
||||||
|
val map = new HashMap<String, Object>();
|
||||||
|
val enumConstants = enumClass.getEnumConstants();
|
||||||
|
for (val constant : enumConstants)
|
||||||
|
map.put(constant.toString(), constant);
|
||||||
|
|
||||||
|
// If the enum also has a numeric value, map those to the constants too
|
||||||
|
// System.out.println("Looking for enum value field");
|
||||||
|
for (Field f : enumClass.getDeclaredFields()) {
|
||||||
|
if (switch (f.getName()) {case "value", "id" -> true; default -> false;}) {
|
||||||
|
// System.out.println("Enum value field found - " + f.getName());
|
||||||
|
boolean acc = f.isAccessible();
|
||||||
|
f.setAccessible(true);
|
||||||
|
try {
|
||||||
|
for (val constant : enumConstants)
|
||||||
|
map.put(String.valueOf(f.getInt(constant)), constant);
|
||||||
|
} catch (IllegalAccessException e) {
|
||||||
|
// System.out.println("Failed to access enum id field.");
|
||||||
|
}
|
||||||
|
f.setAccessible(acc);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return map::get;
|
||||||
|
}
|
||||||
|
private static synchronized Function<String, Object> getEnumTypeParser(Class<?> enumType) {
|
||||||
|
if (enumType == null) {
|
||||||
|
// System.out.println("Called getEnumTypeParser with null enumType");
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
return enumTypeParsers.computeIfAbsent(enumType, TsvUtils::makeEnumTypeParser);
|
||||||
|
}
|
||||||
|
|
||||||
|
private static synchronized Function<String, Object> getTypeParser(Type type) {
|
||||||
|
if (type == null) return parseString;
|
||||||
|
return typeParsers.computeIfAbsent(type, t -> value -> JsonUtils.decode(value, t));
|
||||||
|
}
|
||||||
|
|
||||||
|
private static Type class2Type(Class<?> classType) {
|
||||||
|
return (Type) classType.getGenericSuperclass();
|
||||||
|
}
|
||||||
|
private static Class<?> type2Class(Type type) {
|
||||||
|
if (type instanceof Class) {
|
||||||
|
return (Class<?>) type;
|
||||||
|
} else if (type instanceof ParameterizedType) {
|
||||||
|
return (Class<?>) ((ParameterizedType) type).getRawType();
|
||||||
|
} else {
|
||||||
|
return type.getClass(); // Probably incorrect
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// A helper object that contains a Field and the function to parse a String to create the value for the Field.
|
||||||
|
private static class FieldParser {
|
||||||
|
public final Field field;
|
||||||
|
public final Type type;
|
||||||
|
public final Class<?> classType;
|
||||||
|
public final Function<String, Object> parser;
|
||||||
|
|
||||||
|
FieldParser(Field field) {
|
||||||
|
this.field = field;
|
||||||
|
this.type = field.getGenericType(); // returns specialized type info e.g. java.util.List<java.lang.Integer>
|
||||||
|
this.classType = field.getType();
|
||||||
|
this.parser = getTypeParser(this.type);
|
||||||
|
}
|
||||||
|
|
||||||
|
public Object parse(String token) {
|
||||||
|
return this.parser.apply(token);
|
||||||
|
}
|
||||||
|
|
||||||
|
public void parse(Object obj, String token) throws IllegalAccessException {
|
||||||
|
this.field.set(obj, this.parser.apply(token));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private static Map<String, FieldParser> makeClassFieldMap(Class<?> classType) {
|
||||||
|
val fieldMap = new HashMap<String, FieldParser>();
|
||||||
|
for (Field field : classType.getDeclaredFields()) {
|
||||||
|
field.setAccessible(true); // Yes, we don't bother setting this back. No, it doesn't matter for this project.
|
||||||
|
val fieldParser = new FieldParser(field);
|
||||||
|
|
||||||
|
val a = field.getDeclaredAnnotation(SerializedName.class);
|
||||||
|
if (a == null) { // No annotation, use raw field name
|
||||||
|
fieldMap.put(field.getName(), fieldParser);
|
||||||
|
} else { // Handle SerializedNames and alternatives
|
||||||
|
fieldMap.put(a.value(), fieldParser);
|
||||||
|
for (val alt : a.alternate()) {
|
||||||
|
fieldMap.put(alt, fieldParser);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return fieldMap;
|
||||||
|
}
|
||||||
|
|
||||||
|
private static Map<Class<?>, Map<String, FieldParser>> cachedClassFieldMaps = new HashMap<>();
|
||||||
|
private static synchronized Map<String, FieldParser> getClassFieldMap(Class<?> classType) {
|
||||||
|
return cachedClassFieldMaps.computeIfAbsent(classType, TsvUtils::makeClassFieldMap);
|
||||||
|
}
|
||||||
|
|
||||||
|
private static class StringTree {
|
||||||
|
public final Map<String, StringTree> children = new TreeMap<>();
|
||||||
|
|
||||||
|
public void addPath(String path) {
|
||||||
|
if (path.isEmpty()) return;
|
||||||
|
|
||||||
|
val firstDot = path.indexOf('.');
|
||||||
|
val fieldPath = (firstDot < 0) ? path : path.substring(0, firstDot);
|
||||||
|
val remainder = (firstDot < 0) ? "" : path.substring(firstDot+1);
|
||||||
|
this.children.computeIfAbsent(fieldPath, k -> new StringTree()).addPath(remainder);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@SuppressWarnings("unchecked")
|
||||||
|
private static class StringValueTree {
|
||||||
|
public final SortedMap<String, StringValueTree> children = new TreeMap<>();
|
||||||
|
public final Int2ObjectSortedMap<StringValueTree> arrayChildren = new Int2ObjectRBTreeMap<>();
|
||||||
|
public String value;
|
||||||
|
|
||||||
|
public StringValueTree(StringTree from) {
|
||||||
|
from.children.forEach((k,v) -> {
|
||||||
|
try {
|
||||||
|
this.arrayChildren.put(Integer.parseInt(k), new StringValueTree(v));
|
||||||
|
} catch (NumberFormatException e) {
|
||||||
|
this.children.put(k, new StringValueTree(v));
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setValue(String path, String value) {
|
||||||
|
if (path.isEmpty()) {
|
||||||
|
this.value = value;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
val firstDot = path.indexOf('.');
|
||||||
|
val fieldPath = (firstDot < 0) ? path : path.substring(0, firstDot);
|
||||||
|
val remainder = (firstDot < 0) ? "" : path.substring(firstDot+1);
|
||||||
|
try {
|
||||||
|
this.arrayChildren.get(Integer.parseInt(fieldPath)).setValue(remainder, value);
|
||||||
|
} catch (NumberFormatException e) {
|
||||||
|
this.children.get(fieldPath).setValue(remainder, value);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public JsonElement toJson() {
|
||||||
|
// Determine if this is an object, an array, or a value
|
||||||
|
if (this.value != null) { //
|
||||||
|
return new JsonPrimitive(this.value);
|
||||||
|
}
|
||||||
|
if (!this.arrayChildren.isEmpty()) {
|
||||||
|
val arr = new JsonArray(this.arrayChildren.lastIntKey()+1);
|
||||||
|
arrayChildren.forEach((k,v) -> arr.set(k, v.toJson()));
|
||||||
|
return arr;
|
||||||
|
} else if (this.children.isEmpty()) {
|
||||||
|
return JsonNull.INSTANCE;
|
||||||
|
} else {
|
||||||
|
val obj = new JsonObject();
|
||||||
|
children.forEach((k,v) -> {
|
||||||
|
val j = v.toJson();
|
||||||
|
if (j != JsonNull.INSTANCE)
|
||||||
|
obj.add(k, v.toJson());
|
||||||
|
});
|
||||||
|
return obj;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public <T> T toClass(Class<T> classType, Type type) {
|
||||||
|
// System.out.println("toClass called with Class: "+classType+" \tType: "+type);
|
||||||
|
if (type == null)
|
||||||
|
type = class2Type(classType);
|
||||||
|
|
||||||
|
if (primitiveTypeParsers.containsKey(classType)) {
|
||||||
|
return parsePrimitive(classType, this.value);
|
||||||
|
} else if (classType.isEnum()) {
|
||||||
|
return parseEnum(classType, this.value);
|
||||||
|
} else if (classType.isArray()) {
|
||||||
|
return this.toArray(classType);
|
||||||
|
} else if (List.class.isAssignableFrom(classType)) {
|
||||||
|
// if (type instanceof ParameterizedType)
|
||||||
|
val elementType = ((ParameterizedType) type).getActualTypeArguments()[0];
|
||||||
|
return (T) this.toList(type2Class(elementType), elementType);
|
||||||
|
} else if (Map.class.isAssignableFrom(classType)) {
|
||||||
|
// System.out.println("Class: "+classType+" \tClassTypeParams: "+Arrays.toString(classType.getTypeParameters())+" \tType: "+type+" \tTypeArguments: "+Arrays.toString(((ParameterizedType) type).getActualTypeArguments()));
|
||||||
|
// if (type instanceof ParameterizedType)
|
||||||
|
val keyType = ((ParameterizedType) type).getActualTypeArguments()[0];
|
||||||
|
val valueType = ((ParameterizedType) type).getActualTypeArguments()[1];
|
||||||
|
return (T) this.toMap(type2Class(keyType), type2Class(valueType), valueType);
|
||||||
|
} else {
|
||||||
|
return this.toObj(classType, type);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private <T> T toObj(Class<T> objClass, Type objType) {
|
||||||
|
try {
|
||||||
|
// val obj = objClass.getDeclaredConstructor().newInstance();
|
||||||
|
val obj = newObj(objClass);
|
||||||
|
val fieldMap = getClassFieldMap(objClass);
|
||||||
|
this.children.forEach((name, tree) -> {
|
||||||
|
val field = fieldMap.get(name);
|
||||||
|
if (field == null) return;
|
||||||
|
try {
|
||||||
|
if (primitiveTypes.contains(field.type)) {
|
||||||
|
if ((tree.value != null) && !tree.value.isEmpty())
|
||||||
|
field.parse(obj, tree.value);
|
||||||
|
} else {
|
||||||
|
val value = tree.toClass(field.classType, field.type);
|
||||||
|
// System.out.println("Setting field "+name+" to "+value);
|
||||||
|
field.field.set(obj, value);
|
||||||
|
// field.field.set(obj, tree.toClass(field.classType, field.type));
|
||||||
|
}
|
||||||
|
} catch (Exception e) {
|
||||||
|
// System.out.println("Exception while setting field "+name+" for class "+objClass+" - "+e);
|
||||||
|
Grasscutter.getLogger().error("Exception while setting field "+name+" ("+field.classType+")"+" for class "+objClass+" - ",e);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
return obj;
|
||||||
|
} catch (Exception e) {
|
||||||
|
// System.out.println("Exception while creating object of class "+objClass+" - "+e);
|
||||||
|
Grasscutter.getLogger().error("Exception while creating object of class "+objClass+" - ",e);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public <T> T toArray(Class<T> classType) {
|
||||||
|
// Primitives don't play so nice with generics, so we handle all of them individually.
|
||||||
|
val containedClass = classType.getComponentType();
|
||||||
|
// val arraySize = this.arrayChildren.size(); // Assume dense 0-indexed
|
||||||
|
val arraySize = this.arrayChildren.lastIntKey()+1; // Could be sparse!
|
||||||
|
// System.out.println("toArray called with Class: "+classType+" \tContains: "+containedClass+" \tof size: "+arraySize);
|
||||||
|
if (containedClass == int.class) {
|
||||||
|
val output = new int[arraySize];
|
||||||
|
this.arrayChildren.forEach((idx, tree) -> output[idx] = (int) parseNumber(tree.value));
|
||||||
|
return (T) output;
|
||||||
|
} else if (containedClass == long.class) {
|
||||||
|
val output = new long[arraySize];
|
||||||
|
this.arrayChildren.forEach((idx, tree) -> output[idx] = (long) parseNumber(tree.value));
|
||||||
|
return (T) output;
|
||||||
|
} else if (containedClass == float.class) {
|
||||||
|
val output = new float[arraySize];
|
||||||
|
this.arrayChildren.forEach((idx, tree) -> output[idx] = (float) parseNumber(tree.value));
|
||||||
|
return (T) output;
|
||||||
|
} else if (containedClass == double.class) {
|
||||||
|
val output = new double[arraySize];
|
||||||
|
this.arrayChildren.forEach((idx, tree) -> output[idx] = (double) parseNumber(tree.value));
|
||||||
|
return (T) output;
|
||||||
|
} else if (containedClass == byte.class) {
|
||||||
|
val output = new byte[arraySize];
|
||||||
|
this.arrayChildren.forEach((idx, tree) -> output[idx] = (byte) parseNumber(tree.value));
|
||||||
|
return (T) output;
|
||||||
|
} else if (containedClass == char.class) {
|
||||||
|
val output = new char[arraySize];
|
||||||
|
this.arrayChildren.forEach((idx, tree) -> output[idx] = (char) parseNumber(tree.value));
|
||||||
|
return (T) output;
|
||||||
|
} else if (containedClass == short.class) {
|
||||||
|
val output = new short[arraySize];
|
||||||
|
this.arrayChildren.forEach((idx, tree) -> output[idx] = (short) parseNumber(tree.value));
|
||||||
|
return (T) output;
|
||||||
|
} else if (containedClass == boolean.class) {
|
||||||
|
val output = new boolean[arraySize];
|
||||||
|
this.arrayChildren.forEach((idx, tree) -> {
|
||||||
|
val value = ((tree.value == null) || tree.value.isEmpty()) ? false : Boolean.parseBoolean(tree.value);
|
||||||
|
output[idx] = value;
|
||||||
|
});
|
||||||
|
return (T) output;
|
||||||
|
} else {
|
||||||
|
val output = Array.newInstance(containedClass, arraySize);
|
||||||
|
this.arrayChildren.forEach((idx, tree) -> ((Object[]) output)[idx] = tree.toClass(containedClass, null));
|
||||||
|
return (T) output;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private <E> List<E> toList(Class<E> valueClass, Type valueType) {
|
||||||
|
val arraySize = this.arrayChildren.lastIntKey()+1; // Could be sparse!
|
||||||
|
// System.out.println("toList called with valueClass: "+valueClass+" \tvalueType: "+valueType+" \tof size: "+arraySize);
|
||||||
|
val list = new ArrayList<E>(arraySize);
|
||||||
|
// Safe sparse version
|
||||||
|
for (int i = 0; i < arraySize; i++)
|
||||||
|
list.add(null);
|
||||||
|
this.arrayChildren.forEach((idx, tree) -> list.set(idx, tree.toClass(valueClass, valueType)));
|
||||||
|
return list;
|
||||||
|
}
|
||||||
|
|
||||||
|
private <K,V> Map<K,V> toMap(Class<K> keyClass, Class<V> valueClass, Type valueType) {
|
||||||
|
val map = new HashMap<K,V>();
|
||||||
|
val keyParser = getTypeParser(keyClass);
|
||||||
|
this.children.forEach((key, tree) -> {
|
||||||
|
if ((key != null) && !key.isEmpty())
|
||||||
|
map.put((K) keyParser.apply(key), tree.toClass(valueClass, valueType));
|
||||||
|
});
|
||||||
|
return map;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Flat tab-separated value tables.
|
||||||
|
// Arrays are represented as arrayName.0, arrayName.1, etc. columns.
|
||||||
|
// Maps/POJOs are represented as objName.fieldOneName, objName.fieldTwoName, etc. columns.
|
||||||
|
// This is currently about 25x as slow as TSJ and Gson parsers, likely due to the tree spam.
|
||||||
|
public static <T> List<T> loadTsvToListSetField(Class<T> classType, Path filename) {
|
||||||
|
try (val fileReader = Files.newBufferedReader(filename, StandardCharsets.UTF_8)) {
|
||||||
|
// val fieldMap = getClassFieldMap(classType);
|
||||||
|
// val constructor = classType.getDeclaredConstructor();
|
||||||
|
|
||||||
|
val headerNames = nonRegexSplit(fileReader.readLine(), '\t');
|
||||||
|
val columns = headerNames.size();
|
||||||
|
// If we just crawled through all fields to expand potential subobjects, we might hit recursive data structure explosions (e.g. if something has a Player object)
|
||||||
|
// So we'll only crawl through objects referenced by the header columns
|
||||||
|
val stringTree = new StringTree();
|
||||||
|
headerNames.forEach(stringTree::addPath);
|
||||||
|
|
||||||
|
return fileReader.lines().parallel().map(line -> {
|
||||||
|
// return fileReader.lines().map(line -> {
|
||||||
|
// System.out.println("Processing line of "+filename+" - "+line);
|
||||||
|
val tokens = nonRegexSplit(line, '\t');
|
||||||
|
val m = Math.min(tokens.size(), columns);
|
||||||
|
int t = 0;
|
||||||
|
StringValueTree tree = new StringValueTree(stringTree);
|
||||||
|
try {
|
||||||
|
for (t = 0; t < m; t++) {
|
||||||
|
String token = tokens.get(t);
|
||||||
|
if (!token.isEmpty()) {
|
||||||
|
tree.setValue(headerNames.get(t), token);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// return JsonUtils.decode(tree.toJson(), classType);
|
||||||
|
return tree.toClass(classType, null);
|
||||||
|
} catch (Exception e) {
|
||||||
|
Grasscutter.getLogger().warn("Error deserializing an instance of class "+classType.getCanonicalName());
|
||||||
|
Grasscutter.getLogger().warn("At token #"+t+" of #"+m);
|
||||||
|
Grasscutter.getLogger().warn("Header names are: "+headerNames.toString());
|
||||||
|
Grasscutter.getLogger().warn("Tokens are: "+tokens.toString());
|
||||||
|
Grasscutter.getLogger().warn("Stacktrace is: ", e);
|
||||||
|
// System.out.println("Error deserializing an instance of class "+classType.getCanonicalName());
|
||||||
|
// System.out.println("At token #"+t+" of #"+m);
|
||||||
|
// System.out.println("Header names are: "+headerNames.toString());
|
||||||
|
// System.out.println("Tokens are: "+tokens.toString());
|
||||||
|
// System.out.println("Json is: "+tree.toJson().toString());
|
||||||
|
// System.out.println("Stacktrace is: "+ e);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
}).toList();
|
||||||
|
} catch (Exception e) {
|
||||||
|
Grasscutter.getLogger().error("Error loading TSV file '"+filename+"' - Stacktrace is: ", e);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// This uses a hybrid format where columns can hold JSON-encoded values.
|
||||||
|
// I'll term it TSJ (tab-separated JSON) for now, it has convenient properties.
|
||||||
|
public static <T> List<T> loadTsjToListSetField(Class<T> classType, Path filename) {
|
||||||
|
try (val fileReader = Files.newBufferedReader(filename, StandardCharsets.UTF_8)) {
|
||||||
|
val fieldMap = getClassFieldMap(classType);
|
||||||
|
val constructor = classType.getDeclaredConstructor();
|
||||||
|
|
||||||
|
val headerNames = nonRegexSplit(fileReader.readLine(), '\t');
|
||||||
|
val columns = headerNames.size();
|
||||||
|
val fieldParsers = headerNames.stream().map(fieldMap::get).toList();
|
||||||
|
|
||||||
|
return fileReader.lines().parallel().map(line -> {
|
||||||
|
val tokens = nonRegexSplit(line, '\t');
|
||||||
|
val m = Math.min(tokens.size(), columns);
|
||||||
|
int t = 0;
|
||||||
|
try {
|
||||||
|
T obj = constructor.newInstance();
|
||||||
|
for (t = 0; t < m; t++) {
|
||||||
|
val fieldParser = fieldParsers.get(t);
|
||||||
|
if (fieldParser == null) continue;
|
||||||
|
|
||||||
|
String token = tokens.get(t);
|
||||||
|
if (!token.isEmpty()) {
|
||||||
|
fieldParser.parse(obj, token);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return obj;
|
||||||
|
} catch (Exception e) {
|
||||||
|
Grasscutter.getLogger().warn("Error deserializing an instance of class "+classType.getCanonicalName());
|
||||||
|
Grasscutter.getLogger().warn("At token #"+t+" of #"+m);
|
||||||
|
Grasscutter.getLogger().warn("Header names are: "+headerNames.toString());
|
||||||
|
Grasscutter.getLogger().warn("Tokens are: "+tokens.toString());
|
||||||
|
Grasscutter.getLogger().warn("Stacktrace is: ", e);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
}).toList();
|
||||||
|
} catch (IOException e) {
|
||||||
|
Grasscutter.getLogger().error("Error loading TSV file '"+filename+"' - Stacktrace is: ", e);
|
||||||
|
return null;
|
||||||
|
} catch (NoSuchMethodException e) {
|
||||||
|
Grasscutter.getLogger().error("Error loading TSV file '"+filename+"' - Class is missing NoArgsConstructor");
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
// -----------------------------------------------------------------
|
||||||
|
// Everything below here is for the AllArgsConstructor TSJ parser
|
||||||
|
// -----------------------------------------------------------------
|
||||||
|
// Sadly, this is a little bit slower than the SetField version.
|
||||||
|
// I've left it in as an example of an optimization attempt that didn't work out, since the naive reflection version will tempt people to try things like this.
|
||||||
|
@SuppressWarnings("unchecked")
|
||||||
|
private static <T> Pair<Constructor<T>, String[]> getAllArgsConstructor(Class<T> classType) {
|
||||||
|
for (var c : classType.getDeclaredConstructors()) {
|
||||||
|
val consParameters = (java.beans.ConstructorProperties) c.getAnnotation(java.beans.ConstructorProperties.class);
|
||||||
|
if (consParameters != null) {
|
||||||
|
return Pair.of((Constructor<T>) c, consParameters.value());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
public static <T> List<List<T>> loadTsjsToListsConstructor(Class<T> classType, Path... filenames) throws Exception {
|
||||||
|
val pair = getAllArgsConstructor(classType);
|
||||||
|
if (pair == null) {
|
||||||
|
Grasscutter.getLogger().error("No AllArgsContructor found for class: "+classType);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
val constructor = pair.left();
|
||||||
|
val conArgNames = pair.right();
|
||||||
|
val numArgs = constructor.getParameterCount();
|
||||||
|
|
||||||
|
val argMap = new Object2IntArrayMap<String>();
|
||||||
|
for (int i = 0; i < conArgNames.length; i++) {
|
||||||
|
argMap.put(conArgNames[i], i);
|
||||||
|
}
|
||||||
|
|
||||||
|
val argTypes = new Type[numArgs]; // constructor.getParameterTypes() returns base types like java.util.List instead of java.util.List<java.lang.Integer>
|
||||||
|
for (Field field : classType.getDeclaredFields()) {
|
||||||
|
int index = argMap.getOrDefault(field.getName(), -1);
|
||||||
|
if (index < 0) continue;
|
||||||
|
|
||||||
|
argTypes[index] = field.getGenericType(); // returns specialized type info e.g. java.util.List<java.lang.Integer>
|
||||||
|
|
||||||
|
val a = field.getDeclaredAnnotation(SerializedName.class);
|
||||||
|
if (a != null) { // Handle SerializedNames and alternatives
|
||||||
|
argMap.put(a.value(), index);
|
||||||
|
for (val alt : a.alternate()) {
|
||||||
|
argMap.put(alt, index);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
val argParsers = Stream.of(argTypes).map(TsvUtils::getTypeParser).toList();
|
||||||
|
|
||||||
|
val defaultArgs = new Object[numArgs];
|
||||||
|
for (int i = 0; i < numArgs; i++) {
|
||||||
|
defaultArgs[i] = defaultValues.get(argTypes[i]);
|
||||||
|
}
|
||||||
|
|
||||||
|
return Stream.of(filenames).parallel().map(filename -> {
|
||||||
|
try (val fileReader = Files.newBufferedReader(filename, StandardCharsets.UTF_8)) {
|
||||||
|
val headerNames = nonRegexSplit(fileReader.readLine(), '\t');
|
||||||
|
val columns = headerNames.size();
|
||||||
|
val argPositions = headerNames.stream().mapToInt(name -> argMap.getOrDefault(name, -1)).toArray();
|
||||||
|
|
||||||
|
return fileReader.lines().parallel().map(line -> {
|
||||||
|
val tokens = nonRegexSplit(line, '\t');
|
||||||
|
val args = defaultArgs.clone();
|
||||||
|
val m = Math.min(tokens.size(), columns);
|
||||||
|
int t = 0;
|
||||||
|
try {
|
||||||
|
for (t = 0; t < m; t++) {
|
||||||
|
val argIndex = argPositions[t];
|
||||||
|
if (argIndex < 0) continue;
|
||||||
|
|
||||||
|
String token = tokens.get(t);
|
||||||
|
if (!token.isEmpty()) {
|
||||||
|
args[argIndex] = argParsers.get(argIndex).apply(token);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return (T) constructor.newInstance(args);
|
||||||
|
} catch (Exception e) {
|
||||||
|
Grasscutter.getLogger().warn("Error deserializing an instance of class "+classType.getCanonicalName()+" : "+constructor.getName());
|
||||||
|
Grasscutter.getLogger().warn("At token #"+t+" of #"+m);
|
||||||
|
Grasscutter.getLogger().warn("Arg names are: "+Arrays.toString(conArgNames));
|
||||||
|
Grasscutter.getLogger().warn("Arg types are: "+Arrays.toString(argTypes));
|
||||||
|
Grasscutter.getLogger().warn("Default Args are: "+Arrays.toString(defaultArgs));
|
||||||
|
Grasscutter.getLogger().warn("Args are: "+Arrays.toString(args));
|
||||||
|
Grasscutter.getLogger().warn("Header names are: "+headerNames.toString());
|
||||||
|
Grasscutter.getLogger().warn("Header types are: "+IntStream.of(argPositions).mapToObj(i -> (i >= 0) ? argTypes[i] : null).toList());
|
||||||
|
Grasscutter.getLogger().warn("Tokens are: "+tokens.toString());
|
||||||
|
Grasscutter.getLogger().warn("Stacktrace is: ", e);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
}).toList();
|
||||||
|
} catch (IOException e) {
|
||||||
|
Grasscutter.getLogger().error("Error loading TSV file '"+filename+"' - Stacktrace is: ", e);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
}).toList();
|
||||||
|
}
|
||||||
|
}
|
@ -395,4 +395,22 @@ public final class Utils {
|
|||||||
public static <T> T drawRandomListElement(List<T> list) {
|
public static <T> T drawRandomListElement(List<T> list) {
|
||||||
return drawRandomListElement(list, null);
|
return drawRandomListElement(list, null);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/***
|
||||||
|
* Splits a string by a character, into a list
|
||||||
|
* @param input The string to split
|
||||||
|
* @param separator The character to use as the split points
|
||||||
|
* @return A list of all the substrings
|
||||||
|
*/
|
||||||
|
public static List<String> nonRegexSplit(String input, int separator) {
|
||||||
|
var output = new ArrayList<String>();
|
||||||
|
int start = 0;
|
||||||
|
for (int next = input.indexOf(separator); next > 0; next = input.indexOf(separator, start)) {
|
||||||
|
output.add(input.substring(start, next));
|
||||||
|
start = next + 1;
|
||||||
|
}
|
||||||
|
if (start < input.length())
|
||||||
|
output.add(input.substring(start));
|
||||||
|
return output;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user