Feature/cyclotron freq recon

Source/Data/CMakeLists.txt

@@ -30,6 +30,7 @@ set (DATA_HEADERFILES
     EventAnalysis/KTMultiTrackEventData.hh
     EventAnalysis/KTPowerFitData.hh
     EventAnalysis/KTProcessedMPTData.hh
+    EventAnalysis/KTProcessedCavityEventData.hh
     EventAnalysis/KTProcessedTrackData.hh
     EventAnalysis/KTRPTrackData.hh
     EventAnalysis/KTSequentialLineData.hh
@@ -96,6 +97,7 @@ set (DATA_SOURCEFILES
     EventAnalysis/KTMultiTrackEventData.cc
     EventAnalysis/KTPowerFitData.cc
     EventAnalysis/KTProcessedMPTData.cc
+    EventAnalysis/KTProcessedCavityEventData.cc
     EventAnalysis/KTProcessedTrackData.cc
     EventAnalysis/KTRPTrackData.cc
     EventAnalysis/KTSequentialLineData.cc

Source/Data/EventAnalysis/KTProcessedCavityEventData.cc

@@ -0,0 +1,81 @@
+/*
+ * KTProcessedCavityEventData.cc
+ *
+ *  Created on: Sep 11, 2025
+ *      Author: juniorpe
+ */
+
+#include "KTProcessedCavityEventData.hh"
+
+#include "KTLogger.hh"
+
+namespace Katydid
+{
+
+    const std::string KTProcessedCavityEventData::sName("proc-cavity-event");
+
+    KTProcessedCavityEventData::KTProcessedCavityEventData() :
+            KTExtensibleData< KTProcessedCavityEventData >(),
+            fComponent(0),
+            fAcquisitionID(0.),
+            fEventID(0),
+            fTotalEventSequences(0),
+            fFirstTrackStartCyclotronFrequency(0.),
+            fFirstTrackAxialFrequency(0.),
+            fFirstTrackBandIDs(),
+            fFirstTrackBandClassifications(),
+            fMPTEventSequenceID(),
+            fMPTStartCyclotronFrequency(),
+            fMPTAxialFrequency(),
+            fAllTrackIDs(),
+            fAllTrackEventSequenceIDs(),
+            fAllTrackBandClassifications()
+
+    {
+    }
+
+    KTProcessedCavityEventData::KTProcessedCavityEventData(const KTProcessedCavityEventData& orig) :
+            KTExtensibleData< KTProcessedCavityEventData >(orig),
+
+            fComponent(orig.fComponent),
+            fAcquisitionID(orig.fAcquisitionID),
+            fEventID(orig.fEventID),
+            fTotalEventSequences(orig.fTotalEventSequences),
+            fFirstTrackStartCyclotronFrequency(orig.fFirstTrackStartCyclotronFrequency),
+            fFirstTrackAxialFrequency(orig.fFirstTrackAxialFrequency),
+            fFirstTrackBandIDs(orig.fFirstTrackBandIDs),
+            fFirstTrackBandClassifications(orig.fFirstTrackBandClassifications),
+            fMPTEventSequenceID(orig.fMPTEventSequenceID),
+            fMPTStartCyclotronFrequency(orig.fMPTStartCyclotronFrequency),
+            fMPTAxialFrequency(orig.fMPTAxialFrequency),
+            fAllTrackIDs(orig.fAllTrackIDs),
+            fAllTrackEventSequenceIDs(orig.fAllTrackEventSequenceIDs),
+            fAllTrackBandClassifications(orig.fAllTrackBandClassifications)
+    {
+    }
+
+    KTProcessedCavityEventData::~KTProcessedCavityEventData()
+    {
+    }
+
+    KTProcessedCavityEventData& KTProcessedCavityEventData::operator=(const KTProcessedCavityEventData& rhs)
+    {
+        KTExtensibleData< KTProcessedCavityEventData >::operator=(rhs);
+        fComponent = rhs.fComponent;
+        fAcquisitionID = rhs.fAcquisitionID;
+        fEventID = rhs.fEventID;
+        fTotalEventSequences = rhs.fTotalEventSequences;
+        fFirstTrackStartCyclotronFrequency = rhs.fFirstTrackStartCyclotronFrequency;
+        fFirstTrackAxialFrequency = rhs.fFirstTrackAxialFrequency;
+        fFirstTrackBandIDs = rhs.fFirstTrackBandIDs;
+        fFirstTrackBandClassifications = rhs.fFirstTrackBandClassifications;
+        fMPTEventSequenceID = rhs.fMPTEventSequenceID;
+        fMPTStartCyclotronFrequency = rhs.fMPTStartCyclotronFrequency;
+        fMPTAxialFrequency = rhs.fMPTAxialFrequency;
+        fAllTrackIDs = rhs.fAllTrackIDs;
+        fAllTrackEventSequenceIDs = rhs.fAllTrackEventSequenceIDs;
+        fAllTrackBandClassifications = rhs.fAllTrackBandClassifications;
+        return *this;
+    }
+
+}

Source/Data/EventAnalysis/KTProcessedCavityEventData.hh

@@ -0,0 +1,112 @@
+/*
+ * KTProcessedCavityEventData.hh
+ *
+ *  Created on: Sep 11, 2025
+ *      Author: juniorpe
+ */
+
+#ifndef KTPROCESSEDCAVITYEVENTDATA_HH_
+#define KTPROCESSEDCAVITYEVENTDATA_HH_
+
+#include "KTData.hh"
+#include "KTMultiTrackEventData.hh"
+#include "KTMemberVariable.hh"
+
+#include <vector>
+#include <set>
+
+namespace Katydid
+{
+
+    class KTProcessedCavityEventData : public Nymph::KTExtensibleData< KTProcessedCavityEventData >
+    {
+        public:
+            KTProcessedCavityEventData();
+            KTProcessedCavityEventData(const KTProcessedCavityEventData& orig);
+            virtual ~KTProcessedCavityEventData();
+
+            KTProcessedCavityEventData& operator=(const KTProcessedCavityEventData& rhs);
+
+            MEMBERVARIABLE(unsigned, Component);
+            MEMBERVARIABLE(uint64_t, AcquisitionID);
+            MEMBERVARIABLE(unsigned, EventID);
+            MEMBERVARIABLE(unsigned, TotalEventSequences);
+
+            MEMBERVARIABLE(double, FirstTrackStartCyclotronFrequency);
+            MEMBERVARIABLE(double, FirstTrackAxialFrequency);
+            MEMBERVARIABLE(std::vector<int>, FirstTrackBandIDs);
+            MEMBERVARIABLE(std::vector<int>, FirstTrackBandClassifications);
+
+            MEMBERVARIABLEREF(std::vector<int>, MPTEventSequenceID);
+            MEMBERVARIABLEREF(std::vector<double>, MPTStartCyclotronFrequency);
+            MEMBERVARIABLEREF(std::vector<double>, MPTAxialFrequency);
+
+            MEMBERVARIABLEREF(std::vector<int>, AllTrackIDs);
+            MEMBERVARIABLEREF(std::vector<int>, AllTrackEventSequenceIDs);
+            MEMBERVARIABLEREF(std::vector<int>, AllTrackBandClassifications);
+
+            void ClearProcessedEvent();
+
+            void AddFirstTrackBandInfo(int trackID, int classification);
+            void ClearFirstTrackBandInfo();
+
+            void AddProcessedMPT(int seqID, double startCyclFreq, double axialFreq);
+
+            void AddClassificationData(int trackID, int seqID, int bandClassification);
+            void AddClassificationData(const std::vector<AllTrackData>& tracksInMPT, int seqID, int bandClassification);
+
+        public:
+            static const std::string sName;
+    };
+
+    inline void KTProcessedCavityEventData::ClearProcessedEvent()
+    {
+        fFirstTrackBandIDs.clear();
+        fFirstTrackBandClassifications.clear();
+
+        fMPTEventSequenceID.clear();
+        fMPTStartCyclotronFrequency.clear();
+        fMPTAxialFrequency.clear();
+
+        fAllTrackIDs.clear();
+        fAllTrackEventSequenceIDs.clear();
+        fAllTrackBandClassifications.clear();
+    }
+
+    inline void KTProcessedCavityEventData::AddFirstTrackBandInfo(int trackID, int classification)
+    {
+        fFirstTrackBandIDs.push_back(trackID);
+        fFirstTrackBandClassifications.push_back(classification);
+    }
+
+    inline void KTProcessedCavityEventData::ClearFirstTrackBandInfo()
+    {
+        fFirstTrackBandIDs.clear();
+        fFirstTrackBandClassifications.clear();
+    }
+
+    inline void KTProcessedCavityEventData::AddProcessedMPT(int seqID, double startCyclFreq, double axialFreq)
+    {
+        fMPTEventSequenceID.push_back(seqID);
+        fMPTStartCyclotronFrequency.push_back(startCyclFreq);
+        fMPTAxialFrequency.push_back(axialFreq);
+    }
+
+    inline void KTProcessedCavityEventData::AddClassificationData(int trackID, int seqID, int bandClassification)
+    {
+        fAllTrackIDs.push_back(trackID);
+        fAllTrackEventSequenceIDs.push_back(seqID);
+        fAllTrackBandClassifications.push_back(bandClassification);
+    }
+
+    inline void KTProcessedCavityEventData::AddClassificationData(const std::vector<AllTrackData>& tracksInMPT, int seqID, int bandClassification)
+    {
+        for (const auto& trk : tracksInMPT)
+        {
+            AddClassificationData(trk.fProcTrack.GetTrackID(), seqID, bandClassification);
+        }
+    }
+
+}
+#endif
+

Source/EventAnalysis/CMakeLists.txt

@@ -34,6 +34,7 @@ set (EVENTANALYSIS_HEADERFILES
     KTIterativeTrackClustering.hh
     KTMultiPeakEventBuilder.hh
     KTMultiPeakTrackProcessing.hh
+    KTCavityEventProcessing.hh
     KTMultiPeakTrackBuilder.hh
     KTMultiSliceClustering.hh
     KTOverlappingTrackClustering.hh
@@ -73,6 +74,7 @@ set (EVENTANALYSIS_SOURCEFILES
     KTIterativeTrackClustering.cc
     KTMultiPeakEventBuilder.cc
     KTMultiPeakTrackProcessing.cc
+    KTCavityEventProcessing.cc
     KTMultiPeakTrackBuilder.cc
     KTMultiSliceClustering.cc
     KTOverlappingTrackClustering.cc