/*
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
* its licensors.
*
* For complete copyright and license terms please see the LICENSE at the root of this
* distribution (the "License"). All use of this software is governed by the License,
* or, if provided, by the license below or the license accompanying this file. Do not
* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*
*/
// Original file Copyright Crytek GMBH or its affiliates, used under license.

#include "StdAfx.h"
#include "ObjectLoader.h"
#include "Util/PakFile.h"
#include "Material/MaterialManager.h"

//////////////////////////////////////////////////////////////////////////
// CObjectArchive Implementation.
//////////////////////////////////////////////////////////////////////////
CObjectArchive::CObjectArchive(IObjectManager* objMan, XmlNodeRef xmlRoot, bool loading)
{
    m_objectManager = objMan;
    bLoading = loading;
    bUndo = false;
    m_nFlags = 0;
    node = xmlRoot;
    m_pCurrentErrorReport = GetIEditor()->GetErrorReport();
    m_pGeometryPak = NULL;
    m_pCurrentObject = NULL;
    m_bNeedResolveObjects = false;
    m_bProgressBarEnabled = true;
}

//////////////////////////////////////////////////////////////////////////
CObjectArchive::~CObjectArchive()
{
    if (m_pGeometryPak)
    {
        delete m_pGeometryPak;
    }
    // Always make sure objects are resolved when loading from archive.
    if (bLoading && m_bNeedResolveObjects)
    {
        ResolveObjects();
    }
}

//////////////////////////////////////////////////////////////////////////
void CObjectArchive::SetResolveCallback(CBaseObject* fromObject, REFGUID objectId, ResolveObjRefFunctor1 func)
{
    if (objectId == GUID_NULL)
    {
        func(0);
        return;
    }

    GUID guid(objectId);
    if (m_nFlags & eObjectLoader_ReconstructPrefabs)
    {
        for (int i = 0, iLoadedObjectsSize(m_loadedObjects.size()); i < iLoadedObjectsSize; ++i)
        {
            GUID guidInPrefab;
            if (!m_loadedObjects[i].xmlNode->getAttr("Id", guidInPrefab))
            {
                continue;
            }
            if (guidInPrefab == objectId)
            {
                guid = m_loadedObjects[i].newGuid;
                break;
            }
        }
    }

    CBaseObject* pObject = m_objectManager->FindObject(guid);
    if (pObject && !(m_nFlags & eObjectLoader_MakeNewIDs))
    {
        // Object is already resolved. immediately call callback.
        func(pObject);
    }
    else
    {
        Callback cb;
        cb.fromObject = fromObject;
        cb.func1 = func;
        m_resolveCallbacks.insert(Callbacks::value_type(guid, cb));
    }
}

//////////////////////////////////////////////////////////////////////////
void CObjectArchive::SetResolveCallback(CBaseObject* fromObject, REFGUID objectId, ResolveObjRefFunctor2 func, uint32 userData)
{
    if (objectId == GUID_NULL)
    {
        func(0, userData);
        return;
    }

    CBaseObject* object = m_objectManager->FindObject(objectId);
    if (object && !(m_nFlags & eObjectLoader_MakeNewIDs))
    {
        // Object is already resolved. immidiatly call callback.
        func(object, userData);
    }
    else
    {
        Callback cb;
        cb.fromObject = fromObject;
        cb.func2 = func;
        cb.userData = userData;
        m_resolveCallbacks.insert(Callbacks::value_type(objectId, cb));
    }
}

//////////////////////////////////////////////////////////////////////////
GUID CObjectArchive::ResolveID(REFGUID id)
{
    return stl::find_in_map(m_IdRemap, id, id);
}

