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>
#include <stdlib.h>
#include <stdio.h>
 
using namespace std;
 
int main(int argc, char* argv[]){
  int ratio(1);
  if(argc<2){
    cout << "please append txt file name" << endl;
    cout << "usage: txt2root.exe filename [subratio]" << endl;
    cout << "ex: txt2root.exe jpsi09_radBB.txt 5" << endl;
    return 1;
  }
  if(argc==3){
    ratio = atoi(argv[2]);
  }
 
  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), index(0);
  tree->Branch("run", &run, "run/I");
  tree->Branch("event", &event, "event/I");
  tree->Branch("index", &index, "index/I");
  // read in txt file and assign the numbers to run, event and index of track
  ifstream in(str_txt_file);
  if(!in){
    cout << "cannot open " << str_txt_file << endl;
    return 0;
  }
  char str[255];
  int count(0);
  while(in){
    count ++; 
    in.getline(str, 255);
    sscanf(str ,"%d %d %d", &run, &event, &index);
    if(count%ratio != 0) continue;
    if(!run) continue; // protect of read at the end
    tree->Fill();
  }
 
  tree->Write();
  f.Close();
  cout << "total events: " << count << "  subratio: " << ratio << endl;
  return 0;
}