manualDistributionWidget.h

Go to the documentation of this file.

/*********************************************************************
** Copyright (c) 2023 MaK Technologies, Inc.
** All rights reserved.
*********************************************************************/


#pragma once

#if !defined(Q_MOC_RUN)
#include <vrfGuiCoreQt/vrfGuiCoreQt.h>
#include <QtWidgets/QDialog>
#include <vrvUtil/DtSignalConnectionManager.h>
#include <vrvCore/DtEntityState.h>
#include <vrvCore/DtForceManager.h>
#include <QtGui/QPixmap>
#include <vrfutil/rwUUID.h>
#endif

#include <vrvCore/DtSimEntry.h>
#include <vrvCore/DtStateView.h>
#include <vlpi/simulationAddress.h>

class DtAddressMap;
class DtList;
class DtVrfFileParser;
class QTreeWidget;
class QPushButton;
class QHBoxLayout;
class DtRwVariableBindings;
class QComboBox;
class QTreeWidgetItem;
class QLabel;
class QCheckBox;

namespace makVrv
{
   class DtDe;
}

namespace makVrf
{
   class DtVrfBackendState;
   class DtObjectListViewItem;
   class DtLoadBalancingExtensionWidget;
   class DtLoadBalancerSelectionWidget;
   class DtSearchWidget;

   class DT_DLL_VRFGUICOREQT DtManualDistributionWidget : public QDialog
   {

      Q_OBJECT
         Q_PROPERTY(QColor backendColor1 MEMBER myBackendColor1 USER true)
         Q_PROPERTY(QColor backendColor2 MEMBER myBackendColor2 USER true)
         Q_PROPERTY(QColor backendColor3 MEMBER myBackendColor3 USER true)
         Q_PROPERTY(QColor backendColor4 MEMBER myBackendColor4 USER true)
         Q_PROPERTY(QColor backendColor5 MEMBER myBackendColor5 USER true)
         Q_PROPERTY(QColor backendColor6 MEMBER myBackendColor6 USER true)
         Q_PROPERTY(QColor backendColor7 MEMBER myBackendColor7 USER true)
         Q_PROPERTY(QColor backendColor8 MEMBER myBackendColor8 USER true)
         Q_PROPERTY(QColor backendColor9 MEMBER myBackendColor9 USER true)

   public:

      DtManualDistributionWidget(makVrv::DtDe& de, QWidget* parent = NULL, const Qt::WindowFlags& flags =  Qt::Tool);

      virtual ~DtManualDistributionWidget();

      virtual bool init(DtAddressMap* map, const DtVrfFileParser* parser, const DtList* backends);

      virtual void setBackendColors(const QVariantList&);
      virtual QVariantList getBackendColors() const;

   protected slots:
      virtual void applyLoadBalance();
      virtual void chooseBackend(QTreeWidgetItem*, QMouseEvent*, int);
      virtual void revertBackends();
      virtual void executeTextFilter();
      virtual void changeObjectView();

   protected:
      virtual makVrv::DtVrlinkSimulatedBaseState* stateFromBindings(const DtRwVariableBindings&) const;
      virtual void loadBalancingValidityChanged();
      virtual bool itemHasSubordinates(QTreeWidgetItem*) const;

      virtual void initConnections();
      virtual void initObjectsList(DtAddressMap* map, const DtVrfFileParser* parser);

      virtual void slot_backendDiscovered(const DtVrfBackendState&);
      virtual void slot_backendRemoved(const DtVrfBackendState&);
      virtual QVariant entityType(const DtObjectListViewItem*) const;
      virtual bool sortByEntityType(const DtObjectListViewItem& lhs, const DtObjectListViewItem& rhs) const;
      virtual bool sortByForceType(const DtObjectListViewItem& lhs, const DtObjectListViewItem& rhs) const;
      virtual bool simulationAddressSort(const DtObjectListViewItem& lhs, const DtObjectListViewItem& rhs) const;
      virtual QVariant parentName(const DtObjectListViewItem*) const;
      virtual void setItemToAddress(DtObjectListViewItem* item, const QString& address);
      virtual void setAllChildrenToAddress(DtObjectListViewItem* item, const QString& address);

