Import.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/ImportParams.h"
#include "AL/usdmaya/fileio/NodeFactory.h"
#include "AL/usdmaya/fileio/translators/TranslatorBase.h"

#include "maya/MPxCommand.h"

#include "pxr/pxr.h"
#include <pxr/base/tf/token.h>
#include <pxr/base/tf/hashmap.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 Import
{
public:

  Import(const ImporterParams& params);

  ~Import();

  inline operator bool () const
    { return m_success; }

private:
  void doImport();
  MObject createShape(translators::TranslatorRefPtr translator, const UsdPrim& prim, MObject parent, bool parentUnmerged);
  const ImporterParams& m_params;
  TfHashMap<SdfPath, MObject, SdfPath::Hash> m_instanceObjects;
  TfToken::HashSet m_nonImportablePrims;
  bool m_success;
};

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

  inline ImportCommand()
    : m_params() {}

  inline ~ImportCommand()
    {}

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

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