add trajectory length observer feature

CRPropa · Aug 7, 2024 · 19be5ff · 19be5ff
1 parent fe31aca
commit 19be5ff
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/include/crpropa/module/Observer.h b/include/crpropa/module/Observer.h
@@ -281,6 +281,17 @@ class ObserverTimeEvolution: public ObserverFeature {
 	DetectionState checkDetection(Candidate *candidate) const;
 	std::string getDescription() const;
 };
+
+class ObserverTrajectoryLength: public ObserverFeature {
+private: 
+	double maxLength; 
+public: 
+	ObserverTrajectoryLength(double l);
+
+	DetectionState checkDetection(Candidate *candidate) const;
+};
+
+
 /** @} */
 
 }

diff --git a/src/module/Observer.cpp b/src/module/Observer.cpp
@@ -365,4 +365,17 @@ std::string ObserverSurface::getDescription() const {
 	return ss.str();
 }
 
+// ObserverTrajectoryLength ---------------------------------------------------
+
+ObserverTrajectoryLength::ObserverTrajectoryLength(double l) : maxLength(l) { }
+
+DetectionState ObserverTrajectoryLength::checkDetection(Candidate *cand) const {
+	double currentLength = cand -> getTrajectoryLength(); 
+
+	if (currentLength > maxLength) 
+		return DETECTED;
+
+	cand -> limitNextStep(maxLength - currentLength); 
+	return NOTHING;
+}
 } // namespace crpropa