his2root.cc

Go to the documentation of this file.

#include <fstream>
#include <iostream>
#include <sstream>
#include <string>
 
#include <TFile.h>
#include <TH1.h>
#include <TH2.h>

int main(int argc, char *argv[])
{
   int nHists = 0; // counter for number of histograms processed
   const char *cHisFileName = argv[1];
   std::cout << "Converting file " << cHisFileName << std::endl;
   int big = 0;
   if (argc > 2) {
     big = std::atoi(argv[2]);
   }
   std::stringstream ss;
   std::string line, sListFileName, sHisFileName(cHisFileName), sRootFileName;
   sListFileName = sHisFileName.substr(0, sHisFileName.length() - 3);
   sListFileName.append("list");
   sRootFileName = sHisFileName.substr(0, sHisFileName.length() - 3);
   sRootFileName.append("root");
   std::ifstream listFile;
   listFile.open(sListFileName);
   std::string first;
   int details = 0; // flag to mark if we're in the "detailed info" part of the
   // list file or not
 
   int HID;  /*histogram ID*/
   int dim;  /* dimension of histogram: assuming 1 or 2 */
   int hwpc; /* half words per channel */
   // int len;   /* length in characters of array */
   // int compr; /* compression, not used */
   int minX;          /* minimum (X) value of array */
   int maxX;          /* max value (X) */
   int minY;          /* minimum (Y) value of array */
   int maxY;          /* max value (Y) */
   std::string title; /* title of histogram */
   uint32_t datum;    /* 64 bit integer to store a single channel */
 
   FILE *hisFile;
   int read;
   hisFile = std::fopen(cHisFileName, "rb");
   TFile *outFile = new TFile(sRootFileName.c_str(), "recreate");
 
   while (std::getline(listFile, line)) {
     ss.clear();
     ss.str(line);
     ss >> first;
     if (details == 1) { /* if past the list of IDs */
       std::cout << "processing line: " << line << std::endl;
       HID = stoi(line.substr(0, 5));
       dim = stoi(line.substr(5, 5));
       hwpc = stoi(line.substr(10, 4));
       // compr = stoi(line.substr(23, 8));
       minX = stoi(line.substr(31, 6));
       maxX = stoi(line.substr(37, 6));
       title = line.substr(54);
 
       if (hwpc > 2) {
         std::cout << "warning! more than 32 bit integer, expect bad behaviour" << std::endl;
       }
       if (dim == 2) { /* 2D histogram */
         std::getline(listFile, line);
         ss.clear();
         ss.str(line);
         minY = stoi(line.substr(31, 6));
         maxY = stoi(line.substr(37, 6));
 
         /* make the histogram */
         TString sHistName;
         sHistName.Form("ID%d", HID);
         TH2D *hist = new TH2D(sHistName.Data(), title.c_str(), maxX - minX + 1, minX, maxX + 1, maxY - minY + 1,
                               minY, maxY + 1);
 
         /* fill the histogram */
         for (int iyChannel = 0; iyChannel <= maxY; iyChannel++) {
           int bpc = hwpc*2;
           uint8_t data[bpc*(maxX+1)];
           std::fread(&data[0], (size_t)1, (size_t)(bpc*(maxX+1)), hisFile);
 
           for (int ixChannel = 0; ixChannel <= maxX; ixChannel++) {
             /* changer order of bytes */
             uint32_t val = 0;
             for (int i=0; i<bpc; ++i) {
               if (big) {
                 val += data[bpc*ixChannel+bpc-i-1]<<(8*i);
               } else {
                 val += data[bpc*ixChannel+i]<<(8*i);
               }
             }
 
             hist->SetBinContent(ixChannel+1, iyChannel+1, val);
             hist->SetBinError(ixChannel+1, iyChannel+1, std::sqrt(val));
           }
         }
 
         hist->Write();
       } else { /* 1D histogram */
         TString sHistName;
         sHistName.Form("ID%d", HID);
         TH1D *hist = new TH1D(sHistName.Data(), title.c_str(), maxX - minX + 1, minX, maxX + 1);
 
         /* fill the histogram */
         int bpc = hwpc*2;
         uint8_t data[bpc*(maxX+1)];
         std::fread(&data[0], (size_t)1, (size_t)(bpc*(maxX+1)), hisFile);
               
         for (int ixChannel = 0; ixChannel <= maxX; ixChannel++) {
           /* changer order of bytes */
           uint32_t val = 0;
           for (int i=0; i<bpc; ++i) {
             if (big) {
               val += data[bpc*ixChannel+bpc-i-1]<<(8*i);
             } else {
               val += data[bpc*ixChannel+i]<<(8*i);
             }
           }
                 
           hist->SetBinContent(ixChannel+1, val);
           hist->SetBinError(ixChannel+1, std::sqrt(val));
         }
 
         hist->Write();
       }
 
       nHists = nHists + 1;
     }
     if (first == "HID") {
       details = 1;
     }
 
   } // *.list file loop
   outFile->Close();
   return nHists;
} // his2root