vtk-m/vtkm/rendering/raytracing/GlyphExtractorVector.h

84 lines
2.7 KiB
C
Raw Permalink Normal View History

//============================================================================
// 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_rendering_raytracing_Glyph_Extractor_Vector_h
#define vtk_m_rendering_raytracing_Glyph_Extractor_Vector_h
#include <vtkm/cont/DataSet.h>
namespace vtkm
{
namespace rendering
{
namespace raytracing
{
class GlyphExtractorVector
{
public:
GlyphExtractorVector();
//
// Extract all nodes using a constant size
//
void ExtractCoordinates(const vtkm::cont::CoordinateSystem& coords,
const vtkm::cont::Field& field,
const vtkm::Float32 size);
//
// Set size based on scalar field values. Each is interpolated from min to max
//
void ExtractCoordinates(const vtkm::cont::CoordinateSystem& coords,
const vtkm::cont::Field& field,
const vtkm::Float32 minSize,
const vtkm::Float32 maxSize);
//
// Extract all vertex shapes with constant size
//
void ExtractCells(const vtkm::cont::UnknownCellSet& cells,
const vtkm::cont::Field& field,
vtkm::Float32 size);
//
// Extract all vertex elements with size based on scalar values
//
void ExtractCells(const vtkm::cont::UnknownCellSet& cells,
const vtkm::cont::Field& field,
const vtkm::Float32 minSize,
const vtkm::Float32 maxSize);
vtkm::cont::ArrayHandle<vtkm::Id> GetPointIds();
vtkm::cont::ArrayHandle<vtkm::Vec<vtkm::Float32, 3>> GetSizes();
vtkm::cont::Field GetMagnitudeField();
vtkm::Id GetNumberOfGlyphs() const;
protected:
void SetUniformSize(const vtkm::Float32 size, const vtkm::cont::Field& field);
void SetVaryingSize(const vtkm::Float32 minSize,
const vtkm::Float32 maxSize,
const vtkm::cont::Field& field);
void SetPointIdsFromCoords(const vtkm::cont::CoordinateSystem& coords);
void SetPointIdsFromCells(const vtkm::cont::UnknownCellSet& cells);
void ExtractMagnitudeField(const vtkm::cont::Field& field);
vtkm::cont::ArrayHandle<vtkm::Id> PointIds;
vtkm::cont::ArrayHandle<vtkm::Vec<vtkm::Float32, 3>> Sizes;
vtkm::cont::Field MagnitudeField;
}; // class GlyphExtractorVector
}
}
} //namespace vtkm::rendering::raytracing
#endif //vtk_m_rendering_raytracing_Glyph_Extractor_Vector_h