public override void visit(procedure_definition pd) { // frninja // Classification ISet <string> CollectedLocalsNames = new HashSet <string>(); ISet <string> CollectedFormalParamsNames = new HashSet <string>(); ISet <string> CollectedClassFieldsNames = new HashSet <string>(); ISet <string> CollectedUnitGlobalsNames = new HashSet <string>(); ISet <var_def_statement> CollectedLocals = new HashSet <var_def_statement>(); ISet <var_def_statement> CollectedFormalParams = new HashSet <var_def_statement>(); // Map from ident idName -> captured ident idName IDictionary <string, string> CapturedLocalsNamesMap = new Dictionary <string, string>(); IDictionary <string, string> CapturedFormalParamsNamesMap = new Dictionary <string, string>(); var dld = new DeleteAllLocalDefs(); // mids.vars - все захваченные переменные pd.visit(dld); // Удалить в локальных и блочных описаниях этой процедуры все переменные и вынести их в отдельный список var_def_statement // frninja 08/12/15 bool isInClassMethod; // Collect locals CollectedLocals.UnionWith(dld.LocalDeletedDefs); CollectedLocalsNames.UnionWith(dld.LocalDeletedDefs.SelectMany(vds => vds.vars.idents).Select(id => id.name)); // Collect formal params CollectFormalParams(pd, CollectedFormalParams); CollectFormalParamsNames(pd, CollectedFormalParamsNames); // Collect class fields CollectClassFieldsNames(pd, CollectedClassFieldsNames, out isInClassMethod); // Collect unit globals CollectUnitGlobalsNames(pd, CollectedUnitGlobalsNames); // Create maps :: idName -> captureName CreateCapturedLocalsNamesMap(CollectedLocalsNames, CapturedLocalsNamesMap); CreateCapturedFormalParamsNamesMap(CollectedFormalParamsNames, CapturedFormalParamsNamesMap); // AHAHA test! ReplaceCapturedVariablesVisitor rcapVis = new ReplaceCapturedVariablesVisitor( CollectedLocalsNames, CollectedFormalParamsNames, CollectedClassFieldsNames, new HashSet <string>(), new HashSet <string>(), CollectedUnitGlobalsNames, CapturedLocalsNamesMap, CapturedFormalParamsNamesMap, isInClassMethod ); // Replace (pd.proc_body as block).program_code.visit(rcapVis); }
public override void visit(procedure_definition pd) { // frninja // Classification ISet<string> CollectedLocalsNames = new HashSet<string>(); ISet<string> CollectedFormalParamsNames = new HashSet<string>(); ISet<string> CollectedClassFieldsNames = new HashSet<string>(); ISet<string> CollectedUnitGlobalsNames = new HashSet<string>(); ISet<var_def_statement> CollectedLocals = new HashSet<var_def_statement>(); ISet<var_def_statement> CollectedFormalParams = new HashSet<var_def_statement>(); // Map from ident idName -> captured ident idName IDictionary<string, string> CapturedLocalsNamesMap = new Dictionary<string, string>(); IDictionary<string, string> CapturedFormalParamsNamesMap = new Dictionary<string, string>(); var dld = new DeleteAllLocalDefs(); // mids.vars - все захваченные переменные pd.visit(dld); // Удалить в локальных и блочных описаниях этой процедуры все переменные и вынести их в отдельный список var_def_statement // frninja 08/12/15 bool isInClassMethod; // Collect locals CollectedLocals.UnionWith(dld.LocalDeletedDefs); CollectedLocalsNames.UnionWith(dld.LocalDeletedDefs.SelectMany(vds => vds.vars.idents).Select(id => id.name)); // Collect formal params CollectFormalParams(pd, CollectedFormalParams); CollectFormalParamsNames(pd, CollectedFormalParamsNames); // Collect class fields CollectClassFieldsNames(pd, CollectedClassFieldsNames, out isInClassMethod); // Collect unit globals CollectUnitGlobalsNames(pd, CollectedUnitGlobalsNames); // Create maps :: idName -> captureName CreateCapturedLocalsNamesMap(CollectedLocalsNames, CapturedLocalsNamesMap); CreateCapturedFormalParamsNamesMap(CollectedFormalParamsNames, CapturedFormalParamsNamesMap); // AHAHA test! ReplaceCapturedVariablesVisitor rcapVis = new ReplaceCapturedVariablesVisitor( CollectedLocalsNames, CollectedFormalParamsNames, CollectedClassFieldsNames, new HashSet<string>(), new HashSet<string>(), CollectedUnitGlobalsNames, CapturedLocalsNamesMap, CapturedFormalParamsNamesMap, isInClassMethod ); // Replace (pd.proc_body as block).program_code.visit(rcapVis); }
public override void visit(procedure_definition pd) { // frninja // DEBUG for test // SORRY // Classification ISet<string> CollectedLocalsNames = new HashSet<string>(); ISet<string> CollectedFormalParamsNames = new HashSet<string>(); ISet<string> CollectedClassFieldsNames = new HashSet<string>(); ISet<string> CollectedClassMethodsNames = new HashSet<string>(); ISet<string> CollectedClassPropertiesNames = new HashSet<string>(); ISet<string> CollectedUnitGlobalsNames = new HashSet<string>(); ISet<var_def_statement> CollectedLocals = new HashSet<var_def_statement>(); ISet<var_def_statement> CollectedFormalParams = new HashSet<var_def_statement>(); // Map from ident idName -> captured ident idName IDictionary<string, string> CapturedLocalsNamesMap = new Dictionary<string, string>(); IDictionary<string, string> CapturedFormalParamsNamesMap = new Dictionary<string, string>(); hasYields = false; if (pd.proc_header is function_header) mids = new FindMainIdentsVisitor(); base.visit(pd); if (!hasYields) // т.е. мы разобрали функцию и уже выходим. Это значит, что пока yield будет обрабатываться только в функциях. Так это и надо. return; LoweringVisitor.Accept(pd); // frninja 16/11/15: перенес ниже чтобы работал захват для lowered for var dld = new DeleteAllLocalDefs(); // mids.vars - все захваченные переменные pd.visit(dld); // Удалить в локальных и блочных описаниях этой процедуры все переменные и вынести их в отдельный список var_def_statement // frninja 08/12/15 bool isClassMethod = IsClassMethod(pd); // Collect locals CollectedLocals.UnionWith(dld.LocalDeletedDefs); CollectedLocalsNames.UnionWith(dld.LocalDeletedDefs.SelectMany(vds => vds.vars.idents).Select(id => id.name)); // Collect formal params CollectFormalParams(pd, CollectedFormalParams); CollectFormalParamsNames(pd, CollectedFormalParamsNames); // Collect class fields CollectClassFieldsNames(pd, CollectedClassFieldsNames); // Collect class methods CollectClassMethodsNames(pd, CollectedClassMethodsNames); // Collect class properties CollectClassPropertiesNames(pd, CollectedClassPropertiesNames); // Collect unit globals CollectUnitGlobalsNames(pd, CollectedUnitGlobalsNames); // Create maps :: idName -> captureName CreateCapturedLocalsNamesMap(CollectedLocalsNames, CapturedLocalsNamesMap); CreateCapturedFormalParamsNamesMap(CollectedFormalParamsNames, CapturedFormalParamsNamesMap); // AHAHA test! ReplaceCapturedVariablesVisitor rcapVis = new ReplaceCapturedVariablesVisitor( CollectedLocalsNames, CollectedFormalParamsNames, CollectedClassFieldsNames, CollectedClassMethodsNames, CollectedClassPropertiesNames, CollectedUnitGlobalsNames, CapturedLocalsNamesMap, CapturedFormalParamsNamesMap, isClassMethod ); // Replace (pd.proc_body as block).program_code.visit(rcapVis); mids.vars.Except(dld.LocalDeletedDefsNames); // параметры остались. Их тоже надо исключать - они и так будут обработаны // В результате работы в mids.vars что-то осталось. Это не локальные переменные и с ними непонятно что делать // Обработать параметры! // Как? Ищем в mids formal_parametrs, но надо выделить именно обращение к параметрам - не полям класса, не глобальным переменным var cfa = new ConstructFiniteAutomata((pd.proc_body as block).program_code); cfa.Transform(); (pd.proc_body as block).program_code = cfa.res; // Конструируем определение класса var cct = GenClassesForYield(pd, dld.LocalDeletedDefs, CapturedLocalsNamesMap, CapturedFormalParamsNamesMap); // все удаленные описания переменных делаем описанием класса //UpperNodeAs<declarations>().InsertBefore(pd, cct); if (isClassMethod) { var cd = UpperTo<class_definition>(); if ((object)cd != null) { var td = UpperTo<type_declarations>(); // Insert class predefenition! var iteratorClassPredef = new type_declaration(GetClassName(pd), new class_definition(null)); td.types_decl.Insert(0, iteratorClassPredef); foreach (var helperName in cct.types_decl.Select(ttd => ttd.type_name)) { var helperPredef = new type_declaration(helperName, new class_definition()); td.types_decl.Insert(0, helperPredef); } foreach (var helper in cct.types_decl) { td.types_decl.Add(helper); } //UpperTo<declarations>().InsertAfter(td, cct); } } else { UpperTo<declarations>().InsertBefore(pd, cct); } mids = null; // вдруг мы выйдем из процедуры, не зайдем в другую, а там - оператор! Такого конечно не может быть }
/// <summary> /// Захватываем имена в методе /// </summary> /// <param className="pd">Метод-итератор</param> /// <param className="deletedLocals">Коллекция удаленных локальных переменных</param> /// <param className="capturedLocalsNamesMap">Построенное отображение имен локальных переменных в захваченные имена</param> /// <param className="capturedFormalParamsNamesMap">Построенное отображение имен формальных параметров в захваченные имена</param> private void ReplaceCapturedVariables(procedure_definition pd, IEnumerable<var_def_statement> deletedLocals, out IDictionary<string, string> capturedLocalsNamesMap, out IDictionary<string, string> capturedFormalParamsNamesMap) { // Структуры данных под классификацию имен в методе // Classification ISet<string> CollectedLocalsNames = new HashSet<string>(); ISet<string> CollectedFormalParamsNames = new HashSet<string>(); ISet<string> CollectedClassFieldsNames = new HashSet<string>(); ISet<string> CollectedClassMethodsNames = new HashSet<string>(); ISet<string> CollectedClassPropertiesNames = new HashSet<string>(); ISet<string> CollectedUnitGlobalsNames = new HashSet<string>(); ISet<var_def_statement> CollectedLocals = new HashSet<var_def_statement>(); ISet<var_def_statement> CollectedFormalParams = new HashSet<var_def_statement>(); // Map from ident idName -> captured ident idName capturedLocalsNamesMap = new Dictionary<string, string>(); capturedFormalParamsNamesMap = new Dictionary<string, string>(); // Собираем инфу о именах // Collect locals CollectedLocals.UnionWith(deletedLocals); CollectedLocalsNames.UnionWith(deletedLocals.SelectMany(vds => vds.vars.idents).Select(id => id.name)); // Collect formal params CollectFormalParams(pd, CollectedFormalParams); CollectFormalParamsNames(pd, CollectedFormalParamsNames); // Collect class fields CollectClassFieldsNames(pd, CollectedClassFieldsNames); // Collect class methods CollectClassMethodsNames(pd, CollectedClassMethodsNames); // Collect class properties CollectClassPropertiesNames(pd, CollectedClassPropertiesNames); // Collect unit globals CollectUnitGlobalsNames(pd, CollectedUnitGlobalsNames); // Строим отображения для имён захваченных локальных переменных и формальных параметров // Create maps :: idName -> captureName CreateCapturedLocalsNamesMap(CollectedLocalsNames, capturedLocalsNamesMap); CreateCapturedFormalParamsNamesMap(CollectedFormalParamsNames, capturedFormalParamsNamesMap); // Выполняем замену захват имён в теле метода // AHAHA test! ReplaceCapturedVariablesVisitor rcapVis = new ReplaceCapturedVariablesVisitor( CollectedLocalsNames, CollectedFormalParamsNames, CollectedClassFieldsNames, CollectedClassMethodsNames, CollectedClassPropertiesNames, CollectedUnitGlobalsNames, capturedLocalsNamesMap, capturedFormalParamsNamesMap, IsClassMethod(pd), pd.proc_header.class_keyword, GetClassName(pd) ); // Replace (pd.proc_body as block).program_code.visit(rcapVis); }