ttkProgramBase.h

Go to the documentation of this file.

 
 
#pragma once
 
// VTK IO
#include <vtkDataSet.h>
#include <vtkDataSetAlgorithm.h>
#include <vtkImageData.h>
#include <vtkPolyData.h>
#include <vtkSmartPointer.h>
#include <vtkUnstructuredGrid.h>
#include <vtkXMLImageDataReader.h>
#include <vtkXMLImageDataWriter.h>
#include <vtkXMLPolyDataReader.h>
#include <vtkXMLPolyDataWriter.h>
#include <vtkXMLUnstructuredGridReader.h>
#include <vtkXMLUnstructuredGridWriter.h>
 
// base code includes
#include <ProgramBase.h>
#include <ttkAlgorithm.h>
 
#include <ttkProgramBaseModule.h>
 
class TTKPROGRAMBASE_EXPORT ttkProgramBase : public ttk::ProgramBase {
 
public:
  ttkProgramBase() {
 
    vtkWrapper_ = nullptr;
  };
 
  ~ttkProgramBase() override = default;
  ;
 
  int execute() override;
 
  vtkDataSet *getInput(const int &inputId) {
    if((inputId < 0) || (inputId >= (int)inputs_.size()))
      return nullptr;
    return inputs_[inputId];
  }
 
  int getNumberOfInputs() {
    return inputs_.size();
  };
 
  int run() override {
 
    if(!vtkWrapper_) {
      return -1;
    }
 
    if(!ttkModule_)
      return -2;
 
    ttkModule_->setDebugLevel(debugLevel_);
    ttkModule_->setThreadNumber(threadNumber_);
 
    return execute();
  }
 
  int save() const override;
 
  virtual int setTTKmodule(vtkDataSetAlgorithm *ttkModule) {
 
    vtkWrapper_ = ttkModule;
    ttkModule_ = (Debug *)ttkModule;
 
    return 0;
  }
 
protected:
  std::vector<vtkDataSet *> inputs_;
  std::vector<vtkSmartPointer<vtkXMLImageDataReader>> imageDataReaders_;
  std::vector<vtkSmartPointer<vtkXMLPolyDataReader>> polyDataReaders_;
  std::vector<vtkSmartPointer<vtkXMLUnstructuredGridReader>>
    unstructuredGridReaders_;
  vtkDataSetAlgorithm *vtkWrapper_;
 
  template <class vtkReaderClass>
  int load(const std::string &fileName,
           std::vector<vtkSmartPointer<vtkReaderClass>> &readerList);
 
  int load(const std::vector<std::string> &inputPaths) override;
 
  template <class vtkWriterClass>
  int save(const int &outputPortId) const;
};
 
template <class ttkModule>
class vtkProgram : public ttkProgramBase {
 
public:
  vtkProgram() {
    ttkObject_ = vtkSmartPointer<ttkModule>::New();
    vtkWrapper_ = (vtkDataSetAlgorithm *)ttkObject_.GetPointer();
    ttkModule_ = (Debug *)ttkObject_.GetPointer();
  }
 
  int run() override {
    return ttkProgramBase::run();
  }
 
  vtkSmartPointer<ttkModule> ttkObject_;
};
 
template <class vtkWriterClass>
int ttkProgramBase::save(const int &outputPortId) const {
 
  if(!vtkWrapper_)
    return -1;
 
  std::string extension;
 
  if((vtkWrapper_->GetOutput(outputPortId)->GetDataObjectType()
      == VTK_IMAGE_DATA)) {
    //     ||(vtkWrapper_->GetOutput(outputPortId)->GetDataObjectType()
    //       == TTK_IMAGE_DATA)){
    extension = "vti";
  }
 
  if((vtkWrapper_->GetOutput(outputPortId)->GetDataObjectType()
      == VTK_POLY_DATA)) {
    //     ||(vtkWrapper_->GetOutput(outputPortId)->GetDataObjectType()
    //       == TTK_POLY_DATA)){
    extension = "vtp";
  }
 
  if((vtkWrapper_->GetOutput(outputPortId)->GetDataObjectType()
      == VTK_UNSTRUCTURED_GRID)) {
    //     ||(vtkWrapper_->GetOutput(outputPortId)->GetDataObjectType()
    //       == TTK_UNSTRUCTURED_GRID)){
    extension = "vtu";
  }
 
  std::stringstream fileName;
  fileName << outputPath_ << "_port#" << outputPortId << "." << extension;
 
  vtkSmartPointer<vtkWriterClass> writer
    = vtkSmartPointer<vtkWriterClass>::New();
  writer->SetFileName(fileName.str().data());
  writer->SetInputData(vtkWrapper_->GetOutput(outputPortId));
  std::stringstream msg;
  msg << "[ttkProgramBase] Saving output file `" << fileName.str() << "'..."
      << std::endl;
  printMsg(msg.str());
 
  writer->Write();
 
  return 0;
}
 
template <class vtkReaderClass>
int ttkProgramBase::load(
  const std::string &fileName,
  std::vector<vtkSmartPointer<vtkReaderClass>> &readerList) {
 
  readerList.resize(readerList.size() + 1);
  readerList.back() = vtkSmartPointer<vtkReaderClass>::New();
 
  readerList.back()->SetFileName(fileName.data());
 
  // handle debug messages
  {
    std::stringstream msg;
    msg << "[ttkProgramBase] Reading input data..." << std::endl;
    // choose where to display this message (std::cout, std::cerr, a file)
    // choose the priority of this message (1, nearly always displayed,
    // higher values mean lower priorities)
    printMsg(msg.str());
  }
 
  readerList.back()->Update();
  inputs_.push_back(readerList.back()->GetOutput());
 
  if(!inputs_.back())
    return -1;
 
  if(!inputs_.back()->GetNumberOfPoints())
    return -2;
 
  if(!inputs_.back()->GetNumberOfCells())
    return -3;
 
  {
    std::stringstream msg;
    msg << "[ttkProgramBase]   done! (read "
        << inputs_.back()->GetNumberOfPoints() << " vertices, "
        << inputs_.back()->GetNumberOfCells() << " cells)" << std::endl;
    printMsg(msg.str());
  }
 
  return 0;
}