org.openlca.io.olca.RefSwitcher Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of olca-io Show documentation
The newest version!
package org.openlca.io.olca;

import org.openlca.core.database.CategoryDao;
import org.openlca.core.database.IDatabase;
import org.openlca.core.database.ImpactCategoryDao;
import org.openlca.core.database.NwSetDao;
import org.openlca.core.database.ProcessDao;
import org.openlca.core.database.RefEntityDao;
import org.openlca.core.database.UnitDao;
import org.openlca.core.model.Category;
import org.openlca.core.model.Flow;
import org.openlca.core.model.FlowProperty;
import org.openlca.core.model.FlowPropertyFactor;
import org.openlca.core.model.NwSet;
import org.openlca.core.model.RefEntity;
import org.openlca.core.model.Unit;

/**
 * We copy most of the entities from the source database to the target database
 * by cloning the source entity, changing the references of the source entity to
 * the corresponding references in the destination database, and finally
 * inserting the new entity in the destination database. This class helps to
 * switch a referenced entity from the source database to the corresponding
 * entity in the destination database.
 */
class RefSwitcher {

	private final Seq seq;
	private final IDatabase source;
	private final IDatabase dest;

	RefSwitcher(Config conf) {
		this.source = conf.source();
		this.dest = conf.target();
		this.seq = conf.seq();
	}

	Category switchRef(Category srcCategory) {
		return switchRef(Seq.CATEGORY, new CategoryDao(dest), srcCategory);
	}

	Unit switchRef(Unit srcUnit) {
		if (srcUnit == null)
			return null;
		long id = seq.get(Seq.UNIT, srcUnit.refId);
		if (id == 0)
			return null;
		UnitDao dao = new UnitDao(dest);
		return dao.getForId(id);
	}

	/**
	 * Returns the corresponding flow property factor of the destination flow.
	 */
	FlowPropertyFactor switchRef(FlowPropertyFactor srcFactor, Flow destFlow) {
		if (srcFactor == null || destFlow == null)
			return null;
		FlowProperty srcProp = srcFactor.flowProperty;
		if (srcProp == null)
			return null;
		long propId = seq.get(Seq.FLOW_PROPERTY, srcProp.refId);
		for (FlowPropertyFactor fac : destFlow.flowPropertyFactors) {
			if (fac.flowProperty == null)
				continue;
			if (propId == fac.flowProperty.id)
				return fac;
		}
		return null;
	}

	private  T switchRef(
			int type, RefEntityDao dao, T srcEntity) {
		if (srcEntity == null)
			return null;
		long id = seq.get(type, srcEntity.refId);
		if (id == 0)
			return null;
		return dao.getForId(id);
	}

	Long getDestImpactId(Long srcId) {
		if (srcId == null)
			return null;
		var dao = new ImpactCategoryDao(source);
		var d = dao.getDescriptor(srcId);
		return d != null
				? seq.get(Seq.IMPACT_METHOD, d.refId)
				: null;
	}

	Long getDestProcessId(Long srcId) {
		if (srcId == null)
			return null;
		var dao = new ProcessDao(source);
		var d = dao.getDescriptor(srcId);
		return d != null
				? seq.get(Seq.PROCESS, d.refId)
				: null;
	}

}