      virtual void accept() override;
      virtual void reject() override;

      virtual void initBackendLoads();
      virtual void updateBackendLoads();
      virtual void setBackendForSubordinatesOf(const DtUUID&, const QString& address);
      virtual void initControls();
      virtual void initTrees();
      virtual const QPixmap& colorPixmapForAddress(const DtSimulationAddress&);
      virtual void initTreeAsFlatList();
      virtual void initTreeAsHierarchy();
      virtual void clearForceNodes();
      virtual DtObjectListViewItem* createForceNode(int force);

      virtual void reworkForceManager(const DtVrfFileParser*);
      virtual void restoreForceManager();
      virtual void processForceHostilityFile(const DtFilename&);

   public:
      QTreeWidget* myObjects;
      QHBoxLayout* myButtonLayout;
      QPushButton* myOKButton;
      QPushButton* myCancelButton;

      QTreeWidget* myBackendLoad;
      QLabel*      myBackendLoadLabel;

      DtLoadBalancerSelectionWidget* myLoadBalancerSelectionWidget;
      QHBoxLayout* myLoadBalancingLayout;
      QPushButton* myApplyLoadBalancingOption;
      QPushButton* myRevertLoadBalancingOption;
      DtSearchWidget* mySearchWidget;
      QCheckBox* myShowAsHierarchy = nullptr;
      QHBoxLayout* mySearchLayout = nullptr;

   protected:
      makVrv::DtDe&  myDe;
      makVrv::DtSignalConnectionManager myConnectionManager;
      int myObjectViewId = 0;
      const DtVrfFileParser* myParser = nullptr;
      DtAddressMap* myObjectMap = nullptr;
      QString myRightClickChosenBackend;
      std::set<std::string> myInitialMappedBackends;
      QColor myBackendColor1;
      QColor myBackendColor2;
      QColor myBackendColor3;
      QColor myBackendColor4;
      QColor myBackendColor5;
      QColor myBackendColor6;
      QColor myBackendColor7;
      QColor myBackendColor8;
      QColor myBackendColor9;

      struct ObjectStateData
      {
         ObjectStateData() {};
         ObjectStateData(makVrv::DtStateView<makVrv::DtVrlinkSimulatedBaseState>* v, makVrv::DtSimEntry* e, DtObjectListViewItem* i)
            : view(v), entry(e), item(i) {}

         makVrv::DtStateView<makVrv::DtVrlinkSimulatedBaseState>* view = nullptr;
         makVrv::DtSimEntry* entry = nullptr;
         DtObjectListViewItem* item = nullptr;
      };

      typedef std::map<makVrv::DtUniqueID, ObjectStateData> ObjectStateDataMap;
      ObjectStateDataMap myObjectStateData;
      ObjectStateDataMap myForceNodes;

      typedef std::map<DtUUID, ObjectStateData> ObjectStateDataMapByUUID;
      ObjectStateDataMapByUUID myObjectStateDataByUUID;

      typedef std::map<DtSimulationAddress, QPixmap> ColorByAddress;
      ColorByAddress myColorByAddress;

      QVariantList myBackendColors;

      struct ForceInfo
      {
         std::string       myForceName;
         makVrv::DtForceManager::RGBAColor         myForceColor;
         makVrv::DtForceManager::RGBAColor         myForcePrimaryColor;
         bool              myIncludeInDialogs;
      };

      typedef std::map<int, ForceInfo> ForceInfoMap;
      ForceInfoMap myForceInfoMap;
   };

   class DT_DLL_VRFGUICOREQT DtManualDistributionWidgetCreator
   {
   public:
      DtManualDistributionWidgetCreator(makVrv::DtDe&);

      virtual DtManualDistributionWidget* create(QWidget* parent = 0, const Qt::WindowFlags& flags =  Qt::Tool);

   protected:
      makVrv::DtDe&  myDe;
   };
}