MicromegasRawDataEvaluation.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file MicromegasRawDataEvaluation.h

#ifndef MICROMEGAS_MicromegasRawDataEvaluation_H
#define MICROMEGAS_MicromegasRawDataEvaluation_H

#include "MicromegasCalibrationData.h"
#include "MicromegasMapping.h"

#include <fun4all/SubsysReco.h>
#include <phool/PHObject.h>

#include <TTree.h>

#include <map>
#include <memory>
#include <string>
#include <utility>

class PHCompositeNode;
class TFile;
class TH1;
class TH2;
class TProfile;

class MicromegasRawDataEvaluation : public SubsysReco
{
  public:

  MicromegasRawDataEvaluation( const std::string &name = "MicromegasRawDataEvaluation" );

  int Init(PHCompositeNode*) override;

  int InitRun(PHCompositeNode*) override;

  int process_event(PHCompositeNode*) override;

  int End(PHCompositeNode*) override;

  void set_calibration_file( const std::string& value ) { m_calibration_filename = value; }

  void set_n_sigma( double value ) { m_n_sigma = value; }


  void set_min_adc( double value ) { m_min_adc = value; }

  void set_sample_min( int value ) { m_sample_min = value; }

  void set_sample_max( int value ) { m_sample_max = value; }

  void set_evaluation_outputfile(const std::string &outputfile) {m_evaluation_filename = outputfile;}

  class Sample
  {
    public:
    unsigned int packet_id = 0;

    uint64_t lvl1_bco = 0;

    unsigned int fee_bco = 0;

    unsigned int checksum = 0;
    unsigned int checksum_error = 0;

    unsigned short fee_id = 0;
    unsigned short layer = 0;
    unsigned short tile = 0;

    unsigned short sampa_address = 0;
    unsigned short sampa_channel = 0;

    unsigned short channel = 0;

    unsigned short strip = 0;

    unsigned short sample = 0;
    unsigned short adc = 0;
    
    double pedestal = 0;
    double rms = 0;

    using List = std::vector<Sample>;
  };

  class Waveform
  {
    public:
    unsigned int packet_id = 0;

    uint64_t lvl1_bco = 0;

    unsigned int fee_bco = 0;

    unsigned int checksum = 0;
    unsigned int checksum_error = 0;

    unsigned short fee_id = 0;
    unsigned short layer = 0;
    unsigned short tile = 0;

    unsigned short sampa_address = 0;
    unsigned short sampa_channel = 0;

    unsigned short channel = 0;

    unsigned short strip = 0;

    unsigned short sample_max = 0;
    unsigned short adc_max = 0;
    
    double pedestal = 0;
    double rms = 0;

    bool is_signal = false;

    Waveform() = default;

    Waveform( const Sample& sample )
    { copy_from( sample ); }

    void copy_from( const Sample& );

    using List = std::vector<Waveform>;
  };


  class Container: public PHObject
  {
    public:
    void Reset();
    
    // number of taggers for each packet
    std::vector<int> n_tagger;
    
    // number of waveform for each packet
    std::vector<int> n_waveform;

    Waveform::List waveforms;
    Sample::List samples;
    
    // bco for this event
    std::vector<uint64_t> lvl1_bco_list;
    
    // lvl1 count for this event
    std::vector<uint32_t> lvl1_count_list;
    
    ClassDef(Container,1)
  };

  private:

  std::string m_calibration_filename = "TPOT_Pedestal_000.root";

  MicromegasCalibrationData m_calibration_data;

  MicromegasMapping m_mapping;

  double m_n_sigma = 5;

  /* This removes faulty channels for which calibration has failed */
  double m_min_adc = 50;
  
  int m_sample_min = 0;

  int m_sample_max = 100;

  std::string m_evaluation_filename = "MicromegasRawDataEvaluation.root";
  std::unique_ptr<TFile> m_evaluation_file;

  TTree* m_evaluation_tree = nullptr;

  Container* m_container = nullptr;
  
  using bco_matching_pair_t = std::pair<unsigned int, uint64_t>;

  using fee_bco_matching_map_t = std::map<unsigned short, bco_matching_pair_t>;
  fee_bco_matching_map_t m_fee_bco_matching_map;
  

  using bco_map_t = std::map<uint64_t,unsigned int>;
  bco_map_t m_bco_map;

};

#endif