Classhierarchy.h

Go to the documentation of this file.

/******************************************************************************
 *
 * ROSE Global Database Test
 *
 * Classhierarchy implementation
 *
 *****************************************************************************/

#ifndef CLASSHIERARCHY_H
#define CLASSHIERARCHY_H

#include "/home/thuerey1/local/include/config.h"

#include "DatabaseGraph.h"
#ifndef TABLES_DEFINED
USE_TABLE_TYPES();
#endif

//typedef set<SgMemberFunctionType *> chVertexData;
//typedef set<SgNode *> chVertexData;

class chVertexData {
        public:
                set<SgNode *> defined;
                set<SgNode *> inherited;
                map<SgNode *, set<SgNode *> > multDeclarations;
};

// install boost database graph property map types
namespace boost {
        enum vertex_classhierarchy_t { vertex_classhierarchy };
        BOOST_INSTALL_PROPERTY( vertex, classhierarchy );
}


class Classhierarchy : 
public DatabaseGraph<typesRowdata, EdgeTypeEmpty, 
        setS, vecS, bidirectionalS, property< vertex_classhierarchy_t, chVertexData, GraphvizVertexProperty>, 
        GraphvizEdgeProperty, 
        GraphvizGraphProperty > 
{
        public:
                Classhierarchy( long pid, long type, GlobalDatabaseConnection *gdb ) :
                        DatabaseGraph<typesRowdata, EdgeTypeEmpty, setS, vecS, bidirectionalS, 
                        property< vertex_classhierarchy_t, chVertexData, GraphvizVertexProperty>, 
                        GraphvizEdgeProperty, 
                        GraphvizGraphProperty >(pid,type,gdb)  { };

                ~Classhierarchy() { };

                void inheritVirtualFunctions();

                vector<SgMemberFunctionDeclaration*> searchMemberFunctionCalls(SgMemberFunctionDeclaration*);

        private:
};




/******************************************************************************
 * traversal classes
 *****************************************************************************/


// Build an inherited attribute for the classhierarchy traversal
class ClasshierarchyInhAttr
{
        public:
                ClasshierarchyInhAttr ( SgNode* astNode ) { };

                ClasshierarchyInhAttr ( const ClasshierarchyInhAttr & X, SgNode* astNode ) { };

        private:

};


// tree traversal for class hierarchy generation
class ClasshierarchyTraversal
: public SgTopDownProcessing<ClasshierarchyInhAttr>
{
        public:
                ClasshierarchyTraversal ( SgProject & project )
                        : SgTopDownProcessing<ClasshierarchyInhAttr>() 
                {
                        mProjectId = -1;
                        mFileId = -1;
                        mClasshierarchy = NULL;
                        mpSgProject = NULL;
                };

                // Functions required by the tree traversal mechanism
                ClasshierarchyInhAttr evaluateInheritedAttribute (
                                SgNode* astNode,
                                ClasshierarchyInhAttr inheritedAttribute );


                void removeDeletedFunctionRows( void );


                void setFileScopes( vector<long> *set ){ mpFileScopes = set; };
                vector<long> *getFileScopes( void ) { return mpFileScopes; };

                void setDB( GlobalDatabaseConnection *set ){ mpGDB = set; };
                GlobalDatabaseConnection *getDB( void ) { return mpGDB; };

                void setProjectId( long set ){ mProjectId = set; };
                long getProjectId( void ){ return mProjectId; };

                void setFileId( long set ){ mFileId = set; };
                long getFileId( void ){ return mFileId; };

                void setClasshierarchy( Classhierarchy *set ) { mClasshierarchy = set; }
                Classhierarchy *getClasshierarchy( void ) { return mClasshierarchy; }

                void setSgProject( SgProject *set ) { mpSgProject = set; }
                SgProject *getSgProject( void ) { return mpSgProject; }


        private:
                
                // ! DB connection pointer
                GlobalDatabaseConnection *mpGDB;

                long mProjectId;
                
                long mFileId;
                
                Classhierarchy *mClasshierarchy;

                vector<long> *mpFileScopes;

                SgProject *mpSgProject;
};



#endif