BranchCluster.h

Go to the documentation of this file.

#pragma once
 
 
#include <iostream>
#include <typeinfo>
 
#include <fstream>
#include <string>
#include <boost/serialization/export.hpp>
 
#include "BranchProbability.h"
#include "../Crew/Timp.h"
#include <memory>
#include <boost/archive/text_oarchive.hpp>
#include <boost/archive/text_iarchive.hpp>
#include <boost/serialization/access.hpp>
#include <boost/serialization/vector.hpp>
#include <boost/serialization/unique_ptr.hpp>
#include <boost/serialization/base_object.hpp>
 
 
class BranchCluster : public Time, public Probability 
{
public:
    BranchCluster();
    virtual ~BranchCluster();
    void setCurrentBranchNum(int);
    int getCurrentBranchNum();
    void calBranchProbability();
    void setIDOfTheMotherBranch(int);
    int getIDOfTheMotherBranch();
    void saveInfoToFile();
    void retrieveInfoFromFile(std::ifstream &);
    void setIDOfThisBranchCluster(int);
    int getIDOfThisBranchCluster();
    void clearBranchCluster();
    void setNCountForRestartRELAP5(long);
    int getNCountForRestartRELAP5();
    BranchProbability *findBranch(int);
    void copyBranchCluster(BranchCluster &);
    void saveBranchClusterToFile(std::ofstream &);
    void retrieveBranchClusterFromFile(std::ifstream &);
    void setIDOfMotherCluster(int);
    int getIDOfMotherCluster();
    void savePENodeToFile();
    void savePENodeToFile(std::string strDirName);
    std::vector<std::unique_ptr<BranchProbability>> all_branch;
    friend class boost::serialization::access;
    template<class Archive>
    void serialize(Archive & ar, const unsigned int version) {
//     std::cout << "Serializing class: " << typeid(*this).name() << std::endl;
        ar & boost::serialization::base_object<Time>(*this);
        ar & boost::serialization::base_object<Probability>(*this);
        ar & ID_of_the_mother_branch;
        ar & ID_of_the_mother_cluster;
        ar & Current_branch_num;
        ar & ID_of_this_branch_cluster;
        ar & n_count_for_restart_RELAP5;
        ar & all_branch; // Assuming this is serializable
        // Add more members that need to be serialized
    }
 
private:
    int ID_of_the_mother_branch;
    int ID_of_the_mother_cluster;
    int Current_branch_num; //the current branch in the run time
    int ID_of_this_branch_cluster;
    long n_count_for_restart_RELAP5;
};
 
 
BOOST_CLASS_EXPORT_KEY(BranchCluster)