hepmc - Diff between revs 390 and 432

Subversion Repositories:

(root)/trunk/src/GenEventStreamIO.cc

Rev 390 - Rev 440 - Go to most recent revision - Details - Blame - Last modification - View Log -

     for ( std::vector<long>::iterator rs = m_random_states.begin();
           rs != m_random_states.end(); ++rs ) {
          detail::output( os, *rs );
+    }
     // weights
+    // we need to iterate over the map so that the weights printed
+    // here will be in the same order as the names printed next
     os << ' ' << (int)weights().size() ;
-    for ( WeightContainer::const_iterator w = weights().begin();
-          w != weights().end(); ++w ) {
-         detail::output( os, *w );
+    for ( WeightContainer::const_map_iterator w = weights().map_begin();
+          w != weights().map_end(); ++w ) {
+        detail::output( os, m_weights[w->second] );
+    }
     detail::output( os,'\n');
+    // now add names for weights
+    // note that this prints a new line if and only if the weight container
+    // is not empty
+    if ( ! weights().empty() ) {
+        os << "N " << weights().size() << " " ;
+        for ( WeightContainer::const_map_iterator w = weights().map_begin();
+              w != weights().map_end(); ++w ) {
+            detail::output( os,'"');
+            os << w->first;
+            detail::output( os,'"');
+            detail::output( os,' ');
+        }
+        detail::output( os,'\n');
+    }
     //
     // Units
     os << "U " << name(momentum_unit());
     os << " " << name(length_unit());
     detail::output( os,'\n');
     //
     // write GenCrossSection if it has been set
     if( m_cross_section ) m_cross_section->write(os);
     //
-    os << heavy_ion() ;
-    os << pdf_info() ;
+    // write HeavyIon and PdfInfo if they have been set
+    if( m_heavy_ion ) os << heavy_ion() ;
+    if( m_pdf_info ) os << pdf_info() ;
     //
     // Output all of the vertices - note there is no real order.
     for ( GenEvent::vertex_const_iterator v = vertices_begin();
           v != vertices_end(); ++v ) {
         write_vertex(os, *v);
                       << "setting badbit." << std::endl;
             is.clear(std::ios::badbit);
             return is;
+        }
+    }
-    // get the event line
-    std::string line;
-    std::getline(is,line);
-    std::istringstream iline(line);
-    std::string firstc;
-    iline >> firstc;
-    //
-    // read values into temp variables, then fill GenEvent
-    int event_number = 0, signal_process_id = 0, signal_process_vertex = 0,
-        num_vertices = 0, random_states_size = 0, weights_size = 0,
-        nmpi = -1, bp1 = 0, bp2 = 0;
-    double eventScale = 0, alpha_qcd = 0, alpha_qed = 0;
-    iline >> event_number;
-    if(!iline) detail::find_event_end( is );
-    if( info.io_type() == gen || info.io_type() == extascii ) {
-        iline >> nmpi;
-        if(!iline) detail::find_event_end( is );
-        set_mpi( nmpi );
-    }
-    iline >> eventScale ;
-    if(!iline) detail::find_event_end( is );
-    iline >> alpha_qcd ;
-    if(!iline) detail::find_event_end( is );
-    iline >> alpha_qed;
-    if(!iline) detail::find_event_end( is );
-    iline >> signal_process_id ;
-    if(!iline) detail::find_event_end( is );
-    iline >> signal_process_vertex;
-    if(!iline) detail::find_event_end( is );
-    iline >> num_vertices;
-    if(!iline) detail::find_event_end( is );
-    if( info.io_type() == gen || info.io_type() == extascii ) {
-        iline >> bp1 ;
-        if(!iline) detail::find_event_end( is );
-        iline >> bp2;
-        if(!iline) detail::find_event_end( is );
-    }
-    iline >> random_states_size;
-    if(!iline) detail::find_event_end( is );
-    std::vector<long> random_states(random_states_size);
-    for ( int i = 0; i < random_states_size; ++i ) {
-        iline >> random_states[i];
-        if(!iline) detail::find_event_end( is );
-    }
-    iline >> weights_size;
-    if(!iline) detail::find_event_end( is );
-    WeightContainer weights(weights_size);
-    for ( int ii = 0; ii < weights_size; ++ii ) {
-        iline >> weights[ii];
-        if(!iline) detail::find_event_end( is );
-    }
-    //
-    // fill signal_process_id, event_number, weights, random_states, etc.
-    set_signal_process_id( signal_process_id );
-    set_event_number( event_number );
-    m_weights = weights;
-    set_random_states( random_states );
-    set_event_scale( eventScale );
-    set_alphaQCD( alpha_qcd );
-    set_alphaQED( alpha_qed );
-    if( info.io_type() == gen ) {
-        // get unit information if it exists
-        read_units( is );
-    }
-    // check to see if we have a GenCrossSection line
-    if ( is.peek()=='C' ) {
-        // create cross section
-        GenCrossSection* xs = new GenCrossSection();
-        // check for invalid data
-        try {
-            // read the line
-            xs->read(is);
-        }
-        catch (IO_Exception& e) {
-            detail::find_event_end( is );
-        }
-        if(xs->is_set()) {
-            set_cross_section( *xs );
-        } else {
-            delete xs;
-            xs = 0;
-        }
-    }
-    if( info.io_type() == gen || info.io_type() == extascii ) {
-        // get HeavyIon and PdfInfo
-        HeavyIon* ion = new HeavyIon();
-        // check for invalid data
-        try {
-            is >> ion;
-        }
-        catch (IO_Exception& e) {
-            detail::find_event_end( is );
-        }
-        if(ion->is_valid()) {
-            set_heavy_ion( *ion );
-        } else {
-            delete ion;
-            ion = 0;
-        }
-        PdfInfo* pdf = new PdfInfo();
-        // check for invalid data
-        try {
-            is >> pdf;
-        }
-        catch (IO_Exception& e) {
-            detail::find_event_end( is );
-        }
-        if(pdf->is_valid()) {
-            set_pdf_info( *pdf );
-        } else {
-            delete pdf;
-            pdf = 0;
-        }
-    }
+    int signal_process_vertex = 0;
+    int num_vertices = 0, bp1 = 0, bp2 = 0;
+    // OK - now ready to start reading the event, so set the header flag
+    info.set_reading_event_header(true);
+    // The flag will be set to false when we reach the end of the header
+    while(info.reading_event_header()) {
+        switch(is.peek()) {
+            case 'E':
+            {   // deal with the event line
+                process_event_line( is, num_vertices, bp1, bp2, signal_process_vertex );
+            } break;
+            case 'N':
+            {   // get weight names
+                read_weight_names( is );
+            } break;
+            case 'U':
+            {   // get unit information if it exists
+                if( info.io_type() == gen ) {
+                    read_units( is );
+                }
+            } break;
+            case 'C':
+            {   // we have a GenCrossSection line
+                // create cross section
+                GenCrossSection* xs = new GenCrossSection();
+                // check for invalid data
+                try {
+                    // read the line
+                    xs->read(is);
+                }
+                catch (IO_Exception& e) {
+                    detail::find_event_end( is );
+                }
+                if(xs->is_set()) {
+                    set_cross_section( *xs );
+                } else {
+                    delete xs;
+                    xs = 0;
+                }
+            } break;
+            case 'H':
+            {   // we have a HeavyIon line
+                if( info.io_type() == gen || info.io_type() == extascii ) {
+                    // get HeavyIon
+                    HeavyIon* ion = new HeavyIon();
+                    // check for invalid data
+                    try {
+                        is >> ion;
+                    }
+                    catch (IO_Exception& e) {
+                        detail::find_event_end( is );
+                    }
+                    if(ion->is_valid()) {
+                        set_heavy_ion( *ion );
+                    } else {
+                        delete ion;
+                        ion = 0;
+                    }
+                }
+            } break;
+            case 'F':
+            {   // we have a PdfInfo line
+                if( info.io_type() == gen || info.io_type() == extascii ) {
+                    // get PdfInfo
+                    PdfInfo* pdf = new PdfInfo();
+                    // check for invalid data
+                    try {
+                        is >> pdf;
+                    }
+                    catch (IO_Exception& e) {
+                        detail::find_event_end( is );
+                    }
+                    if(pdf->is_valid()) {
+                        set_pdf_info( *pdf );
+                    } else {
+                        delete pdf;
+                        pdf = 0;
+                    }
+                }
+            } break;
+            case 'V':
+            {
+                // this should be the first vertex line - exit this loop
+                info.set_reading_event_header(false);
+            } break;
+            case 'P':
+            {   // we should not find this line
+                std::cerr << "streaming input: found unexpected line P" << std::endl;
+                info.set_reading_event_header(false);
+            } break;
+            default:
+                // ignore everything else
+                break;
+        } // switch on line type
+    } // while reading_event_header
     //
     // the end vertices of the particles are not connected until
     //  after the event is read --- we store the values in a map until then
     TempParticleMap particle_to_end_vertex;
     //
     if( info.finished_first_event() ) {
         os << info.IO_GenEvent_End() << "\n";
         os << std::flush;
+    }
     return os;
