vtkMRMLFileWriter.h

Go to the documentation of this file.

/*==============================================================================
 
  Program: 3D Slicer
 
  Copyright (c) Kitware Inc.
 
  See COPYRIGHT.txt
  or http://www.slicer.org/copyright/copyright.txt for details.
 
  Unless required by applicable law or agreed to in writing, software
  distributed under the License is distributed on an "AS IS" BASIS,
  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  See the License for the specific language governing permissions and
  limitations under the License.
 
==============================================================================*/
 
#ifndef __vtkMRMLFileWriter_h
#define __vtkMRMLFileWriter_h
 
#include "vtkMRMLFileIOHandler.h"
 
class vtkMRMLIOProperties;

class VTK_MRML_LOGIC_EXPORT vtkMRMLFileWriter : public vtkMRMLFileIOHandler
{
public:
  static vtkMRMLFileWriter* New();
  vtkTypeMacro(vtkMRMLFileWriter, vtkMRMLFileIOHandler);
  void PrintSelf(ostream& os, vtkIndent indent) override;
 
  bool IsWriter() const override { return true; }

  virtual double CanWriteObjectConfidence(vtkObject* object);

  virtual bool CanWriteObject(vtkObject* object);

  virtual std::vector<std::string> GetNameFiltersForObject(vtkObject* object);

  virtual bool Write(vtkMRMLIOProperties* properties);

  const std::vector<std::string>& GetWrittenNodeIDs() const;
  void SetWrittenNodeIDs(const std::vector<std::string>& nodeIDs);
  void AddWrittenNodeID(const std::string& nodeID);
  void ClearWrittenNodeIDs();

  std::string GetNodeClassName() const;
  void SetNodeClassName(const std::string& className);

  void SetNodeClassNames(const std::vector<std::string>& classNames);
  const std::vector<std::string>& GetNodeClassNames() const { return this->NodeClassNames; }
  void AddNodeClassName(const std::string& className);

  vtkGetMacro(ConfidenceForMatchingClass, double);
  vtkSetMacro(ConfidenceForMatchingClass, double);
 
protected:
  vtkMRMLFileWriter();
  ~vtkMRMLFileWriter() override;
 
  std::vector<std::string> NodeClassNames;
  double ConfidenceForMatchingClass{ 0.5 };
  std::vector<std::string> WrittenNodeIDs;
 
private:
  vtkMRMLFileWriter(const vtkMRMLFileWriter&) = delete;
  void operator=(const vtkMRMLFileWriter&) = delete;
};
 
#endif