Changeset 571 in svn for trunk/Utilities/HepMC/src/GenEvent.cc

Timestamp:

Nov 2, 2011, 5:06:22 PM (13 years ago)

Author:

cp3-support

Message:

upgrade HepMC to version 2.06.05

File:

• trunk/Utilities/HepMC/src/GenEvent.cc (modified) (17 diffs)

trunk/Utilities/HepMC/src/GenEvent.cc

@@ -14 +14 @@
 
 #include "GenEvent.h"
+#include "GenCrossSection.h"
 #include "Version.h"
+#include "StreamHelpers.h"
 
 namespace HepMC {
@@ -38 +40 @@
         m_vertex_barcodes(),
         m_particle_barcodes(),
+        m_cross_section(0), 
         m_heavy_ion(0), 
         m_pdf_info(0),
@@ -71 +74 @@
         m_vertex_barcodes(),
         m_particle_barcodes(),
+        m_cross_section(0), 
         m_heavy_ion( new HeavyIon(ion) ), 
         m_pdf_info( new PdfInfo(pdf) ),
@@ -102 +106 @@
         m_vertex_barcodes(),
         m_particle_barcodes(),
+        m_cross_section(0), 
         m_heavy_ion(0), 
         m_pdf_info(0),
@@ -136 +141 @@
         m_vertex_barcodes(),
         m_particle_barcodes(),
+        m_cross_section(0), 
         m_heavy_ion( new HeavyIon(ion) ), 
         m_pdf_info( new PdfInfo(pdf) ),
@@ -162 +168 @@
         m_vertex_barcodes      ( /* inevent.m_vertex_barcodes */ ),
         m_particle_barcodes    ( /* inevent.m_particle_barcodes */ ),
+        m_cross_section        ( inevent.cross_section() ? new GenCrossSection(*inevent.cross_section()) : 0 ),
         m_heavy_ion            ( inevent.heavy_ion() ? new HeavyIon(*inevent.heavy_ion()) : 0 ),
         m_pdf_info             ( inevent.pdf_info() ? new PdfInfo(*inevent.pdf_info()) : 0 ),
@@ -168 +175 @@
     {
         /// deep copy - makes a copy of all vertices!
-        //++s_counter;
         //
 
@@ -234 +240 @@
         m_vertex_barcodes.swap(   other.m_vertex_barcodes );
         m_particle_barcodes.swap( other.m_particle_barcodes );
+        std::swap(m_cross_section        , other.m_cross_section        );
         std::swap(m_heavy_ion            , other.m_heavy_ion            );
         std::swap(m_pdf_info             , other.m_pdf_info             );
@@ -252 +259 @@
     {
         /// Deep destructor.
-        /// deletes all vertices/particles in this evt
-        ///
+        /// deletes all vertices/particles in this GenEvent
+        /// deletes the associated HeavyIon and PdfInfo
         delete_all_vertices();
+        delete m_cross_section;
         delete m_heavy_ion;
         delete m_pdf_info;
-        //--s_counter;
     }
 
@@ -281 +288 @@
                   : 0 )
              << "\n";
-        //ostr << " Current Memory Usage: " 
-        //     << GenEvent::counter() << " events, "
-        //     << GenVertex::counter() << " vertices, "
-        //     << GenParticle::counter() << " particles.\n"; 
         write_units( ostr );
+        write_cross_section(ostr);
         ostr << " Entries this event: " << vertices_size() << " vertices, "
              << particles_size() << " particles.\n"; 
@@ -302 +306 @@
         // Weights
         ostr << " Wgts(" << weights().size() << ")=";
-        for ( WeightContainer::const_iterator wgt = weights().begin();
-              wgt != weights().end(); ++wgt ) { ostr << *wgt << " "; }
-        ostr << "\n";
+        weights().print(ostr);
         ostr << " EventScale " << event_scale() 
              << " [energy] \t alphaQCD=" << alphaQCD() 
@@ -367 +369 @@
         ///
         delete_all_vertices();
+        // remove existing objects and set pointers to null
+        delete m_cross_section;
+        m_cross_section = 0;
         delete m_heavy_ion;
+        m_heavy_ion = 0;
         delete m_pdf_info;
+        m_pdf_info = 0;
         m_signal_process_id = 0;
         m_beam_particle_1 = 0;
         m_beam_particle_2 = 0;
         m_event_number = 0;
+        m_mpi = -1;
         m_event_scale = -1;
         m_alphaQCD = -1;
@@ -378 +386 @@
         m_weights = std::vector<double>();
         m_random_states = std::vector<long>();
-        //m_momentum_unit = ;
-        //m_position_unit = ;
+        // resetting unit information
+        m_momentum_unit = Units::default_momentum_unit();
+        m_position_unit = Units::default_length_unit();
         // error check just to be safe
         if ( m_vertex_barcodes.size() != 0 
@@ -388 +397 @@
                       << m_vertex_barcodes.size() << "  " 
                       << m_particle_barcodes.size() << std::endl;
-            //std::cerr << "Total Number vtx,particle in memory "
-             //         << "after method called = "
-             //         << GenVertex::counter() << "\t"
-                //      << GenParticle::counter() << std::endl;
         }
         return;
@@ -420 +425 @@
                       << m_vertex_barcodes.size() << "  " 
                       << m_particle_barcodes.size() << std::endl;
-            //std::cerr << "Total Number vtx,particle in memory "
-            //          << "after method called = "
-            //          << GenVertex::counter() << "\t"
-                //      << GenParticle::counter() << std::endl;
         }
     }
