vtkTanglegramItem.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 vtkTanglegramItem_h
#define vtkTanglegramItem_h

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

#include "vtkContextItem.h"
#include "vtkSmartPointer.h"       // For SmartPointer ivars
#include "vtkTable.h"  // For get/set

class vtkDendrogramItem;
class vtkLookupTable;
class vtkStringArray;
class vtkTree;

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

  virtual void SetTree1(vtkTree *tree);

  virtual void SetTree2(vtkTree *tree);


  vtkTable * GetTable();
  void SetTable(vtkTable *table);


  vtkGetStringMacro(Tree1Label);
  vtkSetStringMacro(Tree1Label);


  vtkGetStringMacro(Tree2Label);
  vtkSetStringMacro(Tree2Label);

  void SetOrientation(int orientation);

  int GetOrientation();


  vtkGetMacro(MinimumVisibleFontSize, int);
  vtkSetMacro(MinimumVisibleFontSize, int);


  vtkGetMacro(LabelSizeDifference, int);
  vtkSetMacro(LabelSizeDifference, int);


  vtkGetMacro(CorrespondenceLineWidth, float);
  vtkSetMacro(CorrespondenceLineWidth, float);


  float GetTreeLineWidth();
  void SetTreeLineWidth(float width);

  bool Hit(const vtkContextMouseEvent &mouse) override;

  bool MouseDoubleClickEvent(const vtkContextMouseEvent &event) override;

protected:
  vtkTanglegramItem();
  ~vtkTanglegramItem() override;

  void RefreshBuffers(vtkContext2D *painter);

  void PositionTree2();

  void PaintCorrespondenceLines(vtkContext2D *painter);

  void PaintTreeLabels(vtkContext2D *painter);

  void ReorderTree();

  void ReorderTreeAtVertex(vtkIdType parent, vtkTree *tree);

  double GetPositionScoreForVertex(vtkIdType vertex, vtkTree *tree);

  void GenerateLookupTable();

  bool Paint(vtkContext2D *painter) override;

private:
  vtkSmartPointer<vtkDendrogramItem> Dendrogram1;
  vtkSmartPointer<vtkDendrogramItem> Dendrogram2;
  vtkSmartPointer<vtkLookupTable> LookupTable;
  vtkSmartPointer<vtkTable> Table;
  vtkStringArray *Tree1Names;
  vtkStringArray *Tree2Names;
  vtkStringArray *SourceNames;
  double Tree1Bounds[4];
  double Tree2Bounds[4];
  double Spacing;
  double LabelWidth1;
  double LabelWidth2;
  bool PositionSet;
  bool TreeReordered;
  char* Tree1Label;
  char* Tree2Label;
  int Orientation;
  int MinimumVisibleFontSize;
  int LabelSizeDifference;
  float CorrespondenceLineWidth;

  vtkTanglegramItem(const vtkTanglegramItem&) = delete;
  void operator=(const vtkTanglegramItem&) = delete;

};

#endif