#!/usr/bin/env python
# -*- coding: utf-8 -*-

# GIMP plugin to export layer combination possibilities defined by groups
# (c) ShadowKyogre 2013
#
#   History:
#   2013-07-03 (v0.0): First published version
#   2013-07-03 (v0.1): Add getting combinations of layer groups
#
#   This program is free software; you can redistribute it and/or modify
#   it under the terms of the GNU General Public License as published by
#   the Free Software Foundation; either version 2 of the License, or
#   (at your option) any later version.
#
#   This program is distributed in the hope that it will be useful,
#   but WITHOUT ANY WARRANTY; without even the implied warranty of
#   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#   GNU General Public License for more details.
#
#   You should have received a copy of the GNU General Public License
#   along with this program; if not, write to the Free Software
#   Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.

from gimpfu import *
import os
from itertools import product, chain

def deleteAllLayers(image):
	for l in image.layers:
		image.remove_layer(l)

def group_layer_combos(layer, dump=None):
	target = pdb.gimp_image_new(layer.image.width, layer.image.height, RGB)
	if dump is None:
		#dummy image to hold child combinations
		dump = pdb.gimp_image_new(layer.image.width, layer.image.height, RGB)
		dump.disable_undo()
	target.disable_undo()

	src_lyrs=[]
	only_leaves=True
	for lyr in filter(lambda x: x.visible, layer.layers):
		#only consider children in the canditates for product if they are visible
		if hasattr(lyr,"layers"):
			src_lyrs.append(group_layer_combos(lyr, dump=dump)[0])
			only_leaves=False
		else:
			src_lyrs.append([lyr])

	results=[]
	if only_leaves:
		#only leaves indicates we want some choices
		src_lyrs=chain.from_iterable(src_lyrs)
		possibilities=product(src_lyrs)
	else:
		possibilities=product(*src_lyrs)
	#Get permutations for the layer groups
	counter=0
	for prm in possibilities:
		#place the layers in the mixture into a brand spanking new image
		#print([item.name for item in prm])
		for l in reversed(prm):
			n=pdb.gimp_layer_new_from_drawable(l,target)
			if l.parent is not None:
				n.mode=l.parent.mode
				n.opacity=l.parent.opacity*l.opacity/1E2
			target.add_layer(n,0)
		#process!
		result=target.merge_visible_layers(EXPAND_AS_NECESSARY) #or CLIP_TO_IMAGE
		result_copy=pdb.gimp_layer_new_from_drawable(result,dump)
		
		#workaround for now, probably implement a better work around later
		result_copy.name="{}.{}".format(layer.name,counter)
		counter+=1
		#end workaround

		results.append(result_copy)
		deleteAllLayers(target)
	pdb.gimp_image_delete(target)
	return results, dump

def permuteLayers(image, dirname, fnf, get_group_layer_combos):
	#image.undo_group_start()
	target = pdb.gimp_image_new(image.width, image.height, RGB)
	target.disable_undo()
	src_lyrs=[] # get all top lvl layer groups
	dumps=[]
	for lyr in filter(lambda x: x.visible, image.layers):
		#only consider children in the candidates for product if they are visible
		if hasattr(lyr,"layers"):
			if get_group_layer_combos:
				r=group_layer_combos(lyr)
				dumps.append(r[1])
				src_lyrs.append(r[0])
			else:
				src_lyrs.append([l for l in lyr.layers if l.visible])
		else:
			src_lyrs.append([lyr])
	print(src_lyrs)
	#Get permutations for the layer groups
	for prm in product(*src_lyrs):
		#place the layers in the mixture into a brand spanking new image
		for l in reversed(prm):
			n=pdb.gimp_layer_new_from_drawable(l,target)
			if l.parent is not None:
				n.mode=l.parent.mode
				n.opacity=l.parent.opacity*l.opacity/1E2
			target.add_layer(n,0)
		#process!
		expanded_fnf=fnf.format(*prm)
		fullname=os.path.join(dirname,expanded_fnf)
		if not os.path.exists(os.path.dirname(fullname)):
			os.makedirs(os.path.dirname(fullname))
		result=target.merge_visible_layers(EXPAND_AS_NECESSARY) #or CLIP_TO_IMAGE
		#gimp.message(expanded_fnf)
		pdb.gimp_file_save(target, 
				result, 
				fullname, 
				os.path.basename(fullname), 
				run_mode=RUN_NONINTERACTIVE)
		deleteAllLayers(target)
	
	#clean up any dumps that were acquired if group_layer_combos was called
	for dump in dumps:
		deleteAllLayers(dump)
		pdb.gimp_image_delete(dump)
	pdb.gimp_image_delete(target)
	#image.undo_group_end()

### Registrations
	
register(
	'permute-layer-groups',
	'Permute Layer Groups',
	'Save all possible permutations for each layer group combination to files',
	'ShadowKyogre',
	'ShadowKyogre',
	'2013',
	'Permute Layer Groups',
	'*',
	[
		(PF_IMAGE, 'image', 'Input image', None),
		(PF_DIRNAME, 'dirname', 'Output Directory', os.path.expanduser('~')),
		(PF_STRING, 'fnf', 'Filename format:\n'
						'Use {i.name} for the layer name picked out from the '
						'top level group.\n'
						'Numbers start at 0 from the topmost layer', 
						"{0.name}/{1.name}.png"),
		(PF_TOGGLE, 'get_group_layer_combos', 'Get combinations for subgroups?', False),
	],
	[],
	permuteLayers,
	menu='<Image>/File/Save',
)

main()