MarlinUtil.h

#ifndef MarlinUtil_h
#define MarlinUtil_h 1

#include <iostream>
#include <fstream>

#include <cmath>
#include <string>
#include <vector>
#include <utility>
#include <algorithm>

#include <lcio.h>
#include <EVENT/LCEvent.h>
#include <EVENT/LCCollection.h>
#include <EVENT/MCParticle.h>
#include <EVENT/SimTrackerHit.h>
#include <EVENT/SimCalorimeterHit.h>
#include <EVENT/ReconstructedParticle.h>

#include "HelixClass.h"

#include "csvparser.h"


using namespace lcio;


class MarlinUtil {


 public:

  static void printMCParticle(MCParticle* MCP, bool printDaughters = false);
  static std::string getMCName(int PDGCode);
  static int getPDGCode(std::string name);
  static MCParticleVec getAllMCParents(MCParticle* mcPart );
  static MCParticleVec getAllMCDaughters(MCParticle* mcPart);
  static bool isDaughterOf( MCParticle* daughter, MCParticle* parent );
  static bool DecayChainInTree(std::vector<int> DecayChannel, LCEvent* evt);

  static void getMC_Balance(LCEvent* evt, double* accumulatedEnergies);

  static void printTrack(Track* track, double bField=4.0);
  static const double* getMomentum(Track* track, double bField=4.0);
  static const double getAbsMomentum(Track* track, double bField=4.0);
  static void printCluster(Cluster* cluster);
  static void printRecoParticle(ReconstructedParticle* recoParticle, double bField=4.0);
  static int countAllSimTrackerHits(LCEvent* evt,MCParticle* MCP);
  static int countAllSimCalorimeterHits(LCEvent* evt,MCParticle* MCP,double& accumulatedSimCaloEnergy);
  static double getEnergyDepositedInFullCalorimeter(LCEvent* evt);


};




// FIXME: the whole MC particle related stuff should be placed somewhere else (see MarlinUtil as well)


class MCParticleHelper {

 public: 

  MCParticleHelper();
  std::string getMCCharge(int PDGCode);


 private: 

  std::vector<int> _pdgCodesMCParticles;
  std::vector<double> _massMCParticles;
  std::vector<std::string> _nameMCParticles;
  std::vector<std::string> _chargeMCParticles;


};

#endif