A2FileGenerator: added event weight member

Author: werthm

include/A2FileGenerator.hh

@@ -39,6 +39,7 @@ protected:
     EFileGenType fType;                     // type of file generator
     G4String fFileName;                     // input file name
     G4int fNEvents;                         // number of events
+    G4double fWeight;                       // event weight
     A2GenParticle_t fBeam;                  // beam particle
     G4ThreeVector fVertex;                  // primary vertex [mm]
     std::vector<A2GenParticle_t> fPart;     // list of particles
@@ -54,6 +55,7 @@ public:
     EFileGenType GetType() const { return fType; }
     const G4String& GetFileName() const { return fFileName; }
     G4int GetNEvents() const { return fNEvents; }
+    G4double GetWeight() const { return fWeight; }
     const G4ThreeVector& GetVertex() const { return fVertex; }
     const A2GenParticle_t& GetBeam() const { return fBeam; }
 
@@ -69,6 +71,7 @@ public:
     G4bool IsParticleTrack(G4int p) const { return fPart[p].fIsTrack; }
 
     void SetParticleIsTrack(G4int p, G4bool t = true);
+    void SetWeight(G4double w) { fWeight = w; }
 
     void GenerateVertexCylinder(G4double t_length, G4double t_center,
                                 G4double b_diam);

src/A2FileGenerator.cc

@@ -17,6 +17,7 @@ A2FileGenerator::A2FileGenerator(const char* filename, EFileGenType type)
     fType = type;
     fFileName = filename;
     fNEvents = 0;
+    fWeight = 1;
 }
 
 //______________________________________________________________________________
@@ -95,6 +96,7 @@ void A2FileGenerator::Print() const
     G4cout << "Generator type      : " << type << G4endl
            << "File name           : " << fFileName << G4endl
            << "Number of events    : " << fNEvents << G4endl
+           << "Event weight        : " << fWeight << G4endl
            << "Number of particles : " << fPart.size() << G4endl
            << "Primary vertex      : " << fVertex << G4endl
            << "Beam particle" << G4endl;