You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
97 lines
2.7 KiB
C
97 lines
2.7 KiB
C
3 weeks ago
|
/*=========================================================================
|
||
|
|
||
|
Program: Visualization Toolkit
|
||
|
Module: vtkNewickTreeWriter.h
|
||
|
|
||
|
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.
|
||
|
|
||
|
=========================================================================*/
|
||
|
/**
|
||
|
* @class vtkNewickTreeWriter
|
||
|
* @brief write vtkTree data to Newick format.
|
||
|
*
|
||
|
* vtkNewickTreeWriter is writes a vtkTree to a Newick formatted file
|
||
|
* or string.
|
||
|
*/
|
||
|
|
||
|
#ifndef vtkNewickTreeWriter_h
|
||
|
#define vtkNewickTreeWriter_h
|
||
|
|
||
|
#include "vtkDataWriter.h"
|
||
|
#include "vtkIOInfovisModule.h" // For export macro
|
||
|
#include "vtkStdString.h" // For get/set ivars
|
||
|
|
||
|
class vtkTree;
|
||
|
|
||
|
class VTKIOINFOVIS_EXPORT vtkNewickTreeWriter : public vtkDataWriter
|
||
|
{
|
||
|
public:
|
||
|
static vtkNewickTreeWriter* New();
|
||
|
vtkTypeMacro(vtkNewickTreeWriter, vtkDataWriter);
|
||
|
void PrintSelf(ostream& os, vtkIndent indent) override;
|
||
|
|
||
|
//@{
|
||
|
/**
|
||
|
* Get the input to this writer.
|
||
|
*/
|
||
|
vtkTree* GetInput();
|
||
|
vtkTree* GetInput(int port);
|
||
|
//@}
|
||
|
|
||
|
//@{
|
||
|
/**
|
||
|
* Get/Set the name of the input's tree edge weight array.
|
||
|
* This array must be part of the input tree's EdgeData.
|
||
|
* The default name is "weight". If this array cannot be
|
||
|
* found, then no edge weights will be included in the
|
||
|
* output of this writer.
|
||
|
*/
|
||
|
vtkGetMacro(EdgeWeightArrayName, vtkStdString);
|
||
|
vtkSetMacro(EdgeWeightArrayName, vtkStdString);
|
||
|
//@}
|
||
|
|
||
|
//@{
|
||
|
/**
|
||
|
* Get/Set the name of the input's tree node name array.
|
||
|
* This array must be part of the input tree's VertexData.
|
||
|
* The default name is "node name". If this array cannot
|
||
|
* be found, then no node names will be included in the
|
||
|
* output of this writer.
|
||
|
*/
|
||
|
vtkGetMacro(NodeNameArrayName, vtkStdString);
|
||
|
vtkSetMacro(NodeNameArrayName, vtkStdString);
|
||
|
//@}
|
||
|
|
||
|
protected:
|
||
|
vtkNewickTreeWriter();
|
||
|
~vtkNewickTreeWriter() override {}
|
||
|
|
||
|
void WriteData() override;
|
||
|
|
||
|
/**
|
||
|
* Write one vertex. This function calls itself recursively for
|
||
|
* any children of the input vertex.
|
||
|
*/
|
||
|
void WriteVertex(ostream* fp, vtkTree* const input, vtkIdType vertex);
|
||
|
|
||
|
int FillInputPortInformation(int port, vtkInformation* info) override;
|
||
|
|
||
|
vtkStdString EdgeWeightArrayName;
|
||
|
vtkStdString NodeNameArrayName;
|
||
|
|
||
|
vtkAbstractArray* EdgeWeightArray;
|
||
|
vtkAbstractArray* NodeNameArray;
|
||
|
|
||
|
private:
|
||
|
vtkNewickTreeWriter(const vtkNewickTreeWriter&) = delete;
|
||
|
void operator=(const vtkNewickTreeWriter&) = delete;
|
||
|
};
|
||
|
|
||
|
#endif
|