/*
* 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 "CryLegacy_precompiled.h"
#include "FlyHelpers_TacticalPointLanguageExtender.h"
#include "FlyHelpers_Path.h"

#include <PipeUser.h>

namespace FlyHelpers
{
    //////////////////////////////////////////////////////////////////////////
    void CTacticalPointLanguageExtender::Initialize()
    {
        if (gEnv->pAISystem->GetTacticalPointSystem())
        {
            RegisterWithTacticalPointSystem();
            RegisterQueries();
        }
    }


    //////////////////////////////////////////////////////////////////////////
    void CTacticalPointLanguageExtender::Deinitialize()
    {
        if (gEnv->pAISystem->GetTacticalPointSystem())
        {
            UnregisterFromTacticalPointSystem();
        }
    }


    //////////////////////////////////////////////////////////////////////////
    bool CTacticalPointLanguageExtender::GeneratePoints(TGenerateParameters& parameters, SGenerateDetails& details, TObjectType obj, const Vec3& objectPos, TObjectType auxObject, const Vec3& auxObjectPos) const
    {
        const static uint32 s_pointsInDesignerPathCrc = CCrc32::Compute("pointsInDesignerPath");
        const uint32 queryNameCrc = CCrc32::Compute(parameters.query);

        if (queryNameCrc == s_pointsInDesignerPathCrc)
        {
            IF_UNLIKELY (details.fDensity <= 1.0f)
            {
                return false;
            }

            IEntity* pEntity = obj->GetEntity();
            CRY_ASSERT(pEntity);


            IAIObject* pAiObject = pEntity->GetAI();
            IF_UNLIKELY (!pAiObject)
            {
                return false;
            }

            CPipeUser* pPipeUser = pAiObject->CastToCPipeUser();
            IF_UNLIKELY (!pPipeUser)
            {
                return false;
            }

            const char* pathName = pPipeUser->GetPathToFollow();
            CRY_ASSERT(pathName);

            SShape path;
            const bool getPathSuccess = gAIEnv.pNavigation->GetDesignerPath(pathName, path);
            IF_UNLIKELY (!getPathSuccess)
            {
                return false;
            }

            const bool isValidPath = (!path.shape.empty());
            IF_UNLIKELY (!isValidPath)
            {
                return false;
            }

            // TODO: Remove unnecessary creation of this path?
            FlyHelpers::Path flyPath;
            for (size_t i = 0; i < path.shape.size(); ++i)
            {
                const Vec3& point = path.shape[ i ];
                flyPath.AddPoint(point);
            }

            // TODO: Make this an option, or get this property by other means.
            const bool shouldConsiderAsLoopingPath = true;
            if (shouldConsiderAsLoopingPath)
            {
                flyPath.MakeLooping();
            }

            const size_t segmentCount = flyPath.GetSegmentCount();
            for (size_t i = 0; i < flyPath.GetSegmentCount(); ++i)
            {
                const Lineseg segment = flyPath.GetSegment(i);
                const float length = flyPath.GetSegmentLength(i);

                CRY_ASSERT(0 < length);
                const float advance = details.fDensity / length;

                float sampled = 0;
                while (sampled < 1.0f)
                {
                    parameters.result->AddPoint(segment.GetPoint(sampled));
                    sampled += advance;
                }
            }

            return true;
        }
        return false;
    }


    //////////////////////////////////////////////////////////////////////////
    void CTacticalPointLanguageExtender::RegisterWithTacticalPointSystem()
    {
        ITacticalPointSystem* pTacticalPointSystem = gEnv->pAISystem->GetTacticalPointSystem();
        CRY_ASSERT(pTacticalPointSystem);

        const bool successfullyAddedLanguageExtender = pTacticalPointSystem->AddLanguageExtender(this);
        CRY_ASSERT_MESSAGE(successfullyAddedLanguageExtender, "Failed to add tactical point language extender.");
    }


    //////////////////////////////////////////////////////////////////////////
    void CTacticalPointLanguageExtender::RegisterQueries()
    {
        ITacticalPointSystem* pTacticalPointSystem = gEnv->pAISystem->GetTacticalPointSystem();
        CRY_ASSERT(pTacticalPointSystem);

        pTacticalPointSystem->ExtendQueryLanguage("pointsInDesignerPath", eTPQT_GENERATOR, eTPQC_IGNORE);
    }


    //////////////////////////////////////////////////////////////////////////
    void CTacticalPointLanguageExtender::UnregisterFromTacticalPointSystem()
    {
        ITacticalPointSystem* pTacticalPointSystem = gEnv->pAISystem->GetTacticalPointSystem();
        CRY_ASSERT(pTacticalPointSystem);

        const bool successfullyRemovedLanguageExtender = pTacticalPointSystem->RemoveLanguageExtender(this);
        CRY_ASSERT_MESSAGE(successfullyRemovedLanguageExtender, "Failed to remove tactical point language extender.");
    }
}