CSCGasGainCorrectionDBConditions

gain_info

Macros

Line Code
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161
#ifndef _CSCGASGAINCORRECTIONDBCONDITIONS_H
#define _CSCGASGAINCORRECTIONDBCONDITIONS_H

#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/ESProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/EventSetupRecordIntervalFinder.h"
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/SourceFactory.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include <cmath>
#include <memory>

#include "CondFormats/CSCObjects/interface/CSCDBGasGainCorrection.h"
#include "CondFormats/DataRecord/interface/CSCDBGasGainCorrectionRcd.h"
#include "DataFormats/MuonDetId/interface/CSCIndexer.h"
#include <DataFormats/MuonDetId/interface/CSCDetId.h>

class CSCGasGainCorrectionDBConditions : public edm::ESProducer, public edm::EventSetupRecordIntervalFinder {
public:
  CSCGasGainCorrectionDBConditions(const edm::ParameterSet &);
  ~CSCGasGainCorrectionDBConditions() override;

  inline static CSCDBGasGainCorrection *prefillDBGasGainCorrection(bool isForMC, std::string dataCorrFileName);

  typedef std::unique_ptr<CSCDBGasGainCorrection> ReturnType;

  ReturnType produceDBGasGainCorrection(const CSCDBGasGainCorrectionRcd &);

private:
  // ----------member data ---------------------------
  void setIntervalFor(const edm::eventsetup::EventSetupRecordKey &,
                      const edm::IOVSyncValue &,
                      edm::ValidityInterval &) override;

  // Flag for determining if this is for setting MC or data corrections
  bool isForMC;
  // File for reading 55944 gas gain corrections.  MC will be fake;
  std::string dataCorrFileName;
};

#include "CondFormats/CSCObjects/interface/CSCDBGasGainCorrection.h"
#include "CondFormats/DataRecord/interface/CSCDBGasGainCorrectionRcd.h"
#include <DataFormats/MuonDetId/interface/CSCDetId.h>

#include <fstream>
#include <iostream>
#include <vector>

// to workaround plugin library
inline CSCDBGasGainCorrection *CSCGasGainCorrectionDBConditions::prefillDBGasGainCorrection(bool isMC,
                                                                                            std::string filename) {
  if (isMC)
    printf("\n Generating fake DB constants for MC\n");
  else {
    printf("\n Reading gas gain corrections from file %s \n", filename.data());
  }

  CSCIndexer indexer;

  const int MAX_SIZE = 55944;

  CSCDBGasGainCorrection *cndbGasGainCorr = new CSCDBGasGainCorrection();

  CSCDBGasGainCorrection::GasGainContainer &itemvector = cndbGasGainCorr->gasGainCorr;
  itemvector.resize(MAX_SIZE);

  // Filling corrections for MC is very simple
  if (isMC) {
    for (int i = 0; i < MAX_SIZE; i++) {
      itemvector[i].gainCorr = 1.;
    }
    return cndbGasGainCorr;
  }

  struct gain_info {
    int gas_gain_index;
    int endcap;
    int station;
    int ring;
    int chamber;
    int layer;
    int hvsegment;
    int cfeb;
    int nentries;
    float mean;
    float truncated_mean;
    float gas_gain_correction;
  } gains[MAX_SIZE];

  for (int j = 0; j < MAX_SIZE; j++) {
    gains[j].gas_gain_index = -999;
    gains[j].endcap = -999;
    gains[j].station = -999;
    gains[j].ring = -999;
    gains[j].chamber = -999;
    gains[j].layer = -999;
    gains[j].hvsegment = -999;
    gains[j].cfeb = -999;
    gains[j].nentries = -999;
    gains[j].mean = -999.;
    gains[j].truncated_mean = -999.;
    gains[j].gas_gain_correction = -999.;
  }

  FILE *fin = fopen(filename.data(), "r");

  int linecounter = 0;  // set the line counter to the first serial number in the file....

  while (!feof(fin)) {
    // note space at end of format string to convert last \n
    int check = fscanf(fin,
                       "%d %d %d %d %d %d %d %d %d %f %f %f \n",
                       &gains[linecounter].gas_gain_index,
                       &gains[linecounter].endcap,
                       &gains[linecounter].station,
                       &gains[linecounter].ring,
                       &gains[linecounter].chamber,
                       &gains[linecounter].layer,
                       &gains[linecounter].hvsegment,
                       &gains[linecounter].cfeb,
                       &gains[linecounter].nentries,
                       &gains[linecounter].mean,
                       &gains[linecounter].truncated_mean,
                       &gains[linecounter].gas_gain_correction);

    if (check != 12) {
      printf("The input file format is not as expected\n");
      assert(0);
    }

    linecounter++;
  }

  fclose(fin);

  if (linecounter == MAX_SIZE) {
    std::cout << "Total number of gas gains read in = " << linecounter << std::endl;
  } else {
    std::cout << "ERROR:  Total number of gas-gains read in = " << linecounter
              << " while total number expected = " << MAX_SIZE << std::endl;
  }

  // Fill the chip corrections with values from the file
  for (int i = 0; i < MAX_SIZE; i++) {
    itemvector[i].gainCorr = 0.;

    if (gains[i].gas_gain_correction > 0.) {
      itemvector[i].gainCorr = gains[i].gas_gain_correction;
    } else {
      // if there is no value, this should be fixed...
      std::cout << "ERROR:  gas_gain_correction < 0 for index " << gains[i].gas_gain_index << std::endl;
    }
  }

  return cndbGasGainCorr;
}

#endif