UnitGroupImport.java

package org.openlca.jsonld.input;

import java.util.ArrayList;
import java.util.List;

import org.openlca.core.model.FlowProperty;
import org.openlca.core.model.ModelType;
import org.openlca.core.model.Unit;
import org.openlca.core.model.UnitGroup;

import com.google.common.base.Joiner;
import com.google.gson.JsonArray;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;

class UnitGroupImport extends BaseImport<UnitGroup> {

	private UnitGroupImport(String refId, ImportConfig conf) {
		super(ModelType.UNIT_GROUP, refId, conf);
	}

	static UnitGroup run(String refId, ImportConfig conf) {
		return new UnitGroupImport(refId, conf).run();
	}

	@Override
	UnitGroup map(JsonObject json, long id) {
		if (json == null)
			return null;
		UnitGroup g = new UnitGroup();
		In.mapAtts(json, g, id, conf);
		addUnits(g, json);
		// insert the unit group before a default flow property is imported
		// to avoid endless import cycles
		g = conf.db.put(g);
		return setDefaultProperty(json, g);
	}

	private UnitGroup setDefaultProperty(JsonObject json, UnitGroup g) {
		String propId = In.getRefId(json, "defaultFlowProperty");
		if (propId == null)
			return g;
		FlowProperty prop = FlowPropertyImport.run(propId, conf);
		g.setDefaultFlowProperty(prop);
		return conf.db.update(g);
	}

	private void addUnits(UnitGroup g, JsonObject json) {
		JsonArray array = In.getArray(json, "units");
		if (array == null || array.size() == 0)
			return;
		for (JsonElement e : array) {
			if (!e.isJsonObject())
				continue;
			JsonObject obj = e.getAsJsonObject();
			Unit unit = mapUnit(obj);
			boolean refUnit = In.getBool(obj, "referenceUnit", false);
			if (refUnit)
				g.setReferenceUnit(unit);
			g.getUnits().add(unit);
		}
	}

	private Unit mapUnit(JsonObject json) {
		Unit unit = new Unit();
		In.mapAtts(json, unit, 0);
		unit.setConversionFactor(In.getDouble(json, "conversionFactor", 1.0));
		addSynonyms(unit, json);
		return unit;
	}

	private void addSynonyms(Unit unit, JsonObject json) {
		JsonArray array = In.getArray(json, "synonyms");
		if (array == null || array.size() == 0)
			return;
		List<String> synonyms = new ArrayList<>();
		for (JsonElement e : array) {
			if (!e.isJsonPrimitive())
				continue;
			synonyms.add(e.getAsString());
		}
		String synStr = Joiner.on(';').join(synonyms);
		unit.setSynonyms(synStr);
	}
}