@@ -602 +603 @@
     }
 
+    void GenEvent::write_cross_section( std::ostream& os ) const
+    {
+        // write the GenCrossSection information if the cross section was set
+        if( !cross_section() ) return;
+        if( cross_section()->is_set() ) {
+            os << " Cross Section: " << cross_section()->cross_section() ;
+            os << " +/- " << cross_section()->cross_section_error() ;
+            os << std::endl;
+        }
+    }
+
    bool GenEvent::use_momentum_unit( Units::MomentumUnit newunit ) { 
         // currently not exception-safe. 
@@ -651 +663 @@
     }  
 
+    bool GenEvent::is_valid() const {
+        /// A GenEvent is presumed valid if it has both associated
+        /// particles and vertices.   No other information is checked.
+        if ( vertices_empty() ) return false;
+        if ( particles_empty() ) return false;
+        return true;
+    }
+
+    std::ostream & GenEvent::write_beam_particles(std::ostream & os, 
+                         std::pair<HepMC::GenParticle *,HepMC::GenParticle *> pr )
+    {
+        GenParticle* p = pr.first;
+        if(!p) {
+           detail::output( os, 0 );
+        } else {
+           detail::output( os, p->barcode() );
+        }
+        p = pr.second;
+        if(!p) {
+           detail::output( os, 0 );
+        } else {
+           detail::output( os, p->barcode() );
+        }
+
+        return os;
+    }
+
+    std::ostream & GenEvent::write_vertex(std::ostream & os, GenVertex const * v)
+    {
+        if ( !v || !os ) {
+            std::cerr << "GenEvent::write_vertex !v||!os, "
+                      << "v="<< v << " setting badbit" << std::endl;
+            os.clear(std::ios::badbit); 
+            return os;
+        }
+        // First collect info we need
+        // count the number of orphan particles going into v
+        int num_orphans_in = 0;
+        for ( GenVertex::particles_in_const_iterator p1
+                  = v->particles_in_const_begin();
+              p1 != v->particles_in_const_end(); ++p1 ) {
+            if ( !(*p1)->production_vertex() ) ++num_orphans_in;
+        }
+        //
+        os << 'V';
+        detail::output( os, v->barcode() ); // v's unique identifier
+        detail::output( os, v->id() );
+        detail::output( os, v->position().x() );
+        detail::output( os, v->position().y() );
+        detail::output( os, v->position().z() );
+        detail::output( os, v->position().t() );
+        detail::output( os, num_orphans_in );
+        detail::output( os, (int)v->particles_out_size() );
+        detail::output( os, (int)v->weights().size() );
+        for ( WeightContainer::const_iterator w = v->weights().begin(); 
+              w != v->weights().end(); ++w ) {
+            detail::output( os, *w );
+        }
+        detail::output( os,'\n');
+        // incoming particles
+        for ( GenVertex::particles_in_const_iterator p2 
+                  = v->particles_in_const_begin();
+              p2 != v->particles_in_const_end(); ++p2 ) {
+            if ( !(*p2)->production_vertex() ) {
+                write_particle( os, *p2 );
+            }
+        }
+        // outgoing particles
+        for ( GenVertex::particles_out_const_iterator p3 
+                  = v->particles_out_const_begin();
+              p3 != v->particles_out_const_end(); ++p3 ) {
+            write_particle( os, *p3 );
+        }
+        return os;
+    }
+
+    std::ostream & GenEvent::write_particle( std::ostream & os, GenParticle const * p )
+    {
+        if ( !p || !os ) {
+            std::cerr << "GenEvent::write_particle !p||!os, "
+                      << "p="<< p << " setting badbit" << std::endl;
+            os.clear(std::ios::badbit); 
+            return os;
+        }
+        os << 'P';
+        detail::output( os, p->barcode() );
+        detail::output( os, p->pdg_id() );
+        detail::output( os, p->momentum().px() );
+        detail::output( os, p->momentum().py() );
+        detail::output( os, p->momentum().pz() );
+        detail::output( os, p->momentum().e() );
+        detail::output( os, p->generated_mass() );
+        detail::output( os, p->status() );
+        detail::output( os, p->polarization().theta() );
+        detail::output( os, p->polarization().phi() );
+        // since end_vertex is oftentimes null, this CREATES a null vertex
+        // in the map
+        detail::output( os,   ( p->end_vertex() ? p->end_vertex()->barcode() : 0 )  );
+        os << ' ' << p->flow() << "\n";
+
+        return os;
+    }
+
 } // HepMC