public static MatrixNodeI valueOf(SmartMatrix m) {
StringBuffer sb = new StringBuffer();
sb.append('[');
for (int i = 0; i < m.getDimR(); ++i) {
if (i != 0) sb.append(',');
sb.append('[');
for (int j = 0; j < m.getDimC(); ++j) {
if (j != 0) sb.append(',');
sb.append(m.getEntry(i, j));
}
sb.append(']');
}
sb.append(']');
Node r = null;
try {
r = jep.parse(sb.toString());
} catch (ParseException e) {
e.printStackTrace();
}
return (MatrixNodeI) r;
}
public class JepMatrix {
public static MatrixJep jep = new MatrixJep();
public static final ASTConstant zero = (ASTConstant) jep.getNumberFactory().getZero();
public static final MatrixNodeFactory factory = (MatrixNodeFactory) jep.getNodeFactory();
public static final MatrixOperatorSet op = (MatrixOperatorSet) jep.getOperatorSet();
public static final PolynomialCreator simplifier = new PolynomialCreator(jep);
public static MatrixNodeI power(MatrixNodeI m, int exp) {
assert (exp > 0);
if (exp == 1) return m;
int half = exp / 2;
MatrixNodeI mh = power(m, half);
MatrixNodeI result = multiply(mh, mh);
if (exp % 2 == 1) {
result = multiply(result, m);
}
return result;
}
public static MatrixNodeI multiply(MatrixNodeI m1, MatrixNodeI m2) {
assert (m1.getDim().is2D());
assert (m2.getDim().is2D());
assert (m1.getDim().getLastDim() == m2.getDim().getFirstDim());
int m = m1.getDim().getFirstDim();
int n = m1.getDim().getLastDim();
int p = m2.getDim().getLastDim();
Node[] entry = new Node[m * p];
int index = 0;
try {
for (int i = 0; i < m; ++i) {
for (int j = 0; j < p; ++j) {
Node node = zero;
for (int k = 0; k < n; ++k) {
Node item = factory.buildOperatorNode(op.getMultiply(), get(m1, i, k), get(m2, k, j));
node = factory.buildOperatorNode(op.getAdd(), node, item);
}
entry[index] = simplifier.simplify(node);
++index;
}
}
} catch (ParseException e) {
}
return (MatrixNodeI) factory.buildOperatorNode(op.getMList(), entry, Dimensions.valueOf(m, p));
}
public static Node get(MatrixNodeI m, int row, int col) {
return m.jjtGetChild(row * m.getDim().getLastDim() + col);
}
public static MatrixNodeI valueOf(SmartMatrix m) {
StringBuffer sb = new StringBuffer();
sb.append('[');
for (int i = 0; i < m.getDimR(); ++i) {
if (i != 0) sb.append(',');
sb.append('[');
for (int j = 0; j < m.getDimC(); ++j) {
if (j != 0) sb.append(',');
sb.append(m.getEntry(i, j));
}
sb.append(']');
}
sb.append(']');
Node r = null;
try {
r = jep.parse(sb.toString());
} catch (ParseException e) {
e.printStackTrace();
}
return (MatrixNodeI) r;
}
}
