ClassHierarchyGraph.h

Go to the documentation of this file.

#ifndef CLASS_HIERARCHY_GRAPH_H
#define CLASS_HIERARCHY_GRAPH_H

#include <vector>
#include <map>
#include <boost/unordered_set.hpp>

class ROSE_DLL_API ClassHierarchyWrapper
{
public:
    typedef boost::unordered_set<SgClassDefinition*> ClassDefSet;

    typedef boost::unordered_map<std::string, ClassDefSet> MangledNameToClassDefsMap;

private:


    MangledNameToClassDefsMap directParents;

    MangledNameToClassDefsMap directChildren;

    MangledNameToClassDefsMap ancestorClasses;

    MangledNameToClassDefsMap subclasses;

    SgIncidenceDirectedGraph* classGraph;

public:

    ClassHierarchyWrapper(SgNode *node);
    const ClassDefSet& getSubclasses(SgClassDefinition *) const;
    const ClassDefSet& getDirectSubclasses(SgClassDefinition *) const;
    const ClassDefSet& getAncestorClasses(SgClassDefinition *) const;

private:

    static void buildAncestorsMap(const MangledNameToClassDefsMap& parents, MangledNameToClassDefsMap& transitiveParents);
};


#endif