AbstractContour.h

Go to the documentation of this file.

//============================================================================
//  The contents of this file are covered by the Viskores license. See
//  LICENSE.txt for details.
//
//  By contributing to this file, all contributors agree to the Developer
//  Certificate of Origin Version 1.1 (DCO 1.1) as stated in DCO.txt.
//============================================================================
 
//============================================================================
//  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 viskores_filter_contour_AbstractContour_h
#define viskores_filter_contour_AbstractContour_h
 
#include <viskores/filter/Filter.h>
#include <viskores/filter/MapFieldPermutation.h>
#include <viskores/filter/contour/ContourDimension.h>
#include <viskores/filter/contour/viskores_filter_contour_export.h>
#include <viskores/filter/vector_analysis/SurfaceNormals.h>
 
namespace viskores
{
namespace filter
{
namespace contour
{
class VISKORES_FILTER_CONTOUR_EXPORT AbstractContour : public viskores::filter::Filter
{
public:
  void SetNumberOfIsoValues(viskores::Id num)
  {
    if (num >= 0)
    {
      this->IsoValues.resize(static_cast<std::size_t>(num));
    }
  }
 
  viskores::Id GetNumberOfIsoValues() const
  {
    return static_cast<viskores::Id>(this->IsoValues.size());
  }
 
  void SetIsoValue(viskores::Float64 v) { this->SetIsoValue(0, v); }
 
  void SetIsoValue(viskores::Id index, viskores::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<viskores::Float64>& values) { this->IsoValues = values; }
 
  viskores::Float64 GetIsoValue(viskores::Id index = 0) const
  {
    return this->IsoValues[static_cast<std::size_t>(index)];
  }
 
  VISKORES_CONT
  void SetGenerateNormals(bool flag) { this->GenerateNormals = flag; }
  VISKORES_CONT
  bool GetGenerateNormals() const { return this->GenerateNormals; }
 
  VISKORES_CONT
  void SetAddInterpolationEdgeIds(bool flag) { this->AddInterpolationEdgeIds = flag; }
  VISKORES_CONT
  bool GetAddInterpolationEdgeIds() const { return this->AddInterpolationEdgeIds; }
 
  VISKORES_CONT
  void SetComputeFastNormals(bool flag) { this->ComputeFastNormals = flag; }
  VISKORES_CONT
  bool GetComputeFastNormals() const { return this->ComputeFastNormals; }
 
  VISKORES_CONT
  void SetNormalArrayName(const std::string& name) { this->NormalArrayName = name; }
 
  VISKORES_CONT
  const std::string& GetNormalArrayName() const { return this->NormalArrayName; }
 
  VISKORES_CONT void SetInputCellDimension(viskores::filter::contour::ContourDimension dimension)
  {
    this->InputCellDimension = dimension;
  }
 
  VISKORES_CONT viskores::filter::contour::ContourDimension GetInputCellDimension() const
  {
    return this->InputCellDimension;
  }
 
  VISKORES_CONT void SetInputCellDimensionToAuto()
  {
    this->SetInputCellDimension(viskores::filter::contour::ContourDimension::Auto);
  }
 
  VISKORES_CONT void SetInputCellDimensionToAll()
  {
    this->SetInputCellDimension(viskores::filter::contour::ContourDimension::All);
  }
 
  VISKORES_CONT void SetInputCellDimensionToPolyhedra()
  {
    this->SetInputCellDimension(viskores::filter::contour::ContourDimension::Polyhedra);
  }
 
  VISKORES_CONT void SetInputCellDimensionToPolygons()
  {
    this->SetInputCellDimension(viskores::filter::contour::ContourDimension::Polygons);
  }
 
  VISKORES_CONT void SetInputCellDimensionToLines()
  {
    this->SetInputCellDimension(viskores::filter::contour::ContourDimension::Lines);
  }
 
  VISKORES_CONT
  void SetMergeDuplicatePoints(bool on) { this->MergeDuplicatedPoints = on; }
 
  VISKORES_CONT
  bool GetMergeDuplicatePoints() const { return this->MergeDuplicatedPoints; }
 
protected:
  template <typename WorkletType>
  VISKORES_CONT static bool DoMapField(viskores::cont::DataSet& result,
                                       const viskores::cont::Field& field,
                                       WorkletType& worklet)
  {
    if (field.IsPointField())
    {
      viskores::cont::UnknownArrayHandle inputArray = field.GetData();
      viskores::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.
      viskores::cont::ArrayHandle<viskores::Id> permutation = worklet.GetCellIdMap();
      return viskores::filter::MapFieldPermutation(field, permutation, result);
    }
    else if (field.IsWholeDataSetField())
    {
      result.AddField(field);
      return true;
    }
    return false;
  }
 
  VISKORES_CONT void ExecuteGenerateNormals(
    viskores::cont::DataSet& output,
    const viskores::cont::ArrayHandle<viskores::Vec3f>& normals)
  {
    if (this->GenerateNormals)
    {
      if (this->GetComputeFastNormals())
      {
        viskores::filter::vector_analysis::SurfaceNormals surfaceNormals;
        surfaceNormals.SetPointNormalsName(this->NormalArrayName);
        surfaceNormals.SetGeneratePointNormals(true);
        output = surfaceNormals.Execute(output);
      }
      else
      {
        output.AddField(viskores::cont::make_FieldPoint(this->NormalArrayName, normals));
      }
    }
  }
 
  template <typename WorkletType>
  VISKORES_CONT void ExecuteAddInterpolationEdgeIds(viskores::cont::DataSet& output,
                                                    WorkletType& worklet)
  {
    if (this->AddInterpolationEdgeIds)
    {
      viskores::cont::Field interpolationEdgeIdsField(this->InterpolationEdgeIdsArrayName,
                                                      viskores::cont::Field::Association::Points,
                                                      worklet.GetInterpolationEdgeIds());
      output.AddField(interpolationEdgeIdsField);
    }
  }
 
  VISKORES_CONT
  virtual viskores::cont::DataSet DoExecute(
    const viskores::cont::DataSet& result) = 0; // Needs to be overridden by contour implementations
 
  std::vector<viskores::Float64> IsoValues;
  bool GenerateNormals = true;
  bool ComputeFastNormals = false;
 
  viskores::filter::contour::ContourDimension InputCellDimension =
    viskores::filter::contour::ContourDimension::Auto;
 
  bool AddInterpolationEdgeIds = false;
  bool MergeDuplicatedPoints = true;
  std::string NormalArrayName = "normals";
  std::string InterpolationEdgeIdsArrayName = "edgeIds";
};
} // namespace contour
} // namespace filter
} // namespace viskores
 
#endif // viskores_filter_contour_AbstractContour_h