public override void OutAStructDecl(AStructDecl node)
{
if (node.GetClassToken() != null && node.GetIntDim() == null)
{
node.Parent().RemoveChild(node);
}
base.OutAStructDecl(node);
}
public override void CaseAConstructorDecl(AConstructorDecl node)
{
AStructDecl str = Util.GetAncestor <AStructDecl>(node);
AEnrichmentDecl enrichment = Util.GetAncestor <AEnrichmentDecl>(node);
AMethodDecl replacer = new AMethodDecl(new APublicVisibilityModifier(), null, null, null, null, null, new AVoidType(new TVoid("void")),
node.GetName(), new ArrayList(), node.GetBlock());
replacer.GetName().Text += "_Constructor";
while (node.GetFormals().Count > 0)
{
replacer.GetFormals().Add(node.GetFormals()[0]);
}
//Move the method outside the struct
AASourceFile file = Util.GetAncestor <AASourceFile>(node);
if (str != null)
{
str.RemoveChild(node.Parent());
}
else
{
enrichment.RemoveChild(node);
}
int i = file.GetDecl().IndexOf(str ?? (PDecl)enrichment);
file.GetDecl().Insert(i /* + 1*/, replacer);
//Add the struct as a parameter
PType type;
if (str != null)
{
ANamedType structType = new ANamedType(new TIdentifier(str.GetName().Text), null);
finalTrans.data.StructTypeLinks[structType] = str;
type = structType;
}
else
{
type = Util.MakeClone(enrichment.GetType(), finalTrans.data);
}
finalTrans.data.ConstructorMap[node] = replacer;
structFormal = new AALocalDecl(new APublicVisibilityModifier(), null, null, null, null, new APointerType(new TStar("*"), type), new TIdentifier("currentStruct", replacer.GetName().Line, replacer.GetName().Pos), null);
replacer.GetFormals().Add(structFormal);
finalTrans.data.Methods.Add(new SharedData.DeclItem <AMethodDecl>(file, replacer));
//Add return stm
replacer.SetReturnType(new APointerType(new TStar("*"), Util.MakeClone(type, data)));
replacer.Apply(new TransformConstructorReturns(structFormal, data));
//Insert call to base constructor););
if (finalTrans.data.ConstructorBaseLinks.ContainsKey(node))
{
AMethodDecl baseConstructor = finalTrans.data.ConstructorMap[finalTrans.data.ConstructorBaseLinks[node]];
ASimpleInvokeExp invoke = new ASimpleInvokeExp(new TIdentifier(baseConstructor.GetName().Text), new ArrayList());
while (node.GetBaseArgs().Count > 0)
{
invoke.GetArgs().Add(node.GetBaseArgs()[0]);
}
AThisLvalue thisLvalue1 = new AThisLvalue(new TThis("this"));
ALvalueExp thisExp1 = new ALvalueExp(thisLvalue1);
invoke.GetArgs().Add(thisExp1);
AThisLvalue thisLvalue2 = new AThisLvalue(new TThis("this"));
AAssignmentExp assignExp = new AAssignmentExp(new TAssign("="), thisLvalue2, invoke);
ANamedType structType = new ANamedType(new TIdentifier(str.GetName().Text), null);
finalTrans.data.StructTypeLinks[structType] = str;
finalTrans.data.LvalueTypes[thisLvalue1] =
finalTrans.data.LvalueTypes[thisLvalue2] =
finalTrans.data.ExpTypes[thisExp1] =
finalTrans.data.ExpTypes[assignExp] =
finalTrans.data.ExpTypes[invoke] = new APointerType(new TStar("*"), structType);
//finalTrans.data.ExpTypes[invoke] = new AVoidType(new TVoid("void"));
finalTrans.data.SimpleMethodLinks[invoke] = baseConstructor;
((AABlock)replacer.GetBlock()).GetStatements().Insert(0, new AExpStm(new TSemicolon(";"), assignExp));
//Inline if base and current are two different kinds of pointer types (int/string)
AStructDecl baseStruct = null;
AConstructorDecl baseC = finalTrans.data.ConstructorBaseLinks[node];
foreach (KeyValuePair <AStructDecl, List <AConstructorDecl> > pair in finalTrans.data.StructConstructors)
{
bool found = false;
foreach (AConstructorDecl decl in pair.Value)
{
if (baseC == decl)
{
found = true;
break;
}
}
if (found)
{
baseStruct = pair.Key;
break;
}
}
if ((str.GetIntDim() == null) != (baseStruct.GetIntDim() == null))
{
//For the inilining, change the type to the type of the caller
AALocalDecl lastFormal = baseConstructor.GetFormals().OfType <AALocalDecl>().Last();
lastFormal.SetRef(new TRef("ref"));
APointerType oldType = (APointerType)lastFormal.GetType();
structType = new ANamedType(new TIdentifier(str.GetName().Text), null);
finalTrans.data.StructTypeLinks[structType] = str;
APointerType newType = new APointerType(new TStar("*"), structType);
lastFormal.SetType(newType);
foreach (
ALocalLvalue lvalue in
data.LocalLinks.Where(pair => pair.Value == lastFormal).Select(pair => pair.Key))
{
data.LvalueTypes[lvalue] = newType;
if (lvalue.Parent() is ALvalueExp)
{
data.ExpTypes[(PExp)lvalue.Parent()] = newType;
if (lvalue.Parent().Parent() is APointerLvalue)
{
data.LvalueTypes[(PLvalue)lvalue.Parent().Parent()] = newType.GetType();
}
}
}
FixInlineMethods.Inline(invoke, finalTrans);
lastFormal.SetRef(null);
foreach (
ALocalLvalue lvalue in
data.LocalLinks.Where(pair => pair.Value == lastFormal).Select(pair => pair.Key))
{
data.LvalueTypes[lvalue] = oldType;
if (lvalue.Parent() is ALvalueExp)
{
data.ExpTypes[(PExp)lvalue.Parent()] = oldType;
if (lvalue.Parent().Parent() is APointerLvalue)
{
data.LvalueTypes[(PLvalue)lvalue.Parent().Parent()] = oldType.GetType();
}
}
}
lastFormal.SetType(oldType);
}
//Inline it instead, Since the pointer implementations might not be the same (int vs string)
/*AMethodDecl baseConstructor = finalTrans.data.ConstructorMap[finalTrans.data.ConstructorBaseLinks[node]];
*
* AABlock localsBlock = new AABlock(new ArrayList(), new TRBrace("}"));
* ABlockStm cloneBlock = new ABlockStm(new TLBrace("{"), (PBlock) baseConstructor.GetBlock().Clone());
* Dictionary<AALocalDecl, PLvalue> localMap = new Dictionary<AALocalDecl, PLvalue>();
* for (int argNr = 0; argNr < baseConstructor.GetFormals().Count; argNr++)
* {
* AALocalDecl formal = (AALocalDecl) baseConstructor.GetFormals()[i];
* PExp arg;
* if (i < baseConstructor.GetFormals().Count - 1)
* arg = (PExp)node.GetBaseArgs()[i];
* else
* {
* AThisLvalue thisLvalue = new AThisLvalue(new TThis("this"));
* ALvalueExp thisExp = new ALvalueExp(thisLvalue);
*
* ANamedType structType = new ANamedType(new TIdentifier(str.GetName().Text), null);
* finalTrans.data.StructTypeLinks[structType] = str;
*
* finalTrans.data.LvalueTypes[thisLvalue] =
* finalTrans.data.ExpTypes[thisExp] = new APointerType(new TStar("*"), structType);
*
* arg = thisExp;
* }
*
* if (formal.GetRef() != null || formal.GetOut() != null)
* {
* //Use same variable
* localMap[formal] = ((ALvalueExp) arg).GetLvalue();
* }
* else
* {
* //Make a new variable
* AALocalDecl newLocal = new AALocalDecl(new APublicVisibilityModifier(), null, null, null, null,
* Util.MakeClone(formal.GetType(), finalTrans.data),
* new TIdentifier(formal.GetName().Text),
* Util.MakeClone(arg, data));
*
* ALocalLvalue newLocalRef = new ALocalLvalue(new TIdentifier(newLocal.GetName().Text));
*
* localMap[formal] = newLocalRef;
* data.LvalueTypes[newLocalRef] = newLocal.GetType();
* data.LocalLinks[newLocalRef] = newLocal;
*
* localsBlock.GetStatements().Add(new ALocalDeclStm(new TSemicolon(";"), newLocal));
* }
*
* }
*
* CloneMethod cloner = new CloneMethod(finalTrans.data, localMap, cloneBlock);
* baseConstructor.GetBlock().Apply(cloner);
*
* ((AABlock)cloneBlock.GetBlock()).GetStatements().Insert(0, new ABlockStm(new TLBrace("{"), localsBlock));
* ((AABlock)node.GetBlock()).GetStatements().Insert(0, cloneBlock);*/
}
//Fix refferences to other struct stuff);
base.CaseAMethodDecl(replacer);
//Add functionality to refference the current struct in a constructor
//Want to do it as a pointer type, since the constructer can only be called for pointer types
}
