RootTracer.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * See https://github.com/tferber/OrcaKinfit, forked from                 *
 * https://github.com/iLCSoft/MarlinKinfit                                *
 *                                                                        *
 * Further information about the fit engine and the user interface        *
 * provided in MarlinKinfit can be found at                               *
 * https://www.desy.de/~blist/kinfit/doc/html/                            *
 * and in the LCNotes LC-TOOL-2009-001 and LC-TOOL-2009-004 available     *
 * from http://www-flc.desy.de/lcnotes/                                   *
 *                                                                        *
 * Adopted by: Torben Ferber (torben.ferber@desy.de) (TF)                 *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#ifdef MARLIN_USE_ROOT
 
#ifndef __ROOTTRACER_H
#define __ROOTTRACER_H
 
#include <iostream>
#include "analysis/OrcaKinFit/BaseTracer.h"
 
class BaseFitter;
class TFile;
 
#include <TROOT.h>
#include <TChain.h>
#include <TFile.h>
 
//  Class RootTracer:
 
namespace Belle2 {
  namespace OrcaKinFit {
    class BaseFitter;
 
 
 
    class RootTracer: public BaseTracer {
    public:
      RootTracer(const char* filename = "trace.root", const char* option = "RECREATE");
      virtual ~RootTracer();
 
      RootTracer(const RootTracer&) = delete;
      RootTracer& operator=(const RootTracer&) = delete;
 
 
      virtual void initialize(BaseFitter& fitter) override;
      virtual void step(BaseFitter& fitter) override;
      virtual void substep(BaseFitter& fitter,
                           int flag
                          ) override;
      virtual void finish(BaseFitter& fitter) override;
 
      void printFitObjects(BaseFitter& fitter);
      void printConstraints(BaseFitter& fitter);
 
    protected:
      void SetBranchAddresses();
      void CreateBranches();
      void CreateEventBranches(BaseFitter& fitter);
      void FillParameterValues(BaseFitter& fitter);
 
      TFile* file;
      TTree* tree;
      TTree* eventtree;
 
      int istep;
      int isubstep;
 
      Int_t eventnumber;
      Int_t stepnumber;
      Int_t substepnumber;
      Double_t chi2;
 
      enum {NPARMAX = 100};
      Double_t parvalue[NPARMAX];
 
    };
 
  }// end OrcaKinFit namespace
} // end Belle2 namespace
 
#endif // __ROOTTRACER_H
 
#endif // MARLIN_USE_ROOT