source: misc/tools/dependency_graphs/anuga_depgraph2dot.py @ 6115

"""ANUGA specific dependency graph

Usage:
    python anuga_dependencies.py <anuga_script>
"""

from depgraph2dot import pydepgraphdot
import sys, imp

class ANUGAdepgraph(pydepgraphdot):

    def toocommon(self,s,type):
        """ Return true if references to this module are uninteresting. 
            Such references do not get drawn. This is ANUGA specific.
        """    
        
        #imp types:
        #C_BUILTIN = 6
        #C_EXTENSION = 3
        #IMP_HOOK = 9
        #PKG_DIRECTORY = 5
        #PY_CODERESOURCE = 8
        #PY_COMPILED = 2
        #PY_FROZEN = 7
        #PY_RESOURCE = 4
        #PY_SOURCE = 1
        #SEARCH_ERROR = 0


        if s=='__main__':
            # references *to* __main__ are never interesting. omitting them means
            # that main floats to the top of the page
            return 1
            
        #if type in [imp.PY_SOURCE, imp.C_EXTENSION] and s.startswith('anuga'):
        if type in [imp.PY_SOURCE, imp.PKG_DIRECTORY] and s.startswith('anuga'):
        #if type in [imp.PKG_DIRECTORY] and s.startswith('anuga'):
            # Only consider ANUGA source modules and packages
            
            if s.startswith('anuga.config'): return True # Except config.py
            
            sys.stderr.write('Draw dependency to %s\n' % s)
            return False
                
        else:
            return True
        
        
    def weight(self,a,b):
        # Return the weight of the dependency from a to b. Higher weights
        # usually have shorter straighter edges. Return 1 if it has normal weight.
        # A value of 4 is usually good for ensuring that a related pair of modules 
        # are drawn next to each other. This is a default policy - please override.
        #
        if b.split('.')[-1].startswith('_'):
            # A module that starts with an underscore. You need a special reason to
            # import these (for example random imports _random), so draw them close
            # together
            return 4
        return 1            

    def alien(self,a,b):
        # Return non-zero if references to this module are strange, and should be drawn
        # extra-long. the value defines the length, in rank. This is also good for putting some
        # vertical space between seperate subsystems. This is a default policy - please override.
        #
        return 0

    def label(self,s):
        # Convert a module name to a formatted node label. 
        #
        
        # Remove leading 'anuga' in label
        I = s.split('.')
        if I[0] == 'anuga':
            I = I[1:]
        
        return '\\.\\n'.join(I)

            
        
def main():
    ANUGAdepgraph().main(sys.argv[1:])

if __name__=='__main__':
    main()