IOPool/Streamer/src/DumpTools.cc - master - githubmirror/cmssw

IOPool/Streamer/src/DumpTools.cc
232 строки8 KB
Matti Kortelainen
Include ProcessingHistory ID to DiagStreamerFile checks and printouts
11 июн 2026, 22:55
11 июн 2026, 22:55da8b240
/** File contains simple tools to dump Init and Event 
    Messages on screen.
*/

#include "IOPool/Streamer/interface/DumpTools.h"
#include "FWCore/Utilities/interface/Digest.h"
#include "FWCore/Utilities/interface/Algorithms.h"
#include "DataFormats/Streamer/interface/StreamedProducts.h"
#include "IOPool/Streamer/interface/ClassFiller.h"
#include "IOPool/Streamer/interface/uncompress.h"

#include "TBufferFile.h"

#include <iostream>
#include <iterator>
#include <memory>

using namespace edm;

namespace edm::streamer {
  void dumpInitHeader(const InitMsgView* view) {
    std::cout << "code = " << view->code() << ", "
              << "size = " << view->size() << "\n"
              << "run = " << view->run() << ", "
              << "proto = " << view->protocolVersion() << "\n"
              << "release = " << view->releaseTag() << "\n"
              << "processName = " << view->processName() << "\n";
    if (view->protocolVersion() >= 5) {
      std::cout << "outModuleLabel = " << view->outputModuleLabel() << "\n";
    }
    if (view->protocolVersion() >= 6) {
      std::cout << "outputModuleId=0x" << std::hex << view->outputModuleId() << std::dec << std::endl;
    }
    if (view->protocolVersion() >= 8) {
      std::cout << "Checksum for Registry data = " << view->adler32_chksum() << " Hostname = " << view->hostName()
                << std::endl;
    }

    //PSet 16 byte non-printable representation, stored in message.
    uint8 vpset[16];
    view->pset(vpset);

    //Lets convert it to printable hex form
    std::string pset_str(vpset, vpset + sizeof(vpset));
    pset_str += '\0';
    cms::Digest dig(pset_str);
    cms::MD5Result r1 = dig.digest();
    std::string hexy = r1.toString();
    std::cout << "PSetID= " << hexy << std::endl;

    Strings vhltnames, vhltselections, vl1names;
    view->hltTriggerNames(vhltnames);
    if (view->protocolVersion() >= 5) {
      view->hltTriggerSelections(vhltselections);
    }
    view->l1TriggerNames(vl1names);

    std::cout << "HLT names :- \n ";
    edm::copy_all(vhltnames, std::ostream_iterator<std::string>(std::cout, "\n"));

    if (view->protocolVersion() >= 5) {
      std::cout << "HLT selections :- \n ";
      edm::copy_all(vhltselections, std::ostream_iterator<std::string>(std::cout, "\n"));
    }

    std::cout << "L1 names :- \n ";
    edm::copy_all(vl1names, std::ostream_iterator<std::string>(std::cout, "\n"));
    std::cout << "\n";
    std::cout.flush();
  }

  void dumpInitView(const InitMsgView* view) {
    dumpInitHeader(view);
    std::cout << "desc len = " << view->descLength() << "\n";
    //const uint8* pos = view->descData();
    //std::copy(pos,pos+view->descLength(),std::ostream_iterator<uint8>(std::cout,""));
    //std::cout << "\n";
    std::cout.flush();
  }

  void dumpStartMsg(const InitMsgView* view) {
    dumpInitHeader(view);
    std::cout.flush();
  }

  void dumpInitVerbose(const InitMsgView* view) {
    std::cout << ">>>>> INIT Message Dump (begin) >>>>>" << std::endl;
    dumpInitHeader(view);

    TClass* desc = getTClass(typeid(SendJobHeader));
    TBufferFile xbuf(TBuffer::kRead, view->descLength(), (char*)view->descData(), kFALSE);
    std::unique_ptr<SendJobHeader> sd((SendJobHeader*)xbuf.ReadObjectAny(desc));

    if (sd.get() == nullptr) {
      std::cout << "Unable to determine the product registry - "
                << "Registry deserialization error." << std::endl;
    } else {
      std::cout << "Branch Descriptions:" << std::endl;
      SendDescs const& descs = sd->descs();
      SendDescs::const_iterator iDesc(descs.begin()), eDesc(descs.end());
      while (iDesc != eDesc) {
        ProductDescription branchDesc = *iDesc;
        branchDesc.init();
        //branchDesc.write(std::cout);
        std::cout << branchDesc.branchName() << std::endl;
        iDesc++;
      }
    }

    std::cout << "<<<<< INIT Message Dump (end) <<<<<" << std::endl;
    std::cout.flush();
  }

