org.openlca.ipc.handlers.JsonRpc Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of olca-ipc Show documentation
There is a newer version: 2.3.0
package org.openlca.ipc.handlers;

import java.util.Collection;
import java.util.Set;
import java.util.function.BiConsumer;
import java.util.function.BinaryOperator;
import java.util.function.Function;
import java.util.function.Supplier;
import java.util.stream.Collector;

import org.openlca.core.matrix.index.EnviFlow;
import org.openlca.core.matrix.index.TechFlow;
import org.openlca.core.results.Contribution;
import org.openlca.core.results.EnviFlowValue;
import org.openlca.core.results.ImpactValue;
import org.openlca.core.results.LcaResult;
import org.openlca.jsonld.Json;
import org.openlca.jsonld.output.JsonRefs;

import com.google.gson.JsonArray;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;

/**
 * Some utility functions for en/decoding data in JSON-RPC.
 */
class JsonRpc {

	private JsonRpc() {
	}

	static JsonObject encodeResult(LcaResult r, String id, JsonRefs refs) {
		var obj = new JsonObject();
		Json.put(obj, "@id", id);
		obj.addProperty("@id", id);
		if (r == null)
			return obj;
		Json.put(obj, "@type", r.getClass().getSimpleName());

		// tech. index, providers
		if (r.techIndex() != null) {
			var providers = r.techIndex()
					.stream()
					.map(techFlow -> encodeTechFlow(techFlow, refs))
					.collect(toArray());
			Json.put(obj, "providers", providers);
		}

		// flows & flow results
		if (r.hasEnviFlows()) {
			Json.put(obj, "flows", arrayOf(
					r.enviIndex(),
					enviFlow -> encodeEnviFlow(enviFlow, refs)));
			obj.add("flowResults",
					arrayOf(r.getTotalFlows(), v -> encodeFlowValue(v, refs)));
		}

		// impact categories and results
		if (r.hasImpacts()) {
			obj.add("impacts", arrayOf(r.impactIndex(), refs::asRef));
			obj.add("impactResults", arrayOf(
					r.getTotalImpacts(), v -> encodeImpactValue(v, refs)));
		}

		return obj;
	}

	static JsonObject encodeEnviFlow(EnviFlow ef, JsonRefs refs) {
		var obj = new JsonObject();
		Json.put(obj, "flow", refs.asRef(ef.flow()));
		Json.put(obj, "location", refs.asRef(ef.location()));
		Json.put(obj, "isInput", ef.isInput());
		return obj;
	}

	static JsonObject encodeTechFlow(TechFlow techFlow, JsonRefs refs) {
		var obj = new JsonObject();
		Json.put(obj, "provider", refs.asRef(techFlow.provider()));
		Json.put(obj, "flow", refs.asRef(techFlow.flow()));
		return obj;
	}

	static JsonObject encodeFlowValue(EnviFlowValue r, JsonRefs refs) {
		if (r == null)
			return null;
		JsonObject obj = new JsonObject();
		obj.addProperty("@type", "FlowResult");
		obj.add("flow", encodeEnviFlow(r.enviFlow(), refs));
		obj.addProperty("value", r.value());
		return obj;
	}

	static JsonObject encodeImpactValue(ImpactValue r, JsonRefs refs) {
		if (r == null)
			return null;
		JsonObject obj = new JsonObject();
		obj.addProperty("@type", "ImpactResult");
		obj.add("impactCategory", refs.asRef(r.impact()));
		obj.addProperty("value", r.value());
		return obj;
	}

	static  JsonArray encodeContributions(
			Collection> contributions,
			Function itemEncoder) {
		return contributions.stream()
				.map(c -> encodeContribution(c, itemEncoder))
				.collect(toArray());
	}

	static  JsonObject encodeContribution(Contribution c,
			Function itemEncoder) {
		if (c == null)
			return null;
		var obj = new JsonObject();
		Json.put(obj, "@type", "Contribution");
		if (c.item != null) {
			Json.put(obj, "item", itemEncoder.apply(c.item));
		}
		Json.put(obj, "amount", c.amount);
		Json.put(obj, "share", c.share);
		Json.put(obj, "unit", c.unit);
		Json.put(obj, "rest", c.isRest);
		return obj;
	}

	static  JsonArray arrayOf(
			Iterable elements, Function fn) {
		var array = new JsonArray();
		if (elements == null || fn == null)
			return array;
		for (var elem : elements) {
			var jsonElem = fn.apply(elem);
			if (jsonElem != null) {
				array.add(jsonElem);
			}
		}
		return array;
	}

	public static JsonArrayCollector toArray() {
		return new JsonArrayCollector();
	}

	static class JsonArrayCollector
			implements Collector {

		@Override
		public Supplier supplier() {
			return JsonArray::new;
		}

		@Override
		public BiConsumer accumulator() {
			return (array, element) -> {
				if (element != null) {
					array.add(element);
				}
			};
		}

		@Override
		public BinaryOperator combiner() {
			return (array1, array2) -> {
				var combined = new JsonArray();
				combined.addAll(array1);
				combined.addAll(array2);
				return combined;
			};
		}

		@Override
		public Function finisher() {
			return a -> a;
		}

		@Override
		public Set characteristics() {
			return Set.of(Characteristics.IDENTITY_FINISH);
		}
	}

}