DataSetSystem.h

#pragma once
 
#include "ExtensionRegistry.h"
#include "Systems/ComponentSystem.h"
 
#include "../Components/DataSetComponent.h"
 
namespace Cogs
{
  namespace Core
  {
    namespace EchoSounder
    {
      struct PingMetaData
      {
        glm::quat vesselOrientationGlobal;
        glm::vec3 vesselPositionGlobal;
        glm::quat arrayOrientationVessel;
        glm::vec3 arrayPositionVessel;
        glm::quat arrayOrientationGlobal;
        glm::vec3 arrayPositionGlobal;
      };
 
      struct DataSetBuffers
      {
        uint32_t sliceBegin = 0;
        uint32_t sliceCount = 0;
        Memory::TypedBuffer<int64_t> timestamps;
        Memory::TypedBuffer<float> values;
        Memory::TypedBuffer<float> bottomDepths;
        Memory::TypedBuffer<float> bottomReflectivities;
        Memory::TypedBuffer<PingMetaData> metaPings;
        Memory::TypedBuffer<float> distances;
      };
 
      // Data used by re-sampling task, must be kept alive as long
      // as task is running regardless of component gets removed or not.
      struct DataSetPersistent
      {
        // Current data
        Config config;
        DataSetBuffers buffer;
 
        // New data if dataReady is true.
        Config configNew;
        DataSetBuffers bufferNew;
 
        bool dataReady = false;
        bool configReady = false;
        bool configValid = false;
        std::atomic<int> runningTasks = 0;
      };
      
      struct DataSetData
      {
        DataSetData()
        {
          persistent.reset(new DataSetPersistent);
        }
 
        std::shared_ptr<DataSetPersistent> persistent;
 
        uint32_t configGen = 0;
        uint32_t dataGen = 0; //increased when data is changed
        uint32_t clearGen = 0; //increased when data is cleared
        uint32_t slicePreservedBegin = 0;
        uint32_t slicePreservedCount = 0;
      };
 
      struct DataSetSystem
        : public ComponentSystemWithDataPool<DataSetComponent, DataSetData>
      {
      public:
        DataSetSystem(Memory::Allocator * allocator, SizeType capacity) : ComponentSystemWithDataPool(allocator, capacity) {}
 
        void update(Context * context) override;
 
      private:
        bool handleConfigTransition(Context* context, DataSetData& dataData);
        void handleNewPing(DataSetData& dataData, Ping& ping);
        void resizeBuffers(DataSetData& dataData);
 
        void updateGridAngles(DataSetData& dataData);
      };
 
    }
  }
}