SGCosmic.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Based on the Belle single track generator by KUNIYA Toshio             *
 * Contributors: Sergey Yashchenko                                        *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <mdst/dataobjects/MCParticleGraph.h>
 
namespace Belle2 {
  class SGCosmic {
 
  public:
 
    struct Parameters {
      int level;
      int ipRequirement;
      double ipdr;
      double ipdz;
      double ptmin;
      double cylindricalR;
    };
 
    SGCosmic();
 
    ~SGCosmic() {}
 
    bool generateEvent(MCParticleGraph& graph);
 
    bool setParameters(const Parameters& parameters);
 
  protected:
 
    Parameters m_params;
 
  private:
 
    double findMax(const double* dim, const int num);
 
    void genCosmic(const int level, int& charge,
                   double& dr, double& phi, double& Pt, double& dz, double& tanl);
 
    int muChargeFlag(const double);
 
    void mkdist_v1(const int charge, double*);
    int mkDr_pos_v1(const double dr, const float rndm);
    int mkDr_neg_v1(const double dr, const float rndm);
    int mkPhi_pos_v1(const double phi, const float rndm);
    int mkPhi_neg_v1(const double phi, const float rndm);
    int mkPt_pos_v1(const double pt, const float rndm);
    int mkPt_neg_v1(const double pt, const float rndm);
    int mkDz_pos_v1(const double dz, const float rndm);
    int mkDz_neg_v1(const double dz, const float rndm);
    int mkTanl_pos_v1(const double tanl, const float rndm);
    int mkTanl_neg_v1(const double tanl, const float rndm);
 
    void mkdist_v2(const int charge, double*);
    int mkDr_pos_v2(const double dr, const float rndm);
    int mkDr_neg_v2(const double dr, const float rndm);
    int mkPhi_pos_v2(const double phi, const float rndm);
    int mkPhi_neg_v2(const double phi, const float rndm);
    int mkPt_pos_v2(const double pt, const float rndm);
    int mkPt_neg_v2(const double pt, const float rndm);
    int mkDz_pos_v2(const double dz, const float rndm);
    int mkDz_neg_v2(const double dz, const float rndm);
    int mkTanl_pos_v2(const double tanl, const float rndm);
    int mkTanl_neg_v2(const double tanl, const float rndm);
 
  };
 
} //end namespace Belle2