aisDataManager.h

Go to the documentation of this file.

/*******************************************************************************
** Copyright (c) 2022 MAK Technologies, Inc.
** All rights reserved.
*******************************************************************************/

#pragma once


#include <vrfcore/aisDataServiceInterface.h>
#include <vrfutil/localVertexList.h>

#include <atomic>
#include <tbb/spin_mutex.h>
#include <tbb/task_arena.h>
#include <memory>

#include "vrfobjcore/vrfobjcoreDefines.h"

class DtSimulationServices;
class DtEntityType;
class DtAisTrack;
class DtAisDataManagerJob;

namespace MAKVRinTerra
{
   class DtFeatureGeometry;
}

class DT_DLL_vrfobjcore DtAisDataManager : public DtAisDataServiceInterface
{
public:

   typedef std::vector<DtAisTrack> DtAisTrackBuffer;
   typedef std::list<DtAisTrack*> DtAisTrackList;
   typedef std::unordered_map<DtMmsi, DtAisTrack*> DtAisTrackMap;

   DtAisDataManager(DtSimulationServices* simServices);
   
   virtual ~DtAisDataManager() override;

   virtual bool init();

   static void tickCallback(void* usr);

   virtual void tick();

   virtual bool isDataLoaded() const override;

   //\copydoc DtAisDataServiceInterface::findTracksInArea()
   virtual DtAisDataJob::Ptr findTracksInArea(const DtVertexList& areaVertices,
      bool unownedOnly = true) override;

   //\copydoc DtAisDataServiceInterface::findTracksInAreaDuringTime()
   virtual DtAisDataJob::Ptr findTracksInAreaDuringTime(const DtVertexList & areaVertices,
      double startSimTime, double endSimTime, bool unownedOnly) override;

   //\copydoc DtAisDataServiceInterface::trackInfo()
   virtual const DtAisTrack* trackInfo(DtMmsi trackId) const override;

   //\copydoc DtAisDataServiceInterface::takeOwnershipOfTrack()
   virtual DtAisDataJob::Ptr requestOwnershipOfTracks(const DtMmsiSet& trackIds) override;

   //\copydoc DtAisDataServiceInterface::releaseOwnershipOfTrack()
   virtual DtAisDataJob::Ptr releaseOwnershipOfTracks(const DtMmsiSet& trackIds) override;

   //\copydoc DtAisDataServiceInterface::aisTimeFromSimTime()
   virtual double aisTimeFromSimTime(double simTime) const override;

   //\copydoc DtAisDataServiceInterface::simTimeFromAisTime()
   virtual double simTimeFromAisTime(double aisTime) const override;

   //\copydoc DtAisDataServiceInterface::finalDataSimTime()
   virtual double finalDataSimTime() const override;

   //\copydoc DtAisDataServiceInterface::getFirstPointAfterTime()
   virtual bool getFirstPointAfterTime(const DtAisTrack& track, double simTime,
      DtPathPointIter& pathPointIter) const override;

   //\copydoc DtAisDataServiceInterface::getLastPointBeforeTime()
   virtual bool getLastPointBeforeTime(const DtAisTrack& track, double simTime,
      DtPathPointIter& pathPointIter) const override;

   //\copydoc DtAisDataServiceInterface::getEndOfPath()
   virtual DtPathPointIter getEndOfpath(const DtAisTrack& track) const override;

   //\copydoc DtAisDataServiceInterface::resetData()
   virtual void resetData() override;

   //\copydoc DtAisDataServiceInterface::clearData()
   virtual void clearData() override;

   virtual void loadAisData();

   virtual bool isDataLoading() const;
   virtual bool timeToStop() const;
   virtual void stop();

protected:

   friend class DtAisDataManagerJob;

   virtual void performDataClear();

   virtual void performDataReset();

   virtual void startAisDataLoad();

   virtual void calculateTrackTimeOffset();

private:

   DtAisDataManager(const DtAisDataManager& orig);
   DtAisDataManager& operator=(const DtAisDataManager& other);

protected:
   
   DtAisTrackBuffer myTracks;

   DtAisTrackMap myTracksByMmsi;

   DtMmsiSet myUnusedTracks;

   DtMmsiSet myTracksInUse;

   typedef tbb::spin_mutex QueryQueueMutex;
   mutable QueryQueueMutex myQueryQueueMutex;

   typedef std::list<DtAisDataJob::Ptr> JobQueue;
   JobQueue myJobQueue;

   DtSimulationServices* mySimulationServices;
   DtTime myTrackTimeAtSimTime0;
   DtTime myFirstDataAisTime;
   DtTime myFirstDataSimTime;
   DtTime myFinalDataAisTime;
   DtTime myFinalDataSimTime;

   std::atomic<bool> myClearData;
   std::atomic<bool> myResetData;
   std::atomic<bool> myResetTimeOffset;
   std::atomic<bool> myDataIsLoaded;
   std::atomic<bool> myLoadInProgress;

   std::atomic<bool> myWarnedOnce;
   std::atomic<bool> myTimeToStop;
   tbb::task_arena myArena;
};