lsDomain.hpp

Go to the documentation of this file.

#pragma once
 
#include <lsPreCompileMacros.hpp>
 
#include <cstring>
#include <limits>
 
#include <hrleDomain.hpp>
#include <hrleFillDomainWithSignedDistance.hpp>
 
#include <lsPointData.hpp>
 
#include <vcLogger.hpp>
#include <vcSmartPointer.hpp>
#include <vcVectorType.hpp>
 
#define LS_DOMAIN_SERIALIZATION_VERSION 0
 
namespace viennals {
 
using namespace viennacore;
 
// Boundary condition alias for easier access
using BoundaryConditionEnum = viennahrle::BoundaryType;

template <class T, int D> class Domain {
public:
  // TYPEDEFS
  typedef T ValueType;
  typedef viennahrle::Grid<D> GridType;
  typedef viennahrle::Domain<T, D> DomainType;
  typedef BoundaryConditionEnum BoundaryType;
  typedef std::vector<std::pair<viennahrle::Index<D>, T>> PointValueVectorType;
  typedef PointData<T> PointDataType;
  typedef std::vector<bool> VoidPointMarkersType;
 
private:
  // PRIVATE MEMBER VARIABLES
  GridType grid;
  DomainType domain;
  int levelSetWidth = 1;
  PointDataType pointData;
  VoidPointMarkersType voidPointMarkers;
 
public:
  // STATIC CONSTANTS
  static constexpr int dimensions = D;
 
  // PUBLIC MEMBER VARIABLES
  static constexpr T POS_VALUE = std::numeric_limits<T>::max();
  static constexpr T NEG_VALUE = std::numeric_limits<T>::lowest();

  Domain(viennahrle::CoordType gridDelta = 1.0) {
    viennahrle::IndexType gridMin[D], gridMax[D];
    BoundaryType boundaryCons[D];
    for (unsigned i = 0; i < D; ++i) {
      gridMin[i] = 0;
      gridMax[i] = 0;
      boundaryCons[i] = BoundaryType::INFINITE_BOUNDARY;
    }
 
    grid = GridType(gridMin, gridMax, gridDelta, boundaryCons);
    domain.deepCopy(grid, DomainType(grid, T(POS_VALUE)));
  }
 
  Domain(const viennahrle::CoordType *bounds, BoundaryType *boundaryConditions,
         viennahrle::CoordType gridDelta = 1.0) {
    viennahrle::IndexType gridMin[D], gridMax[D];
    for (unsigned i = 0; i < D; ++i) {
      gridMin[i] = std::floor(bounds[2 * i] / gridDelta);
      gridMax[i] = std::ceil(bounds[2 * i + 1] / gridDelta);
    }
 
    grid = GridType(gridMin, gridMax, gridDelta, boundaryConditions);
    domain.deepCopy(grid, DomainType(grid, T(POS_VALUE)));
  }
 
  Domain(std::vector<viennahrle::CoordType> bounds,
         std::vector<unsigned> boundaryConditions,
         viennahrle::CoordType gridDelta = 1.0) {
    BoundaryType boundaryCons[D];
    for (unsigned i = 0; i < D; ++i) {
      boundaryCons[i] = static_cast<BoundaryType>(boundaryConditions[i]);
    }
    auto newDomain =
        SmartPointer<Domain<T, D>>::New(bounds.data(), boundaryCons, gridDelta);
    this->deepCopy(newDomain);
  }

  Domain(PointValueVectorType pointData, viennahrle::CoordType *bounds,
         BoundaryType *boundaryConditions,
         viennahrle::CoordType gridDelta = 1.0) {
    auto newDomain =
        SmartPointer<Domain<T, D>>::New(bounds, boundaryConditions, gridDelta);
    this->deepCopy(newDomain);
    viennahrle::FillDomainWithSignedDistance(domain, pointData, T(NEG_VALUE),
                                             T(POS_VALUE));
  }
 
  Domain(GridType passedGrid) : grid(passedGrid) {
    domain.deepCopy(grid, DomainType(grid, T(POS_VALUE)));
  }
 
  Domain(SmartPointer<Domain> passedDomain) { deepCopy(passedDomain); }
 
  // Convenience function to create a new domain.
  template <class... Args> static auto New(Args &&...args) {
    return SmartPointer<Domain>::New(std::forward<Args>(args)...);
  }

  void finalize(int newWidth) { levelSetWidth = newWidth; }

  void finalize() {}

  void deepCopy(const SmartPointer<Domain<T, D>> passedDomain) {
    grid = passedDomain->grid;
    domain.deepCopy(grid, passedDomain->domain);
    levelSetWidth = passedDomain->levelSetWidth;
    pointData = passedDomain->pointData;
  }

  void insertPoints(PointValueVectorType pointData, bool sort = true) {
    viennahrle::FillDomainWithSignedDistance(domain, pointData, T(NEG_VALUE),
                                             T(POS_VALUE), sort);
  }

  const GridType &getGrid() const { return grid; }

  GridType &getGrid() { return grid; }

  DomainType &getDomain() { return domain; }
 
  const DomainType &getDomain() const { return domain; }

  unsigned getNumberOfSegments() const { return domain.getNumberOfSegments(); }

  unsigned getNumberOfPoints() const { return domain.getNumberOfPoints(); }
 
  int getLevelSetWidth() const { return levelSetWidth; }
 
  void setLevelSetWidth(int width) { levelSetWidth = width; }
 
  // clear all additional data
  void clearMetaData() { pointData.clear(); }

  PointDataType &getPointData() { return pointData; }
 
  const PointDataType &getPointData() const { return pointData; }

  VoidPointMarkersType &getVoidPointMarkers() { return voidPointMarkers; }
 
  const VoidPointMarkersType &getVoidPointMarkers() const {
    return voidPointMarkers;
  }

  void print(std::ostream &out = std::cout) {
    out << "Grid pointer: " << &grid << std::endl;
    out << "lsDomain: " << &domain << std::endl;
    out << "DomainSegments: " << std::endl;
    for (unsigned i = 0; i < getNumberOfSegments(); ++i) {
      out << &(domain.getDomainSegment(i)) << std::endl;
    }
    domain.print(out);
  }

  std::ostream &serialize(std::ostream &stream) {
    // Save header to identify Domain
    stream << "lsDomain";
 
    // now write format version number
    char formatVersion = LS_DOMAIN_SERIALIZATION_VERSION;
    stream.write(&formatVersion, 1);
 
    // serialize grid
    grid.serialize(stream);
 
    // serialize hrleDomain which saves LS values
    domain.serialize(stream);
 
    // serialize Domain members
    // level set width as 32bit uint
    const uint32_t width = levelSetWidth;
    stream.write(reinterpret_cast<const char *>(&width), sizeof(uint32_t));
 
    // serialize pointData if there is any point data associated with this
    // Domain mark whether there is point data or not (1/0)
    char hasPointData = (pointData.empty()) ? 0 : 1;
    stream.write(&hasPointData, 1);
    if (hasPointData == 1) {
      pointData.serialize(stream);
    }
 
    return stream;
  }

  std::istream &deserialize(std::istream &stream) {
    // Check identifier
    char identifier[8];
    stream.read(identifier, 8);
    if (std::memcmp(identifier, "lsDomain", 8) != 0) {
      Logger::getInstance()
          .addError(
              "Reading Domain from stream failed. Header could not be found.")
          .print();
      return stream;
    }
 
    // check format version for compatibility
    char formatVersion;
    stream.read(&formatVersion, 1);
    if (formatVersion > LS_DOMAIN_SERIALIZATION_VERSION) {
      Logger::getInstance()
          .addError("Reading Domain of version " +
                    std::to_string(formatVersion) + " with reader of version " +
                    std::to_string(LS_DOMAIN_SERIALIZATION_VERSION) +
                    " failed.")
          .print();
      return stream;
    }
 
    // read in the grid
    grid.deserialize(stream);
 
    // read in the hrleDoamin
    // grid pointer in hrleDomain is implicitly set
    // to the correct grid already since they were
    // initialized together
    domain.deserialize(stream);
 
    // read in the level set width
    uint32_t width;
    stream.read(reinterpret_cast<char *>(&width), sizeof(uint32_t));
    levelSetWidth = width;
 
    // check wether there is point data to read
    char hasPointData;
    stream.read(&hasPointData, 1);
    if (hasPointData == 1) {
      pointData.clear();
      pointData.deserialize(stream);
    }
 
    return stream;
  }
};
 
// add all template specialisations for this class
PRECOMPILE_PRECISION_DIMENSION(Domain)
 
} // namespace viennals