KVIDZAFromZGridMassHomogenizer.h

#pragma once
 
#include "KVBase.h"
#include "KVIDGridManager.h"
#include "KVIDZAFromZGrid.h"
 
class KVIDZAFromZGridMassHomogenizer : public KVBase
{
    KVIDGridManager theGrids;
    TFile theData;
 
   using mass_dist_stat_t = std::unordered_map<std::string,std::vector<std::string>>;
    std::map<int,mass_dist_stat_t> all_mass_ranges;
 
   void correct_masses_for_grid(const TString& grid_name, int Z, int deltaA,
                                const KVNameValueList& mass_fit_parameters);
   KVIDZAFromZGrid* GetGrid(TObject* obj)
   {
      auto gr = dynamic_cast<KVIDZAFromZGrid*>(obj);
      if(gr && !gr->HasRunList())
         return gr;
      return nullptr;
   }
   void Analyse(KVIDZAFromZGrid* grid,double min_yield);
   void Analyse(const TString& gridname,double min_yield)
   {
      auto gr = GetGrid(gridname);
      Analyse(gr,min_yield);
   }
 
   public:
   KVIDZAFromZGridMassHomogenizer(const TString& gridfile, const TString& datafile);
 
   void Analyse(double min_yield)
   {
 
      for(auto obj : *theGrids.GetGrids())
      {
         Analyse(GetGrid(obj),min_yield);
      }
   }
   KVIDZAFromZGrid* GetGrid(const TString& gridname)
   {
      return GetGrid(theGrids.GetGrid(gridname));
   }
   void Print() const;
   void ShowFits(int Z, const std::string& alist);
   void CorrectMasses(int Z, const std::string& alist, int deltaA,
                      const KVNameValueList& mass_fit_parameters);
   void CorrectMassesForTelescope(int Z, const TString& idtel, int deltaA,
                                  const KVNameValueList& mass_fit_parameters);
   void Clear()
   {
      all_mass_ranges.clear();
   }
   void SaveGrids(const TString& filename);
 
   ClassDefOverride(KVIDZAFromZGridMassHomogenizer,0)
};