txt2root.cxx File Reference

#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>

Go to the source code of this file.

Functions
int	main (int argc, char *argv[])

Function Documentation

main()

int main	(	int	argc,
		char *	argv[] )

Definition at line 15 of file txt2root.cxx.

                                {
  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;
}