/* Copyright (C) 2015
 * swift project Community / Contributors
 *
 * This file is part of swift project. It is subject to the license terms in the LICENSE file found in the top-level
 * directory of this distribution and at http://www.swift-project.org/license.html. No part of swift project,
 * including this file, may be copied, modified, propagated, or distributed except according to the terms
 * contained in the LICENSE file.
 */

#include "blackmisc/simulation/aircraftmodelloader.h"
#include "blackmisc/simulation/fscommon/aircraftcfgparser.h"
#include "blackmisc/simulation/xplane/aircraftmodelloaderxplane.h"
#include "blackmisc/simulation/xplane/xplaneutil.h"
#include "blackmisc/directoryutils.h"
#include "blackmisc/compare.h"
#include "blackmisc/logmessage.h"

#include <QDir>
#include <Qt>
#include <QtGlobal>
#include <QMap>

using namespace BlackMisc;
using namespace BlackMisc::Simulation::Data;
using namespace BlackMisc::Simulation::Settings;
using namespace BlackMisc::Simulation::FsCommon;
using namespace BlackMisc::Simulation::XPlane;

namespace BlackMisc
{
    namespace Simulation
    {
        const CLogCategoryList &IAircraftModelLoader::getLogCategories()
        {
            static const CLogCategoryList cats({ CLogCategory::modelLoader() });
            return cats;
        }

        const QString &IAircraftModelLoader::enumToString(IAircraftModelLoader::LoadFinishedInfo info)
        {
            static const QString loaded("cache loaded");
            static const QString skipped("loading skipped");
            static const QString parsed("parsed data");

            switch (info)
            {
            case CacheLoaded: return loaded;
            case LoadingSkipped: return skipped;
            case ParsedData: return parsed;
            default: break;
            }

            static const QString unknown("??");
            return unknown;
        }

        const QString &IAircraftModelLoader::enumToString(IAircraftModelLoader::LoadModeFlag modeFlag)
        {
            static const QString notSet("not set");
            static const QString directly("load directly");
            static const QString background("load in background");
            static const QString cacheFirst("cache first");
            static const QString cacheSkipped("cache skipped");
            static const QString cacheOnly("cacheOnly");

            switch (modeFlag)
            {
            case NotSet: return notSet;
            case LoadDirectly: return directly;
            case LoadInBackground: return background;
            case CacheFirst: return cacheFirst;
            case CacheSkipped: return cacheSkipped;
            case CacheOnly: return cacheOnly;
            default: break;
            }

            static const QString unknown("??");
            return unknown;
        }

        QString IAircraftModelLoader::enumToString(LoadMode mode)
        {
            QStringList modes;
            if (mode.testFlag(NotSet)) modes << enumToString(NotSet);
            if (mode.testFlag(LoadDirectly)) modes << enumToString(LoadDirectly);
            if (mode.testFlag(LoadInBackground)) modes << enumToString(LoadInBackground);
            if (mode.testFlag(CacheFirst)) modes << enumToString(CacheFirst);
            if (mode.testFlag(CacheSkipped)) modes << enumToString(CacheSkipped);
            return modes.join(", ");
        }

        bool IAircraftModelLoader::needsCacheSynchronized(LoadMode mode)
        {
            return mode.testFlag(CacheFirst) || mode.testFlag(CacheOnly);
        }

        IAircraftModelLoader::IAircraftModelLoader(const CSimulatorInfo &simulator, QObject *parent) :
            CCentralMultiSimulatorModelCachesProvider(parent),
            m_simulator(simulator)
        {
            Q_ASSERT_X(simulator.isSingleSimulator(), Q_FUNC_INFO, "Only one simulator per loader");
            connect(this, &IAircraftModelLoader::loadingFinished, this, &IAircraftModelLoader::onLoadingFinished, Qt::QueuedConnection);
        }

        IAircraftModelLoader::~IAircraftModelLoader() { }

        void IAircraftModelLoader::startLoading(LoadMode mode, const IAircraftModelLoader::ModelConsolidationCallback &modelConsolidation, const QStringList &modelDirectories)
        {
            if (m_loadingInProgress) { return; }
            if (mode == NotSet) { return; }
            m_loadingInProgress = true;
            m_loadingMessages.clear();

            const CSimulatorInfo simulator = this->getSimulator();
            const bool needsCacheSynced = IAircraftModelLoader::needsCacheSynchronized(mode);
            if (needsCacheSynced) { this->synchronizeCache(simulator); }

            const bool useCachedData = !mode.testFlag(CacheSkipped) && this->hasCachedData();
            if (useCachedData && (mode.testFlag(CacheFirst) || mode.testFlag(CacheOnly)))
            {
                // we just just cache data
                static const CStatusMessage status(this, CStatusMessage::SeverityInfo, "Using cached data");
                emit this->loadingFinished(status, simulator, CacheLoaded);
                return;
            }
            if (mode.testFlag(CacheOnly))
            {
                // only cache, but we did not find any data yet (still in progress?)
                // here we rely on the cache load slot, no need to emit here, will
                // be done later in ps_cacheChanged. An alternative was to sync cache here
                m_loadingInProgress = false;
                return;
            }

            // really load from disk?
            const QStringList modelDirs = this->getInitializedModelDirectories(modelDirectories, simulator);
            if (m_skipLoadingEmptyModelDir && modelDirs.isEmpty())
            {
                const CStatusMessage status = CStatusMessage(this, CStatusMessage::SeverityWarning,
                                              "Empty or not existing '%1' directory '%2', skipping read")
                                              << simulator.toQString() << modelDirectories.join(", ");
                m_loadingMessages.push_back(status);
                emit this->loadingFinished(m_loadingMessages, simulator, LoadingSkipped);
                return;
            }

            this->setObjectInfo(simulator);
            this->startLoadingFromDisk(mode, modelConsolidation, modelDirs);
        }

