mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-10-05 09:59:12 +00:00
9d5d9e38a1
The code in `vtkm/cont/Testing.h` now requires a library, which is not built if testing is not built. Thus, the benchmarking code was giving a compile error if benchmarking was on but testing was off. Change the benchmarking to not rely on anything in the Testing framework. This means using classes in `vtkm/source` instead of `MakeTestData`. Also avoid using the `TestValue` defined for the tests. (In one case, we have a simple replacement.) Also had to fix a problem with a header file not defining everything it needed to compile.
166 lines
6.2 KiB
C++
166 lines
6.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_exec_arg_FetchExtrude_h
|
|
#define vtk_m_exec_arg_FetchExtrude_h
|
|
|
|
#include <vtkm/exec/ConnectivityExtrude.h>
|
|
#include <vtkm/exec/arg/FetchTagArrayDirectIn.h>
|
|
#include <vtkm/exec/arg/FetchTagArrayTopologyMapIn.h>
|
|
|
|
#include <vtkm/cont/ArrayHandleXGCCoordinates.h>
|
|
|
|
//optimized fetches for ArrayPortalXGCCoordinates for
|
|
// - 3D Scheduling
|
|
// - WorkletNeighboorhood
|
|
namespace vtkm
|
|
{
|
|
namespace exec
|
|
{
|
|
namespace arg
|
|
{
|
|
|
|
/// \brief Aspect tag to use for getting the visited indices.
|
|
///
|
|
/// The \c AspectTagIncidentElementIndices aspect tag causes the \c Fetch class
|
|
/// to obtain the indices that map to the current topology element.
|
|
///
|
|
struct AspectTagIncidentElementIndices
|
|
{
|
|
};
|
|
|
|
//Optimized fetch for point ids when iterating the cells ConnectivityExtrude
|
|
template <typename FetchType, typename ExecObjectType>
|
|
struct Fetch<FetchType, vtkm::exec::arg::AspectTagIncidentElementIndices, ExecObjectType>
|
|
{
|
|
VTKM_SUPPRESS_EXEC_WARNINGS
|
|
template <typename ScatterAndMaskMode>
|
|
VTKM_EXEC auto Load(
|
|
const vtkm::exec::arg::ThreadIndicesTopologyMap<vtkm::exec::ConnectivityExtrude,
|
|
ScatterAndMaskMode>& indices,
|
|
const ExecObjectType&) const -> vtkm::Vec<vtkm::Id, 6>
|
|
{
|
|
// std::cout << "opimized fetch for point ids" << std::endl;
|
|
const auto& xgcidx = indices.GetIndicesIncident();
|
|
const vtkm::Id offset1 = (xgcidx.Planes[0] * xgcidx.NumberOfPointsPerPlane);
|
|
const vtkm::Id offset2 = (xgcidx.Planes[1] * xgcidx.NumberOfPointsPerPlane);
|
|
vtkm::Vec<vtkm::Id, 6> result;
|
|
result[0] = offset1 + xgcidx.PointIds[0][0];
|
|
result[1] = offset1 + xgcidx.PointIds[0][1];
|
|
result[2] = offset1 + xgcidx.PointIds[0][2];
|
|
result[3] = offset2 + xgcidx.PointIds[1][0];
|
|
result[4] = offset2 + xgcidx.PointIds[1][1];
|
|
result[5] = offset2 + xgcidx.PointIds[1][2];
|
|
return result;
|
|
}
|
|
|
|
VTKM_SUPPRESS_EXEC_WARNINGS
|
|
template <typename ConnectivityType, typename ScatterAndMaskMode>
|
|
VTKM_EXEC auto Load(
|
|
const vtkm::exec::arg::ThreadIndicesTopologyMap<ConnectivityType, ScatterAndMaskMode>& indices,
|
|
const ExecObjectType&) const -> decltype(indices.GetIndicesIncident())
|
|
{
|
|
return indices.GetIndicesIncident();
|
|
}
|
|
|
|
template <typename ThreadIndicesType, typename ValueType>
|
|
VTKM_EXEC void Store(const ThreadIndicesType&, const ExecObjectType&, const ValueType&) const
|
|
{
|
|
// Store is a no-op.
|
|
}
|
|
};
|
|
|
|
//Optimized fetch for point coordinates when iterating the cells of ConnectivityExtrude
|
|
template <typename T>
|
|
struct Fetch<vtkm::exec::arg::FetchTagArrayTopologyMapIn,
|
|
vtkm::exec::arg::AspectTagDefault,
|
|
vtkm::internal::ArrayPortalXGCCoordinates<T>>
|
|
|
|
{
|
|
//Optimized fetch for point arrays when iterating the cells ConnectivityExtrude
|
|
VTKM_SUPPRESS_EXEC_WARNINGS
|
|
template <typename ScatterAndMaskMode>
|
|
VTKM_EXEC auto Load(
|
|
const vtkm::exec::arg::ThreadIndicesTopologyMap<vtkm::exec::ConnectivityExtrude,
|
|
ScatterAndMaskMode>& indices,
|
|
const vtkm::internal::ArrayPortalXGCCoordinates<T>& portal)
|
|
-> decltype(portal.GetWedge(indices.GetIndicesIncident()))
|
|
{
|
|
return portal.GetWedge(indices.GetIndicesIncident());
|
|
}
|
|
|
|
VTKM_SUPPRESS_EXEC_WARNINGS
|
|
template <typename ThreadIndicesType>
|
|
VTKM_EXEC auto Load(const ThreadIndicesType& indices,
|
|
const vtkm::internal::ArrayPortalXGCCoordinates<T>& field) const
|
|
-> decltype(
|
|
detail::FetchArrayTopologyMapInImplementation<typename ThreadIndicesType::Connectivity,
|
|
vtkm::internal::ArrayPortalXGCCoordinates<T>,
|
|
ThreadIndicesType>::Load(indices, field))
|
|
{
|
|
using Implementation =
|
|
detail::FetchArrayTopologyMapInImplementation<typename ThreadIndicesType::Connectivity,
|
|
vtkm::internal::ArrayPortalXGCCoordinates<T>,
|
|
ThreadIndicesType>;
|
|
return Implementation::Load(indices, field);
|
|
}
|
|
|
|
template <typename ThreadIndicesType, typename ValueType>
|
|
VTKM_EXEC void Store(const ThreadIndicesType&,
|
|
const vtkm::internal::ArrayPortalXGCCoordinates<T>&,
|
|
const ValueType&) const
|
|
{
|
|
// Store is a no-op for this fetch.
|
|
}
|
|
};
|
|
|
|
//Optimized fetch for point coordinates when iterating the points of ConnectivityExtrude
|
|
template <typename T>
|
|
struct Fetch<vtkm::exec::arg::FetchTagArrayDirectIn,
|
|
vtkm::exec::arg::AspectTagDefault,
|
|
vtkm::internal::ArrayPortalXGCCoordinates<T>>
|
|
|
|
{
|
|
VTKM_SUPPRESS_EXEC_WARNINGS
|
|
template <typename ThreadIndicesType>
|
|
VTKM_EXEC auto Load(const ThreadIndicesType& indices,
|
|
const vtkm::internal::ArrayPortalXGCCoordinates<T>& points)
|
|
-> decltype(points.Get(indices.GetInputIndex()))
|
|
{
|
|
// std::cout << "optimized fetch for point coordinates" << std::endl;
|
|
return points.Get(indices.GetInputIndex());
|
|
}
|
|
|
|
VTKM_SUPPRESS_EXEC_WARNINGS
|
|
template <typename ScatterAndMaskMode>
|
|
VTKM_EXEC auto Load(
|
|
const vtkm::exec::arg::ThreadIndicesTopologyMap<vtkm::exec::ReverseConnectivityExtrude,
|
|
ScatterAndMaskMode>& indices,
|
|
const vtkm::internal::ArrayPortalXGCCoordinates<T>& points)
|
|
-> decltype(points.Get(indices.GetIndexLogical()))
|
|
{
|
|
// std::cout << "optimized fetch for point coordinates" << std::endl;
|
|
return points.Get(indices.GetIndexLogical());
|
|
}
|
|
|
|
template <typename ThreadIndicesType, typename ValueType>
|
|
VTKM_EXEC void Store(const ThreadIndicesType&,
|
|
const vtkm::internal::ArrayPortalXGCCoordinates<T>&,
|
|
const ValueType&) const
|
|
{
|
|
// Store is a no-op for this fetch.
|
|
}
|
|
};
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
#endif
|