PhysCandPacker.cc

Go to the documentation of this file.

#include "FWCore/Framework/interface/Event.h"
#include "EventFilter/L1TRawToDigi/plugins/PackerFactory.h"

#include "CaloTokens.h"
#include "PhysCandPacker.h"

template<typename T, typename F>
l1t::Blocks
process(unsigned int id1, unsigned int id2, const BXVector<T>& coll, F filter)
{
   std::vector<uint32_t> load[2];

   for (int i = coll.getFirstBX(); i <= coll.getLastBX(); ++i) {
      uint16_t jetbit[4] = {0, 0, 0, 0};
      int n = 0;
      for (auto j =  coll.begin(i); j != coll.end(i) && n < 4; ++j) {
         if (!filter(*j))
            continue;
         //std::cout << j->hwPt() << " @ " << j->hwEta() << ", " << j->hwPhi() << " > " << j->hwQual() << " > " << j->hwIso() << std::endl;
         jetbit[n++] = std::min(j->hwPt(), 0x3F) |
                     (abs(j->hwEta()) & 0x7) << 6 |
                     ((j->hwEta() >> 3) & 0x1) << 9 |
                     (j->hwPhi() & 0x1F) << 10;
      }
      uint32_t word0=(jetbit[0] & 0xFFFF);
      uint32_t word1=(jetbit[1] & 0xFFFF);
      uint32_t word2=(jetbit[2] & 0xFFFF);
      uint32_t word3=(jetbit[3] & 0xFFFF);

      load[0].push_back(word0);
      load[0].push_back(word2);

      load[1].push_back(word1);
      load[1].push_back(word3);

   }

   return {l1t::Block(id1, load[0]),l1t::Block(id2, load[1])};
}

namespace l1t {
  namespace stage1 {
    Blocks
    IsoEGammaPacker::pack(const edm::Event& event, const PackerTokens* toks)
    {
       edm::Handle<EGammaBxCollection> egammas;
       event.getByToken(static_cast<const CaloTokens*>(toks)->getEGammaToken(), egammas);

       return process(85,87, *egammas, [](const l1t::EGamma& eg) -> bool { return eg.hwIso() == 1; });
    }

    Blocks
    NonIsoEGammaPacker::pack(const edm::Event& event, const PackerTokens* toks)
    {
       edm::Handle<EGammaBxCollection> egammas;
       event.getByToken(static_cast<const CaloTokens*>(toks)->getEGammaToken(), egammas);

       return process(89,91, *egammas, [](const l1t::EGamma& eg) -> bool { return eg.hwIso() == 0; });
    }

    Blocks
    CentralJetPacker::pack(const edm::Event& event, const PackerTokens* toks)
    {
       edm::Handle<JetBxCollection> jets;
       event.getByToken(static_cast<const CaloTokens*>(toks)->getJetToken(), jets);

       return process(77,79, *jets, [](const l1t::Jet& jet) -> bool { return !(jet.hwQual() & 2); });
    }

    Blocks
    ForwardJetPacker::pack(const edm::Event& event, const PackerTokens* toks)
    {
       edm::Handle<JetBxCollection> jets;
       event.getByToken(static_cast<const CaloTokens*>(toks)->getJetToken(), jets);

       return process(81,83, *jets, [](const l1t::Jet& jet) -> bool { return jet.hwQual() & 2; });
    }

    Blocks
    TauPacker::pack(const edm::Event& event, const PackerTokens* toks)
    {
       edm::Handle<TauBxCollection> taus;
       event.getByToken(static_cast<const CaloTokens*>(toks)->getTauToken(), taus);

       return process(101,103, *taus, [](const l1t::Tau& tau) -> bool { return true; });
    }

    Blocks
    IsoTauPacker::pack(const edm::Event& event, const PackerTokens* toks)
    {
       edm::Handle<TauBxCollection> taus;
       event.getByToken(static_cast<const CaloTokens*>(toks)->getIsoTauToken(), taus);

       return process(105,107, *taus, [](const l1t::Tau& tau) -> bool { return true; });
    }
  }
}

DEFINE_L1T_PACKER(l1t::stage1::IsoEGammaPacker);
DEFINE_L1T_PACKER(l1t::stage1::NonIsoEGammaPacker);
DEFINE_L1T_PACKER(l1t::stage1::CentralJetPacker);
DEFINE_L1T_PACKER(l1t::stage1::ForwardJetPacker);
DEFINE_L1T_PACKER(l1t::stage1::TauPacker);
DEFINE_L1T_PACKER(l1t::stage1::IsoTauPacker);