EclData.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
//STL
#include <vector>
#include <set>
 
//Root
#include <TTree.h>
#include <TH1F.h>
 
namespace Belle2 {
  class ECLCalDigit;
 
  class EclData {
  private:
    TTree* m_tree;
    int m_branch_ch;
    double m_branch_energy;
    double m_branch_time;
    int m_branch_evtn;
 
    int* m_event_counts;
    int m_event_count_max;
    float* m_energy_sums;
    float m_energy_sums_max;
    float m_energy_total;
 
    int m_last_event_id;
 
    int m_time_max;
    int m_time_range_min;
    int m_time_range_max;
 
    int m_ev_range_min;
    int m_ev_range_max;
 
    int m_en_range_min;
    int m_en_range_max;
 
    std::set<int> m_excluded_ch;
    std::vector<int> m_event_entry;
 
    static const int ring_start_id[70];
 
  public:
    enum EclSubsystem {ALL, BARR, FORW, BACKW};
 
    EclData();
 
    EclData(const EclData& data);
    EclData& operator=(const EclData& other);
 
 
    ~EclData();
 
  private:
    void cloneFrom(const EclData& other);
 
    void initVariables();
    void initEventRanges();
 
  public:
    static int getCrystalCount();
 
    TTree* getTree();
 
    int* getEventCounts();
    int* getEventCountsPerCrystal();
    int getEventCountsMax();
 
    float* getEnergySums();
    float* getEnergySumPerCrystal();
    float getEnergySumsMax();
    float getEnergyTotal();
 
    bool isCrystalInSubsystem(int crystal, EclSubsystem subsys);
 
    int getTimeRangeMin();
    int getTimeRangeMax();
    void setTimeRange(int time_min, int time_max, bool do_update = true);
 
    int getEventRangeMin();
    int getEventRangeMax();
    void setEventRange(int ev_min, int ev_max, bool do_update = true);
 
    void setEnergyThreshold(int en_min, int en_max, bool do_update = true);
 
    int getTimeMin();
    int getTimeMax();
 
    int getLastEventId();
 
    int getChannel(int phi_id, int theta_id);
    int getPhiId(int ch);
    int getThetaId(int ch);
 
    void excludeChannel(int ch, bool do_update = false);
    void includeChannel(int ch, bool do_update = false);
 
    void loadRootFile(const char* path);
 
    void update(bool reset_event_ranges = false);
 
    int addEvent(ECLCalDigit* event, int evtn);
    void fillEnergyHistogram(TH1F* hist, int energy_min, int energy_max, EclSubsystem subsys);
    void fillEnergySumHistogram(TH1F* hist, int energy_min, int energy_max, EclSubsystem subsys);
    void fillTimeHistogram(TH1F* hist, int time_min, int time_max, EclSubsystem subsys);
  };
}