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.
60 lines
2.0 KiB
C++
60 lines
2.0 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_filter_DotProduct_hxx
|
|
#define vtk_m_filter_DotProduct_hxx
|
|
|
|
#include <vtkm/cont/ArrayHandleCast.h>
|
|
|
|
namespace vtkm
|
|
{
|
|
namespace filter
|
|
{
|
|
|
|
//-----------------------------------------------------------------------------
|
|
inline VTKM_CONT DotProduct::DotProduct()
|
|
: vtkm::filter::FilterField<DotProduct>()
|
|
, SecondaryFieldName()
|
|
, SecondaryFieldAssociation(vtkm::cont::Field::Association::ANY)
|
|
, UseCoordinateSystemAsSecondaryField(false)
|
|
, SecondaryCoordinateSystemIndex(0)
|
|
{
|
|
this->SetOutputFieldName("dotproduct");
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
template <typename T, typename StorageType, typename DerivedPolicy>
|
|
inline VTKM_CONT vtkm::cont::DataSet DotProduct::DoExecute(
|
|
const vtkm::cont::DataSet& inDataSet,
|
|
const vtkm::cont::ArrayHandle<T, StorageType>& primary,
|
|
const vtkm::filter::FieldMetadata& fieldMetadata,
|
|
vtkm::filter::PolicyBase<DerivedPolicy> policy)
|
|
{
|
|
vtkm::cont::Field secondaryField;
|
|
if (this->UseCoordinateSystemAsSecondaryField)
|
|
{
|
|
secondaryField = inDataSet.GetCoordinateSystem(this->GetSecondaryCoordinateSystemIndex());
|
|
}
|
|
else
|
|
{
|
|
secondaryField = inDataSet.GetField(this->SecondaryFieldName, this->SecondaryFieldAssociation);
|
|
}
|
|
auto secondary = vtkm::filter::ApplyPolicyFieldOfType<T>(secondaryField, policy, *this);
|
|
|
|
vtkm::cont::ArrayHandle<typename vtkm::VecTraits<T>::ComponentType> output;
|
|
this->Invoke(vtkm::worklet::DotProduct{}, primary, secondary, output);
|
|
|
|
return CreateResult(inDataSet, output, this->GetOutputFieldName(), fieldMetadata);
|
|
}
|
|
}
|
|
} // namespace vtkm::filter
|
|
|
|
#endif
|