From 097c01c1bb1270c21fb64171afe07bf85e39aa53 Mon Sep 17 00:00:00 2001
From: ralfulrich <ralf.ulrich@kit.edu>
Date: Sun, 14 Oct 2018 14:00:38 +0200
Subject: [PATCH] allow more specific types of processes

---
 Framework/Cascade/Cascade.h                   |   4 +-
 Framework/ProcessSequence/BaseProcess.h       |  11 +-
 Framework/ProcessSequence/CMakeLists.txt      |   2 +
 Framework/ProcessSequence/ContinuousProcess.h |  25 +++
 Framework/ProcessSequence/DiscreteProcess.h   |  45 +++++
 Framework/ProcessSequence/ProcessSequence.h   | 154 ++++++++++++++++--
 Framework/ProcessSequence/ProcessSignature.h  |  21 +++
 .../ProcessSequence/testProcessSequence.cc    |  82 +++++++++-
 8 files changed, 317 insertions(+), 27 deletions(-)
 create mode 100644 Framework/ProcessSequence/ContinuousProcess.h
 create mode 100644 Framework/ProcessSequence/DiscreteProcess.h
 create mode 100644 Framework/ProcessSequence/ProcessSignature.h

diff --git a/Framework/Cascade/Cascade.h b/Framework/Cascade/Cascade.h
index 50396b025..b2aa17216 100644
--- a/Framework/Cascade/Cascade.h
+++ b/Framework/Cascade/Cascade.h
@@ -62,7 +62,7 @@ namespace corsika::cascade {
         // DoCascadeEquations(); //
       }
     }
-    
+
     void Step(Particle& particle) {
       [[maybe_unused]] double nextStep = fProcesseList.MinStepLength(particle);
       // corsika::utls::ignore(nextStep);
@@ -80,7 +80,7 @@ namespace corsika::cascade {
         fProcesseList.DoDiscrete(particle, fStack);
       }
     }
-    
+
   private:
     ProcessList& fProcesseList;
     Stack& fStack;
diff --git a/Framework/ProcessSequence/BaseProcess.h b/Framework/ProcessSequence/BaseProcess.h
index 5f6012202..bd6f9c035 100644
--- a/Framework/ProcessSequence/BaseProcess.h
+++ b/Framework/ProcessSequence/BaseProcess.h
@@ -20,6 +20,15 @@ namespace corsika::process {
     const derived& GetRef() const { return static_cast<const derived&>(*this); }
   };
 
-}
+  template <typename T>
+  struct is_base {
+    static const bool value = false;
+  };
+  template <typename T>
+  struct is_base<BaseProcess<T>> {
+    static const bool value = true;
+  };
+
+} // namespace corsika::process
 
 #endif
