/** @return The path of a node */
private Path getNodePath(String nodeName) throws PipelineException {
NodeID targetID = new NodeID(w.user, w.view, nodeName);
NodeStatus targetStat = mclient.status(targetID);
NodeMod targetMod = targetStat.getHeavyDetails().getWorkingVersion();
if (targetMod == null)
throw new PipelineException(
"No working version of the Target Scene Node ("
+ nodeName
+ ") exists "
+ "in the ("
+ w.view
+ ") working area owned by ("
+ PackageInfo.sUser
+ ")!");
Path targetPath;
FileSeq fseq = targetMod.getPrimarySequence();
String suffix = fseq.getFilePattern().getSuffix();
if (!fseq.isSingle() || (suffix == null) || (!suffix.equals("ma") && !suffix.equals("mb")))
throw new PipelineException("The target node (" + nodeName + ") must be a maya scene!");
targetPath =
new Path(PackageInfo.sProdPath, targetID.getWorkingParent() + "/" + fseq.getFile(0));
// System.err.println("$WORKING"+nodeName+"."+suffix);
return targetPath;
} // end getNodePath(String)
/**
* Using a list of shots and the assets to be changed per shot, gets each shot node and modifies
* the references as needed.
*/
private void processNodes() throws PipelineException {
/*each asset, its replacement and the lo-res versions*/
for (String asset : pAssetManager.keySet()) {
String newAsset = pAssetManager.get(asset).getNewAsset();
String hrAsset = asset.replace(lr, "");
String newHrAsset = newAsset.replace(lr, "");
logLine(
"Checking out nodes:\n\t"
+ asset
+ "\n\t"
+ hrAsset
+ "\n\t"
+ newAsset
+ "\n\t"
+ newHrAsset);
mclient.checkOut(w.user, w.view, asset, null, over, froz);
mclient.checkOut(w.user, w.view, newAsset, null, over, froz);
mclient.checkOut(w.user, w.view, hrAsset, null, over, froz);
mclient.checkOut(w.user, w.view, newHrAsset, null, over, froz);
} // end for
TreeMap<String, TreeSet<String>> shotBased = convertListToShotBased();
TreeMap<String, String> nameMap = SonyConstants.getCustomNamespaces(project);
TreeSet<String> oldRef = new TreeSet<String>();
TreeSet<String> newRef = new TreeSet<String>();
for (String shot : shotBased.keySet()) {
// check out the shot
if (shot.endsWith("anim")) mclient.checkOut(w.user, w.view, shot, null, keep, frozU);
else {
continue;
// mclient.checkOut(w.user, w.view, shot, null, keep, pFroz);
}
logLine("Checking out: " + shot);
NodeMod targetMod = mclient.getWorkingVersion(w.user, w.view, shot);
if (!shot.matches(lgtPattern)) {
if (targetMod.isActionEnabled()) {
System.err.println("Anim node with action enabled");
FileSeq fseq = targetMod.getPrimarySequence();
VersionID targetID = targetMod.getWorkingID();
TreeMap<String, VersionID> files = new TreeMap<String, VersionID>();
files.put(fseq.getFile(0).getPath(), targetID);
mclient.revertFiles(w.user, w.view, shot, files);
targetMod.setActionEnabled(false);
}
w.mclient.modifyProperties(w.user, w.view, targetMod);
}
/*--checking the shot to be modified---*/
for (String assetToReplace : shotBased.get(shot)) {
if (assetToReplace.endsWith(lr)) {
AssetInfo temp = pAssetManager.get(assetToReplace);
oldRef.add(assetToReplace);
newRef.add(temp.getNewAsset());
} else {
String hiRes = assetToReplace.replace(lr, "");
AssetInfo temp = pAssetManager.get(hiRes);
if (temp == null) continue;
oldRef.add(assetToReplace);
newRef.add(temp.getNewAsset() + lr);
} // end else
} // end for
if (oldRef.isEmpty() || newRef.isEmpty()) {
logLine("Shot " + shot + " somehow does not need any changes.");
continue;
} // end if
editShotReferences(shot, targetMod, oldRef, newRef, nameMap);
} // end for
/*for(String shot: shotBased.keySet()){
//TODO: Check in nodes
String msg = "Checked in with UpdateAssetGUI. Removed:\n\t";
for(String asset: oldRef)
msg+= (asset + " ");
msg+="\nAdded:\n\t";
for(String asset: newRef)
msg+= (asset + " ");
mclient.checkIn(w.user, w.view,shot, msg, VersionID.Level.Major);
}*/
} // end processShots
/**
* Construct a {@link SubProcessHeavy SubProcessHeavy} instance which when executed will fulfill
* the given action agenda.
*
* <p>
*
* @param agenda The agenda to be accomplished by the action.
* @param outFile The file to which all STDOUT output is redirected.
* @param errFile The file to which all STDERR output is redirected.
* @return The SubProcess which will fulfill the agenda.
* @throws PipelineException If unable to prepare a SubProcess due to illegal, missing or
* imcompatable information in the action agenda or a general failure of the prep method code.
*/
public SubProcessHeavy prep(ActionAgenda agenda, File outFile, File errFile)
throws PipelineException {
NodeID nodeID = agenda.getNodeID();
/* sanity checks */
String opname = null;
File source = null;
File preRender = null;
File postRender = null;
File preFrame = null;
File postFrame = null;
{
{
FileSeq fseq = agenda.getPrimaryTarget();
if (!fseq.hasFrameNumbers())
throw new PipelineException(
"The HfsComposite Action requires that the output images have frame numbers.");
}
/* generate the filename of the Houdini scene to load */
{
String sname = (String) getSingleParamValue("HoudiniScene");
if (sname != null) {
FileSeq fseq = agenda.getPrimarySource(sname);
if (fseq == null)
throw new PipelineException(
"Somehow the Houdini Scene node ("
+ sname
+ ") was not one of the source "
+ "nodes!");
String suffix = fseq.getFilePattern().getSuffix();
if (!fseq.isSingle() || (suffix == null) || !suffix.equals("hip"))
throw new PipelineException(
"The HfsComposite Action requires that the source node specified by the "
+ "Houdini Scene parameter ("
+ sname
+ ") must have a single Houdini scene "
+ "file (.hip) as its primary file sequence!");
NodeID snodeID = new NodeID(nodeID, sname);
source =
new File(PackageInfo.sProdDir, snodeID.getWorkingParent() + "/" + fseq.getFile(0));
} else {
throw new PipelineException(
"The HfsComposite Action requires the Houdini Scene parameter to be set!");
}
}
/* the full name of the geometry output operator */
{
String name = (String) getSingleParamValue("OutputOperator");
if ((name == null) || (name.length() == 0))
throw new PipelineException(
"The HfsComposite Action requires a valid Output Operator name!");
opname = ("/out/" + name);
}
/* command script files */
{
String sname = (String) getSingleParamValue("PreRenderScript");
if (sname != null) {
FileSeq fseq = agenda.getPrimarySource(sname);
if (fseq == null)
throw new PipelineException(
"Somehow the Pre Render Script node ("
+ sname
+ ") was not one of the "
+ "source nodes!");
String suffix = fseq.getFilePattern().getSuffix();
if (!fseq.isSingle() || (suffix == null) || !(suffix.equals("cmd")))
throw new PipelineException(
"The HfsComposite Action requires that the source node specified by the Pre "
+ "Render Script parameter ("
+ sname
+ ") must have a single command script "
+ "(.cmd) as its primary file sequence!");
NodeID snodeID = new NodeID(nodeID, sname);
preRender =
new File(PackageInfo.sProdDir, snodeID.getWorkingParent() + "/" + fseq.getFile(0));
}
}
{
String sname = (String) getSingleParamValue("PostRenderScript");
if (sname != null) {
FileSeq fseq = agenda.getPrimarySource(sname);
if (fseq == null)
throw new PipelineException(
"Somehow the Post Render Script node ("
+ sname
+ ") was not one of the "
+ "source nodes!");
String suffix = fseq.getFilePattern().getSuffix();
if (!fseq.isSingle() || (suffix == null) || !(suffix.equals("cmd")))
throw new PipelineException(
"The HfsComposite Action requires that the source node specified by the "
+ "Post Render Script parameter ("
+ sname
+ ") must have a single command "
+ "script (.cmd) as its primary file sequence!");
NodeID snodeID = new NodeID(nodeID, sname);
postRender =
new File(PackageInfo.sProdDir, snodeID.getWorkingParent() + "/" + fseq.getFile(0));
}
}
{
String sname = (String) getSingleParamValue("PreFrameScript");
if (sname != null) {
FileSeq fseq = agenda.getPrimarySource(sname);
if (fseq == null)
throw new PipelineException(
"Somehow the Pre Frame Script node ("
+ sname
+ ") was not one of the "
+ "source nodes!");
String suffix = fseq.getFilePattern().getSuffix();
if (!fseq.isSingle() || (suffix == null) || !(suffix.equals("cmd")))
throw new PipelineException(
"The HfsComposite Action requires that the source node specified by the Pre "
+ "Frame Script parameter ("
+ sname
+ ") must have a single command script "
+ "(.cmd) as its primary file sequence!");
NodeID snodeID = new NodeID(nodeID, sname);
preFrame =
new File(PackageInfo.sProdDir, snodeID.getWorkingParent() + "/" + fseq.getFile(0));
}
}
{
String sname = (String) getSingleParamValue("PostFrameScript");
if (sname != null) {
FileSeq fseq = agenda.getPrimarySource(sname);
if (fseq == null)
throw new PipelineException(
"Somehow the Post Frame Script node ("
+ sname
+ ") was not one of the "
+ "source nodes!");
String suffix = fseq.getFilePattern().getSuffix();
if (!fseq.isSingle() || (suffix == null) || !(suffix.equals("cmd")))
throw new PipelineException(
"The HfsComposite Action requires that the source node specified by the "
+ "Post Frame Script parameter ("
+ sname
+ ") must have a single command "
+ "script (.cmd) as its primary file sequence!");
NodeID snodeID = new NodeID(nodeID, sname);
postFrame =
new File(PackageInfo.sProdDir, snodeID.getWorkingParent() + "/" + fseq.getFile(0));
}
}
}
/* create the temporary Houdini command script */
File hscript = createTemp(agenda, 0644, "cmd");
try {
FileWriter out = new FileWriter(hscript);
FileSeq fseq =
new FileSeq(
PackageInfo.sProdDir.getPath() + nodeID.getWorkingParent(),
agenda.getPrimaryTarget());
if (fseq.hasFrameNumbers()) {
FilePattern fpat = fseq.getFilePattern();
FrameRange frange = fseq.getFrameRange();
out.write(
"opparm "
+ opname
+ " trange on\n"
+ "opparm "
+ opname
+ " f1 "
+ frange.getStart()
+ "\n"
+ "opparm "
+ opname
+ " f2 "
+ frange.getEnd()
+ "\n"
+ "opparm "
+ opname
+ " f3 "
+ frange.getBy()
+ "\n"
+ "opparm "
+ opname
+ " copoutput '"
+ fpat.getPrefix()
+ ".$F");
if (fpat.getPadding() > 1) out.write(String.valueOf(fpat.getPadding()));
out.write("." + fpat.getSuffix() + "'\n");
} else {
out.write("opparm " + opname + " trange off\n" + "opparm " + opname + " " + fseq + "\n");
}
if (preRender != null) out.write("opparm " + opname + " prerender '" + preRender + "'\n");
if (postRender != null) out.write("opparm " + opname + " postrender '" + postRender + "'\n");
if (preFrame != null) out.write("opparm " + opname + " preframe '" + preFrame + "'\n");
if (postFrame != null) out.write("opparm " + opname + " postframe '" + postFrame + "'\n");
out.write("opparm -c " + opname + " execute\n");
out.close();
} catch (IOException ex) {
throw new PipelineException(
"Unable to write temporary script file ("
+ hscript
+ ") for Job "
+ "("
+ agenda.getJobID()
+ ")!\n"
+ ex.getMessage());
}
/* create the wrapper shell script */
File script = createTemp(agenda, 0755, "bash");
try {
FileWriter out = new FileWriter(script);
out.write("#!/bin/bash\n\n" + "cat " + hscript + " | hscript -v " + source);
out.close();
} catch (IOException ex) {
throw new PipelineException(
"Unable to write temporary script file ("
+ script
+ ") for Job "
+ "("
+ agenda.getJobID()
+ ")!\n"
+ ex.getMessage());
}
try {
return new SubProcessHeavy(
agenda.getNodeID().getAuthor(),
getName() + "-" + agenda.getJobID(),
script.getPath(),
new ArrayList<String>(),
agenda.getEnvironment(),
agenda.getWorkingDir(),
outFile,
errFile);
} catch (Exception ex) {
throw new PipelineException(
"Unable to generate the SubProcess to perform this Action!\n" + ex.getMessage());
}
}
/**
* Construct a {@link SubProcessHeavy SubProcessHeavy} instance which when executed will fulfill
* the given action agenda.
*
* <p>
*
* @param agenda The agenda to be accomplished by the action.
* @param outFile The file to which all STDOUT output is redirected.
* @param errFile The file to which all STDERR output is redirected.
* @return The SubProcess which will fulfill the agenda.
* @throws PipelineException If unable to prepare a SubProcess due to illegal, missing or
* imcompatable information in the action agenda or a general failure of the prep method code.
*/
public SubProcessHeavy prep(ActionAgenda agenda, File outFile, File errFile)
throws PipelineException {
NodeID nodeID = agenda.getNodeID();
/* sanity checks */
File targetScene = null;
ArrayList<File> sourceScenes = new ArrayList<File>();
ArrayList<String> mergePatterns = new ArrayList<String>();
File preBuild = null;
File postBuild = null;
File preScene = null;
File postScene = null;
{
DoubleMap<Integer, String, TreeSet<FileSeq>> sources =
new DoubleMap<Integer, String, TreeSet<FileSeq>>();
for (String sname : agenda.getSourceNames()) {
if (hasSourceParams(sname)) {
FileSeq fseq = agenda.getPrimarySource(sname);
Integer order = (Integer) getSourceParamValue(sname, "Order");
addSourceSeq(order, sname, fseq, sources);
}
for (FileSeq fseq : agenda.getSecondarySources(sname)) {
FilePattern fpat = fseq.getFilePattern();
if (hasSecondarySourceParams(sname, fpat)) {
Integer order = (Integer) getSecondarySourceParamValue(sname, fpat, "Order");
addSourceSeq(order, sname, fseq, sources);
}
}
}
for (Integer order : sources.keySet()) {
for (String sname : sources.get(order).keySet()) {
for (FileSeq fseq : sources.get(order).get(sname)) {
String pattern = null;
if (fseq.equals(agenda.getPrimarySource(sname)))
pattern = (String) getSourceParamValue(sname, "MergePattern");
else {
FilePattern fpat = fseq.getFilePattern();
pattern = (String) getSecondarySourceParamValue(sname, fpat, "MergePattern");
}
if ((pattern == null) || (pattern.length() == 0))
throw new PipelineException(
"The Merge Pattern for file sequence ("
+ fseq
+ ") of source node "
+ "("
+ sname
+ ") was not specified!");
NodeID snodeID = new NodeID(nodeID, sname);
File sfile =
new File(PackageInfo.sProdDir, snodeID.getWorkingParent() + "/" + fseq.getFile(0));
sourceScenes.add(sfile);
mergePatterns.add(pattern);
}
}
}
/* generate the name of the Houdini scene to save */
{
FileSeq fseq = agenda.getPrimaryTarget();
String suffix = fseq.getFilePattern().getSuffix();
if (!fseq.isSingle() || (suffix == null) || !suffix.equals("hip"))
throw new PipelineException(
"The HfsBuild Action requires that the primary target file sequence must "
+ "be a single Houdini scene file!");
targetScene =
new File(PackageInfo.sProdDir, nodeID.getWorkingParent() + "/" + fseq.getFile(0));
}
/* command script files */
{
String sname = (String) getSingleParamValue("PreBuildScript");
if (sname != null) {
FileSeq fseq = agenda.getPrimarySource(sname);
if (fseq == null)
throw new PipelineException(
"Somehow the Pre Build Script node ("
+ sname
+ ") was not one of the "
+ "source nodes!");
String suffix = fseq.getFilePattern().getSuffix();
if (!fseq.isSingle() || (suffix == null) || !(suffix.equals("cmd")))
throw new PipelineException(
"The HfsBuild Action requires that the source node specified by the Pre "
+ "Build Script parameter ("
+ sname
+ ") must have a single command "
+ "script (.cmd) as its primary file sequence!");
NodeID snodeID = new NodeID(nodeID, sname);
preBuild =
new File(PackageInfo.sProdDir, snodeID.getWorkingParent() + "/" + fseq.getFile(0));
}
}
{
String sname = (String) getSingleParamValue("PostBuildScript");
if (sname != null) {
FileSeq fseq = agenda.getPrimarySource(sname);
if (fseq == null)
throw new PipelineException(
"Somehow the Post Build Script node ("
+ sname
+ ") was not one of the "
+ "source nodes!");
String suffix = fseq.getFilePattern().getSuffix();
if (!fseq.isSingle() || (suffix == null) || !(suffix.equals("cmd")))
throw new PipelineException(
"The HfsBuild Action requires that the source node specified by the Post "
+ "Build Script parameter ("
+ sname
+ ") must have a single command "
+ "script (.cmd) as its primary file sequence!");
NodeID snodeID = new NodeID(nodeID, sname);
postBuild =
new File(PackageInfo.sProdDir, snodeID.getWorkingParent() + "/" + fseq.getFile(0));
}
}
{
String sname = (String) getSingleParamValue("PreSceneScript");
if (sname != null) {
FileSeq fseq = agenda.getPrimarySource(sname);
if (fseq == null)
throw new PipelineException(
"Somehow the Pre Scene Script node ("
+ sname
+ ") was not one of the "
+ "source nodes!");
String suffix = fseq.getFilePattern().getSuffix();
if (!fseq.isSingle() || (suffix == null) || !(suffix.equals("cmd")))
throw new PipelineException(
"The HfsBuild Action requires that the source node specified by the Pre "
+ "Scene Script parameter ("
+ sname
+ ") must have a single command script "
+ "(.cmd) as its primary file sequence!");
NodeID snodeID = new NodeID(nodeID, sname);
preScene =
new File(PackageInfo.sProdDir, snodeID.getWorkingParent() + "/" + fseq.getFile(0));
}
}
{
String sname = (String) getSingleParamValue("PostSceneScript");
if (sname != null) {
FileSeq fseq = agenda.getPrimarySource(sname);
if (fseq == null)
throw new PipelineException(
"Somehow the Post Scene Script node ("
+ sname
+ ") was not one of the "
+ "source nodes!");
String suffix = fseq.getFilePattern().getSuffix();
if (!fseq.isSingle() || (suffix == null) || !(suffix.equals("cmd")))
throw new PipelineException(
"The HfsBuild Action requires that the source node specified by the Post "
+ "Scene Script parameter ("
+ sname
+ ") must have a single command script "
+ "(.cmd) as its primary file sequence!");
NodeID snodeID = new NodeID(nodeID, sname);
postScene =
new File(PackageInfo.sProdDir, snodeID.getWorkingParent() + "/" + fseq.getFile(0));
}
}
}
/* create the temporary Houdini command script */
File hscript = createTemp(agenda, 0644, "cmd");
try {
FileWriter out = new FileWriter(hscript);
if (preBuild != null) out.write("source " + preBuild + "\n");
/* load the source scenes */
int wk;
for (wk = 0; wk < sourceScenes.size(); wk++) {
File sourceScene = sourceScenes.get(wk);
String pattern = mergePatterns.get(wk);
if (preScene != null) out.write("source " + preScene + "\n");
out.write("mread");
if (wk > 0) out.write(" -o -m " + pattern);
out.write(" " + sourceScene + "\n");
if (postScene != null) out.write("source " + postScene + "\n");
}
if (postBuild != null) out.write("source " + postBuild + "\n");
/* save the file */
out.write("mwrite " + targetScene + "\n");
out.close();
} catch (IOException ex) {
throw new PipelineException(
"Unable to write temporary command file ("
+ hscript
+ ") for Job "
+ "("
+ agenda.getJobID()
+ ")!\n"
+ ex.getMessage());
}
/* create the wrapper shell script */
File script = createTemp(agenda, 0755, "bash");
try {
FileWriter out = new FileWriter(script);
out.write("#!/bin/bash\n\n" + "cat " + hscript + " | hscript");
out.close();
} catch (IOException ex) {
throw new PipelineException(
"Unable to write temporary script file ("
+ script
+ ") for Job "
+ "("
+ agenda.getJobID()
+ ")!\n"
+ ex.getMessage());
}
try {
return new SubProcessHeavy(
agenda.getNodeID().getAuthor(),
getName() + "-" + agenda.getJobID(),
script.getPath(),
new ArrayList<String>(),
agenda.getEnvironment(),
agenda.getWorkingDir(),
outFile,
errFile);
} catch (Exception ex) {
throw new PipelineException(
"Unable to generate the SubProcess to perform this Action!\n" + ex.getMessage());
}
}
