/****************************************************************************
**
** Copyright (C) 1993-2009 NVIDIA Corporation.
** Copyright (C) 2017 The Qt Company Ltd.
** Contact: https://www.qt.io/licensing/
**
** This file is part of Qt 3D Studio.
**
** $QT_BEGIN_LICENSE:GPL-EXCEPT$
** Commercial License Usage
** Licensees holding valid commercial Qt licenses may use this file in
** accordance with the commercial license agreement provided with the
** Software or, alternatively, in accordance with the terms contained in
** a written agreement between you and The Qt Company. For licensing terms
** and conditions see https://www.qt.io/terms-conditions. For further
** information use the contact form at https://www.qt.io/contact-us.
**
** GNU General Public License Usage
** Alternatively, this file may be used under the terms of the GNU
** General Public License version 3 as published by the Free Software
** Foundation with exceptions as appearing in the file LICENSE.GPL3-EXCEPT
** included in the packaging of this file. Please review the following
** information to ensure the GNU General Public License requirements will
** be met: https://www.gnu.org/licenses/gpl-3.0.html.
**
** $QT_END_LICENSE$
**
****************************************************************************/
#pragma once
#ifndef QT3DS_IMPORT_H
#define QT3DS_IMPORT_H
#include "Qt3DSImportLibPrecompile.h"
#include "Qt3DSImportMesh.h"
#include "Qt3DSImportPath.h"
#include "Qt3DSImportPerformImport.h"
#include "Qt3DSDMStringTable.h"
#include "Qt3DSImportComposerTypes.h"

namespace qt3dsimp {
using namespace std;
typedef QT3DSU64 TIMPHandle;

struct InstanceDesc
{
    TCharPtr m_Id;
    TIMPHandle m_Parent;
    TIMPHandle m_Handle;
    ComposerObjectTypes::Enum m_Type;
    InstanceDesc()
        : m_Id(NULL)
        , m_Parent(0)
        , m_Handle(0)
        , m_Type(ComposerObjectTypes::Unknown)
    {
    }
};

struct Animation
{
    TCharPtr m_InstanceId;
    TCharPtr m_PropertyName;
    QT3DSU32 m_SubPropertyIndex;
    EAnimationType m_Type;
    NVConstDataRef<QT3DSF32> m_Keyframes;
    Animation(TCharPtr iid, TCharPtr pname, QT3DSU32 subPropIndex, EAnimationType bufType,
              NVConstDataRef<QT3DSF32> keyframes)
        : m_InstanceId(iid)
        , m_PropertyName(pname)
        , m_SubPropertyIndex(subPropIndex)
        , m_Type(bufType)
        , m_Keyframes(keyframes)
    {
    }
    Animation()
        : m_InstanceId(L"")
        , m_PropertyName(L"")
        , m_SubPropertyIndex(0)
        , m_Type(EAnimationTypeNone)
    {
    }
};

struct AddRemoveInfo
{
    QT3DSU32 m_Existing; // intersection of new and old
    QT3DSU32 m_NumAdded; // new that doesn't exist in old
    QT3DSU32 m_NumRemoved; // old that doesn't exist in new
    AddRemoveInfo()
        : m_Existing(0)
        , m_NumAdded(0)
        , m_NumRemoved(0)
    {
    }
    AddRemoveInfo(QT3DSU32 exist, QT3DSU32 add, QT3DSU32 remove)
        : m_Existing(exist)
        , m_NumAdded(add)
        , m_NumRemoved(remove)
    {
    }
};

struct ParentChildLink
{
    TCharPtr m_Parent;
    TCharPtr m_Child;
    TCharPtr m_NextSibling;

