KVDataPatch_E884_DataUniformization.cpp

#include "KVDataPatch_E884_DataUniformization.h"
#include "KVExpDB.h"
#include "KVFAZIA.h"
 
 
 
class KVDataPatch_E884_DataUniformization::_impl
{
   public:
   _impl()
   {
      runlists[zn_bi_1] = "26-43";
      bad_id_lists[zn_bi_1][KVFAZIA::IDCodes::ID_SI1_PSA] =
      {"212, 211, 332, 644, 1042, 214, 313, 443, 533, 1041, 1043, 942"};
      bad_id_lists[zn_bi_1][KVFAZIA::IDCodes::ID_SI1_SI2] =
      {"211, 333, 12, 443, 533, 911, 1121, 941, 112, 812, 442"};
      bad_id_lists[zn_bi_1][KVFAZIA::IDCodes::ID_SI2_CSI] =
      {"123, 941, 821, 924"};
   }
   bool is_required(int run)
   {
      for(auto& nl : runlists)
         if(nl.second.Contains(run))
         {
            this_system = nl.first;
            return true;
         }
      return false;
   }
   bool is_ok(KVFAZIA::IDCodes idcode, int index)
   {
      return !bad_id_lists[this_system][idcode].Contains(index);
   }
   private:
   enum system_t
   {
      zn_zn,
      zn_bi_1,
      zn_bi_2,
      zn_al
   };
   system_t this_system;
   std::map<system_t,KVNumberList> runlists;
   using bad_id_list = std::map<KVFAZIA::IDCodes,KVNumberList>;
   std::map<system_t, bad_id_list> bad_id_lists;
};
 
 
KVDataPatch_E884_DataUniformization::KVDataPatch_E884_DataUniformization()
   : KVDataPatch("KVDataPatch_E884_DataUniformization",
                 "Removes all particles identified in telescopes with atypical behaviour based on results of data quality audits"),
     pimpl{std::make_unique<_impl>()}
{}
 
KVDataPatch_E884_DataUniformization::~KVDataPatch_E884_DataUniformization() = default;
 
 
 
Bool_t KVDataPatch_E884_DataUniformization::IsRequired(TString dataset, TString datatype, Int_t run_number, TString, Int_t, const TList*)
{
   return dataset=="INDRAFAZIA.E884" && datatype=="recon" && pimpl->is_required(run_number);
}
 
 
 
 
void KVDataPatch_E884_DataUniformization::ApplyToParticle(KVReconstructedNucleus* rnuc)
{
   if(!pimpl->is_ok((KVFAZIA::IDCodes)rnuc->GetIDCode(),rnuc->GetStoppingDetector()->GetIndex()))
   {
      rnuc->SetIsOK(kFALSE);
      rnuc->GetParameters()->SetValue("EXCLUDED", "Excluded by KVDataPatch_E884_DataUniformization");
   }
}
 
 
 
 
void KVDataPatch_E884_DataUniformization::PrintPatchInfo() const
{
   std::cout << "E884:  Data uniformization, removes all particles identified in telescopes with atypical behaviour\n\n";
   std::cout << "Based on results of data quality audits, we remove (declare as 'bad') any particles identified in\n";
   std::cout << "telescopes which do not follow the average behaviour of neighbouring telescopes (with similar theta)\n";
   std::cout << "concerning:\n";
   std::cout << "   + largest Z identified\n";
   std::cout << "   + mean Z identified;\n";
   std::cout << "   + thresholds for Z or A identification.\n\n";
}