60 lines
1.7 KiB
Java
60 lines
1.7 KiB
Java
package org.warp.jcwdb;
|
|
|
|
import it.unimi.dsi.fastutil.ints.Int2ObjectMap;
|
|
import it.unimi.dsi.fastutil.ints.Int2ObjectOpenHashMap;
|
|
import it.unimi.dsi.fastutil.objects.Object2ObjectMap;
|
|
import it.unimi.dsi.fastutil.objects.Object2ObjectOpenHashMap;
|
|
|
|
public class TypesManager {
|
|
private final Int2ObjectMap<DBTypeParser<?>> types;
|
|
private final Object2ObjectMap<Class<?>, DBTypeParser<?>> typesByClass;
|
|
private DBTypedObjectParser<?> fallbackParser;
|
|
|
|
public TypesManager(JCWDatabase db) {
|
|
types = new Int2ObjectOpenHashMap<>();
|
|
typesByClass = new Object2ObjectOpenHashMap<>();
|
|
DBStandardTypes.registerStandardTypes(db, this);
|
|
}
|
|
|
|
public <T> void registerType(Class<T> clazz, int type, DBTypeParser<T> parser) {
|
|
this.types.put(type, parser);
|
|
this.typesByClass.put(clazz, parser);
|
|
}
|
|
|
|
/**
|
|
* Use this method with the most used classes to save disk space.
|
|
* @param clazz
|
|
* @param id
|
|
* @param <T>
|
|
*/
|
|
public <T> void registerGenericClass(Class<T> clazz, int id) {
|
|
this.fallbackParser.registerClass(clazz, id);
|
|
}
|
|
|
|
public void registerTypeFallback(DBTypedObjectParser<?> parser) {
|
|
this.fallbackParser = parser;
|
|
}
|
|
|
|
public <T> DBTypeParser<T> get(int type) {
|
|
if (types.containsKey(type) == false) {
|
|
if (fallbackParser == null) {
|
|
throw new NoParserFoundException("The type " + type + " can't be parsed.");
|
|
} else {
|
|
return fallbackParser.cast();
|
|
}
|
|
}
|
|
return types.get(type).cast();
|
|
}
|
|
|
|
public <T> DBTypeParser<T> get(Class<T> type) {
|
|
if (typesByClass.containsKey(type) == false) {
|
|
if (fallbackParser == null) {
|
|
throw new NoParserFoundException("The class " + type.getSimpleName() + " can't be parsed.");
|
|
} else {
|
|
return fallbackParser.cast();
|
|
}
|
|
}
|
|
return typesByClass.get(type).cast();
|
|
}
|
|
}
|