Create a new version of paste:

'''
FCP XML to Nuke
v1.1 - Made support for Premiere FCPXML format more robust. Premiere FCP XML stores all sequences in the project, whereas
Final Cut Pro's XML format stores only a single sequence. I added a box that lets you choose what sequence 
of the Premiere XML file to process, and it should work more reliably now.
v1.0 - Initial release.

This script takes a Final Cut XML file with a single flattened video track, and builds Nuke scripts for each clip in the timeline.
It is intended as a simple way to automate workflows between FCP/Premiere and Nuke.
It creates a Nuke script with global first and last frame set, a frameRange node with the proper framerange, and a Write node
set to the output path.
There is an option for creating subdirectories for every Nuke script created. Handles are also an option.
It can parse reel number and clip number from Red and Alexa footage, or can use the clip filename as the base naming for the output files.

This script was somewhat inspired by compflows.blogspot.com, but has been written from scratch and is a bit more flexible (although it only goes from XML->NukeScripts and not back at the moment).
This has only been tested on OSX, but in theory should be cross-platform compatible. Comments and suggestions are welcome!


# The menu.py example entry below adds this script in a folder called "Scripts" in your toolbar.
import fcpxml_to_nuke
nuke.toolbar('Nodes').addMenu('Scripts').addCommand('FCP XML to Nuke', 'fcpxml_to_nuke.process_xml()')
'''

import nuke, os
from xml.dom.minidom import parse

