fcpxml_to_nuke_v02

'''
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 = 'XML File to Process'
    output_dir = 'Directory to Output Nuke Scripts' #"set dir for outputting nuke scripts"
    subdirectories = 'Create subdirectories for each script?'
    render_dir = 'Render Directory for All Write Nodes' #"set render dir for 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 = pathurl_clip.getElementsByTagName('pathurl')[0].firstChild.data.split("file://localhost")[1].replace("%20", " ")
        except:
            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 "These 2 names should be equal: ", file_name, " : ", clip_name
        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':
                        # !!! This is calculating the move from the sequence resolution center, not the clip resolution center, as it should.
                        # Should this be done with a reformat node instead? This is a hard problem.
                        x_move = clip_width * float( param.getElementsByTagName('value')[0].childNodes[1].firstChild.data )
                        y_move = clip_height * float( param.getElementsByTagName('value')[0].childNodes[3].firstChild.data )
                        print effect_name, param_id, x_move, 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 Handles and set first_frame and last_frame
        first_frame = in_point - handle_length
        last_frame = out_point-1 + handle_length
        if timeremap_value:
            first_frame = round(first_frame/(100/timeremap_value))
            last_frame = round(last_frame/(100/timeremap_value))

        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 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
 name Reformat1
 selected true
}}
'''.format())

            nuke_script.write( '''Transform {{
 translate {{{0} {1}}}
 rotate {2}
 scale {3}
 center {{{4} {5}}}
 name Transform1
 selected true
 }}
'''.format( x_move, y_move, rotation_value, scale_value/100, seq_res_x/2, seq_res_y/2 ) )
        # 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/100, first_frame ))

        # 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)
        '''