//////////////////////////////////////////////////////////////////////////
void CObjectArchive::ResolveObjects()
{
    int i;

    if (!bLoading)
    {
        return;
    }

    {
        CWaitProgress wait("Loading Objects", false);
        if (m_bProgressBarEnabled)
        {
            wait.Start();
        }

        GetIEditor()->SuspendUndo();
        //////////////////////////////////////////////////////////////////////////
        // Serialize All Objects from XML.
        //////////////////////////////////////////////////////////////////////////
        int numObj = m_loadedObjects.size();
        for (i = 0; i < numObj; i++)
        {
            if (m_bProgressBarEnabled)
            {
                wait.Step((i * 100) / numObj);
            }

            SLoadedObjectInfo& obj = m_loadedObjects[i];
            m_pCurrentErrorReport->SetCurrentValidatorObject(obj.pObject);
            node = obj.xmlNode;

            obj.pObject->Serialize(*this);

            m_pCurrentErrorReport->SetCurrentValidatorObject(nullptr);

            // Objects can be added to the list here (from Groups).
            numObj = m_loadedObjects.size();
        }
        m_pCurrentErrorReport->SetCurrentValidatorObject(NULL);
        //////////////////////////////////////////////////////////////////////////
        GetIEditor()->ResumeUndo();
    }

    // Sort objects by sort order.
    std::sort(m_loadedObjects.begin(), m_loadedObjects.end());

    // Then rearrange to parent-first, if same sort order.
    for (i = 0; i < m_loadedObjects.size(); i++)
    {
        if (m_loadedObjects[i].pObject->GetParent())
        {
            // Find later in array.
            for (int j = i + 1; j < m_loadedObjects.size() && m_loadedObjects[j].nSortOrder == m_loadedObjects[i].nSortOrder; j++)
            {
                if (m_loadedObjects[j].pObject == m_loadedObjects[i].pObject->GetParent())
                {
                    // Swap the objects.
                    std::swap(m_loadedObjects[i], m_loadedObjects[j]);
                    i--;
                    break;
                }
            }
        }
    }

    //////////////////////////////////////////////////////////////////////////
    // Resolve objects GUIDs
    //////////////////////////////////////////////////////////////////////////
    for (Callbacks::iterator it = m_resolveCallbacks.begin(); it != m_resolveCallbacks.end(); it++)
    {
        Callback& cb = it->second;
        GUID objectId = ResolveID(it->first);
        CBaseObject* object = m_objectManager->FindObject(objectId);
        if (!object)
        {
            QString from;
            if (cb.fromObject)
            {
                from = cb.fromObject->GetName();
            }
            // Cannot resolve this object id.
            CErrorRecord err;
            err.error = QObject::tr("Unresolved ObjectID: %1, Referenced from Object %1").arg(GuidUtil::ToString(objectId)).arg(from);
            err.severity = CErrorRecord::ESEVERITY_ERROR;
            err.flags = CErrorRecord::FLAG_OBJECTID;
            err.pObject = cb.fromObject;
            GetIEditor()->GetErrorReport()->ReportError(err);

            //Warning( "Cannot resolve ObjectID: %s\r\nObject with this ID was not present in loaded file.\r\nFor instance Trigger referencing another object which is not loaded in Level.",GuidUtil::ToString(objectId) );
            continue;
        }
        m_pCurrentErrorReport->SetCurrentValidatorObject(object);
        // Call callback with this object.
        if (cb.func1)
        {
            (cb.func1)(object);
        }
        if (cb.func2)
        {
            (cb.func2)(object, cb.userData);
        }
    }
    m_resolveCallbacks.clear();
    //////////////////////////////////////////////////////////////////////////

    {
        CWaitProgress wait("Creating Objects", false);
        if (m_bProgressBarEnabled)
        {
            wait.Start();
        }
        //////////////////////////////////////////////////////////////////////////
        // Serialize All Objects from XML.
        //////////////////////////////////////////////////////////////////////////
        int numObj = m_loadedObjects.size();
        for (i = 0; i < numObj; i++)
        {
            if (m_bProgressBarEnabled)
            {
                wait.Step((i * 100) / numObj);
            }

            SLoadedObjectInfo& obj = m_loadedObjects[i];
            m_pCurrentErrorReport->SetCurrentValidatorObject(obj.pObject);

            obj.pObject->CreateGameObject();
#if ENABLE_CRY_PHYSICS
            // remove collisions of hidden objects
            if (obj.pObject->CheckFlags(OBJFLAG_INVISIBLE) || obj.pObject->IsHiddenBySpec())
            {
                IPhysicalEntity* pPhEn = obj.pObject->GetCollisionEntity();
                if (pPhEn && gEnv->pPhysicalWorld)
                {
                    gEnv->pPhysicalWorld->DestroyPhysicalEntity(pPhEn, 1);
                }
            }
#endif // ENABLE_CRY_PHYSICS

            CMaterial* pMaterial = obj.pObject->GetRenderMaterial();
            CMaterialManager* pManager = GetIEditor()->GetMaterialManager();

            if (pMaterial && pMaterial->GetMatInfo() && pManager)
            {
                pManager->OnRequestMaterial(pMaterial->GetMatInfo());
            }

            // unset the current validator object because the wait Step 
            // might generate unrelated errors
            m_pCurrentErrorReport->SetCurrentValidatorObject(nullptr);
        }
        m_pCurrentErrorReport->SetCurrentValidatorObject(NULL);
        //////////////////////////////////////////////////////////////////////////
    }

    //////////////////////////////////////////////////////////////////////////
    // Call PostLoad on all these objects.
    //////////////////////////////////////////////////////////////////////////
    {
        int numObj = m_loadedObjects.size();
        for (i = 0; i < numObj; i++)
        {
            SLoadedObjectInfo& obj = m_loadedObjects[i];
            m_pCurrentErrorReport->SetCurrentValidatorObject(obj.pObject);
            node = obj.xmlNode;
            obj.pObject->PostLoad(*this);
        }
    }

    m_bNeedResolveObjects = false;
    m_pCurrentErrorReport->SetCurrentValidatorObject(NULL);
    m_sequenceIdRemap.clear();
    m_pendingIds.clear();
}

