Transform.h

//
// Copyright 2017 Animal Logic
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.//
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
#pragma once
#include <AL/usdmaya/ForwardDeclares.h>

#include "AL/maya/utils/NodeHelper.h"
#include "AL/usdmaya/utils/ForwardDeclares.h"
#include "AL/maya/utils/MayaHelperMacros.h"
#include "maya/MObjectHandle.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 AL::maya::utils::NodeHelper
{
public:

  Transform();
  ~Transform();

  //--------------------------------------------------------------------------------------------------------------------
  // 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(outTime);

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

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

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

  inline const MObject getProxyShape() const
    { return proxyShapeHandle.object(); }

private:

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

  MStatus validateAndSetValue(const MPlug& plug, const MDataHandle& handle, const MDGContext& context) override;
  MPxTransformationMatrix* createTransformationMatrix() override;
  MStatus compute(const MPlug &plug, MDataBlock &datablock) override;
  void postConstructor() override;
  MBoundingBox boundingBox() const override;
  MStatus connectionMade(const MPlug& plug, const MPlug& otherPlug, bool asSrc) override;
  MStatus connectionBroken(const MPlug& plug, const MPlug& otherPlug, bool asSrc) override;
  bool isBounded() const override
    { return true; }
  bool treatAsTransform() const override
    { return false; }

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

  void updateTransform(MDataBlock& dataBlock);

  //--------------------------------------------------------------------------------------------------------------------
  //--------------------------------------------------------------------------------------------------------------------
  bool updateTransformInProgress = false;

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



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


  MObjectHandle proxyShapeHandle;
};

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