include/vrvVrlQt/DtObjectsCountTreeLogic.h

Go to the documentation of this file.

/******************************************************************************
** Copyright (c) 2011 MAK Technologies, Inc.
** All rights reserved.
******************************************************************************/
/******************************************************************************
** $RCSfile:  $ $Revision:  $ $State:  $
******************************************************************************/


#pragma once
#include <vrvVrlQt/vrvVrlQt.h>
#include <vrvVrlQt/DtObjectsCountTreeInterface.h>
#include <vrvUtil/DtSignalConnectionManager.h>
#include <vrvCore/DtEntityState.h>
#include <vrvCore/DtElementAttributes.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);

      ~DtObjectsCountTreeLogic();

      void refreshWidget();

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

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

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

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

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

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

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