Line data    Source code

       1           0 : // Distributed under the MIT License.
       2             : // See LICENSE.txt for details.
       3             : 
       4             : #pragma once
       5             : 
       6             : #include <cstddef>
       7             : #include <mutex>
       8             : 
       9             : #include "DataStructures/DataBox/DataBox.hpp"
      10             : #include "IO/H5/AccessType.hpp"
      11             : #include "IO/H5/File.hpp"
      12             : #include "IO/Observer/Helpers.hpp"
      13             : #include "IO/Observer/Tags.hpp"
      14             : #include "Parallel/ArrayComponentId.hpp"
      15             : #include "Parallel/GlobalCache.hpp"
      16             : #include "Parallel/Info.hpp"
      17             : #include "Parallel/Invoke.hpp"
      18             : #include "Parallel/Local.hpp"
      19             : #include "Parallel/NodeLock.hpp"
      20             : #include "Utilities/Requires.hpp"
      21             : #include "Utilities/TMPL.hpp"
      22             : 
      23             : namespace observers {
      24             : namespace ThreadedActions {
      25             : 
      26             : /*!
      27             :  * \brief Append data to an h5::Dat subfile of `Tags::VolumeFileName`.
      28             :  *
      29             :  * \details This is a streamlined interface for getting data to the volume file
      30             :  * associated with a node; it will simply write the `.dat` object
      31             :  * `subfile_name`, giving it the `file_legend` if it does not yet exist,
      32             :  * appending `data_row` to the end of the dat.
      33             :  */
      34           1 : struct WriteSimpleData {
      35             :   template <typename ParallelComponent, typename DbTagsList,
      36             :             typename Metavariables, typename ArrayIndex>
      37           0 :   static void apply(db::DataBox<DbTagsList>& box,
      38             :                     Parallel::GlobalCache<Metavariables>& cache,
      39             :                     const ArrayIndex& /*array_index*/,
      40             :                     const gsl::not_null<Parallel::NodeLock*> node_lock,
      41             :                     const std::vector<std::string>& file_legend,
      42             :                     const std::vector<double>& data_row,
      43             :                     const std::string& subfile_name) {
      44             :     Parallel::NodeLock* file_lock = nullptr;
      45             :     {
      46             :       const std::lock_guard hold_lock(*node_lock);
      47             :       db::mutate<Tags::H5FileLock>(
      48             :           [&file_lock](const gsl::not_null<Parallel::NodeLock*> in_file_lock) {
      49             :             file_lock = in_file_lock;
      50             :           },
      51             :           make_not_null(&box));
      52             :     }
      53             : 
      54             :     // scoped to close file
      55             :     {
      56             :       const std::lock_guard hold_lock(*file_lock);
      57             :       const auto& file_prefix = Parallel::get<Tags::VolumeFileName>(cache);
      58             :       auto& my_proxy =
      59             :           Parallel::get_parallel_component<ParallelComponent>(cache);
      60             :       h5::H5File<h5::AccessType::ReadWrite> h5file(
      61             :           file_prefix +
      62             :               std::to_string(
      63             :                   Parallel::my_node<int>(*Parallel::local_branch(my_proxy))) +
      64             :               ".h5",
      65             :           true, observers::input_source_from_cache(cache));
      66             :       const size_t version_number = 0;
      67             :       auto& output_dataset =
      68             :           h5file.try_insert<h5::Dat>(subfile_name, file_legend, version_number);
      69             :       output_dataset.append(data_row);
      70             :       h5file.close_current_object();
      71             :     }
      72             :   }
      73             : };
      74             : }  // namespace ThreadedActions
      75             : }  // namespace observers