cdm.base.math.functions.CompareQuantityByUnitOfAmount Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of cdm-java Show documentation

There is a newer version: 6.0.0-dev.89

package cdm.base.math.functions;

import cdm.base.math.CompareOp;
import cdm.base.math.Quantity;
import cdm.base.math.UnitType;
import com.google.inject.ImplementedBy;
import com.rosetta.model.lib.expression.CardinalityOperator;
import com.rosetta.model.lib.functions.RosettaFunction;
import com.rosetta.model.lib.mapper.MapperC;
import com.rosetta.model.lib.mapper.MapperListOfLists;
import com.rosetta.model.lib.mapper.MapperS;
import java.math.BigDecimal;
import java.util.Collections;
import java.util.List;
import javax.inject.Inject;

import static com.rosetta.model.lib.expression.ExpressionOperators.*;

@ImplementedBy(CompareQuantityByUnitOfAmount.CompareQuantityByUnitOfAmountDefault.class)
public abstract class CompareQuantityByUnitOfAmount implements RosettaFunction {
	
	// RosettaFunction dependencies
	//
	@Inject protected CompareNumbers compareNumbers;
	@Inject protected FilterQuantity filterQuantity;

	/**
	* @param quantity1 
	* @param op 
	* @param quantity2 
	* @param unitOfAmount 
	* @return result 
	*/
	public Boolean evaluate(List quantity1, CompareOp op, List quantity2, UnitType unitOfAmount) {
		Boolean result = doEvaluate(quantity1, op, quantity2, unitOfAmount);
		
		return result;
	}

	protected abstract Boolean doEvaluate(List quantity1, CompareOp op, List quantity2, UnitType unitOfAmount);

	public static class CompareQuantityByUnitOfAmountDefault extends CompareQuantityByUnitOfAmount {
		@Override
		protected Boolean doEvaluate(List quantity1, CompareOp op, List quantity2, UnitType unitOfAmount) {
			if (quantity1 == null) {
				quantity1 = Collections.emptyList();
			}
			if (quantity2 == null) {
				quantity2 = Collections.emptyList();
			}
			Boolean result = null;
			return assignOutput(result, quantity1, op, quantity2, unitOfAmount);
		}
		
		protected Boolean assignOutput(Boolean result, List quantity1, CompareOp op, List quantity2, UnitType unitOfAmount) {
			final MapperListOfLists thenResult = MapperC.of(filterQuantity.evaluate(quantity1, unitOfAmount))
				.mapItemToList(q1 -> MapperC.of(filterQuantity.evaluate(quantity2, unitOfAmount))
					.mapItem(q2 -> MapperS.of(compareNumbers.evaluate(q1.map("getValue", quantity -> quantity.getValue()).get(), op, q2.map("getValue", quantity -> quantity.getValue()).get()))));
			result = areEqual(thenResult
				.flattenList(), MapperS.of(true), CardinalityOperator.All).asMapper().get();
			
			return result;
		}
	}
}