dox/Views/Infovis/vtkTreeHeatmapItem.h

Go to the documentation of this file.

/*=========================================================================

  Program:   Visualization Toolkit
  Module:    TestDiagram.cxx

  Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
  All rights reserved.
  See Copyright.txt or http://www.kitware.com/Copyright.htm for details.

     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notice for more information.

=========================================================================*/
#ifndef __vtkTreeHeatmapItem_h
#define __vtkTreeHeatmapItem_h

#include "vtkViewsInfovisModule.h" // For export macro
#include "vtkContextItem.h"

#include "vtkNew.h" // For vtkNew ivars
#include "vtkSmartPointer.h" // For vtkSmartPointer ivars
#include <vector>   // For lookup tables
#include <map>      // For string lookup tables

class vtkDoubleArray;
class vtkGraphLayout;
class vtkLookupTable;
class vtkTable;
class vtkTooltipItem;
class vtkTree;
class vtkPruneTreeFilter;

class VTKVIEWSINFOVIS_EXPORT vtkTreeHeatmapItem : public vtkContextItem
{
public:
  static vtkTreeHeatmapItem *New();
  vtkTypeMacro(vtkTreeHeatmapItem, vtkContextItem);
  virtual void PrintSelf(ostream &os, vtkIndent indent);

  virtual void SetTree(vtkTree *tree);

  vtkTree * GetTree();

  virtual void SetTable(vtkTable *table);

  vtkTable * GetTable();

  void CollapseToNumberOfLeafNodes(unsigned int n);

  // Get the collapsed tree
  vtkTree * GetPrunedTree();

  void SetTreeColorArray(const char *arrayName);

  //BTX
  virtual bool Hit(const vtkContextMouseEvent &mouse);

  virtual bool MouseMoveEvent(const vtkContextMouseEvent &event);


  virtual bool MouseDoubleClickEvent( const vtkContextMouseEvent &event);
  //ETX

protected:
  vtkTreeHeatmapItem();
  ~vtkTreeHeatmapItem();

  virtual void RebuildBuffers();

  virtual void PaintBuffers(vtkContext2D *painter);

  virtual bool IsDirty();

  void ComputeMultipliers();

  void ComputeTreeBounds();

  void InitializeLookupTables();

  virtual bool Paint(vtkContext2D *painter);

  void GenerateLookupTableForStringColumn(vtkIdType column);

  void PaintHeatmapWithoutTree(vtkContext2D *painter);

  bool SetupTextProperty(vtkContext2D *painter);

  std::string GetTooltipText(float x, float y);

  void CountLeafNodes();

  int CountLeafNodes(vtkIdType vertex);

  vtkIdType GetClosestVertex(double x, double y);

  void CollapseSubTree(vtkIdType vertex);

  void ExpandSubTree(vtkIdType vertex);

  vtkIdType GetOriginalId(vtkIdType vertex);

  vtkIdType GetPrunedIdForOriginalId(vtkIdType originalId);

  vtkIdType GetClickedCollapsedSubTree(double x, double y);

  void UpdateVisibleSceneExtent(vtkContext2D *painter);

  bool LineIsVisible(double x0, double y0, double x1, double y1);

private:
  vtkTreeHeatmapItem(const vtkTreeHeatmapItem&); // Not implemented
  void operator=(const vtkTreeHeatmapItem&); // Not implemented

  vtkSmartPointer<vtkTree> Tree;
  vtkSmartPointer<vtkTable> Table;
  vtkSmartPointer<vtkTree> PrunedTree;
  vtkSmartPointer<vtkTree> LayoutTree;
  unsigned long TreeHeatmapBuildTime;
  vtkNew<vtkGraphLayout> Layout;
  vtkNew<vtkTooltipItem> Tooltip;
  vtkNew<vtkPruneTreeFilter> PruneFilter;
  vtkNew<vtkLookupTable> TriangleLookupTable;
  vtkNew<vtkLookupTable> TreeLookupTable;
  vtkDoubleArray* TreeColorArray;
  std::vector< vtkLookupTable * > LookupTables;
  std::vector< vtkIdType > RowMap;
  double MultiplierX;
  double MultiplierY;
  int NumberOfLeafNodes;
  double CellWidth;
  double CellHeight;

  std::map< int, std::map< std::string, double> > StringToDoubleMaps;

  double HeatmapMinX;
  double HeatmapMinY;
  double HeatmapMaxX;
  double HeatmapMaxY;
  double TreeMinX;
  double TreeMinY;
  double TreeMaxX;
  double TreeMaxY;
  double SceneBottomLeft[3];
  double SceneTopRight[3];
  bool JustCollapsedOrExpanded;
  bool ColorTree;
};

#endif