MilleRecordWriter.h

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
 
#ifndef ALICEO2_FWDALIGN_MILLERECORD_WRITER_H
#define ALICEO2_FWDALIGN_MILLERECORD_WRITER_H
 
#include <Rtypes.h>
#include <TString.h>
 
#include "ForwardAlign/MillePedeRecord.h"
 
class TFile;
class TTree;
namespace o2
{
namespace fwdalign
{
 
class MilleRecordWriter
{
 public:
  MilleRecordWriter();
 
  virtual ~MilleRecordWriter();
 
  void setCyclicAutoSave(const long nEntries);
 
  void setDataFileName(TString fname) { mDataFileName = fname; }
 
  void changeDataBranchName(const bool isConstraintsRec = true);
 
  void init();
 
  bool isInitOk() const { return mIsSuccessfulInit; }
 
  o2::fwdalign::MillePedeRecord* getRecord() { return mRecord; };
 
  Long64_t getCurrentDataID() const { return mCurrentDataID; }
 
  void fillRecordTree(const bool doPrint = false);
 
  void terminate();
 
  void setRecordRun(int run);
 
  void setRecordWeight(double wgh);
 
 protected:
  TTree* mDataTree;                       
  TFile* mDataFile;                       
  bool mIsSuccessfulInit;                 
  bool mIsConstraintsRec;                 
  long mNEntriesAutoSave;                 
  TString mDataFileName;                  
  TString mDataTreeName;                  
  TString mDataBranchName;                
  o2::fwdalign::MillePedeRecord* mRecord; 
  Long64_t mCurrentDataID;                
 
  ClassDef(MilleRecordWriter, 0);
};
} // namespace fwdalign
} // namespace o2
 
#endif