txt2root.cxx

Go to the documentation of this file.

#include <iostream>
#include <fstream>
#include <iomanip>
#include <cstring>
#include <string>
#include <vector>
#include <TFile.h>
#include <TTree.h>
#include <TString.h>
 
using namespace std;
 
int main(int argc, char* argv[]){
  if(argc<2){
    cout << "please append txt file name" << endl;
    return 1;
  }
 
  TString str_txt_file(argv[1]);
  TString str_root_file(str_txt_file);
  str_root_file.ReplaceAll(".txt", ".root");
  TFile f(str_root_file, "recreate");
  TTree *tree = new TTree("track", "track");
  int run(0), event(0);
  tree->Branch("run", &run, "run/I");
  tree->Branch("event", &event, "event/I");
  // read in txt file and assign the numbers to run and event
  ifstream in(str_txt_file);
  if(!in){
    cout << "cannot open " << str_txt_file << endl;
    return 0;
  }
  char str[255];
  while(in){
    in.getline(str, 255);
    sscanf(str ,"%d %d", &run, &event);
    if(!run) continue; // protect of read at the end
    tree->Fill();
  }
 
  tree->Write();
  f.Close();
 
  return 0;
}