not losing problem information and moving timeseries

mlmc/src/pdesolver/IStochasticEllipticAssemble.hpp

@@ -21,7 +21,7 @@ public:
         problem->DrawSample(id);
     }
 
-    IStochasticProblem *GetProblem() const {
+    IStochasticEllipticProblem *GetProblem() const {
         return problem;
     };
 

mlmc/src/pdesolver/IStochasticTransportAssemble.hpp

@@ -31,10 +31,16 @@ public:
         this->plot = _plot;
     }
 
-    IStochasticProblem *GetProblem() {
+    IStochasticTransportProblem *GetProblem() {
         return problem;
     };
 
+    TimeSeries GetTimeSeries(const Vector &u) {
+        return TimeSeries(problem->GetStartTime(), problem->GetEndTime(),
+                          problem->GetCFL() * u.GetMesh().MeshWidth().second,
+                          "UniformTimeSeries");
+    }
+
     virtual IDiscretization *GetDisc() = 0;
 
     virtual ~IStochasticTransportAssemble() = default;