0
|
1 #!/usr/bin/env python
|
|
2 import argparse
|
|
3 import os
|
|
4 import shutil
|
|
5 import subprocess
|
|
6 import tempfile
|
|
7 import imagej2_base_utils
|
|
8
|
|
9 parser = argparse.ArgumentParser()
|
|
10 parser.add_argument( '--input', dest='input', help='Path to the input file' )
|
|
11 parser.add_argument( '--input_datatype', dest='input_datatype', help='Datatype of the input image' )
|
|
12 parser.add_argument( '--threshold_min', dest='threshold_min', type=float, help='Minimum threshold value' )
|
|
13 parser.add_argument( '--threshold_max', dest='threshold_max', type=float, help='Maximum threshold value' )
|
|
14 parser.add_argument( '--method', dest='method', help='Threshold method' )
|
|
15 parser.add_argument( '--display', dest='display', help='Display mode' )
|
|
16 parser.add_argument( '--black_background', dest='black_background', help='Black background' )
|
|
17 parser.add_argument( '--stack_histogram', dest='stack_histogram', help='Stack histogram' )
|
|
18 parser.add_argument( '--jython_script', dest='jython_script', help='Path to the Jython script' )
|
|
19 parser.add_argument( '--output', dest='output', help='Path to the output file' )
|
|
20 parser.add_argument( '--output_datatype', dest='output_datatype', help='Datatype of the output image' )
|
|
21 args = parser.parse_args()
|
|
22
|
|
23 tmp_dir = imagej2_base_utils.get_temp_dir()
|
|
24 # ImageJ expects valid image file extensions, so the Galaxy .dat extension does not
|
|
25 # work for some features. The following creates a symlink with an appropriate file
|
|
26 # extension that points to the Galaxy dataset. This symlink is used by ImageJ.
|
|
27 tmp_input_path = imagej2_base_utils.get_input_image_path( tmp_dir, args.input, args.input_datatype )
|
|
28 tmp_output_path = imagej2_base_utils.get_temporary_image_path( tmp_dir, args.output_datatype )
|
|
29 # Define command response buffers.
|
|
30 tmp_out = tempfile.NamedTemporaryFile().name
|
|
31 tmp_stdout = open( tmp_out, 'wb' )
|
|
32 tmp_err = tempfile.NamedTemporaryFile().name
|
|
33 tmp_stderr = open( tmp_err, 'wb' )
|
|
34 # Java writes a lot of stuff to stderr, so we'll specify a file for handling actual errors.
|
|
35 error_log = tempfile.NamedTemporaryFile( delete=False ).name
|
|
36 # Build the command line.
|
|
37 cmd = imagej2_base_utils.get_base_command_imagej2( None, jython_script=args.jython_script )
|
|
38 if cmd is None:
|
|
39 imagej2_base_utils.stop_err( "ImageJ not found!" )
|
|
40 cmd += ' %s' % error_log
|
|
41 cmd += ' %s' % tmp_input_path
|
|
42 cmd += ' %.3f' % args.threshold_min
|
|
43 cmd += ' %.3f' % args.threshold_max
|
|
44 cmd += ' %s' % args.method
|
|
45 cmd += ' %s' % args.display
|
|
46 cmd += ' %s' % args.black_background
|
|
47 cmd += ' %s' % args.stack_histogram
|
|
48 cmd += ' %s' % tmp_output_path
|
|
49 cmd += ' %s' % args.output_datatype
|
|
50 # Run the command.
|
|
51 proc = subprocess.Popen( args=cmd, stderr=tmp_stderr, stdout=tmp_stdout, shell=True )
|
|
52 rc = proc.wait()
|
|
53 # Handle execution errors.
|
|
54 if rc != 0:
|
|
55 error_message = imagej2_base_utils.get_stderr_exception( tmp_err, tmp_stderr, tmp_out, tmp_stdout )
|
|
56 imagej2_base_utils.stop_err( error_message )
|
|
57 # Handle processing errors.
|
|
58 if os.path.getsize( error_log ) > 0:
|
|
59 error_message = open( error_log, 'r' ).read()
|
|
60 imagej2_base_utils.stop_err( error_message )
|
|
61 # Save the output image.
|
|
62 shutil.move( tmp_output_path, args.output )
|
|
63 imagej2_base_utils.cleanup_before_exit( tmp_dir )
|