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 = 7;
#else
    const int m_nDataFields = 6;
#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