lsWriter.hpp

Go to the documentation of this file.

#pragma once
 
#include <fstream>
 
#include <lsDomain.hpp>
#include <lsPreCompileMacros.hpp>
#include <utility>
 
namespace viennals {
 
using namespace viennacore;
 
template <class T, int D> class Writer {
  SmartPointer<Domain<T, D>> levelSet = nullptr;
  std::string fileName;
 
public:
  Writer() = default;
 
  Writer(SmartPointer<Domain<T, D>> passedLevelSet)
      : levelSet(passedLevelSet) {}
 
  Writer(SmartPointer<Domain<T, D>> passedLevelSet, std::string passedFileName)
      : levelSet(passedLevelSet), fileName(std::move(passedFileName)) {}
 
  void setLevelSet(SmartPointer<Domain<T, D>> passedLevelSet) {
    levelSet = passedLevelSet;
  }

  void setFileName(std::string passedFileName) {
    fileName = std::move(passedFileName);
  }
 
  void apply() {
    // check level-set
    if (levelSet == nullptr) {
      Logger::getInstance()
          .addWarning("No level-set was passed to Writer. Not writing.")
          .print();
      return;
    }
    // check filename
    if (fileName.empty()) {
      Logger::getInstance()
          .addWarning("No file name specified for Writer. Not writing.")
          .print();
      return;
    }
 
    if (fileName.find(".lvst") != fileName.length() - 5) {
      Logger::getInstance()
          .addWarning("File name does not end in '.lvst', appending it.")
          .print();
      fileName.append(".lvst");
    }
 
    // Open file for writing and save serialized level set in it
    std::ofstream fout(fileName, std::ios::binary);
 
    levelSet->serialize(fout);
 
    fout.close();
  }
};
 
// add all template specialisations for this class
PRECOMPILE_PRECISION_DIMENSION(Writer)
 
} // namespace viennals