KVDataPatch_E884_DataUniformization.cpp

#include "KVDataPatch_E884_DataUniformization.h"
#include "KVExpDB.h"
#include "KVFAZIA.h"
 
 
 
class KVDataPatch_E884_DataUniformization::_impl
{
   public:
   _impl()
   {
      runlists[zn_zn_1] = "5-14";
      bad_id_lists[zn_zn_1][KVFAZIA::IDCodes::ID_SI1_PSA] =
      {"212, 211, 214, 443, 743, 834, 942, 1041, 1042, 1043, 1111"};
      bad_id_lists[zn_zn_1][KVFAZIA::IDCodes::ID_SI1_SI2] =
      {"112, 812, 12, 333, 442, 443, 941"};
      bad_id_lists[zn_zn_1][KVFAZIA::IDCodes::ID_SI2_CSI] =
      {"613, 821, 924"};
 
      runlists[zn_zn_2] = "15-23 61";
      bad_id_lists[zn_zn_2][KVFAZIA::IDCodes::ID_SI1_PSA] =
      {"212, 211, 214, 342, 531, 743, 811, 833, 834, 1041, 1042, 1043, 1111"};
      bad_id_lists[zn_zn_2][KVFAZIA::IDCodes::ID_SI1_SI2] =
      {"112, 812, 333, 342, 442, 443, 911"};
      bad_id_lists[zn_zn_2][KVFAZIA::IDCodes::ID_SI2_CSI] =
      {"821, 924"};
 
      runlists[zn_bi_1] = "26-43";
      bad_id_lists[zn_bi_1][KVFAZIA::IDCodes::ID_SI1_PSA] =
      {"212, 211, 214, 321, 313, 333, 342, 443, 533, 743, 833, 834, 813, 811, 942, 1031, 1041, 1042, 1043, 1111"};
      bad_id_lists[zn_bi_1][KVFAZIA::IDCodes::ID_SI1_SI2] =
      {"112, 812, 333, 342, 12, 442, 443, 533, 911, 941, 442, 1133"};
      bad_id_lists[zn_bi_1][KVFAZIA::IDCodes::ID_SI2_CSI] =
      {"123, 821, 924"};
 
      runlists[zn_al] = "44-53";
      bad_id_lists[zn_al][KVFAZIA::IDCodes::ID_SI1_PSA] =
      {"212, 211, 214, 342, 443, 811, 832, 1041, 1042, 1043"};
      bad_id_lists[zn_al][KVFAZIA::IDCodes::ID_SI1_SI2] =
      {"112, 342, 812, 442, 941"};
      bad_id_lists[zn_al][KVFAZIA::IDCodes::ID_SI2_CSI] =
      {"821, 924"};
 
      runlists[zn_bi_2] = "54-60";
      bad_id_lists[zn_bi_2][KVFAZIA::IDCodes::ID_SI1_PSA] =
      {"211, 212, 214, 342, 443, 531, 743, 832, 834, 942, 1041, 1042, 1043, 1111"};
      bad_id_lists[zn_bi_2][KVFAZIA::IDCodes::ID_SI1_SI2] =
      {"112, 812, 333, 442, 443, 533, 543, 832, 911, 941"};
      bad_id_lists[zn_bi_2][KVFAZIA::IDCodes::ID_SI2_CSI] =
      {"123, 543, 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);
   }
   void print_excluded()
   {
      std::cout << "\tSI1-PSA:\t" << bad_id_lists[this_system][KVFAZIA::IDCodes::ID_SI1_PSA].AsString() << std::endl;
      std::cout << "\tSI1-SI2:\t" << bad_id_lists[this_system][KVFAZIA::IDCodes::ID_SI1_SI2].AsString() << std::endl;
      std::cout << "\tSI2-CSI:\t" << bad_id_lists[this_system][KVFAZIA::IDCodes::ID_SI2_CSI].AsString() << std::endl;
   }
   private:
   enum system_t
   {
      zn_zn_1,
      zn_zn_2,
      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 FAZIA telescopes with atypical behaviour\n\n";
   std::cout << "Based on results of data quality audits, we exclude from analysis (declare as 'not OK') 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";
   std::cout << "The excluded particles have a string parameter 'EXCLUDED'='Excluded by KVDataPatch_E884_DataUniformization'\n\n";
   std::cout << "For this run, the excluded FAZIA telescopes are:\n";
   pimpl->print_excluded();
}