bool IsConstant(PExp exp)
{
if (exp is ABinopExp)
{
ABinopExp aExp = (ABinopExp)exp;
return(IsConstant(aExp.GetLeft()) && IsConstant(aExp.GetRight()));
}
if (exp is AUnopExp)
{
AUnopExp aExp = (AUnopExp)exp;
return(IsConstant(aExp.GetExp()));
}
if (exp is AIncDecExp)
{
AIncDecExp aExp = (AIncDecExp)exp;
return(IsConstant(aExp.GetLvalue()));
}
if (exp is AIntConstExp || exp is AHexConstExp ||
exp is AOctalConstExp || exp is AFixedConstExp ||
exp is AStringConstExp || exp is ACharConstExp ||
exp is ABooleanConstExp || exp is ANullExp ||
exp is AAssignmentExp || exp is ADelegateExp)
{
return(true);
}
if (exp is ASimpleInvokeExp || exp is ANonstaticInvokeExp ||
exp is ASyncInvokeExp || exp is ANewExp ||
exp is ADelegateInvokeExp)
{
return(false);
}
if (exp is ALvalueExp)
{
ALvalueExp aExp = (ALvalueExp)exp;
return(IsConstant(aExp.GetLvalue()));
}
if (exp is AParenExp)
{
AParenExp aExp = (AParenExp)exp;
return(IsConstant(aExp.GetExp()));
}
if (exp is ACastExp)
{
ACastExp aExp = (ACastExp)exp;
return(IsConstant(aExp.GetExp()));
}
if (exp is AIfExp)
{
AIfExp aExp = (AIfExp)exp;
return(IsConstant(aExp.GetCond()) && IsConstant(aExp.GetThen()) && IsConstant(aExp.GetElse()));
}
if (exp == null)
{
return(false);
}
throw new Exception("Unexpected exp. Got " + exp);
}
public override void OutAIfExp(AIfExp node)
{
//Transform to
/*
* var expIfVar;
* if (<cond>)
* {
* expIfVar = <then>;
* }
* else
* {
* expIfVar = <else>;
* }
* ... expIfVar ...
*/
AALocalDecl expIfVarDecl = new AALocalDecl(new APublicVisibilityModifier(), null, null, null, null, Util.MakeClone(data.ExpTypes[node], data), new TIdentifier("expIfVar"), null);
ALocalLvalue thenExpIfVarLink = new ALocalLvalue(new TIdentifier("expIfVar"));
ALocalLvalue elseExpIfVarLink = new ALocalLvalue(new TIdentifier("expIfVar"));
ALocalLvalue usageExpIfVarLink = new ALocalLvalue(new TIdentifier("expIfVar"));
ALvalueExp usageExpIfVarLinkExp = new ALvalueExp(usageExpIfVarLink);
AAssignmentExp thenAssignment = new AAssignmentExp(new TAssign("="), thenExpIfVarLink, node.GetThen());
AAssignmentExp elseAssignment = new AAssignmentExp(new TAssign("="), elseExpIfVarLink, node.GetElse());
AIfThenElseStm ifStm = new AIfThenElseStm(new TLParen("("), node.GetCond(),
new ABlockStm(new TLBrace("{"),
new AABlock(
new ArrayList()
{
new AExpStm(new TSemicolon(";"),
thenAssignment)
},
new TRBrace("}"))),
new ABlockStm(new TLBrace("{"),
new AABlock(
new ArrayList()
{
new AExpStm(new TSemicolon(";"),
elseAssignment)
},
new TRBrace("}"))));
data.LocalLinks[thenExpIfVarLink] =
data.LocalLinks[elseExpIfVarLink] =
data.LocalLinks[usageExpIfVarLink] = expIfVarDecl;
data.LvalueTypes[thenExpIfVarLink] =
data.LvalueTypes[elseExpIfVarLink] =
data.LvalueTypes[usageExpIfVarLink] =
data.ExpTypes[usageExpIfVarLinkExp] =
data.ExpTypes[thenAssignment] =
data.ExpTypes[elseAssignment] = expIfVarDecl.GetType();
PStm pStm = Util.GetAncestor <PStm>(node);
AABlock pBlock = (AABlock)pStm.Parent();
pBlock.GetStatements().Insert(pBlock.GetStatements().IndexOf(pStm), new ALocalDeclStm(new TSemicolon(";"), expIfVarDecl));
pBlock.GetStatements().Insert(pBlock.GetStatements().IndexOf(pStm), ifStm);
node.ReplaceBy(usageExpIfVarLinkExp);
base.OutAIfExp(node);
}
private static void MakeCloneRefferences(PExp clone, PExp exp, SharedData data)
{
data.ExpTypes[clone] = data.ExpTypes[exp];
if (exp is AIntConstExp || exp is AHexConstExp || exp is AOctalConstExp ||
exp is AFixedConstExp || exp is AStringConstExp || exp is ACharConstExp ||
exp is ABooleanConstExp || exp is ANullExp)
{
//No more required
}
else if (exp is AIncDecExp)
{
AIncDecExp aExp = (AIncDecExp)exp;
AIncDecExp aClone = (AIncDecExp)clone;
MakeCloneRefferences(aClone.GetLvalue(), aExp.GetLvalue(), data);
}
else if (exp is ABinopExp)
{
ABinopExp aExp = (ABinopExp)exp;
ABinopExp aClone = (ABinopExp)clone;
MakeCloneRefferences(aClone.GetLeft(), aExp.GetLeft(), data);
MakeCloneRefferences(aClone.GetRight(), aExp.GetRight(), data);
}
else if (exp is AUnopExp)
{
AUnopExp aExp = (AUnopExp)exp;
AUnopExp aClone = (AUnopExp)clone;
MakeCloneRefferences(aClone.GetExp(), aExp.GetExp(), data);
}
else if (exp is ASimpleInvokeExp)
{
ASimpleInvokeExp aExp = (ASimpleInvokeExp)exp;
ASimpleInvokeExp aClone = (ASimpleInvokeExp)clone;
data.SimpleMethodLinks[aClone] = data.SimpleMethodLinks[aExp];
for (int i = 0; i < aExp.GetArgs().Count; i++)
{
MakeCloneRefferences((PExp)aClone.GetArgs()[i], (PExp)aExp.GetArgs()[i], data);
}
}
else if (exp is ANonstaticInvokeExp)
{
ANonstaticInvokeExp aExp = (ANonstaticInvokeExp)exp;
ANonstaticInvokeExp aClone = (ANonstaticInvokeExp)clone;
data.StructMethodLinks[aClone] = data.StructMethodLinks[aExp];
for (int i = 0; i < aExp.GetArgs().Count; i++)
{
MakeCloneRefferences((PExp)aClone.GetArgs()[i], (PExp)aExp.GetArgs()[i], data);
}
MakeCloneRefferences(aClone.GetReceiver(), aExp.GetReceiver(), data);
}
else if (exp is ALvalueExp)
{
ALvalueExp aExp = (ALvalueExp)exp;
ALvalueExp aClone = (ALvalueExp)clone;
MakeCloneRefferences(aClone.GetLvalue(), aExp.GetLvalue(), data);
}
else if (exp is AAssignmentExp)
{
AAssignmentExp aExp = (AAssignmentExp)exp;
AAssignmentExp aClone = (AAssignmentExp)clone;
MakeCloneRefferences(aClone.GetLvalue(), aExp.GetLvalue(), data);
MakeCloneRefferences(aClone.GetExp(), aExp.GetExp(), data);
}
else if (exp is AParenExp)
{
AParenExp aExp = (AParenExp)exp;
AParenExp aClone = (AParenExp)clone;
MakeCloneRefferences(aClone.GetExp(), aExp.GetExp(), data);
}
else if (exp is AStringConstExp)
{
AStringConstExp aExp = (AStringConstExp)exp;
AStringConstExp aClone = (AStringConstExp)clone;
if (data.ObfuscatedStrings.ContainsKey(aExp))
{
data.ObfuscatedStrings[aClone] = data.ObfuscatedStrings[aExp];
}
if (data.StringsDontJoinRight.Contains(aExp))
{
data.StringsDontJoinRight.Add(aClone);
}
}
else if (exp is ANewExp)
{
ANewExp aExp = (ANewExp)exp;
ANewExp aClone = (ANewExp)clone;
if (data.ConstructorLinks.ContainsKey(aExp))
{
data.ConstructorLinks[aClone] = data.ConstructorLinks[aExp];
}
MakeCloneRefferences(aClone.GetType(), aExp.GetType(), data);
for (int i = 0; i < aExp.GetArgs().Count; i++)
{
MakeCloneRefferences((PExp)aClone.GetArgs()[i], (PExp)aExp.GetArgs()[i], data);
}
}
else if (exp is ACastExp)
{
ACastExp aExp = (ACastExp)exp;
ACastExp aClone = (ACastExp)clone;
MakeCloneRefferences(aClone.GetType(), aExp.GetType(), data);
MakeCloneRefferences(aClone.GetExp(), aExp.GetExp(), data);
}
else if (exp is ADelegateExp)
{
ADelegateExp aExp = (ADelegateExp)exp;
ADelegateExp aClone = (ADelegateExp)clone;
if (data.DelegateCreationMethod.ContainsKey(aExp))
{
data.DelegateCreationMethod[aClone] = data.DelegateCreationMethod[aExp];
}
MakeCloneRefferences(aClone.GetType(), aExp.GetType(), data);
MakeCloneRefferences(aClone.GetLvalue(), aExp.GetLvalue(), data);
}
else if (exp is ADelegateInvokeExp)
{
ADelegateInvokeExp aExp = (ADelegateInvokeExp)exp;
ADelegateInvokeExp aClone = (ADelegateInvokeExp)clone;
MakeCloneRefferences(aClone.GetReceiver(), aExp.GetReceiver(), data);
for (int i = 0; i < aExp.GetArgs().Count; i++)
{
MakeCloneRefferences((PExp)aClone.GetArgs()[i], (PExp)aExp.GetArgs()[i], data);
}
}
else if (exp is AIfExp)
{
AIfExp aExp = (AIfExp)exp;
AIfExp aClone = (AIfExp)clone;
MakeCloneRefferences(aClone.GetCond(), aExp.GetCond(), data);
MakeCloneRefferences(aClone.GetThen(), aExp.GetThen(), data);
MakeCloneRefferences(aClone.GetElse(), aExp.GetElse(), data);
}
else
{
throw new Exception("Unexpect exp. Got " + exp.GetType());
}
}