ITS/AliITSNeuralExport.C

   1 void AliITSNeuralExport(Int_t min_count_4_good = 5)
   2 {
   3         TFile *f_in = new TFile("its_neural_new.root");
   4         TTree *t_in = (TTree*)f_in->Get("TreeT");
   5
   6         AliITSNeuralTrack *in_track = 0;
   7         t_in->SetBranchAddress("Tracks", &in_track);
   8         Int_t in_entries = (Int_t)t_in->GetEntries();
   9
  10         TFile *f_out = new TFile("its_neural_tracks.root", "RECREATE");
  11         TTree *t_out = new TTree("TreeT0", "Tree of exported tracks from neural tracking into V1 IO track");
  12
  13         AliITSIOTrack *out_track = 0;
  14         t_out->Branch("ITStracks", "AliITSIOTrack", &out_track);
  15
  16         for (Int_t i = 0; i < in_entries; i++) {
  17                 t_in->GetEntry(i);
  18                 out_track = in_track->ExportIOtrack(min_count_4_good);
  19                 t_out->Fill();
  20         }
  21
  22         f_out->cd();
  23         t_out->Write();
  24         f_out->Close();
  25 }