source: trunk/source/geometry/solids/test/fred/include/SprayParticleGun.hh @ 1316

//
// ********************************************************************
// * License and Disclaimer                                           *
// *                                                                  *
// * The  Geant4 software  is  copyright of the Copyright Holders  of *
// * the Geant4 Collaboration.  It is provided  under  the terms  and *
// * conditions of the Geant4 Software License,  included in the file *
// * LICENSE and available at  http://cern.ch/geant4/license .  These *
// * include a list of copyright holders.                             *
// *                                                                  *
// * Neither the authors of this software system, nor their employing *
// * institutes,nor the agencies providing financial support for this *
// * work  make  any representation or  warranty, express or implied, *
// * regarding  this  software system or assume any liability for its *
// * use.  Please see the license in the file  LICENSE  and URL above *
// * for the full disclaimer and the limitation of liability.         *
// *                                                                  *
// * This  code  implementation is the result of  the  scientific and *
// * technical work of the GEANT4 collaboration.                      *
// * By using,  copying,  modifying or  distributing the software (or *
// * any work based  on the software)  you  agree  to acknowledge its *
// * use  in  resulting  scientific  publications,  and indicate your *
// * acceptance of all terms of the Geant4 Software license.          *
// ********************************************************************
//
//
// SprayParticleGun.hh
//
// Declaration of test particle gun
//
// G4ParticleGun unfortunately is not written with enough modularity to
// inherit. So, this stuff here is written from scratch, but while refering
// to G4ParticleGun's code.
//

#ifndef SprayParticleGun_HH
#define SprayParticleGun_HH

#include "globals.hh"
#include "G4VPrimaryGenerator.hh"
#include "G4ThreeVector.hh"
#include "G4ParticleDefinition.hh"
#include "G4PrimaryVertex.hh"
#include "G4ParticleMomentum.hh"

class G4Event;
class SprayParticleGunMessenger;

class SprayParticleGun : public G4VPrimaryGenerator
{
        public:
        SprayParticleGun();
        SprayParticleGun( G4ThreeVector start );
        ~SprayParticleGun();
        
        void GeneratePrimaryVertex( G4Event *evt );
        
        inline void SetPosition( G4ThreeVector position ) { start = position; }
        inline G4ThreeVector GetPosition() { return start; }
        
        inline void SetXSpray( G4int xSpray ) {spray[0] = xSpray; }
        inline G4int GetXSpray() { return spray[0]; }
        
        inline void SetYSpray( G4int ySpray ) {spray[1] = ySpray; }
        inline G4int GetYSpray() { return spray[1]; }
        
        inline void SetZSpray( G4int zSpray ) {spray[2] = zSpray; }
        inline G4int GetZSpray() { return spray[2]; }
        
        
        private:
        void SetDefaults();

        protected:
        G4int spray[3];
        G4ThreeVector start;
        G4ParticleDefinition  *particleType;

        private:
        SprayParticleGunMessenger       *messenger;
};

#endif