Callgraph.h

Go to the documentation of this file.

/******************************************************************************
 *
 * ROSE Global Database Test
 *
 * Callgraph implementation
 *
 *****************************************************************************/


#ifndef CALLGRAPH_H
#define CALLGRAPH_H

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

#include "DatabaseGraph.h"
#include "Classhierarchy.h"

#include "TableDefinitions.h"
#ifndef TABLES_DEFINED
USE_TABLE_FUNCTIONS();
#endif

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

class callgraphVertexData {
        public:
                int dummy;
};

// install boost database graph property map types
namespace boost {
        enum vertex_callgraph_t { vertex_callgraph };
        BOOST_INSTALL_PROPERTY( vertex, callgraph );
}


class Callgraph : 
public DatabaseGraph<functionsRowdata, EdgeTypeEmpty, 
        vecS, vecS, bidirectionalS, property< vertex_callgraph_t, callgraphVertexData, GraphvizVertexProperty>, 
        GraphvizEdgeProperty, 
        GraphvizGraphProperty > 
{
        public:
                Callgraph( long pid, long type, GlobalDatabaseConnection *gdb ) :
                        DatabaseGraph<functionsRowdata, EdgeTypeEmpty, vecS, vecS, bidirectionalS, 
                        property< vertex_callgraph_t, callgraphVertexData, GraphvizVertexProperty>, 
                        GraphvizEdgeProperty, 
                        GraphvizGraphProperty >(pid,type,gdb)  { };

                ~Callgraph() { };

        private:
};




/******************************************************************************
 * callgraph generation traversal classes
 *****************************************************************************/

#define SIDEEFFECTS_UNDECIDED 1
#define SIDEEFFECTS_NO                          2
#define SIDEEFFECTS_YES                         3


#define FUNCCALL_IMPLICIT 1
#define FUNCCALL_EXPLICIT 2


// Build an inherited attribute for the callgraph traversal
class CallgraphInhAttr
{
        public:
                CallgraphInhAttr ( SgNode* astNode ) :
                        mFunctionScopes() { };

                CallgraphInhAttr ( const CallgraphInhAttr & X, SgNode* astNode ) { 
                        mFunctionScopes = X.mFunctionScopes;
                };

                bool hasValidFunctionScope( void ){ 
                        return ( mFunctionScopes.size() > 0); }

                long getCurrentFunctionScope( void ){ 
                        assert( mFunctionScopes.size() > 0);
                        return mFunctionScopes[ mFunctionScopes.size()-1 ];
                }
                long pushFunctionScope(long id){ mFunctionScopes.push_back( id ); }

        private:

                vector<long> mFunctionScopes;
                
};




// tree traversal for callgraph generation
class CallgraphTraversal
: public SgTopDownProcessing<CallgraphInhAttr>
{
        public:
                CallgraphTraversal ( SgProject & project )
                        : SgTopDownProcessing<CallgraphInhAttr>() 
                {
                        mProjectId = -1;
                        mFileId = -1;
                        mCallgraph = NULL;
                        mClasshierarchy = NULL;
                        mpFunctionDefinitions = NULL;
                        mpSgProject = NULL;
                };

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


                void removeDeletedFunctionRows( void );


                void setFunctionDefinitions( vector<long> *set ){ mpFunctionDefinitions = set; };
                vector<long> *getFunctionDefinitions( void ) { return mpFunctionDefinitions; };

                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 setCallgraph( DatabaseGraph<functionsRowdata, EdgeTypeEmpty> *set ) { mCallgraph = set; }
                void setCallgraph( Callgraph *set ) { mCallgraph = set; }
                Callgraph *getCallgraph( void ) { return mCallgraph; }

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

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

                set<int> *getCGFileSubgraphs( void ){ return &mCGFileSubgraphs; }

                void setDeclList( list<SgNode*> *set ) { mpDeclQList = set; }
                

        private:

                void addCallEdge(SgFunctionDeclaration *funcDec, long functionScope);
                

                // ! DB connection pointer
                GlobalDatabaseConnection *mpGDB;

                long mProjectId;
                
                long mFileId;
                
                //DatabaseGraph<functionsRowdata, EdgeTypeEmpty> *mCallgraph;
                Callgraph *mCallgraph;
                
                Classhierarchy *mClasshierarchy;

                vector<long> *mpFunctionDefinitions;

                vector<long> *mpFileScopes;

                SgProject *mpSgProject;

                set<int> mCGFileSubgraphs;

                list<SgNode*> *mpDeclQList;

};



#endif