TrgEclTimingCalibration.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.                  *
 **************************************************************************/
#ifndef TRGECLTCALIBRATION_h
#define TRGECLTCALIBRATION_h
 
 
#include <iostream>
#include <fstream>
#include <vector>
 
#include "TMatrixD.h"
#include "TVectorD.h"
#include "TVector3.h"
 
#include <framework/core/Module.h>
#include <trg/ecl/TrgEclMapping.h>
 
#include "trg/ecl/dataobjects/TRGECLHit.h"
#include "trg/ecl/dataobjects/TRGECLTrg.h"
#include "trg/ecl/dataobjects/TRGECLCluster.h"
#include "trg/ecl/dataobjects/TRGECLUnpackerStore.h"
#include "trg/ecl/dataobjects/TRGECLUnpackerEvtStore.h"
 
#include <framework/datastore/StoreObjPtr.h>
#include <framework/datastore/StoreArray.h>
 
 
namespace Belle2 {
  class TRGECLTimingCalModule : public Module {
 
  public:
    TRGECLTimingCalModule();
    virtual ~TRGECLTimingCalModule();
 
  public:
    virtual void initialize() override;
    virtual void beginRun() override;
    virtual void event() override;
    virtual void endRun() override;
    virtual void terminate() override;
 
    void Set_TCposition();
 
    int  Solve_Matrix();
 
    int  ReadData();
 
    int  FillMatrix();
 
    void InitParams();
 
    void Save_Result();
 
    void Calculate_Chisq();
 
    void TRGECLTimingCalClear();
 
    void GetTimeOffset();
  private:
 
    /***** Calibration *****/
    int fSimulation;
    int fCalType;
    int f3Dbhabha_veto;
    int fInclude_FWD;
    int fInclude_BR;
    int fInclude_BWD;
    int fIteration;
 
    double cut_high_energy;
    double cut_low_energy;
    int cut_ntc;
    int TC_ref;
 
    int nevt_selected = 0;
    int nevt_read = 0;
    double TCEnergyCalibrationConstant;
    double cut_chisq;
 
    std::string str_default_name = "TCTimeOffset.root";
    std::string str_ofilename;
    std::string str_timeoffset_fname;
 
    std::vector<int>    TCId;
    std::vector<double> TCEnergy;
    std::vector<double> TCTiming;
    std::vector<TVector3> TCPosition;
 
    TMatrixDSym _Matrix;
    std::vector<std::vector<double>> Chisq1;
    std::vector<std::vector<double>> Chisq2;
    std::vector<std::vector<double>> Chisq3;
 
    std::vector<int> Nevent_TC;
    std::vector<double> chisq_result;
    TVectorD _Vector;
    std::vector<bool> b_Inc_TC;
    std::vector<bool> b_Exclude_TC;
    std::vector<double> tcal_result;
    std::vector<double> tcal_result_err;
 
    int FlagMatrixSolved = 0;
 
    std::vector<double> TimeOffset;
 
  };
}
#endif