Tool to extract PSD and XCF Gimp layers

Python_GUI/extractPSDLayers/__init__.py

@@ -0,0 +1 @@
+

Python_GUI/extractPSDLayers/extractPSDLayers.py

@@ -0,0 +1,180 @@
+#This Source Code Form is subject to the terms of the Mozilla Public
+#License, v. 2.0. If a copy of the MPL was not distributed with this
+#file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+# Derived from Fabrice Fernandez extractEXRLayers. Implemented by CGVIRUS
+
+import os
+import string
+import NatronEngine
+from NatronGui import *
+
+
+# EXTRACT PSD LAYERS #
+
+def extractPSDLayers():
+
+	# get current Natron instance running in memory #
+	app = natron.getGuiInstance(0)
+
+	# get selected nodes # 
+	selectedNodes = app.getSelectedNodes()
+
+	# cycle through every selected node #
+	for currentNode in selectedNodes:
+
+		# get node ID #
+		currentID = currentNode.getPluginID()
+
+		# check if selected node is a 'Read' node #
+		if currentID == 'fr.inria.built-in.Read':
+
+			# get 'Read' node position #
+			readPosition = currentNode.getPosition()
+
+
+
+
+			###################
+			# CREATE ROOT DOT #
+			###################
+
+			# create root dot #
+			rootDot = app.createNode('fr.inria.built-in.Dot')
+
+			# set root dot position #
+			rootDot.setPosition(readPosition[0] + 45 , readPosition[1] + 300)
+
+			# connect root dot to 'Read' #
+			rootDot.connectInput(0,currentNode)
+
+			# get root dot position #
+			rootDotPosition = rootDot.getPosition()
+
+
+
+
+
+			# get all available layers in PSD #
+			if hasattr(currentNode.getParam('layer'),'getOptions'):
+				layersList = currentNode.getParam('layer').getOptions()
+			else:
+				layersList = currentNode.getParam('outputLayer').getOptions()
+
+			# sort list alphabetically #
+			list.sort(layersList)
+
+			backdropLength = len(layersList)
+
+			# initialize counter #
+			channelCounter = 0
+
+			# cycle through every layer #
+			for choice in layersList:
+
+				if choice != 'Default' or 'Color.RGBA':
+
+					# full layer name #
+					LayerName = choice
+
+					# layer name #
+					layerName = os.path.splitext(choice)[0]
+
+					print layerName
+
+					# layer channels (RGBA,RGB,XYZ,UV,A,Z) #
+					layerChannels = os.path.splitext(choice)[1]
+
+					# remove '.' from channels #
+					layerChannels = layerChannels.replace('.','')
+
+
+
+
+					# create the first 'Shuffle' node #
+					if channelCounter == 0 :
+
+						###################
+						# CREATE SHUFFLE  #
+						###################
+
+						# create a 'Shuffle' node #
+						newShuffle = app.createNode('net.sf.openfx.ShufflePlugin')
+
+						# set 'Shuffle' label #
+						newShuffle.setLabel(str(layerName))
+
+						# enable preview #
+						newShuffle.getParam('enablePreview').setValue(1)
+						newShuffle.getParam('enablePreview').setValue(0)
+						newShuffle.getParam('enablePreview').setValue(1)
+
+						# set 'Shuffle' node position #
+						newShuffle.setPosition(rootDotPosition[0] - 45 , rootDotPosition[1] + 200)
+
+						# set 'Shuffle' color #
+						newShuffle.setColor(1, 0.5, 0.15)
+
+						# connect 'Shuffle' to root dot #
+						newShuffle.connectInput(0,rootDot)
+
+						# create a 'Backdrop' #
+						newBackdrop = app.createNode('fr.inria.built-in.BackDrop')
+						newBackdrop.setPosition(rootDotPosition[0] - 200 , rootDotPosition[1] - 120)
+						newBackdrop.setSize( (backdropLength)*400, 500 )
+						newBackdrop.setColor(0.5, 0.35, 0.12)
+
+
+					# create all the other 'Shuffle' nodes #
+					if channelCounter != 0 :
+
+						# create a new dot #
+						newDot = app.createNode('fr.inria.built-in.Dot')
+
+						# set root dot position #
+						newDot.setPosition(rootDotPosition[0] + 400 , rootDotPosition[1])
+
+						# connect root dot to previous dot #
+						newDot.connectInput(0,rootDot)
+
+						# replace old Dot position value
+						rootDotPosition = newDot.getPosition()
+
+
+						###################
+						# CREATE SHUFFLE  #
+						###################
+
+
+						# create a 'Shuffle' node #
+						newShuffle = app.createNode('net.sf.openfx.ShufflePlugin')
+
+						# set 'Shuffle' label #
+						newShuffle.setLabel(str(layerName))
+
+						# enable preview #
+						newShuffle.getParam('enablePreview').setValue(1)
+						newShuffle.getParam('enablePreview').setValue(0)
+						newShuffle.getParam('enablePreview').setValue(1)
+
+						# set 'Shuffle' node position #
+						newShuffle.setPosition(rootDotPosition[0] - 45 , rootDotPosition[1] + 200)
+
+						# set 'Shuffle' color #
+						newShuffle.setColor(1, 0.5, 0.15)
+
+						# connect 'Shuffle' to root dot #
+						newShuffle.connectInput(0,newDot)
+
+
+
+
+					##########################
+					# SET SHUFFLE PARAMETERS #
+					##########################
+
+					newShuffleValue = 'B.' + str(layerName) + '.' + 'R'
+					newShuffle.getParam('outputR').set(newShuffleValue)
+
+
+					# increase counter #
+					channelCounter += 1

