Export.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/fileio/ExportParams.h"
#include "AL/usdmaya/fileio/translators/TranslatorBase.h"

#include "maya/MPxCommand.h"

#include "pxr/pxr.h"
#include "AL/usdmaya/utils/ForwardDeclares.h"
#include "AL/maya/utils/Api.h"
#include "AL/maya/utils/MayaHelperMacros.h"

PXR_NAMESPACE_USING_DIRECTIVE


namespace AL {
namespace usdmaya {
namespace fileio {

//----------------------------------------------------------------------------------------------------------------------
//----------------------------------------------------------------------------------------------------------------------
class Export
{
public:

  Export(const ExporterParams& params);

  ~Export();

private:

  enum ReferenceType
  {
    kNoReference,
    kShapeReference,
    kTransformReference,
  };

  void exportSceneHierarchy(MDagPath path, SdfPath& defaultPrim);
  void exportShapesCommonProc(MDagPath shapePath, MFnTransform& fnTransform, SdfPath& usdPath, ReferenceType refType);
  void exportShapesOnlyUVProc(MDagPath shapePath, MFnTransform& fnTransform, SdfPath& usdPath);
  UsdPrim exportMeshUV(MDagPath path, const SdfPath& usdPath);
  UsdPrim exportAssembly(MDagPath path, const SdfPath& usdPath);
  UsdPrim exportPluginLocatorNode(MDagPath path, const SdfPath& usdPath);
  UsdPrim exportPluginShape(MDagPath path, const SdfPath& usdPath);
  void exportIkChain(MDagPath effectorPath, const SdfPath& usdPath);
  void exportGeometryConstraint(MDagPath effectorPath, const SdfPath& usdPath);
  void copyTransformParams(UsdPrim prim, MFnTransform& fnTransform);
  SdfPath determineUsdPath(MDagPath path, const SdfPath& usdPath, ReferenceType refType);
  void addReferences(MDagPath shapePath, MFnTransform& fnTransform, SdfPath& usdPath,
                     const SdfPath& instancePath, ReferenceType refType);
  inline bool isPrimDefined(SdfPath &usdPath);
  struct Impl;
  void doExport();
  const ExporterParams& m_params;
  Impl* m_impl;
  translators::TranslatorManufacture m_translatorManufacture;

};

//----------------------------------------------------------------------------------------------------------------------
//----------------------------------------------------------------------------------------------------------------------
class ExportCommand
  : public MPxCommand
{
public:
  AL_MAYA_DECLARE_COMMAND();

  ExportCommand();

  ~ExportCommand();

private:
  MStatus doIt(const MArgList& args);
  MStatus redoIt();
  MStatus undoIt();
  bool isUndoable() const { return true; };
  ExporterParams m_params;
};

//----------------------------------------------------------------------------------------------------------------------
} // fileio
} // usdmaya
} // AL
//----------------------------------------------------------------------------------------------------------------------