LinearSortingPolicy: ignore unavailable extension error and adapt tre…

…atment for it
key4hep · Feb 16, 2024 · 9889d01 · 9889d01
1 parent 36bd6ba
commit 9889d01
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/plugins/LinearSortingPolicy.cpp b/plugins/LinearSortingPolicy.cpp
@@ -91,9 +91,13 @@ namespace {
 
           dd4hep::rec::DoubleParameters* para = nullptr;
           try { // use existing map, or create a new one
-            para = ddsurf->detElement().extension<dd4hep::rec::DoubleParameters>();
+            para = ddsurf->detElement().extension<dd4hep::rec::DoubleParameters>(false);
+            if(not para) {
+              para = new dd4hep::rec::DoubleParameters;
+              ddsurf->detElement().addExtension<dd4hep::rec::DoubleParameters>(para);
+            }
             para->doubleParameters["SortingPolicy"] = rValue;
-          } catch(...){
+          } catch(...) {
             para = new dd4hep::rec::DoubleParameters;
             para->doubleParameters["SortingPolicy"] = rValue;
             ddsurf->detElement().addExtension<dd4hep::rec::DoubleParameters>(para);