2 ##########################################################################
4 # Copyright 2011 Jose Fonseca
7 # Permission is hereby granted, free of charge, to any person obtaining a copy
8 # of this software and associated documentation files (the "Software"), to deal
9 # in the Software without restriction, including without limitation the rights
10 # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
11 # copies of the Software, and to permit persons to whom the Software is
12 # furnished to do so, subject to the following conditions:
14 # The above copyright notice and this permission notice shall be included in
15 # all copies or substantial portions of the Software.
17 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
18 # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
19 # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
20 # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
21 # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
22 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
25 ##########################################################################/
39 def __init__(self, trace, calls):
40 self.output = tempfile.NamedTemporaryFile()
52 self.dump = subprocess.Popen(
55 universal_newlines = True,
59 if platform.system() == 'Windows':
65 start_delete = '\33[9m\33[31m'
67 start_insert = '\33[32m'
71 def diff(ref_trace, src_trace):
73 ref_dumper = Dumper(ref_trace, options.ref_calls)
74 src_dumper = Dumper(src_trace, options.src_calls)
76 # TODO use difflib instead
77 if options.diff == 'diff':
80 '--speed-large-files',
81 '--old-line-format=' + start_delete + '%l' + end_delete + '\n',
82 '--new-line-format=' + start_insert + '%l' + end_insert + '\n',
84 elif options.diff == 'sdiff':
87 '--width=%u' % options.width,
88 '--speed-large-files',
90 elif options.diff == 'wdiff':
95 '--start-delete=' + start_delete,
96 '--end-delete=' + end_delete,
97 '--start-insert=' + start_insert,
98 '--end-insert=' + end_insert,
102 diff_args += [ref_dumper.output.name, src_dumper.output.name]
104 ref_dumper.dump.wait()
105 src_dumper.dump.wait()
108 if sys.stdout.isatty():
109 less = subprocess.Popen(
110 args = ['less', '-FRXn'],
111 stdin = subprocess.PIPE
114 diff_stdout = less.stdin
118 diff = subprocess.Popen(
120 stdout = diff_stdout,
121 universal_newlines = True,
131 def which(executable):
132 '''Search for the executable on the PATH.'''
134 if platform.system() == 'Windows':
138 dirs = os.environ['PATH'].split(os.path.pathsep)
140 path = os.path.join(dir, executable)
142 if os.path.exists(path + ext):
150 return curses.tigetnum('cols')
157 # Determine default options
158 default_width = columns()
160 # Parse command line options
161 optparser = optparse.OptionParser(
162 usage='\n\t%prog [options] -- TRACE_FILE TRACE_FILE',
164 optparser.add_option(
165 '-a', '--apitrace', metavar='PROGRAM',
166 type='string', dest='apitrace', default='apitrace',
167 help='apitrace command [default: %default]')
168 optparser.add_option(
170 type="choice", choices=('diff', 'sdiff', 'wdiff'),
171 dest="diff", default=None,
172 help="diff program: wdiff, sdiff, or diff [default: auto]")
173 optparser.add_option(
174 '-c', '--calls', metavar='CALLSET',
175 type="string", dest="calls", default='1-10000',
176 help="calls to compare [default: %default]")
177 optparser.add_option(
178 '--ref-calls', metavar='CALLSET',
179 type="string", dest="ref_calls", default=None,
180 help="calls to compare from reference trace")
181 optparser.add_option(
182 '--src-calls', metavar='CALLSET',
183 type="string", dest="src_calls", default=None,
184 help="calls to compare from source trace")
185 optparser.add_option(
186 '-w', '--width', metavar='NUM',
187 type="int", dest="width", default=default_width,
188 help="columns [default: %default]")
191 (options, args) = optparser.parse_args(sys.argv[1:])
193 optparser.error("incorrect number of arguments")
195 if options.diff is None:
197 options.diff = 'wdiff'
199 sys.stderr.write('warning: wdiff not found\n')
201 options.diff = 'sdiff'
203 sys.stderr.write('warning: sdiff not found\n')
204 options.diff = 'diff'
206 if options.ref_calls is None:
207 options.ref_calls = options.calls
208 if options.src_calls is None:
209 options.src_calls = options.calls
214 if __name__ == '__main__':