KVIDZAFromZGridMassCorrector.h

#pragma once
 
#include "KVIDZAFromZGrid.h"
 
class KVIDZAFromZGridMassCorrector : public KVBase
{
   KVIDZAFromZGrid* refGrid = nullptr;
   KVIDZAFromZGrid* modGrid = nullptr;
 
   std::map<int,KVNumberList> not_good = {{2,"1-2 5 7 9-12"},{3,"1-5"},{4,"1-6 8"},{5,"1-7 9"},{6,"1-9"},{7,"1-12"},{8,"1-14"}};
 
   public:
   static int fOnlyShowZ;
 
   KVIDZAFromZGridMassCorrector(KVIDZAFromZGrid* _reference,
                                KVIDZAFromZGrid* _to_correct)
      : refGrid{_reference}, modGrid{_to_correct}
   {}
 
   using mass_correction_t = std::pair<int,int>;
   std::vector<mass_correction_t> DetermineMassCorrections(double min_yield);
 
   void ApplyCorrections(const std::vector<KVIDZAFromZGridMassCorrector::mass_correction_t>&, TH1* pid_dist,
                         const KVNameValueList& mass_fit_parameters);
 
   ClassDefOverride(KVIDZAFromZGridMassCorrector,0)
};