FabricSpliceBaseInterface.h

#ifndef _FabricSpliceBaseInterface_H_
#define _FabricSpliceBaseInterface_H_

#include "FabricSpliceConversion.h"
#include "plugin.h"

#include <vector>

#include <maya/MFnDependencyNode.h> 
#include <maya/MPlug.h> 
#include <maya/MPxNode.h> 
#include <maya/MTypeId.h> 
#include <maya/MNodeMessage.h>
#include <maya/MStringArray.h>
#include <maya/MFnCompoundAttribute.h>

#include <FabricSplice.h>

#define MAYASPLICE_CATCH_BEGIN(statusPtr) \
  if(statusPtr) \
    *statusPtr=MS::kSuccess; \
  try {

#define MAYASPLICE_CATCH_END(statusPtr) } \
  catch (FabricSplice::Exception e) { \
    mayaLogErrorFunc(e.what()); \
    if (statusPtr) \
      *statusPtr=MS::kFailure; \
  } \
  catch (FabricCore::Exception e) { \
    mayaLogErrorFunc(e.getDesc_cstr()); \
    if (statusPtr) \
      *statusPtr=MS::kFailure; \
  }

class FabricSpliceBaseInterface {

public:

  FabricSpliceBaseInterface();
  ~FabricSpliceBaseInterface();
  void constructBaseInterface();

  virtual MObject getThisMObject() = 0;
  virtual MPlug getSaveDataPlug() = 0;

  static std::vector<FabricSpliceBaseInterface*> getInstances();
  static FabricSpliceBaseInterface * getInstanceByName(const std::string & name);

  void addMayaAttribute(const MString &portName, const MString &dataType, const MString &arrayType, const FabricSplice::Port_Mode &portMode, MStatus *stat = 0);
  void addPort(const MString &portName, const MString &dataType, const FabricSplice::Port_Mode &portMode, const MString & dgNode, bool autoInitObjects, const MString & extension, const FabricCore::Variant & defaultValue, MStatus *stat = 0);
  void removeMayaAttribute(const MString &portName, MStatus *stat = 0);
  void removePort(const MString &portName, MStatus *stat = 0);
  void addKLOperator(const MString &operatorName, const MString &operatorCode, const MString &operatorEntry, const MString & dgNode, const FabricCore::Variant & portMap, MStatus *stat = 0);
  void setKLOperatorEntry(const MString &operatorName, const MString &operatorEntry, MStatus *stat = 0);
  void setKLOperatorIndex(const MString &operatorName, unsigned int operatorIndex, MStatus *stat = 0);
  void setKLOperatorCode(const MString &operatorName, const MString &operatorCode, const MString &operatorEntry, MStatus *stat = 0);
  std::string getKLOperatorCode(const MString &operatorName, MStatus *stat = 0);
  void setKLOperatorFile(const MString &operatorName, const MString &filename, const MString &entry, MStatus *stat = 0);
  void removeKLOperator(const MString &operatorName, const MString & dgNode, MStatus *stat = 0);
  void storePersistenceData(MString file, MStatus *stat = 0);
  void restoreFromPersistenceData(MString file, MStatus *stat = 0);
  void resetInternalData(MStatus *stat = 0);
  MStringArray getKLOperatorNames();
  MStringArray getPortNames();
  FabricSplice::DGPort getPort(MString name);
  void saveToFile(MString fileName);
  MStatus loadFromFile(MString fileName);
  void setPortPersistence(const MString &portName, bool persistence);
  FabricSplice::DGGraph & getSpliceGraph() { return _spliceGraph; }
  void setDgDirtyEnabled(bool enabled) { _dgDirtyEnabled = enabled; }

  static void onNodeAdded(MObject &node, void *clientData);
  static void onNodeRemoved(MObject &node, void *clientData);

  void managePortObjectValues(bool destroy);

protected:
  void invalidatePlug(MPlug & plug);
  virtual void invalidateNode();
  void incEvalID();
  void setupMayaAttributeAffects(MString portName, FabricSplice::Port_Mode portMode, MObject newAttribute, MStatus *stat = 0);

  // private members and helper methods
  static std::vector<FabricSpliceBaseInterface*> _instances;
  bool _restoredFromPersistenceData;
  unsigned int _dummyValue;

  FabricSplice::DGGraph _spliceGraph;
  MStringArray _dirtyPlugs;
  std::vector<std::string> mSpliceMayaDataOverride;
  bool _isTransferingInputs;
  bool _portObjectsDestroyed;

  void transferInputValuesToSplice(MDataBlock& data);
  void evaluate();
  void transferOutputValuesToMaya(MDataBlock& data, bool isDeformer = false);
  void collectDirtyPlug(MPlug const &inPlug);
  void affectChildPlugs(MPlug &plug, MPlugArray &affectedPlugs);
  void setDependentsDirty(MObject thisMObject, MPlug const &inPlug, MPlugArray &affectedPlugs);
  void copyInternalData(MPxNode *node);

  // static MString sManipulationCommand;
  // MString _manipulationCommand;
  bool _dgDirtyEnabled;

#if _SPLICE_MAYA_VERSION < 2014
  static std::map<std::string, int> _nodeCreatorCounts;
#endif

private:
  bool plugInArray(const MPlug &plug, const MPlugArray &array);
};

#endif