DtObjectsCountTreeLogic.h

Go to the documentation of this file.

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


#pragma once
#include <vrvVrlQt/vrvVrlQt.h>
#include <vrvVrlQt/DtObjectsCountTreeInterface.h>
#include <vrvUtil/DtSignalConnectionManager.h>
#include <vrvCore/DtEntityState.h>
#include <vrvCore/DtElementAttributeManager.h>
#include <vector>

namespace makVrv
{
   class DT_DLL_VRVVRLQT DtObjectsCountTreeLogic
   {
   public:
      typedef std::pair<std::string, DtObjectsCountTreeInterface::TacticalGraphicType> TacticalTypePair;
      typedef std::vector<TacticalTypePair> TacticalTypePairList;
      typedef std::vector<int> EntityIDList;

   public:
      DtObjectsCountTreeLogic(DtDe& de, DtObjectsCountTreeInterface* iface);

      virtual ~DtObjectsCountTreeLogic();

      DtObjectsCountTreeLogic() = delete;

      DtObjectsCountTreeLogic(const DtObjectsCountTreeLogic&) = delete;

      DtObjectsCountTreeLogic& operator=(const DtObjectsCountTreeLogic&) = delete;

   public:
      virtual void refreshCounts();

   protected:
      virtual void slot_countEntitiesAdded(const DtElementData::ElementList& elementList);
      virtual void slot_countEntitiesRemoved(const DtElementData::UniqueIdList& elementIdList);
      virtual void slot_countDestroyedAndTacticals(const DtElementAttributeManager::EntryAttributesList& attributeList);
      virtual void slot_tacticalGraphicRenamed( const DtUniqueID& tacticalId, const std::string& overlay);
      virtual void slot_tacticalGraphicRemoved( const DtUniqueID& tacticalId);

      virtual void countOneEntity(const DtElementEntry* entry, const bool subtract = false);
      virtual void countOneDestroyed(const DtElementEntry* entry, const DtElementAttributeDamageState* damageState);
      virtual void countOneTactical(const DtElementAttributeDisplayName* dispName, 
         const DtElementAttributeKindAndCategory* kindCat, bool subtract = false);

      virtual void rememberEntityID(int entityID);
      virtual void forgetEntityID(int entityID);
      virtual bool knownEntityID(int entityID);

      virtual void rememberDestroyedID(int destroyedID);
      virtual void forgetDestroyedID(int destroyedID);
      virtual bool knownDestroyedID(int destroyedID);

      virtual void rememberTacticalData(std::string dispName, DtObjectsCountTreeInterface::TacticalGraphicType tacticalType);
      virtual void forgetTacticalData(const std::string& dispName);
      virtual bool knownTacticalData(const std::string& dispName);
      virtual DtObjectsCountTreeInterface::TacticalGraphicType findTacticalDataType(const std::string& dispName);

      virtual DtObjectsCountTreeInterface::TacticalGraphicType tacticalType(std::pair<int,int> kindCat);

      virtual void slot_postInitialize();
      virtual void updateEntityStats();
      virtual void updateTacticalStats();

   protected:
      DtDe& myDe;
      DtObjectsCountTreeInterface* myWidget;
      DtSignalConnectionManager myManager;
      EntityIDList entityIDs;
      EntityIDList destroyedIDs;
      TacticalTypePairList tacticalData;

      //Cached counts for the objects
      int myLiveEntityCount;
      int myDestroyedEntityCount;

      int myLiveAggregateCount;
      int myDestroyedAggregateCount;

      int myWaypointCount;
      int myRouteCount;
      int myAreasCount;
      int myObstaclesCount;
      int myPhaselinesCount;
   };
}