//============================================================================ // 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_CellAverage_h #define vtk_m_worklet_CellAverage_h #include namespace vtkm { namespace worklet { //simple functor that returns the average point value. class CellAverage : public vtkm::worklet::WorkletMapTopologyPointToCell { public: typedef void ControlSignature(FieldInFrom inPoints, TopologyIn topology, FieldOut outCells); typedef void ExecutionSignature(_1, FromCount, _3); typedef _2 InputDomain; template VTKM_EXEC_EXPORT void operator()(const PointValueVecType &pointValues, const vtkm::IdComponent &numPoints, OutType &average) const { OutType sum = static_cast(pointValues[0]); for (vtkm::IdComponent pointIndex = 1; pointIndex < numPoints; ++pointIndex) { sum = sum + static_cast(pointValues[pointIndex]); } average = sum / static_cast(numPoints); } }; } } // namespace vtkm::worklet #endif // vtk_m_worklet_CellAverage_h