1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86
|
// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
/**
* @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
VTK_ABI_NAMESPACE_BEGIN
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 = default;
void WriteData() override;
/**
* Write one vertex. This function calls itself recursively for
* any children of the input vertex.
*/
void WriteVertex(ostream* fp, vtkTree* 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;
};
VTK_ABI_NAMESPACE_END
#endif
|