defsAndUsesTraversal.h

Go to the documentation of this file.

#pragma once
#include "rose.h"
#include "staticSingleAssignment.h"

namespace ssa_private
{

    class ChildUses
    {
    private:
        SgVarRefExp* currentVar;

        std::vector<SgNode*> uses;

    public:

        ChildUses() : currentVar(NULL)
        {
        }

        ChildUses(SgNode* useNode, SgVarRefExp* var)
        {
            uses.push_back(useNode);
            currentVar = var;
        }

        ChildUses(const std::vector<SgNode*>& useTree, SgVarRefExp* var = NULL)
        {
            if (useTree.size() > 0)
                uses.assign(useTree.begin(), useTree.end());
            currentVar = var;
        }

        std::vector<SgNode*>& getUses()
        {
            return uses;
        }

        void setUses(const std::vector<SgNode*>& newUses)
        {
            uses.assign(newUses.begin(), newUses.end());
        }

        SgVarRefExp* getCurrentVar() const
        {
            return currentVar;
        }
    };

    class DefsAndUsesTraversal : public AstBottomUpProcessing<ChildUses>
    {
        StaticSingleAssignment* ssa;

        const bool treatPointersAsStructs;
        
    public:

        DefsAndUsesTraversal(StaticSingleAssignment* ssa, bool treatPointersAsStructs = true) : ssa(ssa),
                treatPointersAsStructs(treatPointersAsStructs)
        {
        }

        virtual ChildUses evaluateSynthesizedAttribute(SgNode* node, SynthesizedAttributesList attrs);

    private:

        void addUsesToNode(SgNode* node, std::vector<SgNode*> uses);

        void addDefForVarAtNode(SgVarRefExp* currentVar, SgNode* defNode);
    };

} //namespace ssa_private