ProxyShapeCommands.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 "AL/usdmaya/fileio/ImportParams.h"
#include "AL/usdmaya/nodes/ProxyShape.h"

#include <map>

#include "maya/MPxCommand.h"
#include "maya/MDagModifier.h"
#include "maya/MObject.h"
#include "maya/MObjectArray.h"
#include "maya/MSelectionList.h"

#include "pxr/pxr.h"
#include "pxr/usd/usd/stage.h"

PXR_NAMESPACE_USING_DIRECTIVE

namespace AL {
namespace usdmaya {
namespace cmds {

//----------------------------------------------------------------------------------------------------------------------
//----------------------------------------------------------------------------------------------------------------------
class ProxyShapeCommandBase
  : public MPxCommand
{
public:

  static MSyntax setUpCommonSyntax();

  MDagPath getShapePath(const MArgDatabase& args);

  nodes::ProxyShape* getShapeNode(const MArgDatabase& args);

  UsdStageRefPtr getShapeNodeStage(const MArgDatabase& args);

  MArgDatabase makeDatabase(const MArgList& args);
};

//----------------------------------------------------------------------------------------------------------------------
//----------------------------------------------------------------------------------------------------------------------
class ProxyShapeImport
  : public MPxCommand
{
  MDagModifier m_modifier;
  MDagModifier m_modifier2;
  MObjectArray m_parentTransforms;
  MObject m_shape;
  MString m_proxy_name;
  bool m_createdParent = true;
public:
  AL_MAYA_DECLARE_COMMAND();
private:
  bool isUndoable() const override;
  MStatus undoIt() override;
  MStatus redoIt() override;
  MStatus doIt(const MArgList& args) override;
};

//----------------------------------------------------------------------------------------------------------------------
//----------------------------------------------------------------------------------------------------------------------
class ProxyShapeFindLoadable
  : public ProxyShapeCommandBase
{
public:
  AL_MAYA_DECLARE_COMMAND();
private:
  bool isUndoable() const override;
  MStatus doIt(const MArgList& args) override;
};

//----------------------------------------------------------------------------------------------------------------------
//----------------------------------------------------------------------------------------------------------------------
class ProxyShapeImportAllTransforms
  : public ProxyShapeCommandBase
{
  MDagModifier m_modifier;
  MDagModifier m_modifier2;
public:
  AL_MAYA_DECLARE_COMMAND();
private:
  bool isUndoable() const override;
  MStatus undoIt() override;
  MStatus redoIt() override;
  MStatus doIt(const MArgList& args) override;
};

//----------------------------------------------------------------------------------------------------------------------
//----------------------------------------------------------------------------------------------------------------------
class ProxyShapeRemoveAllTransforms
  : public ProxyShapeCommandBase
{
  MDagModifier m_modifier;
public:
  AL_MAYA_DECLARE_COMMAND();
private:
  bool isUndoable() const override;
  MStatus undoIt() override;
  MStatus redoIt() override;
  MStatus doIt(const MArgList& args) override;
};

//----------------------------------------------------------------------------------------------------------------------
//----------------------------------------------------------------------------------------------------------------------
class ProxyShapeImportPrimPathAsMaya
  : public ProxyShapeCommandBase
{
  MDagModifier m_modifier;
  SdfPath m_path;
  MDagPath m_transformPath;
  fileio::ImporterParams m_importParams;
  MObjectArray m_nodesToKill;
  bool m_asProxyShape;

  MObject makePrimTansforms(
      nodes::ProxyShape* shapeNode,
      UsdPrim usdPrim);

  MObject makeUsdTransformChain(
      std::map<std::string, MObject>& mapped,
      const UsdPrim& usdPrim,
      const MPlug& outStage,
      const MPlug& outTime,
      const MObject& parentXForm);
public:
  AL_MAYA_DECLARE_COMMAND();
private:
  bool isUndoable() const override;
  MStatus undoIt() override;
  MStatus redoIt() override;
  MStatus doIt(const MArgList& args) override;
};

//----------------------------------------------------------------------------------------------------------------------
//----------------------------------------------------------------------------------------------------------------------
class ProxyShapePrintRefCountState
  : public ProxyShapeCommandBase
{
public:
  AL_MAYA_DECLARE_COMMAND();
private:
  bool isUndoable() const override;
  MStatus doIt(const MArgList& args) override;
};

//----------------------------------------------------------------------------------------------------------------------
//----------------------------------------------------------------------------------------------------------------------
class ProxyShapeResync
  : public ProxyShapeCommandBase
{
  SdfPath m_resyncPrimPath;
  UsdPrim m_resyncPrim;
  nodes::ProxyShape* m_shapeNode;
  bool m_runPrePrimChanged;
  bool m_runPostPrimChanged;
public:
  AL_MAYA_DECLARE_COMMAND();
private:
  bool isUndoable() const override;
  MStatus doIt(const MArgList& args) override;
  MStatus redoIt() override;
};

//----------------------------------------------------------------------------------------------------------------------
//----------------------------------------------------------------------------------------------------------------------
class ProxyShapeSelect
  : public ProxyShapeCommandBase
{
  nodes::SelectionUndoHelper* m_helper;
public:
  ProxyShapeSelect () : m_helper(0) {}
  ~ProxyShapeSelect() { delete m_helper; }
  AL_MAYA_DECLARE_COMMAND();
private:
  bool isUndoable() const override;
  MStatus doIt(const MArgList& args) override;
  MStatus undoIt() override;
  MStatus redoIt() override;
  MStatus _redoIt(bool isInternal);
};

//----------------------------------------------------------------------------------------------------------------------
//----------------------------------------------------------------------------------------------------------------------
class ProxyShapePostSelect
  : public ProxyShapeCommandBase
{
  nodes::ProxyShape* m_proxy;
public:
  ~ProxyShapePostSelect() {}
  AL_MAYA_DECLARE_COMMAND();
private:
  bool isUndoable() const override;
  MStatus doIt(const MArgList& args) override;
  MStatus undoIt() override;
  MStatus redoIt() override;
};

//----------------------------------------------------------------------------------------------------------------------
//----------------------------------------------------------------------------------------------------------------------
class InternalProxyShapeSelect
  : public ProxyShapeCommandBase
{
  nodes::ProxyShape* m_proxy;
  nodes::SelectionList m_previous;
  nodes::SelectionList m_new;
public:
  AL_MAYA_DECLARE_COMMAND();
private:
  bool isUndoable() const override;
  MStatus doIt(const MArgList& args) override;
  MStatus undoIt() override;
  MStatus redoIt() override;
};


//----------------------------------------------------------------------------------------------------------------------
//----------------------------------------------------------------------------------------------------------------------
class TranslatePrim
  : public ProxyShapeCommandBase
{
  fileio::translators::TranslatorParameters tp;

  nodes::ProxyShape* m_proxy;
  SdfPathVector m_importPaths;
  SdfPathVector m_teardownPaths;

public:
  AL_MAYA_DECLARE_COMMAND();
private:
  bool isUndoable() const override;
  MStatus doIt(const MArgList& args) override;
  MStatus redoIt() override;
};


AL_USDMAYA_PUBLIC
void constructProxyShapeCommandGuis();

AL_USDMAYA_PUBLIC
void constructPickModeCommandGuis();

//----------------------------------------------------------------------------------------------------------------------
} // cmds
} // usdmaya
} // AL
//----------------------------------------------------------------------------------------------------------------------