KVDataQualityAudit.h

#ifndef __KVDATAQUALITYAUDIT_H
#define __KVDATAQUALITYAUDIT_H
 
#include "KVBase.h"
 
#include <KVNumberList.h>
#include <KVReconstructedNucleus.h>
#include <KVUniqueNameList.h>
 
class KVDataQualityAudit : public KVBase {
private:
   KVUniqueNameList telescopes;
 
   Long64_t merge(KVDataQualityAudit*);
 
public:
   KVDataQualityAudit() : KVDataQualityAudit("") {}
   KVDataQualityAudit(const Char_t* name, const Char_t* title = "")
      : KVBase(name, title)
   {
      telescopes.SetOwner();
   }
   virtual ~KVDataQualityAudit() {}
   void Add(const KVReconstructedNucleus& N);
   void Print(Option_t* opt = "") const;
   const KVSeqCollection* GetTelescopeList() const
   {
      return &telescopes;
   }
   Bool_t HasTelescope(const TString& tel_name) const
   {
      return telescopes.FindObject(tel_name) != nullptr;
   }
   Long64_t Merge(TCollection*);
 
   struct isotope {
      Double_t counts{0};
      Float_t emin{-1.0};
      UShort_t A{0};
      void add(const KVReconstructedNucleus& N);
      void print() const;
      void merge(const isotope& isoto)
      {
         if (emin > 0) {
            if (isoto.emin > 0) emin = std::min(emin, isoto.emin);
         }
         else if (isoto.emin > 0) emin = isoto.emin;
         counts += isoto.counts;
      }
 
      ClassDef(isotope, 1)
   };
   struct element {
      Double_t counts{0};
      Float_t emin{-1.0};
      UShort_t A{0};// default (calculated) mass given when no isotopic identification available
      UChar_t Z{0};
      std::map<int, isotope> isotopes;
      void add(const KVReconstructedNucleus& N);
      void print() const;
      void merge(const element&);
      Bool_t HasIsotopes() const
      {
         return !isotopes.empty();
      }
      Bool_t HasIsotope(int A) const
      {
         return isotopes.find(A) != std::end(isotopes);
      }
      const isotope& GetIsotope(int A) const
      {
         return isotopes.at(A);
      }
      KVNumberList GetIsotopeList() const
      {
         KVNumberList l;
         for (auto& p : isotopes) l.Add(p.first);
         return l;
      }
      double get_mean_isotopic_mass() const;
      int get_max_isotopic_mass() const;
      std::map<int, double> get_isotopic_distribution() const;
      int get_min_isotopic_mass() const;
      double get_minimum_isotopic_threshold_mev_per_nuc() const;
      int get_default_mass() const
      {
         return A;
      }
 
      ClassDef(element, 1)
   };
   class idtelescope : public TNamed {
      std::map<int, element> elements;
   public:
      idtelescope(const TString& name = "") : TNamed(name, name) {}
      virtual ~idtelescope() {}
      ROOT_COPY_CTOR(idtelescope, TNamed)
      ROOT_COPY_ASSIGN_OP(idtelescope)
      void Copy(TObject& o) const
      {
         TNamed::Copy(o);
         idtelescope& other = (idtelescope&)o;
         other.elements = elements;
      }
      void add(const KVReconstructedNucleus& N);
      void Print(Option_t* opt = "") const;
      void merge(const idtelescope*);
      Bool_t HasElement(int Z) const
      {
         return elements.find(Z) != std::end(elements);
      }
      const element& GetElement(int Z) const
      {
         return elements.at(Z);
      }
      KVNumberList GetElementList() const
      {
 
         KVNumberList l;
         for (auto& p : elements) l.Add(p.first);
         return l;
      }
      std::map<int, double> get_element_distribution() const;
      double get_mean_Z() const;
      int get_max_Z() const;
      double get_mean_A() const;
      int get_max_A() const;
      int get_min_A() const;
      int get_min_Z() const;
      int get_max_Z_with_isotopes() const;
 
      ClassDef(idtelescope, 1)
   };
 
   Bool_t HasTelescope(const idtelescope* idt) const
   {
      return HasTelescope(idt->GetName());
   }
   idtelescope* GetTelescope(const TString& tel_name) const
   {
      return (idtelescope*)telescopes.FindObject(tel_name);
   }
   idtelescope* GetTelescope(const idtelescope* idt) const
   {
      return GetTelescope(idt->GetName());
   }
 
private:
   void add(const idtelescope*);
 
   ClassDef(KVDataQualityAudit, 1) //Audit of experimental data identification and calibrations
};
 
#endif