DrivenTransforms.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 "../../Api.h"

#include "pxr/usd/sdf/path.h"
#include "pxr/usd/usd/prim.h"

#include "maya/MPxData.h"
#include "maya/MVector.h"
#include "maya/MMatrix.h"

#include <vector>
#include <string>
#include "AL/maya/utils/ForwardDeclares.h"
#include "AL/usd/utils/ForwardDeclares.h"

PXR_NAMESPACE_USING_DIRECTIVE

namespace AL {
namespace usdmaya {
namespace nodes {
namespace proxy {

//----------------------------------------------------------------------------------------------------------------------
//----------------------------------------------------------------------------------------------------------------------
class DrivenTransforms
{
public:

  inline DrivenTransforms()
    : m_drivenPrimPaths(), m_drivenMatrix(), m_drivenVisibility(), m_dirtyMatrices(), m_dirtyVisibilities() {}

  inline size_t transformCount() const
    { return m_drivenPrimPaths.size(); }

  AL_USDMAYA_PUBLIC
  void resizeDrivenTransforms(const size_t primPathCount);

  inline void setDrivenPrimPaths(const SdfPathVector& primPaths)
    { m_drivenPrimPaths = primPaths; }

  AL_USDMAYA_PUBLIC
  bool update(UsdStageRefPtr stage, const MTime& currentTime);

  inline void dirtyVisibility(const int32_t primIndex, bool newValue)
    {
      m_drivenVisibility[primIndex] = newValue;
      m_dirtyVisibilities.push_back(primIndex);
    }

  inline void dirtyMatrix(const int32_t primIndex, const MMatrix& newValue)
    {
      m_drivenMatrix[primIndex] = newValue;
      m_dirtyMatrices.push_back(primIndex);
    }

  inline const SdfPathVector& drivenPrimPaths() const
    { return m_drivenPrimPaths; }

  inline const std::vector<int32_t>& dirtyMatrices() const
    { return m_dirtyMatrices; }

  inline const std::vector<int32_t>& dirtyVisibilities() const
    { return m_dirtyVisibilities; }

  inline const std::vector<MMatrix>& drivenMatrices() const
    { return m_drivenMatrix; }

  inline const std::vector<bool>& drivenVisibilities() const
    { return m_drivenVisibility; }

private:
  void updateDrivenVisibility(std::vector<UsdPrim>& drivenPrims, const MTime& currentTime);
  void updateDrivenTransforms(std::vector<UsdPrim>& drivenPrims, const MTime& currentTime);
private:
  SdfPathVector m_drivenPrimPaths;
  std::vector<MMatrix> m_drivenMatrix;
  std::vector<bool> m_drivenVisibility;
  std::vector<int32_t> m_dirtyMatrices;
  std::vector<int32_t> m_dirtyVisibilities;
};

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