Chromium Code Reviews| Index: webrtc/tools/barcode_tools/barcode_decoder.py |
| diff --git a/webrtc/tools/barcode_tools/barcode_decoder.py b/webrtc/tools/barcode_tools/barcode_decoder.py |
| index e615fa83b83c55a2a4da1e11e914eb94fb9cc494..4b71ad3178ad8b090e3f7d8e63e5f1fa0820ffba 100755 |
| --- a/webrtc/tools/barcode_tools/barcode_decoder.py |
| +++ b/webrtc/tools/barcode_tools/barcode_decoder.py |
| @@ -21,7 +21,7 @@ import helper_functions |
| sys.stderr = sys.stdout |
| -def convert_yuv_to_png_files(yuv_file_name, yuv_frame_width, yuv_frame_height, |
| +def ConvertYuvToPngFiles(yuv_file_name, yuv_frame_width, yuv_frame_height, |
| output_directory, ffmpeg_path): |
| """Converts a YUV video file into PNG frames. |
| @@ -61,7 +61,7 @@ def convert_yuv_to_png_files(yuv_file_name, yuv_frame_width, yuv_frame_height, |
| return True |
| -def decode_frames(input_directory, zxing_path): |
| +def DecodeFrames(input_directory, zxing_path): |
| """Decodes the barcodes overlaid in each frame. |
| The function uses the Zxing command-line tool from the Zxing C++ distribution |
| @@ -85,11 +85,11 @@ def decode_frames(input_directory, zxing_path): |
| print 'Decoding barcodes from PNG files with %s...' % zxing_path |
| return helper_functions.perform_action_on_all_files( |
| directory=input_directory, file_pattern='frame_', |
| - file_extension='png', start_number=1, action=_decode_barcode_in_file, |
| + file_extension='png', start_number=1, action=_DecodeBarcodeInFile, |
| command_line_decoder=zxing_path) |
| -def _decode_barcode_in_file(file_name, command_line_decoder): |
| +def _DecodeBarcodeInFile(file_name, command_line_decoder): |
| """Decodes the barcode in the upper left corner of a PNG file. |
| Args: |
| @@ -116,7 +116,7 @@ def _decode_barcode_in_file(file_name, command_line_decoder): |
| return True |
| -def _generate_stats_file(stats_file_name, input_directory='.'): |
| +def _GenerateStatsFile(stats_file_name, input_directory='.'): |
| """Generate statistics file. |
| The function generates a statistics file. The contents of the file are in the |
| @@ -128,7 +128,7 @@ def _generate_stats_file(stats_file_name, input_directory='.'): |
| stats_file = open(stats_file_name, 'w') |
| print 'Generating stats file: %s' % stats_file_name |
| - for i in range(1, _count_frames_in(input_directory=input_directory) + 1): |
| + for i in range(1, _CountFramesIn(input_directory=input_directory) + 1): |
| frame_number = helper_functions.zero_pad(i) |
| barcode_file_name = file_prefix + frame_number + '.txt' |
| png_frame = file_prefix + frame_number + '.png' |
| @@ -136,10 +136,10 @@ def _generate_stats_file(stats_file_name, input_directory='.'): |
| entry = 'frame_' + entry_frame_number + ' ' |
| if os.path.isfile(barcode_file_name): |
| - barcode = _read_barcode_from_text_file(barcode_file_name) |
| + barcode = _ReadBarcodeFromTextFile(barcode_file_name) |
| os.remove(barcode_file_name) |
| - if _check_barcode(barcode): |
| + if _CheckBarcode(barcode): |
| entry += (helper_functions.zero_pad(int(barcode[0:11])) + '\n') |
| else: |
| entry += 'Barcode error\n' # Barcode is wrongly detected. |
| @@ -152,7 +152,7 @@ def _generate_stats_file(stats_file_name, input_directory='.'): |
| stats_file.close() |
| -def _read_barcode_from_text_file(barcode_file_name): |
| +def _ReadBarcodeFromTextFile(barcode_file_name): |
| """Reads the decoded barcode for a .txt file. |
| Args: |
| @@ -166,7 +166,7 @@ def _read_barcode_from_text_file(barcode_file_name): |
| return barcode |
| -def _check_barcode(barcode): |
| +def _CheckBarcode(barcode): |
| """Check weather the UPC-A barcode was decoded correctly. |
| This function calculates the check digit of the provided barcode and compares |
| @@ -200,7 +200,7 @@ def _check_barcode(barcode): |
| return dsum == int(barcode[11]) |
| -def _count_frames_in(input_directory='.'): |
| +def _CountFramesIn(input_directory='.'): |
| """Calculates the number of frames in the input directory. |
| The function calculates the number of frames in the input directory. The |
| @@ -225,7 +225,7 @@ def _count_frames_in(input_directory='.'): |
| return num - 1 |
| -def _parse_args(): |
| +def _ParseArgs(): |
| """Registers the command-line options.""" |
| usage = "usage: %prog [options]" |
| parser = optparse.OptionParser(usage=usage) |
| @@ -256,7 +256,7 @@ def _parse_args(): |
| return options |
| -def _main(): |
| +def main(): |
| """The main function. |
| A simple invocation is: |
| @@ -265,10 +265,10 @@ def _main(): |
| --yuv_frame_width=640 --yuv_frame_height=480 |
| --stats_file=<path_and_name_to_stats_file> |
| """ |
| - options = _parse_args() |
| + options = _ParseArgs() |
| # Convert the overlaid YUV video into a set of PNG frames. |
| - if not convert_yuv_to_png_files(options.yuv_file, options.yuv_frame_width, |
| + if not ConvertYuvToPngFiles(options.yuv_file, options.yuv_frame_width, |
| options.yuv_frame_height, |
| output_directory=options.png_working_dir, |
| ffmpeg_path=options.ffmpeg_path): |
|
oprypin_webrtc
2017/03/09 08:41:49
Misaligned
kjellander_webrtc
2017/03/09 08:58:13
Done.
|
| @@ -276,16 +276,16 @@ def _main(): |
| return -1 |
| # Decode the barcodes from the PNG frames. |
| - if not decode_frames(input_directory=options.png_working_dir, |
| + if not DecodeFrames(input_directory=options.png_working_dir, |
| zxing_path=options.zxing_path): |
| print 'An error occurred decoding barcodes from PNG frames.' |
| return -2 |
| # Generate statistics file. |
| - _generate_stats_file(options.stats_file, |
| + _GenerateStatsFile(options.stats_file, |
| input_directory=options.png_working_dir) |
|
oprypin_webrtc
2017/03/09 08:41:49
Misaligned here and above
kjellander_webrtc
2017/03/09 08:58:13
Done.
|
| print 'Completed barcode decoding.' |
| return 0 |
| if __name__ == '__main__': |
| - sys.exit(_main()) |
| + sys.exit(main()) |