pascalabcnet/SyntaxVisitors/BaseVisitors/SmallHelperVisitors.cs

// Copyright (c) Ivan Bondarev, Stanislav Mihalkovich (for details please see \doc\copyright.txt)
// This code is distributed under the GNU LGPL (for details please see \doc\license.txt)
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

using PascalABCCompiler;
using PascalABCCompiler.SyntaxTree;

namespace SyntaxVisitors
{
    public class FindLocalDefsVisitor : WalkingVisitorNew // Запускать только для подпрограмм
    {
        public ISet<string> ids = new HashSet<string>();
        private bool indef = false;
        public override void visit(ident id)
        {
            if (indef)
                ids.Add(id.name);
        }
        public override void visit(var_def_statement defs)
        {
            indef = true;
            ProcessNode(defs.vars); // исключаем типы - просматриваем только имена переменных
            indef = false;
        }

        public void Print()
        {
            foreach (var x in ids)
                Console.Write(x + ", ");
            Console.WriteLine();
        }
    }

    public class FindMainIdentsVisitor : WalkingVisitorNew  // в выражении yield это надо будет. 
    {
        public HashSet<string> vars = new HashSet<string>();
        public override void visit(ident id)
        {
            vars.Add(id.name);
        }
        public override void visit(dot_node dn)
        {
            ProcessNode(dn.left);
            if (dn.right.GetType() != typeof(ident))
                ProcessNode(dn.right);
        }
    }

    public class ExprHasNameVisitor : WalkingVisitorNew  // есть ли в выражении переменная с данным именем (не включая вложенные лямбды) (используется для поиска Result)
    {
        private string varname;
        public ident id = null;
        public static ident HasName(expression ex, string varname)
        {
            var v = new ExprHasNameVisitor(varname);
            v.ProcessNode(ex);
            return v.id;
        }
        public ExprHasNameVisitor(string varname)
        {
            this.varname = varname.ToLower();
        }
        public override void visit(ident i)
        {
            if (i.name.ToLower() == varname)
                id = i;
        }
        public override void visit(dot_node dn)
        {
            ProcessNode(dn.left);
            if (dn.right.GetType() != typeof(ident))
                ProcessNode(dn.right);
        }
        public override void visit(function_lambda_definition fd)
        {            
        }
    }

}