Client.h

Go to the documentation of this file.

 
 
#ifndef CLIENT_H
#define CLIENT_H
 
#include <mpi.h>
#include "ParallelUtilities.h"
#include "ParallelGlobal.h"
#include "../Solver/ThermalHydraulicSolver.h"
#include "../Solver/AdsidacCallback.h"
#include <filesystem>
#include <memory>
#include <unordered_map>
 
namespace fs = std::filesystem;
 
using PU = ParallelUtilities;
 
class Client {
public:
    explicit Client(int rank);
 
    int Work();
 
    void parallel_restartforanewsequence_(int branch_index);
 
    void execute_relap();
 
    int getNextLocalBranchID() { return ++local_branch_counter_; }
 
    int parent_seq_id_ = -1;       
    double branch_time_ = 0.0;     
    int branches_this_seq_ = 0;    
 
    MPI_Status status_client{};    
 
    struct HeldBranchData {
        std::shared_ptr<RestartNodeInfo> rinfo;         
        std::shared_ptr<BranchRecord> branch_record;    
        std::string file_path;      
        int ref_count;              
    };
 
    void hold_branch_data(int cid, std::shared_ptr<RestartNodeInfo> rinfo,
                          BranchRecord& br, const std::string& path, int refs);
 
    void restore_relap_state(const std::string& file_path, bool clear_buf);
 
    void complete_sequence(int seq_id);
 
private:
    int rank;                       
    int local_branch_counter_;      
    std::unique_ptr<ThermalHydraulicSolver> solver_;  
    AdsidacCallback callback_;      
    std::shared_ptr<RestartNodeInfo> restart_info_;  
    std::shared_ptr<BranchRecord> branch_center_;    
    KeyParameterHistory KPS;        
    std::unordered_map<int, HeldBranchData> held_data_;  
 
    std::vector<char> base_cache_;
    bool ensure_base_cache();          
    void restore_buffer(std::vector<char> raw);  
    void self_merge_outputs();      
};
 
#endif //CLIENT_H