KKGenInterface.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
/* Basf2 headers. */
#include <mdst/dataobjects/MCParticleGraph.h>
#include <framework/utilities/ConditionalGaussGenerator.h>
 
/* C++ headers. */
#include <string>
#include <unordered_map>
 
#define nmxhep 4000
 
struct hepevt_type {
  int nevhep;               
  int nhep;                 
  int isthep[nmxhep];       
  int idhep[nmxhep];        
  int jmohep[nmxhep][2];    
  int jdahep[nmxhep][2];    
  double phep[nmxhep][5];   
  double vhep[nmxhep][4];   
};
 
struct pydat2_type {
  int KCHG[4][500];  
  double PMAS[4][500]; 
  double PARF[2000]; 
  double VCKM[4][4]; 
};
 
extern hepevt_type hepevt_;
extern pydat2_type pydat2_;
 
extern "C" {
  void kk_init_(const char*, const char*, const char*, int*, const char*);
  void kk_begin_run_(double*, double*);
  void kk_init_seed_();
  void kk_term_(double*, double*);
  void kk_event_(int*);
  void kk_shifttaudecayvtx_();
  int pycomp_(int&);
 
}
 
namespace Belle2 {
  class KKGenInterface {
 
  public:
 
    KKGenInterface() {}
 
    ~KKGenInterface() {}
 
    KKGenInterface(const KKGenInterface& m) = delete;
 
    KKGenInterface& operator= (const KKGenInterface& m) = delete;
 
    int setup(const std::string& KKdefaultFileName, const std::string& tauinputFileName, const std::string& taudecaytableFileName,
              const std::string& KKMCOutputFileName);
 
    void set_beam_info(double Ecms0, double Ecms0Spread);
 
    int simulateEvent(MCParticleGraph& graph, const ConditionalGaussGenerator& lorentzGenerator, ROOT::Math::XYZVector vertex);
 
    void term();
 
  private:
 
    int addParticles2Graph(MCParticleGraph& graph, ROOT::Math::XYZVector vertex);
 
    void updateGraphParticle(int, MCParticleGraph::GraphParticle* gParticle,
                             ROOT::Math::XYZVector vertex);
 
    std::unordered_map<int, int> m_mapPythiaIDtoPDG;
 
  };
 
}