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++

/*=========================================================================
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