public FunctionDeclarationCG(Compiler.Nodes.FunctionDeclaration originalNode) : base(originalNode.CodeElement())
{
this.OriginalNode = originalNode;
//Check if we need to generate something special for this Procedure
bool needToGenerateParametersIntoLinkage = originalNode.CodeElement().Profile.InputParameters.Count + originalNode.CodeElement().Profile.InoutParameters.Count + originalNode.CodeElement().Profile.OutputParameters.Count +
(originalNode.CodeElement().Profile.ReturningParameter != null ? 1 : 0) > 0;
//we'll generate things for public call
var containsPublicCall = originalNode.ProcStyleCalls != null && originalNode.ProcStyleCalls.Count > 0;
ProgramHashName = originalNode.Hash;
//Get procedure original name and truncate it to 22 chars if over.
OriginalProcName = originalNode.Name.Substring(0, Math.Min(originalNode.Name.Length, 22));
foreach (var child in originalNode.Children)
{
if (child is Compiler.Nodes.ProcedureDivision)
{
Compiler.Nodes.LinkageSection linkageSection = null;
Compiler.Nodes.DataDivision dataDivision = null;
//Create DataDivision and LinkageSection if needed
//DataDivision must be created before ProcedureDivision because Program Node doesn't manage order of their children
//DataDivision manage order of their children so it's ok
if (needToGenerateParametersIntoLinkage || containsPublicCall)
{
dataDivision = GetOrCreateNode <Compiler.Nodes.DataDivision>(originalNode, () => new DataDivision());
linkageSection = GetOrCreateNode <Compiler.Nodes.LinkageSection>(dataDivision, () => new LinkageSection(originalNode), dataDivision);
//declare procedure parameters into linkage
DeclareProceduresParametersIntoLinkage(originalNode, linkageSection, originalNode.Profile);
}
//Replace ProcedureDivision node with a new one and keep all original children
var sentences = new List <Node>();
foreach (var sentence in child.Children)
{
sentences.Add(sentence);
}
var pdiv = new ProcedureDivision(originalNode, sentences);
children.Add(pdiv);
//Generate code if this procedure call a public procedure in another source
if (containsPublicCall)
{
var workingStorageSection = GetOrCreateNode <Compiler.Nodes.WorkingStorageSection>(dataDivision, () => new WorkingStorageSection(originalNode), dataDivision);
ProgramImports imports = ProgramImportsAttribute.GetProgramImports(originalNode);
Node[] toAddRange =
{
new GeneratedNode2("01 TC-Call PIC X VALUE 'T'.", true),
new GeneratedNode2(" 88 TC-FirstCall VALUE 'T'.", true),
new GeneratedNode2(" 88 TC-NthCall VALUE 'F'", true),
new GeneratedNode2(" X'00' thru 'S'", true),
new GeneratedNode2(" 'U' thru X'FF'.", true)
};
workingStorageSection.AddRange(toAddRange, 0);
GenerateCodeToCallPublicProc(originalNode, pdiv, workingStorageSection, linkageSection);
}
}
else
{
if (child.CodeElement is FunctionDeclarationEnd)
{
children.Add(new ProgramEnd(new URI(ProgramHashName), OriginalProcName));
}
else
{
// TCRFUN_CODEGEN_NO_ADDITIONAL_DATA_SECTION
// TCRFUN_CODEGEN_DATA_SECTION_AS_IS
children.Add(child);
}
}
}
}
protected void GenerateCodeToCallPublicProc(FunctionDeclaration originalNode, ProcedureDivision procedureDivision, Compiler.Nodes.WorkingStorageSection workingStorageSection, Compiler.Nodes.LinkageSection linkageSection)
{
ProgramImports imports = ProgramImportsAttribute.GetProgramImports(originalNode);
foreach (var pgm in imports.Programs.Values)
{
workingStorageSection.Add(
new GeneratedNode2("01 TC-" + pgm.Name + " pic X(08) value '" + pgm.Name.ToUpperInvariant() + "'.\n", true), 0);
}
List <Node> toAddRange = new List <Node>();
toAddRange.Add(new GeneratedNode2("*Common to all librairies used by the program.", true));
toAddRange.Add(new GeneratedNode2("01 TC-Library-PntTab.", false));
toAddRange.Add(new GeneratedNode2(" 05 TC-Library-PntNbr PIC S9(04) COMP.", true));
toAddRange.Add(new GeneratedNode2(
" 05 TC-Library-Item OCCURS 1000\n DEPENDING ON TC-Library-PntNbr\n INDEXED BY TC-Library-Idx.",
false));
toAddRange.Add(new GeneratedNode2(" 10 TC-Library-Item-Idt PIC X(08).", true));
toAddRange.Add(new GeneratedNode2(" 10 TC-Library-Item-Pnt PROCEDURE-POINTER.", true));
foreach (var pgm in imports.Programs.Values)
{
foreach (var proc in pgm.Procedures.Values)
{
proc.IsNotByExternalPointer = true;
toAddRange.Add(new GeneratedNode2(" ", true));
toAddRange.Add(new GeneratedNode2("*To call program " + proc.Hash + proc.Name + " in module " + proc.ProcStyleCall.FunctionDeclaration.QualifiedName.Tail, false));
toAddRange.Add(new GeneratedNode2("*Which is generated code for " + proc.ProcStyleCall.FunctionDeclaration.QualifiedName, false));
toAddRange.Add(new GeneratedNode2("*Declared in source file " + proc.ProcStyleCall.FunctionDeclaration.CodeElement.TokenSource.SourceName, false));
toAddRange.Add(new GeneratedNode2("01 TC-" + pgm.Name + "-" + proc.Hash + "-Item.", false));
toAddRange.Add(new GeneratedNode2(" 05 TC-" + pgm.Name + "-" + proc.Hash + "-Idt PIC X(08).", true));
toAddRange.Add(new GeneratedNode2(" 05 TC-" + pgm.Name + "-" + proc.Hash + " PROCEDURE-POINTER.",
true));
}
}
linkageSection.AddRange(toAddRange, 0);
if (imports.HasPublicProcedures)
{
Node whereToGenerate;
//Generate a PERFORM, this must be the first instruction unless we have a Paragraph or a section
var firstChildOfPDiv = procedureDivision.Children.First();
if (firstChildOfPDiv is Section)
{
var temp = firstChildOfPDiv.Children.First();
if (temp is Paragraph)
{
whereToGenerate = temp;
}
else
{
whereToGenerate = firstChildOfPDiv;
}
}
else if (firstChildOfPDiv is Paragraph)
{
whereToGenerate = firstChildOfPDiv;
}
else
{
whereToGenerate = procedureDivision;
}
//After #655, TC-Initializations is not used
whereToGenerate.Add(new GeneratedNode2(" PERFORM TC-INITIALIZATIONS", true), 0);
//Generate "TC-Initializations" paragraph
procedureDivision.Add(
new GeneratedNode2("*=================================================================", true));
procedureDivision.Add(new ParagraphGen("TC-INITIALIZATIONS"));
procedureDivision.Add(
new GeneratedNode2("*=================================================================", true));
procedureDivision.Add(new GeneratedNode2(" IF TC-FirstCall", true));
procedureDivision.Add(new GeneratedNode2(" SET TC-NthCall TO TRUE", true));
foreach (var pgm in imports.Programs.Values)
{
foreach (var proc in pgm.Procedures.Values)
{
procedureDivision.Add(
new GeneratedNode2(
" SET ADDRESS OF TC-" + pgm.Name + "-" + proc.Hash + "-Item TO NULL", true));
}
}
procedureDivision.Add(new GeneratedNode2(" END-IF", true));
procedureDivision.Add(new GeneratedNode2(" .", true));
}
//Generate "TC-LOAD-POINTERS-" paragraph
foreach (var pgm in imports.Programs.Values)
{
procedureDivision.Add(new GeneratedNode2("*=================================================================", true));
procedureDivision.Add(new ParagraphGen("TC-LOAD-POINTERS-" + pgm.Name));
procedureDivision.Add(new GeneratedNode2("*=================================================================", true));
procedureDivision.Add(new GeneratedNode2(" CALL 'ZCALLPGM' USING TC-" + pgm.Name, true));
procedureDivision.Add(new GeneratedNode2(" ADDRESS OF TC-Library-PntTab", true));
procedureDivision.Add(new GeneratedNode2(" PERFORM VARYING TC-Library-Idx FROM 1 BY 1", true));
procedureDivision.Add(new GeneratedNode2(" UNTIL TC-Library-Idx > TC-Library-PntNbr", true));
procedureDivision.Add(new GeneratedNode2(" EVALUATE TC-Library-Item-Idt (TC-Library-Idx)", true));
foreach (var proc in pgm.Procedures.Values)
{
procedureDivision.Add(new GeneratedNode2(" WHEN '" + proc.Hash + "'", true));
procedureDivision.Add(new GeneratedNode2(" SET ADDRESS OF", true));
procedureDivision.Add(new GeneratedNode2(" TC-" + pgm.Name + "-" + proc.Hash + "-Item", true));
procedureDivision.Add(new GeneratedNode2(" TO ADDRESS OF", true));
procedureDivision.Add(new GeneratedNode2(" TC-Library-Item(TC-Library-Idx)", true));
}
procedureDivision.Add(new GeneratedNode2(" WHEN OTHER", true));
procedureDivision.Add(new GeneratedNode2(" CONTINUE", true));
procedureDivision.Add(new GeneratedNode2(" END-EVALUATE", true));
procedureDivision.Add(new GeneratedNode2(" END-PERFORM", true));
procedureDivision.Add(new GeneratedNode2(" .", true));
}
}