e_starlightStandalone.cpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file e_starlightStandalone.cpp

//
//    Copyright 2010
//
//    This file is part of starlight.
//
//    starlight is free software: you can redistribute it and/or modify
//    it under the terms of the GNU General Public License as published by
//    the Free Software Foundation, either version 3 of the License, or
//    (at your option) any later version.
//
//    starlight is distributed in the hope that it will be useful,
//    but WITHOUT ANY WARRANTY; without even the implied warranty of
//    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
//    GNU General Public License for more details.
//
//    You should have received a copy of the GNU General Public License
//    along with starlight. If not, see <http://www.gnu.org/licenses/>.
//
//
// File and Version Information:
// $Rev:: 270                         $: revision of last commit
// $Author:: jnystrand                $: author of last commit
// $Date:: 2016-07-08 16:31:51 +0100 #$: date of last commit
//
// Description:
//
//
//


#include <iostream>
#include <chrono>
#include "reportingUtils.h"
#include "e_starlight.h"
#include "inputParameters.h"
#include "eventfilewriter.h"
#include "e_starlightStandalone.h"


using namespace std;


e_starlightStandalone::e_starlightStandalone()
  : _configFileName   ("slight.in"),
    _starlight        (0),
    _nmbEventsTot     (1),
    _nmbEventsPerFile (_nmbEventsTot)
{ }


e_starlightStandalone::~e_starlightStandalone()
{ }


bool
e_starlightStandalone::init()
{
  _inputParameters = new inputParameters();
  // read input parameters from config file
  _inputParameters->configureFromFile(_configFileName);
  if (!_inputParameters->init()) {
    printWarn << "problems initializing input parameters. cannot initialize starlight." << endl;
    return false;
  }

  // copy input file to one with baseFileName naming scheme
        std::string inputCopyName, _baseFileName;
        _baseFileName = _inputParameters->baseFileName();
  if (_baseFileName != "slight") {
         inputCopyName = _baseFileName +".in";

        ofstream inputCopyFile;
  inputCopyFile.open(inputCopyName.c_str());

        std::ifstream infile(_configFileName.c_str());
        if ((!infile) || (!infile.good()))
        {
          return -1;
        }

        int lineSize = 256;
        char tmp[lineSize];
        while (!infile.getline(tmp, lineSize).eof())
         {
       cout << tmp << endl;
         inputCopyFile << tmp << endl;
         }
        inputCopyFile.close();
  }

  // get the number of events
  // for now we write everything to one file
  _nmbEventsTot     = _inputParameters->nmbEvents();
  _nmbEventsPerFile = _nmbEventsTot;

  // create the starlight object
  _starlight = new e_starlight();

        // give starlight the input parameters
        _starlight->setInputParameters(_inputParameters);
  
  // initialize starlight
  return _starlight->init();
}


bool
e_starlightStandalone::run()
{
  if (!_starlight) {
    printWarn << "null pointer to starlight object. make sure that init() was called. "
              << "cannot generate events." << endl;
    return false;
  }

  // open output file
    eventFileWriter fileWriter;
  eventFileWriter fileWriterLUND;
  fileWriter.writeFullPythiaInfo(_inputParameters->pythiaFullEventRecord());
  fileWriter.writeFullHepMC3Info(_inputParameters->hepmc3FullEventRecord());
        _baseFileName = _inputParameters->baseFileName();
        _eventDataFileName = _baseFileName +".out";
  fileWriter.open(_eventDataFileName);
  //
  fileWriter.writeInit(*_inputParameters);
  //
  if(_inputParameters->lundFullEventRecord()){
    fileWriterLUND.open("slight_LUND.txt");
    //
    fileWriterLUND.writeInitLUND(*_inputParameters);
    //
  }
  printInfo << "generating events:" << endl;
  unsigned int nmbEvents = 0;
  std::chrono::steady_clock::time_point begin= std::chrono::steady_clock::now();
  while (nmbEvents < _nmbEventsTot) {
    for (unsigned int iEvent = 0; (iEvent < _nmbEventsPerFile) && (nmbEvents < _nmbEventsTot);
         ++iEvent, ++nmbEvents) {   
      progressIndicator(iEvent, _nmbEventsTot, true, 4);
      eXEvent event = _starlight->produceEvent();
      // Boost event from back to lab reference frame
      boostEvent(event);
      reflectEvent(event);
      fileWriter.writeEvent(event, iEvent);
      if(_inputParameters->lundFullEventRecord()){
        fileWriterLUND.writeEventLUND(event, iEvent);
      }
    }
  }
  std::chrono::steady_clock::time_point end= std::chrono::steady_clock::now();                                   
  float running_total = 1E-3*std::chrono::duration_cast<std::chrono::milliseconds>(end - begin).count();
  cout<<"Total time "<<running_total<<" s ("<<1E3*running_total/nmbEvents<<" ms/ev)"<<endl;
  fileWriter.close();
  if(_inputParameters->lundFullEventRecord()){
    fileWriterLUND.close();
  }

  if( _starlight->nmbAttempts() == 0 )return true; 

  double _branchingRatio = _inputParameters->inputBranchingRatio();
  printInfo << "number of attempts = " << _starlight->nmbAttempts() << ", "
            << "number of accepted events = " << _starlight->nmbAccepted() << endl;
        double selectedCrossSection =
    ((double)_starlight->nmbAccepted()/_starlight->nmbAttempts())*_branchingRatio*_starlight->getTotalCrossSection(); 
  if (selectedCrossSection > 1.){
    cout<< " The cross section of the generated sample is "<<selectedCrossSection<<" barn."<<endl;
  } else if (1.E3*selectedCrossSection > 1.){
    cout<< " The cross section of the generated sample is "<<1.E3*selectedCrossSection<<" mb."<<endl;
        } else if (1.E6*selectedCrossSection > 1.){
    cout<< " The cross section of the generated sample is "<<1.E6*selectedCrossSection<<" microbarn."<<endl;
        } else if (1.E9*selectedCrossSection > 1.){
    cout<< " The cross section of the generated sample is "<<1.E9*selectedCrossSection<<" nanobarn."<<endl;
        } else if (1.E12*selectedCrossSection > 1.){
    cout<< " The cross section of the generated sample is "<<1.E12*selectedCrossSection<<" picobarn."<<endl;
        } else {
    cout<< " The cross section of the generated sample is " <<1.E15*selectedCrossSection<<" femtob."<<endl;
        }  

  return true;
}
void e_starlightStandalone::boostEvent(eXEvent &event)
{
  
   // Calculate CMS boost 
   double rap2 = -acosh(_inputParameters->targetBeamLorentzGamma()); 
   double boost = _inputParameters->rap_CM();
   event.boost(boost, rap2); //  Boost back to laboratory reference frame. Electron initially in target frame and V.M. in CMS
                             //  Assuming electron is electronBeam


}


void e_starlightStandalone::reflectEvent(eXEvent &event)
{
   event.reflect(); //Change all z to -z
}