LoadBalanceCosts.H

Go to the documentation of this file.

/* Copyright 2019-2020 Michael Rowan, Yinjian Zhao
 *
 * This file is part of WarpX.
 *
 * License: BSD-3-Clause-LBNL
 */
 
#ifndef WARPX_DIAGNOSTICS_REDUCEDDIAGS_LOADBALANCECOSTS_H_
#define WARPX_DIAGNOSTICS_REDUCEDDIAGS_LOADBALANCECOSTS_H_
 
#include "ReducedDiags.H"
 
#include <AMReX_Vector.H>
 
#include <string>
#include <vector>

class LoadBalanceCosts : public ReducedDiags
{
public:

    amrex::Vector<char> m_data_string_recvbuf;

    int m_data_string_recvbuf_length = 0;

    std::vector<std::string> m_data_string;

    amrex::Vector<int> m_data_string_recvcount; // array of size N_procs, how many message root recv from sender
    amrex::Vector<int> m_data_string_disp;      // array of size N_procs, where to place data in IOProc

#ifdef AMREX_USE_GPU
    static const int m_nDataFields = 9;
#else
    static const int m_nDataFields = 8;
#endif

    int m_nBoxesMax = -1;

    explicit LoadBalanceCosts(const std::string& rd_name);

    void ComputeDiags(int step) final;

    void WriteToFile(int step) const final;
 
};
 
#endif