# HG changeset patch # User iuc # Date 1642249529 0 # Node ID cbd605a243361487ab31ca144921eae042d34681 # Parent 4aed704725899e2a24f69a96ba070c97972a42b7 "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/idr_download commit b68715960d1107593db13dd9e1dbd8d4b905cc6f" diff -r 4aed70472589 -r cbd605a24336 idr_download_by_ids.py --- a/idr_download_by_ids.py Wed Nov 24 21:01:02 2021 +0000 +++ b/idr_download_by_ids.py Sat Jan 15 12:25:29 2022 +0000 @@ -7,6 +7,7 @@ from tempfile import TemporaryDirectory from libtiff import TIFF +from omero.cli import cli_login from omero.gateway import BlitzGateway # noqa from omero.constants.namespaces import NSBULKANNOTATIONS # noqa @@ -111,6 +112,7 @@ def download_image_data( image_ids_or_dataset_id, dataset=False, + download_original=False, channel=None, z_stack=0, frame=0, coord=(0, 0), width=0, height=0, region_spec='rectangle', skip_failed=False, download_tar=False, omero_host='idr.openmicroscopy.org', omero_secured=False, config_file=None @@ -129,7 +131,7 @@ omero_username = 'public' omero_password = 'public' - if region_spec not in ['rectangle', 'center']: + if not download_original and region_spec not in ['rectangle', 'center']: raise ValueError( 'Got unknown value "{0}" as region_spec argument' .format(region_spec) @@ -224,124 +226,159 @@ 'database. Aborting!' .format(image_warning_id) ) + if not download_original: + try: + # try to extract image properties + # if anything goes wrong here skip the image + # or abort. + image_name = os.path.splitext(image.getName())[0] + image_warning_id = '{0} (ID: {1})'.format( + image_name, image_id + ) - try: - # try to extract image properties - # if anything goes wrong here skip the image - # or abort. - image_name = os.path.splitext(image.getName())[0] - image_warning_id = '{0} (ID: {1})'.format( - image_name, image_id - ) + if region_spec == 'rectangle': + tile = get_clipping_region(image, *coord, width, height) + elif region_spec == 'center': + tile = get_clipping_region( + image, + *_center_to_ul(*coord, width, height) + ) + + ori_z, z_stack = z_stack, confine_plane(image, z_stack) + ori_frame, frame = frame, confine_frame(image, frame) + num_channels = image.getSizeC() + if channel is None: + channel_index = 0 + else: + channel_index = find_channel_index(image, channel) + except Exception as e: + # respect skip_failed on unexpected errors + if skip_failed: + warn(str(e), image_warning_id, warn_skip=True) + continue + else: + raise - if region_spec == 'rectangle': - tile = get_clipping_region(image, *coord, width, height) - elif region_spec == 'center': - tile = get_clipping_region( - image, - *_center_to_ul(*coord, width, height) + # region sanity checks and warnings + if tile[2] < width or tile[3] < height: + # The downloaded image region will have smaller dimensions + # than the specified width x height. + warn( + 'Downloaded image dimensions ({0} x {1}) will be smaller ' + 'than the specified width and height ({2} x {3}).' + .format(tile[2], tile[3], width, height), + image_warning_id + ) + + # z-stack sanity checks and warnings + if z_stack != ori_z: + warn( + 'Specified image plane ({0}) is out of bounds. Using {1} ' + 'instead.' + .format(ori_z, z_stack), + image_warning_id + ) + + # frame sanity checks and warnings + if frame != ori_frame: + warn( + 'Specified image frame ({0}) is out of bounds. Using ' + 'frame {1} instead.' + .format(ori_frame, frame), + image_warning_id ) - ori_z, z_stack = z_stack, confine_plane(image, z_stack) - ori_frame, frame = frame, confine_frame(image, frame) - num_channels = image.getSizeC() + # channel index sanity checks and warnings if channel is None: - channel_index = 0 - else: - channel_index = find_channel_index(image, channel) - except Exception as e: - # respect skip_failed on unexpected errors - if skip_failed: - warn(str(e), image_warning_id, warn_skip=True) - continue + if num_channels > 1: + warn( + 'No specific channel selected for multi-channel ' + 'image. Using first of {0} channels.' + .format(num_channels), + image_warning_id + ) else: - raise - - # region sanity checks and warnings - if tile[2] < width or tile[3] < height: - # The downloaded image region will have smaller dimensions - # than the specified width x height. - warn( - 'Downloaded image dimensions ({0} x {1}) will be smaller ' - 'than the specified width and height ({2} x {3}).' - .format(tile[2], tile[3], width, height), - image_warning_id - ) + if channel_index == -1 or channel_index >= num_channels: + if skip_failed: + warn( + str(channel) + + ' is not a known channel name for this image.', + image_warning_id, + warn_skip=True + ) + continue + else: + raise ValueError( + '"{0}" is not a known channel name for image {1}. ' + 'Aborting!' + .format(channel, image_warning_id) + ) - # z-stack sanity checks and warnings - if z_stack != ori_z: - warn( - 'Specified image plane ({0}) is out of bounds. Using {1} ' - 'instead.' - .format(ori_z, z_stack), - image_warning_id + # download and save the region as TIFF + fname = '__'.join( + [image_name, str(image_id)] + [str(x) for x in tile] ) + try: + if fname[-5:] != '.tiff': + fname += '.tiff' - # frame sanity checks and warnings - if frame != ori_frame: - warn( - 'Specified image frame ({0}) is out of bounds. Using ' - 'frame {1} instead.' - .format(ori_frame, frame), - image_warning_id - ) + fname = fname.replace(' ', '_') + + im_array = get_image_array(image, tile, z_stack, channel_index, frame) - # channel index sanity checks and warnings - if channel is None: - if num_channels > 1: - warn( - 'No specific channel selected for multi-channel ' - 'image. Using first of {0} channels.' - .format(num_channels), - image_warning_id - ) - else: - if channel_index == -1 or channel_index >= num_channels: + if download_tar: + fname = os.path.join(tempdir, fname) + try: + tiff = TIFF.open(fname, mode='w') + tiff.write_image(im_array) + finally: + tiff.close() + # move image into tarball + if download_tar: + archive.add(fname, os.path.basename(fname)) + os.remove(fname) + except Exception as e: if skip_failed: - warn( - str(channel) - + ' is not a known channel name for this image.', - image_warning_id, - warn_skip=True - ) + # respect skip_failed on unexpected errors + warn(str(e), image_warning_id, warn_skip=True) continue else: - raise ValueError( - '"{0}" is not a known channel name for image {1}. ' - 'Aborting!' - .format(channel, image_warning_id) - ) - - # download and save the region as TIFF - fname = '__'.join( - [image_name, str(image_id)] + [str(x) for x in tile] - ) - try: - if fname[-5:] != '.tiff': - fname += '.tiff' - - fname = fname.replace(' ', '_') - - im_array = get_image_array(image, tile, z_stack, channel_index, frame) - - if download_tar: - fname = os.path.join(tempdir, fname) + raise + else: try: - tiff = TIFF.open(fname, mode='w') - tiff.write_image(im_array) - finally: - tiff.close() - # move image into tarball - if download_tar: - archive.add(fname, os.path.basename(fname)) - os.remove(fname) - except Exception as e: - if skip_failed: + # try to extract image properties + # if anything goes wrong here skip the image + # or abort. + image_name = os.path.splitext(image.getName())[0] + image_warning_id = '{0} (ID: {1})'.format( + image_name, image_id + ) + original_image_name = image.getFileset().listFiles()[0].getName() + fname = image_name + "__" + str(image_id) + os.path.splitext(original_image_name)[1] + fname = fname.replace(' ', '_') + fname = fname.replace('/', '_') + download_directory = "./" + if download_tar: + download_directory = tempdir + with cli_login("-u", omero_username, "-s", omero_host, "-w", omero_password) as cli: + cli.invoke(["download", f"Image:{image_id}", download_directory]) + if cli.rv != 0: + raise Exception("Download failed.") + # This will download to download_directory/original_image_name + os.rename(os.path.join(download_directory, original_image_name), + os.path.join(download_directory, fname)) + # move image into tarball + if download_tar: + archive.add(os.path.join(download_directory, fname), + os.path.basename(fname)) + os.remove(os.path.join(download_directory, fname)) + except Exception as e: # respect skip_failed on unexpected errors - warn(str(e), image_warning_id, warn_skip=True) - continue - else: - raise + if skip_failed: + warn(str(e), image_warning_id, warn_skip=True) + continue + else: + raise def _center_to_ul(center_x, center_y, width, height): @@ -369,6 +406,10 @@ 'read ids from stdin).' ) p.add_argument( + '--download-original', dest='download_original', action='store_true', + help="download the original file uploaded to omero" + ) + p.add_argument( '-c', '--channel', help='name of the channel to retrieve data for ' '(note: the first channel of each image will be downloaded if ' diff -r 4aed70472589 -r cbd605a24336 idr_download_by_ids.xml --- a/idr_download_by_ids.xml Wed Nov 24 21:01:02 2021 +0000 +++ b/idr_download_by_ids.xml Sat Jan 15 12:25:29 2022 +0000 @@ -1,5 +1,5 @@ - + - download images from any OMERO instance using image IDs @@ -46,14 +46,18 @@ #if str($image_ids.source) == 'omeroDatasetID': --dataset #end if - #set $channel = str($channel).strip() - #if $channel: - -c '$channel' - #end if - -f $frame - -z $z_section - #if str($clip_image.select): - ${clip_image.select} ${clip_image.x_coord} ${clip_image.y_coord} ${clip_image.width} ${clip_image.height} + #if $image_region.original == "original": + --download-original + #else: + #set $channel = str($image_region.channel).strip() + #if $channel: + -c '$channel' + #end if + -f $image_region.frame + -z $image_region.z_section + #if str($image_region.clip_image.select): + ${image_region.clip_image.select} ${image_region.clip_image.x_coord} ${image_region.clip_image.y_coord} ${image_region.clip_image.width} ${image_region.clip_image.height} + #end if #end if $skip_failed $download_tar @@ -124,26 +128,36 @@ - - - - - - - - + + + + + - - - - - - + + + + + + + + + + + + + + + + + + + - - - - - - - + + + + + + + + + + + - @@ -191,15 +208,18 @@ - - - - - - - + + + + + + + + + + + - @@ -211,15 +231,18 @@ - - - - - - - + + + + + + + + + + + - @@ -231,12 +254,15 @@ - - - - + + + + + + + + - @@ -259,12 +285,15 @@ - - - - + + + + + + + + - @@ -275,12 +304,15 @@ - - - - + + + + + + + + - @@ -305,15 +337,18 @@ - - - - - - - + + + + + + + + + + + - @@ -335,15 +370,18 @@ - - - - - - - + + + + + + + + + + + - @@ -357,15 +395,18 @@ - - - - - - - + + + + + + + + + + + - @@ -402,6 +443,22 @@ + + + + + + + + + + + + + + + +