/**
* 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
@Override
public void run(String[] args) {
try {
TreeMap<String, LinkedList<String>> parsedArgs = argParser(args);
if (!checkArgsAndSetParams(parsedArgs)) {
return;
}
} catch (PipelineException ex) {
System.err.println("There was a problem reading the arguments.\n" + ex.getMessage());
printHelp();
return;
}
for (String turntableName : turntables) {
logLine("Doing turntable: " + turntableName);
for (String passName : passes) {
logLine("\tDoing pass: "******"\t\tDoing asset: " + as.assetName);
LairTurntable tt = new LairTurntable(as, turntableName, passName);
w = new Wrapper(user, view, toolset, client);
logLine("\t\tFreezing all the textures");
getLatest(w, as.texGroup, over, froz);
logLine("\t\tChecking out the final model scene");
getNewest(w, as.finalScene, keep, pFroz);
getNewest(w, as.lr_finalScene, keep, pFroz);
logLine("\t\tChecking out the shader scene.");
getNewest(w, as.shdScene, keep, pFroz);
logLine("\t\tChecking out the turntable scene.");
getNewest(w, tt.turntableScene, keep, pFroz);
logLine("\t\tChecking out the overRide scene.");
getNewest(w, tt.ttCamOverMI, keep, pFroz);
logLine("\t\tChecking out the options scene.");
getNewest(w, tt.ttOptionsMI, keep, pFroz);
logLine("\t\tChecking out the camera MI node.");
getNewest(w, tt.ttCamMI, keep, pFroz);
logLine("\t\tChecking out the light MI node.");
getNewest(w, tt.ttLightMI, keep, pFroz);
logLine("\t\tChecking out the shade MI node.");
getNewest(w, tt.ttShadeMI, keep, pFroz);
logLine("\t\tChecking out the geo MI node.");
getNewest(w, tt.ttGeoMI, keep, pFroz);
logLine("\t\tChecking out the images node.");
getNewest(w, tt.ttImages, keep, pFroz);
logLine("\t\tFixing the turntable node");
{
NodeMod mod = client.getWorkingVersion(user, view, tt.turntableScene);
if (toolset != null) mod.setToolset(toolset);
client.modifyProperties(user, view, mod);
}
logLine("\t\tFixing the camera override node");
{
NodeMod mod = client.getWorkingVersion(user, view, tt.ttCamOverMI);
if (toolset != null) mod.setToolset(toolset);
BaseAction act = mod.getAction();
act.setSingleParamValue("ImageWidth", 1280);
act.setSingleParamValue("ImageHeight", 720);
act.setSingleParamValue("AspectRatio", 1.777);
act.setSingleParamValue("Aperture", 1.41732);
act.setSingleParamValue("OverrideFocal", false);
act.setSingleParamValue("OverrideClipping", false);
mod.setAction(act);
client.modifyProperties(user, view, mod);
}
logLine("\t\tFixing the camera node");
{
NodeMod mod = client.getWorkingVersion(user, view, tt.ttCamMI);
if (toolset != null) mod.setToolset(toolset);
client.modifyProperties(user, view, mod);
}
logLine("\t\tFixing the light node");
{
NodeMod mod = client.getWorkingVersion(user, view, tt.ttLightMI);
if (toolset != null) mod.setToolset(toolset);
client.modifyProperties(user, view, mod);
}
logLine("\t\tFixing the geo node");
{
NodeMod mod = client.getWorkingVersion(user, view, tt.ttGeoMI);
if (toolset != null) mod.setToolset(toolset);
BaseAction act = mod.getAction();
act.setSingleParamValue("CustomText", true);
mod.setAction(act);
client.modifyProperties(user, view, mod);
}
logLine("\t\tFixing the shader node");
{
NodeMod mod = client.getWorkingVersion(user, view, tt.ttShadeMI);
if (toolset != null) mod.setToolset(toolset);
BaseAction act = mod.getAction();
BaseAction act2 = LairConstants.actionMayaMiShader();
act2.setSingleParamValues(act);
mod.setAction(act2);
JobReqs req = mod.getJobRequirements();
req.addSelectionKey("MentalRay");
mod.setJobRequirements(req);
mod.setExecutionMethod(ExecutionMethod.Parallel);
mod.setBatchSize(100);
client.modifyProperties(user, view, mod);
}
logLine("\t\tFixing the images node");
{
NodeMod mod = client.getWorkingVersion(user, view, tt.ttImages);
if (toolset != null) mod.setToolset(toolset);
BaseAction act = mod.getAction();
act.setSingleParamValue("TexturePath", "$WORKING");
mod.setAction(act);
JobReqs req = mod.getJobRequirements();
req.addSelectionKey("MentalRay");
mod.setJobRequirements(req);
mod.setExecutionMethod(ExecutionMethod.Parallel);
mod.setBatchSize(5);
client.modifyProperties(user, view, mod);
}
logLine(
"All Done. Remember to set your export set on the mod node "
+ "if you want to export a custom part of the body");
} catch (PipelineException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
}
}
}
/**
* Perform execution of the tool.
*
* <p>
*
* @param mclient The network connection to the plmaster(1) daemon.
* @param qclient The network connection to the plqueuemgr(1) daemon.
* @return Whether to continue and collect user input for the next phase of the tool.
* @throws PipelineException If unable to sucessfully execute this phase of the tool.
*/
public synchronized boolean executePhase(MasterMgrClient mclient, QueueMgrClient qclient)
throws PipelineException {
boolean newCheckOut = false;
String matName = null;
String texName = null;
{
/*find out if it is a fresh checkout*/
JToolDialog tool = new JToolDialog("MaterialsCheckOut", new JPanel(), "Continue");
JConfirmDialog dialog = new JConfirmDialog(tool, "Do you want a completely fresh checkout");
dialog.setVisible(true);
newCheckOut = dialog.wasConfirmed();
}
{
/*-check out the asset and mat nodes-*/
jcheckOut(mclient, pUser, pView, pPrimary, null, keep, pFroz);
NodeMod assetMod = mclient.getWorkingVersion(pUser, pView, pPrimary);
{
JobReqs jreqs = assetMod.getJobRequirements();
jreqs.addSelectionKey("LinuxOnly");
assetMod.setJobRequirements(jreqs);
mclient.modifyProperties(pUser, pView, assetMod);
}
Set<String> assetSrcs = assetMod.getSourceNames();
err.println("The asset sources are: ");
for (String src : assetSrcs) {
err.println(src);
if (src.matches(matPattern)) {
matName = src;
err.println("Found mat node:\n\t " + src);
if (newCheckOut) {
err.println("Clean mat checkout");
jcheckOut(mclient, pUser, pView, src, null, over, frozU);
} else {
OverallNodeState state =
mclient.status(pUser, pView, src).getHeavyDetails().getOverallNodeState();
if (!state.equals(OverallNodeState.Modified)) {
err.println("Mat node has not been modified");
jcheckOut(mclient, pUser, pView, src, null, keep, pFroz);
}
}
} else if (src.matches(matPattern + "Exp")) {
err.println("Matexp is:\n\t" + src);
jcheckOut(mclient, pUser, pView, src, null, keep, modi);
{
NodeMod expMod = mclient.getWorkingVersion(pUser, pView, src);
JobReqs jreqs = expMod.getJobRequirements();
jreqs.addSelectionKey("LinuxOnly");
expMod.setJobRequirements(jreqs);
mclient.modifyProperties(pUser, pView, expMod);
}
} else {
jcheckOut(mclient, pUser, pView, src, null, over, froz);
}
} // end for
err.println("mat node: " + matName);
if (matName == null)
throw new PipelineException("This asset node does not have an " + "attached mat node");
}
{
/*find the texture node and check out so it can be changed. If new checkout, fresh textures*/
NodeMod matMod = mclient.getWorkingVersion(pUser, pView, matName);
TreeSet<String> matSrcs = new TreeSet<String>(matMod.getSourceNames());
for (String src : matSrcs) {
if (src.matches(texPattern)) {
texName = src;
err.println("Found tex node:\n\t " + src);
if (newCheckOut) {
err.println("Clean");
jcheckOut(mclient, pUser, pView, src, null, over, frozU);
} else {
err.println("Old stuff");
jcheckOut(mclient, pUser, pView, src, null, keep, pFroz);
jcheckOut(mclient, pUser, pView, src, null, keep, modi);
}
continue;
} // end if
jcheckOut(mclient, pUser, pView, src, null, over, froz);
} // end for
err.println("tex node: " + texName);
if (texName == null)
throw new PipelineException(
"This asset node does not have an associated " + "texture node");
}
err.println("Checked out the asset, mat and texture nodes");
{
/*check out finalise scripts*/
jcheckOut(
mclient,
pUser,
pView,
"/projects/lr/assets/tools/mel/finalize-character",
null,
over,
froz);
jcheckOut(
mclient, pUser, pView, "/projects/lr/assets/tools/mel/finalize-set", null, over, froz);
jcheckOut(
mclient, pUser, pView, "/projects/lr/assets/tools/mel/finalize-prop", null, over, froz);
}
err.close();
return false;
} // end executePhase
private boolean dylanate(
MasterMgrClient mclient,
String switchName,
DoubleMap<String, String, TreeSet<VersionID>> plugs)
throws PipelineException {
err.println("\nProcessing: " + switchName);
String animName = null;
{
NodeMod switchMod = null;
try {
switchMod = mclient.getWorkingVersion(pUser, pView, switchName);
} catch (PipelineException e) {
mclient.checkOut(pUser, pView, switchName, null, keep, pFroz);
switchMod = mclient.getWorkingVersion(pUser, pView, switchName);
}
TreeSet<String> switchSrcs = new TreeSet<String>(switchMod.getSourceNames());
for (String src : switchSrcs) {
if (src.matches(animPattern)) {
animName = src;
err.println("Found anim node: " + src);
mclient.checkOut(pUser, pView, src, null, over, frozU);
continue;
} // end if
mclient.checkOut(pUser, pView, src, null, over, frozU);
LinkMod lMod = switchMod.getSource(src);
LinkPolicy rel = lMod.getPolicy();
System.err.println(src + ": " + rel);
if (!rel.equals(REF)) {
System.err.println("umm");
lMod.setPolicy(REF);
switchMod.setSource(lMod);
mclient.modifyProperties(pUser, pView, switchMod);
}
} // end for
err.println("anim node: " + animName);
mclient.modifyProperties(pUser, pView, switchMod);
if (animName == null)
throw new PipelineException("This switch node does not have an associated anim node");
}
err.println("Checked out the anim and switch nodes");
String actionName = null;
VersionID ttVer = null;
boolean toCache = false;
/*change the action setting on the Switch node*/
{
NodeMod switchMod = mclient.getWorkingVersion(pUser, pView, switchName);
Path p = new Path(switchName);
Path syfRoot = new Path(p.getParentPath().getParentPath(), "syf");
System.err.println("Going to look for caches in: " + syfRoot);
ArrayList<String> syfDirs = getChildrenDirs(mclient, syfRoot.toString());
for (String dir : syfDirs) {
System.err.println(dir);
Path dPath = new Path(syfRoot, dir);
ArrayList<String> simDir = getChildrenNodes(mclient, dPath.toString());
for (String pCache : simDir) {
Path cPath = new Path(dPath, pCache);
if (cPath.toString().matches(cltPattern)) {
System.err.println("\t" + cPath.toString());
try {
mclient.checkOut(pUser, pView, cPath.toString(), null, over, froz);
mclient.lock(pUser, pView, cPath.toString(), null);
mclient.link(pUser, pView, switchMod.getName(), cPath.toString(), DEP, LINKALL, null);
} catch (PipelineException e) {
e.printStackTrace();
}
toCache = true;
}
}
}
{
BaseAction action = switchMod.getAction();
if (!switchMod.getToolset().equals(pToolset)) {
switchMod.setToolset(pToolset);
mclient.modifyProperties(pUser, pView, switchMod);
}
if (!toCache) {
actionName = modRep;
} else {
actionName = modRep + "Syflex";
}
ttVer = plugs.get("SCEA", actionName).last();
if ((action == null)
|| (!action.getName().equals(actionName))
|| (!action.getVersionID().equals(ttVer))) {
System.err.println(
"Action name is incorrect - the switch node "
+ switchName
+ " doesn't have a "
+ actionName
+ "Action");
action = plug.newAction(actionName, ttVer, "SCEA");
action.setSingleParamValue("Source", animName);
action.setSingleParamValue("Response", "Ignore");
if (toCache) action.setSingleParamValue(aApplyCache, true);
switchMod.setAction(action);
mclient.modifyProperties(pUser, pView, switchMod);
} else {
if (!action.getSingleParamValue("Response").equals("Ignore")) {
// action = plug.newAction(actionName, ttVer, "SCEA");
// action.setSingleParamValue("Source", animName);
action.setSingleParamValue("Response", "Ignore");
// if(toCache)
// action.setSingleParamValue(aApplyCache, true);
switchMod.setAction(action);
mclient.modifyProperties(pUser, pView, switchMod);
} // end if
} // end else
}
}
{
/*-check out and lock the animation sources-*/
NodeMod animMod = mclient.getWorkingVersion(pUser, pView, animName);
Set<String> animSrcs = animMod.getSourceNames();
for (String src : animSrcs) {
if (src.matches(loresPattern)) {
pLoresSrcs.add(src);
err.println("Adding lores src " + src);
}
mclient.checkOut(pUser, pView, src, null, keep, pFroz);
} // end for
err.println("lores:" + pLoresSrcs);
}
/*-sync the animation assets with the switch assets.*/
// also remove unnecessary hires models
{
NodeMod switchMod = mclient.getWorkingVersion(pUser, pView, switchName);
for (String src : switchMod.getSourceNames()) {
if (src.matches(hiresPattern)) {
err.println("Found hires source " + src);
if (pLoresSrcs.contains(src + "_lr")) {
pHiresSrcs.add(src);
err.println("The hires source matched the anim node.");
} else {
err.println("Gotta remove " + src);
} // end else
} // end if
} // end for
}
// add necessary hires models
for (String lores : pLoresSrcs) {
String hr = lores.replace("_lr", "");
err.println("Looking at lores source " + lores + " which matches " + hr);
if (!pHiresSrcs.contains(hr)) {
err.println("Gotta add " + hr + " to " + switchName);
pHiresSrcs.add(hr);
}
}
{
NodeMod switchMod = mclient.getWorkingVersion(pUser, pView, switchName);
TreeSet<String> switchSrcs = new TreeSet<String>(switchMod.getSourceNames());
err.println("Final hiRes list:" + pHiresSrcs + "\n");
err.println("switch now has:\n\t" + switchSrcs + "\n");
err.println("Looking for things to add.");
for (String src : pHiresSrcs) {
if ((src.matches(hiresPattern) && (!switchSrcs.contains(src)))) {
err.print("Linking ");
mclient.checkOut(pUser, pView, src, null, over, frozU);
mclient.link(pUser, pView, pPrimary, src, REF, LINKALL, null);
switchSrcs.add(src);
}
err.println("src from hiRes list: " + src);
}
}
{
NodeMod switchMod = mclient.getWorkingVersion(pUser, pView, switchName);
TreeSet<String> switchSrcs = new TreeSet<String>(switchMod.getSourceNames());
err.println("switch now has:\n\t" + switchSrcs + "\n");
for (String src : switchSrcs) {
if ((src.matches(hiresPattern) && (!pHiresSrcs.contains(src)))) {
err.print("Unlinking ");
mclient.unlink(pUser, pView, pPrimary, src);
}
err.println("src from switch node list: " + src);
}
}
/*queue the switch node*/
mclient.submitJobs(pUser, pView, switchName, null);
return false;
}