mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-09-19 18:45:43 +00:00
51e3b2bb1d
It's easy to put accidently put something that is not a valid tag in a ControlSignature or ExecutionSignature. Previously, when you did that you got a weird error at the end of a very long template instantiation chain that made it difficult to find the offending worklet. This adds some type checks when the dispatcher is instantated to check the signatures. It doesn't point directly to the signature or its parameter, but it is much closer.
74 lines
2.6 KiB
C++
74 lines
2.6 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.
|
|
//
|
|
// Copyright 2014 Sandia Corporation.
|
|
// Copyright 2014 UT-Battelle, LLC.
|
|
// Copyright 2014. Los Alamos National Security
|
|
//
|
|
// Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
|
|
// the U.S. Government retains certain rights in this software.
|
|
//
|
|
// Under the terms of Contract DE-AC52-06NA25396 with Los Alamos National
|
|
// Laboratory (LANL), the U.S. Government retains certain rights in
|
|
// this software.
|
|
//============================================================================
|
|
#ifndef vtk_m_worklet_WorkletMapField_h
|
|
#define vtk_m_worklet_WorkletMapField_h
|
|
|
|
#include <vtkm/worklet/internal/WorkletBase.h>
|
|
|
|
#include <vtkm/TypeListTag.h>
|
|
|
|
#include <vtkm/cont/arg/ControlSignatureTagBase.h>
|
|
#include <vtkm/cont/arg/TransportTagArrayIn.h>
|
|
#include <vtkm/cont/arg/TransportTagArrayOut.h>
|
|
#include <vtkm/cont/arg/TypeCheckTagArray.h>
|
|
|
|
#include <vtkm/exec/arg/FetchTagArrayDirectIn.h>
|
|
#include <vtkm/exec/arg/FetchTagArrayDirectOut.h>
|
|
|
|
namespace vtkm {
|
|
namespace worklet {
|
|
|
|
/// Base class for worklets that do a simple mapping of field arrays. All
|
|
/// inputs and outputs are on the same domain. That is, all the arrays are the
|
|
/// same size.
|
|
///
|
|
class WorkletMapField : public vtkm::worklet::internal::WorkletBase
|
|
{
|
|
public:
|
|
/// \brief A control signature tag for input fields.
|
|
///
|
|
/// This tag takes a template argument that is a type list tag that limits
|
|
/// the possible value types in the array.
|
|
///
|
|
template<typename TypeList = AllTypes>
|
|
struct FieldIn : vtkm::cont::arg::ControlSignatureTagBase {
|
|
typedef vtkm::cont::arg::TypeCheckTagArray<TypeList> TypeCheckTag;
|
|
typedef vtkm::cont::arg::TransportTagArrayIn TransportTag;
|
|
typedef vtkm::exec::arg::FetchTagArrayDirectIn FetchTag;
|
|
};
|
|
|
|
/// \brief A control signature tag for output fields.
|
|
///
|
|
/// This tag takes a template argument that is a type list tag that limits
|
|
/// the possible value types in the array.
|
|
///
|
|
template<typename TypeList = AllTypes>
|
|
struct FieldOut : vtkm::cont::arg::ControlSignatureTagBase {
|
|
typedef vtkm::cont::arg::TypeCheckTagArray<TypeList> TypeCheckTag;
|
|
typedef vtkm::cont::arg::TransportTagArrayOut TransportTag;
|
|
typedef vtkm::exec::arg::FetchTagArrayDirectOut FetchTag;
|
|
};
|
|
};
|
|
|
|
}
|
|
} // namespace vtkm::worklet
|
|
|
|
#endif //vtk_m_worklet_WorkletMapField_h
|