FieldToColors.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_field_transform_FieldToColors_h
#define viskores_filter_field_transform_FieldToColors_h
 
#include <viskores/cont/ColorTable.h>
#include <viskores/filter/Filter.h>
#include <viskores/filter/field_transform/viskores_filter_field_transform_export.h>
 
namespace viskores
{
namespace filter
{
namespace field_transform
{
class VISKORES_FILTER_FIELD_TRANSFORM_EXPORT FieldToColors : public viskores::filter::Filter
{
public:
  VISKORES_CONT
  explicit FieldToColors(const viskores::cont::ColorTable& table = viskores::cont::ColorTable());
 
  // Documentation of enumerations is behaving a little weird in Doxygen (version 1.9.7).
  // You cannot have a blank line in the documentation. Everything below it will be treated
  // as preformatted text. Also, the first line always seem to behave like `@brief` is used
  // even when it is not. It's easier to just document an associated method and copy the
  // documentation
 
  enum struct InputMode
  {
    Scalar,
    Magnitude,
    Component,
  };
 
  enum struct OutputMode
  {
    RGB,
    RGBA,
  };
 
  void SetColorTable(const viskores::cont::ColorTable& table)
  {
    this->Table = table;
    this->ModifiedCount = -1;
  }
  const viskores::cont::ColorTable& GetColorTable() const { return this->Table; }
 
  void SetMappingMode(InputMode mode) { this->InputModeType = mode; }
  void SetMappingToScalar() { this->InputModeType = InputMode::Scalar; }
  void SetMappingToMagnitude() { this->InputModeType = InputMode::Magnitude; }
  void SetMappingToComponent() { this->InputModeType = InputMode::Component; }
  InputMode GetMappingMode() const { return this->InputModeType; }
  bool IsMappingScalar() const { return this->InputModeType == InputMode::Scalar; }
  bool IsMappingMagnitude() const { return this->InputModeType == InputMode::Magnitude; }
  bool IsMappingComponent() const { return this->InputModeType == InputMode::Component; }
 
  void SetMappingComponent(viskores::IdComponent comp) { this->Component = comp; }
  viskores::IdComponent GetMappingComponent() const { return this->Component; }
 
  void SetOutputMode(OutputMode mode) { this->OutputModeType = mode; }
  void SetOutputToRGB() { this->OutputModeType = OutputMode::RGB; }
  void SetOutputToRGBA() { this->OutputModeType = OutputMode::RGBA; }
  OutputMode GetOutputMode() const { return this->OutputModeType; }
  bool IsOutputRGB() const { return this->OutputModeType == OutputMode::RGB; }
  bool IsOutputRGBA() const { return this->OutputModeType == OutputMode::RGBA; }
 
  void SetNumberOfSamplingPoints(viskores::Int32 count);
  viskores::Int32 GetNumberOfSamplingPoints() const { return this->SampleCount; }
 
private:
  VISKORES_CONT viskores::cont::DataSet DoExecute(const viskores::cont::DataSet& input) override;
 
  viskores::cont::ColorTable Table;
  InputMode InputModeType = InputMode::Scalar;
  OutputMode OutputModeType = OutputMode::RGBA;
  viskores::cont::ColorTableSamplesRGB SamplesRGB;
  viskores::cont::ColorTableSamplesRGBA SamplesRGBA;
  viskores::IdComponent Component = 0;
  viskores::Int32 SampleCount = 256;
  viskores::Id ModifiedCount = -1;
};
 
} // namespace field_transform
} // namespace filter
} // namespace viskores
 
#endif // viskores_filter_field_transform_FieldToColors_h