treesAndGraphWidget.h

Go to the documentation of this file.

/****************************************************************************
 * Copyright (c) 2017 VT MAK
 * All rights reserved.
 ****************************************************************************/


#pragma once

#include "rtiMsConfig.h"
#include <QWidget>
#include <QModelIndex>
#include <vector>
#include <set>


class QVBoxLayout;
class QHBoxLayout;
class QTreeView;
class QAbstractItemModel;
class QLabel;
class QSignalMapper;
class QMenu;
class QAction;
class QString;
class QSplitter;
class QFile;
class QToolBar;
class QTextStream;
class DtGraphDialog;

class DT_DLL_ASSISTANTAPP DtTreesAndGraphWidget : public QWidget
{
   Q_OBJECT

public:

   DtTreesAndGraphWidget(QWidget* parent = 0, Qt::WindowFlags f = 0);

   virtual ~DtTreesAndGraphWidget();

   virtual QSize sizeHint() const;

   enum DtRowNameLoc
   {
      // Indicates the row header should be used as the name
      DtRowNameIsHeader,
      // Indicates the first column should be used as the name
      DtRowNameIsFirstColumn,
      // Indicates that rows are unnamed
      DtNoRowNames,
   };

   unsigned addTree(const QString& name, const QString& description,
      QAbstractItemModel* model, DtRowNameLoc rowNameLoc, bool hideHeaders = false);
//   unsigned addTree(const QString& name, QAbstractItemModel* model,
//      DtRowNameLoc rowNameLoc, bool hideHeader = false);

   void initComplete();

   void enableValueGraphing(unsigned treeIndex, unsigned column,
      const QString& units);

   void enableRateGraphing(unsigned treeIndex, unsigned column,
      const QString& units);

   typedef std::pair<unsigned, QModelIndex> DtTreeAndModelIndex;

   void addSet(DtTreeAndModelIndex modelIndex);

   void deleteSet(DtTreeAndModelIndex modelIndex);

   void updateGraphs(double currentTime);

   void updateGraphsForTree(unsigned treeIndex, double currentTime);

   void saveGuiConfiguration(QFile& configFile);

   void loadGuiConfiguration(QFile& configFile);

   void setFederateId(const QString& federateId) { myFederateId = federateId; }

   void setFederateDescription(const QString& federateDescr) { myFederateDescription = federateDescr; }

protected:

   void initTrees();

   void connectSignals();

   void getRowAndColNames(DtTreeAndModelIndex modelIndex, QString& rowName,
      QString& colName);

   void enableDisableGraphOptions();

   enum TableFileFormat
   {
      TextTableFormat,
      CsvTableFormat
   };

   void writeTreeToStream(const QTreeView* tree, QTextStream& fileStrm,
      TableFileFormat format);

public slots:

   void deleteDialog(QWidget* graphDialog);

   void processItemSelection(const QModelIndex& index);

   void showContextMenu(const QPoint& point);

   void graphSelectedItem();

   void graphSelectedItemRate();

   void updateSortingEnable();

   void saveTreesToFile();

   void hideToolbar();
private:

   DtTreesAndGraphWidget(const DtTreesAndGraphWidget&);

   DtTreesAndGraphWidget &operator=(const DtTreesAndGraphWidget&);
protected:

   QSplitter* mySplitter;

   std::vector<QVBoxLayout*> myTreeLayouts;

   std::vector<QLabel*> myTreeLabels;

   std::vector<QHBoxLayout*> myTreeLabelLayouts;

   std::vector<QTreeView*> myTreeViews;

   std::vector<DtRowNameLoc> myTreeRowNameLocs;

   std::vector<std::map<unsigned, QString> > myTreeValGraphableCols;

   std::vector<std::map<unsigned, QString> > myTreeRateGraphableCols;


   std::map<DtTreeAndModelIndex, DtGraphDialog*> myGraphByModelIndex;
   std::map<DtGraphDialog*, DtTreeAndModelIndex> myModelIndexByGraph;


   std::map<DtTreeAndModelIndex, DtGraphDialog*> myRateGraphByModelIndex;
   std::map<DtGraphDialog*, DtTreeAndModelIndex> myModelIndexByRateGraph;

   QSignalMapper* myCloseGraphSignalMapper;


   QMenu* myContextMenu;
   QToolBar* myToolBar;
   QAction* myGraphItemAction;
   QAction* myGraphRateAction;
   QAction* mySaveToFileAction;

   DtTreeAndModelIndex myCurrentModelIndex;

   QString myFederateId;

   QString myFederateDescription;

   static const QString theTreeBreakText;
};