GeneratorTParticle.h

Go to the documentation of this file.

// Copyright 2023-2099 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
#ifndef ALICEO2_GENERATORTPARTICLE_H_
#define ALICEO2_GENERATORTPARTICLE_H_
#include <FairGenerator.h>
#include <Generators/Generator.h>
#include <Generators/GeneratorFileOrCmd.h>
#include <Generators/GeneratorTParticleParam.h>
 
// Forward decls
class TChain;
class TParticle;
class TClonesArray;
 
namespace o2
{
namespace conf
{
class SimConfig;
}
namespace eventgen
{
class GeneratorTParticle : public Generator, public GeneratorFileOrCmd
{
 public:
  GeneratorTParticle();
  GeneratorTParticle(const std::string& name)
    : Generator(name.c_str(), "ALICEo2 TParticle Generator")
  {
  }
  virtual ~GeneratorTParticle();
 
  Bool_t Init() override;
 
  void setup(const GeneratorFileOrCmdParam& param0,
             const GeneratorTParticleParam& param,
             const conf::SimConfig& config);
  Bool_t generateEvent() override;
 
  Bool_t importParticles() override;
 
  void setTreeName(const std::string& val) { mTreeName = val; }
  void setBranchName(const std::string& val) { mBranchName = val; }
 
 protected:
  std::string mTreeName = "T";
  std::string mBranchName = "Particles";
  unsigned int mEntry = 0;
  TChain* mChain = 0;
  TClonesArray* mTParticles;
 
  ClassDefOverride(GeneratorTParticle, 1);
};
} // namespace eventgen
} // namespace o2
#endif
// Local Variables:
//  mode: C++
// End:
//
// EOF
//