diff --git a/Framework/ProcessSequence/CMakeLists.txt b/Framework/ProcessSequence/CMakeLists.txt
index f0e14f729..c24567821 100644
--- a/Framework/ProcessSequence/CMakeLists.txt
+++ b/Framework/ProcessSequence/CMakeLists.txt
@@ -11,6 +11,8 @@ set (
 set (
   CORSIKAprocesssequence_HEADERS
   BaseProcess.h
+  ContinuousProcess.h
+  DiscreteProcess.h
   ProcessSequence.h
   ProcessReturn.h
   )
diff --git a/Framework/ProcessSequence/ContinuousProcess.h b/Framework/ProcessSequence/ContinuousProcess.h
new file mode 100644
index 000000000..fa3b5fa5c
--- /dev/null
+++ b/Framework/ProcessSequence/ContinuousProcess.h
@@ -0,0 +1,25 @@
+#ifndef _include_corsika_continuousprocess_h_
+#define _include_corsika_continuousprocess_h_
+
+#include <corsika/process/ProcessReturn.h> // for convenience
+
+namespace corsika::process {
+
+  /**
+     \class ContinuousProcess
+
+     The structural base type of a process object in a
+     ProcessSequence. Both, the ProcessSequence and all its elements
+     are of type ContinuousProcess<T>
+
+   */
+
+  template <typename derived>
+  struct ContinuousProcess {
+    derived& GetRef() { return static_cast<derived&>(*this); }
+    const derived& GetRef() const { return static_cast<const derived&>(*this); }
+  };
+
+} // namespace corsika::process
+
+#endif
diff --git a/Framework/ProcessSequence/DiscreteProcess.h b/Framework/ProcessSequence/DiscreteProcess.h
new file mode 100644
index 000000000..2a1f17473
--- /dev/null
+++ b/Framework/ProcessSequence/DiscreteProcess.h
@@ -0,0 +1,45 @@
+#ifndef _include_corsika_discreteprocess_h_
+#define _include_corsika_discreteprocess_h_
+
+#include <corsika/process/ProcessReturn.h> // for convenience
+
+#include <iostream> // debug
+
+namespace corsika::process {
+
+  /**
+     \class DiscreteProcess
+
+     The structural base type of a process object in a
+     ProcessSequence. Both, the ProcessSequence and all its elements
+     are of type DiscreteProcess<T>
+
+   */
+
+  template <typename derived>
+  struct DiscreteProcess {
+
+    // DiscreteProcess() {
+    // static_assert(mustProvide<derived>::mustProvide, "");
+    //}
+
+    derived& GetRef() { return static_cast<derived&>(*this); }
+    const derived& GetRef() const { return static_cast<const derived&>(*this); }
+
+    // here starts the interface part
+    // -> enforce derived to implement DoDiscrete...
+    template <typename Particle, typename Stack>
+    inline EProcessReturn DoDiscrete(Particle&, Stack&) const; // {}
+
+    //  private:
+    template <typename D, typename T, typename S>
+    inline EProcessReturn DoContinuous(D& d, T&, S&) const {
+      std::cout << "yeah" << std::endl;
+      return EProcessReturn::eOk;
+    } // find out how to make this FINAL
+    // void DoContinuous;
+  };
+
+} // namespace corsika::process
+
+#endif
diff --git a/Framework/ProcessSequence/ProcessSequence.h b/Framework/ProcessSequence/ProcessSequence.h
index 9afe793f2..a6d51e4b5 100644
--- a/Framework/ProcessSequence/ProcessSequence.h
+++ b/Framework/ProcessSequence/ProcessSequence.h
@@ -13,14 +13,112 @@
 #define _include_ProcessSequence_h_
 
 #include <corsika/process/BaseProcess.h>
+#include <corsika/process/ContinuousProcess.h>
+#include <corsika/process/DiscreteProcess.h>
 #include <corsika/process/ProcessReturn.h>
 
-#include <cmath>
-#include <iostream>
-#include <typeinfo>
+//#include <type_traits> // still needed ?
 
 namespace corsika::process {
 
+  /* namespace detail { */
+
+  /*   /\* template<typename TT1, typename TT2, typename Type = void> *\/ */
+  /*   /\*   struct CallHello { *\/ */
+  /*   /\* 	static void Call(const TT1&, const TT2&) { *\/ */
+  /*   /\* 	  std::cout << "normal" << std::endl; *\/ */
+  /*   /\* 	} *\/ */
+  /*   /\*   }; *\/ */
+
+  /*   /\* template<typename TT1, typename TT2> *\/ */
+  /*   /\*   struct CallHello<TT1, TT2, typename
+   * std::enable_if<std::is_base_of<ContinuousProcess<TT2>, TT2>::value>::type> *\/ */
+  /*   /\*   { *\/ */
+  /*   /\* 	static void Call(const TT1&, const TT2&) { *\/ */
+  /*   /\* 	  std::cout << "special" << std::endl; *\/ */
+  /*   /\* 	}	 *\/ */
+  /*   /\*   }; *\/ */
+
+  /*   template<typename T1, typename T2, typename Particle, typename Trajectory, typename
+   * Stack> //, typename Type = void> */
+  /*     struct DoContinuous { */
+  /* 	static EProcessReturn Call(const T1& A, const T2& B, Particle& p, Trajectory& t,
+   * Stack& s) { */
+  /* 	  EProcessReturn ret = EProcessReturn::eOk; */
+  /* 	  if constexpr (!std::is_base_of<DiscreteProcess<T1>, T1>::value)  { */
+  /* 	      A.DoContinuous(p, t, s); */
+  /* 	    } */
+  /* 	  if constexpr (!std::is_base_of<DiscreteProcess<T2>, T2>::value)  { */
+  /* 	      B.DoContinuous(p, t, s); */
+  /* 	    } */
+  /* 	  return ret; */
+  /* 	} */
+  /*     }; */
+
+  /*   /\* */
+  /*   template<typename T1, typename T2, typename Particle, typename Trajectory, typename
+   * Stack> */
+  /*     struct DoContinuous<T1,T2,Particle,Trajectory,Stack, typename
+   * std::enable_if<std::is_base_of<DiscreteProcess<T1>, T1>::value>::type> { */
+  /* 	static EProcessReturn Call(const T1& A, const T2& B, Particle& p, Trajectory& t,
+   * Stack& s) { */
+  /* 	  EProcessReturn ret = EProcessReturn::eOk; */
+  /* 	  A.DoContinuous(p, t, s); */
+  /* 	  B.DoContinuous(p, t, s); */
+  /* 	  return ret; */
+  /* 	} */
+  /*     }; */
+
+  /*       template<typename T1, typename T2, typename Particle, typename Trajectory,
+   * typename Stack> */
+  /*     struct DoContinuous<T1,T2,Particle,Trajectory,Stack, typename
+   * std::enable_if<std::is_base_of<DiscreteProcess<T2>, T2>::value>::type> { */
+  /* 	static EProcessReturn Call(const T1& A, const T2&, Particle& p, Trajectory& t,
+   * Stack& s) { */
+  /* 	  EProcessReturn ret = EProcessReturn::eOk; */
+  /* 	  A.DoContinuous(p, t, s); */
+  /* 	  B.DoContinuous(p, t, s); */
+  /* 	  return ret; */
+  /* 	} */
+  /*     }; */
+  /*   *\/ */
+
+  /*   template<typename T1, typename T2, typename Particle, typename Stack>//, typename
+   * Type = void> */
+  /*     struct DoDiscrete { */
+  /* 	static EProcessReturn Call(const T1& A, const T2& B, Particle& p, Stack& s)  { */
+  /* 	  if constexpr (!std::is_base_of<ContinuousProcess<T1>, T1>::value) { */
+  /* 	      A.DoDiscrete(p, s); */
+  /* 	    } */
+  /* 	  if constexpr (!std::is_base_of<ContinuousProcess<T2>, T2>::value) { */
+  /* 	      B.DoDiscrete(p, s); */
+  /* 	    } */
+  /* 	  return EProcessReturn::eOk; */
+  /* 	} */
+  /*     }; */
+  /*   /\* */
+  /*   template<typename T1, typename T2, typename Particle, typename Stack> */
+  /*     struct DoDiscrete<T1,T2,Particle,Stack, typename
+   * std::enable_if<std::is_base_of<ContinuousProcess<T1>, T1>::value>::type> { */
+  /*     static EProcessReturn Call(const T1&, const T2& B, Particle& p, Stack& s) { */
+  /* 	// A.DoDiscrete(p, s); */
+  /*       B.DoDiscrete(p, s); */
+  /*       return EProcessReturn::eOk; */
+  /*     } */
+  /*   }; */
+
+  /*   template<typename T1, typename T2, typename Particle, typename Stack> */
+  /*     struct DoDiscrete<T1,T2,Particle,Stack, typename
+   * std::enable_if<std::is_base_of<ContinuousProcess<T2>, T2>::value>::type> { */
+  /*     static EProcessReturn Call(const T1& A, const T2&, Particle& p, Stack& s) { */
+  /* 	A.DoDiscrete(p, s); */
+  /*       //B.DoDiscrete(p, s); */
+  /*       return EProcessReturn::eOk; */
+  /*     } */
+  /*   }; */
+  /*   *\/ */
+  /* } // end namespace detail */
+
   /**
      \class ProcessSequence
 
@@ -42,13 +140,20 @@ namespace corsika::process {
         : A(in_A)
         , B(in_B) {}
 
+    // example for a trait-based call:
+    // void Hello() const  { detail::CallHello<T1,T2>::Call(A, B); }
+
     template <typename Particle, typename Trajectory, typename Stack>
     inline EProcessReturn DoContinuous(Particle& p, Trajectory& t, Stack& s) const {
       EProcessReturn ret = EProcessReturn::eOk;
-      /*ret |=*/A.DoContinuous(p, t, s);
-      /*ret |=*/B.DoContinuous(p, t, s);
+      if constexpr (!std::is_base_of<DiscreteProcess<T1>, T1>::value) {
+        A.DoContinuous(p, t, s);
+      }
+      if constexpr (!std::is_base_of<DiscreteProcess<T2>, T2>::value) {
+        B.DoContinuous(p, t, s);
+      }
       return ret;
-    } // add trajectory
+    }
 
     template <typename D>
     inline double MinStepLength(D& d) const {
@@ -64,11 +169,16 @@ namespace corsika::process {
           p, length); // need to do this also to decide which Trajectory to return!!!!
     }
     */
