WarpPI/rules/src/main/java/rules/NumberRule3.java
2018-09-22 11:17:30 +02:00

81 lines
2.5 KiB
Java

package rules;
/*
SETTINGS: (please don't move this part)
PATH=NumberRule3
*/
import it.cavallium.warppi.math.Function;
import it.cavallium.warppi.math.FunctionOperator;
import it.cavallium.warppi.math.MathContext;
import it.cavallium.warppi.math.functions.Multiplication;
import it.cavallium.warppi.math.functions.Number;
import it.cavallium.warppi.math.functions.Subtraction;
import it.cavallium.warppi.math.functions.Sum;
import it.cavallium.warppi.math.functions.SumSubtraction;
import it.cavallium.warppi.math.rules.Rule;
import it.cavallium.warppi.math.rules.RuleType;
import it.unimi.dsi.fastutil.objects.ObjectArrayList;
/**
* Number rule
* a - a = 0
* -a + a = 0
* a ± a = {0, 2a}
*
* @author Andrea Cavalli
*
*/
public class NumberRule3 implements Rule {
// Rule name
@Override
public String getRuleName() {
return "NumberRule3";
}
// Rule type
@Override
public RuleType getRuleType() {
return RuleType.CALCULATION;
}
/* Rule function
Returns:
- null if it's not executable on the function "f"
- An ObjectArrayList<Function> if it did something
*/
@Override
public ObjectArrayList<Function> execute(final Function f) {
boolean isExecutable = false;
if (f instanceof Subtraction) {
final FunctionOperator sub = (FunctionOperator) f;
if (sub.getParameter1().equals(sub.getParameter2()))
isExecutable = true;
} else if (f instanceof Sum) {
final FunctionOperator sub = (FunctionOperator) f;
if (sub.getParameter1() instanceof Multiplication)
if (((FunctionOperator) sub.getParameter1()).getParameter1() instanceof Number && ((FunctionOperator) sub.getParameter1()).getParameter1().equals(new Number(f.getMathContext(), -1))) {
final Function neg = ((FunctionOperator) sub.getParameter1()).getParameter2();
if (neg.equals(sub.getParameter2()))
isExecutable = true;
}
} else if (f instanceof SumSubtraction) {
final FunctionOperator sub = (FunctionOperator) f;
if (sub.getParameter1().equals(sub.getParameter2()))
isExecutable = true;
}
if (isExecutable) {
final MathContext root = f.getMathContext();
final ObjectArrayList<Function> result = new ObjectArrayList<>();
if (f instanceof SumSubtraction) {
final Function mul = new Multiplication(root, new Number(root, 2), ((FunctionOperator) f).getParameter1());
result.add(mul);
}
result.add(new Number(root, 0));
return result;
} else
return null;
}
}