CondFormats/SiPixelObjects/interface/SiPixelTemplateDBObject.h - master - githubmirror/cmssw

CondFormats/SiPixelObjects/interface/SiPixelTemplateDBObject.h
151 строка5 KB
Cms Build
Clang-Format
11 июн 2019, 11:29
11 июн 2019, 11:29e9d3fcc
#ifndef CondFormats_SiPixelObjects_SiPixelTemplateDBObject_h
#define CondFormats_SiPixelObjects_SiPixelTemplateDBObject_h 1

#include "CondFormats/Serialization/interface/Serializable.h"

#include <vector>
#include <map>
#include <cstdint>
#include "FWCore/MessageLogger/interface/MessageLogger.h"

// ******************************************************************************************
//! \class SiPixelTemplateDBObject
//!
// ******************************************************************************************

class SiPixelTemplateDBObject {
public:
  struct Reader {
    explicit Reader(SiPixelTemplateDBObject const& idb) : index_(0), isInvalid_(false), db(idb) {}

    //- Fills integer from dbobject
    Reader& operator>>(int& i) {
      isInvalid_ = false;
      if (index_ <= db.maxIndex()) {
        i = (int)db.sVector()[index_];
        index_++;
      } else
        setInvalid();
      return *this;
    }
    //- Fills float from dbobject
    Reader& operator>>(float& f) {
      isInvalid_ = false;
      if (index_ <= db.maxIndex()) {
        f = db.sVector()[index_];
        index_++;
      } else
        setInvalid();
      return *this;
    }

    //- Functions to monitor integrity of dbobject
    void setInvalid() { isInvalid_ = true; }
    bool fail() { return isInvalid_; }
    int index() const { return index_; }
    int numOfTempl() const { return db.numOfTempl(); }
    float version() const { return db.version(); }
    std::vector<float> const& sVector() const { return db.sVector(); }

    //- Able to set the index for template header
    void incrementIndex(int i) { index_ += i; }

    int index_;
    bool isInvalid_;
    SiPixelTemplateDBObject const& db;
  };

  Reader reader() const { return Reader(*this); }

  SiPixelTemplateDBObject() : index_(0), maxIndex_(0), numOfTempl_(1), version_(-99.9), isInvalid_(false), sVector_(0) {
    sVector_.reserve(1000000);
  }
  virtual ~SiPixelTemplateDBObject() {}

  //- Allows the dbobject to be read out like cout
  friend std::ostream& operator<<(std::ostream& s, const SiPixelTemplateDBObject& dbobject);

  //- Fills integer from dbobject
  SiPixelTemplateDBObject& operator>>(int& i) {
    isInvalid_ = false;
    if (index_ <= maxIndex_) {
      i = (int)(*this).sVector_[index_];
      index_++;
    } else
      (*this).setInvalid();
    return *this;
  }
  //- Fills float from dbobject
  SiPixelTemplateDBObject& operator>>(float& f) {
    isInvalid_ = false;
    if (index_ <= maxIndex_) {
      f = (*this).sVector_[index_];
      index_++;
    } else
      (*this).setInvalid();
    return *this;
  }

  //- Functions to monitor integrity of dbobject
  void setVersion(float version) { version_ = version; }
  void setInvalid() { isInvalid_ = true; }
  bool fail() { return isInvalid_; }

  //- Setter functions
  void push_back(float entry) { sVector_.push_back(entry); }
  void setIndex(int index) { index_ = index; }
  void setMaxIndex(int maxIndex) { maxIndex_ = maxIndex; }
  void setNumOfTempl(int numOfTempl) { numOfTempl_ = numOfTempl; }

  //- Accessor functions
  int index() const { return index_; }
  int maxIndex() const { return maxIndex_; }
  int numOfTempl() const { return numOfTempl_; }
  float version() const { return version_; }
  std::vector<float> const& sVector() const { return sVector_; }

  //- Able to set the index for template header
  void incrementIndex(int i) { index_ += i; }

  //- Allows storage of header (type = char[80]) in dbobject
  union char2float {
    char c[4];
    float f;
  };

  //- To be used to select template calibration based on detid
  void putTemplateIDs(std::map<unsigned int, short>& t_ID) { templ_ID = t_ID; }
  const std::map<unsigned int, short>& getTemplateIDs() const { return templ_ID; }

  bool putTemplateID(const uint32_t& detid, short& value) {
    std::map<unsigned int, short>::const_iterator id = templ_ID.find(detid);
    if (id != templ_ID.end()) {
      edm::LogError("SiPixelTemplateDBObject")
          << "Template ID for DetID " << detid << " is already stored. Skipping this put" << std::endl;
      return false;
    } else
      templ_ID[detid] = value;
    return true;
  }

  short getTemplateID(const uint32_t& detid) const {
    std::map<unsigned int, short>::const_iterator id = templ_ID.find(detid);
    if (id != templ_ID.end())
      return id->second;
    else
      edm::LogError("SiPixelTemplateDBObject") << "Template ID for DetID " << detid << " is not stored" << std::endl;
    return 0;
  }

private:
  int index_;
  int maxIndex_;
  int numOfTempl_;
  float version_;
  bool isInvalid_;
  std::vector<float> sVector_;
  std::map<unsigned int, short> templ_ID;

  COND_SERIALIZABLE;
};  //end SiPixelTemplateDBObject
#endif