orbatSelectorPalette.h

Go to the documentation of this file.

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



#pragma once

#include <vrvUtil/signalslib.h>
#include <vrvUtil/DtVirtualBaseClass.h>

#include <vrfGuiCoreQt/vrfGuiCoreQt.h>
#include <vrfGuiCore/vrfGuiOrbatManager.h>
#include <vrvCore/DtEntityState.h>
#include <vrvCore/DtStateView.h>
#include <vrvUtil/DtSignalConnectionManager.h>

#include <QtWidgets/QDialog>
#include <QtWidgets/QTreeWidget>

namespace makVrv
{
   class DtDe;
}

class DtRwOrbatData;
class QLabel;
class QHBoxLayout;
class QPushButton;

#define DtVrlCreateOrbatObjectFromPalette "text/x-vrlCreateOrbatObjectFromPalette@Type"

namespace makVrf
{
   class DtVrfObjectListViewItem;

   class DT_DLL_VRFGUICOREQT DtOrbatTreeWidget : public QTreeWidget
   {
   public:
      DtOrbatTreeWidget(makVrv::DtDe&, QWidget*);

      virtual ~DtOrbatTreeWidget();

      virtual void addChildForCopyToVector(DtVrfObjectListViewItem*, QStringList& nodes, std::set<DtUUID>& processed,
         bool consideredUnparented = false) const;

   protected:
      virtual void startDrag(Qt::DropActions supportedActions);

      virtual QMimeData* mimeData(const QList<QTreeWidgetItem*> items) const;

      virtual bool atLeastOneNotInOrbatView(DtVrfObjectListViewItem*) const;

   protected:
      makVrv::DtDe& myDe;
   };

   class DT_DLL_VRFGUICOREQT DtOrbatSelectorPalette : public QDialog
   {
      Q_OBJECT

   public:

      struct ItemData
      {
         makVrv::DtUniqueID id;
         makVrv::DtVrlinkSimulatedEntityState* state;
         DtVrfObjectListViewItem* item;
      };

   public:
      DtOrbatSelectorPalette( makVrv::DtDe& de, QWidget* parent = 0, const Qt::WindowFlags& flags = Qt::Tool);

      virtual ~DtOrbatSelectorPalette();

      virtual void setupFrom(const DtVrfGuiOrbatManager::OrbatTreeWithRwOrbatData&);

      virtual void clear();

      virtual void setFilename(const DtFilename&);

   protected:
      virtual void setupUi();

      virtual DtOrbatTreeWidget* createOrbatTreeWidget();

      virtual void addItems(const DtVrfGuiOrbatManager::OrbatTreeWithRwOrbatData&, QTreeWidgetItem* parent);

      virtual bool sortByOrbatOrder(const DtVrfObjectListViewItem& lhs, const DtVrfObjectListViewItem& rhs) const;
      virtual QVariant textColorForOrbatItem(const DtVrfObjectListViewItem* item) const;

      virtual void orbatChanged();

      virtual void filterItems();

      virtual void setupForceRootNodes();

      virtual ItemData createItem(const DtRwOrbatData& data, const DtString& keyword);

      virtual void nullChildren(DtVrfObjectListViewItem*);

   protected slots:
      virtual void onImportAll();

   public:
      DtOrbatTreeWidget* myOrbatTreeWidget;
      QLabel* myCreateHelpText;
      QPushButton* myImportAllButton;
      QPushButton* myCloseButton;
      QHBoxLayout* myButtonLayout;

   protected:
      makVrv::DtDe& myDe;
      DtFilename myFilename;

      typedef std::map<DtUUID, ItemData> Items;
      Items myItems;

      makVrv::DtSignalConnectionManager myConnections;

      typedef std::map<int, ItemData> ForceRootItems;

      ForceRootItems  myForceRootItems;
   };

   class DT_DLL_VRFGUICOREQT DtOrbatSelectorPaletteCreator : public makVrv::DtVirtualBaseClass
   {
   public:
      DtOrbatSelectorPaletteCreator(makVrv::DtDe&);

      virtual ~DtOrbatSelectorPaletteCreator();

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

      static DtOrbatSelectorPaletteCreator& instance(makVrv::DtDe& de);
      static void registerInstance(makVrv::DtDe& de, DtOrbatSelectorPaletteCreator*);

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