ó
dòôdc           @` sU  d  Z  d d l m Z m Z m Z m Z d d l Td d l Z d d l Z d d l	 Z	 d d l
 Z
 e
 j d ƒ d d l Z d d l Z d d l Z d d l Z d d l m Z d d l Z d d l m Z d d l m Z d d	 l m Z d d
 l m Z d d l m Z d d l m Z d d l m Z d e j f d „  ƒ  YZ  e j! e  ƒ d S(   u>   Preview widget displaying a scaled-down image to be processed.i    (   t   absolute_importt   divisiont   print_functiont   unicode_literals(   t   *Nu   2.0(   t   pdb(   t	   pygimplib(   t   actions(   t   builtin_constraints(   t
   exceptions(   t   utils(   t   messages(   t   preview_baset   ImagePreviewc           B` s³  e  Z d  Z i e j d% e j e j f f d 6Z d Z	 d Z
 d Z d Z d „  Z e d „  ƒ Z e j d „  ƒ Z e d	 „  ƒ Z d
 „  Z e d „ Z e d „ Z d „  Z d% d „ Z d% d% d „ Z d „  Z d „  Z d „  Z d „  Z d „  Z d% d% d „ Z d% d% d „ Z d% d% d „ Z  d „  Z! d „  Z" d „  Z# d „  Z$ d „  Z% e d „ Z& d „  Z' d „  Z( d  „  Z) d! „  Z* d" „  Z+ e, e d% d% d% d# „ ƒ Z- e, d$ „  ƒ Z. RS(&   u¨  
  This class defines a widget displaying a preview of an image to be processed,
  including its name.
  
  Signals:
  
  * `'preview-updated'` - The preview was updated by calling `update()`. This
    signal is not emitted if the update is locked.
    
    Arguments:
    
    * `error` - If `None`, the preview was updated successfully. Otherwise,
      this is an `Exception` object describing the error that occurred during
      the update.
    * `update_duration_seconds` - Duration of the update in seconds as a float.
      The duration only considers the update of the image contents (i.e. does
      not consider the duration of updating the label of the image name).
  s   preview-updatedu   _manual_updatei   i   i   c         C` sK  t  ƒ  j ƒ  | |  _ | |  _ d  |  _ d  |  _ d  |  _ d  |  _ t	 |  _
 t |  _ t |  _ d  |  _ d  |  _ d  |  _ d  |  _ d  |  _ d  |  _ d  |  _ |  j ƒ  |  j ƒ  d „  t j t j ƒ  ƒ Dƒ \ |  _ |  _ |  j d |  j ƒ |  j j d |  j ƒ |  j  j d |  j! ƒ |  j" j d |  j# ƒ |  j$ j d |  j% ƒ d  S(   Nc         s` s,   |  ]" } t  t | ƒ d  d d ƒ Vq d S(   i   i   i   N(   t   intt   hex(   t   .0t   shade(    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pys	   <genexpr>_   s    u   size-allocateu   clickedu   toggled(&   t   supert   __init__t   _batchert	   _settingst   Nonet   _itemt   _preview_pixbuft   _previous_preview_pixbuf_widtht   _previous_preview_pixbuf_heightt   Truet    draw_checkboard_alpha_backgroundt   Falset   _is_updatingt    _is_preview_image_allocated_sizet   _preview_widtht   _preview_heightt   _preview_scaling_factort   _resize_image_action_idt   _merge_items_action_idt   _scale_item_action_idt   _resize_item_action_idt   prepare_image_for_renderingt	   _init_guit   gimpt   checks_get_shadest
   check_typet    _preview_alpha_check_color_firstt!   _preview_alpha_check_color_secondt   connectt   _on_size_allocatet   _preview_imaget   _on_preview_image_size_allocatet   _button_menut   _on_button_menu_clickedt   _menu_item_update_automaticallyt*   _on_menu_item_update_automatically_toggledt   _button_refresht   _on_button_refresh_clicked(   t   selft   batchert   settings(    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyR   @   s4    																

+	c         C` s   |  j  S(   N(   R   (   R8   (    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyt   itemi   s    c         C` s7   | |  _  | d  k r3 d  |  _ d  |  _ d  |  _ n  d  S(   N(   R   R   R   R   R   (   R8   t   value(    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyR;   m   s
    			c         C` s   |  j  S(   N(   R4   (   R8   (    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyt   menu_item_update_automaticallyu   s    c         C` sñ   t  ƒ  j ƒ  } | r d  S|  j d  k r, d  St j |  j j ƒ sO |  j ƒ  d  S|  j j	 ƒ  |  j j
 t j j k rÃ t |  _ |  j j	 ƒ  |  j j ƒ  |  j |  j j ƒ |  j rí |  j ƒ  qí n* |  j j	 ƒ  |  j ƒ  |  j |  j j ƒ d  S(   N(   R   t   updateR;   R   R   t   gimp_item_is_validt   rawt   cleart   _placeholder_imaget   hidet   typet   pgt   itemtreet   TYPE_FOLDERR   R   t   _folder_imageR0   t   showt   _set_item_name_labelt   nameR   t   _set_contentst   _show_folder_image(   R8   t   update_locked(    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyR>   y   s&    
		
c         C` sA   d  |  _ |  j j ƒ  |  j j ƒ  |  j j ƒ  |  j | ƒ d  S(   N(   R   R;   R0   RA   RC   RH   t   _show_placeholder_image(   R8   t   use_item_name(    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyRA   •   s
    	c         C` s<   |  j  r8 |  j j ƒ  r8 |  j |  j j ƒ  |  j ƒ n  d S(   uu   
    Resize the preview if the widget is smaller than the previewed image so that
    the image fits the widget.
    N(   R   R0   t
   get_mappedt   _resize_previewt   get_allocationR   (   R8   t"   update_when_larger_than_image_size(    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyt   resizeœ   s    c         C` sL   |  j  j ƒ  } |  j d k	 oK | j |  j j ƒ  k oK | j |  j j ƒ  k S(   uu   
    Return `True` if the preview widget is larger than the image. If no image is
    previewed, return `False`.
    N(   R0   RS   R   R   t   widtht	   get_widtht   heightt
   get_height(   R8   t
   allocation(    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyt   is_larger_than_image¤   s    c         C` sÏ   | d  k ri |  j d  k	 r` |  j j d  k	 r` |  j j j |  j j k r` |  j j j } t } q{ t } n | |  j j k } | rË |  j j | } |  j j j j	 | ƒ rË | |  _ |  j
 |  j j ƒ qË n  d  S(   N(   R   R;   R   t	   item_treeR@   t   IDR   R   t   filtert   is_matchRJ   RK   (   R8   t   raw_item_idt   should_updateR;   (    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyt   update_item¯   s    			c         C` s:  | d k r d g } n  | d k r0 d g } n  |  j j |  j d d d t ƒ|  j j |  j | d t ƒ|  _ |  j j |  j d d d t ƒ|  j j |  j d g d t ƒ|  _ |  j j |  j	 d d d t ƒ|  j j |  j
 | d t ƒ|  _	 |  j j |  j d d d t ƒ|  j j |  j d g d t ƒ|  _ d S(	   uö  Adds procedures that prepare an image for rendering in the preview.
    
    Specifically, the image to be previewed is resized, scaled and later merged
    into a single layer.
    
    Subsequent calls to this method will remove the previously added procedures.
    
    The optional action groups allow to customize at which point during
    processing the resize and scale procedures are applied. By default, these
    procedures are applied before applying other procedures added by the user.
    u   before_process_items_contentsu   before_process_item_contentst   groupsu   allt   ignore_if_not_existst   ignore_if_existsu   after_process_item_contentsN(   R   R   t   remove_actionR#   R   t   add_proceduret   _resize_image_for_batcherR$   t   _merge_items_for_batcherR%   t   _scale_item_for_batcherR&   t   _resize_item_for_batcher(   R8   t   resize_image_action_groupst   scale_item_action_groups(    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyR'   Á   s(    								c         C` s¹   |  j  d  k r d  St j ƒ  } t j j ƒ    |  j |  j  j ƒ \ } } Wd  QX| d  k	 ro |  j j	 | ƒ n |  j
 d t ƒ |  j ƒ  t |  _ t j ƒ  | } |  j d | | ƒ d  S(   NRP   u   preview-updated(   R;   R   t   timeRE   t   pdbutilst   redirect_messagest   _get_in_memory_previewR@   R0   t   set_from_pixbufRA   R   t
   queue_drawR   R   t   emit(   R8   t   start_update_timet   preview_pixbuft   errort   update_duration_seconds(    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyRL   è   s    !
	c         C` s7  t  j ƒ  |  _ |  j j t  j ƒ |  j j t  j t  j t  j ƒ ƒ t  j	 t
 d ƒ ƒ |  _ |  j j t ƒ t  j ƒ  |  _ |  j j |  j ƒ |  j j ƒ  t  j ƒ  |  _ |  j j t
 d ƒ ƒ |  j j t  j ƒ |  j j t  j |  j j t  j t  j ƒ ƒ ƒ |  j j ƒ  |  j j ƒ  |  j j t ƒ t  j ƒ  |  _ |  j j |  j d t d t ƒ|  j j |  j d t d t ƒt  j ƒ  |  _ |  j j t ƒ t  j ƒ  |  _  |  j  j! t  j" t  j# ƒ |  j  j t ƒ t  j ƒ  |  _$ |  j$ j! t  j% t  j# ƒ |  j$ j t ƒ t  j& ƒ  |  _' |  j' j( t) j* ƒ |  j+ |  j, ƒ |  j- |  j. ƒ |  j |  j d t d t ƒ|  j |  j d t d t ƒ|  j |  j  d t d t ƒ|  j |  j$ d t d t ƒ|  j |  j' d t d t ƒt  j/ |  j  j0 d ƒ ƒ |  _1 t  j/ |  j$ j0 d ƒ ƒ |  _2 |  j  |  _3 |  j1 |  _4 |  j5 ƒ  d  S(   Nu   Update Preview Automaticallyu   Update Previewt   expandt   fillu	   icon-size(6   t   gtkt   ButtonR2   t
   set_relieft   RELIEF_NONEt   addt   Arrowt
   ARROW_DOWNt	   SHADOW_INt   CheckMenuItemt   _R4   t
   set_activeR   t   Menut   _menu_settingst   appendt   show_allR6   t   set_tooltip_textt   image_new_from_pixbuft   render_icont   STOCK_REFRESHt   ICON_SIZE_MENURC   t   set_no_show_allt   HBoxt   _hbox_buttonst
   pack_startR   t   ImageR0   RB   t   set_from_stockt   STOCK_DIALOG_QUESTIONt   ICON_SIZE_DIALOGRH   t   STOCK_DIRECTORYt   Labelt   _label_item_namet   set_ellipsizet   pangot   ELLIPSIZE_MIDDLEt   set_spacingt   _WIDGET_SPACINGt   set_border_widtht   _BORDER_WIDTHt   icon_size_lookupt   get_propertyt   _placeholder_image_sizet   _folder_image_sizet   _current_placeholder_imaget   _current_placeholder_image_sizeRO   (   R8   (    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyR(   ÿ   sX    "c         C` sf  |  j  | j | j ƒ \ |  _ |  _ |  j | j |  _ |  j ƒ  \ } } | d  k se t j	 | ƒ ro d  | f S| j
 s’ t j j | ƒ d  | f S| j t j k r´ t j | ƒ n  | j
 d } | j d  k	 rã | j t j ƒ n  |  j  | j | j ƒ \ |  _ |  _ |  j | |  j |  j ƒ \ |  _ |  _ } |  j | |  j |  j | ƒ } t j | ƒ | | f S(   Ni    (   t   _get_preview_sizeRV   RX   R    R!   R"   t   _get_image_previewR   R   t   gimp_image_is_validt   layersRE   Ro   t   try_delete_imaget	   base_typet	   gimpenumst   RGBt   gimp_image_convert_rgbt   maskt   remove_maskt
   MASK_APPLYt   _get_preview_datat   _get_preview_pixbuft   gimp_image_delete(   R8   t   raw_itemt   image_previewRw   t   raw_item_previewt   preview_datat   raw_item_preview_pixbuf(    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyRq   :  s,    
	
$c         C` s  |  j  j d  k	 rF x1 |  j  j j ƒ  D] } | j ƒ  | j ƒ  q% Wn  |  j  j t j d t	 j
 g d |  j j j g g ƒ} d  } yP |  j  j d t d |  j  j d t d t d t d t t j |  j d	 ƒ  } WnÒ t j k
 rê } n½ t j k
 rK} t j t j | ƒ d
 t | ƒ d t j ƒ  d t j j |  ƒ ƒ| } d  } n\ t  k
 r¦} t j t! d ƒ d
 t | ƒ d t j ƒ  d t j j |  ƒ ƒ| } d  } n X|  j  j" | t	 j
 g ƒ |  j  j d  k	 rüx' |  j  j j ƒ  D] } | j# ƒ  qåWn  | | f S(   NRc   t   argst   keep_image_copyR\   t
   is_previewt   process_contentst   process_namest   process_exportu   maint   failure_messaget   detailst   parentu4   There was a problem with updating the image preview:($   R   R\   R   t   iter_allt
   push_statet   resett   add_constraintR   t   is_item_in_selected_itemsR   t   DEFAULT_CONSTRAINTS_GROUPR;   R@   R]   t   runR   R   t   utils_t   get_settings_for_batcherR   R	   t   BatcherCancelErrort   ActionErrort	   messages_t   display_failure_messaget   get_failing_action_messaget   strt	   tracebackt
   format_excRE   t   guit   get_toplevel_windowt	   ExceptionR„   Rf   t	   pop_state(   R8   R;   t    only_selected_item_constraint_idRw   R·   t   e(    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyR¨   ^  sT    
			
	c      	   O` st   | j  } t j | t d t t | j |  j ƒ ƒ ƒ t d t t | j |  j ƒ ƒ ƒ d d ƒ t j	 t
 j ƒ d  S(   Ni   i    (   t   current_imageR   t   gimp_image_resizet   maxR   t   roundRV   R"   RX   t   gimp_context_set_interpolationR­   t   INTERPOLATION_LINEAR(   R8   R9   R»   t   kwargst   image(    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyRh   ™  s    	""c         C` s1   t  j | j t j ƒ } | | j _ | | _ d  S(   N(   R   t   gimp_image_merge_visible_layersRÛ   R­   t   EXPAND_AS_NECESSARYt   active_layert   current_raw_item(   R8   R9   R;   R¶   t   raw_item_merged(    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyRi   ¥  s    c         C` s‹   | d  k s t j | ƒ r( | j } n  t j | | j d |  j | j d |  j | j d | j |  j | j d | j |  j ƒ d  S(   Ni    i   (	   R   R   R?   Ræ   t   gimp_item_transform_scalet   offsetsR"   RV   RX   (   R8   R9   R;   R¶   (    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyRj   ¬  s    c         C` s   t  j | j ƒ d  S(   N(   R   t   gimp_layer_resize_to_image_sizeRæ   (   R8   R9   R;   R¶   (    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyRk   ·  s    c      	   C` sz   t  j j | t  j j | j d | | | | j ƒ } | |  _ | j rv |  j | | j |  j	 |  j
 |  j |  j ƒ } n  | S(   Ni   (   R{   t   gdkt   pixbuf_new_from_datat   COLORSPACE_RGBt	   has_alphat   bppR   t   _add_alpha_background_to_pixbuft   opacityR   t   _PREVIEW_ALPHA_CHECK_SIZER,   R-   (   R8   R¶   t   preview_widtht   preview_heightR¹   Rº   (    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyR´   º  s$    				c         C` s  |  j  j ƒ  } | j } | j } | | k rˆ t | | ƒ } t t | | | ƒ ƒ } | | k rà | } t t | | | ƒ ƒ } qà nX t | | ƒ } t t | | | ƒ ƒ } | | k rà | } t t | | | ƒ ƒ } n  | d k rõ d } n  | d k r
d } n  | | f S(   Ni    i   (   R0   RS   RV   RX   t   minR   RÞ   (   R8   RV   RX   t   preview_widget_allocationt   preview_widget_widtht   preview_widget_heightRô   Ró   (    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyR§   Ô  s&    		 		c         C` sù   | d  k r d  S| j | j ƒ  k r> | j | j ƒ  k r> d  S|  j | j ƒ  | j ƒ  ƒ \ } } |  j | k r„ |  j | k r„ d  S| j | | t	 j
 j ƒ } |  j | d |  j |  j |  j |  j ƒ } |  j j | ƒ |  j ƒ  | |  _ | |  _ d  S(   Nid   (   R   RV   RW   RX   RY   R§   R   R   t   scale_simpleR{   Rë   t   INTERP_BILINEARRð   R   Rò   R,   R-   R0   Rr   Rs   (   R8   t   preview_allocationRv   t   scaled_preview_widtht   scaled_preview_heightt   scaled_preview_pixbuf(    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyRR   ï  s.    
	c         C` s¶   |  j  r² |  j j ƒ  r² | j |  j } | j |  j j ƒ  j |  j |  j	 j ƒ  j |  j |  j d } | |  j
 d k  s’ | |  j
 d k  r¢ |  j j ƒ  q² |  j j ƒ  n  d  S(   Ni   i    i   (   R   R0   RQ   RV   R    RX   R‘   RS   Rž   R™   R¦   R¥   RC   RI   (   R8   t   previewRZ   t   preview_widget_allocated_widtht   preview_widget_allocated_height(    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyR/     s    Bc         C` s#   |  j  s |  j ƒ  t |  _  n  d  S(   N(   R   RL   R   (   R8   Râ   RZ   (    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyR1      s    	
c         C` sE   |  j  |  _ |  j |  _ |  j  j ƒ  | sA |  j t d ƒ ƒ n  d  S(   Nu   No selection(   RB   R¥   R£   R¦   RI   RJ   R„   (   R8   RP   (    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyRO   %  s
    c         C` s)   |  j  |  _ |  j |  _ |  j  j ƒ  d  S(   N(   RH   R¥   R¤   R¦   RI   (   R8   (    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyRM   .  s    c         C` s2   |  j  j d j t j t j j | ƒ ƒ ƒ ƒ d  S(   Nu	   <i>{}</i>(   R™   t
   set_markupt   formatt   gobjectt   markup_escape_textRE   R
   t   safe_encode_gtk(   R8   t	   item_name(    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyRJ   4  s    	c         C` s   t  j j |  j | ƒ d  S(   N(   RE   RÕ   t   menu_popup_below_widgetR‡   (   R8   t   button(    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyR3   8  s    c         C` s`   |  j  j ƒ  r< |  j j ƒ  |  j t |  j ƒ |  j ƒ  n  |  j j ƒ  |  j t	 |  j ƒ d  S(   N(
   R4   t
   get_activeR6   RC   t   lock_updateR   t   _MANUAL_UPDATE_LOCKR>   RI   R   (   R8   t	   menu_item(    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyR5   ;  s    c         C` sS   |  j  |  j k rE |  j t |  j  ƒ |  j ƒ  |  j t |  j  ƒ n
 |  j ƒ  d  S(   N(   R  t
   _lock_keysR  R   R>   R   (   R8   R	  (    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyR7   D  s
    
c         C` s+  | r™ t  j j t  j j t d |  j ƒ  |  j ƒ  ƒ } |  j | d d |  j ƒ  |  j ƒ  d d d d t  j j t	 t
 | d d ƒ ƒ d d | | | ƒ nŽ t  j j t  j j t d |  j ƒ  |  j ƒ  ƒ } | j d ƒ |  j | d d |  j ƒ  |  j ƒ  d d d d t  j j t	 t
 | d d ƒ ƒ ƒ | S(   Ni   i    g      ð?g      Y@iÿ   I ÿÿÿ    (   R{   Rë   t   PixbufRí   R   RW   RY   t   composite_colort   INTERP_NEARESTR   RÞ   R   Rz   t	   composite(   t   pixbufRñ   t   use_checkboard_backgroundt
   check_sizet   check_color_firstt   check_color_secondt   pixbuf_with_alpha_background(    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyRð   L  sX    						
						c         C` sC   t  j |  | | ƒ \ } } } } } | | t j d | ƒ j ƒ  f S(   Nt   B(   R   t   gimp_drawable_thumbnailt   arrayt   tostring(   R¶   Ró   Rô   t   actual_preview_widtht   actual_preview_heightt   unused_R¹   (    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyR³   …  s    $N(/   t   __name__t
   __module__t   __doc__R  t   SIGNAL_RUN_FIRSTR   t   TYPE_PYOBJECTt
   TYPE_FLOATt   __gsignals__R  Rž   R    Rò   R   t   propertyR;   t   setterR=   R>   R   RA   RU   R[   Rb   R'   RL   R(   Rq   R¨   Rh   Ri   Rj   Rk   R´   R§   RR   R/   R1   RO   RM   RJ   R3   R5   R7   t   staticmethodRð   R³   (    (    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyR   !   sT   "	)		&		;	$	;				 									2("   R"  t
   __future__R    R   R   R   t   future.builtinsR  Rn   RÓ   t   pygtkt   requireR{   R  R›   R)   R   R­   t   export_layersR   RE   R   R   R	   R
   RË   t   export_layers.guiR   RÏ   R   t   preview_base_t   PreviewR   t   type_register(    (    (    sI   /home/josie/.config/GIMP/2.10/plug-ins/export_layers/gui/preview_image.pyt   <module>   s0   "
ÿ ÿ q                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                    # -*- coding: utf-8 -*-

"""Preview widget displaying the names of items to be batch-processed."""

from __future__ import absolute_import, division, print_function, unicode_literals
from future.builtins import *

import collections
import os
import traceback

import pygtk
pygtk.require('2.0')
import gtk
import gobject

from gimp import pdb

from export_layers import pygimplib as pg

from export_layers import exceptions
from export_layers import utils as utils_
from export_layers.gui import preview_base as preview_base_

from export_layers.gui import messages as messages_


class NamePreview(preview_base_.Preview):
  """A widget displaying a preview of batch-processed items - names and their
  folder structure.
  
  Additional features:
  * toggling "filter mode" - unselected items are not sensitive.
  * assigning tags to items.
  
  Attributes:
  
  * `is_filtering` - If enabled, unselected items are not sensitive.
  
  Signals:
  
  * `'preview-selection-changed'` - The selection in the preview was modified
    by the user or by calling `set_selected_items()`.
  * `'preview-updated'` - The preview was updated by calling `update()`. This
    signal is not emitted if the update is locked.
    
    Arguments:
    
    * `error` - If `None`, the preview was updated successfully. Otherwise,
      this is an `Exception` object describing the error that occurred during
      the update.
  * `'preview-tags-changed'` - An existing tag was added to or removed from an
    item.
  """
  
  __gsignals__ = {
    b'preview-selection-changed': (gobject.SIGNAL_RUN_FIRST, None, ()),
    b'preview-updated': (gobject.SIGNAL_RUN_FIRST, None, (gobject.TYPE_PYOBJECT,)),
    b'preview-tags-changed': (gobject.SIGNAL_RUN_FIRST, None, ()),
  }
  
  _ADD_TAG_POPUP_HBOX_SPACING = 5
  _ADD_TAG_POPUP_BORDER_WIDTH = 5
  
  _COLUMNS = (
    _COLUMN_ICON_ITEM,
    _COLUMN_ICON_TAG_VISIBLE,
    _COLUMN_ITEM_NAME_SENSITIVE,
    _COLUMN_ITEM_NAME,
    _COLUMN_ITEM_ID,
    _COLUMN_ITEM_TYPE) = (
    [0, gtk.gdk.Pixbuf],
    [1, gobject.TYPE_BOOLEAN],
    [2, gobject.TYPE_BOOLEAN],
    [3, gobject.TYPE_STRING],
    [4, gobject.TYPE_INT],
    [5, gobject.TYPE_INT])
  
  def __init__(
        self,
        batcher,
        settings,
        initial_item_tree=None,
        collapsed_items=None,
        selected_items=None,
        selected_items_filter_name='selected_in_preview',
        available_tags_setting=None):
    super().__init__()
    
    self._batcher = batcher
    self._settings = settings
    self._initial_item_tree = initial_item_tree
    self._collapsed_items = collapsed_items if collapsed_items is not None else set()
    self._selected_items = selected_items if selected_items is not None else []
    self._selected_items_filter_name = selected_items_filter_name
    self._available_tags_setting = available_tags_setting
    
    self.is_filtering = False
    
    # key: `Item.raw.ID` or (`Item.raw.ID`, 'folder') instance
    # value: `gtk.TreeIter` instance
    self._tree_iters = collections.defaultdict(pg.utils.return_none_func)
    
    self._row_expand_collapse_interactive = True
    self._toggle_tag_interactive = True
    self._clearing_preview = False
    self._row_select_interactive = True
    self._initial_scroll_to_selection = True
    
    self._icon_image_filepath = os.path.join(
      pg.config.PLUGIN_SUBDIRPATH, 'images', 'icon_image.png')
    self._icon_tag_filepath = os.path.join(
      pg.config.PLUGIN_SUBDIRPATH, 'images', 'icon_tag.png')
    
    self._init_gui()
  
  @property
  def batcher(self):
    return self._batcher
  
  @property
  def tree_view(self):
    return self._tree_view
  
  @property
  def collapsed_items(self):
    return self._collapsed_items
  
  @property
  def selected_items(self):
    return self._selected_items
  
  def update(self, reset_items=False, update_existing_contents_only=False):
    """Updates the preview (add/remove item, move item to a different parent
    item group, etc.).
    
    If `reset_items` is `True`, perform full update - add new items, remove
    non-existent items, etc. Note that setting this to `True` may introduce a
    performance penalty for hundreds of items.
    
    If `update_existing_contents_only` is `True`, only update the contents of
    the existing items. Note that the items will not be reparented,
    expanded/collapsed or added/removed even if they need to be. This option is
    useful if you know the item structure will be preserved.
    
    If an exception was captured during the update, the method is terminated
    prematurely. It is the responsibility of the caller to handle the error
    (e.g. lock or clear the preview).
    """
    update_locked = super().update()
    if update_locked:
      return
    
    if not update_existing_contents_only:
      self.clear()
    
    error = self._process_items(reset_items=reset_items)
    
    if error:
      self.emit('preview-updated', error)
      return
    
    items = self._get_items_to_process()
    
    if not update_existing_contents_only:
      self._insert_items(items)
      self._set_expanded_items()
    else:
      self._update_items(items)
    
    self._set_selection()
    self._set_item_tree_sensitive_for_selected(items)
    
    self._update_available_tags()
    
    self._tree_view.columns_autosize()
    
    self.emit('preview-updated', None)
  
  def clear(self):
    """
    Clear the entire preview.
    """
    self._clearing_preview = True
    self._tree_model.clear()
    self._tree_iters.clear()
    self._clearing_preview = False
  
  def set_collapsed_items(self, collapsed_items):
    """
    Set the collapsed state of items in the preview.
    """
    self._collapsed_items = collapsed_items
    self._set_expanded_items()
  
  def set_selected_items(self, selected_items):
    """
    Set the selection of items in the preview.
    """
    self._selected_items = selected_items
    self._set_selection()
    self.emit('preview-selection-changed')
  
  def get_items_from_selected_rows(self):
    return [self._batcher.item_tree[item_key]
            for item_key in self._get_keys_from_current_selection()]
  
  def get_item_from_cursor(self):
    tree_path, unused_ = self._tree_view.get_cursor()
    if tree_path is not None:
      item_key = self._get_key_from_tree_iter(self._tree_model.get_iter(tree_path))
      return self._batcher.item_tree[item_key]
    else:
      return None
  
  def _init_gui(self):
    self._tree_model = gtk.TreeStore(*[column[1] for column in self._COLUMNS])
    
    self._tree_view = gtk.TreeView(model=self._tree_model)
    self._tree_view.set_headers_visible(False)
    self._tree_view.set_enable_search(False)
    self._tree_view.get_selection().set_mode(gtk.SELECTION_MULTIPLE)
    
    self._init_icons()
    
    self._init_tags_menu()
    
    column = gtk.TreeViewColumn(b'')
    
    cell_renderer_icon_item = gtk.CellRendererPixbuf()
    column.pack_start(cell_renderer_icon_item, expand=False)
    column.set_attributes(cell_renderer_icon_item, pixbuf=self._COLUMN_ICON_ITEM[0])
    
    cell_renderer_icon_tag = gtk.CellRendererPixbuf()
    cell_renderer_icon_tag.set_property('pixbuf', self._icons['tag'])
    column.pack_start(cell_renderer_icon_tag, expand=False)
    column.set_attributes(
      cell_renderer_icon_tag,
      visible=self._COLUMN_ICON_TAG_VISIBLE[0])
    
    cell_renderer_item_name = gtk.CellRendererText()
    column.pack_start(cell_renderer_item_name, expand=False)
    column.set_attributes(
      cell_renderer_item_name,
      text=self._COLUMN_ITEM_NAME[0],
      sensitive=self._COLUMN_ITEM_NAME_SENSITIVE[0])
    
    self._tree_view.append_column(column)
    
    self._scrolled_window = gtk.ScrolledWindow()
    self._scrolled_window.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
    self._scrolled_window.add(self._tree_view)
    
    self.pack_start(self._scrolled_window)
    
    self._tree_view.connect('row-collapsed', self._on_tree_view_row_collapsed)
    self._tree_view.connect('row-expanded', self._on_tree_view_row_expanded)
    self._tree_view.get_selection().connect('changed', self._on_tree_selection_changed)
    self._tree_view.connect('event', self._on_tree_view_right_button_press_event)
  
  def _init_icons(self):
    self._icons = {}
    self._icons['folder'] = self._tree_view.render_icon(
      gtk.STOCK_DIRECTORY, gtk.ICON_SIZE_MENU)
    self._icons['item'] = gtk.gdk.pixbuf_new_from_file_at_size(
      self._icon_image_filepath, -1, self._icons['folder'].props.height)
    self._icons['tag'] = gtk.gdk.pixbuf_new_from_file_at_size(
      self._icon_tag_filepath, -1, self._icons['folder'].props.height)
    
    self._icons['group'] = self._icons['item'].copy()
    
    scaling_factor = 0.8
    width_unscaled = self._icons['folder'].props.width
    width = int(width_unscaled * scaling_factor)
    height_unscaled = self._icons['folder'].props.height
    height = int(height_unscaled * scaling_factor)
    x_offset_unscaled = self._icons['group'].props.width - self._icons['folder'].props.width
    x_offset = x_offset_unscaled + width_unscaled - width
    y_offset_unscaled = self._icons['group'].props.height - self._icons['folder'].props.height
    y_offset = y_offset_unscaled + height_unscaled - height
    
    self._icons['folder'].composite(
      self._icons['group'],
      x_offset, y_offset, width, height, x_offset, y_offset,
      scaling_factor, scaling_factor, gtk.gdk.INTERP_BILINEAR, 255)
  
  def _init_tags_menu(self):
    self._tags_menu_items = {}
    self._tags_remove_submenu_items = {}
    
    self._tags_menu_relative_position = None
    
    self._tags_menu = gtk.Menu()
    self._tags_remove_submenu = gtk.Menu()
    
    self._tags_menu.append(gtk.SeparatorMenuItem())
    
    self._menu_item_add_tag = gtk.MenuItem(_('Add New Tag...'))
    self._menu_item_add_tag.connect('activate', self._on_menu_item_add_tag_activate)
    self._tags_menu.append(self._menu_item_add_tag)
    
    self._menu_item_remove_tag = gtk.MenuItem(_('Remove Tag'))
    self._menu_item_remove_tag.set_submenu(self._tags_remove_submenu)
    self._tags_menu.append(self._menu_item_remove_tag)
    
    for tag, tag_display_name in self._available_tags_setting.default_value.items():
      self._add_tag_menu_item(tag, tag_display_name)
    
    self._tags_menu.show_all()
  
  def _update_available_tags(self):
    used_tags = set()
    for item in self._batcher.item_tree.iter(filtered=False):
      for tag in item.tags:
        used_tags.add(tag)
        if tag not in self._tags_menu_items:
          self._add_tag_menu_item(tag, tag)
          self._add_remove_tag_menu_item(tag, tag)
    
    for tag, menu_item in self._tags_remove_submenu_items.items():
      menu_item.set_sensitive(tag not in used_tags)
    
    for tag in self._available_tags_setting.value:
      if tag not in self._tags_menu_items:
        self._add_tag_menu_item(tag, tag)
        self._add_remove_tag_menu_item(tag, tag)
    
    self._menu_item_remove_tag.set_sensitive(
      bool(self._tags_remove_submenu.get_children()))
    
    self._sort_tags_menu_items()
    
    for tag in self._tags_menu_items:
      if tag not in self._available_tags_setting.value:
        self._available_tags_setting.value[tag] = tag
    
    self._available_tags_setting.save()
  
  def _sort_tags_menu_items(self):
    for new_tag_position, tag in (
          enumerate(sorted(self._tags_menu_items, key=lambda tag: tag.lower()))):
      self._tags_menu.reorder_child(self._tags_menu_items[tag], new_tag_position)
      if tag in self._tags_remove_submenu_items:
        self._tags_remove_submenu.reorder_child(
          self._tags_remove_submenu_items[tag], new_tag_position)
  
  def _add_tag_menu_item(self, tag, tag_display_name):
    self._tags_menu_items[tag] = gtk.CheckMenuItem(tag_display_name)
    self._tags_menu_items[tag].connect('toggled', self._on_tags_menu_item_toggled, tag)
    self._tags_menu_items[tag].show()
    self._tags_menu.prepend(self._tags_menu_items[tag])
    
    return self._tags_menu_items[tag]
  
  def _add_remove_tag_menu_item(self, tag, tag_display_name):
    self._tags_remove_submenu_items[tag] = gtk.MenuItem(tag_display_name)
    self._tags_remove_submenu_items[tag].connect(
      'activate', self._on_tags_remove_submenu_item_activate, tag)
    self._tags_remove_submenu_items[tag].show()
    self._tags_remove_submenu.prepend(self._tags_remove_submenu_items[tag])
  
  def _on_tree_view_right_button_press_event(self, tree_view, event):
    if event.type == gtk.gdk.BUTTON_PRESS and event.button == 3:
      item_keys = []
      stop_event_propagation = False
      
      # Get the current selection. We cannot use `TreeSelection.get_selection()`
      # because this event is fired before the selection is updated.
      selection_at_pos = self._tree_view.get_path_at_pos(int(event.x), int(event.y))
      
      if (selection_at_pos is not None
          and self._tree_view.get_selection().count_selected_rows() > 1):
        item_keys = self._get_keys_from_current_selection()
        stop_event_propagation = True
      else:
        if selection_at_pos is not None:
          tree_iter = self._tree_model.get_iter(selection_at_pos[0])
          item_keys = [self._get_key_from_tree_iter(tree_iter)]
      
      self._toggle_tag_interactive = False
      
      items = [self._batcher.item_tree[item_key] for item_key in item_keys]
      for tag, tags_menu_item in self._tags_menu_items.items():
        tags_menu_item.set_active(all(tag in item.tags for item in items))
      
      self._toggle_tag_interactive = True
      
      if len(item_keys) >= 1:
        self._tags_menu.popup(None, None, None, event.button, event.time)
        
        toplevel_window = pg.gui.get_toplevel_window(self)
        if toplevel_window is not None:
          self._tags_menu_relative_position = toplevel_window.get_window().get_pointer()
      
      return stop_event_propagation
  
  def _on_tags_menu_item_toggled(self, tags_menu_item, tag):
    if self._toggle_tag_interactive:
      pdb.gimp_image_undo_group_start(self._batcher.input_image)
      
      for item_key in self._get_keys_from_current_selection():
        item = self._batcher.item_tree[item_key]
        
        if tags_menu_item.get_active():
          item.add_tag(tag)
        else:
          item.remove_tag(tag)
      
      pdb.gimp_image_undo_group_end(self._batcher.input_image)
      
      # Modifying just one item could result in renaming other items
      # differently, hence update the whole preview.
      self.update(update_existing_contents_only=True)
      
      self.emit('preview-tags-changed')
  
  def _on_menu_item_add_tag_activate(self, menu_item_add_tag):
    def _on_popup_focus_out_event(popup, event):
      popup.destroy()
    
    def _on_popup_key_press_event(popup, event):
      key_name = gtk.gdk.keyval_name(event.keyval)
      if key_name in ['Return', 'KP_Enter']:
        entry_text = entry_add_tag.get_text()
        if entry_text and entry_text not in self._tags_menu_items:
          menu_item = self._add_tag_menu_item(entry_text, entry_text)
          menu_item.set_active(True)
          self._add_remove_tag_menu_item(entry_text, entry_text)
        
        popup.destroy()
        return True
      elif key_name == 'Escape':
        popup.destroy()
        return True
    
    def _set_popup_position(popup, window):
      if self._tags_menu_relative_position is not None:
        window_absolute_position = window.get_window().get_origin()
        popup.move(
          window_absolute_position[0] + self._tags_menu_relative_position[0],
          window_absolute_position[1] + self._tags_menu_relative_position[1])
        
        self._tags_menu_relative_position = None
    
    popup_add_tag = gtk.Window(gtk.WINDOW_TOPLEVEL)
    popup_add_tag.set_decorated(False)
    popup_add_tag.set_type_hint(gtk.gdk.WINDOW_TYPE_HINT_POPUP_MENU)
    
    toplevel = pg.gui.get_toplevel_window(self)
    if toplevel is not None:
      popup_add_tag.set_transient_for(toplevel)
    
    _set_popup_position(popup_add_tag, toplevel)
    
    label_tag_name = gtk.Label(_('Tag Name:'))
    
    entry_add_tag = gtk.Entry()
    
    hbox = gtk.HBox()
    hbox.set_spacing(self._ADD_TAG_POPUP_HBOX_SPACING)
    hbox.pack_start(label_tag_name, expand=False, fill=False)
    hbox.pack_start(entry_add_tag, expand=False, fill=False)
    hbox.set_border_width(self._ADD_TAG_POPUP_BORDER_WIDTH)
    
    frame = gtk.Frame()
    frame.add(hbox)
    
    popup_add_tag.add(frame)
    
    popup_add_tag.connect('focus-out-event', _on_popup_focus_out_event)
    popup_add_tag.connect('key-press-event', _on_popup_key_press_event)
    
    popup_add_tag.show_all()
  
  def _on_tags_remove_submenu_item_activate(self, tags_remove_submenu_item, tag):
    self._tags_remove_submenu.remove(tags_remove_submenu_item)
    self._tags_menu.remove(self._tags_menu_items[tag])
    
    del self._tags_menu_items[tag]
    del self._tags_remove_submenu_items[tag]
    del self._available_tags_setting.value[tag]
    
    self._menu_item_remove_tag.set_sensitive(
      bool(self._tags_remove_submenu.get_children()))
    
    self._available_tags_setting.save()
  
  def _on_tree_view_row_collapsed(self, tree_view, tree_iter, tree_path):
    if self._row_expand_collapse_interactive:
      self._collapsed_items.add(self._get_key_from_tree_iter(tree_iter))
      self._tree_view.columns_autosize()
  
  def _on_tree_view_row_expanded(self, tree_view, tree_iter, tree_path):
    if self._row_expand_collapse_interactive:
      item_key = self._get_key_from_tree_iter(tree_iter)
      if item_key in self._collapsed_items:
        self._collapsed_items.remove(item_key)
      
      self._set_expanded_items(tree_path)
      
      self._tree_view.columns_autosize()
  
  def _on_tree_selection_changed(self, tree_selection):
    if not self._clearing_preview and self._row_select_interactive:
      previous_selected_items = self._selected_items
      self._selected_items = self._get_keys_from_current_selection()
      
      self.emit('preview-selection-changed')
      
      if self.is_filtering and self._selected_items != previous_selected_items:
        self.update()
  
  def _get_keys_from_current_selection(self):
    unused_, tree_paths = self._tree_view.get_selection().get_selected_rows()
    return [
      self._get_key_from_tree_iter(self._tree_model.get_iter(tree_path))
      for tree_path in tree_paths]
  
  def _get_key(self, item):
    if item.type != pg.itemtree.TYPE_FOLDER:
      return item.raw.ID
    else:
      return (item.raw.ID, pg.itemtree.FOLDER_KEY)
  
  def _get_key_from_tree_iter(self, tree_iter):
    item_id = self._tree_model.get_value(tree_iter, column=self._COLUMN_ITEM_ID[0])
    item_type = self._tree_model.get_value(tree_iter, column=self._COLUMN_ITEM_TYPE[0])
    
    if item_type != pg.itemtree.TYPE_FOLDER:
      return item_id
    else:
      return (item_id, pg.itemtree.FOLDER_KEY)
  
  def _get_items_to_process(self):
    if self.is_filtering:
      with self._batcher.item_tree.filter.remove_temp(name=self._selected_items_filter_name):
        return list(self._batcher.item_tree)
    else:
      return list(self._batcher.item_tree)
  
  def _process_items(self, reset_items=False):
    if not reset_items:
      if self._initial_item_tree is not None:
        item_tree = self._initial_item_tree
        self._initial_item_tree = None
      else:
        item_tree = self._batcher.item_tree
    else:
      item_tree = None
    
    if item_tree is not None:
      # We need to reset item attributes explicitly before processing since
      # existing item trees are not automatically refreshed.
      for item in item_tree.iter_all():
        item.reset()
    
    error = None
    
    try:
      self._batcher.run(
        item_tree=item_tree,
        is_preview=True,
        process_contents=False,
        process_names=True,
        process_export=False,
        **utils_.get_settings_for_batcher(self._settings['main']))
    except exceptions.BatcherCancelError as e:
      pass
    except exceptions.ActionError as e:
      messages_.display_failure_message(
        messages_.get_failing_action_message(e),
        failure_message=str(e),
        details=traceback.format_exc(),
        parent=pg.gui.get_toplevel_window(self))
      
      error = e
    except Exception as e:
      messages_.display_failure_message(
        _('There was a problem with updating the name preview:'),
        failure_message=str(e),
        details=traceback.format_exc(),
        parent=pg.gui.get_toplevel_window(self))
      
      error = e
    
    return error
  
  def _update_items(self, items):
    updated_parents = set()
    for item in items:
      self._update_parent_items(item, updated_parents)
      self._update_item(item)
  
  def _insert_items(self, items):
    inserted_parents = set()
    for item in items:
      self._insert_parent_items(item, inserted_parents)
      self._insert_item(item)
  
  def _insert_item(self, item):
    if item.parent:
      parent_tree_iter = self._tree_iters[self._get_key(item.parent)]
    else:
      parent_tree_iter = None
    
    tree_iter = self._tree_model.append(
      parent_tree_iter,
      [self._get_icon_from_item(item),
       bool(item.tags),
       True,
       pg.utils.safe_encode_gtk(item.name),
       item.raw.ID,
       item.type])
    
    self._tree_iters[self._get_key(item)] = tree_iter
    
    return tree_iter
  
  def _update_item(self, item):
    self._tree_model.set(
      self._tree_iters[self._get_key(item)],
      self._COLUMN_ICON_TAG_VISIBLE[0],
      bool(item.tags),
      self._COLUMN_ITEM_NAME_SENSITIVE[0],
      True,
      self._COLUMN_ITEM_NAME[0],
      pg.utils.safe_encode_gtk(item.name))
  
  def _insert_parent_items(self, item, inserted_parents):
    for parent in item.parents:
      if parent not in inserted_parents:
        self._insert_item(parent)
        inserted_parents.add(parent)
  
  def _update_parent_items(self, item, updated_parents):
    for parent in item.parents:
      if parent not in updated_parents:
        self._update_item(parent)
        updated_parents.add(parent)
  
  def _set_item_tree_sensitive_for_selected(self, items):
    if self.is_filtering:
      self._set_items_sensitive(items, False)
      self._set_items_sensitive(
        [self._batcher.item_tree[item_key] for item_key in self._selected_items], True)
  
  def _get_item_sensitive(self, item):
    return self._tree_model.get_value(
      self._tree_iters[self._get_key(item)], self._COLUMN_ITEM_NAME_SENSITIVE[0])
  
  def _set_items_sensitive(self, items, sensitive):
    processed_parents = set()
    for item in items:
      self._set_item_sensitive(item, sensitive)
      self._set_parent_items_sensitive(item, processed_parents)
  
  def _set_item_sensitive(self, item, sensitive):
    if self._get_key(item) in self._tree_iters:
      self._tree_model.set_value(
        self._tree_iters[self._get_key(item)],
        self._COLUMN_ITEM_NAME_SENSITIVE[0],
        sensitive)
  
  def _set_parent_items_sensitive(self, item, processed_parents):
    for parent in reversed(list(item.parents)):
      if parent not in processed_parents:
        parent_sensitive = any(
          self._get_item_sensitive(child) for child in parent.children
          if self._get_key(child) in self._tree_iters)
        self._set_item_sensitive(parent, parent_sensitive)
        
        processed_parents.add(parent)
  
  def _get_icon_from_item(self, item):
    if item.type == pg.itemtree.TYPE_ITEM:
      return self._icons['item']
    elif item.type == pg.itemtree.TYPE_GROUP:
      return self._icons['group']
    elif item.type == pg.itemtree.TYPE_FOLDER:
      return self._icons['folder']
    else:
      return None
  
  def _set_expanded_items(self, tree_path=None):
    """
    Set the expanded state of items in the tree view.
    
    If `tree_path` is specified, set the states only for the child elements in
    the tree path, otherwise set the states in the whole tree view.
    """
    self._row_expand_collapse_interactive = False
    
    if tree_path is None:
      self._tree_view.expand_all()
    else:
      self._tree_view.expand_row(tree_path, True)
    
    self._remove_no_longer_valid_collapsed_items()
    
    for item_key in self._collapsed_items:
      if item_key in self._tree_iters:
        item_tree_iter = self._tree_iters[item_key]
        if item_tree_iter is None:
          continue
        
        item_tree_path = self._tree_model.get_path(item_tree_iter)
        if tree_path is None or self._tree_view.row_expanded(item_tree_path):
          self._tree_view.collapse_row(item_tree_path)
    
    self._row_expand_collapse_interactive = True
  
  def _remove_no_longer_valid_collapsed_items(self):
    if self._batcher.item_tree is None:
      return
    
    self._collapsed_items = set(
      [item_key for item_key in self._collapsed_items if item_key in self._batcher.item_tree])
  
  def _set_selection(self):
    self._row_select_interactive = False
    
    self._selected_items = [
      item_key for item_key in self._selected_items if item_key in self._tree_iters]
    
    for item_key in self._selected_items:
      tree_iter = self._tree_iters[item_key]
      if tree_iter is not None:
        self._tree_view.get_selection().select_iter(tree_iter)
    
    if self._initial_scroll_to_selection and self._selected_items:
      self._set_initial_scroll_to_selection()
      self._initial_scroll_to_selection = False
    
    self._row_select_interactive = True
  
  def _set_cursor(self, previous_cursor=None):
    self._row_select_interactive = False
    
    if previous_cursor is not None and self._tree_model.get_iter(previous_cursor) is not None:
      self._tree_view.set_cursor(previous_cursor)
    
    self._row_select_interactive = True
  
  def _set_initial_scroll_to_selection(self):
    if self._selected_items:
      tree_iter = self._tree_iters[self._selected_items[0]]
      if tree_iter is not None:
        first_selected_item_path = (
          self._tree_model.get_path(self._tree_iters[self._selected_items[0]]))
        if first_selected_item_path is not None:
          self._tree_view.scroll_to_cell(first_selected_item_path, None, True, 0.5, 0.0)


gobject.type_register(NamePreview)
                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                              