FinalStatePartons.cc

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

/*******************************************************************************
 * Copyright (c) The JETSCAPE Collaboration, 2018
 *
 * Modular, task-based framework for simulating all aspects of heavy-ion collisions
 * 
 * For the list of contributors see AUTHORS.
 *
 * Report issues at https://github.com/JETSCAPE/JETSCAPE/issues
 *
 * or via email to bugs.jetscape@gmail.com
 *
 * Distributed under the GNU General Public License 3.0 (GPLv3 or later).
 * See COPYING for details.
 ******************************************************************************/

#include <iostream>
#include <fstream>
#include <memory>
#include <chrono>
#include <thread>

#include "gzstream.h"
#include "PartonShower.h"
#include "JetScapeLogger.h"
#include "JetScapeReader.h"
#include "JetScapeBanner.h"
#include "fjcore.hh"

#include <GTL/dfs.h>

using namespace std;
using namespace fjcore;

using namespace Jetscape;

// You could overload here and then simply use ofstream << p;
// ostream & operator<<(ostream & ostr, const fjcore::PseudoJet & jet);


// -------------------------------------

int main(int argc, char** argv)
{

  // JetScapeLogger::Instance()->SetInfo(false);
  JetScapeLogger::Instance()->SetDebug(false);
  JetScapeLogger::Instance()->SetRemark(false);
  // //SetVerboseLevel (9 a lot of additional debug output ...)
  // //If you want to suppress it: use SetVerboseLevle(0) or max  SetVerboseLevel(9) or 10
  JetScapeLogger::Instance()->SetVerboseLevel(0);
  
  auto reader=make_shared<JetScapeReaderAscii>("test_out.dat");
  std::ofstream dist_output ("JetscapeFinalStatePartons.txt"); //Format is SN, PID, E, Px, Py, Pz, Eta, Phi
  
  while (!reader->Finished())
    {
      reader->Next();
      
      // cout<<"Analyze current event: "<<reader->GetCurrentEvent()<<endl;
      auto mShowers=reader->GetPartonShowers();     

      dist_output<<"Event "<< reader->GetCurrentEvent()+1<<endl;
      for (int i=0;i<mShowers.size();i++)
  {
    //  cout<<" Analyze parton shower: "<<i<<endl;
    // Let's create a file
    for ( int ipart = 0; ipart< mShowers[i]->GetFinalPartons().size(); ++ipart){
      Parton p = *mShowers[i]->GetFinalPartons().at(ipart);
      dist_output << ipart   << "\t"
      << p.pid() << "\t"
      << p.pstat() << "\t"
      << p.e()   << "\t"
      << p.px()  << "\t" 
      << p.py()  << "\t" 
      << p.pz()  << "\t"
      << p.eta()<<  "\t"<< p.phi() << endl;
    }
  }

    }
  
  reader->Close(); 
}