UECalibration.h

Go to the documentation of this file.

#ifndef __HiJetAlgos_UECalibration_h__
#define __HiJetAlgos_UECalibration_h__

// SVD Block Predictor
#include <fstream>
#include <sstream>

#include "FWCore/ParameterSet/interface/FileInPath.h"

struct UECalibration {
   UECalibration(const std::vector<float> &v)
    {
        np[0] = 3;  // Number of reduced PF ID (track, ECAL, HCAL)
        np[1] = 15; // Number of pseudorapidity block
        np[2] = 5;  // Fourier series order
        np[3] = 2;  // Re or Im
        np[4] = 82; // Number of feature parameter

        ni0[0] = np[1];
        ni0[1] = 344;

        ni1[0] = np[1];
        ni1[1] = 344;

        ni2[0] = np[1];
        ni2[1] = 82;

        unsigned int Nnp_full = np[0] * np[1] * np[2] * np[3] * np[4];
        unsigned int Nnp = np[0] * np[1] * (1 + (np[2] - 1) * np[3]) * np[4];
        unsigned int Nni0 = ni0[0]*ni0[1];
        unsigned int Nni1 = ni1[0]*ni1[1];
        unsigned int Nni2 = ni2[0]*ni2[1];

        memset(ue_predictor_pf, 0, Nnp_full * sizeof(float));
        memset(ue_interpolation_pf0, 0, Nni0 * sizeof(float));
        memset(ue_interpolation_pf1, 0, Nni1 * sizeof(float));
        memset(ue_interpolation_pf2, 0, Nni2 * sizeof(float));

        for (index = 0; index < Nnp + Nni0 + Nni1 + Nni2; index++) {      
            float val = v[index];
            int bin0, bin1, bin2, bin3, bin4;

            if (index < Nnp){
                bin4 = index % np[4];

                unsigned int u = (index / np[4]) % (1 + (np[2] - 1) * np[3]);

                if (u == 0) {
                    bin2 = 0;
                    bin3 = 0;
                }
                else {
                    bin2 = (u - 1) / np[3] + 1;
                    bin3 = (u - 1) % np[3];
                }

              bin1 = ((index / np[4]) / (1 + (np[2] - 1) * np[3])) % np[1];
              bin0 = ((index / np[4]) / (1 + (np[2] - 1) * np[3])) / np[1];

              ue_predictor_pf[bin0][bin1][bin2][bin3][bin4] = val;
          }
          else if (index < Nnp + Nni0) {
              bin1 = (index - Nnp) % ni0[1];
              bin0 = (index - Nnp) / ni0[1];
              ue_interpolation_pf0[bin0][bin1] = val;
          }
          else if (index < Nnp + Nni0 + Nni1) {
              bin1 = (index - Nnp - Nni0) % ni1[1];
              bin0 = (index - Nnp - Nni0) / ni1[1];
              ue_interpolation_pf1[bin0][bin1] = val;
          }
          else {
              bin1 = (index - Nnp - Nni0 - Nni1) % ni2[1];
              bin0 = (index - Nnp - Nni0 - Nni1) / ni2[1];
              ue_interpolation_pf2[bin0][bin1] = val;
          }
      }
    }

  UECalibration(std::string calibrationFile){
       np[0] = 3;   // Number of reduced PF ID (track, ECAL, HCAL)
       np[1] = 15;  // Number of pseudorapidity block
       np[2] = 5;   // Fourier series order
       np[3] = 2;   // Re or Im
       np[4] = 82;  // Number of feature parameter

      ni0[0] = np[1];
      ni0[1] = 344;

      ni1[0] = np[1];
      ni1[1] = 344;

      ni2[0] = np[1];
      ni2[1] = 82;

      index = 0;

      unsigned int Nnp_full = np[0] * np[1] * np[2] * np[3] * np[4];
      unsigned int Nnp = np[0] * np[1] * (1 + (np[2] - 1) * np[3]) * np[4];
      unsigned int Nni0 = ni0[0]*ni0[1];
      unsigned int Nni1 = ni1[0]*ni1[1];
      unsigned int Nni2 = ni2[0]*ni2[1];

      memset(ue_predictor_pf, 0, Nnp_full * sizeof(float));
      memset(ue_interpolation_pf0, 0, Nni0 * sizeof(float));
      memset(ue_interpolation_pf1, 0, Nni1 * sizeof(float));
      memset(ue_interpolation_pf2, 0, Nni2 * sizeof(float));

      edm::FileInPath ueData(calibrationFile.data());
      std::string qpDataName = ueData.fullPath();
      std::ifstream in( qpDataName.c_str() );
      std::string line;

      while( std::getline( in, line)){
     if(!line.size() || line[0]=='#') {
        continue;
     }
     std::istringstream linestream(line);
     float val;
     int bin0, bin1, bin2, bin3, bin4;
     if(index < Nnp){
        //      cout<<"predictor "<<bin0<<" "<<bin1<<" "<<bin2<<" "<<bin3<<" "<<bin4<<" "<<val<<endl;
        linestream>>bin0>>bin1>>bin2>>bin3>>bin4>>val;
        ue_predictor_pf[bin0][bin1][bin2][bin3][bin4] = val;
     }else if(index < Nnp + Nni0){
        //            cout<<"inter_0 "<<bin0<<" "<<bin1<<" "<<bin2<<" "<<bin3<<" "<<bin4<<" "<<val<<endl;
        linestream>>bin0>>bin1>>val;
        ue_interpolation_pf0[bin0][bin1] = val;
     }else if(index < Nnp + Nni0 + Nni1){
        //            cout<<"inter_1 "<<bin0<<" "<<bin1<<" "<<bin2<<" "<<bin3<<" "<<bin4<<" "<<val<<endl;
            linestream>>bin0>>bin1>>val;
            ue_interpolation_pf1[bin0][bin1] = val;
     }else if(index < Nnp + Nni0 + Nni1 + Nni2){
        //            cout<<"inter_2 "<<bin0<<" "<<bin1<<" "<<bin2<<" "<<bin3<<" "<<bin4<<" "<<val<<endl;
        linestream>>bin0>>bin1>>val;
        ue_interpolation_pf2[bin0][bin1] = val;
     }
     ++index;
      }
   }

   unsigned int index,
      np[5], 
      ni0[2], 
      ni1[2], 
      ni2[2];

   float ue_predictor_pf[3][15][5][2][82], 
      ue_interpolation_pf0[15][344], 
      ue_interpolation_pf1[15][344], 
      ue_interpolation_pf2[15][82];
   
};


#endif