Fork me on GitHub

Changeset 466 in svn


Ignore:
Timestamp:
Jul 10, 2009, 4:54:59 PM (15 years ago)
Author:
Xavier Rouby
Message:

bug fixed in the number of desired events to process

Location:
trunk
Files:
5 edited

Legend:

Unmodified
Added
Removed

  • trunk/CHANGELOG

    r457 r466  
    66     (input event files, detector card, trigger card, beam files, particle table)
    77   - it is now possible to read "delphes root files" as an input for Delphes (GEN tree)
     8   - bug fix: number of events to process is correct event if multiple files in the inputlist
    89
    910 /-----------------------------------------\

  • trunk/src/HEPEVTConverter.cc

    r443 r466  
    5757//------------------------------------------------------------------------------
    5858
    59 //Nevents not yet implemented! 08.03.2009
    60 
    6159HEPTreeReader::HEPTreeReader(TTree *tree, HEPEvent *event) : fChain(tree), fCurrentTree(-1), fEvent(event)
    6260{
    … …  
    174172{
    175173  string buffer;
     174  int nevt_already_processed=0;
    176175
    177176  gROOT->SetBatch();
    … …  
    190189  while(1)
    191190    {
    192      
     191      if (Nevt>0 && nevt_already_processed >=Nevt) break; // enough events already processed
    193192      TChain *chain = new TChain("h101");
    194193      infile >> buffer;
    … …  
    203202        else checking_the_file.close();
    204203
    205       if(!infile.good()) break;
     204      if(!infile.good()) break;  // end of the list
    206205      chain->Add(buffer.c_str());
    207206     
    … …  
    214213     
    215214      TRootC::GenParticle *element;
    216       allEntries = (Nevt<0)?allEntries:min((int)allEntries,Nevt); // do not miss the "+1" 
     215      allEntries = (Nevt<0)?allEntries:min((int)allEntries,Nevt);
    217216      // Loop over all events
    218217      for(entry = 0; entry < allEntries; ++entry)
    219218        {
     219          if (Nevt>0 && nevt_already_processed >=Nevt) break; // enough events already processed
    220220          // Load selected branches with data from specified event
    221221          treeReader->ReadEntry(entry);
    … …  
    253253            }
    254254          treeWriter->Fill();
     255          ++nevt_already_processed;
    255256        }
    256257      delete chain;

  • trunk/src/HepMCConverter.cc

    r443 r466  
    270270  // generated particles from HEPEVT
    271271  ExRootTreeBranch *branchGenParticle = treeWriter->NewBranch("Particle", TRootC::GenParticle::Class());
    272  
     272 
     273  int nevt_already_processed=0;
     274 
    273275  // Open a stream connected to an event file:
    274276  ifstream infile(inputFileList.c_str());
    … …  
    285287    {
    286288      infile >> filename;
    287       if(!infile.good()) break;
     289      if(!infile.good()) break; // end of listfile reached
     290      if (Nevt>0 && nevt_already_processed >=Nevt) break; // enough events already processed   
     291
    288292      ifstream checking_the_file(filename.c_str());
    289293      if(!checking_the_file.good())
    … …  
    305309        {
    306310          if(Nevt>0 && entry>=Nevt) break;
     311          if(Nevt>0 && nevt_already_processed >=Nevt) break; // enough events already processed
    307312
    308313          treeWriter->Clear();
    … …  
    314319          treeWriter->Fill();
    315320          ++entry;
     321          ++nevt_already_processed;
    316322        }
    317323    }
    … …  
    322328}
    323329
    324 
    325 
    326 

  • trunk/src/LHEFConverter.cc

    r450 r466  
    171171  ExRootTreeBranch *branchParticle = treeWriter->NewBranch("Particle", TRootLHEFParticle::Class());
    172172  //ExRootTreeBranch *branchParticle = treeWriter->NewBranch("Particle", TRootC::GenParticle::Class());
     173
     174  int nevt_already_processed=0;
    173175 
    174176  // Open a stream connected to an event file:
    … …  
    186188      infile >> filename;
    187189      if(!infile.good()) break;
     190      if (Nevt>0 && nevt_already_processed >=Nevt) break; // enough events already processed
     191
    188192        ifstream checking_the_file(filename.c_str());
    189193        if(!checking_the_file.good())
    … …  
    206210          Long64_t entry = 0;
    207211          while(inputReader->readEvent()) {
     212              if (Nevt>0 && nevt_already_processed >=Nevt) break; // enough events already processed
    208213              treeWriter->Clear();
    209214              AnalyseEvent(inputReader, branchEvent, entry + 1);
    … …  
    211216              treeWriter->Fill();
    212217              ++entry;
     218              ++nevt_already_processed;
    213219              if(allEntries<entry+1) break;
    214220          }

  • trunk/src/STDHEPConverter.cc

    r443 r466  
    6060//---------------------------------------------------------------------------
    6161
    62 void STDHEPConverter::AnalyseParticles(ExRootTreeBranch *branch)
    63 {
    64   int Nmax = -1;  // useless for the moment -- should be changed in order
    65   if(Nmax>0) Nmax = min(Nmax,myhepevt.nhep); // to use Nevt in DataConverter
    66   else Nmax = myhepevt.nhep;
    67 
     62void STDHEPConverter::AnalyseParticles(ExRootTreeBranch *branch)
     63{
    6864  TRootC::GenParticle *element;
    6965
    … …  
    7268  Int_t number;
    7369
    74   for(number = 0; number < Nmax; ++number)
     70  for(number = 0; number < myhepevt.nhep; ++number)
    7571  {
    7672    element = static_cast<TRootC::GenParticle*>(branch->NewEntry());
    … …  
    119115  int ierr, entryType;
    120116  int istr = 0;
    121   int nevt = 0;
     117  int nevt = 0, nevt_already_processed=0;
    122118
    123119  ExRootTreeWriter *treeWriter = new ExRootTreeWriter(outputFileName, "GEN");
    … …  
    136132                    << left  << setw(20) << inputFileList                    <<""
    137133                    << right << setw(19) <<"for input **"<<""<<endl;
    138 
    139134      exit(1);
    140135    }
     136
     137  // Parsing the input list
    141138  while(1)
    142139    {
    143140      infile >> filename;
    144       if(!infile.good()) break;
    145         ifstream checking_the_file(filename.c_str());
    146         if(!checking_the_file.good())
    147            {
     141      if(!infile.good()) break; // end of the list
     142      if (Nevt>0 && nevt_already_processed >=Nevt) break; // enough events already processed
     143
     144      ifstream checking_the_file(filename.c_str());
     145      if(!checking_the_file.good()) {
    148146             cerr << left  << setw(30) <<"**   ERROR: Can't find file  "<<""
    149147                  << left  << setw(20) << filename                      <<""
    150148                  << right << setw(19) <<"for input **"<<""<<endl;
    151149             continue;
    152            }
    153         else checking_the_file.close();
     150       } else checking_the_file.close();
    154151     
    155152      ierr = StdHepXdrReadInit(const_cast<char*>(filename.c_str()), &nevt, istr);
    156      
    157       if(ierr != 0)
    158         {
     153
     154      if(ierr != 0) {
    159155          cerr <<"**                ERROR: Can't open file for input                 **"<<endl;
    160156          break;
    161         }
     157      }
     158
    162159
    163160      Long64_t allEntries = (Nevt<0)?nevt:min(nevt,Nevt+1); // do not miss the "+1"
    … …  
    170167            {
    171168              ierr = StdHepXdrRead(&entryType, istr);
     169              if (Nevt>0 && nevt_already_processed >=Nevt) break; // enough events already processed
    172170             
    173               if(ierr != 0)
    174                 {
     171              if(ierr != 0) {
    175172                  cerr << left  << setw(49) <<"**   ERROR: Unexpected end of file after"<<""
    176173                       << left  << setw(10) << entry                                    <<""
    177174                       << right << setw(10) <<"entries **"<<""<<endl;
    178175                  break;
    179                 }
     176              } // if error
    180177             
    181178              // analyse only entries with standard HEPEVT common block
    182               if(entryType == 1)
    183                 {
     179              if(entryType == 1) {
    184180                  // add empty events for missing event numbers
    185                   while(recordNumber < myhepevt.nevhep)
    186                     {
     181                  while(recordNumber < myhepevt.nevhep) {
    187182                      treeWriter->Clear();
    188183                      AnalyseEvent(branchGenEvent, recordNumber);
    189184                      treeWriter->Fill();
    190185                      ++recordNumber;
    191                     }
     186                      ++nevt_already_processed;
     187                  } // while
    192188                 
    193189                  treeWriter->Clear();
    194                  
    195190                  AnalyseEvent(branchGenEvent, myhepevt.nevhep);
    196191                  AnalyseParticles(branchGenParticle);
    197                  
    198192                  treeWriter->Fill();
    199                  
    200193                  ++recordNumber;
    201                  
    202                 }
    203             }
    204         }
     194                  ++nevt_already_processed;
     195                } // if
     196            } // for
     197        } // if allEntries >0
    205198      StdHepXdrEnd(istr);
    206     }
     199    } // while inputlist not empty
    207200  treeWriter->Write();
    208201 
Note: See TracChangeset for help on using the changeset viewer.