vtkCosmicTreeLayoutStrategy.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-FileCopyrightText: Copyright (c) Sandia Corporation
// SPDX-License-Identifier: BSD-3-Clause
#ifndef vtkCosmicTreeLayoutStrategy_h
#define vtkCosmicTreeLayoutStrategy_h
 
#include "vtkGraphLayoutStrategy.h"
#include "vtkInfovisLayoutModule.h" // For export macro
 
VTK_ABI_NAMESPACE_BEGIN
class vtkDoubleArray;
class vtkDataArray;
class vtkPoints;
class vtkTree;
 
class VTKINFOVISLAYOUT_EXPORT vtkCosmicTreeLayoutStrategy : public vtkGraphLayoutStrategy
{
public:
  static vtkCosmicTreeLayoutStrategy* New();
  void PrintSelf(ostream& os, vtkIndent indent) override;
  vtkTypeMacro(vtkCosmicTreeLayoutStrategy, vtkGraphLayoutStrategy);
 
  void Layout() override;
 
 
  vtkSetMacro(SizeLeafNodesOnly, vtkTypeBool);
  vtkGetMacro(SizeLeafNodesOnly, vtkTypeBool);
  vtkBooleanMacro(SizeLeafNodesOnly, vtkTypeBool);
 
 
  vtkSetMacro(LayoutDepth, int);
  vtkGetMacro(LayoutDepth, int);
 
 
  vtkSetMacro(LayoutRoot, vtkIdType);
  vtkGetMacro(LayoutRoot, vtkIdType);
 
 
  vtkSetStringMacro(NodeSizeArrayName);
  vtkGetStringMacro(NodeSizeArrayName);
 
protected:
  enum RadiusMode
  {
    NONE,   
    LEAVES, 
    ALL     
  };
 
  vtkCosmicTreeLayoutStrategy();
  ~vtkCosmicTreeLayoutStrategy() override;
 
  void LayoutChildren(vtkTree* tree, vtkPoints* newPoints, vtkDoubleArray* radii,
    vtkDoubleArray* scale, vtkIdType root, int depth, RadiusMode mode);
 
  void OffsetChildren(vtkTree* tree, vtkPoints* pts, vtkDoubleArray* radii, vtkDoubleArray* scale,
    double parent[4], vtkIdType root, int depth, RadiusMode mode);
 
  vtkDoubleArray* CreateRadii(vtkIdType numVertices, double initialValue, vtkDataArray* inputRadii);
 
  vtkDoubleArray* CreateScaleFactors(vtkIdType numVertices);
 
  vtkTypeBool SizeLeafNodesOnly;
  int LayoutDepth;
  vtkIdType LayoutRoot;
  char* NodeSizeArrayName;
 
private:
  vtkCosmicTreeLayoutStrategy(const vtkCosmicTreeLayoutStrategy&) = delete;
  void operator=(const vtkCosmicTreeLayoutStrategy&) = delete;
};
 
VTK_ABI_NAMESPACE_END
#endif // vtkCosmicTreeLayoutStrategy_h