mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-09-08 21:33:55 +00:00
42bc9a393c
With recent changes to allow a configuration to change the default types, storage, and cell sets, it is possible to feed filters and other components types they were not previously expecting. Fix feature gaps where these components were not accepting the types they should.
46 lines
1.2 KiB
C++
46 lines
1.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_DotProduct_h
|
|
#define vtk_m_worklet_DotProduct_h
|
|
|
|
#include <vtkm/worklet/WorkletMapField.h>
|
|
|
|
#include <vtkm/Math.h>
|
|
#include <vtkm/VectorAnalysis.h>
|
|
|
|
namespace vtkm
|
|
{
|
|
namespace worklet
|
|
{
|
|
|
|
class DotProduct : public vtkm::worklet::WorkletMapField
|
|
{
|
|
public:
|
|
using ControlSignature = void(FieldIn, FieldIn, FieldOut);
|
|
|
|
template <typename T, vtkm::IdComponent Size>
|
|
VTKM_EXEC void operator()(const vtkm::Vec<T, Size>& v1,
|
|
const vtkm::Vec<T, Size>& v2,
|
|
T& outValue) const
|
|
{
|
|
outValue = static_cast<T>(vtkm::Dot(v1, v2));
|
|
}
|
|
|
|
template <typename T>
|
|
VTKM_EXEC void operator()(T s1, T s2, T& outValue) const
|
|
{
|
|
outValue = static_cast<T>(s1 * s2);
|
|
}
|
|
};
|
|
}
|
|
} // namespace vtkm::worklet
|
|
|
|
#endif // vtk_m_worklet_Normalize_h
|