AbstractContour.h

Go to the documentation of this file.

//============================================================================
//  Copyright (c) Kitware, Inc.
//  All rights reserved.
//  See LICENSE.txt for details.
//
//  This software is distributed WITHOUT ANY WARRANTY; without even
//  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
//  PURPOSE.  See the above copyright notice for more information.
//============================================================================
 
#ifndef vtk_m_filter_contour_AbstractContour_h
#define vtk_m_filter_contour_AbstractContour_h
 
#include <vtkm/filter/Filter.h>
#include <vtkm/filter/MapFieldPermutation.h>
#include <vtkm/filter/contour/ContourDimension.h>
#include <vtkm/filter/contour/vtkm_filter_contour_export.h>
#include <vtkm/filter/vector_analysis/SurfaceNormals.h>
 
namespace vtkm
{
namespace filter
{
namespace contour
{
class VTKM_FILTER_CONTOUR_EXPORT AbstractContour : public vtkm::filter::Filter
{
public:
  void SetNumberOfIsoValues(vtkm::Id num)
  {
    if (num >= 0)
    {
      this->IsoValues.resize(static_cast<std::size_t>(num));
    }
  }
 
  vtkm::Id GetNumberOfIsoValues() const { return static_cast<vtkm::Id>(this->IsoValues.size()); }
 
  void SetIsoValue(vtkm::Float64 v) { this->SetIsoValue(0, v); }
 
  void SetIsoValue(vtkm::Id index, vtkm::Float64 v)
  {
    std::size_t i = static_cast<std::size_t>(index);
    if (i >= this->IsoValues.size())
    {
      this->IsoValues.resize(i + 1);
    }
    this->IsoValues[i] = v;
  }
 
  void SetIsoValues(const std::vector<vtkm::Float64>& values) { this->IsoValues = values; }
 
  vtkm::Float64 GetIsoValue(vtkm::Id index = 0) const
  {
    return this->IsoValues[static_cast<std::size_t>(index)];
  }
 
  VTKM_CONT
  void SetGenerateNormals(bool flag) { this->GenerateNormals = flag; }
  VTKM_CONT
  bool GetGenerateNormals() const { return this->GenerateNormals; }
 
  VTKM_CONT
  void SetAddInterpolationEdgeIds(bool flag) { this->AddInterpolationEdgeIds = flag; }
  VTKM_CONT
  bool GetAddInterpolationEdgeIds() const { return this->AddInterpolationEdgeIds; }
 
  VTKM_CONT
  void SetComputeFastNormals(bool flag) { this->ComputeFastNormals = flag; }
  VTKM_CONT
  bool GetComputeFastNormals() const { return this->ComputeFastNormals; }
 
  VTKM_CONT
  void SetNormalArrayName(const std::string& name) { this->NormalArrayName = name; }
 
  VTKM_CONT
  const std::string& GetNormalArrayName() const { return this->NormalArrayName; }
 
  VTKM_CONT void SetInputCellDimension(vtkm::filter::contour::ContourDimension dimension)
  {
    this->InputCellDimension = dimension;
  }
 
  VTKM_CONT vtkm::filter::contour::ContourDimension GetInputCellDimension() const
  {
    return this->InputCellDimension;
  }
 
  VTKM_CONT void SetInputCellDimensionToAuto()
  {
    this->SetInputCellDimension(vtkm::filter::contour::ContourDimension::Auto);
  }
 
  VTKM_CONT void SetInputCellDimensionToAll()
  {
    this->SetInputCellDimension(vtkm::filter::contour::ContourDimension::All);
  }
 
  VTKM_CONT void SetInputCellDimensionToPolyhedra()
  {
    this->SetInputCellDimension(vtkm::filter::contour::ContourDimension::Polyhedra);
  }
 
  VTKM_CONT void SetInputCellDimensionToPolygons()
  {
    this->SetInputCellDimension(vtkm::filter::contour::ContourDimension::Polygons);
  }
 
  VTKM_CONT void SetInputCellDimensionToLines()
  {
    this->SetInputCellDimension(vtkm::filter::contour::ContourDimension::Lines);
  }
 
  VTKM_CONT
  void SetMergeDuplicatePoints(bool on) { this->MergeDuplicatedPoints = on; }
 
  VTKM_CONT
  bool GetMergeDuplicatePoints() const { return this->MergeDuplicatedPoints; }
 
protected:
  template <typename WorkletType>
  VTKM_CONT static bool DoMapField(vtkm::cont::DataSet& result,
                                   const vtkm::cont::Field& field,
                                   WorkletType& worklet)
  {
    if (field.IsPointField())
    {
      vtkm::cont::UnknownArrayHandle inputArray = field.GetData();
      vtkm::cont::UnknownArrayHandle outputArray = inputArray.NewInstanceBasic();
 
      auto functor = [&](const auto& concrete) {
        using ComponentType = typename std::decay_t<decltype(concrete)>::ValueType::ComponentType;
        auto fieldArray = outputArray.ExtractArrayFromComponents<ComponentType>();
        worklet.ProcessPointField(concrete, fieldArray);
      };
      inputArray.CastAndCallWithExtractedArray(functor);
      result.AddPointField(field.GetName(), outputArray);
      return true;
    }
    else if (field.IsCellField())
    {
      // Use the precompiled field permutation function.
      vtkm::cont::ArrayHandle<vtkm::Id> permutation = worklet.GetCellIdMap();
      return vtkm::filter::MapFieldPermutation(field, permutation, result);
    }
    else if (field.IsWholeDataSetField())
    {
      result.AddField(field);
      return true;
    }
    return false;
  }
 
  VTKM_CONT void ExecuteGenerateNormals(vtkm::cont::DataSet& output,
                                        const vtkm::cont::ArrayHandle<vtkm::Vec3f>& normals)
  {
    if (this->GenerateNormals)
    {
      if (this->GetComputeFastNormals())
      {
        vtkm::filter::vector_analysis::SurfaceNormals surfaceNormals;
        surfaceNormals.SetPointNormalsName(this->NormalArrayName);
        surfaceNormals.SetGeneratePointNormals(true);
        output = surfaceNormals.Execute(output);
      }
      else
      {
        output.AddField(vtkm::cont::make_FieldPoint(this->NormalArrayName, normals));
      }
    }
  }
 
  template <typename WorkletType>
  VTKM_CONT void ExecuteAddInterpolationEdgeIds(vtkm::cont::DataSet& output, WorkletType& worklet)
  {
    if (this->AddInterpolationEdgeIds)
    {
      vtkm::cont::Field interpolationEdgeIdsField(this->InterpolationEdgeIdsArrayName,
                                                  vtkm::cont::Field::Association::Points,
                                                  worklet.GetInterpolationEdgeIds());
      output.AddField(interpolationEdgeIdsField);
    }
  }
 
  VTKM_CONT
  virtual vtkm::cont::DataSet DoExecute(
    const vtkm::cont::DataSet& result) = 0; // Needs to be overridden by contour implementations
 
  std::vector<vtkm::Float64> IsoValues;
  bool GenerateNormals = true;
  bool ComputeFastNormals = false;
 
  vtkm::filter::contour::ContourDimension InputCellDimension =
    vtkm::filter::contour::ContourDimension::Auto;
 
  bool AddInterpolationEdgeIds = false;
  bool MergeDuplicatedPoints = true;
  std::string NormalArrayName = "normals";
  std::string InterpolationEdgeIdsArrayName = "edgeIds";
};
} // namespace contour
} // namespace filter
} // namespace vtkm
 
#endif // vtk_m_filter_contour_AbstractContour_h