Method SetTree

Views/Infovis/vtkDendrogramItem.cxx:93–165

------------------------------------------------------------------------------

//------------------------------------------------------------------------------
void vtkDendrogramItem::SetTree(vtkTree* tree)
{
  if (tree == nullptr || tree->GetNumberOfVertices() == 0)
  {
    this->Tree = vtkSmartPointer<vtkTree>::New();
    this->PrunedTree = vtkSmartPointer<vtkTree>::New();
    this->LayoutTree = vtkSmartPointer<vtkTree>::New();
    return;
  }

  this->Tree = tree;

  vtkSmartPointer<vtkUnsignedIntArray> originalVertexIsPruned =
    vtkArrayDownCast<vtkUnsignedIntArray>(this->Tree->GetVertexData()->GetArray("VertexIsPruned"));

  // initialize some additional arrays for the tree's vertex data
  vtkNew<vtkUnsignedIntArray> vertexIsPruned;
  vertexIsPruned->SetNumberOfComponents(1);
  vertexIsPruned->SetName("VertexIsPruned");
  vertexIsPruned->SetNumberOfValues(this->Tree->GetNumberOfVertices());
  vertexIsPruned->FillComponent(0, 0.0);
  this->Tree->GetVertexData()->AddArray(vertexIsPruned);

  vtkNew<vtkIdTypeArray> originalId;
  originalId->SetNumberOfComponents(1);
  originalId->SetName("OriginalId");
  vtkIdType numVertices = this->Tree->GetNumberOfVertices();
  originalId->SetNumberOfValues(numVertices);
  for (vtkIdType i = 0; i < numVertices; ++i)
  {
    originalId->SetValue(i, i);
  }
  this->Tree->GetVertexData()->AddArray(originalId);

  // make a copy of the full tree for later pruning
  this->PrunedTree->DeepCopy(this->Tree);

  // If the tree already had an array named VertexIsPruned,
  // use that array to construct the pruned tree.
  if (originalVertexIsPruned)
  {
    for (vtkIdType i = 0; i < originalVertexIsPruned->GetNumberOfValues(); ++i)
    {
      if (originalVertexIsPruned->GetValue(i))
      {
        this->CollapseSubTree(i);
      }
    }
  }

  // setup the lookup table that's used to color the triangles representing
  // collapsed subtrees.  First we find maximum possible value.
  vtkIdType root = this->Tree->GetRoot();
  if (this->Tree->GetNumberOfChildren(root) == 1)
  {
    root = this->Tree->GetChild(root, 0);
  }
  int numLeavesInBiggestSubTree = 0;