        QString IAircraftModelLoader::getFirstModelDirectoryOrDefault() const
        {
            const QString md = m_settings.getFirstModelDirectoryOrDefault(m_simulator);
            return md;
        }

        void IAircraftModelLoader::setModels(const CAircraftModelList &models)
        {
            this->setModelsForSimulator(models, m_simulator);
        }

        int IAircraftModelLoader::updateModels(const CAircraftModelList &models)
        {
            return this->updateModelsForSimulator(models, m_simulator);
        }

        IAircraftModelLoader *IAircraftModelLoader::createModelLoader(const CSimulatorInfo &simulator, QObject *parent)
        {
            Q_ASSERT_X(simulator.isSingleSimulator(), Q_FUNC_INFO, "Single simulator");
            if (simulator.isXPlane()) { return new CAircraftModelLoaderXPlane(parent); }
            return CAircraftCfgParser::createModelLoader(simulator, parent);
        }

        QStringList IAircraftModelLoader::getInitializedModelDirectories(const QStringList &modelDirectories, const CSimulatorInfo &simulator) const
        {
            QStringList modelDirs = modelDirectories.isEmpty() ? m_settings.getModelDirectoriesOrDefault(simulator) : modelDirectories;
            modelDirs = CFileUtils::fixWindowsUncPaths(modelDirs);
            return CDirectoryUtils::getExistingUnemptyDirectories(modelDirs);
        }

        bool IAircraftModelLoader::hasCachedData() const
        {
            return !this->getCachedModels(m_simulator).isEmpty();
        }

        void IAircraftModelLoader::setObjectInfo(const CSimulatorInfo &simulatorInfo)
        {
            this->setObjectName("Model loader for: '" + simulatorInfo.toQString(true) + "'");
        }

        void IAircraftModelLoader::onLoadingFinished(const CStatusMessageList &statusMsgs, const CSimulatorInfo &simulator, IAircraftModelLoader::LoadFinishedInfo info)
        {
            Q_UNUSED(info);
            if (!this->supportsSimulator(simulator)) { return; } // none of my business
            this->setObjectInfo(simulator);

            // remark: in the past status used to be bool, now it is CStatusMessage
            // so there is some redundancy here between status and m_loadingMessages
            m_loadingInProgress = false;

            const QMap<int, int> counts = statusMsgs.countSeverities();
            const int errors = counts.value(SeverityError);
            const int warnings = counts.value(SeverityWarning);

            if (statusMsgs.hasWarningOrErrorMessages())
            {
                CLogMessage(this).log(m_loadingMessages.worstSeverity(),
                                      "Message loading produced %1 error and %2 warning messages") << errors << warnings;
            }
            else
            {
                CLogMessage(this).info("Loading finished, success for '%1'") << simulator.toQString();
            }
        }

        IAircraftModelLoader *CMultiAircraftModelLoaderProvider::loaderInstance(const CSimulatorInfo &simulator)
        {
            Q_ASSERT_X(simulator.isSingleSimulator(), Q_FUNC_INFO, "No single simulator");
            switch (simulator.getSimulator())
            {
            case CSimulatorInfo::FSX:
                {
                    if (!m_loaderFsx) { m_loaderFsx = this->initLoader(CSimulatorInfo::fsx()); }
                    return m_loaderFsx;
                }
            case CSimulatorInfo::P3D:
                {
                    if (!m_loaderP3D) { m_loaderP3D = this->initLoader(CSimulatorInfo::p3d()); }
                    return m_loaderP3D;
                }
            case CSimulatorInfo::XPLANE:
                {
                    if (!m_loaderXP) { m_loaderXP = this->initLoader(CSimulatorInfo::xplane()); }
                    return m_loaderXP;
                }
            case CSimulatorInfo::FS9:
                {
                    if (!m_loaderFS9) { m_loaderFS9 = this->initLoader(CSimulatorInfo::fs9()); }
                    return m_loaderFS9;
                }
            default:
                Q_ASSERT_X(false, Q_FUNC_INFO, "Wrong simulator");
                break;
            }
            return nullptr;
        }

        CMultiAircraftModelLoaderProvider &CMultiAircraftModelLoaderProvider::multiModelLoaderInstance()
        {
            static CMultiAircraftModelLoaderProvider loader;
            return loader;
        }

        IAircraftModelLoader *CMultiAircraftModelLoaderProvider::initLoader(const CSimulatorInfo &simulator)
        {
            IAircraftModelLoader *loader = IAircraftModelLoader::createModelLoader(simulator, this);
            connect(loader, &IAircraftModelLoader::loadingFinished, this, &CMultiAircraftModelLoaderProvider::loadingFinished);
            connect(loader, &IAircraftModelLoader::diskLoadingStarted, this, &CMultiAircraftModelLoaderProvider::diskLoadingStarted);
            return loader;
        }
    } // ns
} // ns