KVIDZAFromZGridBatchMassCorrector.h

#pragma once
#include "KVBase.h"
#include "KVIDGridManager.h"
#include "KVIDZAFromZGrid.h"
 
class KVIDZAFromZGridBatchMassCorrector : public KVBase
{
    KVIDGridManager reference_grids;
    KVIDGridManager grids_to_correct;
    TFile reference_id_maps;
   bool apply_corrections = false;
   void RefitReferenceGridToData(const TString& ref_grid_name, const KVNameValueList& mass_fit_params);
 
   public:
   KVIDZAFromZGridBatchMassCorrector(const TString& ref_grids_file,
                                     const TString& bad_grids_file,
                                     const TString& bad_grids_id_map_file);
   KVIDZAFromZGridBatchMassCorrector(const TString& ref_grids_file,
                                     const TString& ref_id_map_file);
 
   void SetApplyCorrections(bool yes=true){ apply_corrections=yes; }
   void CorrectGrid(const TString& bad_grid_name, std::function<KVString (const KVString&)> deduce_ref_grid_name_from_bad_grid_name,
                    const KVNameValueList& mass_fit_parameters,
                    double min_yield_cutoff = 0.8,
                    std::function<KVString(const KVString&)> deduce_bad_histo_name = [](const KVString& n){ return n; });
   void CorrectGrid(KVIDZAFromZGrid* bad_grid, std::function<KVString (const KVString&)> deduce_ref_grid_name_from_bad_grid_name,
                    const KVNameValueList& mass_fit_parameters,
                    double min_yield_cutoff = 0.8,
                    std::function<KVString(const KVString&)> deduce_bad_histo_name = [](const KVString& n){ return n; });
   void CorrectGridToReference(KVIDZAFromZGrid* bad_grid, KVIDZAFromZGrid* ref_grid,
                               const KVNameValueList& mass_fit_parameters,
                    double min_yield_cutoff = 0.8,
                    std::function<KVString(const KVString&)> deduce_bad_histo_name = [](const KVString& n){ return n; });
   void CorrectGridToReference(const TString& bad_grid_name, const TString& ref_grid_name,
                    const KVNameValueList& mass_fit_parameters,
                    double min_yield_cutoff = 0.8,
                    std::function<KVString(const KVString&)> deduce_bad_histo_name = [](const KVString& n){ return n; });
   void CorrectGrids(std::function<KVString (const KVString&)> deduce_ref_grid_name_from_bad_grid_name,
                     const KVNameValueList& mass_fit_parameters,
                     double min_yield_cutoff = 0.8,
                     std::function<KVString(const KVString&)> deduce_bad_histo_name = [](const KVString& n){ return n; }
                     );
 
   void SaveCorrectedGrids(const TString& filename);
 
   void RefitAllReferenceGridsToData(const KVNameValueList& mass_fit_params);
   void SaveReferenceGrids(const TString& filename);
 
   void DrawReferenceFit(const TString& grid_name, Int_t Z) const;
   void DrawOtherFit(const TString& grid_name, Int_t Z) const;
 
   void StartViewerReferenceGrids()
   {
      reference_grids.StartViewer();
   }
   void StartViewerBadGrids()
   {
      grids_to_correct.StartViewer();
   }
 
   ClassDef(KVIDZAFromZGridBatchMassCorrector,0)
};