Transform.h

#pragma once
#include "AL/usdmaya/Common.h"
#include "AL/maya/NodeHelper.h"
#include "maya/MPxTransform.h"

namespace AL {
namespace usdmaya {
namespace nodes {

//----------------------------------------------------------------------------------------------------------------------
//          transform type directly from inside Maya. It works by providing a custom MPxTransform node which uses a
//----------------------------------------------------------------------------------------------------------------------
class Transform
  : public MPxTransform,
    public maya::NodeHelper
{
public:

  //--------------------------------------------------------------------------------------------------------------------
  // Type Info & Registration
  //--------------------------------------------------------------------------------------------------------------------
  AL_MAYA_DECLARE_NODE();

  //--------------------------------------------------------------------------------------------------------------------
  // Input Attributes
  //--------------------------------------------------------------------------------------------------------------------
  AL_DECL_ATTRIBUTE(primPath);
  AL_DECL_ATTRIBUTE(inStageData);
  AL_DECL_ATTRIBUTE(time);
  AL_DECL_ATTRIBUTE(timeOffset);
  AL_DECL_ATTRIBUTE(timeScalar);
  AL_DECL_ATTRIBUTE(localTranslateOffset);
  AL_DECL_ATTRIBUTE(pushToPrim);
  AL_DECL_ATTRIBUTE(readAnimatedValues);

  //--------------------------------------------------------------------------------------------------------------------
  // Output Attributes
  //--------------------------------------------------------------------------------------------------------------------
  AL_DECL_ATTRIBUTE(outStageData);
  AL_DECL_ATTRIBUTE(outTime);

  //--------------------------------------------------------------------------------------------------------------------
  //--------------------------------------------------------------------------------------------------------------------

  inline TransformationMatrix* transform() const
    { return reinterpret_cast<TransformationMatrix*>(transformationMatrixPtr()); }

  UsdPrim getUsdPrim(MDataBlock& dataBlock) const;

  MPxNode::SchedulingType schedulingType() const override
    { return kParallel; }

private:

  //--------------------------------------------------------------------------------------------------------------------
  //--------------------------------------------------------------------------------------------------------------------

  MStatus validateAndSetValue(const MPlug& plug, const MDataHandle& handle, const MDGContext& context) override;
  MPxTransformationMatrix* createTransformationMatrix() override;
  MStatus connectionMade(const MPlug &plug, const MPlug &otherPlug, bool asSrc) override;
  MStatus connectionBroken(const MPlug &plug, const MPlug &otherPlug, bool asSrc) override;
  MStatus compute(const MPlug &plug, MDataBlock &datablock) override;
  void postConstructor() override;

  //--------------------------------------------------------------------------------------------------------------------
  //--------------------------------------------------------------------------------------------------------------------

  void updateTransform(MDataBlock& dataBlock);
  bool isStageValid() const;

  //--------------------------------------------------------------------------------------------------------------------
  //--------------------------------------------------------------------------------------------------------------------



  //--------------------------------------------------------------------------------------------------------------------
  //--------------------------------------------------------------------------------------------------------------------

};

//----------------------------------------------------------------------------------------------------------------------
} // nodes
} // usdmaya
} // AL
//----------------------------------------------------------------------------------------------------------------------