expvintl
459 lines
15 KiB
C++
459 lines
15 KiB
C++
#include "file/token.h"
|
|
#include "file/stream.h"
|
|
|
|
#include "shaderMaterial/shaderMaterialGeoParamValueTexture.h"
|
|
#include "shaderMaterial/shaderMaterialGeoParamDescriptionTexture.h"
|
|
#include "rageShaderMaterialMayaParam.h"
|
|
#include "rageShaderMaterialMaya.h"
|
|
#include "rageShaderMaya65.h"
|
|
|
|
#define REQUIRE_IOSTREAM
|
|
#pragma warning(push)
|
|
#pragma warning(disable : 4668)
|
|
#include <maya/MPlug.h>
|
|
#include <maya/MPlugArray.h>
|
|
#include <maya/MFnAttribute.h>
|
|
#include <maya/MItDependencyNodes.h>
|
|
#include <maya/MSelectionList.h>
|
|
#include <maya/MDGModifier.h>
|
|
#pragma warning(pop)
|
|
|
|
using namespace rage;
|
|
|
|
extern bool g_bDisplayMGlobalExecuteCommands;
|
|
|
|
|
|
bool rageShaderMaterialMaya::AddToMayaNode(MObject obNode, const rageShaderMaterial& obMaterial, bool* DeletedExistingDataDuringCreation)
|
|
{
|
|
MFnDependencyNode obFnThis(obNode);
|
|
|
|
Assert(DeletedExistingDataDuringCreation);
|
|
|
|
(*DeletedExistingDataDuringCreation) = false;
|
|
|
|
if (!obFnThis.isLocked() && !obFnThis.isFromReferencedFile())
|
|
{
|
|
if (obFnThis.attributeCount())
|
|
{
|
|
bool attribsAreEqual = true;
|
|
// Check the shader attributes
|
|
for (int i=0; i<obMaterial.GetShaderParamCount(); i++)
|
|
{
|
|
if (!rageShaderMaterialMayaParam::IsAttributeSameValue(obNode, obMaterial.GetShaderParam(i)))
|
|
{
|
|
attribsAreEqual = false;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (attribsAreEqual)
|
|
{
|
|
//All the same so exit
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
if (MFnDependencyNode(obNode).isLocked())
|
|
{
|
|
//message that says we are locked so no connections changed
|
|
Displayf("message that says we are locked so no connections changed");
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
|
|
// Texture overide support
|
|
MPlug obTextureOveridePlug = obFnThis.findPlug("TextureOveride");
|
|
bool bUseExistingTextureNodes;
|
|
obTextureOveridePlug.getValue(bUseExistingTextureNodes);
|
|
|
|
// Delete all the old attrs
|
|
for(unsigned int i=0; i<obFnThis.attributeCount(); i++ )
|
|
{
|
|
bool bDeleteAttribute = false;
|
|
MFnAttribute obFnAttribute(obFnThis.attribute(i));
|
|
if(obFnAttribute.isDynamic())
|
|
{
|
|
// It is a dynamic attribute, so maybe delete it
|
|
// Is it a child attribute?
|
|
bool bChildAttribute = (obFnAttribute.parent() != MObject::kNullObj);
|
|
if(!bChildAttribute)
|
|
{
|
|
// It is a dynamic attribute and not a child attribute, so maybe delete it
|
|
if(bUseExistingTextureNodes && obFnAttribute.isUsedAsColor())
|
|
{
|
|
// It is a dynamic attribute, not a child attribute, it is a texture attribute and might have to retain it
|
|
// Does a param with the same name exist in the given material
|
|
bDeleteAttribute = true;
|
|
MString strExistingParamName = obFnAttribute.name();
|
|
for (int j=0; j<obMaterial.GetParamCount(); j++)
|
|
{
|
|
// Set the values
|
|
const shaderMaterialGeoParamValue * param = obMaterial.GetParam(j);
|
|
// Displayf("(\"%s\" == \"%s\") = %d", strExistingParamName.asChar(), param->GetName(), (strExistingParamName == MString(param->GetName())));
|
|
if(strExistingParamName == MString(param->GetParamDescription()->GetUIName()))
|
|
{
|
|
// It still exists, so don't kill it
|
|
bDeleteAttribute = false;
|
|
break;
|
|
}
|
|
}
|
|
if(bDeleteAttribute)
|
|
{
|
|
// Displayf("Going to delete %s", obFnAttribute.name().asChar());
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// It is a dynamic attribute, not a child attribute and I do not have to retain the existing textures, so delete it!
|
|
bDeleteAttribute = true;
|
|
}
|
|
}
|
|
}
|
|
if(bDeleteAttribute)
|
|
{
|
|
//(*DeletedExistingDataDuringCreation) = true;
|
|
// Safe to delete
|
|
//Displayf("Deleting %s", obFnAttribute.name().asChar());
|
|
MStatus obStatus = obFnThis.removeAttribute(obFnThis.attribute(i), MFnDependencyNode::kLocalDynamicAttr);
|
|
if(obStatus != MS::kSuccess)
|
|
{
|
|
Displayf("An error occured removing attribute %s from node %s : %s %s %d", obFnAttribute.name().asChar(), obFnThis.name().asChar(), obStatus.errorString().asChar(), __FILE__, __LINE__);
|
|
return false;
|
|
}
|
|
i--;
|
|
}
|
|
}
|
|
}
|
|
|
|
// Add the new shader attributes
|
|
for (int i=0; i<obMaterial.GetShaderParamCount(); i++)
|
|
{
|
|
MPlug obExistingPlug = obFnThis.findPlug(obMaterial.GetShaderParam(i)->GetParamDescription()->GetUIName());
|
|
if(obExistingPlug.isNull())
|
|
{
|
|
//Displayf("Add Attribute [%s]", obMaterial.GetShaderParam(i)->GetParamDescription()->GetUIName());
|
|
// Don't exist, so add me
|
|
//if (!rageShaderMaya::getIgnoreReferencedNodesDuringWireUpTextures())
|
|
//{
|
|
rageShaderMaterialMayaParam::AddAttributesToMayaNode(obNode, obMaterial.GetShaderParam(i));
|
|
//}
|
|
}
|
|
else
|
|
{
|
|
//Displayf("Not going to add %s because already exists", obExistingPlug.name().asChar());
|
|
}
|
|
}
|
|
|
|
// Restore the locked-ness
|
|
return true;
|
|
}
|
|
|
|
bool rageShaderMaterialMaya::RemoveFromMayaNode(MObject obNode, const rageShaderMaterial& /*obMaterial*/)
|
|
{
|
|
MFnDependencyNode obFnThis(obNode);
|
|
if (obFnThis.isLocked())
|
|
{
|
|
//nothing we can do
|
|
return true;
|
|
}
|
|
|
|
MPlug obTextureOveridePlug = obFnThis.findPlug("TextureOveride");
|
|
bool bUseExistingTextureNodes;
|
|
obTextureOveridePlug.getValue(bUseExistingTextureNodes);
|
|
|
|
// Delete all the attrs of the material
|
|
for(unsigned int i=0; i<obFnThis.attributeCount(); i++ )
|
|
{
|
|
bool bDeleteAttribute = false;
|
|
MFnAttribute obFnAttribute(obFnThis.attribute(i));
|
|
if(obFnAttribute.isDynamic())
|
|
{
|
|
// It is a dynamic attribute, so maybe delete it
|
|
// Is it a child attribute?
|
|
bool bChildAttribute = (obFnAttribute.parent() != MObject::kNullObj);
|
|
if(!bChildAttribute)
|
|
{
|
|
if(bUseExistingTextureNodes && obFnAttribute.isUsedAsColor())
|
|
{
|
|
bDeleteAttribute = false;
|
|
}
|
|
else
|
|
{
|
|
bDeleteAttribute = true;
|
|
}
|
|
}
|
|
}
|
|
|
|
if(bDeleteAttribute)
|
|
{
|
|
// Safe to delete
|
|
//Displayf("Deleting %s", obFnAttribute.name().asChar());
|
|
MStatus obStatus = obFnThis.removeAttribute(obFnThis.attribute(i), MFnDependencyNode::kLocalDynamicAttr);
|
|
if(obStatus != MS::kSuccess)
|
|
{
|
|
Errorf("An error occured removing attribute %s from node %s : %s %s %d", obFnAttribute.name().asChar(), obFnThis.name().asChar(), obStatus.errorString().asChar(), __FILE__, __LINE__);
|
|
return false;
|
|
}
|
|
i--;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool rageShaderMaterialMaya::WireUpTextureNode(MObject obNode, const shaderMaterialGeoParamValue& obParam)
|
|
{
|
|
// Initialize the dependency object
|
|
MFnDependencyNode obFnShaderNode(obNode);
|
|
//Displayf("Node %s", obFnShaderNode.name().asChar());
|
|
if (obFnShaderNode.isLocked())
|
|
{
|
|
//Displayf("Node %s is locked no texture connection attempted\n", obFnShaderNode.name().asChar());
|
|
//early out cause we cant change anything that is locked.
|
|
return true;
|
|
}
|
|
|
|
if (obFnShaderNode.isFromReferencedFile())
|
|
{
|
|
//Displayf("Node %s is from referenced file, no texture connection attempted\n", obFnShaderNode.name().asChar());
|
|
//early out cause we cant change anything that is from a referenced file.
|
|
return true;
|
|
}
|
|
|
|
// Got a valid texture param
|
|
shaderMaterialGeoParamValueTexture* pobTextureParam = (shaderMaterialGeoParamValueTexture*) &obParam;
|
|
const shaderMaterialGeoParamDescriptionTexture* textureDesc = (const shaderMaterialGeoParamDescriptionTexture*)pobTextureParam->GetParamDescription();
|
|
|
|
Assert(textureDesc);
|
|
|
|
// Get texture name
|
|
const char * pcTextureFilename = pobTextureParam->GetValue();
|
|
|
|
if (!pcTextureFilename)
|
|
{
|
|
Errorf("No texture name from RSM: %s PARAM: %s\n", obFnShaderNode.name().asChar(), obParam.GetName());
|
|
return false;
|
|
}
|
|
|
|
if (stricmp(pcTextureFilename, "none"))
|
|
{
|
|
//Displayf("Using Texture: %s from RSM: %s PARAM: %s", pcTextureFilename, obFnShaderNode.name().asChar(), obParam.GetName());
|
|
|
|
MItDependencyNodes depIterator( MFn::kFileTexture);
|
|
// Do I already have a file node matching that texture file name?
|
|
MObject obFileNode = MObject::kNullObj;
|
|
MString strFileNodeName = "";
|
|
MStatus obStatus;
|
|
for (; !depIterator.isDone(); depIterator.next() )
|
|
{
|
|
MFnDependencyNode node( depIterator.item() );
|
|
if (rageShaderMaya::getIgnoreReferencedNodesDuringWireUpTextures() && node.isFromReferencedFile())
|
|
{
|
|
continue;
|
|
}
|
|
else
|
|
{
|
|
MPlug plug = node.findPlug("fileTextureName");
|
|
if (!plug.isNull())
|
|
{
|
|
MString textureFileName;
|
|
plug.getValue(textureFileName);
|
|
|
|
//Displayf("Comparing %s and %s (%s)", pcTextureFilename, textureFileName.asChar());
|
|
|
|
if (stricmp(fiAssetManager::FileName(textureFileName.asChar()), fiAssetManager::FileName(pcTextureFilename))==0)
|
|
{
|
|
//check the full path to make sure it is the same.
|
|
if (stricmp(textureFileName.asChar(), pcTextureFilename)==0)
|
|
{
|
|
strFileNodeName = node.name();
|
|
obFileNode = depIterator.item();
|
|
//Displayf("Found file node: %s", strFileNodeName.asChar());
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if(strFileNodeName == "")
|
|
{
|
|
// Unable to find an existing filename that works, so make one
|
|
// Can't find a good node, so create the maya texture file node
|
|
MGlobal::executeCommand("shadingNode -asTexture file", strFileNodeName, g_bDisplayMGlobalExecuteCommands);
|
|
|
|
// Convert node name into actual API node
|
|
MSelectionList list;
|
|
list.add(strFileNodeName);
|
|
list.getDependNode(0, obFileNode);
|
|
|
|
// Set the filename of the new node
|
|
MFnDependencyNode obFnFileNode(obFileNode);
|
|
MPlug obFileTextureNamePlug = obFnFileNode.findPlug("fileTextureName");
|
|
obFileTextureNamePlug.setValue(pcTextureFilename);
|
|
|
|
//Displayf("Create File node with texture %s", pcTextureFilename);
|
|
}
|
|
|
|
// Get plugs for connections
|
|
MFnDependencyNode obFnFileNode( obFileNode );
|
|
MPlug obOutColorPlug = obFnFileNode.findPlug("outColor", &obStatus);
|
|
if(obOutColorPlug.isNull())
|
|
{
|
|
// Oh dear
|
|
MGlobal::displayError("Unable to find outColor attribute on \""+ obFnFileNode.name() +"\"");
|
|
return false;
|
|
}
|
|
CHECK_STATUS( obStatus );
|
|
|
|
MString strTextureParamUIName = pobTextureParam->GetParamDescription()->GetUIName();
|
|
MString strSafeTextureParamUIName = rageShaderMaterialMayaParam::GetCleanVersionOfString(strTextureParamUIName.asChar());
|
|
MPlug obInColorPlug = obFnShaderNode.findPlug(strSafeTextureParamUIName, &obStatus);
|
|
if(obInColorPlug.isNull())
|
|
{
|
|
// Oh dear
|
|
MGlobal::displayError("Unable to find \""+ strSafeTextureParamUIName +"\" attribute on \""+ obFnShaderNode.name() +"\"");
|
|
return false;
|
|
}
|
|
CHECK_STATUS( obStatus );
|
|
|
|
|
|
// Make sure I'm not already connected
|
|
MPlugArray aobPlugsConnectedToOutColor;
|
|
obOutColorPlug.connectedTo(aobPlugsConnectedToOutColor, true, true, &obStatus);
|
|
CHECK_STATUS( obStatus );
|
|
|
|
MPlugArray aobPlugsConnectedToInColor;
|
|
obInColorPlug.connectedTo(aobPlugsConnectedToInColor, true, true, &obStatus);
|
|
CHECK_STATUS( obStatus );
|
|
|
|
//Displayf("aobPlugsConnectedToInColor.length() = %d", aobPlugsConnectedToInColor.length());
|
|
//for(unsigned i=0; i<aobPlugsConnectedToInColor.length(); i++)
|
|
//{
|
|
// Displayf("aobPlugsConnectedToInColor[%d] = %s\n", i, aobPlugsConnectedToInColor[i].name().asChar());
|
|
//}
|
|
|
|
//Displayf("aobPlugsConnectedToOutColor.length() = %d", aobPlugsConnectedToOutColor.length());
|
|
//for(unsigned i=0; i<aobPlugsConnectedToOutColor.length(); i++)
|
|
//{
|
|
// Displayf("aobPlugsConnectedToOutColor[%d] = %s\n", i, aobPlugsConnectedToOutColor[i].name().asChar());
|
|
//}
|
|
|
|
MString plugName = obFnShaderNode.name();
|
|
plugName += ".";
|
|
plugName += rageShaderMaterialMayaParam::GetCleanVersionOfString(textureDesc->GetUIName());
|
|
bool bAlreadyConnected = false;
|
|
for(unsigned i=0; i<aobPlugsConnectedToOutColor.length(); i++)
|
|
{
|
|
if (!stricmp(plugName.asChar(), aobPlugsConnectedToOutColor[i].name().asChar()))
|
|
{
|
|
MPlugArray aobPlugsConnectedToOutColorPlug;
|
|
aobPlugsConnectedToOutColor[i].connectedTo(aobPlugsConnectedToOutColorPlug, true, true, &obStatus);
|
|
|
|
//Displayf("aobPlugsConnectedToOutColorPlug.length() = %d", aobPlugsConnectedToOutColorPlug.length());
|
|
//for(unsigned j=0; j<aobPlugsConnectedToOutColorPlug.length(); j++)
|
|
//{
|
|
// Displayf("aobPlugsConnectedToOutColorPlug[%d] = %s\n", j, aobPlugsConnectedToOutColorPlug[j].name().asChar());
|
|
//}
|
|
|
|
for(unsigned j=0; j<aobPlugsConnectedToOutColorPlug.length(); j++)
|
|
{
|
|
for(unsigned k=0; k<aobPlugsConnectedToInColor.length(); k++)
|
|
{
|
|
//Displayf("Comparing %s and %s\n", aobPlugsConnectedToInColor[k].name().asChar(), aobPlugsConnectedToOutColorPlug[j].name().asChar());
|
|
if(aobPlugsConnectedToInColor[k] == aobPlugsConnectedToOutColorPlug[j])
|
|
{
|
|
// Already connected so forget about it
|
|
//Displayf("\nAlready connected\n");
|
|
bAlreadyConnected = true;
|
|
break;
|
|
}
|
|
}
|
|
if (bAlreadyConnected)
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
if (bAlreadyConnected)
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
|
|
if(!bAlreadyConnected)
|
|
{
|
|
MDGModifier obModifier;
|
|
|
|
if (obInColorPlug.isLocked())
|
|
{
|
|
//Displayf("Atrribute %s is locked no texture connection attempted\n", obInColorPlug.name().asChar());
|
|
//early out cause we cant change anything that is locked.
|
|
return true;
|
|
}
|
|
|
|
if (obOutColorPlug.isLocked())
|
|
{
|
|
//Displayf("Atrribute %s is locked no texture connection attempted\n", obOutColorPlug.name().asChar());
|
|
//early out cause we cant change anything that is locked.
|
|
return true;
|
|
}
|
|
|
|
MFnAttribute obA1( obInColorPlug.attribute() );
|
|
if (!obA1.isWritable())
|
|
{
|
|
//Displayf("Atrribute %s from plug %s is not writable no texture connection attempted\n", obA1.name().asChar(), obInColorPlug.name().asChar());
|
|
//early out cause we cant change anything that is locked.
|
|
return true;
|
|
}
|
|
|
|
|
|
//Having to force this connection seems wrong, but we are running into an error case
|
|
// which is that the file nodes' outColor attribute (obOutColorPlug) is not writable
|
|
// which is causing the connection to error resulting in the connection not happening.
|
|
//By forcing the connection we get the desired effect of shared file nodes.
|
|
//The command bellow forces the connection between the file node and the RSM without an error.
|
|
//The code above this is not changed because we still want all checks to happen to protect
|
|
// referenced in RSM nodes and other "locked" cases.
|
|
//Executing this script line is the same as doing the drag drop step to link a file node
|
|
// to an RSM.
|
|
|
|
//connectAttr -force DiffuseTexture2.outColor p_gen_basketIndian01_.DiffuseTexture;
|
|
MString command = "connectAttr -force ";
|
|
command += obOutColorPlug.name();
|
|
command += " ";
|
|
command += obInColorPlug.name();
|
|
command += ";";
|
|
|
|
// Connect the attributes
|
|
//Displayf("%d Connecting %s -> %s", (int)__LINE__, obOutColorPlug.name().asChar(), obInColorPlug.name().asChar());
|
|
CHECK_STATUS(obModifier.commandToExecute(command));
|
|
MStatus obStatus = obModifier.doIt();
|
|
CHECK_STATUS(obStatus);
|
|
if(obStatus != MS::kSuccess)
|
|
{
|
|
// An error occured
|
|
Displayf("An error occurred connecting %s to %s\n", obOutColorPlug.name().asChar(), obInColorPlug.name().asChar());
|
|
}
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool rageShaderMaterialMaya::ExpandTextureNodes(MObject obNode, const rageShaderMaterial& obMaterial)
|
|
{
|
|
// Wire up all the nodes
|
|
for (int iParam=0; iParam<obMaterial.GetParamCount(); iParam++)
|
|
{
|
|
// Lets find all the texture pobParameters
|
|
const shaderMaterialGeoParamValue * pobParam = obMaterial.GetParam(iParam);
|
|
if (pobParam && pobParam->GetType() == grcEffect::VT_TEXTURE)
|
|
{
|
|
WireUpTextureNode(obNode, *pobParam);
|
|
}
|
|
}
|
|
return true;
|
|
}
|