vtk-m2/vtkm/cont/MultiBlock.cxx

//============================================================================
//  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 2015 Sandia Corporation.
//  Copyright 2015 UT-Battelle, LLC.
//  Copyright 2015 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.
//============================================================================

#include <vtkm/StaticAssert.h>
#include <vtkm/cont/ErrorExecution.h>
#include <vtkm/cont/ArrayHandle.h>
#include <vtkm/cont/Field.h>
#include <vtkm/cont/DataSet.h>
#include <vtkm/cont/DynamicArrayHandle.h>
#include <vtkm/cont/DeviceAdapterAlgorithm.h>
#include <vtkm/cont/MultiBlock.h>
namespace vtkm {
namespace cont {

VTKM_CONT
  MultiBlock::MultiBlock(const vtkm::cont::DataSet &ds)
  {
    this->blocks.push_back(ds);
  }

  VTKM_CONT
  MultiBlock::MultiBlock(const vtkm::cont::MultiBlock &src)
  {
    this->blocks = src.GetBlocks();
  }
VTKM_CONT
  MultiBlock::MultiBlock(const std::vector<vtkm::cont::DataSet> &mblocks)
  {
    this->blocks = mblocks;
  }
VTKM_CONT
  MultiBlock::MultiBlock()
  {
  }
VTKM_CONT
  MultiBlock::~MultiBlock(){}

VTKM_CONT
MultiBlock & MultiBlock::operator=(const vtkm::cont::MultiBlock &src)
  {
    this->blocks = src.GetBlocks();
    return *this;
  }
VTKM_CONT
vtkm::cont::Field MultiBlock::GetField(const std::string &field_name,
                             const int &block_index)
  {
    assert(block_index >= 0);
    assert(block_index < static_cast<int>( blocks.size() ) );
    return blocks[static_cast<std::size_t>(block_index)].GetField(field_name);
  }
VTKM_CONT
vtkm::Id MultiBlock::GetNumberOfBlocks() const
  {
    return static_cast<vtkm::Id>(this->blocks.size());
  }
VTKM_CONT
const vtkm::cont::DataSet & MultiBlock:: GetBlock(vtkm::Id blockId) const
  {
    return this->blocks[(std::size_t) blockId];
  }
VTKM_CONT
const std::vector<vtkm::cont::DataSet> & MultiBlock::GetBlocks() const
  {
    return this->blocks;
  }
VTKM_CONT
void MultiBlock::AddBlock(vtkm::cont::DataSet &ds)
  {
    this->blocks.push_back(ds);
  }
VTKM_CONT
void MultiBlock::AddBlocks(std::vector<vtkm::cont::DataSet> &mblocks)
  {
    for(std::size_t i = 0; i < mblocks.size(); i++)
    {
      AddBlock(mblocks[i]);
    }
  }
VTKM_CONT
vtkm::Bounds MultiBlock::GetBounds(vtkm::Id coordinate_system_index) const
  {
    const vtkm::Id index = coordinate_system_index;
    const size_t num_blocks = blocks.size();

    vtkm::Bounds bounds;

    for(size_t i = 0; i < num_blocks; ++i)
    {
      vtkm::Bounds block_bounds = GetBlockBounds(i, index);
      bounds.Include(block_bounds);
    }

    return bounds;
  }
VTKM_CONT
vtkm::Bounds MultiBlock::GetBlockBounds(const std::size_t &block_index,
                               vtkm::Id coordinate_system_index) const
  {
    const vtkm::Id index = coordinate_system_index;
    vtkm::cont::CoordinateSystem coords;
    try
    {
      coords = blocks[block_index].GetCoordinateSystem(index);
    }
    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();
  }

VTKM_CONT
vtkm::cont::ArrayHandle<vtkm::Range> MultiBlock::GetGlobalRange(const int &index) const
  {
    assert(blocks.size() > 0);
    vtkm::cont::Field field = blocks.at(0).GetField(index);
    std::string field_name = field.GetName();
    return this->GetGlobalRange(field_name);
  }
VTKM_CONT
vtkm::cont::ArrayHandle<vtkm::Range> MultiBlock::GetGlobalRange(const std::string &field_name) const
  {
    bool valid_field = true;
    const size_t num_blocks = blocks.size();

    vtkm::cont::ArrayHandle<vtkm::Range> range;
    vtkm::Id num_components = 0;

    for(size_t i = 0; i < num_blocks; ++i)
    {
      if(!blocks[0].HasField(field_name))
      {
        valid_field = false;
        break;
      }

      const vtkm::cont::Field &field = blocks[i].GetField(field_name);
      vtkm::cont::ArrayHandle<vtkm::Range> sub_range;
      sub_range = field.GetRange();

      if(i == 0)
      {
        num_components = sub_range.GetPortalConstControl().GetNumberOfValues();
        range = sub_range;
        continue;
      }

      vtkm::Id components = sub_range.GetPortalConstControl().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());
      }

      for(vtkm::Id c = 0; c < components; ++c)
      {
        vtkm::Range s_range = sub_range.GetPortalControl().Get(c);
        vtkm::Range c_range = range.GetPortalControl().Get(c);
        c_range.Include(s_range);
        range.GetPortalControl().Set(c, c_range);
      }

    }

    if(!valid_field)
    {
      std::string msg = "GetRange call failed. ";
      msg += " Field " +  field_name + " did not exist in at least one block.";
      throw ErrorExecution(msg);
    }

    return range;
  }
VTKM_CONT
void MultiBlock::PrintSummary(std::ostream &stream) const
  {
    stream<<"block "<<"\n";

    for(size_t block_index = 0; block_index < blocks.size(); ++block_index)
    {
      stream<<"block "<<block_index<<"\n";
      blocks[block_index].PrintSummary(stream);
    }
  }
}
} // namespace vtkm::cont