mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-09-19 10:35:42 +00:00
106 lines
3.2 KiB
C++
106 lines
3.2 KiB
C++
//============================================================================
|
|
// 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_worklet_colorconversion_LookupTable_h
|
|
#define vtk_m_worklet_colorconversion_LookupTable_h
|
|
|
|
#include <vtkm/cont/ColorTableSamples.h>
|
|
|
|
#include <vtkm/exec/ColorTable.h>
|
|
#include <vtkm/worklet/WorkletMapField.h>
|
|
#include <vtkm/worklet/colorconversion/Conversions.h>
|
|
|
|
#include <float.h>
|
|
|
|
namespace vtkm
|
|
{
|
|
namespace worklet
|
|
{
|
|
namespace colorconversion
|
|
{
|
|
|
|
using LookupTableTypes = vtkm::List<vtkm::Vec3ui_8, vtkm::Vec4ui_8, vtkm::Vec3f_32, vtkm::Vec4f_64>;
|
|
|
|
struct LookupTable : public vtkm::worklet::WorkletMapField
|
|
{
|
|
vtkm::Float32 Shift;
|
|
vtkm::Float32 Scale;
|
|
vtkm::Range TableRange;
|
|
vtkm::Int32 NumberOfSamples;
|
|
|
|
//needs to support Nan, Above, Below Range colors
|
|
VTKM_CONT
|
|
template <typename T>
|
|
LookupTable(const T& colorTableSamples)
|
|
{
|
|
this->Shift = static_cast<vtkm::Float32>(-colorTableSamples.SampleRange.Min);
|
|
double rangeDelta = colorTableSamples.SampleRange.Length();
|
|
if (rangeDelta < DBL_MIN * colorTableSamples.NumberOfSamples)
|
|
{
|
|
// if the range is tiny, anything within the range will map to the bottom
|
|
// of the color scale.
|
|
this->Scale = 0.0;
|
|
}
|
|
else
|
|
{
|
|
this->Scale = static_cast<vtkm::Float32>(colorTableSamples.NumberOfSamples / rangeDelta);
|
|
}
|
|
this->TableRange = colorTableSamples.SampleRange;
|
|
this->NumberOfSamples = colorTableSamples.NumberOfSamples;
|
|
}
|
|
|
|
using ControlSignature = void(FieldIn in, WholeArrayIn lookup, FieldOut color);
|
|
using ExecutionSignature = void(_1, _2, _3);
|
|
|
|
template <typename T, typename WholeFieldIn, typename U, int N>
|
|
VTKM_EXEC void operator()(const T& in,
|
|
const WholeFieldIn lookupTable,
|
|
vtkm::Vec<U, N>& output) const
|
|
{
|
|
vtkm::Float64 v = (static_cast<vtkm::Float64>(in));
|
|
vtkm::Int32 idx = 1;
|
|
|
|
//This logic uses how ColorTableSamples is constructed. See
|
|
//vtkm/cont/ColorTableSamples to see why we use these magic offset values
|
|
if (vtkm::IsNan(v))
|
|
{
|
|
idx = this->NumberOfSamples + 3;
|
|
}
|
|
else if (v < this->TableRange.Min)
|
|
{ //If we are below the color range
|
|
idx = 0;
|
|
}
|
|
else if (v == this->TableRange.Min)
|
|
{ //If we are at the ranges min value
|
|
idx = 1;
|
|
}
|
|
else if (v > this->TableRange.Max)
|
|
{ //If we are above the ranges max value
|
|
idx = this->NumberOfSamples + 2;
|
|
}
|
|
else if (v == this->TableRange.Max)
|
|
{ //If we are at the ranges min value
|
|
idx = this->NumberOfSamples;
|
|
}
|
|
else
|
|
{
|
|
v = (v + this->Shift) * this->Scale;
|
|
// When v is very close to p.Range[1], the floating point calculation giving
|
|
// idx may map above the highest value in the lookup table. That is why it
|
|
// is padded
|
|
idx = 1 + static_cast<vtkm::Int32>(v);
|
|
}
|
|
output = lookupTable.Get(idx);
|
|
}
|
|
};
|
|
}
|
|
}
|
|
}
|
|
#endif
|