def process_xml():
	'''
	Imports an FCP XML file, locates each clip in the timeline on Track 1, 
	and for each clip, builds a nuke script for that file and puts it in the output directory specified

	New Features that would be nice to have: 
	Customized naming patterns based on reel/clip number.
	Handle FCP XML from Premiere or FCP / FCPX ( Are there differences in the XML structure for these? )
	Choose additional output naming and directory formatting patterns
	'''

	# Build the Nuke Panel where locations and stuff is specified.
	p = nuke.Panel("FCP XML Import")
	xml_file = '/Volumes/heka/Programming/nuke-scripts/fcpxml_to_nuke/fcpxml_test 960 0.xml' #FCP XML Import
	output_dir = '/Volumes/heka/Programming/nuke-scripts/fcpxml_to_nuke/output' #Directory to Output Nuke Scripts
	subdirectories = 'Create subdirectories for each script?'
	render_dir = '/Volumes/heka/Programming/nuke-scripts/fcpxml_to_nuke/output' #Render Directory for All Write Nodes
	handle_length = "0 5 10 15 20 30 35 40"
	clip_name_format = "Bypass RED Alexa"

	p.addFilenameSearch("FCP XML File", xml_file)
	p.addBooleanCheckBox("Create subdirectories for each script", subdirectories)
	p.addFilenameSearch("Output Directory", output_dir)
	p.addFilenameSearch("Render Directory", render_dir)
	p.addEnumerationPulldown("Handle Length", handle_length)
	p.addEnumerationPulldown("Clip Name Format", clip_name_format)
	p.setWidth(600)
	if not p.show():
		return

	# Assign vars from Nuke Panel user-entered data 
	xml_file 	= p.value("FCP XML File")
	output_dir	= p.value("Output Directory")
	subdirectories = p.value("Create subdirectories for each script")
	render_dir 	= p.value("Render Directory")
	handle_length = int( p.value("Handle Length") )
	clip_name_format = p.value("Clip Name Format")

	# Create paths for render directory if it does not exist
	if not os.path.isdir(render_dir):
		os.mkdir(render_dir)
	if not os.path.isdir(output_dir):
		os.mkdir(output_dir)


	############################
	# Begin Parsing XML File
	############################
	dom = parse( xml_file )

	# Prompt user to choose which sequence to process, because Premiere's XML export includes all sequences.
	# Much of this complexity is to handle sequences with spaces in their names, because the nuke.panel
	# addEnumerationPulldown is space-demarcated. The logic below handles spaces in sequence names and allows the user 
	# to choose which sequence to process, and stores the sequence to process as a variable to access

	sequences = []
	i = 0
	# Makes a list of all sequence names
	for seq in dom.getElementsByTagName('sequence'):
		seqname = seq.getElementsByTagName('name')[0].firstChild.data
		sequences.append( [seqname] )
		# If a sequence name has space characters, replace them with underscores for the Nuke enumeration pulldown panel
		if seqname.find(' ') != -1:
			sequences[i].append( seqname.replace(' ', '_') )
		i += 1
	print "Sequences are: ", sequences
	# ??? Do all of the below messy work to prepare for deciding between multiple sequences, including dealing with spaces in sequence names
	# but only if there is more than one sequence in the XML file. If not, we'll just set the 'row' var to 0, and roll with that.
	if len(sequences) > 1:
		# Makes a " " demarcated string with all sequence names for the nuke enumeration pulldown panel
		seq_enum = ''
		i = 0
		for seq in sequences:
			if len(sequences[i]) == 1:
				seq_enum += sequences[i][0] + ' '
			else:
				seq_enum += sequences[i][1] + ' '
			i += 1

		# Create a Nuke panel for the user to choose which sequence to process
		seq_panel = nuke.Panel("Choose Sequence To Process")
		seq_panel.addEnumerationPulldown("Choose Sequence", seq_enum)
		seq_panel.setWidth(400)
		if not seq_panel.show():
			return
		chosen_sequence = seq_panel.value("Choose Sequence")

		# Gets the index of the chosen sequence in the list of sequences, stores the sequence XML object as a variable
		for row, i in enumerate(sequences):
				try:
					column = i.index( chosen_sequence )
				except ValueError:
					continue
				break
	else:
		row = 0
	chosen_seqobj = dom.getElementsByTagName('sequence')[row]
	print "Chosen sequence is: ", chosen_seqobj.getElementsByTagName('name')[0].firstChild.data

	seq_res_x = int( chosen_seqobj.getElementsByTagName('format')[0].getElementsByTagName('width')[0].firstChild.data )
	seq_res_y = int( chosen_seqobj.getElementsByTagName('format')[0].getElementsByTagName('height')[0].firstChild.data )
	print "Sequence resolution is ", seq_res_x,"x",seq_res_y

	# Set optional effect parameters to False
	timeremap_value = False
	scale_value = False
	x_move = False
	y_move = False
	rotation_value = False

	seq_clip_number = 1
	track = chosen_seqobj.getElementsByTagName('track')[0]
	for clip in track.getElementsByTagName('clipitem'):
		masterclipid 	= clip.getElementsByTagName('masterclipid')[0].firstChild.data
		clip_name 		= clip.getElementsByTagName("name")[0].firstChild.data
		in_point 		= int( clip.getElementsByTagName('in')[0].firstChild.data )
		out_point 		= int( clip.getElementsByTagName('out')[0].firstChild.data )
		clip_duration 	= int( clip.getElementsByTagName("duration")[0].firstChild.data )
		
		# Fetch the pathurl of the clip by cycling through all <pathurl> nodes and comparing the filename of the clip to the clip_name
		# This is necessary because in Premiere XMLs, the pathurl for a clip is not always stored in the clipitem node, but rather in a seperate node in the master-clip
		
		#??? Instead: Check for pathurl node in current clip node. If it doesn't exist, cycle through all 
		# clipitem nodes to find another that matches the name node with the current clip_name.
		# If it finds a matching named clipitem, search for a pathurl in that node.
		try:
			file_path = clip.getElementsByTagName('pathurl')[0].firstChild.data.split("file://localhost")[1].replace("%20", " ")

		except:
			print 'Failed to get pathurl in current clipitem. Searching other clipitems for matching name.'
			for pathurl_clip in dom.getElementsByTagName('clipitem'):
				if pathurl_clip.getElementsByTagName('name')[0].firstChild.data == clip_name:
					try:
						file_path = pathurl_clip.getElementsByTagName('pathurl')[0].firstChild.data.split("file://localhost")[1].replace("%20", " ")
						break
					except:
						continue
		print clip_name, in_point, out_point, clip_duration, file_path

		# Get resolution of this clip in the clipitem node, 
		# Else, look for a masterclip with the same name and try to get the resolution from there, Else: fail?
		try:
			clip_width = int( clip.getElementsByTagName('width')[0].firstChild.data )
			clip_height = int( clip.getElementsByTagName('height')[0].firstChild.data )
			print "Clip resolution is: ", clip_width, "x", clip_height
		except:	
			for masterclip in dom.getElementsByTagName('clipitem'):
				if masterclip.getElementsByTagName('name')[0].firstChild.data == clip_name:
					#!!! This is triggered if a clip is used more than once in the sequence.
					try:
						#print "found master clip: ", masterclip.getElementsByTagName('name')[0].firstChild.data, " and "
						clip_width = int( masterclip.getElementsByTagName('width')[0].firstChild.data )
						clip_height = int( masterclip.getElementsByTagName('height')[0].firstChild.data )
						break
					except:
						continue
		# Get all effects applied to this clip
		for effect in clip.getElementsByTagName('effect'):
			effect_name = effect.childNodes[1].firstChild.data
			if effect_name == 'Time Remap':
				# Loop through all parameters of the effect
				for param in effect.getElementsByTagName('parameter'):
					param_id = param.childNodes[1].firstChild.data
					if param_id == 'speed':
						timeremap_value = float( param.getElementsByTagName('value')[0].firstChild.data )
						print effect_name, param_id, timeremap_value

			if effect_name == 'Basic Motion':
				for param in effect.getElementsByTagName('parameter'):
					param_id = param.childNodes[1].firstChild.data
					if param_id == 'scale':
						scale_value = float( param.getElementsByTagName('value')[0].firstChild.data )
						print effect_name, param_id, scale_value
					if param_id == 'rotation':
						rotation_value = float( param.getElementsByTagName('value')[0].firstChild.data )
						print effect_name, param_id, rotation_value
					if param_id == 'center':
						x_move = float( param.getElementsByTagName('value')[0].childNodes[1].firstChild.data )
						y_move = float( param.getElementsByTagName('value')[0].childNodes[3].firstChild.data )
						print effect_name, param_id, x_move, y_move
						'''
						So.... Figuring out how Premiere handles position values:
						Prem 0-0 clip is centered upper left: value = .5,.5
						prem 1920-1080, clip is centered lower right: value = .5 .5??
						prem 1060-640, value: 0.052083 0.092593
						prem center bottom: 960-1080, value: 0.0 0.5
						prem center top: 960 0, value: 0.0 -0.5
						prem UR 1919 0, value: 0.499479 -0.5
						
						location 	x, y
						center 		0, 0
						UL			-0.5, -0.5
						UR 			0.5, -0.5
						LR 			0.5, 0.5
						LL 			0, 0.5
						y-up is negative
						x-right is positive
						The range from edge to edge of sequence space is 1. 
						-.5 is left, .5 is right. 
						-.5 is up, .5 is down.
						.125, 0 would be 1200x540 = (seq_res_x * x_move) = how many pixels to move from center) = 1920*.125 + 1920/2 = 1200
						for x: seq_res_x * x_move
						for y: seq_res_y * -y_move
						'''
		
		# Gets the shot name, which is the formatted clip_name with clip# and reel#, with the sequence clipnumber.
		# uses camera type (Red, Alexa, etc), and the clip_name string (the filename of the clip used in FCP) 
		# Also takes the seq_clip_number for returning the correct shot_name (the name that will be used to name the nuke script)
		if clip_name_format == 'RED':
			# This works for Red footage of format: A###_C###_RANDOMDATA
			reel_number = clip_name.split('_')[0][1:]
			clip_number = clip_name.split('_')[1][1:]

		if clip_name_format == 'Alexa':
			# Alexa footage is A###C###_######_R####
			reel_number = int(clip_name.split('C')[0][1:])
			clip_number = int( clip_name.split('C')[1].split('_')[0] )

		if clip_name_format == 'Bypass':
			shot_name = "%02d0_%s" %(seq_clip_number, os.path.splitext(clip_name)[0])
		else:
			# shot_name is the string that defines the name that the nuke script is saved to. seq_clip_number+0_A{reelnumber}_C{clipnumber}
			shot_name = "%02d0_A%sC%s" %(seq_clip_number, reel_number, clip_number)


		############################
		# Build Nuke Script
		############################
		
		# if the subdirectories checkbox is checked, set the output_shotdir to be a subdirectory named with the shot_name
		if subdirectories:
			output_shotdir = output_dir
			output_shotdir = os.path.join(output_dir, shot_name)
		else:
			output_shotdir = output_dir
		# If the output_shotdir does not exist, create it (auto-creates subdirectories)
		if not os.path.isdir(output_shotdir):
				os.mkdir(output_shotdir)

		
		###########################
		# Compute values to plug into the Nuke Script

		# Compute Handles and set first_frame and last_frame
		first_frame = in_point - handle_length
		last_frame = out_point-1 + handle_length

		if timeremap_value:
			timeremap_value 	= timeremap_value/100
			new_clip_duration 	= last_frame - first_frame
			clip_duration 		= clip_duration * timeremap_value
			first_frame 		= first_frame * timeremap_value
			last_frame 			= first_frame + new_clip_duration
			
			'''
			you have the duration of the original clip, and the in and out points of the retimed clip 
			originalIn = newIn * retime
			originalOut = newOut * retime
			new duration = lastFrame - first_frame
			'''


		# Set Format 
		if seq_res_x == 1920 and seq_res_y == 1080:
			fcp_xml_resolution = 'HD'
		else:
			fcp_xml_resolution = 'from_xml'


		#!!! This creates a nuke script by appending text to the .nk file instead of using nuke.nodeCopy(), which is slow and messy, and so that root node settings can be added.
		# The strings that are written are triple-quoted. newlines are created with '\n'. {} chars in the string have to be doubled so as not to throw a KeyError
		nuke_file = os.path.join(output_shotdir, "%s_v001.nk"%(shot_name))
		nuke_script = open(nuke_file, 'a+')
		# Create Root node
		nuke_script.write('''Root {{\n inputs 0\n name {0}\n project_directory \"\[python \{{nuke.script_directory()\}}]\"\n first_frame {1}\n last_frame {2}\n format \"{3} {4} 0 0 {3} {4} 1 {5}\"\n proxy_type scale\n}}\n'''.format(nuke_file, first_frame, last_frame, seq_res_x, seq_res_y, fcp_xml_resolution))
		# Create Read node
		nuke_script.write('''Read {{\n inputs 0\n file \"{0}\"\n first 0\n last {1}\n frame_mode offset\n frame 1\n origlast {1}\n origset true\n name Read1\n selected true\n xpos -425\n ypos -40\n}}\n'''.format(file_path, clip_duration))
		# Create TimeRemap if there is retiming on the clip
		if timeremap_value:
			nuke_script.write('''Group {{
 name RetimeFromFrame
 selected true
 addUserKnob {{20 Retime t "Retime From Frame Parameters"}}
 addUserKnob {{41 StartFrame l "SourceStart Frame" t "The source frame from which retiming starts. For example, if you have a clip that you are using a range from frames 200-300 in, and you want to retime that clip to be 50\% speed, you would set this to be 200. \\n\\nThis gizmo references the root.first_frame value to determine the \\\"in-point\\\" of the clip." T RetimeControls.StartFrame}}
 addUserKnob {{41 PlaybackSpeed l "Playback Speed" t "Retime speed as a fraction of one. That is, 0.5 = 50\% speed, 2 = 200\% speed." T RetimeControls.PlaybackSpeed}}
}}
 Input {{
  inputs 0
  name Input1
  xpos 0
 }}
 TimeOffset {{
  time_offset {{{{-RetimeControls.StartFrame/RetimeScreen.timingSpeed}}}}
  name Retime_TimeOffset
  tile_color 0xff0000ff
  xpos 0
  ypos 132
 }}
 OFXuk.co.thefoundry.time.oflow_v100 {{
  method Motion
  timing Speed
  timingFrame 1
  timingSpeed {{{{RetimeControls.PlaybackSpeed}}}}
  filtering Normal
  warpMode Normal
  correctLuminance false
  automaticShutterTime false
  shutterTime 0
  shutterSamples 1
  vectorDetail 0.2
  smoothness 0.5
  blockSize 6
  Tolerances 0
  weightRed 0.3
  weightGreen 0.6
  weightBlue 0.1
  showVectors false
  cacheBreaker false
  name RetimeScreen
  tile_color 0xff0000ff
  selected true
  xpos 0
  ypos 156
 }}
 TimeOffset {{
  time_offset {{{{root.first_frame}}}}
  name GlobalStart_Offset
  tile_color 0xff0000ff
  xpos 0
  ypos 180
 }}
 Output {{
  name Output1
  xpos 0
  ypos 393
 }}
 NoOp {{
  inputs 0
  name RetimeControls
  xpos -174
  ypos 126
  addUserKnob {{20 User}}
  addUserKnob {{7 PlaybackSpeed l "Playback Speed" R 0 100}}
  PlaybackSpeed {0}
  addUserKnob {{3 StartFrame l "Start Frame" t "Offset video start frame"}}
  StartFrame {1}
 }}
end_group\n'''.format( timeremap_value, first_frame ))
		
		# Create FrameRange node 
		nuke_script.write('''FrameRange {{\n first_frame {0}\n last_frame {1}\n name FrameRange1\n label "\\[knob first_frame]-\\[knob last_frame]"\n selected true\n}}\n'''.format(first_frame, last_frame))
		
		# Create a Transform node with pans and scales, if they exist
		if 	scale_value or x_move or y_move or rotation_value:
			if not scale_value:
				scale_value = 1
			if not x_move:
				x_move = 0
			if not y_move:
				y_move = 0
			if not rotation_value:
				rotation_value = 0
			# Create a reformat node if there are pans or scales
			nuke_script.write('''Reformat {{
 resize none
 black_outside true
 name Reformat1
 selected true
}}
'''.format())
			nuke_script.write( '''Transform {{
 translate {{{0} {1}}}
 rotate {2}
 scale {3}
 center {{{4} {5}}}
 name Transform1
 selected true
 }}
'''.format( (seq_res_x * x_move), (seq_res_y * -y_move), -rotation_value, scale_value/100, seq_res_x/2, seq_res_y/2 ) )

		# Create Write node
		nuke_script.write('''Write {{\n file \"{0}\"\n file_type mov\n codec apch\n fps 23.976\n checkHashOnRead false\n name Write1\n selected true\n}}\n'''.format('{0}_v001.mov'.format(os.path.join(render_dir, shot_name)) ) )
		# Create Viewer node
		nuke_script.write('''Viewer {\n name Viewer1\n selected true\n}\n''')
		# Create Backdrop node
		nuke_script.write('''BackdropNode {{\n inputs 0\n name BackdropNode1\n tile_color 0x26434dff\n label \"<img src=\\\"Read.png\\\"> Read Plate <br/><font size=1> {0} <br/> {1}-{2}<br/>{3} frame handles\"\n note_font_size 30\n selected true\n xpos -500\n ypos -150\n bdwidth 234\n bdheight 254\n}}\n'''.format(shot_name, in_point, out_point, handle_length))


		seq_clip_number += 1
		# Reset option effect parameters to false for next clip iteration
		timeremap_value = False
		scale_value = False
		x_move = False
		y_move = False
		rotation_value = False
		### End of for loop which processes each clip in timeline.


	nuke.message('All clips processed successfully!')
	return