+}
+std::istream & GenEvent::process_event_line( std::istream & is,
+                                             int & num_vertices,
+                                             int & bp1, int & bp2,
+                                             int & signal_process_vertex )
+{
+    //
+    if ( !is ) {
+        std::cerr << "GenEvent::process_event_line setting badbit." << std::endl;
+        is.clear(std::ios::badbit);
+        return is;
+    }
+    //
+    StreamInfo & info = get_stream_info(is);
+    std::string line;
+    std::getline(is,line);
+    std::istringstream iline(line);
+    std::string firstc;
+    iline >> firstc;
+    //
+    // read values into temp variables, then fill GenEvent
+    int event_number = 0, signal_process_id = 0,
+        random_states_size = 0, nmpi = -1;
+    double eventScale = 0, alpha_qcd = 0, alpha_qed = 0;
+    iline >> event_number;
+    if(!iline) detail::find_event_end( is );
+    if( info.io_type() == gen || info.io_type() == extascii ) {
+        iline >> nmpi;
+        if(!iline) detail::find_event_end( is );
+        set_mpi( nmpi );
+    }
+    iline >> eventScale ;
+    if(!iline) detail::find_event_end( is );
+    iline >> alpha_qcd ;
+    if(!iline) detail::find_event_end( is );
+    iline >> alpha_qed;
+    if(!iline) detail::find_event_end( is );
+    iline >> signal_process_id ;
+    if(!iline) detail::find_event_end( is );
+    iline >> signal_process_vertex;
+    if(!iline) detail::find_event_end( is );
+    iline >> num_vertices;
+    if(!iline) detail::find_event_end( is );
+    if( info.io_type() == gen || info.io_type() == extascii ) {
+        iline >> bp1 ;
+        if(!iline) detail::find_event_end( is );
+        iline >> bp2;
+        if(!iline) detail::find_event_end( is );
+    }
+    iline >> random_states_size;
+    if(!iline) detail::find_event_end( is );
+    std::vector<long> random_states(random_states_size);
+    for ( int i = 0; i < random_states_size; ++i ) {
+        iline >> random_states[i];
+        if(!iline) detail::find_event_end( is );
+    }
+    WeightContainer::size_type weights_size = 0;
+    iline >> weights_size;
+    if(!iline) detail::find_event_end( is );
+    std::vector<double> wgt(weights_size);
+    for ( WeightContainer::size_type ii = 0; ii < weights_size; ++ii ) {
+        iline >> wgt[ii];
+        if(!iline) detail::find_event_end( is );
+    }
+    // weight names will be added later if they exist
+    if( weights_size > 0 ) m_weights = wgt;
+    //
+    // fill signal_process_id, event_number, random_states, etc.
+    set_signal_process_id( signal_process_id );
+    set_event_number( event_number );
+    set_random_states( random_states );
+    set_event_scale( eventScale );
+    set_alphaQCD( alpha_qcd );
+    set_alphaQED( alpha_qed );
+    //
+    return is;
+}
+std::istream & GenEvent::read_weight_names( std::istream & is )
+{
+    // now check for a named weight line
+    if ( !is ) {
+        std::cerr << "GenEvent::read_weight_names setting badbit." << std::endl;
+        is.clear(std::ios::badbit);
+        return is;
+    }
+    // Test to be sure the next entry is of type "N"
+    // If we have no named weight line, this is not an error
+    // releases prior to 2.06.00 do not have named weights
+    if ( is.peek() !='N') {
+        return is;
+    }
+    // now get this line and process it
+    std::string line;
+    std::getline(is,line);
+    std::istringstream wline(line);
+    std::string firstc;
+    WeightContainer::size_type name_size = 0;
+    wline >> firstc >> name_size;
+    if(!wline) detail::find_event_end( is );
+    if( firstc != "N") {
+        std::cout << "debug: first character of named weights is " << firstc << std::endl;
+        std::cout << "debug: We should never get here" << std::endl;
+        is.clear(std::ios::badbit);
+        return is;
+    }
+    if( m_weights.size() != name_size ) {
+        std::cout << "debug: weight sizes do not match "<< std::endl;
+        std::cout << "debug: weight vector size is " << m_weights.size() << std::endl;
+        std::cout << "debug: weight name size is " << name_size << std::endl;
+        is.clear(std::ios::badbit);
+        return is;
+    }
+    std::string name;
+    std::string::size_type i1 = line.find("\"");
+    std::string::size_type i2;
+    std::string::size_type len = line.size();
+    WeightContainer namedWeight;
+    for ( WeightContainer::size_type ii = 0; ii < name_size; ++ii ) {
+        // weight names may contain blanks
+        if(i1 >= len) {
+            std::cout << "debug: attempting to read past the end of the named weight line " << std::endl;
+            std::cout << "debug: We should never get here" << std::endl;
+            std::cout << "debug: Looking for the end of this event" << std::endl;
+            detail::find_event_end( is );
+        }
+        i2 = line.find("\"",i1+1);
+        name = line.substr(i1+1,i2-i1-1);
+        namedWeight[name] = m_weights[ii];
+        i1 = line.find("\"",i2+1);
+    }
+    m_weights = namedWeight;
+    return is;
+}
 std::istream & GenEvent::read_units( std::istream & is )
+{
     //
     if ( !is ) {
-        std::cerr << "StreamHelpers read_units setting badbit." << std::endl;
+        std::cerr << "GenEvent::read_units setting badbit." << std::endl;
         is.clear(std::ios::badbit);
         return is;
+    }
     //
     StreamInfo & info = get_stream_info(is);