GMTSetup.cc

Go to the documentation of this file.

#include "FWCore/Framework/interface/stream/EDProducerBase.h"
#include "FWCore/Utilities/interface/InputTag.h"

#include "EventFilter/L1TRawToDigi/plugins/PackerFactory.h"
#include "EventFilter/L1TRawToDigi/plugins/PackingSetupFactory.h"
#include "EventFilter/L1TRawToDigi/plugins/UnpackerFactory.h"

#include "EventFilter/L1TRawToDigi/plugins/implementations_stage2/MuonUnpacker.h"
#include "EventFilter/L1TRawToDigi/plugins/implementations_stage2/IntermediateMuonUnpacker.h"

#include "GMTSetup.h"

#include <array>
#include <string>

namespace l1t {
   namespace stage2 {
      std::unique_ptr<PackerTokens>
      GMTSetup::registerConsumes(const edm::ParameterSet& cfg, edm::ConsumesCollector& cc)
      {
         return std::unique_ptr<PackerTokens>(new GMTTokens(cfg, cc));
      }

      void
      GMTSetup::fillDescription(edm::ParameterSetDescription& desc)
      {
         desc.addOptional<edm::InputTag>("BMTFInputLabel")->setComment("for stage2");
         desc.addOptional<edm::InputTag>("OMTFInputLabel")->setComment("for stage2");
         desc.addOptional<edm::InputTag>("EMTFInputLabel")->setComment("for stage2");
         desc.addOptional<edm::InputTag>("ImdInputLabelBMTF")->setComment("uGMT intermediate muon from BMTF after first sorting stage");
         desc.addOptional<edm::InputTag>("ImdInputLabelEMTFNeg")->setComment("uGMT intermediate muon from neg. EMTF side after first sorting stage");
         desc.addOptional<edm::InputTag>("ImdInputLabelEMTFPos")->setComment("uGMT intermediate muon from pos. EMTF side after first sorting stage");
         desc.addOptional<edm::InputTag>("ImdInputLabelOMTFNeg")->setComment("uGMT intermediate muon from neg. OMTF side after first sorting stage");
         desc.addOptional<edm::InputTag>("ImdInputLabelOMTFPos")->setComment("uGMT intermediate muon from pos. OMTF side after first sorting stage");
      }

      PackerMap
      GMTSetup::getPackers(int fed, unsigned int fw)
      {
         PackerMap res;
         if (fed == 1402) {
            // Use amc_no and board id 1 for packing
            res[{1, 1}] = {
               PackerFactory::get()->make("stage2::RegionalMuonGMTPacker"),
               PackerFactory::get()->make("stage2::GMTMuonPacker"),
               PackerFactory::get()->make("stage2::IntermediateMuonPacker"),
            };
         }
         return res;
      }

      void
      GMTSetup::registerProducts(edm::stream::EDProducerBase& prod)
      {
         prod.produces<RegionalMuonCandBxCollection>("BMTF");
         prod.produces<RegionalMuonCandBxCollection>("OMTF");
         prod.produces<RegionalMuonCandBxCollection>("EMTF");
         prod.produces<MuonBxCollection>("Muon");
         for (int i=1; i<6; ++i) {
            prod.produces<MuonBxCollection>(("MuonCopy"+std::to_string(i)).c_str());
         }
         prod.produces<MuonBxCollection>("imdMuonsBMTF");
         prod.produces<MuonBxCollection>("imdMuonsEMTFNeg");
         prod.produces<MuonBxCollection>("imdMuonsEMTFPos");
         prod.produces<MuonBxCollection>("imdMuonsOMTFNeg");
         prod.produces<MuonBxCollection>("imdMuonsOMTFPos");
      }

      std::unique_ptr<UnpackerCollections>
      GMTSetup::getCollections(edm::Event& e)
      {
         return std::unique_ptr<UnpackerCollections>(new GMTCollections(e));
      }

      UnpackerMap
      GMTSetup::getUnpackers(int fed, int board, int amc, unsigned int fw)
      {
         UnpackerMap res;

         // input muons
         auto gmt_in_unp = UnpackerFactory::get()->make("stage2::RegionalMuonGMTUnpacker");
         for (int iLink = 72; iLink < 144; iLink += 2)
            res[iLink] = gmt_in_unp;

         // internal muons
         auto gmt_imd_unp = static_pointer_cast<l1t::stage2::IntermediateMuonUnpacker>(UnpackerFactory::get()->make("stage2::IntermediateMuonUnpacker"));
         gmt_imd_unp->setAlgoVersion(fw);
         for (int oLink = 49; oLink < 63; oLink += 2)
            res[oLink] = gmt_imd_unp;

         // output muons (6 copies)
         std::array<std::shared_ptr<l1t::stage2::MuonUnpacker>, 6> gmt_out_unps;
         int i = 0;
         for (auto gmt_out_unp:gmt_out_unps) {
            gmt_out_unp = static_pointer_cast<l1t::stage2::MuonUnpacker>(UnpackerFactory::get()->make("stage2::MuonUnpacker"));
            gmt_out_unp->setAlgoVersion(fw);
            gmt_out_unp->setFedNumber(fed);
            gmt_out_unp->setMuonCopy(i);

            int oLinkMin = i*8+1;
            for (int oLink = oLinkMin; oLink < oLinkMin+8; oLink += 2)
               res[oLink] = gmt_out_unp;

            ++i;
         }

         return res;
      }
   }
}

DEFINE_L1T_PACKING_SETUP(l1t::stage2::GMTSetup);