Libs/Optimize/Domain/ImageDomainWithGradN.h

Namespaces

Name
shapeworks User usage reporting (telemetry)

Classes

	Name
class	shapeworks::ImageDomainWithGradN

Source code

```cpp

pragma once

include "ImageDomainWithGradients.h"

include "itkDerivativeImageFilter.h"

include "itkDiscreteGaussianImageFilter.h"

include "itkFixedArray.h"

include "itkGradientImageFilter.h"

include "itkImage.h"

include "itkImageDuplicator.h"

include "vnl/vnl_matrix_fixed.h"

namespace shapeworks { template class ImageDomainWithGradN : public ImageDomainWithGradients { public: typedef ImageDomainWithGradients Superclass; using Pointer = std::shared_ptr;

typedef typename Superclass::PointType PointType; typedef typename Superclass::ImageType ImageType;

typedef typename Superclass::GradNType GradNType; typedef typename Superclass::GradNType VnlMatrixType;

void SetImage(ImageType* I, double narrow_band) { Superclass::SetImage(I, narrow_band);

const auto grad = this->GetVDBGradient();

// Compute the gradient of normals component-wise
for (int i = 0; i < 3; i++) {
  auto norm_i = openvdb::FloatGrid::create();
  norm_i->setTransform(this->transform());
  auto norm_i_accessor = norm_i->getAccessor();
  for (openvdb::VectorGrid::ValueOnCIter it = grad->cbeginValueOn(); it.test(); ++it) {
    const openvdb::Vec3f& v = *it;
    norm_i_accessor.setValue(it.getCoord(), v[i] / v.length());
  }

  m_VDBGradNorms[i] = openvdb::tools::gradient(*norm_i);
}

} // end setimage

inline GradNType SampleGradNAtPoint(const PointType& p, int idx) const override { const auto coord = this->ToVDBCoord(p);

GradNType grad_n;
for (int i = 0; i < 3; i++) {
  auto grad_ni = openvdb::tools::BoxSampler::sample(m_VDBGradNorms[i]->tree(), coord);
  grad_n.set(i, 0, grad_ni[0]);
  grad_n.set(i, 1, grad_ni[1]);
  grad_n.set(i, 2, grad_ni[2]);
}
return grad_n;

}

void DeletePartialDerivativeImages() override { for (unsigned int i = 0; i < DIMENSION; i++) { m_VDBGradNorms[i] = 0; } }

void DeleteImages() override { Superclass::DeleteImages(); DeletePartialDerivativeImages(); }

protected: ImageDomainWithGradN() {} virtual ~ImageDomainWithGradN(){};

void PrintSelf(std::ostream& os, itk::Indent indent) const { Superclass::PrintSelf(os, indent); }

private: typename openvdb::VectorGrid::Ptr m_VDBGradNorms[3]; };

} // end namespace shapeworks ```

---

Updated on 2026-03-31 at 16:02:11 +0000