Files
bullet3/Extras/FCollada/FCDocument/FCDEffect.cpp
2006-05-25 19:18:29 +00:00

269 lines
8.0 KiB
C++

/*
Copyright (C) 2005-2006 Feeling Software Inc.
MIT License: http://www.opensource.org/licenses/mit-license.php
*/
/*
Based on the FS Import classes:
Copyright (C) 2005-2006 Feeling Software Inc
Copyright (C) 2005-2006 Autodesk Media Entertainment
MIT License: http://www.opensource.org/licenses/mit-license.php
*/
#include "StdAfx.h"
#include "FCDocument/FCDocument.h"
#include "FCDocument/FCDAnimated.h"
#include "FCDocument/FCDEffect.h"
#include "FCDocument/FCDEffectStandard.h"
#include "FCDocument/FCDEffectProfileFX.h"
#include "FCDocument/FCDEffectParameter.h"
#include "FCDocument/FCDEffectParameterFactory.h"
#include "FCDocument/FCDEffectParameterList.h"
#include "FCDocument/FCDLibrary.h"
#include "FCDocument/FCDImage.h"
#include "FUtils/FUStringConversion.h"
#include "FUtils/FUDaeParser.h"
#include "FUtils/FUDaeWriter.h"
using namespace FUDaeParser;
using namespace FUDaeWriter;
FCDEffect::FCDEffect(FCDocument* document) : FCDEntity(document, "Effect")
{
parameters = new FCDEffectParameterList(GetDocument(), true);
}
FCDEffect::~FCDEffect()
{
SAFE_DELETE(parameters);
CLEAR_POINTER_VECTOR(profiles);
}
void FCDEffect::AddParameter(FCDEffectParameter* parameter)
{
parameters->push_back(parameter);
}
// Flatten this effect: trickling down all the parameters to the technique level
void FCDEffect::Flatten()
{
for (FCDEffectParameterList::iterator itP = parameters->begin(); itP != parameters->end(); ++itP)
{
FCDEffectParameterList generators;
if ((*itP)->IsModifier())
{
// Overwrite the generators
FindParametersByReference((*itP)->GetReference(), generators);
for (FCDEffectParameterList::iterator itQ = generators.begin(); itQ != generators.end(); ++itQ)
{
if ((*itP) != (*itQ))
{
(*itP)->Overwrite(*itQ);
}
}
}
else
{
// Add this parameter to hierarchies below
for (FCDEffectProfileList::iterator itR = profiles.begin(); itR != profiles.end(); ++itR)
{
if ((*itR)->GetType() != FUDaeProfileType::COMMON)
{
((FCDEffectProfileFX*) (*itR))->AddParameter((*itP)->Clone());
}
}
}
}
CLEAR_POINTER_VECTOR(*parameters);
for (FCDEffectProfileList::iterator itR = profiles.begin(); itR != profiles.end(); ++itR)
{
(*itR)->Flatten();
}
}
// Search for a profile of the given type
FCDEffectProfile* FCDEffect::FindProfile(FUDaeProfileType::Type type)
{
for (FCDEffectProfileList::iterator itR = profiles.begin(); itR != profiles.end(); ++itR)
{
if ((*itR)->GetType() == type) return (*itR);
}
return NULL;
}
const FCDEffectProfile* FCDEffect::FindProfile(FUDaeProfileType::Type type) const
{
for (FCDEffectProfileList::const_iterator itR = profiles.begin(); itR != profiles.end(); ++itR)
{
if ((*itR)->GetType() == type) return (*itR);
}
return NULL;
}
// Create a new effect profile.
FCDEffectProfile* FCDEffect::AddProfile(FUDaeProfileType::Type type)
{
// Delete any old profile of the same type
FCDEffectProfile* profile = FindProfile(type);
if (profile != NULL)
{
ReleaseProfile(profile);
profile = NULL;
}
// Create the correct profile for this type.
if (type == FUDaeProfileType::COMMON) profile = new FCDEffectStandard(GetDocument(), this);
else profile = new FCDEffectProfileFX(GetDocument(), this, type);
profiles.push_back(profile);
return profile;
}
// Releases the effect profile.
void FCDEffect::ReleaseProfile(FCDEffectProfile* profile)
{
FCDEffectProfileList::iterator itR = std::find(profiles.begin(), profiles.end(), profile);
if (itR != profiles.end())
{
delete *itR;
profiles.erase(itR);
}
}
// Look for the effect parameter with the correct semantic, in order to bind/set its value
FCDEffectParameter* FCDEffect::FindParameterBySemantic(const string& semantic)
{
FCDEffectParameter* p = parameters->FindSemantic(semantic);
for (FCDEffectProfileList::iterator itR = profiles.begin(); itR != profiles.end() && p == NULL; ++itR)
{
p = (*itR)->FindParameterBySemantic(semantic);
}
return p;
}
void FCDEffect::FindParametersBySemantic(const string& semantic, FCDEffectParameterList& _parameters)
{
parameters->FindSemantic(semantic, _parameters);
for (FCDEffectProfileList::iterator itR = profiles.begin(); itR != profiles.end(); ++itR)
{
(*itR)->FindParametersBySemantic(semantic, _parameters);
}
}
void FCDEffect::FindParametersByReference(const string& reference, FCDEffectParameterList& _parameters)
{
parameters->FindReference(reference, _parameters);
for (FCDEffectProfileList::iterator itR = profiles.begin(); itR != profiles.end(); ++itR)
{
(*itR)->FindParametersBySemantic(reference, _parameters);
}
}
// Parse COLLADA document's <effect> element
// Also parses the <material> element for COLLADA 1.3 backward compatibility
FUStatus FCDEffect::LoadFromXML(xmlNode* effectNode)
{
FUStatus status = FCDEntity::LoadFromXML(effectNode);
if (!status) return status;
CLEAR_POINTER_VECTOR(*parameters);
// COLLADA 1.3 backward compatibility: look for a <material><shader> node and parse this into a standard effect.
if (IsEquivalent(effectNode->name, DAE_MATERIAL_ELEMENT))
{
xmlNode* shaderNode = FindChildByType(effectNode, DAE_SHADER_ELEMENT);
if (shaderNode != NULL)
{
FCDEffectProfile* profile = AddProfile(FUDaeProfileType::COMMON);
status.AppendStatus(profile->LoadFromXML(shaderNode));
}
}
else
{
// Accept solely <effect> elements at this point.
if (!IsEquivalent(effectNode->name, DAE_EFFECT_ELEMENT))
{
return status.Warning(FS("Unknown element in effect library."), effectNode->line);
}
for (xmlNode* child = effectNode->children; child != NULL; child = child->next)
{
if (child->type != XML_ELEMENT_NODE) continue;
if (IsEquivalent(child->name, DAE_IMAGE_ELEMENT))
{
FCDImage* image = GetDocument()->GetImageLibrary()->AddEntity();
status.AppendStatus(image->LoadFromXML(child));
images.push_back(image);
}
else if (IsEquivalent(child->name, DAE_FXCMN_SETPARAM_ELEMENT) || IsEquivalent(child->name, DAE_FXCMN_NEWPARAM_ELEMENT))
{
AddParameter(FCDEffectParameterFactory::LoadFromXML(GetDocument(), child, &status));
}
else if (IsEquivalent(child->name, DAE_EXTRA_ELEMENT))
{
// Valid element.. but not processed.
}
else
{
// Check for a valid profile element.
FUDaeProfileType::Type type = FUDaeProfileType::FromString((const char*) child->name);
if (type != FUDaeProfileType::UNKNOWN)
{
FCDEffectProfile* profile = AddProfile(type);
status.AppendStatus(profile->LoadFromXML(child));
}
else
{
status.Warning(FS("Unsupported profile or unknown element in effect: ") + TO_FSTRING(GetDaeId()), child->line);
}
}
}
}
return status;
}
// Returns a copy of the effect, with all the animations/textures attached
FCDEffect* FCDEffect::Clone()
{
FCDEffect* clone = new FCDEffect(GetDocument());
FCDEntity::Clone(clone);
for (FCDEffectProfileList::iterator itR = profiles.begin(); itR != profiles.end(); ++itR)
{
FCDEffectProfile* p = (*itR)->Clone(clone);
clone->profiles.push_back(p);
}
for (FCDEffectImageList::iterator itI = images.begin(); itI != images.end(); ++itI)
{
FCDImage* p = *itI;
clone->images.push_back(p);
}
SAFE_DELETE(clone->parameters);
clone->parameters = parameters->Clone();
return clone;
}
// Write out the <material> element to the COLLADA xml tree
xmlNode* FCDEffect::WriteToXML(xmlNode* parentNode) const
{
xmlNode* effectNode = WriteToEntityXML(parentNode, DAE_EFFECT_ELEMENT);
// Write out the parameters
for (FCDEffectParameterList::iterator itP = parameters->begin(); itP != parameters->end(); ++itP)
{
(*itP)->WriteToXML(effectNode);
}
// Write out the profiles
for (FCDEffectProfileList::const_iterator itR = profiles.begin(); itR != profiles.end(); ++itR)
{
(*itR)->WriteToXML(effectNode);
}
FCDEffect::WriteToExtraXML(effectNode);
return effectNode;
}