pythia6.cc

Go to the documentation of this file.

/***********************************************************************
*                                                                      *
* File: pythia6_jet.cc                                                 *
* This is a simple test program.                                       *
* Copyright \ufffd 2005 Torbj\ufffdrn Sj\ufffdstrand                   *
*                                                                      *
* Modified by sj and ca                                                *
*                                                                      *
***********************************************************************/
//#include "Pythia.h"

//#include "LHAFortran.h"

//#include "HepMCInterface.h"

// this is needed to generate some random numbers
#include <stdlib.h>
#include <time.h>

#include <stdio.h>
#include <iostream>
#include <fstream>
#include <sstream>
#include "HepMC/PythiaWrapper.h"
#include "HepMC/IO_HEPEVT.h"
#include "HepMC/IO_Ascii.h"
#include "HepMC/GenEvent.h"
#include "HepMC/SimpleVector.h"
#include "MyPythia6Wrapper.h"

#include "TH1.h"
#include "TFile.h"
#include "TMath.h"
#include <TCanvas.h>

#include "TF1.h"
#include <TH2.h>
#include <TStyle.h>
#include "TLegend.h"
#include <TTree.h>
#include <TString.h>

using namespace std;

// include our own classes
#include "../../include/TopAnalysis.h"
#include "../../include/TauAnalysis.h"
#include "../../include/DiJetAnalysis.h"
#include "../../include/WplusJetAnalysis.h"
#include "../../include/ZAnalysis.h"
#include "../../include/UEAnalysis.h"
#include "readpythiaConfigFile.h"
#include "../../include/Configuration.h"

/**************************************************************************/

