From f1b11ebfaba8df276ce60cb9abbf4fc54a5d2322 Mon Sep 17 00:00:00 2001
From: Emery Nibigira <emery.nibigira@cern.ch>
Date: Mon, 2 Dec 2024 10:50:54 +0100
Subject: [PATCH] added new types for miniaod

---
 .../SUSYBSMObjects/interface/HSCParticle.h       | 16 ++++++++++++++++
 .../SUSYBSMObjects/src/classes_def.xml           |  4 ++--
 2 files changed, 18 insertions(+), 2 deletions(-)

diff --git a/AnalysisDataFormats/SUSYBSMObjects/interface/HSCParticle.h b/AnalysisDataFormats/SUSYBSMObjects/interface/HSCParticle.h
index df867db33a4e5..6605efa2cf6a0 100644
--- a/AnalysisDataFormats/SUSYBSMObjects/interface/HSCParticle.h
+++ b/AnalysisDataFormats/SUSYBSMObjects/interface/HSCParticle.h
@@ -10,6 +10,11 @@
 #include "DataFormats/GeometryVector/interface/GlobalPoint.h"
 #include <vector>
 
+#include "DataFormats/PatCandidates/interface/Muon.h"
+#include "DataFormats/PatCandidates/interface/PackedCandidate.h"
+#include "DataFormats/PatCandidates/interface/IsolatedTrack.h"
+#include "DataFormats/TrackReco/interface/DeDxHitInfo.h"
+
 #include "DataFormats/MuonReco/interface/MuonTimeExtra.h"
 #include "AnalysisDataFormats/SUSYBSMObjects/interface/HSCPCaloInfo.h"
 
@@ -45,6 +50,8 @@ namespace susybsm {
     HSCParticle() {}
 
     // check available infos
+    bool hasTrack() const { return track_.packedCandRef().isNonnull(); }
+    bool hasMuon() const { return muon_.isNonnull(); }
     bool hasMuonRef() const { return muonRef_.isNonnull(); }
     bool hasMTMuonRef() const { return MTMuonRef_.isNonnull(); }
     bool hasTrackRef() const { return trackRef_.isNonnull(); }
@@ -53,6 +60,9 @@ namespace susybsm {
     bool hasCaloInfo() const { return caloInfoRef_.isNonnull(); }
 
     // set infos
+    void setDeDxHitInfo(const reco::DeDxHitInfo* data) {dedxHitInfo_ = data;}
+    void setTrack(const pat::IsolatedTrack& data) { track_ = data; }
+    void setMuon(const pat::MuonRef& data) { muon_ = data; }
     void setMuon(const reco::MuonRef& data) { muonRef_ = data; }
     void setMTMuon(const reco::MuonRef& data) { MTMuonRef_ = data; }
     void setTrack(const reco::TrackRef& data) { trackRef_ = data; }
@@ -61,6 +71,9 @@ namespace susybsm {
     void setCaloInfo(const HSCPCaloInfoRef& data) { caloInfoRef_ = data; }
 
     // get infos
+    const reco::DeDxHitInfo* dedxHitInfo() const {return dedxHitInfo_; }
+    pat::IsolatedTrack track() const { return track_; }
+    pat::MuonRef muon() const { return muon_; }
     reco::TrackRef trackRef() const { return trackRef_; }
     reco::TrackRef trackIsoRef() const { return trackIsoRef_; }
     reco::MuonRef muonRef() const { return muonRef_; }
@@ -74,6 +87,9 @@ namespace susybsm {
     int type() const;
 
   private:
+    const reco::DeDxHitInfo* dedxHitInfo_ = nullptr;
+    pat::IsolatedTrack track_;
+    pat::MuonRef muon_;
     reco::TrackRef trackRef_;     //TrackRef from refitted track collection (dE/dx purposes)
     reco::TrackRef trackIsoRef_;  //TrackRef from general track collection (isolation purposes)
     reco::MuonRef muonRef_;
diff --git a/AnalysisDataFormats/SUSYBSMObjects/src/classes_def.xml b/AnalysisDataFormats/SUSYBSMObjects/src/classes_def.xml
index f0a73307deeb4..f1a41fe196ea8 100644
--- a/AnalysisDataFormats/SUSYBSMObjects/src/classes_def.xml
+++ b/AnalysisDataFormats/SUSYBSMObjects/src/classes_def.xml
@@ -1,6 +1,6 @@
 <lcgdict>
-   <class name="susybsm::HSCParticle" ClassVersion="11">
-    <version ClassVersion="11" checksum="2205741811"/>
+   <class name="susybsm::HSCParticle" ClassVersion="12">
+    <version ClassVersion="12" checksum="3951465682"/>
    </class>
    <class name="susybsm::RPCBetaMeasurement" ClassVersion="10">
     <version ClassVersion="10" checksum="2621605308"/>