    ParentChildLink(TCharPtr p, TCharPtr c, TCharPtr ns)
        : m_Parent(p)
        , m_Child(c)
        , m_NextSibling(ns)
    {
    }
    ParentChildLink() {}
    bool operator<(const ParentChildLink &other) const
    {
        // This works because at this point id's are in the string table
        // so equal strings are pointed to by the same pointer.
        return m_Parent < other.m_Parent;
    }
};

template <typename TDataType>
struct AddRemoveData
{
    NVConstDataRef<TDataType> m_Existing;
    NVConstDataRef<TDataType> m_Added;
    NVConstDataRef<TDataType> m_Removed;
    AddRemoveData() {}
    AddRemoveData(NVConstDataRef<TDataType> exist, NVConstDataRef<TDataType> add,
                  NVConstDataRef<TDataType> rem)
        : m_Existing(exist)
        , m_Added(add)
        , m_Removed(rem)
    {
    }
};

// Contains the basic information to perform an import or a refresh.
// Note that during an import or refresh, the user-id information on the
// descriptors may change.  So you can't ever use exactly this information
// on the descriptors.  You have to always query the descriptor.  Specifically,
// Create instance, delete instance, and update instance can change user id
// information.
struct ImportReport
{
    // There is a precondition you can rely on that parents are added
    // before children.  Furthermore, instances that are added won't
    // appear in the below parent child link added section.
    AddRemoveData<InstanceDesc> m_Instances;
    // For children, existing does tell you the intersection of the two
    // sets *but* existing is in the order of the new set.
    // This allows clients to detect child reordering where it may have
    // a substantial effect (such as materials which would then get mapped
    // to a different mesh instance).
    AddRemoveData<ParentChildLink> m_Links;

    AddRemoveData<Pair<TCharPtr, TCharPtr>> m_Images;
    AddRemoveData<Pair<TCharPtr, TCharPtr>> m_Meshes;
    AddRemoveData<Pair<TCharPtr, TCharPtr>> m_PathBuffers;
    AddRemoveData<Animation> m_Animations;

    ImportReport() {}
    ImportReport(AddRemoveData<InstanceDesc> instances, AddRemoveData<ParentChildLink> links,
                 AddRemoveData<Pair<TCharPtr, TCharPtr>> imageBuffers,
                 AddRemoveData<Pair<TCharPtr, TCharPtr>> meshes,
                 AddRemoveData<Pair<TCharPtr, TCharPtr>> pathBuffers,
                 AddRemoveData<Animation> animBuffers)
        : m_Instances(instances)
        , m_Links(links)
        , m_Images(imageBuffers)
        , m_Meshes(meshes)
        , m_PathBuffers(pathBuffers)
        , m_Animations(animBuffers)
    {
    }
};

template <typename TDataType>
struct DatatypeOrError
{
    bool m_Error;
    ImportErrorData m_ErrorData;
    TDataType m_Value;
    DatatypeOrError(const TDataType &value)
        : m_Error(false)
        , m_Value(value)
    {
    }
    DatatypeOrError(const ImportErrorData &error, const TDataType &badValue = TDataType())
        : m_Error(true)
        , m_ErrorData(error)
        , m_Value(badValue)
    {
    }

    TDataType &operator->()
    {
        QT3DS_ASSERT(m_Error == false);
        return m_Value;
    }
    TDataType &operator*()
    {
        QT3DS_ASSERT(m_Error == false);
        return m_Value;
    }
    operator TDataType()
    {
        QT3DS_ASSERT(m_Error == false);
        return m_Value;
    }
};
class Import;
typedef DatatypeOrError<Import *> ImportPtrOrError;
typedef DatatypeOrError<TCharPtr> CharPtrOrError;

template <typename TDataType>
struct SImportConverter
{
    TImportModelValue Convert(const TDataType &inType) { return inType; }
};

template <>
struct SImportConverter<TDataStrPtr>
{
    const wchar_t *Convert(const wchar_t *inValue) { return inValue; }
};
// Long4s are string refs, so they are set by string
template <>
struct SImportConverter<SLong4>
{
    SValue Convert(const wchar_t *inValue) { return SStringRef(inValue); }
};

class Import
{
protected:
    virtual ~Import(){}

public:
    // So users can check if a file is an import file.
    //<UICImportLib version="${fileversion}"
    static QT3DSU32 GetImportFileVersion() { return 1; }

    // Cache this string in the string table and return
    // a representation that will be around until this import object
    // is destroyed.
    virtual TCharPtr RegisterStr(TCharPtr data) = 0;
    // Returns the source directory relative to the dest directory
    // or the full path if it is on a different drive
    virtual QString GetSrcFile() const = 0;
    // Returns the full path of the destination directory
    virtual QString GetDestDir() const = 0;
    // Returns the full path of the image directory
    virtual QString GetImageDir() const = 0;
    // Returns the full path of the mesh directory
    virtual QString GetMeshDir() const = 0;
    // Returns the full path to the path buffer directory
    virtual QString GetPathBufferDir() const = 0;
    virtual void Release() = 0;
    // returns false if fname couldn't be opened for write
    // Fname is appended to the directory this object was created with.
    // This is necessary in order to keep relative paths valid within
    // the import file.  Returns an ID that uniquely identifies this import
    // object within the final document.
    virtual QT3DSU32 Save(TCharPtr fname) const = 0;

