IOPool/Streamer/test/WriteStreamerFile.cpp - master - githubmirror/cmssw

IOPool/Streamer/test/WriteStreamerFile.cpp
144 строки5 KB
Chris Jones
Moved code to edm::streamer namespace
02 май 2024, 00:33
02 май 2024, 00:33bcbcf69
/** Example code shows you to write Streamer files.
    All values are dummy here, The Init message contains different
    values from what Event Header contains, this is only
    for the demonstration, obviously.

    Change total number of written events using,

     #define NO_OF_EVENTS 10

    Crank it up to 10000 if you like to scrol your screen ;-).

Disclaimer: Most of the code here is randomly written during
               testing various parts, its not a supported testing code.
               Changes can and will be made, when and if required.
*/

#include <iostream>
#include "IOPool/Streamer/interface/MsgTools.h"
#include "IOPool/Streamer/interface/EventMsgBuilder.h"
#include "IOPool/Streamer/interface/InitMsgBuilder.h"
#include "IOPool/Streamer/interface/InitMessage.h"

#include "FWCore/Utilities/interface/Adler32Calculator.h"
#include "FWCore/Utilities/interface/Exception.h"
#include "IOPool/Streamer/interface/StreamerOutputFile.h"

#include "zlib.h"

#define NO_OF_EVENTS 10

using namespace edm::streamer;

int main() try {
  typedef std::vector<uint8> Buffer;
  Buffer buf(1024);
  Buffer buf2(11024);

  // ----------- init

  char psetid[] = "1234567890123456";
  char test_value[] = "This is a test, This is a";
  char test_value_event[] = "This is a test Event, This is a";
  Strings hlt_names;
  Strings l1_names;

  hlt_names.push_back("a");
  hlt_names.push_back("b");
  hlt_names.push_back("c");
  hlt_names.push_back("d");
  hlt_names.push_back("e");
  hlt_names.push_back("f");
  hlt_names.push_back("g");
  hlt_names.push_back("h");
  hlt_names.push_back("i");

  l1_names.push_back("t10");
  l1_names.push_back("t11");
  l1_names.push_back("t12");
  l1_names.push_back("t13");
  l1_names.push_back("t14");
  l1_names.push_back("t15");
  l1_names.push_back("t16");
  l1_names.push_back("t17");
  l1_names.push_back("t18");
  l1_names.push_back("t19");
  l1_names.push_back("t20");

  char reltag[] = "CMSSW_0_8_0_pre7";
  std::string processName = "HLT";
  std::string outputModuleLabel = "HLTOutput";

  uLong crc = crc32(0L, Z_NULL, 0);
  Bytef* crcbuf = (Bytef*)outputModuleLabel.data();
  crc = crc32(crc, crcbuf, outputModuleLabel.length());

  uint32 adler32_chksum = (uint32)cms::Adler32((char*)&test_value[0], sizeof(test_value));

  InitMsgBuilder init(&buf[0],
                      buf.size(),
                      12,
                      Version((const uint8*)psetid),
                      (const char*)reltag,
                      processName.c_str(),
                      outputModuleLabel.c_str(),
                      crc,
                      hlt_names,
                      hlt_names,
                      l1_names,
                      adler32_chksum);

  init.setDataLength(sizeof(test_value));
  std::copy(&test_value[0], &test_value[0] + sizeof(test_value), init.dataAddress());

  //Do a dumpInit here if you need to see the event.

  //Start the Streamer file
  std::cout << "Trying to Write a Streamer file" << std::endl;
  std::string initfilename = "teststreamfile.dat";
  StreamerOutputFile stream_writer(initfilename);

  std::cout << "Trying to Write Out The Init message into Streamer File: " << initfilename << std::endl;
  stream_writer.write(init);

  // ------- event

  std::vector<bool> l1bit(11);
  uint8 hltbits[] = "4567";
  const int hltsize = 9; /** I am interested in 9 bits only */

  l1bit[0] = true;
  l1bit[4] = true;
  l1bit[8] = false;  //l1bit[12]=true;
  l1bit[1] = true;
  l1bit[5] = false;
  l1bit[9] = false;  //l1bit[13]=false;
  l1bit[2] = false;
  l1bit[6] = true;
  l1bit[10] = true;  //l1bit[14]=false;
  l1bit[3] = false;
  l1bit[7] = false;  //l1bit[11]=true;  //l1bit[15]=true;
  //l1bit[16]=false;  l1bit[17]=false;  l1bit[18]=true;  l1bit[19]=true;

  //Lets Build 10 Events and then Write them into Streamer file.

  adler32_chksum = (uint32)cms::Adler32((char*)&test_value_event[0], sizeof(test_value_event));
  std::string host_name = "mytestnode.cms";

  for (uint32 eventId = 2000; eventId != 2000 + NO_OF_EVENTS; ++eventId) {
    EventMsgBuilder emb(
        &buf2[0], buf2.size(), 45, eventId, 2, 0xdeadbeef, 3, l1bit, hltbits, hltsize, adler32_chksum, host_name.c_str());
    emb.setOrigDataSize(78);
    emb.setEventLength(sizeof(test_value_event));
    std::copy(&test_value_event[0], &test_value_event[0] + sizeof(test_value_event), emb.eventAddr());

    //Lets write this to our streamer file .
    std::cout << "Writing Event# : " << eventId << " To Streamer file" << std::endl;
    stream_writer.write(emb);
  }

  return 0;
} catch (cms::Exception const& e) {
  std::cerr << e.explainSelf() << std::endl;
  return 1;
}