forked from yuemingl/SymJava
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathConcat.java
More file actions
83 lines (72 loc) · 2.53 KB
/
Concat.java
File metadata and controls
83 lines (72 loc) · 2.53 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
package symjava.symbolic;
import java.util.Map;
import symjava.symbolic.arity.NaryOp;
import symjava.symbolic.utils.Utils;
import com.sun.org.apache.bcel.internal.Constants;
import com.sun.org.apache.bcel.internal.generic.AASTORE;
import com.sun.org.apache.bcel.internal.generic.ANEWARRAY;
import com.sun.org.apache.bcel.internal.generic.ArrayType;
import com.sun.org.apache.bcel.internal.generic.ConstantPoolGen;
import com.sun.org.apache.bcel.internal.generic.InstructionConstants;
import com.sun.org.apache.bcel.internal.generic.InstructionFactory;
import com.sun.org.apache.bcel.internal.generic.InstructionHandle;
import com.sun.org.apache.bcel.internal.generic.InstructionList;
import com.sun.org.apache.bcel.internal.generic.MethodGen;
import com.sun.org.apache.bcel.internal.generic.ObjectType;
import com.sun.org.apache.bcel.internal.generic.PUSH;
import com.sun.org.apache.bcel.internal.generic.Type;
/**
* Concatenate symbol matrices or vectors with the same dimension(s)
*
*/
public class Concat extends NaryOp {
public Concat(Expr ...args) {
super(args);
updateLabel();
}
@Override
public Expr simplify() {
return this;
}
@Override
public boolean symEquals(Expr other) {
return false;
}
@Override
public Expr diff(Expr x) {
return null;
}
@Override
public void updateLabel() {
this.label = "["+Utils.joinLabels(args, ", ")+"]";
this.sortKey = this.label;
}
@Override
public InstructionHandle bytecodeGen(String clsName, MethodGen mg,
ConstantPoolGen cp, InstructionFactory factory,
InstructionList il, Map<String, Integer> argsMap, int argsStartPos,
Map<Expr, Integer> funcRefsMap) {
InstructionHandle startPos = null;
//if(arg1.getType() == TYPE.MATRIX && arg2.getType() == TYPE.MATRIX) {
il.append(new PUSH(cp, args.length));
il.append(new ANEWARRAY(cp.addClass("Jama.Matrix")));
for(int i=0; i<args.length; i++) {
il.append(InstructionConstants.DUP);
il.append(new PUSH(cp, i));
InstructionHandle tmp = args[i].bytecodeGen(clsName, mg, cp, factory, il, argsMap, argsStartPos, funcRefsMap);
if(startPos == null) startPos = tmp;
il.append(new AASTORE());
}
il.append(factory.createInvoke("symjava.symbolic.utils.BytecodeOpSupport", "concat",
new ObjectType("Jama.Matrix"), new Type[] { new ArrayType(new ObjectType("Jama.Matrix"),1) },
Constants.INVOKESTATIC));
return startPos;
}
@Override
public TypeInfo getTypeInfo() {
TypeInfo ti = new TypeInfo(TYPE.VECTOR, new int[1]);
for(int i=0; i<args.length; i++)
ti.dim[0] += args[i].getTypeInfo().dim[0];
return ti;
}
}