InitialParticleGeneration.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2015 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Martin Ritter                                            *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
#ifndef GENERATORS_UTILITIES_INITIALPARTICLEGENERATION_H
#define GENERATORS_UTILITIES_INITIALPARTICLEGENERATION_H
 
#include <framework/datastore/StoreObjPtr.h>
#include <framework/database/DBObjPtr.h>
#include <framework/utilities/MultivariateNormalGenerator.h>
#include <framework/dbobjects/BeamParameters.h>
#include <framework/dataobjects/MCInitialParticles.h>
 
namespace Belle2 {
  class InitialParticleGeneration {
  public:
    explicit InitialParticleGeneration(int allowedFlags = 0): m_allowedFlags(allowedFlags | MCInitialParticles::c_generateCMS) {};
 
    MCInitialParticles& generate();
 
    TVector3 updateVertex(bool force = false);
 
    const BeamParameters& getBeamParameters() const { return *m_beamParams; }
 
    void initialize();
 
    void setAllowedFlags(int allowedFlags)
    {
      m_allowedFlags = allowedFlags | MCInitialParticles::c_generateCMS;
    }
 
  private:
 
    MCInitialParticles& generate(int allowedFlags);
 
    TVector3 generateVertex(const TVector3& initial, const TMatrixDSym& cov, MultivariateNormalGenerator& gen);
    TLorentzVector generateBeam(const TLorentzVector& initial, const TMatrixDSym& cov, MultivariateNormalGenerator& gen);
    DBObjPtr<BeamParameters> m_beamParams;
    StoreObjPtr<MCInitialParticles> m_event;
    MultivariateNormalGenerator m_generateHER;
    MultivariateNormalGenerator m_generateLER;
    MultivariateNormalGenerator m_generateVertex;
    int m_allowedFlags;
  };
 
 
} //Belle2 namespace
#endif // GENERATORS_UTILITIES_INITIALPARTICLEGENERATION_H