LayerVisitor.h

#pragma once
#include "AL/usdmaya/Common.h"

#include <stack>
#include <vector>

namespace AL {
namespace usdmaya {
namespace nodes {

//----------------------------------------------------------------------------------------------------------------------
//----------------------------------------------------------------------------------------------------------------------
class LayerVisitor
{
public:

  LayerVisitor(ProxyShape* shape);

  virtual ~LayerVisitor() {}

  void visitAll();

  virtual void onVisit() {}

  inline Layer* thisLayer() const
    { return m_stack.top().m_thisLayer; }

  inline Layer* parentLayer() const
    { return m_stack.top().m_parentLayer; }

  inline bool isSubLayer() const
    { return m_stack.top().m_isSubLayer; }

  inline uint32_t depth() const
    { return static_cast<uint32_t>(m_stack.size()); }

private:
  void visit();
  struct StackItem
  {
    StackItem(Layer* layer, const bool isSubLayer);
    std::vector<Layer*> m_subLayers;
    std::vector<Layer*> m_childLayers;
    Layer* m_thisLayer;
    Layer* m_parentLayer;
    int32_t m_index;
    bool m_isSubLayer;
  };

  std::stack<StackItem> m_stack;
  ProxyShape* m_shape;
};

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