public static void handleTIRAbstractAssignToVarStmt(
TIRAbstractAssignStmt node, IRx10ASTGenerator target, StmtBlock block) {
boolean isDecl;
String LHS;
boolean isReductionStmt = Helper.isReductionStmt(node, target, block);
target.symbolMapKey = ((TIRAbstractAssignToVarStmt) node).getTargetName().getID();
LHS = target.symbolMapKey;
/*
* If checks whether it is already defined, in which case it's
* assignment otherwise it's a declaration
*/
if (true == target.symbolMap.containsKey(target.symbolMapKey)
&& (target.parforSwitch == false || isReductionStmt == true)) {
isDecl = false;
AssignStmt assign_stmt = new AssignStmt();
// IDInfo LHSinfo = new IDInfo();
// assign_stmt.setLHS(LHSinfo);
assign_stmt.setLHS(Helper.generateIDInfo(target.analysis, target.index, node, LHS));
assign_stmt
.getLHS()
.setName(((TIRAbstractAssignToVarStmt) node).getTargetName().getID().toString());
boolean tf = true;
if (null != assign_stmt.getLHS().getShape())
for (int i = 0; i < assign_stmt.getLHS().getShape().size(); i++) {
if (null != assign_stmt.getLHS().getShape().get(i))
tf &= ("1").equals(assign_stmt.getLHS().getShape().get(i).toString());
}
// tf=true => scalar
if (target.parforSwitch == false || isReductionStmt == true)
target.symbolMap.put(target.symbolMapKey, assign_stmt.getLHS());
setRHSValue(isDecl, assign_stmt, node, tf, target);
block.addStmt(assign_stmt);
if (target.parforSwitch == false || isReductionStmt == true)
target.symbolMap.put(target.symbolMapKey, assign_stmt.getLHS());
// TODO : Handle expressions of various types
// Set parent's value in various expressions
if (true == isReductionStmt) {
assign_stmt.setAtomic(true);
} else {
assign_stmt.setAtomic(false);
}
} else {
isDecl = true;
DeclStmt decl_stmt = new DeclStmt();
IDInfo LHSinfo = new IDInfo();
decl_stmt.setLHS(Helper.generateIDInfo(target.analysis, target.index, node, LHS));
decl_stmt.getLHS().setName(((TIRAbstractAssignToVarStmt) node).getLHS().getVarName());
boolean tf = true;
if (null != decl_stmt.getLHS().getShape()) {
for (int i = 0; i < decl_stmt.getLHS().getShape().size(); i++) {
if (null != decl_stmt.getLHS().getShape().get(i))
tf &= ("1").equals(decl_stmt.getLHS().getShape().get(i).toString());
}
}
setRHSValue(isDecl, decl_stmt, node, tf, target);
if (target.currentBlock.size() > 1
&& (target.parforSwitch == false || isReductionStmt == true)) {
target.symbolMap.put(target.symbolMapKey, decl_stmt.getLHS());
DeclStmt pseudoDecl = new DeclStmt();
pseudoDecl.setLHS(decl_stmt.getLHS());
target.currentBlock.get(0).addStmt(pseudoDecl);
AssignStmt pseudoAssign = new AssignStmt();
pseudoAssign.setLHS(decl_stmt.getLHS());
pseudoAssign.setRHS(decl_stmt.getRHS());
block.addStmt(pseudoAssign);
if (target.parforSwitch == false || isReductionStmt == true)
target.symbolMap.put(target.symbolMapKey, decl_stmt.getLHS());
}
//
else {
target.symbolMap.put(target.symbolMapKey, decl_stmt.getLHS());
// if (target.currentBlock.size() > 1)
// target.currentBlock.get(0).addStmt(decl_stmt);
// else
block.addStmt(decl_stmt);
if (target.parforSwitch == false || isReductionStmt == true)
target.symbolMap.put(target.symbolMapKey, decl_stmt.getLHS());
}
if (true == isReductionStmt) {
decl_stmt.setAtomic(true);
} else {
decl_stmt.setAtomic(false);
}
}
}
public static void handleTIRContinueStmt(
TIRContinueStmt node, IRx10ASTGenerator target, StmtBlock block) {
block.addStmt(new ContinueStmt());
}
public static void handleTIRAbstractAssignToListStmt(
TIRAbstractAssignStmt node, IRx10ASTGenerator target, StmtBlock block) {
boolean isReductionStmt = Helper.isReductionStmt(node, target, block);
if (1 == ((TIRAbstractAssignToListStmt) node).getTargets().asNameList().size()) {
String LHS;
boolean isDecl;
target.symbolMapKey = ((TIRAbstractAssignToListStmt) node).getTargetName().getID();
LHS = target.symbolMapKey;
if (true == target.symbolMap.containsKey(target.symbolMapKey)
&& (IRx10ASTGenerator.parforSwitch == false || isReductionStmt == true)) {
isDecl = false;
// IDInfo LHSinfo = new IDInfo();
// assign_stmt.setLHS(LHSinfo);
AssignStmt list_single_assign_stmt = new AssignStmt();
list_single_assign_stmt.setLHS(
Helper.generateIDInfo(target.analysis, target.index, node, LHS));
list_single_assign_stmt
.getLHS()
.setName(((TIRAbstractAssignToListStmt) node).getTargets().getChild(0).getVarName());
setRHSValue(false, list_single_assign_stmt, node, false, target);
target.symbolMap.put(target.symbolMapKey, list_single_assign_stmt.getLHS());
System.out.println("#####!" + target.symbolMapKey);
block.addStmt(list_single_assign_stmt);
if (target.parforSwitch == false || isReductionStmt == true)
target.symbolMap.put(target.symbolMapKey, list_single_assign_stmt.getLHS());
if (true == isReductionStmt) {
list_single_assign_stmt.setAtomic(true);
} else {
list_single_assign_stmt.setAtomic(false);
}
target.symbolMap.put(target.symbolMapKey, list_single_assign_stmt.getLHS());
} else {
isDecl = true;
DeclStmt decl_stmt = new DeclStmt();
IDInfo LHSinfo = new IDInfo();
decl_stmt.setLHS(Helper.generateIDInfo(target.analysis, target.index, node, LHS));
decl_stmt
.getLHS()
.setName(((TIRAbstractAssignToListStmt) node).getTargets().getChild(0).getVarName());
setRHSValue(isDecl, decl_stmt, node, false, target);
System.out.println("#####!" + target.symbolMapKey);
// block.addStmt(decl_stmt);
if (target.parforSwitch == false || isReductionStmt == true)
target.symbolMap.put(target.symbolMapKey, decl_stmt.getLHS());
DeclStmt pseudoDecl = new DeclStmt();
pseudoDecl.setLHS(decl_stmt.getLHS());
//
if (true == isReductionStmt) {
decl_stmt.setAtomic(true);
} else {
decl_stmt.setAtomic(false);
}
AssignStmt pseudoAssign = new AssignStmt();
pseudoAssign.setLHS(decl_stmt.getLHS());
pseudoAssign.setRHS(decl_stmt.getRHS());
if (true == isReductionStmt) {
pseudoAssign.setAtomic(true);
} else {
pseudoAssign.setAtomic(false);
}
if (target.currentBlock.size() > 1
&& (target.parforSwitch == false || isReductionStmt == true)) {
target.currentBlock.get(0).addStmt(pseudoDecl);
block.addStmt(pseudoAssign);
if (target.parforSwitch == false || isReductionStmt == true)
target.symbolMap.put(target.symbolMapKey, decl_stmt.getLHS());
// target.currentBlock.get(target.currentBlock.size()-1).addStmt(decl_stmt);
System.out.println(block.getParent().toString() + "^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^");
}
//
else {
block.addStmt(decl_stmt);
if (target.parforSwitch == false || isReductionStmt == true)
target.symbolMap.put(target.symbolMapKey, decl_stmt.getLHS());
target.symbolMap.put(target.symbolMapKey, decl_stmt.getLHS());
}
}
}
// TODO: Fix this to be handled in parfor
else { // this branch handles the case with multiple targets on LHS
String multiVarName = "";
AssignStmt list_assign_stmt = new AssignStmt();
MultiAssignLHS LHSinfo = new MultiAssignLHS();
list_assign_stmt.setMultiAssignLHS(LHSinfo);
for (ast.Name name : ((TIRAbstractAssignToListStmt) node).getTargets().asNameList()) {
list_assign_stmt
.getMultiAssignLHS()
.addIDInfo(Helper.generateIDInfo(target.analysis, target.index, node, name.getID()));
if (!target.symbolMap.containsKey(name.getID())) {
DeclStmt partDeclStmt = new DeclStmt();
partDeclStmt.setLHS(
Helper.generateIDInfo(target.analysis, target.index, node, name.getID()));
target.currentBlock.get(0).addStmt(partDeclStmt);
target.symbolMap.put(name.getID(), partDeclStmt.getLHS());
}
multiVarName = multiVarName + "_" + name.getID();
}
DeclStmt superDeclStmt = new DeclStmt();
superDeclStmt.setLHS(new IDInfo());
superDeclStmt.getLHS().setName(multiVarName);
ShapeFactory s = new ShapeFactory();
ArrayList<String> list_dims = new ArrayList<String>();
list_dims.add("1");
int num = ((TIRAbstractAssignToListStmt) node).getNumTargets();
list_dims.add(Integer.toString(num));
superDeclStmt.getLHS().setShape((list_dims));
superDeclStmt.getLHS().setType(new Type("Any"));
if (!target.symbolMap.containsKey(multiVarName)) {
target.currentBlock.get(0).addStmt(superDeclStmt);
target.symbolMap.put(multiVarName, superDeclStmt.getLHS());
}
AssignStmt pseudoAssign = new AssignStmt();
pseudoAssign.setLHS(superDeclStmt.getLHS());
setRHSValue(false, pseudoAssign, node, false, target);
System.out.println(
node.getRHS().getPrettyPrinted() + "#####################################");
block.addStmt(pseudoAssign);
target.symbolMap.put(multiVarName, pseudoAssign.getLHS());
ArrayAccess accessMultiRet;
for (int i = 0; i < list_assign_stmt.getMultiAssignLHS().getNumIDInfo(); i++) {
pseudoAssign = new AssignStmt();
accessMultiRet = new ArrayAccess();
pseudoAssign.setLHS(list_assign_stmt.getMultiAssignLHS().getIDInfo(i));
accessMultiRet.setArrayID(new IDUse(multiVarName));
accessMultiRet.getIndicess().add(new IntLiteral(Integer.toString(i)));
pseudoAssign.setRHS(accessMultiRet);
pseudoAssign.setTypeCast(true);
block.addStmt(pseudoAssign);
target.symbolMap.put(pseudoAssign.getLHS().getName(), pseudoAssign.getLHS());
}
}
}