    // Add a mapping from an named id to a handle
    virtual Option<InstanceDesc> GetInstanceByHandle(TIMPHandle inst) const = 0;
    virtual Option<InstanceDesc> FindInstanceById(TCharPtr inst) const = 0;
    virtual Option<InstanceDesc> FindAnyInstanceById(TCharPtr inst) const = 0;
    virtual QT3DSU32 GetNumInstances() const = 0;
    virtual QT3DSU32 GetInstances(NVDataRef<InstanceDesc> outDescs) const = 0;
    virtual QT3DSU32 GetNumProperties(TIMPHandle instance) const = 0;
    virtual QT3DSU32 GetProperties(TIMPHandle inst, NVDataRef<PropertyValue> outBuffer) const = 0;
    virtual Option<SValue> GetInstancePropertyValue(TIMPHandle inst,
                                                    ComposerPropertyNames::Enum val) const = 0;
    virtual QT3DSU32 GetNumChildren(TIMPHandle instance) const = 0;
    virtual QT3DSU32 GetChildren(TIMPHandle instance, NVDataRef<InstanceDesc> childBuffer) const = 0;
    // Invalid instances will not be saved out to the import file and should be ignored
    // in the import report.
    virtual void MarkInstanceInvalid(TIMPHandle inst) = 0;

    virtual TIMPHandle CreateInstance(TCharPtr id, ComposerObjectTypes::Enum inType) = 0;
    // The new instance ends up attached to the same parent
    // and just after inSource in the parent's children lists.
    // This performs a deep copy and fixes up references if they occur in the
    // new hierarchy.
    virtual TIMPHandle CopyInstance(TIMPHandle inSource) = 0;

    // Returns true if inst exists, false otherwise.

    virtual bool SetInstanceProperties(TIMPHandle inst, NVConstDataRef<PropertyValue> inBuffer) = 0;

    template <typename TPropertyType, typename TDataType>
    void SetInstancePropertyValue(TIMPHandle inst,
                                  const SImportPropertyDefinition<TPropertyType> &inProperty,
                                  const TDataType &val)
    {
        DoSetInstancePropertyValue(inst, inProperty.m_Name,
                                   SImportConverter<TPropertyType>().Convert(val));
    }

    virtual bool AddChild(TIMPHandle inst, TIMPHandle child) = 0;

    virtual QT3DSU32 GetNumImages() const = 0;
    // Returns imgOriginalPath,imgDestPath pair
    virtual QT3DSU32 GetImages(NVDataRef<Pair<TCharPtr, TCharPtr>> imgPaths) const = 0;
    // Copies the an appropriate location in our import directory
    // Returns the path of the added image.  This may mangle the name slightly
    // In case of a conflict.
    // If this function returns L"", there was a problem copying the data
    // and you need to abort the import process immediately.  For a new import,
    // then delete the existing imported images.  For refresh, the user may just
    // be stuck with a half-updated dataset.
    // This path is relative to the current working directory when this object was created
    // or absolute.
    // Returns a relative path, from dest directory, where the image was saved out to.
    virtual CharPtrOrError AddImage(TCharPtr imgPath) = 0;
    // Assuming this image path is relative to the current working directory, find the image
    virtual Option<TCharPtr> FindImageByPath(TCharPtr imgPath) const = 0;
    // Assuming this path is relative to the import source document, find the image.
    virtual Option<TCharPtr> FindImageByRelativePath(TCharPtr imgPath) const = 0;

    virtual QT3DSU32 GetNumMeshes() const = 0;
    // Returns meshName,meshDestPath pair
    virtual QT3DSU32 GetMeshes(NVDataRef<Pair<TCharPtr, TCharPtr>> bufferPaths) const = 0;

    // Copies the vertex buffer into the appropriate location, renaming if necessary.
    // Mesh name is used to write out a reasonable buffer *and* on refresh to know
    // if a buffer is changed/updated or not
    // If this function returns L"", there was a problem copying the data
    // and you need to abort the import process immediately. For a new import,
    // then delete the existing imported images.  For refresh, the user may just
    // be stuck with a half-updated dataset.
    // Returns a relative path, from the mesh directory, where the mesh was saved out
    // to.
    virtual CharPtrOrError AddMesh(const Mesh &meshBuffer, TCharPtr meshName) = 0;
    virtual bool HasMesh(TCharPtr meshName) const = 0;
    // Return the mesh path with the current version number
    virtual Option<TCharPtr> FindMeshReferencePathByName(TCharPtr meshName) const = 0;
    virtual Option<TCharPtr> FindMeshFilePathByName(TCharPtr meshName) const = 0;

