#ifndef SOURCE
#define SOURCE

#include "G4VUserPrimaryGeneratorAction.hh"
#include "G4ParticleGun.hh"

class G4Event;

class GeneratorAction : public G4VUserPrimaryGeneratorAction {
public:
    GeneratorAction();
    ~GeneratorAction();
    virtual void GeneratePrimaries(G4Event* e);

private:
    G4ParticleGun* particleGun;
};

#endif