68 lines
2.1 KiB
Java
68 lines
2.1 KiB
Java
package io.gitlab.jfronny.muscript.ast.bool;
|
|
|
|
import io.gitlab.jfronny.muscript.ast.compare.Equal;
|
|
import io.gitlab.jfronny.muscript.ast.compare.Greater;
|
|
import io.gitlab.jfronny.muscript.ast.literal.NumberLiteral;
|
|
import io.gitlab.jfronny.muscript.compiler.*;
|
|
import io.gitlab.jfronny.muscript.data.Scope;
|
|
import io.gitlab.jfronny.muscript.ast.BoolExpr;
|
|
import io.gitlab.jfronny.muscript.ast.Expr;
|
|
import io.gitlab.jfronny.muscript.ast.literal.BoolLiteral;
|
|
|
|
import java.io.IOException;
|
|
import java.util.stream.Stream;
|
|
|
|
public class Not extends BoolExpr {
|
|
public final BoolExpr inner;
|
|
|
|
public Not(CodeLocation location, BoolExpr inner) {
|
|
super(Order.Unary, location);
|
|
this.inner = inner;
|
|
}
|
|
|
|
@Override
|
|
public Boolean get(Scope dataRoot) {
|
|
return !inner.get(dataRoot);
|
|
}
|
|
|
|
@Override
|
|
public BoolExpr optimize() {
|
|
BoolExpr inner = this.inner.optimize();
|
|
if (inner instanceof Not not) return not.inner;
|
|
if (inner instanceof BoolLiteral literal) return literal(location, !literal.value);
|
|
return new Not(location, inner);
|
|
}
|
|
|
|
@Override
|
|
public Stream<Expr<?>> extractSideEffects() {
|
|
return inner.extractSideEffects();
|
|
}
|
|
|
|
@Override
|
|
public void decompile(ExprWriter writer) throws IOException {
|
|
if (inner instanceof Equal eq) {
|
|
parenthesize(eq.left, writer, false);
|
|
writer.append(" != ");
|
|
parenthesize(eq.right, writer, true);
|
|
} else if (inner instanceof Greater gt) {
|
|
if (gt.left instanceof NumberLiteral && !(gt.right instanceof NumberLiteral)) {
|
|
parenthesize(gt.right, writer, false);
|
|
writer.append(" >= ");
|
|
parenthesize(gt.left, writer, true);
|
|
} else {
|
|
parenthesize(gt.left, writer, false);
|
|
writer.append(" <= ");
|
|
parenthesize(gt.right, writer, true);
|
|
}
|
|
} else {
|
|
writer.append("!");
|
|
parenthesize(inner, writer, false);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public boolean equals(Object obj) {
|
|
return obj instanceof Not not && inner.equals(not.inner);
|
|
}
|
|
}
|