1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
|
#!/usr/bin/env python2
import os, sys
import re
def displayHelp():
print """
Usage: WhatModulesVTK.py vtkSourceTree applicationFile|applicationFolder
Generate a FindPackage(VTK COMPONENTS) that lists all modules
referenced by a set of files.
For example:
Running from the VTK source,
./Utilities/Maintenance/WhatModulesVTK.py . Filters/Modeling/Testing/Cxx/TestRotationalExtrusion.cxx
Produces
All modules referenced in the files:
find_package(VTK COMPONENTS
CommonCore
FiltersCore
FiltersModeling
FiltersSources
RenderingCore
RenderingOpenGL2
TestingCore
TestingRendering
)
Your application code includes 8 of 170 vtk modules.
"""
exit(0)
def IncludesToPaths(path):
'''
Build a dict that maps include files to paths.
'''
includeToPath = dict()
prog = re.compile(r"((?:vtk|QVTK).*\.h)")
for root, dirs, files in os.walk(path):
for f in files:
if prog.match(f):
includeFile = prog.findall(f)[0]
includeToPath[includeFile] = root
return includeToPath
def FindModules(path):
'''
Build a dict that maps paths to modules.
'''
pathToModule = dict()
fileProg = re.compile(r"vtk\.module$")
for root, dirs, files in os.walk(path):
for f in files:
if fileProg.match(f):
with open(os.path.join(root, f), "r") as fid:
contents = fid.read()
args = contents.split()
try:
idx = args.index('NAME')
except ValueError:
raise RuntimeError('%s is missing a NAME field' % os.path.join(root, f))
pathToModule[root] = args[idx + 1]
return pathToModule
def FindIncludes(path):
'''
Build a set that contains vtk includes.
'''
includes = set()
includeProg = re.compile(r"((?:vtk|QVTK).*\.h)")
with open(path, "r") as fid:
contents = fid.read()
incs = includeProg.findall(contents)
includes.update(incs)
return includes
def FindModuleFiles(path):
'''
Get a list of module files in the VTK directory.
'''
moduleFiles = [os.path.join(root, name)
for root, dirs, files in os.walk(path)
for name in files
if name == ("vtk.module")]
return moduleFiles
def MakeFindPackage(modules):
'''
Make a useful find_package command.
'''
# Print a useful cmake command
res = "find_package(VTK COMPONENTS\n"
for module in sorted(modules):
res += " " + module.replace('VTK::', '') + "\n"
res += ")"
return res
from pprint import pprint as pp
def main(vtkSourceDir, sourceFiles):
'''
Start the program
'''
# Generate dict's for mapping includes to modules
includesToPaths = IncludesToPaths(vtkSourceDir + "/")
pathsToModules = FindModules(vtkSourceDir + "/")
# Test to see if VTK source is provided
if len(pathsToModules) == 0:
raise IOError, vtkSourceDir +\
" is not a VTK source directory. It does not contain any vtk.module files."
# Parse the module files making a dictionary of each module and its
# dependencies or what it implements.
moduleFiles = FindModuleFiles(vtkSourceDir + "/")
# Build a set of includes for all command line files
allIncludes = set()
for f in sourceFiles:
if os.path.isfile(f):
allIncludes.update(FindIncludes(f))
else:
# We have a folder so look through all the files.
for path, dirs, files in os.walk(f):
for fn in files:
allIncludes.update(FindIncludes(os.path.join(path,fn)))
if len(allIncludes) == 0:
raise IOError, f + " does not exist"
# Build a set that contains all modules referenced in command line files
allModules = set()
for inc in allIncludes:
if inc in includesToPaths:
module = includesToPaths[inc]
if module in pathsToModules:
allModules.add(pathsToModules[includesToPaths[inc]])
modules = {'All modules referenced in the files:': allModules,
}
for k, v in modules.iteritems():
print k
print MakeFindPackage(v)
print "Your application code includes " + str(len(v)) +\
" of " + str(len(pathsToModules)) + " vtk modules."
if __name__ == '__main__':
if len(sys.argv) < 3:
displayHelp()
exit(0)
main(sys.argv[1], sys.argv[2:])
print('This program is deprecated.')
print('Please consider using FindNeededModules.py instead.')
|
