Mercurial > repos > bgruening > imagej2_enhance_contrast

annotate jython_utils.py @ 0:5c1467b54342 draft default tip

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
author bgruening
date Sat, 18 Mar 2017 09:42:41 -0400
parents
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
1 import imagej2_base_utils
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
2 from ij import IJ
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
3
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
4 IMAGE_PLUS_IMAGE_TYPE_FIELD_VALUES = { '0':'GRAY8', '1':'GRAY16', '2':'GRAY32',
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
5 '3':'COLOR_256', '4':'COLOR_RGB' }
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
6
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
7 def asbool( val ):
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
8 return str( val ).lower() in [ 'yes', 'true' ]
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
9
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
10 def convert_before_saving_as_tiff( image_plus ):
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
11 # The bUnwarpJ plug-in produces TIFF image stacks consisting of 3
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
12 # slices which can be viewed in ImageJ. The 3 slices are: 1) the
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
13 # registered image, 2) the target image and 3) the black/white warp
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
14 # image. When running bUnwarpJ from the command line (as these
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
15 # Galaxy wrappers do) the initial call to IJ.openImage() (to open the
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
16 # registered source and target images produced by bUnwarpJ) in the
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
17 # tool's jython_script.py returns an ImagePlus object with a single
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
18 # slice which is the "generally undesired" slice 3 discussed above.
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
19 # However, a call to IJ.saveAs() will convert the single-slice TIFF
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
20 # into a 3-slice TIFF image stack (as described above) if the selected
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
21 # format for saving is TIFF. Galaxy supports only single-layered
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
22 # images, so to work around this behavior, we have to convert the
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
23 # image to something other than TIFF so that slices are eliminated.
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
24 # We can then convert back to TIFF for saving. There might be a way
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
25 # to do this without converting twice, but I spent a lot of time looking
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
26 # and I have yet to discover it.
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
27 tmp_dir = imagej2_base_utils.get_temp_dir()
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
28 tmp_out_png_path = imagej2_base_utils.get_temporary_image_path( tmp_dir, 'png' )
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
29 IJ.saveAs( image_plus, 'png', tmp_out_png_path )
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
30 return IJ.openImage( tmp_out_png_path )
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
31
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
32 def get_binary_options( black_background, iterations=1, count=1, pad_edges_when_eroding='no' ):
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
33 options = [ 'edm=Overwrite', 'iterations=%d' % iterations, 'count=%d' % count ]
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
34 if asbool( pad_edges_when_eroding ):
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
35 options.append( 'pad' )
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
36 if asbool( black_background ):
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
37 options.append( "black" )
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
38 return " ".join( options )
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
39
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
40 def get_display_image_type( image_type ):
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
41 return IMAGE_PLUS_IMAGE_TYPE_FIELD_VALUES.get( str( image_type ), None )
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
42
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
43 def handle_error( error_log, msg ):
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
44 # Java writes a lot of stuff to stderr, so the received error_log
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
45 # will log actual errors.
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
46 elh = open( error_log, 'wb' )
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
47 elh.write( msg )
5c1467b54342 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/image_processing/imagej2 commit b'7e5cd452018ae9507c2d1cd13dd688a747550393\n'
bgruening
parents:
diff changeset
48 elh.close()