-    
+
     template <typename Particle, typename Stack>
-    void DoDiscrete(Particle& p, Stack& s) const {
-      A.DoDiscrete(p, s);
-      B.DoDiscrete(p, s);
+    inline EProcessReturn DoDiscrete(Particle& p, Stack& s) const {
+      if constexpr (!std::is_base_of<ContinuousProcess<T1>, T1>::value) {
+        A.DoDiscrete(p, s);
+      }
+      if constexpr (!std::is_base_of<ContinuousProcess<T2>, T2>::value) {
+        B.DoDiscrete(p, s);
+      }
+      return EProcessReturn::eOk;
     }
 
     /// TODO the const_cast is not nice, think about the constness here
@@ -78,13 +188,27 @@ namespace corsika::process {
     }
   };
 
-  /// the + operator that assembles more BaseProcess objects into a ProcessSequence
-  template <typename T1, typename T2>
-  inline const ProcessSequence<T1, T2> operator+(const BaseProcess<T1>& A,
-                                                 const BaseProcess<T2>& B) {
-    return ProcessSequence<T1, T2>(A.GetRef(), B.GetRef());
+  /// the +operator assembles many BaseProcess, ContinuousProcess, and
+  /// DiscreteProcess objects into a ProcessSequence, all combinatoris
+  /// must be allowed, this is why we define a macro to define all
+  /// combinations here:
+
+#define OPSEQ(C1, C2)                                                                \
+  template <typename T1, typename T2>                                                \
+  inline const ProcessSequence<T1, T2> operator+(const C1<T1>& A, const C2<T2>& B) { \
+    return ProcessSequence<T1, T2>(A.GetRef(), B.GetRef());                          \
   }
 
+  OPSEQ(BaseProcess, BaseProcess)
+  OPSEQ(BaseProcess, DiscreteProcess)
+  OPSEQ(BaseProcess, ContinuousProcess)
+  OPSEQ(ContinuousProcess, BaseProcess)
+  OPSEQ(ContinuousProcess, DiscreteProcess)
+  OPSEQ(ContinuousProcess, ContinuousProcess)
+  OPSEQ(DiscreteProcess, BaseProcess)
+  OPSEQ(DiscreteProcess, DiscreteProcess)
+  OPSEQ(DiscreteProcess, ContinuousProcess)
+
   /*
     template <typename T1>
     struct depth_lhs
diff --git a/Framework/ProcessSequence/ProcessSignature.h b/Framework/ProcessSequence/ProcessSignature.h
new file mode 100644
index 000000000..d9193750c
--- /dev/null
+++ b/Framework/ProcessSequence/ProcessSignature.h
@@ -0,0 +1,21 @@
+#ifndef _include_process_processsignature_h_
+#define _include_process_processsignature_h_
+
+#define FORCE_SIGNATURE(nameTrait, nameMethod, signatureMethod)                \
+  template <typename U>                                                        \
+  class nameTrait {                                                            \
+  private:                                                                     \
+    template <typename T, T>                                                   \
+    struct helper;                                                             \
+    template <typename T>                                                      \
+    static std::uint8_t check(helper<signatureMethod, &nameMethod>*);          \
+    template <typename T>                                                      \
+    static std::uint16_t check(...);                                           \
+                                                                               \
+  public:                                                                      \
+    static constexpr bool value = sizeof(check<U>(0)) == sizeof(std::uint8_t); \
+  }
+
+// FORCE_SIGNATURE(thisMustBeDefined, T::thisMustBeDefined, int(*)(void));
+
+#endif
diff --git a/Framework/ProcessSequence/testProcessSequence.cc b/Framework/ProcessSequence/testProcessSequence.cc
index b144230eb..2d4b46b15 100644
--- a/Framework/ProcessSequence/testProcessSequence.cc
+++ b/Framework/ProcessSequence/testProcessSequence.cc
@@ -22,43 +22,94 @@
 using namespace std;
 using namespace corsika::process;
 
-class Process1 : public BaseProcess<Process1> {
+class ContinuousProcess1 : public ContinuousProcess<ContinuousProcess1> {
+public:
+  ContinuousProcess1() {}
+  void Init() { cout << "ContinuousProcess1::Init" << endl; }
+  template <typename D, typename T, typename S>
+  inline EProcessReturn DoContinuous(D& d, T&, S&) const {
+    cout << "ContinuousProcess1::DoContinuous" << endl;
+    for (int i = 0; i < 10; ++i) d.p[i] += 0.933;
+    return EProcessReturn::eOk;
+  }
+
+  template <typename Particle, typename Stack>
+  inline EProcessReturn DoDiscrete(Particle&, Stack&) const {
+    cout << "ContinuousProcess1::DoDiscrete" << endl;
+    return EProcessReturn::eOk;
+  }
+};
+
+class ContinuousProcess2 : public ContinuousProcess<ContinuousProcess2> {
+public:
+  ContinuousProcess2() {}
+  void Init() { cout << "ContinuousProcess2::Init" << endl; }
+  template <typename D, typename T, typename S>
+  inline EProcessReturn DoContinuous(D& d, T&, S&) const {
+    cout << "ContinuousProcess2::DoContinuous" << endl;
+    for (int i = 0; i < 20; ++i) d.p[i] += 0.933;
+    return EProcessReturn::eOk;
+  }
+
+  template <typename Particle, typename Stack>
+  inline EProcessReturn DoDiscrete(Particle&, Stack&) const {
+    cout << "ContinuousProcess2::DoDiscrete" << endl;
+    return EProcessReturn::eOk;
+  }
+};
+
+class Process1 : public DiscreteProcess<Process1> {
 public:
   Process1() {}
-  void Init() {} // cout << "Process1::Init" << endl; }
+  void Init() { cout << "Process1::Init" << endl; }
   template <typename D, typename T, typename S>
   inline EProcessReturn DoContinuous(D& d, T&, S&) const {
     for (int i = 0; i < 10; ++i) d.p[i] += 1 + i;
     return EProcessReturn::eOk;
   }
+  template <typename Particle, typename Stack>
+  inline EProcessReturn DoDiscrete(Particle&, Stack&) const {
+    cout << "Process1::DoDiscrete" << endl;
+    return EProcessReturn::eOk;
+  }
 };
 
-class Process2 : public BaseProcess<Process2> {
+class Process2 : public DiscreteProcess<Process2> {
 public:
   Process2() {}
-  void Init() {} // cout << "Process2::Init" << endl; }
+  void Init() { cout << "Process2::Init" << endl; }
   template <typename D, typename T, typename S>
   inline EProcessReturn DoContinuous(D& d, T&, S&) const {
     for (int i = 0; i < 10; ++i) d.p[i] *= 0.7;
     return EProcessReturn::eOk;
   }
+  template <typename Particle, typename Stack>
+  inline EProcessReturn DoDiscrete(Particle&, Stack&) const {
+    cout << "Process2::DoDiscrete" << endl;
+    return EProcessReturn::eOk;
+  }
 };
 
-class Process3 : public BaseProcess<Process3> {
+class Process3 : public DiscreteProcess<Process3> {
 public:
   Process3() {}
-  void Init() {} // cout << "Process3::Init" << endl; }
+  void Init() { cout << "Process3::Init" << endl; }
   template <typename D, typename T, typename S>
   inline EProcessReturn DoContinuous(D& d, T&, S&) const {
     for (int i = 0; i < 10; ++i) d.p[i] += 0.933;
     return EProcessReturn::eOk;
   }
+  template <typename Particle, typename Stack>
+  inline EProcessReturn DoDiscrete(Particle&, Stack&) const {
+    cout << "Process3::DoDiscrete" << endl;
+    return EProcessReturn::eOk;
+  }
 };
 
 class Process4 : public BaseProcess<Process4> {
 public:
   Process4() {}
-  void Init() {} // cout << "Process4::Init" << endl; }
+  void Init() { cout << "Process4::Init" << endl; }
   template <typename D, typename T, typename S>
   inline EProcessReturn DoContinuous(D& d, T&, S&) const {
     for (int i = 0; i < 10; ++i) d.p[i] /= 1.2;
@@ -85,17 +136,30 @@ TEST_CASE("Cascade", "[Cascade]") {
 
     const auto sequence = m1 + m2 + m3 + m4;
 
+    ContinuousProcess1 cp1;
+    ContinuousProcess2 cp2;
+
+    const auto sequence2 = cp1 + m2 + m3 + cp2;
+
     DummyData p;
     DummyTrajectory t;
     DummyStack s;
 
+    cout << "-->init" << endl;
+    sequence2.Init();
+    cout << "-->docont" << endl;
+    sequence2.DoContinuous(p, t, s);
+    cout << "-->dodisc" << endl;
+    sequence2.DoDiscrete(p, s);
+    cout << "-->done" << endl;
+
     sequence.Init();
 
     const int n = 100;
-    INFO("Running loop with n=" << n);
+    cout << "Running loop with n=" << n << endl;
     for (int i = 0; i < n; ++i) { sequence.DoContinuous(p, t, s); }
 
-    for (int i = 0; i < 10; i++) { INFO("data[" << i << "]=" << p.p[i]); }
+    for (int i = 0; i < 10; i++) { cout << "data[" << i << "]=" << p.p[i] << endl; }
   }
 
   SECTION("sectionThree") {}
-- 
GitLab