lsCalculateCurvatures.hpp

Go to the documentation of this file.

#pragma once
 
#include <hrleCartesianPlaneIterator.hpp>
#include <lsCurvatureFormulas.hpp>
#include <lsDomain.hpp>
#include <lsExpand.hpp>
 
#include <vcLogger.hpp>
#include <vcSmartPointer.hpp>
 
namespace viennals {
 
using namespace viennacore;
 
enum struct CurvatureEnum : unsigned {
  MEAN_CURVATURE = 0,
  GAUSSIAN_CURVATURE = 1,
  MEAN_AND_GAUSSIAN_CURVATURE = 2
};
 
// Calculates the Mean Curvature and/or Gaussian Curvature (3D)
// for the passed lsDomain for all points with level set values <= 0.5. The
// result is saved in the lsDomain.
template <class T, int D> class CalculateCurvatures {
  SmartPointer<Domain<T, D>> levelSet = nullptr;
  T maxValue = 0.5;
  CurvatureEnum type = CurvatureEnum::MEAN_CURVATURE;
 
public:
  static constexpr char meanCurvatureLabel[] = "MeanCurvatures";
  static constexpr char gaussianCurvatureLabel[] = "GaussianCurvatures";
 
  CalculateCurvatures() = default;
 
  CalculateCurvatures(SmartPointer<Domain<T, D>> passedLevelSet)
      : levelSet(passedLevelSet) {}
 
  CalculateCurvatures(SmartPointer<Domain<T, D>> passedLevelSet,
                      CurvatureEnum method)
      : levelSet(passedLevelSet), type(method) {}
 
  void setLevelSet(SmartPointer<Domain<T, D>> passedLevelSet) {
    levelSet = passedLevelSet;
  }
 
  void setCurvatureType(CurvatureEnum passedType) {
    // in 2D there is only one option so ignore
    if constexpr (D == 3) {
      type = passedType;
    } else {
      if (passedType != type) {
        VIENNACORE_LOG_WARNING(
            "CalculateCurvatures: Could not set curvature type because 2D "
            "only supports mean curvature.");
      }
    }
  }
 
  void setMaxValue(const T passedMaxValue) { maxValue = passedMaxValue; }
 
  void apply() {
    if (levelSet == nullptr) {
      VIENNACORE_LOG_ERROR("No level set was passed to CalculateCurvatures.");
      return;
    }
 
    // need second neighbours
    if (unsigned minWidth = std::ceil((maxValue * 8) + 1);
        levelSet->getLevelSetWidth() < minWidth) {
      VIENNACORE_LOG_WARNING("CalculateCurvatures: Level set width must be "
                             "at least " +
                             std::to_string(minWidth) +
                             ". Expanding level set to " +
                             std::to_string(minWidth) + ".");
      Expand<T, D>(levelSet, minWidth).apply();
    }
 
    std::vector<std::vector<T>> meanCurvaturesVector(
        levelSet->getNumberOfSegments());
    std::vector<std::vector<T>> gaussCurvaturesVector(
        levelSet->getNumberOfSegments());
 
    auto grid = levelSet->getGrid();
    const bool calculateMean =
        (type == CurvatureEnum::MEAN_CURVATURE) ||
        (type == CurvatureEnum::MEAN_AND_GAUSSIAN_CURVATURE);
    const bool calculateGauss =
        (type == CurvatureEnum::GAUSSIAN_CURVATURE) ||
        (type == CurvatureEnum::MEAN_AND_GAUSSIAN_CURVATURE);

#pragma omp parallel for
    for (unsigned p = 0; p < levelSet->getNumberOfSegments(); ++p) {
 
      auto &meanCurvatures = meanCurvaturesVector[p];
      auto &gaussCurvatures = gaussCurvaturesVector[p];
 
      if (calculateMean) {
        meanCurvatures.reserve(
            levelSet->getDomain().getDomainSegment(p).getNumberOfPoints());
      }
      if (calculateGauss) {
        gaussCurvatures.reserve(
            levelSet->getDomain().getDomainSegment(p).getNumberOfPoints());
      }
 
      viennahrle::Index<D> const startVector =
          (p == 0) ? grid.getMinGridPoint()
                   : levelSet->getDomain().getSegmentation()[p - 1];
 
      viennahrle::Index<D> const endVector =
          (p != static_cast<int>(levelSet->getNumberOfSegments() - 1))
              ? levelSet->getDomain().getSegmentation()[p]
              : grid.incrementIndices(grid.getMaxGridPoint());
 
      for (viennahrle::CartesianPlaneIterator<typename Domain<T, D>::DomainType>
               neighborIt(levelSet->getDomain(), startVector);
           neighborIt.getIndices() < endVector; neighborIt.next()) {
 
        auto &center = neighborIt.getCenter();
        if (!center.isDefined()) {
          continue;
        } else if (std::abs(center.getValue()) > maxValue) {
          if (calculateMean)
            meanCurvatures.push_back(0.);
          if (calculateGauss)
            gaussCurvatures.push_back(0.);
          continue;
        }
 
        // calculate curvatures
        if (calculateMean) {
          meanCurvatures.push_back(lsInternal::meanCurvature(neighborIt));
        }
        if (calculateGauss) {
          gaussCurvatures.push_back(lsInternal::gaussianCurvature(neighborIt));
        }
      }
    }
 
    // put all curvature values in the correct ordering
    if (calculateMean) {
      unsigned numberOfCurvatures = 0;
      for (unsigned i = 0; i < levelSet->getNumberOfSegments(); ++i) {
        numberOfCurvatures += meanCurvaturesVector[i].size();
      }
      meanCurvaturesVector[0].reserve(numberOfCurvatures);
 
      for (unsigned i = 1; i < levelSet->getNumberOfSegments(); ++i) {
        meanCurvaturesVector[0].insert(meanCurvaturesVector[0].end(),
                                       meanCurvaturesVector[i].begin(),
                                       meanCurvaturesVector[i].end());
      }
    }
 
    if (calculateGauss) {
      unsigned numberOfCurvatures = 0;
      for (unsigned i = 0; i < levelSet->getNumberOfSegments(); ++i) {
        numberOfCurvatures += gaussCurvaturesVector[i].size();
      }
      gaussCurvaturesVector[0].reserve(numberOfCurvatures);
 
      for (unsigned i = 1; i < levelSet->getNumberOfSegments(); ++i) {
        gaussCurvaturesVector[0].insert(gaussCurvaturesVector[0].end(),
                                        gaussCurvaturesVector[i].begin(),
                                        gaussCurvaturesVector[i].end());
      }
    }
 
    // insert into pointData of levelSet
    if (calculateMean) {
      auto &pointData = levelSet->getPointData();
      auto scalarDataPointer =
          pointData.getScalarData(meanCurvatureLabel, true);
      // if it does not exist, insert new normals vector
      if (scalarDataPointer == nullptr) {
        pointData.insertNextScalarData(meanCurvaturesVector[0],
                                       meanCurvatureLabel);
      } else {
        // if it does exist, just swap the old with the new values
        *scalarDataPointer = std::move(meanCurvaturesVector[0]);
      }
    }
 
    if (calculateGauss) {
      auto &pointData = levelSet->getPointData();
      auto scalarDataPointer =
          pointData.getScalarData(gaussianCurvatureLabel, true);
      // if it does not exist, insert new normals vector
      if (scalarDataPointer == nullptr) {
        pointData.insertNextScalarData(gaussCurvaturesVector[0],
                                       gaussianCurvatureLabel);
      } else {
        // if it does exist, just swap the old with the new values
        *scalarDataPointer = std::move(gaussCurvaturesVector[0]);
      }
    }
  }
};
 
} // namespace viennals