Changeset 1315 for trunk/source/processes/hadronic/models/cascade/cascade/include/G4CascadeMomentum.hh

Timestamp:

Jun 18, 2010, 11:42:07 AM (14 years ago)

Author:

garnier

Message:

update geant4-09-04-beta-cand-01 interfaces-V09-03-09 vis-V09-03-08

File:

• trunk/source/processes/hadronic/models/cascade/cascade/include/G4CascadeMomentum.hh (modified) (3 diffs)

trunk/source/processes/hadronic/models/cascade/cascade/include/G4CascadeMomentum.hh

@@ -24 +24 @@
 // ********************************************************************
 //
-// $Id: G4CascadeMomentum.hh,v 1.1 2008/09/22 10:06:32 gcosmo Exp $
-// GEANT4 tag $Name: geant4-09-03 $
-//
+// $Id: G4CascadeMomentum.hh,v 1.5 2010/03/16 22:10:26 mkelsey Exp $
+// GEANT4 tag $Name: geant4-09-04-beta-cand-01 $
 //
 // Class G4CascadeMomentum
@@ -35 +34 @@
 // std::vector<double> in the cascade mode code, which causes
 // problems for performance due to excess memory allocations.
-
+//
+// NOTE:  The Bertini code does not pass legitimate four-vectors when
+//        creating new particles; the new getLV() function takes an
+//        optional mass argument (in Bertini units [GeV]) so that a
+//        valid G4LorentzVector can be returned.
+//
 // Author: Peter Elmer, Princeton University                  7-Aug-2008
+// Update: Michael Kelsey, SLAC (support G4LorentzVector)     8-Jan-2010
 // --------------------------------------------------------------------
 #ifndef G4CASCADE_MOMENTUM_HH
@@ -44 +49 @@
 
 #include "G4Types.hh"
+#include "G4LorentzVector.hh"
+#include "G4ThreeVector.hh"
 
 class G4CascadeMomentum
 {
   public:
-
     G4CascadeMomentum() {for (int i=0; i<4; ++i) data_[i]=0.0;}
 
-    G4double& operator[](int i)
-    {
+    // WARNING!  This metric is (t,x,y,z), DIFFERENT FROM HepLV!
+    G4CascadeMomentum(const G4LorentzVector& lv) {
+      setLV(lv);
+    }
+
+    G4double& operator[](int i) {
       assert(i>=0 && i<4);
       return data_[i];
     }
-    const G4double& operator[](int i) const
-    {
+    const G4double& operator[](int i) const {
       assert(i>=0 && i<4);
       return data_[i];
     }
 
+    operator const G4LorentzVector&() const {
+      return getLV();                   // Casting can't do mass repairs
+    }
+
+    const G4LorentzVector& getLV(G4double mass=-1.) const {
+      if (mass>=0.) lv.setVectM(get3V(),mass);          // Force input mass!
+      else lv.set(data_[1],data_[2],data_[3],data_[0]);
+      
+      return lv;
+    }
+
+    G4ThreeVector get3V() const {
+      return getLV().vect();
+    }
+
+    void setLV(const G4LorentzVector& lv) {
+      data_[0] = lv.t();                // NOTE DIFFERENT METRIC CONVENTION!
+      data_[1] = lv.x();
+      data_[2] = lv.y();
+      data_[3] = lv.z();
+    }
+
+    G4CascadeMomentum& operator=(const G4LorentzVector& lv) {
+      setLV(lv);
+      return *this;
+    }
+
   private:
+    G4double data_[4];
+    mutable G4LorentzVector lv;         // Buffer for conversion operations
+};
 
-    G4double data_[4];
-};
 #endif