GMTCollections.h

Go to the documentation of this file.

#ifndef GMTCollections_h
#define GMTCollections_h

#include "DataFormats/L1TMuon/interface/RegionalMuonCand.h"
#include "DataFormats/L1Trigger/interface/Muon.h"

#include "L1TObjectCollections.h"

#include <array>

namespace l1t {
   namespace stage2 {
      class GMTCollections : public L1TObjectCollections {
         public:
            // If the zero suppression deletes all the blocks used to
            // fill a collection the BX range cannot be determined.
            // Set default values here to then create an empty collection
            // with a defined BX range.
            GMTCollections(edm::Event& e, const int iFirstBx=-2, const int iLastBx=2, const int oFirstBx=-2, const int oLastBx=2) :
               L1TObjectCollections(e),
               regionalMuonCandsBMTF_(std::make_unique<RegionalMuonCandBxCollection>(0, iFirstBx, iLastBx)),
               regionalMuonCandsOMTF_(std::make_unique<RegionalMuonCandBxCollection>(0, iFirstBx, iLastBx)),
               regionalMuonCandsEMTF_(std::make_unique<RegionalMuonCandBxCollection>(0, iFirstBx, iLastBx)),
               muons_(),
               imdMuonsBMTF_(std::make_unique<MuonBxCollection>(0, oFirstBx, oLastBx)),
               imdMuonsEMTFNeg_(std::make_unique<MuonBxCollection>(0, oFirstBx, oLastBx)),
               imdMuonsEMTFPos_(std::make_unique<MuonBxCollection>(0, oFirstBx, oLastBx)),
               imdMuonsOMTFNeg_(std::make_unique<MuonBxCollection>(0, oFirstBx, oLastBx)),
               imdMuonsOMTFPos_(std::make_unique<MuonBxCollection>(0, oFirstBx, oLastBx))
            {
               std::generate(muons_.begin(), muons_.end(), [&oFirstBx, &oLastBx]{ return std::make_unique<MuonBxCollection>(0, oFirstBx, oLastBx); });
            };

            virtual ~GMTCollections();

            inline RegionalMuonCandBxCollection* getRegionalMuonCandsBMTF() { return regionalMuonCandsBMTF_.get(); };
            inline RegionalMuonCandBxCollection* getRegionalMuonCandsOMTF() { return regionalMuonCandsOMTF_.get(); };
            inline RegionalMuonCandBxCollection* getRegionalMuonCandsEMTF() { return regionalMuonCandsEMTF_.get(); };
            inline MuonBxCollection* getMuons(const unsigned int copy) override { return muons_[copy].get(); };
            inline MuonBxCollection* getImdMuonsBMTF() { return imdMuonsBMTF_.get(); };
            inline MuonBxCollection* getImdMuonsEMTFNeg() { return imdMuonsEMTFNeg_.get(); };
            inline MuonBxCollection* getImdMuonsEMTFPos() { return imdMuonsEMTFPos_.get(); };
            inline MuonBxCollection* getImdMuonsOMTFNeg() { return imdMuonsOMTFNeg_.get(); };
            inline MuonBxCollection* getImdMuonsOMTFPos() { return imdMuonsOMTFPos_.get(); };

         private:
            std::unique_ptr<RegionalMuonCandBxCollection> regionalMuonCandsBMTF_;
            std::unique_ptr<RegionalMuonCandBxCollection> regionalMuonCandsOMTF_;
            std::unique_ptr<RegionalMuonCandBxCollection> regionalMuonCandsEMTF_;
            std::array<std::unique_ptr<MuonBxCollection>, 6> muons_;
            std::unique_ptr<MuonBxCollection> imdMuonsBMTF_;
            std::unique_ptr<MuonBxCollection> imdMuonsEMTFNeg_;
            std::unique_ptr<MuonBxCollection> imdMuonsEMTFPos_;
            std::unique_ptr<MuonBxCollection> imdMuonsOMTFNeg_;
            std::unique_ptr<MuonBxCollection> imdMuonsOMTFPos_;
      };
   }
}

#endif