Fork me on GitHub

Changes in / [1808db5:fb98f40] in git


Ignore:
Files:
1 deleted
1 edited

Legend:

Unmodified
Added
Removed
  • readers/DelphesPythia8.cpp

    r1808db5 rfb98f40  
    144144  interrupted = true;
    145145}
    146 
    147 
    148 // Single-particle gun. The particle must be a colour singlet.
    149 // Input: flavour, energy, direction (theta, phi).
    150 // If theta < 0 then random choice over solid angle.
    151 // Optional final argument to put particle at rest => E = m.
    152 // from pythia8 example 21
    153 
    154 void fillParticle(int id, double ee_max, double thetaIn, double phiIn,
    155   Pythia8::Event& event, Pythia8::ParticleData& pdt, Pythia8::Rndm& rndm, bool atRest = false) {
    156 
    157   // Reset event record to allow for new event.
    158   event.reset();
    159 
    160   // Angles uniform in solid angle.
    161   double cThe, sThe, phi, ee;
    162   cThe = 2. * rndm.flat() - 1.;
    163   sThe = Pythia8::sqrtpos(1. - cThe * cThe);
    164   phi = 2. * M_PI * rndm.flat();
    165   ee = pow(10,1+(log10(ee_max)-1)*rndm.flat());
    166   double mm = pdt.mSel(id);
    167   double pp = Pythia8::sqrtpos(ee*ee - mm*mm);
    168 
    169   // Store the particle in the event record.
    170   event.append( id, 1, 0, 0, pp * sThe * cos(phi), pp * sThe * sin(phi), pp * cThe, ee, mm);
    171 
    172 }
    173 
    174 void fillPartons(int type, double ee_max, Pythia8::Event& event, Pythia8::ParticleData& pdt,
    175   Pythia8::Rndm& rndm) {
    176 
    177   // Reset event record to allow for new event.
    178   event.reset();
    179 
    180   // Angles uniform in solid angle.
    181   double cThe, sThe, phi, ee;
    182 
    183   // Information on a q qbar system, to be hadronized.
    184 
    185   cThe = 2. * rndm.flat() - 1.;
    186   sThe = Pythia8::sqrtpos(1. - cThe * cThe);
    187   phi = 2. * M_PI * rndm.flat();
    188   ee = pow(10,1+(log10(ee_max)-1)*rndm.flat());
    189   double mm = pdt.m0(type);
    190   double pp = Pythia8::sqrtpos(ee*ee - mm*mm);
    191   if (type == 21)
    192   {
    193     event.append( 21, 23, 101, 102, pp * sThe * cos(phi), pp * sThe * sin(phi), pp * cThe, ee);
    194     event.append( 21, 23, 102, 101, -pp * sThe * cos(phi), -pp * sThe * sin(phi), -pp * cThe, ee);
    195   }
    196   else
    197   {
    198     event.append(  type, 23, 101,   0, pp * sThe * cos(phi), pp * sThe * sin(phi), pp * cThe, ee, mm);
    199     event.append( -type, 23,   0, 101, -pp * sThe * cos(phi), -pp * sThe * sin(phi), -pp * cThe, ee, mm);
    200   }
    201 }
    202 
    203146
    204147//---------------------------------------------------------------------------
     
    270213    // Initialize Pythia
    271214    pythia = new Pythia8::Pythia;
    272     //Pythia8::Event& event = pythia->event;
    273     //Pythia8::ParticleData& pdt = pythia->particleData;
    274215
    275216    if(pythia == NULL)
     
    289230    timesAllowErrors = pythia->mode("Main:timesAllowErrors");
    290231
    291     // Check if particle gun
    292     if (!pythia->flag("Main:spareFlag1"))
    293     {
    294       inputFile = fopen(pythia->word("Beams:LHEF").c_str(), "r");
    295       if(inputFile)
    296       {
    297         reader = new DelphesLHEFReader;
    298         reader->SetInputFile(inputFile);
    299 
    300         branchEventLHEF = treeWriter->NewBranch("EventLHEF", LHEFEvent::Class());
    301         branchWeightLHEF = treeWriter->NewBranch("WeightLHEF", LHEFWeight::Class());
    302 
    303         allParticleOutputArrayLHEF = modularDelphes->ExportArray("allParticlesLHEF");
    304         stableParticleOutputArrayLHEF = modularDelphes->ExportArray("stableParticlesLHEF");
    305         partonOutputArrayLHEF = modularDelphes->ExportArray("partonsLHEF");
    306       }
     232    inputFile = fopen(pythia->word("Beams:LHEF").c_str(), "r");
     233    if(inputFile)
     234    {
     235      reader = new DelphesLHEFReader;
     236      reader->SetInputFile(inputFile);
     237
     238      branchEventLHEF = treeWriter->NewBranch("EventLHEF", LHEFEvent::Class());
     239      branchWeightLHEF = treeWriter->NewBranch("WeightLHEF", LHEFWeight::Class());
     240
     241      allParticleOutputArrayLHEF = modularDelphes->ExportArray("allParticlesLHEF");
     242      stableParticleOutputArrayLHEF = modularDelphes->ExportArray("stableParticlesLHEF");
     243      partonOutputArrayLHEF = modularDelphes->ExportArray("partonsLHEF");
    307244    }
    308245
     
    323260      while(reader && reader->ReadBlock(factory, allParticleOutputArrayLHEF,
    324261        stableParticleOutputArrayLHEF, partonOutputArrayLHEF) && !reader->EventReady());
    325 
    326       if (pythia->flag("Main:spareFlag1"))
    327       {
    328         if (pythia->mode("Main:spareMode1") == 11 || pythia->mode("Main:spareMode1") == 13 || pythia->mode("Main:spareMode1") == 22)
    329         {
    330           fillParticle( pythia->mode("Main:spareMode1"), pythia->parm("Main:spareParm1"), -1., 0.,pythia->event, pythia->particleData, pythia->rndm, 0);
    331         }
    332         else fillPartons( pythia->mode("Main:spareMode1"), pythia->parm("Main:spareParm1"), pythia->event, pythia->particleData, pythia->rndm);
    333       }
    334262
    335263      if(!pythia->next())
Note: See TracChangeset for help on using the changeset viewer.