generatorinputproducer.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/base/memory.h"
#include "gwnavgeneration/input/clientinputconsumer.h"
#include "gwnavgeneration/generator/generatorsector.h"

// a forward declare of ClientInputConsumer would be enough but #include is handy
// to avoid the derivation to #include manually <gwnavgeneration/input/clientinputconsumer.h>
// and <gwnavgeneration/generator/generatorsector.h>

namespace Kaim
{

class GeneratorSystem;

class GeneratorInputProducer : public RefCountBaseNTS<GeneratorInputProducer, MemStat_NavDataGen>
{
public:
    GeneratorInputProducer() {}

    KyResult ProduceSectorInputs(const GeneratorSector& sector, ClientInputConsumer& sectorInputConsumer)
    {
        for (UPInt i = 0; i < sector.m_additionalTagVolumes.GetSize(); ++i)
            sectorInputConsumer.ConsumeTagVolume(sector.m_additionalTagVolumes[i]);

        for (UPInt i = 0; i < sector.m_additionalSeedPoints.GetSize(); ++i)
            sectorInputConsumer.ConsumeSeedPoint(sector.m_additionalSeedPoints[i]);

        // HeightFields & IndexedMeshes
        for (UPInt i = 0; i < sector.m_inputFileNames.GetSize(); ++i)
        {
            const String& inputFileName = sector.m_inputFileNames[i];
            const String ext = inputFileName.GetExtension();

            if (ext.CompareNoCase(".HeightField") == 0)
            {
                if (sectorInputConsumer.ConsumeHeightFieldFile(inputFileName) == KY_ERROR)
                    return KY_ERROR;
            }
            else if (ext.CompareNoCase(".IndexedMesh") == 0)
            {
                if (sectorInputConsumer.ConsumeIndexedMeshFile(inputFileName) == KY_ERROR)
                    return KY_ERROR;
            }

            // Note it may seem reasonable to support .obj file here: if (".obj") { sectorInputConsumer.ConsumeObjFile(inputFileName); }
            // The problem is that .obj are not always 1unit=1m and zup. And we consider that handling a specific coordSystem should be part
            // of the Produce() virtual function implementation. This is why one should use ObjFileInputProducer as producer and call its SetupCoordSystem() function.
        }

        if (Produce(sector, sectorInputConsumer) == KY_ERROR)
            return KY_ERROR;

        return KY_SUCCESS;
    }

private:
    virtual KyResult Produce(const GeneratorSector& sector, ClientInputConsumer& sectorInputConsumer) = 0;
};
}