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
    const int m_nDataFields = 9;
#else
    const int m_nDataFields = 8;
#endif
 
    int m_nBoxesMax = -1;
 
    LoadBalanceCosts(std::string rd_name);
 
    virtual void ComputeDiags(int step) override final;
 
    virtual void WriteToFile(int step) const override final;
 
};
 
#endif