tkisotope_builder.h

/********************************************************************************
 *   Copyright (c) : Université de Lyon 1, CNRS/IN2P3, UMR5822,                 *
 *                   IP2I, F-69622 Villeurbanne Cedex, France                   *
 *                   Normandie Université, ENSICAEN, UNICAEN, CNRS/IN2P3,       *
 *                   LPC Caen, F-14000 Caen, France                             *
 *   Contibutor(s) :                                                            *
 *      Jérémie Dudouet jeremie.dudouet@cnrs.fr [2020]                          *
 *      Diego Gruyer    diego.gruyer@cnrs.fr    [2020]                          *
 *                                                                              *
 *   Licensed under the MIT License <http://opensource.org/licenses/MIT>.       *   
 *   SPDX-License-Identifier: MIT                                               *
 ********************************************************************************/
 
#ifndef tkisotope_builder_H
#define tkisotope_builder_H
 
#include <algorithm>
#include <iterator>
#include <vector>
 
#include "tkn_config.h"
#include "tkstring.h"
#include "tkdatabase.h"
#include "json.hpp"
#include "tkdb_builder.h"
 
#ifdef HAS_ROOT
#include "TClass.h"
#endif
 
using json = nlohmann::json;
using namespace std;
 
namespace tkn {
 
class tkisotope_builder : public tkdb_builder {
 
protected:
 
    struct json_nucleus {
        tkstring name="";
        int Z,A,N;
        tkdb_table::measure_data_struct mass_excess;
        tkdb_table::measure_data_struct abundance;
        tkdb_table::measure_data_struct quadrupoleDeformation;
 
        tkdb_table::measure_data_struct radius;
        tkdb_table::measure_data_struct electric_quadrupole;
        tkdb_table::measure_data_struct magnetic_dipole;
 
        tkdb_table::measure_data_struct FY235U;
        tkdb_table::measure_data_struct FY238U;
        tkdb_table::measure_data_struct FY239Pu;
        tkdb_table::measure_data_struct FY241Pu;
        tkdb_table::measure_data_struct FY252Cf;
 
        tkdb_table::measure_data_struct cFY235U;
        tkdb_table::measure_data_struct cFY238U;
        tkdb_table::measure_data_struct cFY239Pu;
        tkdb_table::measure_data_struct cFY241Pu;
        tkdb_table::measure_data_struct cFY252Cf;
 
        vector<pair<tkstring, double>> decay_modes;
 
        int year=0.;
 
        void print_data(const tkstring &_name, const tkdb_table::measure_data_struct &data) {
            if(data.value!=-1) cout<<left<<setw(33)<<_name<<": "<< setw(10) << data.value << " (" << setw(8) << data.err << ")" << setw(5) << data.unit << data.info << endl;
        }
 
        vector<pair<tkdb_table::measure_data_struct,tkstring>> sort_map(map<tkstring, tkdb_table::measure_data_struct> &map_in, bool inverse = false) {
 
            vector<pair<tkdb_table::measure_data_struct,tkstring>> sorted_map;
            sorted_map.reserve(map_in.size());
            std::transform(map_in.begin(), map_in.end(), std::back_inserter(sorted_map),
                           [](const auto &it) { return std::make_pair(it.second, it.first); });
            std::sort(std::begin(sorted_map),  std::end(sorted_map), [inverse](pair<tkdb_table::measure_data_struct,tkstring> a, pair<tkdb_table::measure_data_struct,tkstring> b) {
                if(inverse) return a.first.value > b.first.value;
                else return a.first.value <= b.first.value;
            });
 
            return sorted_map;
        }
 
        void print() {
            if(name.is_empty() || Z==-1 || N==-1 || A==-1) {
                glog << error << "empty nucleus" << do_endl;
                return;
            }
            cout<<left<<"Nucleus: "<<setw(5)<<name<<endl;
            cout<<"Z="<<setw(3)<<Z<<", N="<<setw(3)<<N<<", A="<<setw(3)<<A<<endl;
            print_data("Mass Excess",mass_excess);
            print_data("Quadrupole deformation",quadrupoleDeformation);
            print_data("Abundance",abundance);
            if(decay_modes.size()) {
                cout<<left<<setw(33)<<"Decay modes"<<": ";
                for(auto &mode: decay_modes) cout<< mode.first << " : " << tkstring::form("%g",mode.second) << " ; ";
                cout<<endl;
            }
            print_data("Radius",radius);
            print_data("Electirc quadrupole",electric_quadrupole);
            print_data("Magnetic dipole",magnetic_dipole);
 
        }
    };
 
    int fIsotopeIdx     = 0;
 
    std::map<int,tkn::tkstring> columns;
 
public:
    tkisotope_builder(tkdatabase *_database, const char* _table_name="ISOTOPE");
    virtual ~tkisotope_builder() override;
 
protected:
    void read_measure(const tkstring &key, const json &entry, tkdb_table::measure_data_struct &_data, bool _fromstd=false);
    json_nucleus read_nucleus(json &entry);
    void fill_isotope(const json_nucleus &_json_nuc);
 
public:
    int fill_database(const char* _json_filename, const char *_gs_filename, int _only_charge=0, int _only_mass=0);
 
#ifdef HAS_ROOT
    ClassDefOverride(tkisotope_builder,0);
#endif
};
 
}
#endif