2017-07-20 19:11:12 +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.
|
|
|
|
//
|
2017-09-20 21:33:44 +00:00
|
|
|
// Copyright 2015 National Technology & Engineering Solutions of Sandia, LLC (NTESS).
|
2017-07-20 19:11:12 +00:00
|
|
|
// Copyright 2015 UT-Battelle, LLC.
|
|
|
|
// Copyright 2015 Los Alamos National Security.
|
|
|
|
//
|
2017-09-20 21:33:44 +00:00
|
|
|
// Under the terms of Contract DE-NA0003525 with NTESS,
|
2017-07-20 19:11:12 +00:00
|
|
|
// 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.
|
|
|
|
//============================================================================
|
2017-08-01 22:01:02 +00:00
|
|
|
|
2017-07-20 19:11:12 +00:00
|
|
|
#include <vtkm/StaticAssert.h>
|
|
|
|
#include <vtkm/cont/ArrayHandle.h>
|
|
|
|
#include <vtkm/cont/DataSet.h>
|
|
|
|
#include <vtkm/cont/DeviceAdapterAlgorithm.h>
|
2017-08-18 20:28:33 +00:00
|
|
|
#include <vtkm/cont/DynamicArrayHandle.h>
|
|
|
|
#include <vtkm/cont/ErrorExecution.h>
|
|
|
|
#include <vtkm/cont/Field.h>
|
2017-07-20 19:11:12 +00:00
|
|
|
#include <vtkm/cont/MultiBlock.h>
|
2017-08-18 20:28:33 +00:00
|
|
|
namespace vtkm
|
|
|
|
{
|
|
|
|
namespace cont
|
|
|
|
{
|
2017-08-02 03:15:41 +00:00
|
|
|
|
|
|
|
VTKM_CONT
|
2017-08-18 20:28:33 +00:00
|
|
|
MultiBlock::MultiBlock(const vtkm::cont::DataSet& ds)
|
|
|
|
{
|
2017-12-06 01:06:19 +00:00
|
|
|
this->Blocks.insert(this->Blocks.end(), ds);
|
2017-08-18 20:28:33 +00:00
|
|
|
}
|
|
|
|
|
2017-08-02 20:17:31 +00:00
|
|
|
VTKM_CONT
|
2017-08-18 20:28:33 +00:00
|
|
|
MultiBlock::MultiBlock(const vtkm::cont::MultiBlock& src)
|
|
|
|
{
|
2017-12-06 01:06:19 +00:00
|
|
|
this->Blocks = src.GetBlocks();
|
2017-08-18 20:28:33 +00:00
|
|
|
}
|
|
|
|
|
2017-08-02 03:15:41 +00:00
|
|
|
VTKM_CONT
|
2017-08-18 20:28:33 +00:00
|
|
|
MultiBlock::MultiBlock(const std::vector<vtkm::cont::DataSet>& mblocks)
|
|
|
|
{
|
2017-12-06 01:06:19 +00:00
|
|
|
this->Blocks = mblocks;
|
2017-08-18 20:28:33 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
VTKM_CONT
|
|
|
|
MultiBlock::MultiBlock(vtkm::Id size)
|
|
|
|
{
|
2017-12-06 01:06:19 +00:00
|
|
|
this->Blocks.reserve(static_cast<std::size_t>(size));
|
2017-08-18 20:28:33 +00:00
|
|
|
}
|
|
|
|
|
2017-08-02 19:18:55 +00:00
|
|
|
VTKM_CONT
|
2017-08-02 20:17:31 +00:00
|
|
|
MultiBlock::MultiBlock()
|
2017-08-18 20:28:33 +00:00
|
|
|
{
|
|
|
|
}
|
|
|
|
|
2017-08-02 03:15:41 +00:00
|
|
|
VTKM_CONT
|
2017-08-18 20:28:33 +00:00
|
|
|
MultiBlock::~MultiBlock()
|
|
|
|
{
|
|
|
|
}
|
2017-08-02 03:15:41 +00:00
|
|
|
|
2017-08-02 02:16:00 +00:00
|
|
|
VTKM_CONT
|
2017-08-18 20:28:33 +00:00
|
|
|
MultiBlock& MultiBlock::operator=(const vtkm::cont::MultiBlock& src)
|
|
|
|
{
|
2017-12-06 01:06:19 +00:00
|
|
|
this->Blocks = src.GetBlocks();
|
2017-08-18 20:28:33 +00:00
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
|
2017-08-02 02:16:00 +00:00
|
|
|
VTKM_CONT
|
2017-08-18 20:28:33 +00:00
|
|
|
vtkm::cont::Field MultiBlock::GetField(const std::string& field_name, const int& block_index)
|
|
|
|
{
|
|
|
|
assert(block_index >= 0);
|
2017-12-06 01:06:19 +00:00
|
|
|
assert(static_cast<std::size_t>(block_index) < this->Blocks.size());
|
|
|
|
return this->Blocks[static_cast<std::size_t>(block_index)].GetField(field_name);
|
2017-08-18 20:28:33 +00:00
|
|
|
}
|
|
|
|
|
2017-08-02 02:16:00 +00:00
|
|
|
VTKM_CONT
|
2017-07-20 19:11:12 +00:00
|
|
|
vtkm::Id MultiBlock::GetNumberOfBlocks() const
|
2017-08-18 20:28:33 +00:00
|
|
|
{
|
2017-12-06 01:06:19 +00:00
|
|
|
return static_cast<vtkm::Id>(this->Blocks.size());
|
2017-08-18 20:28:33 +00:00
|
|
|
}
|
|
|
|
|
2017-08-20 18:00:11 +00:00
|
|
|
VTKM_CONT
|
|
|
|
const vtkm::cont::DataSet& MultiBlock::GetBlock(vtkm::Id blockId) const
|
2017-08-18 20:28:33 +00:00
|
|
|
{
|
2017-12-06 01:06:19 +00:00
|
|
|
return this->Blocks[static_cast<std::size_t>(blockId)];
|
2017-08-18 20:28:33 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
VTKM_CONT
|
|
|
|
const std::vector<vtkm::cont::DataSet>& MultiBlock::GetBlocks() const
|
|
|
|
{
|
2017-12-06 01:06:19 +00:00
|
|
|
return this->Blocks;
|
2017-08-18 20:28:33 +00:00
|
|
|
}
|
2017-08-30 16:37:15 +00:00
|
|
|
|
2017-08-18 20:28:33 +00:00
|
|
|
VTKM_CONT
|
|
|
|
void MultiBlock::AddBlock(vtkm::cont::DataSet& ds)
|
|
|
|
{
|
2017-12-06 01:06:19 +00:00
|
|
|
this->Blocks.insert(this->Blocks.end(), ds);
|
2017-08-18 20:28:33 +00:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
void MultiBlock::AddBlocks(std::vector<vtkm::cont::DataSet>& mblocks)
|
|
|
|
{
|
2017-12-06 01:06:19 +00:00
|
|
|
this->Blocks.insert(this->Blocks.end(), mblocks.begin(), mblocks.end());
|
2017-08-18 20:28:33 +00:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
VTKM_CONT
|
|
|
|
void MultiBlock::InsertBlock(vtkm::Id index, vtkm::cont::DataSet& ds)
|
|
|
|
{
|
2017-12-06 01:06:19 +00:00
|
|
|
if (index <= static_cast<vtkm::Id>(this->Blocks.size()))
|
|
|
|
this->Blocks.insert(this->Blocks.begin() + index, ds);
|
2017-08-18 20:28:33 +00:00
|
|
|
else
|
2017-08-24 14:50:14 +00:00
|
|
|
{
|
|
|
|
std::string msg = "invalid insert position\n ";
|
|
|
|
throw ErrorExecution(msg);
|
|
|
|
}
|
2017-08-18 20:28:33 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
VTKM_CONT
|
2017-08-24 14:50:14 +00:00
|
|
|
void MultiBlock::ReplaceBlock(vtkm::Id index, vtkm::cont::DataSet& ds)
|
2017-08-18 20:28:33 +00:00
|
|
|
{
|
2017-12-06 01:06:19 +00:00
|
|
|
if (index < static_cast<vtkm::Id>(this->Blocks.size()))
|
|
|
|
this->Blocks.at(static_cast<std::size_t>(index)) = ds;
|
2017-08-18 20:28:33 +00:00
|
|
|
else
|
2017-08-24 14:50:14 +00:00
|
|
|
{
|
|
|
|
std::string msg = "invalid replace position\n ";
|
|
|
|
throw ErrorExecution(msg);
|
|
|
|
}
|
2017-08-18 20:28:33 +00:00
|
|
|
}
|
|
|
|
|
2017-08-02 02:16:00 +00:00
|
|
|
VTKM_CONT
|
2017-07-20 19:11:12 +00:00
|
|
|
vtkm::Bounds MultiBlock::GetBounds(vtkm::Id coordinate_system_index) const
|
2017-08-18 20:28:33 +00:00
|
|
|
{
|
|
|
|
return this->GetBounds(coordinate_system_index,
|
|
|
|
VTKM_DEFAULT_COORDINATE_SYSTEM_TYPE_LIST_TAG(),
|
|
|
|
VTKM_DEFAULT_COORDINATE_SYSTEM_STORAGE_LIST_TAG());
|
|
|
|
}
|
2017-07-20 19:11:12 +00:00
|
|
|
|
2017-08-18 20:28:33 +00:00
|
|
|
template <typename TypeList>
|
|
|
|
VTKM_CONT vtkm::Bounds MultiBlock::GetBounds(vtkm::Id coordinate_system_index, TypeList) const
|
|
|
|
{
|
|
|
|
VTKM_IS_LIST_TAG(TypeList);
|
|
|
|
return this->GetBounds(
|
|
|
|
coordinate_system_index, TypeList(), VTKM_DEFAULT_COORDINATE_SYSTEM_STORAGE_LIST_TAG());
|
|
|
|
}
|
|
|
|
template <typename TypeList, typename StorageList>
|
|
|
|
VTKM_CONT vtkm::Bounds MultiBlock::GetBounds(vtkm::Id coordinate_system_index,
|
|
|
|
TypeList,
|
|
|
|
StorageList) const
|
|
|
|
{
|
|
|
|
VTKM_IS_LIST_TAG(TypeList);
|
|
|
|
VTKM_IS_LIST_TAG(StorageList);
|
2017-07-20 19:11:12 +00:00
|
|
|
|
2017-08-18 20:28:33 +00:00
|
|
|
const vtkm::Id index = coordinate_system_index;
|
2017-12-06 01:06:19 +00:00
|
|
|
const size_t num_blocks = this->Blocks.size();
|
2017-08-18 20:28:33 +00:00
|
|
|
|
|
|
|
vtkm::Bounds bounds;
|
|
|
|
for (size_t i = 0; i < num_blocks; ++i)
|
2017-07-20 19:11:12 +00:00
|
|
|
{
|
2017-08-18 20:28:33 +00:00
|
|
|
vtkm::Bounds block_bounds = this->GetBlockBounds(i, index, TypeList(), StorageList());
|
|
|
|
bounds.Include(block_bounds);
|
2017-07-20 19:11:12 +00:00
|
|
|
}
|
|
|
|
|
2017-08-18 20:28:33 +00:00
|
|
|
return bounds;
|
|
|
|
}
|
|
|
|
|
2017-08-02 02:16:00 +00:00
|
|
|
VTKM_CONT
|
2017-08-18 20:28:33 +00:00
|
|
|
vtkm::Bounds MultiBlock::GetBlockBounds(const std::size_t& block_index,
|
|
|
|
vtkm::Id coordinate_system_index) const
|
|
|
|
{
|
|
|
|
return this->GetBlockBounds(block_index,
|
|
|
|
coordinate_system_index,
|
|
|
|
VTKM_DEFAULT_COORDINATE_SYSTEM_TYPE_LIST_TAG(),
|
|
|
|
VTKM_DEFAULT_COORDINATE_SYSTEM_STORAGE_LIST_TAG());
|
|
|
|
}
|
|
|
|
|
|
|
|
template <typename TypeList>
|
|
|
|
VTKM_CONT vtkm::Bounds MultiBlock::GetBlockBounds(const std::size_t& block_index,
|
|
|
|
vtkm::Id coordinate_system_index,
|
|
|
|
TypeList) const
|
|
|
|
{
|
|
|
|
VTKM_IS_LIST_TAG(TypeList);
|
|
|
|
return this->GetBlockBounds(block_index,
|
|
|
|
coordinate_system_index,
|
|
|
|
TypeList(),
|
|
|
|
VTKM_DEFAULT_COORDINATE_SYSTEM_STORAGE_LIST_TAG());
|
|
|
|
}
|
|
|
|
|
|
|
|
template <typename TypeList, typename StorageList>
|
|
|
|
VTKM_CONT vtkm::Bounds MultiBlock::GetBlockBounds(const std::size_t& block_index,
|
|
|
|
vtkm::Id coordinate_system_index,
|
|
|
|
TypeList,
|
|
|
|
StorageList) const
|
|
|
|
{
|
|
|
|
VTKM_IS_LIST_TAG(TypeList);
|
|
|
|
VTKM_IS_LIST_TAG(StorageList);
|
|
|
|
|
|
|
|
const vtkm::Id index = coordinate_system_index;
|
|
|
|
vtkm::cont::CoordinateSystem coords;
|
|
|
|
try
|
2017-07-20 19:11:12 +00:00
|
|
|
{
|
2017-12-06 01:06:19 +00:00
|
|
|
coords = this->Blocks[block_index].GetCoordinateSystem(index);
|
2017-08-02 19:18:55 +00:00
|
|
|
}
|
2017-08-18 20:28:33 +00:00
|
|
|
catch (const vtkm::cont::Error& error)
|
|
|
|
{
|
|
|
|
std::stringstream msg;
|
|
|
|
msg << "GetBounds call failed. vtk-m error was encountered while "
|
|
|
|
<< "attempting to get coordinate system " << index << " from "
|
|
|
|
<< "block " << block_index << ". vtkm error message: " << error.GetMessage();
|
|
|
|
throw ErrorExecution(msg.str());
|
|
|
|
}
|
|
|
|
return coords.GetBounds(TypeList(), StorageList());
|
|
|
|
}
|
|
|
|
|
2017-08-02 02:16:00 +00:00
|
|
|
VTKM_CONT
|
2017-08-18 20:28:33 +00:00
|
|
|
vtkm::cont::ArrayHandle<vtkm::Range> MultiBlock::GetGlobalRange(const int& index) const
|
|
|
|
{
|
|
|
|
return this->GetGlobalRange(index, VTKM_DEFAULT_TYPE_LIST_TAG(), VTKM_DEFAULT_STORAGE_LIST_TAG());
|
|
|
|
}
|
|
|
|
|
|
|
|
template <typename TypeList>
|
|
|
|
VTKM_CONT vtkm::cont::ArrayHandle<vtkm::Range> MultiBlock::GetGlobalRange(const int& index,
|
|
|
|
TypeList) const
|
|
|
|
{
|
|
|
|
VTKM_IS_LIST_TAG(TypeList);
|
|
|
|
return this->GetGlobalRange(index, TypeList(), VTKM_DEFAULT_STORAGE_LIST_TAG());
|
|
|
|
}
|
|
|
|
|
|
|
|
template <typename TypeList, typename StorageList>
|
|
|
|
VTKM_CONT vtkm::cont::ArrayHandle<vtkm::Range> MultiBlock::GetGlobalRange(const int& index,
|
|
|
|
TypeList,
|
|
|
|
StorageList) const
|
|
|
|
{
|
|
|
|
VTKM_IS_LIST_TAG(TypeList);
|
|
|
|
VTKM_IS_LIST_TAG(StorageList);
|
|
|
|
|
2017-12-06 01:06:19 +00:00
|
|
|
assert(this->Blocks.size() > 0);
|
|
|
|
vtkm::cont::Field field = this->Blocks.at(0).GetField(index);
|
2017-08-18 20:28:33 +00:00
|
|
|
std::string field_name = field.GetName();
|
|
|
|
return this->GetGlobalRange(field_name, TypeList(), StorageList());
|
|
|
|
}
|
|
|
|
|
|
|
|
VTKM_CONT
|
|
|
|
vtkm::cont::ArrayHandle<vtkm::Range> MultiBlock::GetGlobalRange(const std::string& field_name) const
|
|
|
|
{
|
|
|
|
return this->GetGlobalRange(
|
|
|
|
field_name, VTKM_DEFAULT_TYPE_LIST_TAG(), VTKM_DEFAULT_STORAGE_LIST_TAG());
|
|
|
|
}
|
|
|
|
|
|
|
|
template <typename TypeList>
|
|
|
|
VTKM_CONT vtkm::cont::ArrayHandle<vtkm::Range> MultiBlock::GetGlobalRange(
|
|
|
|
const std::string& field_name,
|
|
|
|
TypeList) const
|
|
|
|
{
|
|
|
|
VTKM_IS_LIST_TAG(TypeList);
|
|
|
|
return this->GetGlobalRange(field_name, TypeList(), VTKM_DEFAULT_STORAGE_LIST_TAG());
|
|
|
|
}
|
|
|
|
|
|
|
|
template <typename TypeList, typename StorageList>
|
|
|
|
VTKM_CONT vtkm::cont::ArrayHandle<vtkm::Range>
|
|
|
|
MultiBlock::GetGlobalRange(const std::string& field_name, TypeList, StorageList) const
|
|
|
|
{
|
|
|
|
bool valid_field = true;
|
2017-12-06 01:06:19 +00:00
|
|
|
const size_t num_blocks = this->Blocks.size();
|
2017-08-18 20:28:33 +00:00
|
|
|
|
|
|
|
vtkm::cont::ArrayHandle<vtkm::Range> range;
|
|
|
|
vtkm::Id num_components = 0;
|
|
|
|
|
|
|
|
for (size_t i = 0; i < num_blocks; ++i)
|
2017-07-20 19:11:12 +00:00
|
|
|
{
|
2017-12-06 01:06:19 +00:00
|
|
|
if (!this->Blocks[i].HasField(field_name))
|
2017-08-18 20:28:33 +00:00
|
|
|
{
|
|
|
|
valid_field = false;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
2017-12-06 01:06:19 +00:00
|
|
|
const vtkm::cont::Field& field = this->Blocks[i].GetField(field_name);
|
2017-08-18 20:28:33 +00:00
|
|
|
vtkm::cont::ArrayHandle<vtkm::Range> sub_range = field.GetRange(TypeList(), StorageList());
|
2017-07-20 19:11:12 +00:00
|
|
|
|
2017-08-18 20:28:33 +00:00
|
|
|
vtkm::cont::ArrayHandle<vtkm::Range>::PortalConstControl sub_range_control =
|
|
|
|
sub_range.GetPortalConstControl();
|
|
|
|
vtkm::cont::ArrayHandle<vtkm::Range>::PortalControl range_control = range.GetPortalControl();
|
2017-07-20 19:11:12 +00:00
|
|
|
|
2017-08-18 20:28:33 +00:00
|
|
|
if (i == 0)
|
2017-07-20 19:11:12 +00:00
|
|
|
{
|
2017-08-18 20:28:33 +00:00
|
|
|
num_components = sub_range_control.GetNumberOfValues();
|
|
|
|
range = sub_range;
|
|
|
|
continue;
|
|
|
|
}
|
2017-08-02 19:18:55 +00:00
|
|
|
|
2017-08-18 20:28:33 +00:00
|
|
|
vtkm::Id components = sub_range_control.GetNumberOfValues();
|
|
|
|
|
|
|
|
if (components != num_components)
|
|
|
|
{
|
|
|
|
std::stringstream msg;
|
|
|
|
msg << "GetRange call failed. The number of components (" << components << ") in field "
|
|
|
|
<< field_name << " from block " << i << " does not match the number of components "
|
|
|
|
<< "(" << num_components << ") in block 0";
|
|
|
|
throw ErrorExecution(msg.str());
|
2017-07-20 19:11:12 +00:00
|
|
|
}
|
|
|
|
|
2017-08-18 20:28:33 +00:00
|
|
|
|
|
|
|
for (vtkm::Id c = 0; c < components; ++c)
|
2017-07-20 19:11:12 +00:00
|
|
|
{
|
2017-08-18 20:28:33 +00:00
|
|
|
vtkm::Range s_range = sub_range_control.Get(c);
|
|
|
|
vtkm::Range c_range = range_control.Get(c);
|
|
|
|
c_range.Include(s_range);
|
|
|
|
range_control.Set(c, c_range);
|
2017-07-20 19:11:12 +00:00
|
|
|
}
|
2017-08-18 20:28:33 +00:00
|
|
|
}
|
2017-07-20 19:11:12 +00:00
|
|
|
|
2017-08-18 20:28:33 +00:00
|
|
|
if (!valid_field)
|
|
|
|
{
|
|
|
|
std::string msg = "GetRange call failed. ";
|
|
|
|
msg += " Field " + field_name + " did not exist in at least one block.";
|
|
|
|
throw ErrorExecution(msg);
|
2017-07-20 19:11:12 +00:00
|
|
|
}
|
2017-08-18 20:28:33 +00:00
|
|
|
|
|
|
|
return range;
|
|
|
|
}
|
|
|
|
|
2017-08-02 02:16:00 +00:00
|
|
|
VTKM_CONT
|
2017-08-18 20:28:33 +00:00
|
|
|
void MultiBlock::PrintSummary(std::ostream& stream) const
|
|
|
|
{
|
|
|
|
stream << "block "
|
|
|
|
<< "\n";
|
2017-07-20 19:11:12 +00:00
|
|
|
|
2017-12-06 01:06:19 +00:00
|
|
|
for (size_t block_index = 0; block_index < this->Blocks.size(); ++block_index)
|
2017-08-18 20:28:33 +00:00
|
|
|
{
|
|
|
|
stream << "block " << block_index << "\n";
|
2017-12-06 01:06:19 +00:00
|
|
|
this->Blocks[block_index].PrintSummary(stream);
|
2017-07-20 19:11:12 +00:00
|
|
|
}
|
|
|
|
}
|
2017-08-18 20:28:33 +00:00
|
|
|
}
|
2017-07-20 19:11:12 +00:00
|
|
|
} // namespace vtkm::cont
|