#include #include #include #include #include "TROOT.h" #include "TFile.h" #include "JAcoustics/JEvent.hh" #include "JAcoustics/JSupport.hh" #include "JSupport/JSingleFileScanner.hh" #include "JLang/JException.hh" #include "JLang/JComparator.hh" #include "JLang/JComparison.hh" #include "Jeep/JParser.hh" #include "Jeep/JMessage.hh" namespace { using JLANG::JValueOutOfRange; using JACOUSTICS::JTransmission; using JACOUSTICS::JEvent; enum JCompare_t { ToA_t = 1, ToE_t = 2 }; int cta = 0; // time selection const struct { /** * Compare transmissions. * * \param first first transmission * \param second second transmission * \return true if first transmission less than second; else false */ inline bool operator()(const JTransmission& first, const JTransmission& second) const { if (cta != ToA_t && cta != ToE_t) { THROW(JValueOutOfRange, "Invalid time selection " << cta); } if (first.getRunNumber() == second.getRunNumber()) { if (first.getID() == second.getID()) { if ((cta == ToA_t && first.getToA() == second.getToA()) || (cta == ToE_t && first.getToE() == second.getToE())) { if (first.getQ() == second.getQ()) { return first.getW() < second.getW(); } else { return first.getQ() < second.getQ(); } } else { if (cta == ToA_t) { return first.getToA() < second.getToA(); } if (cta == ToE_t) { return first.getToE() < second.getToE(); } THROW(JValueOutOfRange, "Invalid time selection " << cta); } } else { return first.getID() < second.getID(); } } else { return first.getRunNumber() < second.getRunNumber(); } } /** * Compare events. * * \param first first event * \param second second event * \return true if first event less than second; else false */ inline bool operator()(const JEvent& first, const JEvent& second) const { if (first.getOID() == second.getOID()) { if (first.getID() == second.getID()) { if (first.size() == second.size()) { for (JEvent::const_iterator p0 = first .begin(), p1 = second.begin(); p0 != first.end() && p1 != second.end(); ++p0, ++p1) { if ((*this)(*p0, *p1)) { return true; } } return false; } else { return first.size() < second.size(); } } else { return first.getID() < second.getID(); } } else { return first.getOID() < second.getOID(); } } } compare; /** * Printer. */ struct printer { int debug; size_t width; /** * Print data. * * \param out output stream * \param filename file name * \param index index * \param evt event * \param prefix prefix * \param postfix postfix * \return output stream */ std::ostream& operator()(std::ostream& out, const std::string& filename, const int index, const JEvent& evt, const std::string& prefix, const std::string& postfix) const { using namespace std; using namespace JPP; if (debug >= debug_t) { out << prefix << (prefix == "" ? "" : " ") << setw(width) << left << filename << right << ' ' << "[" << FILL(6,'0') << index << "]" << FILL() << ' ' << evt.getOID() << ' ' << setw(6) << evt.getCounter() << ' ' << setw(2) << evt.getID() << ' ' << FIXED(12,6) << evt. begin()->getToA() << ' ' << FIXED(12,6) << evt.rbegin()->getToA() << (postfix == "" ? "" : " ") << postfix << endl; } return out; } /** * Write transmission to output stream. * * \param out output stream * \param object transmission * \param prefix prefix * \param postfix postfix * \return output stream */ std::ostream& operator()(std::ostream& out, const JTransmission& object, const std::string& prefix, const std::string& postfix) const { using namespace std; using namespace JPP; if (debug >= debug_t) { out << prefix << (prefix == "" ? "" : " ") << setw(8) << object.getID() << ' ' << setw(8) << object.getRunNumber() << ' ' << FIXED(13,7) << object.getToA() << ' ' << FIXED(13,7) << object.getToE() << ' ' << FIXED(8,0) << object.getQ() << (postfix == "" ? "" : " ") << postfix << endl; } return out; } }; /** * Compare acoustics events by time of arrival of first hit. * * \param first first event * \param second second event * \return true if first event earliear than second; else false */ static inline bool toa(const JEvent& first, const JEvent& second) { return first.begin()->getToA() < second.begin()->getToA(); } /** * Compare acoustics events by time of emission of first hit. * * \param first first event * \param second second event * \return true if first event earliear than second; else false */ static inline bool toe(const JEvent& first, const JEvent& second) { return first.begin()->getToE() < second.begin()->getToE(); } /** * Options. */ enum JOption_t { Null_t = 0, Sort_t = 1 }; } /** * \file * * Program to compare acoustics event data. * \author mdejong */ int main(int argc, char **argv) { using namespace std; using namespace JPP; vector inputFile; JLimit numberOfEvents; int option; int debug; try { JParser<> zap("Program to compare acoustics event data."); zap['f'] = make_field(inputFile, "two outputs of JAcousticsEventBuilder[.sh]"); zap['n'] = make_field(numberOfEvents) = JLimit::max(); zap['O'] = make_field(option, Null_t << " -> nothing " << Sort_t << " -> sort") = Null_t, Sort_t; zap['C'] = make_field(cta, ToA_t << " -> ToA " << ToE_t << " -> ToE") = ToA_t, ToE_t; zap['d'] = make_field(debug) = 2; zap(argc, argv); } catch(const exception &error) { FATAL(error.what() << endl); } if (inputFile.size() != 2u) { FATAL("Wrong number of input files " << inputFile.size() << endl); } const size_t width = max(inputFile[0].size(), inputFile[1].size()); const printer print = { debug, width }; vector buffer[2]; for (int i = 0; i != 2; ++i) { for (JSingleFileScanner in(inputFile[i], numberOfEvents); in.hasNext(); ) { buffer[i].push_back(*in.next()); } } if (option == Sort_t) { for (int i = 0; i != 2; ++i) { for (JEvent& evt : buffer[i]) { sort(evt.begin(), evt.end(), compare); } if (cta == ToA_t) { sort(buffer[i].begin(), buffer[i].end(), toa); } if (cta == ToE_t) { sort(buffer[i].begin(), buffer[i].end(), toe); } for (vector::iterator p = buffer[i].begin(); p != buffer[i].end(); ++p) { sort(p->begin(), p->end(), compare); } } } int count[] = { 0, 0 }; for (vector::const_iterator p0 = buffer[0].begin(), p1 = buffer[1].begin(); p0 != buffer[0].end() && p1 != buffer[1].end(); ) { for ( ; p0 != buffer[0].end() && p1 != buffer[1].end() && ((cta == ToA_t && toa(*p0,*p1)) || (cta == ToE_t && toe(*p0,*p1))); ++p0, ++count[1]) { print(cout, inputFile[0], distance(buffer[0].cbegin(),p0), *p0, ">>", ""); } for ( ; p0 != buffer[0].end() && p1 != buffer[1].end() && ((cta == ToA_t && toa(*p1,*p0)) || (cta == ToE_t && toe(*p1,*p0))); ++p1, ++count[1]) { print(cout, inputFile[1], distance(buffer[1].cbegin(),p1), *p1, "<<", ""); } if (p0 != buffer[0].end() && p1 != buffer[1].end()) { if (!compare(*p0,*p1) && !compare(*p1,*p0)) { ++count[0]; print(cout, inputFile[0], distance(buffer[0].cbegin(),p0), *p0, "", "\\"); print(cout, inputFile[1], distance(buffer[1].cbegin(),p1), *p1, "", "/ "); } else { ++count[1]; print(cout, inputFile[0], distance(buffer[0].cbegin(),p0), *p0, "", "*"); print(cout, inputFile[1], distance(buffer[1].cbegin(),p1), *p1, "", "*"); if (p0->getOID() == p1->getOID() && p0->getCounter() == p1->getCounter() && p0->getID() == p1->getID()) { JEvent::const_iterator i0 = p0->begin(); JEvent::const_iterator i1 = p1->begin(); while (i0 != p0->end() && i1 != p1->end()) { for ( ; i0 != p0->end() && i1 != p1->end() && compare(*i0,*i1); ++i0) { print(cout, *i0, ">>", ""); } for ( ; i0 != p0->end() && i1 != p1->end() && compare(*i1,*i0); ++i1) { print(cout, *i1, "<<", ""); } if (i0 != p0->end() && i1 != p1->end()) { if (!compare(*i0, *i1) && !compare(*i1,*i0)) { ++i0; ++i1; } } } for ( ; i0 != p0->end(); ++i0) { print(cout, *i0, ">>", ""); } for ( ; i1 != p1->end(); ++i1) { print(cout, *i1, "<<", ""); } } else { print(cout, inputFile[0], distance(buffer[0].cbegin(),p0), *p0, ">>", ""); print(cout, inputFile[1], distance(buffer[1].cbegin(),p1), *p1, "<<", ""); } } if (toa(*p0,*p1) || toa(*p1,*p0)) { } else { ++p0; ++p1; } } } STATUS("Number of differences / events: " << count[1] << " / " << count[0] << endl); if (buffer[0].size() != buffer[1].size()) { FATAL("Different size " << buffer[0].size() << ' ' << buffer[1].size() << endl); } if (count[1] != 0) { FATAL("Number of differences " << count[1] << endl); } }