vtk-m/vtkm/cont/Field.h

527 lines
14 KiB
C
Raw Normal View History

2015-02-10 17:36:10 +00:00
//============================================================================
// 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.
2015-02-10 17:36:10 +00:00
//
// 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_cont_Field_h
#define vtk_m_cont_Field_h
2015-02-10 17:36:10 +00:00
2015-07-16 19:10:01 +00:00
#include <vtkm/Math.h>
#include <vtkm/Range.h>
#include <vtkm/Types.h>
2015-07-16 19:10:01 +00:00
#include <vtkm/VecTraits.h>
#include <vtkm/BinaryOperators.h>
2015-07-16 19:10:01 +00:00
2015-02-10 17:36:10 +00:00
#include <vtkm/cont/ArrayHandle.h>
2015-07-16 19:10:01 +00:00
#include <vtkm/cont/ArrayHandleTransform.h>
#include <vtkm/cont/ArrayHandleUniformPointCoordinates.h>
2015-07-16 19:10:01 +00:00
#include <vtkm/cont/DeviceAdapterAlgorithm.h>
2015-02-10 17:36:10 +00:00
#include <vtkm/cont/DynamicArrayHandle.h>
#include <vtkm/cont/internal/ArrayPortalFromIterators.h>
2015-02-10 17:36:10 +00:00
namespace vtkm {
namespace cont {
2015-07-16 19:10:01 +00:00
namespace internal {
template<typename DeviceAdapterTag>
class ComputeRange
2015-07-16 19:10:01 +00:00
{
public:
ComputeRange(ArrayHandle<vtkm::Range>& range) : Range(&range) {}
2015-07-16 19:10:01 +00:00
template<typename ArrayHandleType>
void operator()(const ArrayHandleType &input) const
{
typedef typename ArrayHandleType::ValueType ValueType;
typedef vtkm::VecTraits<ValueType> VecType;
const vtkm::IdComponent NumberOfComponents = VecType::NUM_COMPONENTS;
2015-07-16 19:10:01 +00:00
typedef vtkm::cont::DeviceAdapterAlgorithm<DeviceAdapterTag> Algorithm;
2015-07-16 19:10:01 +00:00
//not the greatest way of doing this for performance reasons. But
//this implementation should generate the smallest amount of code
const vtkm::Pair<ValueType, ValueType> initial(
input.GetPortalConstControl().Get(0),
input.GetPortalConstControl().Get(0));
2015-07-16 19:10:01 +00:00
vtkm::Pair<ValueType, ValueType> result =
Algorithm::Reduce(input, initial, vtkm::MinAndMax<ValueType>());
2015-07-16 19:10:01 +00:00
this->Range->Allocate(NumberOfComponents);
for (vtkm::IdComponent i = 0; i < NumberOfComponents; ++i)
{
this->Range->GetPortalControl().Set(
i, vtkm::Range(VecType::GetComponent(result.first, i),
VecType::GetComponent(result.second, i)));
}
2015-07-16 19:10:01 +00:00
}
// Special implementation for regular point coordinates, which are easy
// to determine.
void operator()(const vtkm::cont::ArrayHandle<
vtkm::Vec<vtkm::FloatDefault,3>,
vtkm::cont::ArrayHandleUniformPointCoordinates::StorageTag>
&array)
2015-07-16 19:10:01 +00:00
{
vtkm::internal::ArrayPortalUniformPointCoordinates portal =
array.GetPortalConstControl();
// In this portal we know that the min value is the first entry and the
// max value is the last entry.
vtkm::Vec<vtkm::FloatDefault,3> minimum = portal.Get(0);
vtkm::Vec<vtkm::FloatDefault,3> maximum =
portal.Get(portal.GetNumberOfValues()-1);
this->Range->Allocate(3);
vtkm::cont::ArrayHandle<vtkm::Range>::PortalControl outPortal =
this->Range->GetPortalControl();
outPortal.Set(0, vtkm::Range(minimum[0], maximum[0]));
outPortal.Set(1, vtkm::Range(minimum[1], maximum[1]));
outPortal.Set(2, vtkm::Range(minimum[2], maximum[2]));
2015-07-16 19:10:01 +00:00
}
private:
vtkm::cont::ArrayHandle<vtkm::Range> *Range;
2015-07-16 19:10:01 +00:00
};
} // namespace internal
2015-02-10 17:36:10 +00:00
/// A \c Field encapsulates an array on some piece of the mesh, such as
/// the points, a cell set, a point logical dimension, or the whole mesh.
2015-02-10 17:36:10 +00:00
///
class Field
{
public:
2015-05-11 20:15:58 +00:00
enum AssociationEnum
{
ASSOC_ANY,
ASSOC_WHOLE_MESH,
ASSOC_POINTS,
ASSOC_CELL_SET,
ASSOC_LOGICAL_DIM
};
/// constructors for points / whole mesh
VTKM_CONT
2015-07-30 14:22:25 +00:00
Field(std::string name,
AssociationEnum association,
const vtkm::cont::DynamicArrayHandle &data)
2015-07-30 14:22:25 +00:00
: Name(name),
Association(association),
AssocCellSetName(),
AssocLogicalDim(-1),
Data(data),
Range(),
2015-07-30 14:22:25 +00:00
ModifiedFlag(true)
{
VTKM_ASSERT(this->Association == ASSOC_WHOLE_MESH ||
this->Association == ASSOC_POINTS);
2015-07-30 14:22:25 +00:00
}
template<typename T, typename Storage>
VTKM_CONT
Field(std::string name,
AssociationEnum association,
const ArrayHandle<T, Storage> &data)
: Name(name),
Association(association),
AssocCellSetName(),
AssocLogicalDim(-1),
Data(data),
Range(),
ModifiedFlag(true)
2015-05-11 20:15:58 +00:00
{
VTKM_ASSERT((this->Association == ASSOC_WHOLE_MESH) ||
(this->Association == ASSOC_POINTS));
2015-05-11 20:15:58 +00:00
}
template <typename T>
VTKM_CONT
Field(std::string name,
AssociationEnum association,
const std::vector<T> &data)
: Name(name),
Association(association),
AssocCellSetName(),
AssocLogicalDim(-1),
Range(),
ModifiedFlag(true)
{
VTKM_ASSERT((this->Association == ASSOC_WHOLE_MESH) ||
(this->Association == ASSOC_POINTS));
this->CopyData(&data[0], static_cast<vtkm::Id>(data.size()));
}
template <typename T>
VTKM_CONT
Field(std::string name,
AssociationEnum association,
const T *data,
vtkm::Id nvals)
: Name(name),
Association(association),
AssocCellSetName(),
AssocLogicalDim(-1),
Range(),
ModifiedFlag(true)
{
VTKM_ASSERT((this->Association == ASSOC_WHOLE_MESH) ||
(this->Association == ASSOC_POINTS));
this->CopyData(data, nvals);
}
/// constructors for cell set associations
VTKM_CONT
Field(std::string name,
AssociationEnum association,
const std::string& cellSetName,
const vtkm::cont::DynamicArrayHandle &data)
: Name(name),
Association(association),
AssocCellSetName(cellSetName),
AssocLogicalDim(-1),
Data(data),
Range(),
ModifiedFlag(true)
{
VTKM_ASSERT(this->Association == ASSOC_CELL_SET);
}
2015-07-30 14:22:25 +00:00
template <typename T, typename Storage>
VTKM_CONT
Field(std::string name,
AssociationEnum association,
const std::string& cellSetName,
const vtkm::cont::ArrayHandle<T, Storage> &data)
: Name(name),
Association(association),
AssocCellSetName(cellSetName),
AssocLogicalDim(-1),
Data(data),
Range(),
ModifiedFlag(true)
2015-05-11 20:15:58 +00:00
{
VTKM_ASSERT(this->Association == ASSOC_CELL_SET);
2015-05-11 20:15:58 +00:00
}
template <typename T>
VTKM_CONT
Field(std::string name,
AssociationEnum association,
const std::string& cellSetName,
const std::vector<T> &data)
: Name(name),
Association(association),
AssocCellSetName(cellSetName),
AssocLogicalDim(-1),
Range(),
ModifiedFlag(true)
{
VTKM_ASSERT(this->Association == ASSOC_CELL_SET);
this->CopyData(&data[0], static_cast<vtkm::Id>(data.size()));
}
template <typename T>
VTKM_CONT
Field(std::string name,
AssociationEnum association,
const std::string& cellSetName,
const T *data,
vtkm::Id nvals)
: Name(name),
Association(association),
AssocCellSetName(cellSetName),
AssocLogicalDim(-1),
Range(),
ModifiedFlag(true)
{
VTKM_ASSERT(this->Association == ASSOC_CELL_SET);
this->CopyData(data, nvals);
}
/// constructors for logical dimension associations
VTKM_CONT
Field(std::string name,
AssociationEnum association,
vtkm::IdComponent logicalDim,
const vtkm::cont::DynamicArrayHandle &data)
: Name(name),
Association(association),
AssocCellSetName(),
AssocLogicalDim(logicalDim),
Data(data),
Range(),
ModifiedFlag(true)
{
VTKM_ASSERT(this->Association == ASSOC_LOGICAL_DIM);
}
2015-07-30 14:22:25 +00:00
template <typename T, typename Storage>
VTKM_CONT
Field(std::string name,
AssociationEnum association,
vtkm::IdComponent logicalDim,
const vtkm::cont::ArrayHandle<T, Storage> &data)
: Name(name),
Association(association),
AssocLogicalDim(logicalDim),
Data(data),
Range(),
ModifiedFlag(true)
2015-05-11 20:15:58 +00:00
{
VTKM_ASSERT(this->Association == ASSOC_LOGICAL_DIM);
2015-05-11 20:15:58 +00:00
}
template <typename T>
VTKM_CONT
Field(std::string name,
AssociationEnum association,
vtkm::IdComponent logicalDim,
const std::vector<T> &data)
: Name(name),
Association(association),
AssocLogicalDim(logicalDim),
Range(),
ModifiedFlag(true)
{
VTKM_ASSERT(this->Association == ASSOC_LOGICAL_DIM);
this->CopyData(&data[0], static_cast<vtkm::Id>(data.size()));
}
template <typename T>
VTKM_CONT
Field(std::string name,
AssociationEnum association,
vtkm::IdComponent logicalDim,
const T *data, vtkm::Id nvals)
: Name(name),
Association(association),
AssocLogicalDim(logicalDim),
Range(),
ModifiedFlag(true)
{
VTKM_ASSERT(this->Association == ASSOC_LOGICAL_DIM);
CopyData(data, nvals);
}
VTKM_CONT
Field()
: Name(),
Association(ASSOC_ANY),
AssocCellSetName(),
AssocLogicalDim(),
Data(),
Range(),
ModifiedFlag(true)
{
//Generate an empty field
}
VTKM_CONT
const std::string &GetName() const
2015-05-19 20:21:15 +00:00
{
return this->Name;
2015-05-19 20:21:15 +00:00
}
VTKM_CONT
AssociationEnum GetAssociation() const
2015-05-11 20:15:58 +00:00
{
return this->Association;
2015-05-11 20:15:58 +00:00
}
VTKM_CONT
std::string GetAssocCellSet() const
2015-05-11 20:15:58 +00:00
{
return this->AssocCellSetName;
2015-05-11 20:15:58 +00:00
}
VTKM_CONT
vtkm::IdComponent GetAssocLogicalDim() const
2015-05-11 20:15:58 +00:00
{
return this->AssocLogicalDim;
2015-05-11 20:15:58 +00:00
}
2015-07-16 19:10:01 +00:00
template<typename DeviceAdapterTag, typename TypeList, typename StorageList>
VTKM_CONT
const vtkm::cont::ArrayHandle<vtkm::Range>& GetRange(DeviceAdapterTag,
TypeList,
StorageList) const
2015-07-16 19:10:01 +00:00
{
if (this->ModifiedFlag)
{
internal::ComputeRange<DeviceAdapterTag> computeRange(this->Range);
this->Data.ResetTypeAndStorageLists(TypeList(),StorageList()).CastAndCall(computeRange);
2015-07-16 19:10:01 +00:00
this->ModifiedFlag = false;
}
return this->Range;
2015-07-16 19:10:01 +00:00
}
template<typename DeviceAdapterTag, typename TypeList, typename StorageList>
VTKM_CONT
void GetRange(vtkm::Range *range,
DeviceAdapterTag,
TypeList,
StorageList) const
2015-07-16 19:10:01 +00:00
{
this->GetRange(DeviceAdapterTag(), TypeList(), StorageList());
2015-07-16 19:10:01 +00:00
vtkm::Id length = this->Range.GetNumberOfValues();
2015-07-16 19:10:01 +00:00
for (vtkm::Id i = 0; i < length; ++i)
{
range[i] = this->Range.GetPortalConstControl().Get(i);
2015-07-16 19:10:01 +00:00
}
}
template<typename DeviceAdapterTag, typename TypeList>
VTKM_CONT
const vtkm::cont::ArrayHandle<vtkm::Range>& GetRange(DeviceAdapterTag,
TypeList) const
2015-07-16 19:10:01 +00:00
{
return this->GetRange(DeviceAdapterTag(),
TypeList(),
VTKM_DEFAULT_STORAGE_LIST_TAG());
2015-07-16 19:10:01 +00:00
}
template<typename DeviceAdapterTag, typename TypeList>
VTKM_CONT
void GetRange(vtkm::Range *range, DeviceAdapterTag, TypeList) const
2015-07-16 19:10:01 +00:00
{
this->GetRange(range,
DeviceAdapterTag(),
TypeList(),
VTKM_DEFAULT_STORAGE_LIST_TAG());
2015-07-16 19:10:01 +00:00
}
template<typename DeviceAdapterTag>
VTKM_CONT
const vtkm::cont::ArrayHandle<vtkm::Range>& GetRange(DeviceAdapterTag) const
2015-07-16 19:10:01 +00:00
{
return this->GetRange(DeviceAdapterTag(),
VTKM_DEFAULT_TYPE_LIST_TAG(),
VTKM_DEFAULT_STORAGE_LIST_TAG());
2015-07-16 19:10:01 +00:00
}
template<typename DeviceAdapterTag>
VTKM_CONT
void GetRange(vtkm::Range *range, DeviceAdapterTag) const
2015-07-16 19:10:01 +00:00
{
this->GetRange(range,
DeviceAdapterTag(),
VTKM_DEFAULT_TYPE_LIST_TAG(),
VTKM_DEFAULT_STORAGE_LIST_TAG());
2015-07-16 19:10:01 +00:00
}
VTKM_CONT
const vtkm::cont::DynamicArrayHandle &GetData() const
2015-07-30 14:22:25 +00:00
{
return this->Data;
}
VTKM_CONT
2015-07-30 14:22:25 +00:00
vtkm::cont::DynamicArrayHandle &GetData()
2015-02-10 17:36:10 +00:00
{
2015-07-16 19:10:01 +00:00
this->ModifiedFlag = true;
return this->Data;
2015-02-10 17:36:10 +00:00
}
2015-05-11 20:15:58 +00:00
template <typename T>
VTKM_CONT
void SetData(const vtkm::cont::ArrayHandle<T> &newdata)
{
this->Data = newdata;
this->ModifiedFlag = true;
}
VTKM_CONT
void SetData(const vtkm::cont::DynamicArrayHandle &newdata)
2015-02-10 17:36:10 +00:00
{
this->Data = newdata;
2015-07-16 19:10:01 +00:00
this->ModifiedFlag = true;
2015-02-10 17:36:10 +00:00
}
2015-05-19 20:21:15 +00:00
template <typename T>
VTKM_CONT
void CopyData(const T *ptr, vtkm::Id nvals)
2015-02-10 17:36:10 +00:00
{
//allocate main memory using an array handle
vtkm::cont::ArrayHandle<T> tmp;
tmp.Allocate(nvals);
//copy into the memory owned by the array handle
std::copy(ptr,
ptr + static_cast<std::size_t>(nvals),
vtkm::cont::ArrayPortalToIteratorBegin(tmp.GetPortalControl()));
//assign to the dynamic array handle
this->Data = tmp;
2015-07-16 19:10:01 +00:00
this->ModifiedFlag = true;
2015-02-10 17:36:10 +00:00
}
2015-05-20 19:26:10 +00:00
VTKM_CONT
virtual void PrintSummary(std::ostream &out) const
2015-05-20 19:26:10 +00:00
{
out<<" "<<this->Name;
2015-05-20 19:26:10 +00:00
out<<" assoc= ";
switch (this->GetAssociation())
2015-05-20 19:26:10 +00:00
{
2015-08-12 17:37:36 +00:00
case ASSOC_ANY: out<<"Any "; break;
2015-05-20 19:26:10 +00:00
case ASSOC_WHOLE_MESH: out<<"Mesh "; break;
case ASSOC_POINTS: out<<"Points "; break;
case ASSOC_CELL_SET: out<<"Cells "; break;
case ASSOC_LOGICAL_DIM: out<<"LogicalDim "; break;
}
this->Data.PrintSummary(out);
2015-05-20 19:26:10 +00:00
out<<"\n";
}
2015-02-10 17:36:10 +00:00
private:
std::string Name; ///< name of field
2015-05-19 20:21:15 +00:00
AssociationEnum Association;
std::string AssocCellSetName; ///< only populate if assoc is cells
vtkm::IdComponent AssocLogicalDim; ///< only populate if assoc is logical dim
2015-05-11 20:15:58 +00:00
vtkm::cont::DynamicArrayHandle Data;
mutable vtkm::cont::ArrayHandle<vtkm::Range> Range;
2015-07-16 19:10:01 +00:00
mutable bool ModifiedFlag;
2015-02-10 17:36:10 +00:00
};
template<typename Functor>
void CastAndCall(const vtkm::cont::Field& field, const Functor &f)
{
field.GetData().CastAndCall(f);
}
namespace internal {
template<>
struct DynamicTransformTraits<vtkm::cont::Field>
{
typedef vtkm::cont::internal::DynamicTransformTagCastAndCall DynamicTag;
};
2015-02-10 17:36:10 +00:00
} // namespace internal
2015-02-10 17:36:10 +00:00
} // namespace cont
} // namespace vtkm
#endif //vtk_m_cont_Field_h