def __init__( self ):

"""Initializes the Maya render job submission window."""

QMainWindow.__init__( self )

self.setupUi( self ) #self, self

QObject.connect(self.submitButton, SIGNAL("clicked()"),

self.doSubmit)

QObject.connect(self.browseButton, SIGNAL("clicked()"),

self.setMayaProjectPath)

self.requirementsListWidget.setSelectionMode (QAbstractItemView.MultiSelection)

if len (sys.argv) < 2:

sys.argv.append ('???')

sys.argv.extend (['1', '1', '1'])

scene, start, end, by = sys.argv[1:5] # _TODO: proper command line args

scene = scene.replace ('\\', '/')

self.sceneText.setText(scene)

self.startSpinBox.setValue( eval (start ) )

self.endSpinBox.setValue( eval( end ) )

self.populateProjectComboBox()

self.populateExecutableComboBox()

self.populateRequirementsListWidget()

def populateProjectComboBox(self):

"""Populates the projects dropdown box."""

# get current list of projects from the database

tuples = None

with transaction() as t:

t.cur.execute("select * from Hydra_projects")

tuples = t.cur.fetchall()

# flatten list of tuples fetched from the database

projectsList = [t for (t,) in tuples]

# populate the dropdown

for project in projectsList:

self.projectComboBox.addItem(project)

# show default project selection

[thisNode] = Hydra_rendernode.fetch(

"where host = '%s'" % Utils.myHostName())

idx = self.projectComboBox.findText(

thisNode.project,

flags=Qt.MatchExactly|Qt.MatchCaseSensitive)

self.projectComboBox.setCurrentIndex(idx)

def populateExecutableComboBox(self):

"""Populates the executables dropdown box."""

# get current list of executables from the database

tuples = None

with transaction() as t:

t.cur.execute("select * from Hydra_executable")

tuples = t.cur.fetchall()

# flatten list of tuples fetched from the database

executables = [t for (t,) in tuples]

# populate the dropdown

for program in executables:

self.executableComboBox.addItem(program)

def populateRequirementsListWidget(self):

"""Populates the requirements list widget."""

# get current list of capabilities from the database

tuples = None

with transaction() as t:

t.cur.execute("select * from Hydra_capabilities order by name")

tuples = t.cur.fetchall()

# flatten list of tuples fetched from the database

capabilities = [t for (t,) in tuples]

# populate the dropdown

for item in capabilities:

self.requirementsListWidget.addItem(item)

## # show default project selection

## [thisNode] = Hydra_rendernode.fetch(

## "where host = '%s'" % Utils.myHostName())

## idx = self.projectComboBox.findText(

## thisNode.project,

## flags=Qt.MatchExactly|Qt.MatchCaseSensitive)

## self.projectComboBox.setCurrentIndex(idx)

# called by pressing the "submit" button in the Qt main window

def doSubmit( self ):

"""Submits a job ticket for this scene to be split into

tasks and processed."""

logger.debug ('doSubmit')

reqs = sorted ([str (req.text ())

for req in self.requirementsListWidget.selectedItems ()])

reqs_pattern = '%' + '%'.join (reqs) + '%' if reqs else '%'

print reqs_pattern

sceneFile = str( self.sceneText.text() ).replace ('\\', '/')

startFrame = self.startSpinBox.value( )

endFrame = self.endSpinBox.value( )

numJobs = self.numJobsSpinBox.value( )

batchSize = int(math.ceil((endFrame - startFrame + 1.0) / numJobs))

logger.debug ("numJobs %s batchSize %s", numJobs, batchSize)

priority = self.prioritySpinBox.value( )

project = str(self.projectComboBox.currentText())

executable = str(self.executableComboBox.currentText())

mayaProjectPath = str(self.projectDirLineEdit.text())

if not os.path.exists(os.path.join (mayaProjectPath, "workspace.mel")):

# try to find workspace.mel

mayaProjectPath = self.getMayaProjectPath(sceneFile)

if not mayaProjectPath:

logger.debug("workspace.mel not found")

aboutBox(self, "Error", """

The project path cannot be set because workspace.mel could not

be located. Please set the project path manually.""")

return

if yesNoBox(self, "Confirm",

"Maya project path set to:<br>" +

mayaProjectPath +

"<br> Is this correct?") == QMessageBox.No:

aboutBox(self, "Abort",

"Submission aborted. Please set the Maya project path manually.")

return

self.projectDirLineEdit.setText(mayaProjectPath)

# executable names a class in the JobTicket module

ticketClass = getattr (JobTicket, executable)

ticketClass(sceneFile, mayaProjectPath, startFrame, endFrame, batchSize,

priority, project, reqs_pattern).submit()

aboutBox(self, "Success",

"Job submitted. Please close the submitter window.")

def getMayaProjectPath(self, scenePath):

"""

Walks up the file tree looking for workspace.mel,

returns the path if found"""

# remove Maya scene file name from the end of the path

mayaProjectPath = os.path.dirname (scenePath)

lastPath = None

wrkspc = "workspace.mel"

while not os.path.exists(os.path.join(mayaProjectPath, wrkspc)):

logger.debug ("%s not in %s", wrkspc, mayaProjectPath)

lastPath = mayaProjectPath

mayaProjectPath = os.path.dirname (mayaProjectPath)

if lastPath == mayaProjectPath:

return ""

return mayaProjectPath

def setMayaProjectPath(self):

"""

Opens a file browser dialog for finding workspace.mel,

tries to start the user somewhere sensible"""

currentDir = str( self.projectDirLineEdit.text() )

startDir = None

if len(currentDir) == 0:

sceneFile = str( self.sceneText.text() )

startDir = self.getMayaProjectPath(sceneFile)

if not startDir:

startDir = os.getcwd()

else:

startDir = currentDir

mayaProjectPath = QFileDialog.getOpenFileName(

parent=self,

caption="Find workspace.mel",

directory=startDir,

filter="workspace.mel")

if mayaProjectPath:

# remove "workspace.mel" from the end of the path

mayaProjectPath = str(mayaProjectPath).split('/')

mayaProjectPath.pop()

mayaProjectPath = '/'.join(mayaProjectPath) + '/'