vtkTanglegramItem.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
#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
#include "vtkWrappingHints.h" // For VTK_MARSHALAUTO

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

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


  virtual void SetTree1(vtkTree* tree);
  vtkTree* GetTree1();


  virtual void SetTree2(vtkTree* tree);
  vtkTree* GetTree2();


  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;
};

VTK_ABI_NAMESPACE_END
#endif