initGui.py

@@ -26,6 +26,7 @@
 # Tools -> Channel
 from Python_GUI.autoAlpha.autoAlpha import *
 from Python_GUI.extractExrLayers.extractExrLayers import *
+from Python_GUI.extractPSDLayers.extractPSDLayers import *
 
 # Tools -> Generate
 from Python_GUI.postageStamp.postageStamp import *
@@ -86,6 +87,7 @@
 
 NatronGui.natron.addMenuCommand('Tools/Channel/Auto alpha','autoAlpha')
 NatronGui.natron.addMenuCommand('Tools/Channel/Extract EXR layers','extractExrLayers')
+NatronGui.natron.addMenuCommand('Tools/Channel/Extract PSD or XCF layers','extractPSDLayers')
 
 NatronGui.natron.addMenuCommand('Tools/Generate/PostageStamp','postageStamp', QtCore.Qt.Key.Key_P, QtCore.Qt.KeyboardModifier.ControlModifier | QtCore.Qt.AltModifier)
 NatronGui.natron.addMenuCommand('Tools/Generate/Roto to tracker','rotoToTracker')
@@ -104,8 +106,8 @@
 NatronGui.natron.addMenuCommand('Edit/Invert selection','invertSelection', QtCore.Qt.Key.Key_I, QtCore.Qt.KeyboardModifier.ControlModifier | QtCore.Qt.ShiftModifier)
 NatronGui.natron.addMenuCommand('Edit/Remove input','removeInput', QtCore.Qt.Key.Key_D, QtCore.Qt.KeyboardModifier.ControlModifier)
 
-NatronGui.natron.addMenuCommand('Tools/Other/Blending mode+','mergeBlendingDown', QtCore.Qt.Key.Key_Down, QtCore.Qt.KeyboardModifier)
-NatronGui.natron.addMenuCommand('Tools/Other/Blending mode-','mergeBlendingUp', QtCore.Qt.Key.Key_Up, QtCore.Qt.KeyboardModifier)
+NatronGui.natron.addMenuCommand('Tools/Other/Blending mode+','mergeBlendingDown', QtCore.Qt.Key.Key_Down, QtCore.Qt.AltModifier)
+NatronGui.natron.addMenuCommand('Tools/Other/Blending mode-','mergeBlendingUp', QtCore.Qt.Key.Key_Up, QtCore.Qt.AltModifier)
 NatronGui.natron.addMenuCommand('Tools/Other/Link roto to tracker','rotoLink', QtCore.Qt.Key.Key_L, QtCore.Qt.KeyboardModifier.ControlModifier | QtCore.Qt.ShiftModifier)
 
 NatronGui.natron.addMenuCommand('Tools/Roto/Circle','fullCircle()')
@@ -143,6 +145,7 @@
 print '\n'
 print '   + Tools/Channel/Auto Alpha'
 print '   + Tools/Channel/Extract EXR layers'
+print '   + Tools/Channel/Extract PSD/XCF layers'
 print '   +'
 print '   + Tools/Generate/PostageStamp'
 print '   + Tools/Generate/Roto to tracker'