'''
		### ??? ALTERNATIVE METHOD FOR CREATING THE SCRIPT FILES
		### This approach uses the script that import_xml is executed from as a base for creating nodes, and then using nuke.nodeCopy() to 'paste' the data into each script file.
		### The approach I ended up using instead just uses python to format .nk scripts as text, inputting the variables where relevant. It is much faster and more efficient.
		for node in nuke.allNodes():
			node.setSelected(True)
		nuke.nodeDelete()

		# Create Read node
		read = nuke.createNode("Read")
		read.knob("file").setValue(file_path)
		read.knob("frame_mode").setValue('offset')
		read.knob("frame").setValue('1')
		read.knob("first").setValue(0)
		read.knob("last").setValue(clip_duration)

		# Create a NoOp node to hold shot info
		#!!! This is not needed
		#shotInfo = nuke.createNode('NoOp')
		#shotInfo.knob('name').setValue(shot_name+"_info")
		#shotInfo.addKnob( nuke.String_Knob("The original name of the clip", "clip_name", clip_name) )
		#shotInfo.addKnob( nuke.String_Knob("Base name of the nuke script", "shot_name", shot_name) )

		# Create FrameRange node
		frame_range = nuke.createNode("FrameRange")
		frame_range.knob('label').setValue('[knob first_frame]-[knob last_frame]')
		frame_range.knob('first_frame').setValue( in_point - handle_length ) 
		frame_range.knob('last_frame').setValue( out_point-1 + handle_length )
		
		# Create Write node
		write = nuke.createNode("Write")
		write.knob('file').setValue('{0}{1}_v001.mov'.format(render_dir, shot_name))
		#write.knob("file_type").setValue("mov")
		#write.knob("mov.codec").setValue("apch")
		#write.knob("mov.fps").setValue("23.976")

		# Create Viewer Node
		nuke.createNode('Viewer')

		# Informational Backdrop Node
		bd_node = nuke.createNode("BackdropNode")
		bd_node.knob("tile_color").setValue(0x26434dff)
		bd_node.knob("note_font_size").setValue(30)
		bd_node.knob("bdwidth").setValue(234)
		bd_node.knob("bdheight").setValue(254)
		bd_node.knob("label").setValue('<img src=\"Read.png\"> Read Plate <br/><font size=1> %s <br/> %s-%s'%(shot_name,int(in_point),int(out_point)))

		# Set root script values
		#nuke.toNode('root').knob('project_directory').setValue('[python {nuke.script_directory()}]')
		#nuke.toNode('root').knob('first_frame').setValue( in_point-handle_length)
		#nuke.toNode('root').knob('first_frame').setValue( (out_point-1) + handle_length)
		#nuke.toNode('root').knob('format').setValue('HD')

		# Select all created nodes and copy them into a new script
		for node in nuke.allNodes():
			node.setSelected(True)
		nuke.nodeCopy(nuke_file)
		'''