From 331e63fed4dfde4a412b3ac0e11e4a9d79d6e7ba Mon Sep 17 00:00:00 2001 From: "Gunther H. Weber" Date: Sat, 31 Jul 2021 17:00:19 -0700 Subject: [PATCH] Added missing LoadArrays include. --- .../contourtree_distributed/LoadArrays.h | 115 ++++++++++++++++++ 1 file changed, 115 insertions(+) create mode 100644 vtkm/worklet/contourtree_distributed/LoadArrays.h diff --git a/vtkm/worklet/contourtree_distributed/LoadArrays.h b/vtkm/worklet/contourtree_distributed/LoadArrays.h new file mode 100644 index 000000000..1d0b0d4b3 --- /dev/null +++ b/vtkm/worklet/contourtree_distributed/LoadArrays.h @@ -0,0 +1,115 @@ +//============================================================================ +// 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 (c) 2018, The Regents of the University of California, through +// Lawrence Berkeley National Laboratory (subject to receipt of any required approvals +// from the U.S. Dept. of Energy). All rights reserved. +// +// Redistribution and use in source and binary forms, with or without modification, +// are permitted provided that the following conditions are met: +// +// (1) Redistributions of source code must retain the above copyright notice, this +// list of conditions and the following disclaimer. +// +// (2) Redistributions in binary form must reproduce the above copyright notice, +// this list of conditions and the following disclaimer in the documentation +// and/or other materials provided with the distribution. +// +// (3) Neither the name of the University of California, Lawrence Berkeley National +// Laboratory, U.S. Dept. of Energy nor the names of its contributors may be +// used to endorse or promote products derived from this software without +// specific prior written permission. +// +// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND +// ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED +// WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. +// IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, +// INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, +// BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, +// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF +// LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE +// OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED +// OF THE POSSIBILITY OF SUCH DAMAGE. +// +#ifndef vtk_m_worklet_contourtree_distributed_load_arrays_h +#define vtk_m_worklet_contourtree_distributed_load_arrays_h + +#include +#include + +namespace vtkm +{ +namespace worklet +{ +namespace contourtree_distributed +{ + +// Types used in binary test files +typedef size_t FileSizeType; +typedef unsigned long long FileIndexType; +const FileIndexType FileIndexMask = 0x07FFFFFFFFFFFFFFLL; +typedef double FileDataType; + +inline void ReadIndexArray(std::ifstream& is, vtkm::cont::ArrayHandle& indexArray) +{ + FileSizeType sz; + is.read(reinterpret_cast(&sz), sizeof(sz)); + //std::cout << "Reading index array of size " << sz << std::endl; + indexArray.Allocate(sz); + auto writePortal = indexArray.WritePortal(); + + for (vtkm::Id i = 0; i < static_cast(sz); ++i) + { + FileIndexType x; + is.read(reinterpret_cast(&x), sizeof(x)); + // Covert from index type size in file (64 bit) to index type currently used by + // shifting the flag portion of the index accordingly + vtkm::Id shiftedFlagVal = (x & FileIndexMask) | + ((x & ~FileIndexMask) >> ((sizeof(FileIndexType) - sizeof(vtkm::Id)) << 3)); + writePortal.Set(i, shiftedFlagVal); + } +} + +inline void ReadIndexArrayVector(std::ifstream& is, + std::vector>& indexArrayVector) +{ + FileSizeType sz; + is.read(reinterpret_cast(&sz), sizeof(sz)); + //std::cout << "Reading vector of " << sz << " index arrays" << std::endl; + indexArrayVector.resize(sz); + + for (vtkm::Id i = 0; i < static_cast(sz); ++i) + { + ReadIndexArray(is, indexArrayVector[i]); + } +} + +template +inline void ReadDataArray(std::ifstream& is, vtkm::cont::ArrayHandle& dataArray) +{ + FileSizeType sz; + is.read(reinterpret_cast(&sz), sizeof(sz)); + //std::cout << "Reading data array of size " << sz << std::endl; + dataArray.Allocate(sz); + auto writePortal = dataArray.WritePortal(); + + for (vtkm::Id i = 0; i < static_cast(sz); ++i) + { + FileDataType x; + is.read(reinterpret_cast(&x), sizeof(x)); + //std::cout << "Read " << x << std::endl; + writePortal.Set(i, x); + } +} + +} // namespace contourtree_distributed +} // namespace worklet +} // namespace vtkm + +#endif