    virtual QT3DSU32 GetNumPathBuffers() const = 0;
    // Returns imgOriginalPath,imgDestPath pair
    virtual QT3DSU32 GetPathBuffers(NVDataRef<Pair<TCharPtr, TCharPtr>> pathBufferPaths) const = 0;
    virtual CharPtrOrError AddPathBuffer(const SPathBuffer &pathBuffer, TCharPtr pathName) = 0;
    // Assuming this Path path is relative to the current working directory, find the Path
    virtual Option<TCharPtr> FindPathBufferByPath(TCharPtr pathBufferPath) const = 0;
    // Assuming this path is relative to the import source document, find the Path.
    virtual Option<TCharPtr> FindPathBufferByRelativePath(TCharPtr pathBufferPath) const = 0;

    virtual QT3DSU32 GetNumAnimations() const = 0;
    virtual QT3DSU32 GetAnimations(NVDataRef<Animation> outBuffers) const = 0;
    // Data is copied into this object, you can release the anim buffer data after this
    virtual Option<Animation> FindAnimation(TCharPtr instance, TCharPtr propName,
                                            QT3DSU32 subPropIndex) const = 0;
    template <typename TDataType>
    void AddAnimation(TCharPtr instance, const SImportPropertyDefinition<TDataType> &inProperty,
                      QT3DSU32 subPropIndex, EAnimationType type, NVConstDataRef<QT3DSF32> values)
    {
        size_t arity = getDatatypeAnimatableArity(TypeToDataType<TDataType>());
        if (arity) {
            if (subPropIndex >= arity) {
                QT3DS_ASSERT(false);
                subPropIndex = 0;
            }
            DoAddAnimation(instance, ComposerPropertyNames::Convert(inProperty.GetName()),
                           subPropIndex, type, values);
        } else {
            QT3DS_ASSERT(false);
        }
    }

    // Call after the import process is complete in order to get all the new items.
    virtual ImportReport CompileReport() const = 0;

protected:
    // Careful with this.  If the property value contains heap memory
    // Then this may crash if you are coming from a dll.
    virtual void DoAddAnimation(TCharPtr instance, TCharPtr propName, QT3DSU32 subPropIndex,
                                EAnimationType type, NVConstDataRef<QT3DSF32> values) = 0;
    virtual bool DoSetInstancePropertyValue(TIMPHandle inst, ComposerPropertyNames::Enum pname,
                                            const TImportModelValue &val) = 0;
    virtual bool DoSetInstancePropertyValue(TIMPHandle inst, ComposerPropertyNames::Enum pname,
                                            TCharPtr val) = 0;

public:
    // Create blank import object
    // It is the caller's responsibility to ensure that imgPath and meshBufPath exist and
    // are writable.
    // Of srcDirectoryOrFile is a file, we get the directory of that file using
    // FilePath::GetDirName( srcDirectoryOrFile )
    // We then get its absolute path, get our absolute path, and attempt to create
    // a relative path from dest directory to src directory.
    static ImportPtrOrError Create(TCharPtr srcFile, TCharPtr destDirectory);

    // Load an import object from a file.  This can obviously fail
    // Also, an import file can contain multiple documents so you need to identify
    // which document you are talking about.
    // 0 means load the latest document.
    static ImportPtrOrError Load(TCharPtr pathToFile, QT3DSU32 inDocumentId = 0);

    // Create an import object that will update heavy resources (like images and mesh objects)
    // and it CompileReport will return objects that indicate the differences between the new
    // document and the existing document.
    // You have a chance to refresh from a document that isn't in the same location.
    // We use relative paths throughout the system so that image src paths are relative
    // to the src document directory and such, so we can still import sanely.
    // Dest directory is set to where original's dest directory was set to.
    static ImportPtrOrError CreateForRefresh(Import &original, TCharPtr srcFile);

    // Return the highest import version in a given document.  Returns zero upon
    // failure, else an integer that is valid for load.
    static QT3DSU32 GetHighestImportRevision(TCharPtr pathToFile);
};
}

#endif