dc@`sdZddlmZmZmZmZddlTddlZddlZddl Z ddl m Z ddl Z ddl m Zddl mZejd Zd Zd Zd Zdd ZdZdZdZdZedZdZdZdZdZ dZ!dZ"e#ddZ$de#dZ&dde#e#e#dZ'ee#eedZ(dZ)eje j*dZ+de,fd YZ-da.d!Z/d"Z0d#Z1dS($uAdditional functions dealing with GIMP objects (images, layers, etc.) not defined in the GIMP procedural database (PDB) or the GIMP Python API. i(tabsolute_importtdivisiontprint_functiontunicode_literals(t*N(tpdbi(t invocation(tutilscc`s+tj|z dVWdtj|XdS(u Wrap the enclosing block of code into one GIMP undo group for the specified image. Use this function as a context manager: with undo_group(image): # do stuff N(Rtgimp_image_undo_group_starttgimp_image_undo_group_end(timage((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pyt undo_groups  cC`sP|j |jdko%|jknoO|j |jdkoM|jkSS(u Return `True` if the layer is inside the image canvas (partially or completely). Return `False` if the layer is completely outside the image canvas. ii(twidthtoffsetstheight(R tlayer((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pytis_layer_inside_image's*cC`sT|}tj|j|j|j}tj|tj|tj|tj||jt j krtj |tj |ntj |\}}xB|D]:}|j|}|jtj|j|j|jqW|jdk rtj||jntj||jtjdkrPtj|}|rPtj||qPn|S(u Create a new image with metadata (dimensions, base type, parasites, etc.) copied from `image_to_copy_metadata_from`. Layers, channels or paths are not copied. For a full image copy, use `pdb.gimp_image_duplicate()`. ii N(ii (Rtgimp_image_newR Rt base_typetgimp_image_set_resolutiontgimp_image_get_resolutiontgimp_image_set_unittgimp_image_get_unitt gimpenumstINDEXEDtgimp_image_set_colormaptgimp_image_get_colormaptgimp_image_get_parasite_listt parasite_findtparasite_attachtgimptParasitetnametflagstdatatfilenametNonetgimp_image_set_filenametversiontgimp_image_get_metadatatgimp_image_set_metadata(timage_to_copy_metadata_fromR t new_imagetunused_tparasite_namesR tparasitetimage_string_metadata((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pytcreate_image_from_metadata2s& #cC`s5gtjD]$}tj|j|kr |^q S(uReturns a list of currently opened images in GIMP matching the given file path. Images are returned as `gimp.Image` instances. Matching is performed via the `gimp.Image.filename` attribute. (Rt image_listtpgutilstsafe_decode_gimpR#(timage_filepathR ((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pytfind_images_by_filepathZs cC`svt|}|rl|dkr:t|t|d}n%|dkr_t|t| }n||}nd}|S(uReturns the currently opened image in GIMP matching the given file path. If no match is found, `None` is returned. The image is returned as `gimp.Image` instance. Matching is performed via the `gimp.Image.filename` attribute. For multiple matches, the first matching image is returned by default. There may be multiple opened images from the same file path, but there is no way to tell which image is the one the user desires to work with. To adjust which image to return, pass a custom `index` value indicating the position to return. If the index is out of bounds, the highest possible index is returned given a positive value and the lowest possible index given a negative value. iiN(R4tmintlentmaxR$(R3tindextimagesR ((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pytfind_image_by_filepathhs    c`sCttdrtjStfdtjDdSdS(uReturns a currently opened image in GIMP matching the given ID. If no match is found, `None` is returned. The image is returned as `gimp.Image` instance. Matching is performed via the `gimp.Image.ID` attribute. u _id2imagec3`s$|]}|jkr|VqdS(N(tID(t.0R (timage_id(sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pys sN(thasattrRt _id2imagetnextR0R$(R=((R=sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pytfind_image_by_ids  cC`s|jtj}t|dkr(dSt|dt||}|dkrTdSt|dkrj|S|}d}x:|dD].}t||j}|dkrdS|}qW|S(uReturns a `gimp.Item` given the image, item class name and item path. The item class name corresponds to one of the GIMP item classes, e.g. `'Layer'` or `'Channel'`. The item path consists of the item name and all of its parent layer groups, separated by '/'. For example, if the item name is 'Left' its parent groups are 'Hands' (immediate parent) and 'Body (parent of 'Hands'), then the item path is 'Body/Hands/Left'. iiN(tsplitR1tGIMP_ITEM_PATH_SEPARATORR6R$t_find_item_by_name_in_childrent_get_children_from_imagetchildren(R titem_class_namet item_pathtitem_path_componentstmatching_image_childtparentt matching_itemtparent_or_item_name((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pyt!get_item_from_image_and_item_paths"    cC`s(x!|D]}|j|kr|SqWdS(N(R R$(t item_nameRFtchild((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pyRDs cC`sstt|}|tjtjfkr.|jS|tjkrD|jS|tjkrZ|jSt dj |dS(NuEinvalid item type "{}"; must be Layer, GroupLayer, Channel or Vectors( tgetattrRtLayert GroupLayertlayerstChanneltchannelstVectorstvectorst TypeErrortformat(R RGt item_type((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pyREscC`s|dkrdSg}|rZ|jdk rS|jjdk rS|j|jjqZdSntj|jjd}t|}tj j d||gD}|j ||g|S(uReturns item as a list of [item class name, item path] or [image file path, item class name, item path]. Item class name and item path are described in `get_item_from_image_and_item_path()`. uutf-8cs`s!|]}tj|jVqdS(N(R1R2R (R<tparent_or_item((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pys sN( R$R R#tappendR1t safe_decodet __class__t__name__t_get_item_parentsRCtjointextend(titemt include_imaget item_as_pathRGtparentsRH((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pytget_item_as_paths !  cC`s?g}|j}x)|dk r:|jd||j}qW|S(Ni(RKR$tinsert(RdRgtcurrent_parent((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pyRas   cC`s(x!|jD]}tj||q WdS(u1 Remove all layers from the specified image. N(RTRtgimp_image_remove_layer(R R((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pytremove_all_layersscC`s(x!|jD]}tj||q WdS(u1 Remove all layers from the specified image. N(RVRtgimp_image_remove_channel(R tchannel((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pytremove_all_channelsscC`s(x!|jD]}tj||q WdS(u: Remove all paths (vectors) from the specified image. N(RXRtgimp_image_remove_vectors(R tpath((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pytremove_all_pathsscC`s"t|t|t|dS(uJ Remove all items (layers, channels, paths) from the specified image. N(RlRoRr(R ((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pytremove_all_itemss  cC`s#tj|rtj|ndS(u\ Delete the specified image. If the image does not exist, do not raise any exception. N(Rtgimp_image_is_validtgimp_image_delete(R ((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pyttry_delete_imagescC`stj|tjj|}|t|jks?|dkrHd}ntj|j||}tjj||_|rtjj |jd|_ntj ||dt|jtj ||S(uA Load an image as a layer given its file path to an existing `image`. Return the layer. The layer is loaded at the end of the image. Layers names are basenames of the corresponding files. If `strip_file_extension` is `True`, remove the file extension from layer names. If the file contains multiple layers, specify the index of the desired layer to load. Only top-level layers are supported (i.e. not layers inside layer groups). If the index is greater than the number of layers in the loaded image or is negative, load and return the last layer. iiN( Rtgimp_file_loadtosRqtbasenameR6RTtgimp_layer_new_from_drawableR tsplitexttgimp_image_insert_layerR$Ru(tfilepathR tstrip_file_extensiontlayer_to_load_indext loaded_imageR((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pyt load_layer%s!  cC`sb|dk}|r'tjdd}nx|D]}t|||q.W|r^tj|n|S(up Load multiple layers to one image. Return the image. The layers are loaded at the end of the image. If `image` is `None`, create a new image. If `image` is not `None`, load the layers to the specified image. Layers names are basenames of the corresponding files. If `strip_file_extension` is `True`, remove the file extension from layer names. iN(R$RtImageRRtgimp_image_resize_to_layers(t filepathsR R~tcreate_new_imageR}((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pyt load_layersFs  cC`stj||}tj|||||rtj|tt|tjstjdkrrtj |tntj |tqn|rtj |t n|rtj |rt|}n|S(u Copy the specified layer into the specified image, parent layer group and position in the group. Return the copied layer. If `parent` is `None`, insert the layer in the main stack (outside of any layer group). If `remove_lock_attributes` is `True`, remove all lock-related attributes (lock position, alpha channel, etc.) for the layer copy. If `set_visible` is `True`, set the layer's visible state to `True`. If `merge_group` is `True` and the layer is a group, merge the group into a single layer. ii (ii (RRzR|tgimp_item_set_lock_contenttFalset isinstanceRRSR&tgimp_item_set_lock_positiontgimp_layer_set_lock_alphatgimp_item_set_visibletTruetgimp_item_is_grouptmerge_layer_group(RR RKtpositiontremove_lock_attributest set_visiblet merge_groupt layer_copy((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pytcopy_and_paste_layer_sc`sddd}fd}d}d}dfd} dd tfd d Dotfd d D} | stStfd d D} |r| rtStjd d tj} || } tj| || | ||t}x3| j D](}tj |r/t }| |q/q/W|| }|r|r|rx@| j D]5}|j dk r||qtj |tjqW|| }ntj| |S(uQ Return `True` if the contents of all specified layers are identical, `False` otherwise. Layer groups are also supported. The default values of the optional parameters correspond to how the layers are displayed in the image canvas. If `compare_alpha_channels` is `True`, perform comparison of alpha channels. If `compare_has_alpha` is `True`, compare the presence of alpha channels in all layers - if some layers have alpha channels and others do not, then do not perform full comparison and return `False`. If `apply_layer_attributes` is `True`, take the layer attributes (opacity, mode) into consideration when comparing, otherwise ignore them. If `apply_layer_masks` is `True`, apply layer masks if they are enabled. If the masks are disabled or `apply_layer_masks` is `False`, layer masks are ignored. icS`sptj|}tj||||x'|D]}t|||ddtq,Wx|jD]}t|_qYW|S(NiR(Rtgimp_layer_group_newR|RRRFtvisible(R RTRKRt layer_groupR((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pyt _copy_layerss  c`sx|jD]}tj|r.t|}q |jdksO|jtjkr|rj|||}qd|_tj|_n|jdk r |rtj |rtj |tj qtj |tj q q WdS(NgY@(RFRRRtopacitytmodeRt NORMAL_MODEtmaskR$tgimp_layer_get_apply_masktgimp_layer_remove_maskt MASK_APPLYt MASK_DISCARD(R Rtapply_layer_attributestapply_layer_masksR(t_apply_layer_attributes(sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pyt_process_layerss! cS`stj|jd_x|jdD]}t|_q!Wx|jdD]}t|_tjd kr~t j |tj d d}nt j |tj dd}|d}|dk}|stSt|_qBWtS( Niiii ig?ig(ii gp?( RtDIFFERENCE_MODERFRRRRRR&Rtgimp_drawable_histogramtHISTOGRAM_VALUEtgimp_histogram(RRthistogram_datat percentilet identical((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pyt _is_identicals      cS`sFtj|jtj|t}tj|tj|tjdS(N( Rtgimp_edit_copyRtgimp_edit_pasteRtgimp_floating_sel_anchorRRR(Rt floating_sel((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pyt_set_mask_to_layers cS`sKtj|}tj|||dtj|||dt|}|S(Ni(RRR|tgimp_image_reorder_itemR(R Rt parent_groupt temp_group((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pyRs  c`s||dS(N((R(t$_extract_alpha_channel_to_layer_maskt_remove_alpha_channel(sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pyt)_prepare_for_comparison_of_alpha_channelss cS`s9tj|tj}tj||tj|tdS(N(Rtgimp_layer_create_maskRtADD_ALPHA_MASKtgimp_layer_add_masktgimp_layer_set_apply_maskR(RR((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pyRscS`stj|dS(N(Rtgimp_layer_flatten(R((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pyRsc3`s%|]}dj|jkVqdS(iN(R (R<R(RT(sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pys sic3`s%|]}dj|jkVqdS(iN(R(R<R(RT(sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pys sc3`s%|]}dj|jkVqdS(iN(ttype(R<R(RT(sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pys sN(R$tallRRRRtRGBRRRFtgimp_drawable_has_alphaRRtgimp_drawable_fillt WHITE_FILLRu(RTtcompare_alpha_channelstcompare_has_alphaRRRRRRRtall_layers_have_same_sizetall_layers_are_same_image_typeR Rt has_alphaRR((RRRRTsJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pytcompare_layerssB              c C`s@tj|s*tdj|jn|j}t|d}|jdk r|jtj ||f}tj ||ddng|j D]}|j ^q}x|j D]}t |_ qWt|_ tj|tj}x)t|j |D]\}}||_ qW|r6tj |||d|dnWdQX|S(u Merge layers in the specified layer group belonging to the specified image into one layer. This function can handle both top-level and nested layer groups. u"{}": not a layer groupiiN((RRRYRZR R R RKR$tgimp_image_get_item_positionRRTRRRtgimp_image_merge_visible_layersRtEXPAND_AS_NECESSARYtzip(RR torig_parent_and_posRtorig_layer_visibilitytmerged_layer_groupt orig_visible((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pyRs(     !cc`s7tj}tj|z dVWdtj|XdS(u Temporarily redirect GIMP messages to the specified message handler. Use this function as a context manager: with redirect_messages(): # do stuff N(Rtgimp_message_get_handlertgimp_message_set_handler(tmessage_handlertorig_message_handler((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pytredirect_messagesBs   tGimpMessageFilecB`sGeZdZejdddZdZdZdZ dZ RS(ue This class provides a file-like way to write output as GIMP messages. You can use this class to redirect output or error output to the GIMP console. Parameters: * `message_handler` - Handler to which messages are output. Possible values are the same as for `pdb.gimp_message_get_handler()`. * `message_prefix` - If not `None`, prepend this string to each message. * `message_delay_milliseconds` - Delay in milliseconds before displaying the output. This is useful to aggregate multiple messages into one in order to avoid printing an excessive number of message headers. icC`sU||_|dk r!t|nd|_||_d|_d|_|j|_dS(Nui(t_message_handlerR$tstrt_message_prefixt_message_delay_millisecondst _buffer_sizet_orig_message_handlert_message_buffer(tselfRtmessage_prefixtmessage_delay_milliseconds((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pyt__init__gs  !   cC`s<tj|_tj|j|j||j|_dS(N(RRRRRt_writetwrite(RR"((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pyRvs cC`s^t|j|jkr@|j|7_tj|j|jntj|j|jdS(N(R6RRt pginvocationttimeout_add_strictRtflushttimeout_remove_strict(RR"((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pyRs cC`s)tjtj|j|j|_dS(N(RtmessageR1tsafe_encode_gimpRR(R((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pyRscC`s&|jdk r"tj|jndS(N(RR$RR(R((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pytclosesN( R`t __module__t__doc__Rt ERROR_CONSOLER$RRRRR(((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pyRUs  cC`s2tr dStjtjtjtjtjadS(u Prevent the default progress bar in GIMP from updating by installing a dummy progress callback. Subsequent calls to this function without a matching call to `unsuppress_gimp_progress()` have no effect. N(t_dummy_progress_callbackRtprogress_installR1t empty_func(((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pytsuppress_gimp_progressscC`s!ts dStjtdadS(u Undo suppression of the default progress bar in GIMP caused by calling `suppress_gimp_progress()`. Calling this function before calling `suppress_gimp_progress()` has no effect. N(RRtprogress_uninstallR$(((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pytunsuppress_gimp_progresss cC`s(t|do't|do't|S(uP Return `True` if the given function is a PDB procedure, `False` otherwise. u proc_nameuparams(R>tcallable(tfunction((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pytis_pdb_procedures(2Rt __future__RRRRtfuture.builtinsRxt contextlibRRRtRRRR1tcontextmanagerR RR/R4R:RARNRDRERRhRaRlRoRrRsRvRRR$RRRRRRtobjectRRRRR(((sJ/home/josie/.config/GIMP/2.10/plug-ins/export_layers/pygimplib/pdbutils.pytsR"      (    %       ! & *?  # -*- coding: utf-8 -*- """Interface for updating the progress of the work done so far.""" from __future__ import absolute_import, division, print_function, unicode_literals from future.builtins import * class ProgressUpdater(object): """ This class wraps the behavior of a progress bar used in GUIs. This class in particular can be used if no progress update is desired. To use this in the GUI for a progress bar, subclass this class and override the `_fill_progress_bar()` and `_set_text_progress_bar()` methods. Attributes: * `progress_bar` - Progress bar (GUI element). * `num_total_tasks` - Number of total tasks to complete. * `num_finished_tasks` (read-only) - Number of tasks finished so far. """ def __init__(self, progress_bar, num_total_tasks=0): self.progress_bar = progress_bar self.num_total_tasks = num_total_tasks self._num_finished_tasks = 0 @property def num_finished_tasks(self): return self._num_finished_tasks def update_tasks(self, num_tasks=1): """ Advance the progress bar by a given number of tasks finished. Raises: * `ValueError` - Number of finished tasks exceeds the number of total tasks. """ if self._num_finished_tasks + num_tasks > self.num_total_tasks: raise ValueError('number of finished tasks exceeds the number of total tasks') self._num_finished_tasks += num_tasks self._fill_progress_bar() def update_text(self, text): """ Update text in the progress bar. Use `None` or an empty string to remove the text. """ if text is None: text = '' self._set_text_progress_bar(text) def reset(self): """ Empty the progress bar and remove its text. """ self._num_finished_tasks = 0 if self.num_total_tasks > 0: self._fill_progress_bar() self._set_text_progress_bar('') def _fill_progress_bar(self): """ Fill in `num_finished_tasks`/`num_total_tasks` fraction of the progress bar. This is a method to be overridden by a subclass that implements a GUI-specific progress updater. """ pass def _set_text_progress_bar(self, text): """ Set the text of the progress bar. This is a method to be overridden by a subclass that implements a GUI-specific progress updater. """ pass