org.openlca.proto.io.output.UnitGroupWriter Maven / Gradle / Ivy

Go to download
package org.openlca.proto.io.output;

import java.util.Arrays;
import java.util.Objects;

import org.openlca.core.model.UnitGroup;
import org.openlca.proto.ProtoType;
import org.openlca.proto.ProtoUnit;
import org.openlca.proto.ProtoUnitGroup;
import org.openlca.util.Strings;

public class UnitGroupWriter {

  private final WriterConfig config;

  public UnitGroupWriter(WriterConfig config) {
    this.config = config;
  }

  public ProtoUnitGroup write(UnitGroup group) {
    var proto = ProtoUnitGroup.newBuilder();
    if (group == null)
      return proto.build();
    proto.setType(ProtoType.UnitGroup);
    Out.map(group, proto);
		config.dep(group.defaultFlowProperty, proto::setDefaultFlowProperty);
    writeUnits(group, proto);
    return proto.build();
  }

  private void writeUnits(UnitGroup group, ProtoUnitGroup.Builder proto) {
    for (var unit : group.units) {
      var protoUnit = ProtoUnit.newBuilder();
      protoUnit.setId(Strings.orEmpty(unit.refId));
      protoUnit.setName(Strings.orEmpty(unit.name));
      protoUnit.setDescription(Strings.orEmpty(unit.description));
      protoUnit.setConversionFactor(unit.conversionFactor);
      if (unit.synonyms != null) {
        Arrays.stream(unit.synonyms.split(";"))
          .map(String::trim)
          .filter(Strings::notEmpty)
          .forEach(protoUnit::addSynonyms);
      }
      protoUnit.setIsRefUnit(Objects.equals(unit, group.referenceUnit));
      proto.addUnits(protoUnit.build());
    }
  }
}