int main(int argc, const char* argv[]) {
    
  if(argc <= 1 ) {
    std::cout << "Usage: " << argv[0] << " <Configuration File>" << std::endl;
    return 0;
  }
  
  
  Configuration* currentconfig = new Configuration();
  if(currentconfig->read(argv[1]))
    {
      std::cout << " ... skip !" << std::endl;
      return 0;
    }
  
  DiJetAnalysis dijet;
  WplusJetAnalysis wplusjet;
  ZAnalysis z0;
  TauAnalysis tau;
  TopAnalysis top;
  UEAnalysis ue;

  // create analysis Objects and initialize them
  if(currentconfig->dijet){
    dijet.Init(currentconfig->max_eta,currentconfig->min_pt);
    dijet.InitJetFinder(currentconfig->coneRadius,currentconfig->overlapThreshold,currentconfig->jet_ptmin);
  }
  
  if(currentconfig->wplusjet){
    wplusjet.Init(currentconfig->max_eta,currentconfig->min_pt);
    wplusjet.InitJetFinder(currentconfig->coneRadius,currentconfig->overlapThreshold,currentconfig->jet_ptmin);
  }
  
  if(currentconfig->z){
    z0.Init(currentconfig->max_eta,currentconfig->min_pt);
    z0.InitJetFinder(currentconfig->coneRadius,currentconfig->overlapThreshold,currentconfig->jet_ptmin);
  }
  if(currentconfig->tau){
    tau.Init(currentconfig->max_eta,currentconfig->min_pt);
    tau.InitJetFinder(currentconfig->coneRadius,currentconfig->overlapThreshold,currentconfig->jet_ptmin);
  }
  if(currentconfig->top){  
    top.Init(currentconfig->max_eta,currentconfig->min_pt);
    top.InitJetFinder(currentconfig->coneRadius,currentconfig->overlapThreshold,currentconfig->jet_ptmin);
  }
  if(currentconfig->ue){
    ue.Init(currentconfig->max_eta,currentconfig->min_pt);
    ue.InitJetFinder(currentconfig->coneRadius,currentconfig->overlapThreshold,currentconfig->jet_ptmin);
  }
  
  
  //........................................HEPEVT
  //  Standard Pythia >6.1 uses HEPEVT with NMXHEP=4000 entries and 8-byte
  //  floating point numbers. We need to explicitly pass this information
  //  to the HEPEVT_Wrapper.
  pyjets.n=0; // ensure dummyness of the next call
  call_pyhepc(1); // mstu(8) is set to NMXHEP in this dummy call (version >=6.404)
  //HepMC::HEPEVT_Wrapper::set_max_number_entries(pydat1.mstu[8-1]);
  HepMC::HEPEVT_Wrapper::set_max_number_entries(10000);
  HepMC::HEPEVT_Wrapper::set_sizeof_real(8);
  
  //
  //........................................PYTHIA INITIALIZATIONS
  // (Some platforms may require the initialization of pythia PYDATA block
  //  data as external - if you get pythia initialization errors try
  //  commenting in/out the below call to initpydata() )
  // initpydata();
  //
  
  // set random number seed (mandatory!) ... this may be overwritten by function readConfigFile
  pydatr.mrpy[1-1] = currentconfig->rseed ;
  
  //int tunemode=300;
  //pytune( &tunemode );
  
  std::cout << "Configuration Parameter List:" << std::endl;
  
  for (unsigned int files=0; files<(currentconfig->ConfigFileNames).size(); files++)
    readConfigFile((currentconfig->ConfigFileNames)[files]);

  // master switches for FSR, ISR and MI
  pypars.mstp[61-1]=int(currentconfig->ISR); 
  pypars.mstp[71-1]=int(currentconfig->FSR); 
  pypars.mstp[81-1]=int(currentconfig->MI); 

  // Call pythia initialization
  call_pyinit("CMS", "p", "p", 14000.);
  
  //........................................HepMC INITIALIZATIONS
  //
  // Instantiate an IO strategy for reading from HEPEVT.
  HepMC::IO_HEPEVT hepevtio;
  //
  // Instantiate an IO strategy to write the data to file - it uses the 
  //  same ParticleDataTable
  //HepMC::IO_Ascii ascii_io("test_pythia_hepmc.dat",std::ios::out);
  
  //.......EVENT LOOP:
  
  int nShow = 50;
  int nShowPace = max(1,int(currentconfig->nevents)/nShow);
  
  for (unsigned int i = 1 ; i <= currentconfig->nevents; i++) {
    
    // Begin event loop. 
    if (i%nShowPace == 0) cout << "pythia6:  Now begin event "
                               << i << endl;
    
    call_pyevnt();      // generate one event with Pythia
    // pythia pyhepc routine converts common PYJETS in common HEPEVT
    call_pyhepc(1);
    HepMC::GenEvent* evt = hepevtio.read_next_event();
    // add some information to the event
    evt->set_event_number(i);
    evt->set_signal_process_id(20);
    // write the event out to the ascii file
    //    ascii_io << evt;
    
    int ret_dijet = SUCCESS;
    int ret_wplusjet = SUCCESS;
    int ret_z0 = SUCCESS;
    int ret_tau = SUCCESS;
    int ret_top = SUCCESS;
    int ret_ue = SUCCESS;

    //call analysis
    if(currentconfig->dijet)       ret_dijet = dijet.Process(evt);
    if(currentconfig->wplusjet)    ret_wplusjet = wplusjet.Process(evt);
    if(currentconfig->z)           ret_z0 = z0.Process(evt);
    if(currentconfig->tau)  ret_tau = tau.Process(evt);
    if(currentconfig->top)  ret_top = top.Process(evt);
    if(currentconfig->ue)   ret_ue = ue.Process(evt);
    
    if (ret_dijet==FAILURE || ret_wplusjet==FAILURE || ret_z0==FAILURE 
        || ret_tau==FAILURE || ret_top==FAILURE || ret_ue==FAILURE)
      {
        std::cout << "Processing event number " << i << std::endl;    
        std::cout << "throw FAILURE"<<std::endl;
        //std::cout<<"pylist"<<std::endl;
        //call_pylist(2);
        //std::cout<<std::endl;
        //std::cout<<"HepMC Output"<<std::endl;
        //evt->print();
        std::cout<<std::endl;
      }

    //delete all the jet objects to be free for the next event
    if(currentconfig->dijet)       dijet.DeleteJetObject(evt);
    if(currentconfig->wplusjet)    wplusjet.DeleteJetObject(evt);
    if(currentconfig->z)         z0.DeleteJetObject(evt);
    if(currentconfig->tau)       tau.DeleteJetObject(evt);
    if(currentconfig->top)       top.DeleteJetObject(evt);
    if(currentconfig->ue)        ue.DeleteJetObject(evt);

    // we also need to delete the created event from memory
    delete evt;
  } // end of event loop
  
  
  // write out some information from Pythia to the screen
  call_pystat(1);    
  
  TFile* f = new TFile(currentconfig->OutputFileName.c_str(),"RECREATE");
  if(currentconfig->dijet)       dijet.Finalize(f);
  if(currentconfig->wplusjet)    wplusjet.Finalize(f);
  if(currentconfig->z)         z0.Finalize(f);
  if(currentconfig->tau)       tau.Finalize(f);
  if(currentconfig->top)       top.Finalize(f);
  if(currentconfig->ue)        ue.finalize(f);
  
  std::cout<<"pythia6:  program ends"<<std::endl;
  f->Close();
  // Done.
  return 0;
  
}

---