body.getUnits().remove(unit);
doneSomething = true;
}
}
} else if (unit instanceof AssignStmt) {
AssignStmt stmt = (AssignStmt) unit;
/*Type assignmentType =*/stmt.getLeftOp().getType();
if (stmt.getLeftOp() instanceof Local
&& stmt.getRightOp() instanceof LengthExpr) {
if (debug) {
System.out.println("handling as length expr");
}
LengthExpr lengthExpr = (LengthExpr) stmt.getRightOp();
Local baseLocal = (Local) lengthExpr.getOp();
if (debug) {
System.out.println("operating on " + baseLocal);
}
Map fieldToReplacementArrayLocal = (Map) localToFieldToLocal
.get(baseLocal);
if (fieldToReplacementArrayLocal != null) {
doneSomething = true;
// Get the length of a random one of the replacement fields.
List replacementList = new ArrayList(
fieldToReplacementArrayLocal.keySet());
Collections.sort(replacementList, new Comparator() {
public int compare(Object o1, Object o2) {
SootField f1 = (SootField) o1;
SootField f2 = (SootField) o2;
return f1.getName().compareTo(f2.getName());
}
});
SootField field = (SootField) replacementList
.get(replacementList.size() - 1);
if (debug) {
System.out.println("replace with "
+ fieldToReplacementArrayLocal
.get(field));
}
lengthExpr
.setOp((Local) fieldToReplacementArrayLocal
.get(field));
if (debug) {
System.out.println("unit now = " + unit);
}
// body.getUnits().remove(unit);
}
} else if (stmt.getLeftOp() instanceof InstanceFieldRef) {
// Replace references to fields of tokens.
// FIXME: assign to all aliases as well.
if (debug) {
System.out
.println("is assignment to Instance FieldRef");
}
InstanceFieldRef r = (InstanceFieldRef) stmt
.getLeftOp();
SootField field = r.getField();
if (r.getBase().getType() instanceof RefType) {
RefType type = (RefType) r.getBase().getType();
//System.out.println("BaseType = " + type);
if (SootUtilities.derivesFrom(type.getSootClass(),
PtolemyUtilities.tokenClass)) {
if (debug) {
System.out.println("handling " + unit
+ " token operation");
}
// We have a reference to a field of a token class.
Local baseLocal = (Local) r.getBase();
Local instanceLocal = _getInstanceLocal(body,
baseLocal, field, localToFieldToLocal,
debug);
if (debug) {
System.out.println("instanceLocal = "
+ instanceLocal);
}
if (instanceLocal != null) {
stmt.getLeftOpBox().setValue(instanceLocal);
doneSomething = true;
}
}
}
} else if (stmt.getRightOp() instanceof InstanceFieldRef) {
// Replace references to fields of tokens.
if (debug) {
System.out
.println("is assignment from Instance FieldRef");
}
InstanceFieldRef r = (InstanceFieldRef) stmt
.getRightOp();
SootField field = r.getField();
if (r.getBase().getType() instanceof RefType) {
RefType type = (RefType) r.getBase().getType();
//System.out.println("BaseType = " + type);
if (SootUtilities.derivesFrom(type.getSootClass(),
PtolemyUtilities.tokenClass)) {
if (debug) {
System.out.println("handling " + unit
+ " token operation");
}
// We have a reference to a field of a token class.
Local baseLocal = (Local) r.getBase();
Local instanceLocal = _getInstanceLocal(body,
baseLocal, field, localToFieldToLocal,
debug);
if (debug) {
System.out.println("instanceLocal = "
+ instanceLocal);
}
if (instanceLocal != null) {
stmt.getRightOpBox()
.setValue(instanceLocal);
doneSomething = true;
}
}
}
}
}
}
}
if (debug) {
System.out.println("Specializing types for " + entityClass);
}
// Specialize the token types. Any field we created above
// should now have its correct concrete type.
// TypeSpecializer.specializeTypes(debug, entityClass, unsafeLocalSet);
// Now go through the methods again and handle all token assignments,
// replacing them with assignments on the native replacements.
for (Iterator methods = entityClass.getMethods().iterator(); methods
.hasNext();) {
SootMethod method = (SootMethod) methods.next();
if (debug) {
System.out.println("Replacing token assignments in method "
+ method);
}
JimpleBody body = (JimpleBody) method.retrieveActiveBody();
// InstanceEqualityEliminator.removeInstanceEqualities(body, null, true);
for (Iterator units = body.getUnits().snapshotIterator(); units
.hasNext();) {
Unit unit = (Unit) units.next();
if (debug) {
System.out.println("ttn3 unit = " + unit);
}
// Hack to work around the presence of NIL fields:
// Assume they are null references.
// for (Iterator boxes = unit.getUseAndDefBoxes().iterator(); boxes
// .hasNext();) {
// ValueBox box = (ValueBox) boxes.next();
// Value value = box.getValue();
// if (value instanceof FieldRef) {
// FieldRef ref = (FieldRef)value;
// SootField field = ref.getField();
// if (field.getName().equals("NIL")) {
// doneSomething = true;
// box.setValue(NullConstant.v());
// System.out.println("replacing as Null");
// }
// }
// }
if (unit instanceof AssignStmt) {
AssignStmt stmt = (AssignStmt) unit;
Type assignmentType = stmt.getLeftOp().getType();
if (PtolemyUtilities.isTokenType(assignmentType)) {
if (stmt.getLeftOp() instanceof Local
&& (stmt.getRightOp() instanceof Local || stmt
.getRightOp() instanceof Constant)) {
if (debug) {
System.out
.println("handling as local-immediate assign");
}
doneSomething |= _handleImmediateAssignment(body,
stmt, localToFieldToLocal,
localToIsNotNullLocal, stmt.getLeftOp(),
stmt.getRightOp(), debug);
} else if (stmt.getLeftOp() instanceof Local
&& stmt.getRightOp() instanceof CastExpr) {
if (debug) {
System.out.println("handling as local cast");
}
Value rightLocal = ((CastExpr) stmt.getRightOp())
.getOp();
doneSomething |= _handleImmediateAssignment(body,
stmt, localToFieldToLocal,
localToIsNotNullLocal, stmt.getLeftOp(),
rightLocal, debug);
} else if (stmt.getLeftOp() instanceof FieldRef
&& stmt.getRightOp() instanceof Local) {
if (debug) {
System.out
.println("handling as assignment to Field");
}
FieldRef oldFieldRef = (FieldRef) stmt.getLeftOp();
SootField field = oldFieldRef.getField();
Map fieldToReplacementField = (Map) entityFieldToTokenFieldToReplacementField
.get(field);
Map fieldToReplacementLocal = (Map) localToFieldToLocal
.get(stmt.getRightOp());
// System.out.println("fieldToReplacementField = " + fieldToReplacementField);
// System.out.println("fieldToReplacementLocal = " + fieldToReplacementLocal);
if ((fieldToReplacementLocal != null)
&& (fieldToReplacementField != null)) {
doneSomething = true;
// Replace references to fields with token types.
{
SootField replacementField = (SootField) entityFieldToIsNotNullField
.get(field);
//System.out.println("replacementField = " + replacementField);
FieldRef isNotNullFieldRef;
if (oldFieldRef instanceof InstanceFieldRef) {
isNotNullFieldRef = Jimple
.v()
.newInstanceFieldRef(
((InstanceFieldRef) oldFieldRef)
.getBase(),
replacementField
.makeRef());
} else {
isNotNullFieldRef = Jimple.v()
.newStaticFieldRef(
replacementField
.makeRef());
}
body
.getUnits()
.insertBefore(
Jimple
.v()
.newAssignStmt(
isNotNullFieldRef,
(Local) localToIsNotNullLocal
.get(stmt
.getRightOp())),
unit);
}
if (debug) {
System.out.println("local = "
+ stmt.getLeftOp());
}
for (Iterator tokenFields = fieldToReplacementField
.keySet().iterator(); tokenFields
.hasNext();) {
SootField tokenField = (SootField) tokenFields
.next();
if (debug) {
System.out.println("tokenField = "
+ tokenField);
}
SootField replacementField = (SootField) fieldToReplacementField
.get(tokenField);
Local replacementLocal = (Local) fieldToReplacementLocal
.get(tokenField);
if (debug) {
System.out
.println("replacementLocal = "
+ replacementLocal);
}
FieldRef fieldRef;
if (stmt.getLeftOp() instanceof InstanceFieldRef) {
Local base = (Local) ((InstanceFieldRef) stmt
.getLeftOp()).getBase();
fieldRef = Jimple.v()
.newInstanceFieldRef(
base,
replacementField
.makeRef());
} else {
fieldRef = Jimple.v()
.newStaticFieldRef(
replacementField
.makeRef());
}
body.getUnits().insertBefore(
Jimple.v().newAssignStmt(fieldRef,
replacementLocal), unit);
}
stmt.getRightOpBox().setValue(NullConstant.v());
// body.getUnits().remove(unit);
}
} else if (stmt.getLeftOp() instanceof Local
&& stmt.getRightOp() instanceof FieldRef) {
if (debug) {
System.out
.println("handling as assignment from Field");
}
FieldRef oldFieldRef = (FieldRef) stmt.getRightOp();
Map fieldToReplacementLocal = (Map) localToFieldToLocal
.get(stmt.getLeftOp());
SootField field = oldFieldRef.getField();
Map fieldToReplacementField = (Map) entityFieldToTokenFieldToReplacementField
.get(field);
// There are some fields that represent
// singleton tokens. Deal with them
// specially.
// This is awkward, but it is simpler than
// doing a dataflow analysis of the static
// initializer of the token class to
// figure out what the value is.
boolean isSingleton = false;
if (field.getName().equals("TRUE")
|| field.getName().equals("FALSE")
|| field.getName().equals("ZERO")
|| field.getName().equals("ONE")) {
isSingleton = true;
}
if ((isSingleton)
&& (fieldToReplacementLocal != null)) {
doneSomething = true;
// Replace references to fields with
// token types. The special fields
// should never be null
body.getUnits().insertBefore(
Jimple.v().newAssignStmt(
(Local) localToIsNotNullLocal
.get(stmt.getLeftOp()),
IntConstant.v(1)), unit);
if (debug) {
System.out.println("local = "
+ stmt.getLeftOp());
}
for (Iterator localFields = fieldToReplacementLocal
.keySet().iterator(); localFields
.hasNext();) {
SootField localField = (SootField) localFields
.next();
if (debug) {
System.out.println("localField = "
+ localField);
}
// if (localField.getName().equals("_isNil")) {
// Local replacementLocal = (Local) fieldToReplacementLocal
// .get(localField);
// body
// .getUnits()
// .insertBefore(
// Jimple
// .v()
// .newAssignStmt(
// replacementLocal,
// IntConstant
// .v(1)),
// unit);
// } else
if (localField.getName().equals(
"_unitCategoryExponents")) {
Local replacementLocal = (Local) fieldToReplacementLocal
.get(localField);
body
.getUnits()
.insertBefore(
Jimple
.v()
.newAssignStmt(
replacementLocal,
NullConstant
.v()),
unit);
} else if (localField.getName().equals(
"_value")) {
Local replacementLocal = (Local) fieldToReplacementLocal
.get(localField);
if (field
.getSignature()
.equals(
"<ptolemy.data.BooleanToken: ptolemy.data.BooleanToken TRUE>")) {
body.getUnits().insertBefore(
Jimple.v().newAssignStmt(
replacementLocal,
IntConstant.v(1)),
unit);
} else if (field
.getSignature()
.equals(
"<ptolemy.data.BooleanToken: ptolemy.data.BooleanToken FALSE>")) {
body.getUnits().insertBefore(
Jimple.v().newAssignStmt(
replacementLocal,
IntConstant.v(0)),
unit);
} else if (field
.getSignature()
.equals(
"<ptolemy.data.UnsignedByteToken: ptolemy.data.UnsignedByteToken ONE>")) {
body.getUnits().insertBefore(
Jimple.v().newAssignStmt(
replacementLocal,
IntConstant.v(1)),
unit);
} else if (field
.getSignature()
.equals(
"<ptolemy.data.UnsignedByteToken: ptolemy.data.UnsignedByteToken ZERO>")) {
body.getUnits().insertBefore(
Jimple.v().newAssignStmt(
replacementLocal,
IntConstant.v(0)),
unit);
} else if (field
.getSignature()
.equals(
"<ptolemy.data.IntToken: ptolemy.data.IntToken ONE>")) {
body.getUnits().insertBefore(
Jimple.v().newAssignStmt(
replacementLocal,
IntConstant.v(1)),
unit);
} else if (field
.getSignature()
.equals(
"<ptolemy.data.IntToken: ptolemy.data.IntToken ZERO>")) {
body.getUnits().insertBefore(
Jimple.v().newAssignStmt(
replacementLocal,
IntConstant.v(0)),
unit);
} else if (field
.getSignature()
.equals(
"<ptolemy.data.LongToken: ptolemy.data.LongToken ONE>")) {
body.getUnits().insertBefore(
Jimple.v().newAssignStmt(
replacementLocal,
LongConstant.v(1)),
unit);
} else if (field
.getSignature()
.equals(
"<ptolemy.data.LongToken: ptolemy.data.LongToken ZERO>")) {
body.getUnits().insertBefore(
Jimple.v().newAssignStmt(
replacementLocal,
LongConstant.v(0)),
unit);
} else if (field
.getSignature()
.equals(
"<ptolemy.data.DoubleToken: ptolemy.data.DoubleToken ONE>")) {
body.getUnits().insertBefore(
Jimple.v().newAssignStmt(
replacementLocal,
DoubleConstant
.v(1.0)),
unit);
} else if (field
.getSignature()
.equals(
"<ptolemy.data.DoubleToken: ptolemy.data.DoubleToken ZERO>")) {
body.getUnits().insertBefore(
Jimple.v().newAssignStmt(
replacementLocal,
DoubleConstant
.v(0.0)),
unit);
}
} else {
throw new RuntimeException(
"Unknown Field in Token: "
+ localField
.getSignature());
}
}
stmt.getRightOpBox().setValue(NullConstant.v());
//body.getUnits().remove(unit);
} else if ((fieldToReplacementLocal != null)
&& (fieldToReplacementField != null)) {
doneSomething = true;
// Replace references to fields with token types.
// FIXME properly handle isNotNull field?
{
SootField replacementField = (SootField) entityFieldToIsNotNullField
.get(field);
// System.out.println("replacementField = " + replacementField);
FieldRef isNotNullFieldRef;
if (oldFieldRef instanceof InstanceFieldRef) {
isNotNullFieldRef = Jimple
.v()
.newInstanceFieldRef(
((InstanceFieldRef) oldFieldRef)
.getBase(),
replacementField
.makeRef());
} else {
isNotNullFieldRef = Jimple.v()
.newStaticFieldRef(
replacementField
.makeRef());
}
body
.getUnits()
.insertBefore(
Jimple
.v()
.newAssignStmt(
(Local) localToIsNotNullLocal
.get(stmt
.getLeftOp()),
isNotNullFieldRef),
unit);
}
if (debug) {
System.out.println("local = "
+ stmt.getLeftOp());
}
for (Iterator tokenFields = fieldToReplacementField
.keySet().iterator(); tokenFields
.hasNext();) {
SootField tokenField = (SootField) tokenFields
.next();
if (debug) {
System.out.println("tokenField = "
+ tokenField);
}
SootField replacementField = (SootField) fieldToReplacementField
.get(tokenField);
Local replacementLocal = (Local) fieldToReplacementLocal
.get(tokenField);
FieldRef fieldRef;
if (stmt.getRightOp() instanceof InstanceFieldRef) {
Local base = (Local) ((InstanceFieldRef) stmt
.getRightOp()).getBase();
fieldRef = Jimple.v()
.newInstanceFieldRef(
base,
replacementField
.makeRef());
} else {
fieldRef = Jimple.v()
.newStaticFieldRef(
replacementField
.makeRef());
}
if (debug) {
System.out
.println("replacementLocal = "
+ replacementLocal);
}
body.getUnits()
.insertBefore(
Jimple.v().newAssignStmt(
replacementLocal,
fieldRef), unit);
}
stmt.getRightOpBox().setValue(NullConstant.v());
// body.getUnits().remove(unit);
}
} else if (stmt.getLeftOp() instanceof ArrayRef
&& stmt.getRightOp() instanceof Local) {
if (debug) {
System.out
.println("handling as assignment to Array");
}
ArrayRef arrayRef = (ArrayRef) stmt.getLeftOp();
Local baseLocal = (Local) arrayRef.getBase();
Map fieldToReplacementArrayLocal = (Map) localToFieldToLocal
.get(baseLocal);
Map fieldToReplacementLocal = (Map) localToFieldToLocal
.get(stmt.getRightOp());
if (debug) {
System.out
.println("fieldToReplacementArrayLocal = "
+ fieldToReplacementArrayLocal);
}
if (debug) {
System.out.println("fieldToReplacementLocal = "
+ fieldToReplacementLocal);
}
if ((fieldToReplacementLocal != null)
&& (fieldToReplacementArrayLocal != null)) {
doneSomething = true;
body
.getUnits()
.insertBefore(
Jimple
.v()
.newAssignStmt(
Jimple
.v()
.newArrayRef(
(Local) localToIsNotNullLocal
.get(baseLocal),
arrayRef
.getIndex()),
(Local) localToIsNotNullLocal
.get(stmt
.getRightOp())),
unit);
if (debug) {
System.out.println("local = "
+ stmt.getLeftOp());
}
for (Iterator tokenFields = fieldToReplacementLocal
.keySet().iterator(); tokenFields
.hasNext();) {
SootField tokenField = (SootField) tokenFields
.next();
if (debug) {
System.out.println("tokenField = "
+ tokenField);
}
Local replacementArrayLocal = (Local) fieldToReplacementArrayLocal
.get(tokenField);
Local replacementLocal = (Local) fieldToReplacementLocal
.get(tokenField);
body
.getUnits()
.insertBefore(
Jimple
.v()
.newAssignStmt(
Jimple
.v()
.newArrayRef(
replacementArrayLocal,
arrayRef
.getIndex()),
replacementLocal),
unit);
}
// Have to remove here, because otherwise we'll try to
// index into a null array.
//stmt.getRightOpBox().setValue(NullConstant.v());
body.getUnits().remove(unit);
}
} else if (stmt.getLeftOp() instanceof Local
&& stmt.getRightOp() instanceof ArrayRef) {
if (debug) {
System.out
.println("handling as assignment from Array");
}
ArrayRef arrayRef = (ArrayRef) stmt.getRightOp();
Map fieldToReplacementLocal = (Map) localToFieldToLocal
.get(stmt.getLeftOp());
Local baseLocal = (Local) arrayRef.getBase();
Map fieldToReplacementArrayLocal = (Map) localToFieldToLocal
.get(baseLocal);
if ((fieldToReplacementLocal != null)
&& (fieldToReplacementArrayLocal != null)) {
doneSomething = true;
body
.getUnits()
.insertBefore(
Jimple
.v()
.newAssignStmt(
(Local) localToIsNotNullLocal
.get(stmt
.getLeftOp()),
Jimple
.v()
.newArrayRef(
(Local) localToIsNotNullLocal
.get(baseLocal),
arrayRef
.getIndex())),
unit);
if (debug) {
System.out.println("local = "
+ stmt.getLeftOp());
}
for (Iterator tokenFields = fieldToReplacementLocal
.keySet().iterator(); tokenFields
.hasNext();) {
SootField tokenField = (SootField) tokenFields
.next();
if (debug) {
System.out.println("tokenField = "
+ tokenField);
}
Local replacementArrayLocal = (Local) fieldToReplacementArrayLocal
.get(tokenField);
Local replacementLocal = (Local) fieldToReplacementLocal
.get(tokenField);
if (debug) {
System.out
.println("replacementLocal = "
+ replacementLocal);
}
if (debug) {
System.out
.println("replacementArrayLocal = "
+ replacementArrayLocal);
}
body
.getUnits()
.insertBefore(
Jimple
.v()
.newAssignStmt(
replacementLocal,
Jimple
.v()
.newArrayRef(
replacementArrayLocal,
arrayRef
.getIndex())),
unit);
}
stmt.getRightOpBox().setValue(NullConstant.v());
//body.getUnits().remove(unit);
}
} else if (stmt.getLeftOp() instanceof Local
&& stmt.getRightOp() instanceof NewArrayExpr) {
if (debug) {
System.out
.println("handling as new array object");
}
NewArrayExpr newExpr = (NewArrayExpr) stmt
.getRightOp();
// We have an assignment to a local from a new array.
Map map = (Map) localToFieldToLocal.get(stmt
.getLeftOp());
if (map != null) {
doneSomething = true;
Type isNotNullType = SootUtilities
.createIsomorphicType(newExpr
.getBaseType(), BooleanType.v());
Local isNotNullLocal = (Local) localToIsNotNullLocal
.get(stmt.getLeftOp());
body.getUnits().insertBefore(
Jimple.v().newAssignStmt(
isNotNullLocal,
Jimple.v().newNewArrayExpr(
isNotNullType,
newExpr.getSize())),
unit);
for (Iterator tokenFields = map.keySet()
.iterator(); tokenFields.hasNext();) {
SootField tokenField = (SootField) tokenFields
.next();
if (debug) {
System.out.println("tokenField = "
+ tokenField);
}
Local replacementLocal = (Local) map
.get(tokenField);
Type replacementType = SootUtilities
.createIsomorphicType(newExpr
.getBaseType(), tokenField
.getType());
// Initialize fields?
body
.getUnits()
.insertBefore(
Jimple
.v()
.newAssignStmt(
replacementLocal,
Jimple
.v()
.newNewArrayExpr(
replacementType,
newExpr
.getSize())),
unit);
}
stmt.getRightOpBox().setValue(NullConstant.v());
//body.getUnits().remove(unit);
}
} else if (stmt.getLeftOp() instanceof Local
&& stmt.getRightOp() instanceof NewExpr) {
if (debug) {
System.out.println("handling as new object");
}
/* NewExpr newExpr = (NewExpr)*/stmt.getRightOp();
// We have an assignment from one local token to another.
Map map = (Map) localToFieldToLocal.get(stmt
.getLeftOp());
if (map != null) {
doneSomething = true;
Local isNotNullLocal = (Local) localToIsNotNullLocal
.get(stmt.getLeftOp());
if (debug) {
System.out.println("Stmt = " + stmt);
}
body.getUnits().insertBefore(
Jimple.v().newAssignStmt(
isNotNullLocal,
IntConstant.v(1)), unit);
for (Iterator tokenFields = map.keySet()
.iterator(); tokenFields.hasNext();) {
SootField tokenField = (SootField) tokenFields
.next();
Local replacementLocal = (Local) map
.get(tokenField);
// Initialize fields?
if (debug) {
System.out.println("tokenField = "
+ tokenField);
}
// FIXME: ??
Value replacementValue = _getNullValueForType(replacementLocal
.getType());
body.getUnits().insertBefore(
Jimple.v().newAssignStmt(
replacementLocal,
replacementValue), unit);
}
stmt.getRightOpBox().setValue(NullConstant.v());
// body.getUnits().remove(unit);
}
}