/**
* Updates the asset references for a shot within Maya and then in pipeline.
*
* @param shotName The name of the shot being processed.
* @param pRemoveRef The list of assets being dereferenced from the shot.
* @param pReplaceRef The list of assets being referenced into the shot.
* @param nameMap
*/
private void editShotReferences(
String shotName,
NodeMod targetMod,
TreeSet<String> pRemoveRef,
TreeSet<String> pReplaceRef,
TreeMap<String, String> nameMap)
throws PipelineException {
logLine("Editing shot: " + shotName);
boolean anim = !shotName.matches(lgtPattern);
/* writing the mel script */
if (anim) {
File script = null;
try {
script = File.createTempFile("UpdateAssetGUI.", ".mel", PackageInfo.sTempPath.toFile());
FileCleaner.add(script);
} // end try
catch (IOException ex) {
throw new PipelineException(
"Unable to create the temporary MEL script used to collect "
+ "texture information from the Maya scene!");
} // end catch
try {
PrintWriter out = new PrintWriter(new BufferedWriter(new FileWriter(script)));
for (String asset : pReplaceRef) {
String nameSpace;
if (asset.endsWith(lr))
nameSpace = nameMap.get(getShortName(asset.substring(0, asset.length() - 3)));
else {
System.err.println("This should not be happening, a hi res asset in a lo-res node.");
continue;
// nameSpace = nameMap.get(getShortName(asset));
}
out.println("print (\"referencing file: $WORKING" + asset + ".ma\");");
out.println(
"file -reference -namespace \"" + nameSpace + "\" \"$WORKING" + asset + ".ma\";");
} // end for
for (String asset : pRemoveRef) {
out.println("print (\"dereferencing file: $WORKING" + asset + ".ma\");");
out.println("file -rr \"$WORKING" + asset + ".ma\";");
} // end for
out.println("// SAVE");
out.println("file -save;");
out.close();
} // end try
catch (IOException ex) {
throw new PipelineException(
"Unable to write the temporary MEL script(" + script + ") used add the references!");
} // end catch
NodeID targetID = new NodeID(w.user, w.view, shotName);
// NodeStatus targetStat = mclient.status(targetID);
/* run Maya to collect the information */
try {
Path targetPath = getNodePath(shotName);
ArrayList<String> args = new ArrayList<String>();
args.add("-batch");
args.add("-script");
args.add(script.getPath());
args.add("-file");
args.add(targetPath.toOsString());
Path wdir = new Path(PackageInfo.sProdPath.toOsString() + targetID.getWorkingParent());
TreeMap<String, String> env =
mclient.getToolsetEnvironment(
w.user, w.view, targetMod.getToolset(), PackageInfo.sOsType);
Map<String, String> nenv = env;
String midefs = env.get("PIPELINE_MI_SHADER_PATH");
if (midefs != null) {
nenv = new TreeMap<String, String>(env);
Path dpath = new Path(new Path(wdir, midefs));
nenv.put("MI_CUSTOM_SHADER_PATH", dpath.toOsString());
}
String command = "maya";
if (PackageInfo.sOsType.equals(OsType.Windows)) command += ".exe";
SubProcessLight proc =
new SubProcessLight("UpdateAssetGUI", command, args, env, wdir.toFile());
try {
proc.start();
proc.join();
if (!proc.wasSuccessful()) {
throw new PipelineException(
"Did not correctly edit the reference due to a maya error.!\n\n"
+ proc.getStdOut()
+ "\n\n"
+ proc.getStdErr());
} // end if
} // end try
catch (InterruptedException ex) {
throw new PipelineException(ex);
} // end catch
} // end try
catch (Exception ex) {
throw new PipelineException(ex);
} // end catch
}
/*-edit the references in pipeline once they are done in the file-*/
BaseAction targetAction = targetMod.getAction();
for (String asset : pReplaceRef) {
mclient.link(
w.user, w.view, shotName, asset, LinkPolicy.Reference, LinkRelationship.All, null);
if (anim) {
/*Set the namespaces*/
String nameSpace = nameMap.get(getShortName(asset.substring(0, asset.length() - 3)));
System.err.println(nameSpace);
targetAction.initSourceParams(asset);
targetAction.setSourceParamValue(asset, "PrefixName", nameSpace);
targetMod.setAction(targetAction);
}
}
w.mclient.modifyProperties(w.user, w.view, targetMod);
for (String asset : pRemoveRef) mclient.unlink(w.user, w.view, shotName, asset);
if (!anim) {
System.err.println("Queuing the switchLgt node " + shotName);
mclient.submitJobs(w.user, w.view, shotName, null);
}
} // end editShotReferences
/** Add the UI components for the given file sequence to the panel. */
private void addFileSeqPanel(FileSeq fseq) {
boolean isPresentInWorking = false;
if ((pStatus != null) && pStatus.hasLightDetails()) {
NodeDetailsLight details = pStatus.getLightDetails();
NodeMod mod = details.getWorkingVersion();
if (mod != null) {
if (mod.getPrimarySequence().equals(fseq)) isPresentInWorking = true;
else {
for (FileSeq sfseq : mod.getSecondarySequences()) {
if (sfseq.equals(fseq)) {
isPresentInWorking = true;
break;
}
}
}
}
}
/* collate the row information */
ArrayList<VersionID> vids = new ArrayList<VersionID>();
ArrayList<FileSeq> singles = new ArrayList<FileSeq>();
TreeSet<FileSeq> enabled = new TreeSet<FileSeq>();
TreeMap<FileSeq, FileState> fstates = new TreeMap<FileSeq, FileState>();
TreeMap<FileSeq, NativeFileInfo> finfos = new TreeMap<FileSeq, NativeFileInfo>();
TreeMap<FileSeq, QueueState> qstates = new TreeMap<FileSeq, QueueState>();
TreeMap<FileSeq, Boolean[]> novel = new TreeMap<FileSeq, Boolean[]>();
{
TreeMap<FileSeq, Integer> wsingles = new TreeMap<FileSeq, Integer>();
if ((pStatus != null) && pStatus.hasLightDetails()) {
if (isPresentInWorking) {
if (pStatus.hasHeavyDetails()) {
NodeDetailsHeavy details = pStatus.getHeavyDetails();
FileState[] fs = details.getFileStates(fseq);
QueueState[] qs = details.getQueueStates();
NativeFileInfo[] infos = details.getFileInfos(fseq);
if ((fs != null) && (qs != null) && (infos != null)) {
int wk;
for (wk = 0; wk < fs.length; wk++) {
FileSeq sfseq = new FileSeq(fseq, wk);
wsingles.put(sfseq, wk);
fstates.put(sfseq, fs[wk]);
finfos.put(sfseq, infos[wk]);
qstates.put(sfseq, qs[wk]);
if (fs[wk] != FileState.CheckedIn) enabled.add(sfseq);
}
}
} else {
NodeDetailsLight details = pStatus.getLightDetails();
int wk;
for (wk = 0; wk < fseq.numFrames(); wk++) {
FileSeq sfseq = new FileSeq(fseq, wk);
wsingles.put(sfseq, wk);
if (details.getVersionState() == VersionState.CheckedIn) {
fstates.put(sfseq, FileState.CheckedIn);
qstates.put(sfseq, QueueState.Undefined);
} else {
enabled.add(sfseq);
}
}
}
}
{
vids.addAll(pNovelty.keySet());
Collections.reverse(vids);
int idx = 0;
for (VersionID vid : vids) {
TreeMap<FileSeq, boolean[]> table = pNovelty.get(vid);
for (FileSeq nfseq : table.keySet()) {
if (fseq.similarTo(nfseq)) {
boolean[] flags = table.get(nfseq);
int wk;
for (wk = 0; wk < flags.length; wk++) {
FileSeq sfseq = new FileSeq(nfseq, wk);
if (!wsingles.containsKey(sfseq)) wsingles.put(sfseq, null);
Boolean[] rflags = novel.get(sfseq);
if (rflags == null) {
rflags = new Boolean[pNovelty.size()];
novel.put(sfseq, rflags);
}
rflags[idx] = new Boolean(flags[wk]);
}
break;
}
}
idx++;
}
}
}
TreeMap<Integer, FileSeq> order = new TreeMap<Integer, FileSeq>();
for (FileSeq sfseq : wsingles.keySet()) {
int frame = -1;
if (sfseq.hasFrameNumbers()) frame = sfseq.getFrameRange().getStart();
order.put(frame, sfseq);
}
singles.addAll(order.values());
}
/* add the panel */
{
JFileSeqPanel panel =
new JFileSeqPanel(
this,
pManagerPanel,
pStatus,
pPrivilegeDetails,
fseq,
vids,
pOffline,
singles,
fstates,
finfos,
qstates,
enabled,
novel,
pIsListLayout);
if (pIsListLayout) pFileSeqsBox.add(panel);
else pFileSeqsTab.addTab(fseq.getFilePattern().toString(), sTabIcon, panel);
pFileSeqPanels.put(fseq, panel);
}
}