//////////////////////////////////////////////////////////////////////////
void CObjectArchive::SaveObject(CBaseObject* pObject, bool bSaveInGroupObjects, bool bSaveInPrefabObjects)
{
    // Not save objects in prefabs.
    if (!bSaveInPrefabObjects && pObject->CheckFlags(OBJFLAG_PREFAB))
    {
        return;
    }

    if (pObject->CheckFlags(OBJFLAG_DONT_SAVE))
    {
        return;
    }

    // Not save objects in group.
    if ((!bSaveInGroupObjects && pObject->GetGroup()) && !bSaveInPrefabObjects)
    {
        return;
    }

    if (m_savedObjects.find(pObject) == m_savedObjects.end())
    {
        m_pCurrentObject = pObject;
        m_savedObjects.insert(pObject);
        // If this object was not saved before.
        XmlNodeRef objNode = node->newChild("Object");
        XmlNodeRef prevRoot = node;
        node = objNode;

        if (bSaveInPrefabObjects)
        {
            m_nFlags |= eObjectLoader_SavingInPrefabLib;
        }

        pObject->Serialize(*this);
        node = prevRoot;
    }
}

//////////////////////////////////////////////////////////////////////////
CBaseObject* CObjectArchive::LoadObject(const XmlNodeRef& objNode, CBaseObject* pPrevObject)
{
    XmlNodeRef prevNode = node;
    node = objNode;
    CBaseObject* pObject;
    bool bMakeNewID = (m_nFlags & eObjectLoader_MakeNewIDs) ? true : false;

    pObject = m_objectManager->NewObject(*this, pPrevObject, bMakeNewID);
    if (pObject)
    {
        SLoadedObjectInfo obj;
        obj.nSortOrder = pObject->GetClassDesc()->GameCreationOrder();
        obj.pObject = pObject;
        obj.newGuid = pObject->GetId();
        obj.xmlNode = node;
        m_loadedObjects.push_back(obj);
        m_bNeedResolveObjects = true;
    }
    node = prevNode;
    return pObject;
}

