spatializedpointcollectorinaabbquery.h

Go to the documentation of this file.

/*
* Copyright 2016 Autodesk, Inc. All rights reserved.
* Use of this software is subject to the terms of the Autodesk license agreement and any attachments or Appendices thereto provided at the time of installation or download,
* or which otherwise accompanies this software in either electronic or hard copy form, or which is signed by you and accepted by Autodesk.
*/

#pragma once


#include "gwnavruntime/querysystem/iquery.h"
#include "gwnavruntime/queries/querydynamicoutput.h"
#include "gwnavruntime/queries/utils/basespatializedpointcollectorinaabbquery.h"
#include "gwnavruntime/queries/utils/spatializedpointcollectorcontext.h"
#include "gwnavruntime/spatialization/spatializedpoint.h"
#include "gwnavruntime/database/database.h"
#include "gwnavruntime/database/querystatistics.h"



namespace Kaim
{

class SpatializedPointCollectorFilter_CollectAll
{
public:
    bool ShouldCollectSpatializedPoint(SpatializedPoint* /*spatializedPoint*/) { return true; }
};

class SpatializedPointCollectorFilter_SelectiveCollect
{
public:
    SpatializedPointCollectorFilter_SelectiveCollect()
    {
        for (KyUInt32 i = 0; i < SpatializedPointObjectType_Count; ++i)
            m_selection[i] = false;
    }

    void Select(SpatializedPointObjectType type) { m_selection[type] = true; }

    bool ShouldCollectSpatializedPoint(SpatializedPoint* spatializedPoint)
    {
        return m_selection[spatializedPoint->GetObjectType()];
    }

public:
    bool m_selection[SpatializedPointObjectType_Count];
};

template <class SpatializedPointCollectorFilter = SpatializedPointCollectorFilter_CollectAll>
class SpatializedPointCollectorInAABBQuery : public BaseSpatializedPointCollectorInAABBQuery
{
public:
    // ------------------------------ Functions -----------------------------

    SpatializedPointCollectorInAABBQuery();
    SpatializedPointCollectorInAABBQuery(const SpatializedPointCollectorFilter& collectorFilter);
    virtual ~SpatializedPointCollectorInAABBQuery() {}

    // ---------------------------------- Functions to set up the query ----------------------------------

    void BindToDatabase(Database* database);

    void Initialize(const Vec3f& startPos, const Box3f& extentBox);

    // ---------------------------------- write accessors for query inputs ----------------------------------

    void SetStartTrianglePtr(const NavTrianglePtr& startTrianglePtr);

    void SetQueryDynamicOutput(QueryDynamicOutput* queryDynamicOutput);

    // ---------------------------------- write accessors for query parameters ----------------------------------

    void SetPositionSpatializationRange(const PositionSpatializationRange& positionSpatializationRange);

    // ---------------------------------- Query framework functions ----------------------------------

    virtual void Advance(WorkingMemory* workingMemory);

    void PerformQuery(WorkingMemory* workingMemory = nullptr);

    // ------------------------------- Accessors ----------------------------------

    SpatializedPointCollectorInAABBQueryResult GetResult()                 const;
    const Vec3f&                               GetStartPos()               const;
    const Box3f&                               GetExtentBox()              const;
    const NavTrianglePtr&                      GetStartTrianglePtr()       const;
    QueryDynamicOutput*                        GetQueryDynamicOutput()     const;

    const PositionSpatializationRange& GetPositionSpatializationRange() const;

public:
    void PerformQueryWithInputCoordPos(WorkingMemory* workingMemory);
    void SetStartIntegerPos(const WorldIntegerPos& startIntegerPos);
    const WorldIntegerPos& GetStartIntegerPos() const;

private:
    void SetResult(SpatializedPointCollectorInAABBQueryResult result);

    void RunCollectorTraversal(SpatializedPointCollectorContext* collectorContext, ScopedDynamicOutput& scopedDynamicOutput);
    KyResult CollectInNavFloor(ScopedDynamicOutput& scopedDynamicOutput, const NavFloorRawPtr& navFloorRawPtr);
    KyResult CollectInNavGraphEdge(ScopedDynamicOutput& scopedDynamicOutput, const NavGraphEdgeRawPtr& navGraphEdgeRawPtr);
    KyResult ProcessNavFloorNode(SpatializedPointCollectorContext* collectorContext, ScopedDynamicOutput& scopedDynamicOutput, const NavFloorRawPtr& navFloorRawPtr);
    KyResult ProcessNavGraphEdgeNode(SpatializedPointCollectorContext* collectorContext, ScopedDynamicOutput& scopedDynamicOutput, const NavGraphEdgeRawPtr& navGraphEdgeRawPtr);
public:
    SpatializedPointCollectorFilter m_spatializedPointCollectorFilter;
};

}

#include "gwnavruntime/queries/spatializedpointcollectorinaabbquery.inl"