  void dumpInit(uint8* buf) {
    InitMsgView view(buf);
    dumpInitHeader(&view);

    std::cout << "desc len = " << view.descLength() << "\n";
    //const uint8* pos = view.descData();
    //std::copy(pos,pos+view.descLength(),std::ostream_iterator<uint8>(std::cout,""));
    //std::cout << "\n";
    std::cout.flush();
  }

  void printBits(unsigned char c) {
    for (int i = 7; i >= 0; --i) {
      int bit = ((c >> i) & 1);
      std::cout << " " << bit;
    }
  }

  void dumpEventHeader(const EventMsgView* eview) {
    std::string_view algo = "uncompressed";
    unsigned long origsize = eview->origDataSize();
    if (origsize != 0 && origsize != 78) {
      using namespace edm::streamer::uncompress;
      auto ealgo = compressionAlgo(static_cast<unsigned char const*>(eview->eventData()), eview->eventLength());
      switch (ealgo) {
        case Algo::kZLIB:
          algo = "ZLIB";
          break;
        case Algo::kZSTD:
          algo = "ZSTD";
          break;
        case Algo::kLZMA:
          algo = "LZMA";
          break;
      }
    }

    std::cout << "code=" << eview->code() << "\n"
              << "size=" << eview->size() << "\n"
              << "protocolVersion=" << eview->protocolVersion() << "\n"
              << "run=" << eview->run() << "\n"
              << "event=" << eview->event() << "\n"
              << "lumi=" << eview->lumi() << "\n"
              << "origDataSize=" << eview->origDataSize() << "\n"
              << "outModId=0x" << std::hex << eview->outModId() << std::dec << "\n"
              << "compression=" << algo << "\n"
              << "adler32 chksum= " << eview->adler32_chksum() << "\n"
              << "host name= " << eview->hostName() << "\n"
              << "event length=" << eview->eventLength() << "\n"
              << "droppedEventsCount=" << eview->droppedEventsCount() << "\n";

    std::vector<bool> l1_out;
    eview->l1TriggerBits(l1_out);

    std::cout << "\nl1 size= " << l1_out.size() << "\n l1 bits=\n";
    edm::copy_all(l1_out, std::ostream_iterator<bool>(std::cout, " "));

    std::vector<unsigned char> hlt_out;
    if (eview->hltCount() > 0) {
      hlt_out.resize(1 + (eview->hltCount() - 1) / 4);
    }
    eview->hltTriggerBits(&hlt_out[0]);

    std::cout << "\nhlt Count:" << eview->hltCount();
    std::cout << "\nhlt bits=\n(";
    for (int i = (hlt_out.size() - 1); i != -1; --i)
      printBits(hlt_out[i]);
    std::cout << ")\n";
    std::cout.flush();
  }

  void dumpEventView(const EventMsgView* eview, const SendEvent* sendEvent) {
    dumpEventHeader(eview);
    //const uint8* edata = eview->eventData();
    //std::cout << "\nevent data=\n(";
    //std::copy(&edata[0],&edata[0]+eview->eventLength(),
    //     std::ostream_iterator<char>(std::cout,""));
    //std::cout << ")\n";
    if (sendEvent) {
      auto history = sendEvent->processHistory();
      history.reduce();
      std::cout << "\nReduced ProcessHistory ID=" << history.id() << "\n";
    }
    std::cout.flush();
  }

  void dumpEventIndex(const EventMsgView* eview) {
    dumpEventHeader(eview);
    std::cout.flush();
  }

  void dumpEvent(uint8* buf) {
    EventMsgView eview(buf);

    dumpEventHeader(&eview);

    //const uint8* edata = eview.eventData();
    //std::cout << "\nevent data=\n(";
    //std::copy(&edata[0],&edata[0]+eview.eventLength(),
    //     std::ostream_iterator<char>(std::cout,""));
    //std::cout << ")\n";
    std::cout.flush();
  }

  void dumpFRDEventView(const FRDEventMsgView* fview) {
    std::cout << "\n>>>>> FRDEvent Message Dump (begin) >>>>>" << std::endl;
    std::cout.flush();

    std::cout << "size = " << fview->size() << "\n"
              << "version = " << fview->version() << "\n"
              << "run = " << fview->run() << "\n"
              << "lumi = " << fview->lumi() << "\n"
              << "event = " << fview->event() << "\n";
    std::cout.flush();

    std::cout << ">>>>> FRDEvent Message Dump (end) >>>>>" << std::endl;
    std::cout.flush();
  }
}  // namespace edm::streamer