//////////////////////////////////////////////////////////////////////////
void CObjectArchive::LoadObjects(XmlNodeRef& rootObjectsNode)
{
    int numObjects = rootObjectsNode->getChildCount();
    for (int i = 0; i < numObjects; i++)
    {
        XmlNodeRef objNode = rootObjectsNode->getChild(i);
        LoadObject(objNode, NULL);
    }
}

//////////////////////////////////////////////////////////////////////////
void CObjectArchive::ReportError(CErrorRecord& err)
{
    if (m_pCurrentErrorReport)
    {
        m_pCurrentErrorReport->ReportError(err);
    }
}

//////////////////////////////////////////////////////////////////////////
void CObjectArchive::SetErrorReport(CErrorReport* errReport)
{
    if (errReport)
    {
        m_pCurrentErrorReport = errReport;
    }
    else
    {
        m_pCurrentErrorReport = GetIEditor()->GetErrorReport();
    }
}

//////////////////////////////////////////////////////////////////////////
void CObjectArchive::ShowErrors()
{
    GetIEditor()->GetErrorReport()->Display();
}

//////////////////////////////////////////////////////////////////////////
void CObjectArchive::MakeNewIds(bool bEnable)
{
    if (bEnable)
    {
        m_nFlags |= eObjectLoader_MakeNewIDs;
    }
    else
    {
        m_nFlags &= ~(eObjectLoader_MakeNewIDs);
    }
}

//////////////////////////////////////////////////////////////////////////
void CObjectArchive::EnableReconstructPrefabObject(bool bEnable)
{
    if (bEnable)
    {
        m_nFlags |= eObjectLoader_ReconstructPrefabs;
    }
    else
    {
        m_nFlags &= ~(eObjectLoader_ReconstructPrefabs);
    }
}

//////////////////////////////////////////////////////////////////////////
void CObjectArchive::SetShouldResetInternalMembers(bool reset)
{
    if (reset)
    {
        m_nFlags |= eObjectLoader_ResetInternalMembers;
    }
    else
    {
        m_nFlags &= ~(eObjectLoader_ResetInternalMembers);
    }
}

//////////////////////////////////////////////////////////////////////////
void CObjectArchive::RemapID(REFGUID oldId, REFGUID newId)
{
    m_IdRemap[oldId] = newId;
}

//////////////////////////////////////////////////////////////////////////
CPakFile* CObjectArchive::GetGeometryPak(const char* sFilename)
{
    if (m_pGeometryPak)
    {
        return m_pGeometryPak;
    }
    m_pGeometryPak = new CPakFile;
    m_pGeometryPak->Open(sFilename);
    return m_pGeometryPak;
}

//////////////////////////////////////////////////////////////////////////
CBaseObject* CObjectArchive::GetCurrentObject()
{
    return m_pCurrentObject;
}

//////////////////////////////////////////////////////////////////////////
void CObjectArchive::AddSequenceIdMapping(uint32 oldId, uint32 newId)
{
    assert(oldId != newId);
    assert(GetIEditor()->GetMovieSystem()->FindSequenceById(oldId) || stl::find(m_pendingIds, oldId));
    assert(GetIEditor()->GetMovieSystem()->FindSequenceById(newId) == NULL);
    assert(stl::find(m_pendingIds, newId) == false);
    m_sequenceIdRemap[oldId] = newId;
    m_pendingIds.push_back(newId);
}

//////////////////////////////////////////////////////////////////////////
uint32 CObjectArchive::RemapSequenceId(uint32 id) const
{
    std::map<uint32, uint32>::const_iterator itr = m_sequenceIdRemap.find(id);
    if (itr == m_sequenceIdRemap.end())
    {
        return id;
    }
    else
    {
        return itr->second;
    }
}

//////////////////////////////////////////////////////////////////////////
bool CObjectArchive::IsAmongPendingIds(uint32 id) const
{
    return stl::find(m_pendingIds, id);
}