/*
 * Copyright (c) Contributors to the Open 3D Engine Project.
 * For complete copyright and license terms please see the LICENSE at the root of this distribution.
 *
 * SPDX-License-Identifier: Apache-2.0 OR MIT
 *
 */


#pragma once
#ifndef CRYINCLUDE_EDITOR_PLUGIN_H
#define CRYINCLUDE_EDITOR_PLUGIN_H
#include "Include/IEditorClassFactory.h"

#include "Util/GuidUtil.h"
#include <map>

//! Class factory is a common repository of all registered plugin classes,
//! Classes here can found by their class ID or all classes of given system class retrieved
class CRYEDIT_API CClassFactory
    : public IEditorClassFactory
{
public:
    CClassFactory();
    ~CClassFactory();

    //! Access class factory singleton.
    static CClassFactory* Instance();
    //! Register a new class to the factory
    void RegisterClass(IClassDesc* pClassDesc);
    //! Find class in the factory by class name
    IClassDesc* FindClass(const char* className) const;
    //! Find class in the factory by class ID
    IClassDesc* FindClass(const GUID& rClassID) const;
    void UnregisterClass(const char* pClassName);
    void UnregisterClass(const GUID& rClassID);
    //! Get classes matching specific requirements ordered alphabetically by name.
    void GetClassesBySystemID(ESystemClassID aSystemClassID, std::vector<IClassDesc*>& rOutClasses);
    void GetClassesByCategory(const char* pCategory, std::vector<IClassDesc*>& rOutClasses);

private:
    void RegisterAutoTypes();

    typedef std::map<QString, IClassDesc*> TNameMap;
    typedef std::map<GUID, IClassDesc*, guid_less_predicate> TGuidMap;

    AZ_PUSH_DISABLE_DLL_EXPORT_MEMBER_WARNING
    TNameMap m_nameToClass;
    TGuidMap m_guidToClass;
    std::vector<IClassDesc*> m_classes;
    AZ_POP_DISABLE_DLL_EXPORT_MEMBER_WARNING

#if defined(DEBUG_CLASS_NAME_REGISTRATION)
    // This vector will mirror that of CClassFactory::m_classes.
    // When a class description is destroyed without being unregistered first
    // (i.e. from within plugins is a common scenario) you will get a crash
    // in ~CClassFactory as it's Releasing m_classes.
    // When you hit that crash, take the index and look into this vector at
    // the class name.  That class did not unregister before it was released.
    std::vector<string> m_debugClassNames;
#endif // DEBUG_CLASS_NAME_REGISTRATION

    static CClassFactory* s_pInstance;
};

//! Auto registration for classes
class CAutoRegisterClassHelper
{
public:
    CAutoRegisterClassHelper(IClassDesc* pClassDesc)
    {
        m_pClassDesc = pClassDesc;
        m_pNext = 0;

        if (!s_pLast)
        {
            s_pFirst = this;
        }
        else
        {
            s_pLast->m_pNext = this;
        }

        s_pLast = this;
    }

    IClassDesc* m_pClassDesc;
    CAutoRegisterClassHelper* m_pNext;
    static CAutoRegisterClassHelper* s_pFirst;
    static CAutoRegisterClassHelper* s_pLast;
};

// Use this define to automatically register a new class description.
#define REGISTER_CLASS_DESC(ClassDesc) \
    CAutoRegisterClassHelper g_AutoRegHelper##ClassDesc(new ClassDesc);

#endif // CRYINCLUDE_EDITOR_PLUGIN_H
