diff --git a/sf-writer/sf_h5_writer.cpp b/sf-writer/sf_h5_writer.cpp index 0b34769..7a8c4ba 100644 --- a/sf-writer/sf_h5_writer.cpp +++ b/sf-writer/sf_h5_writer.cpp @@ -4,7 +4,8 @@ #include #include #include -#include "mpi.h" +#include +#include "BufferMultiReader.hpp" #include "config.hpp" #include "SfFormat.cpp" @@ -13,12 +14,13 @@ using namespace std; int main (int argc, char *argv[]) { - if (argc != 5) { + if (argc != 6) { cout << endl; - cout << "Usage: sf_h5_writer [device_name]"; + cout << "Usage: sf_h5_writer [device_name] [root_folder]"; cout << " [output_file] [start_pulse_id] [stop_pulse_id]"; cout << endl; cout << "\tdevice_name: Name of detector to write."; + cout << "\troot_folder: Base of the buffer."; cout << "\toutput_file: Complete path to the output file."; cout << "\tstart_pulse_id: Start pulse_id of retrieval." << endl; cout << "\tstop_pulse_id: Stop pulse_id of retrieval." << endl; @@ -28,25 +30,65 @@ int main (int argc, char *argv[]) } string device_name = string(argv[1]); - string output_file = string(argv[2]); - uint64_t start_pulse_id = (uint64_t) atoi(argv[3]); - uint64_t stop_pulse_id = (uint64_t) atoi(argv[4]); + string root_folder = string(argv[2]); + string output_file = string(argv[3]); + uint64_t start_pulse_id = (uint64_t) atoi(argv[4]); + uint64_t stop_pulse_id = (uint64_t) atoi(argv[5]); - if (MPI_Init ( &argc, &argv ) != MPI_SUCCESS) { - throw runtime_error("Cannot MPI init."); + BufferMultiReader reader(device_name, root_folder); + UdpFrameMetadata metadata; + uint16_t* frame_buffer = new uint16_t[32*512*1024]; + + H5Writer writer(output_file); + + for (size_t pulse_id=start_pulse_id; + pulse_id <= stop_pulse_id; + pulse_id++) { + + metadata = reader.get(pulse_id, (void*) frame_buffer); + + writer.write_data("pulse_id", pulse_id, + (char*)&(metadata.pulse_id), + {1}, 8, "uint64", "little"); + + writer.write_data("frame", pulse_id, + (char*)&(metadata.frame_index), + {1}, 8, "uint64", "little"); + + writer.write_data("daq_rec", pulse_id, + (char*)&(metadata.daq_rec), + {1}, 8, "uint64", "little"); + + uint64_t is_good_frame = 0; + if (metadata.n_recv_packets == 4096) { + is_good_frame = 1; + } + + writer.write_data("is_good_frame", pulse_id, + (char*)&(is_good_frame), + {1}, 8, "uint64", "little"); } - int rank_id; - if(MPI_Comm_rank (MPI_COMM_WORLD, &rank_id) != MPI_SUCCESS) { - throw runtime_error("Cannot get the MPI rank."); - } - - - - - if (MPI_Finalize() != MPI_SUCCESS) { - throw runtime_error("Cannot finalize MPI."); - }; + reader.close(); return 0; } + +//unordered_map header_values { +// {"pulse_id", HeaderDataType("uint64")}, +// {"frame", HeaderDataType("uint64")}, +// {"is_good_frame", HeaderDataType("uint64")}, +// {"daq_rec", HeaderDataType("int64")}, +// +// {"pulse_id_diff", HeaderDataType("int64", n_modules)}, +// {"framenum_diff", HeaderDataType("int64", n_modules)}, +// +// {"missing_packets_1", HeaderDataType("uint64", n_modules)}, +// {"missing_packets_2", HeaderDataType("uint64", n_modules)}, +// {"daq_recs", HeaderDataType("uint64", n_modules)}, +// +// {"pulse_ids", HeaderDataType("uint64", n_modules)}, +// {"framenums", HeaderDataType("uint64", n_modules)}, +// +// {"module_number", HeaderDataType("uint64", n_modules)} +// }); \ No newline at end of file