added coshat problems

mlmc/src/MLMCMain.C

 #include "MLMCMain.h"
 
-
 using namespace std;
 
 void MLMCMain::initialize() {
@@ -59,9 +58,12 @@ MatrixGraphs *MLMCMain::getSampleMatrixGraphs() {
 }
 
 StochasticField *MLMCMain::getStochasticField() {
-    if (problemName == "StochasticLaplace1D" ||  problemName == "StochasticLaplace2D")
+    if (problemName == "StochasticLaplace1D" || problemName == "StochasticLaplace2D")
         return new StochasticField(*meshes, "CirculantEmbedding");
-    if (problemName == "StochasticPollution1D" || problemName == "StochasticPollution2D"|| problemName == "Pollution2D")
+    if (problemName == "StochasticPollution1D" || problemName == "StochasticPollution2D"
+        || problemName == "StochasticPollution2DCosHat"
+        || problemName == "DeterministicPollution2D"
+        || problemName == "DeterministicPollution2DCosHat")
         return new StochasticField(*meshes, "HybridFluxGenerator");
 }
 
@@ -74,8 +76,12 @@ StochasticProblem *MLMCMain::getStochasticProblem() {
         return new StochasticPollution1D();
     if (problemName == "StochasticPollution2D")
         return new StochasticPollution2D();
-    if (problemName == "Pollution2D")
-        return new Pollution2D();
+    if (problemName == "StochasticPollution2DCosHat")
+        return new StochasticPollution2DCosHat();
+    if (problemName == "DeterministicPollution2D")
+        return new DeterministicPollution2D();
+    if (problemName == "DeterministicPollution2DCosHat")
+        return new DeterministicPollution2DCosHat();
     Exit("\nStochastic problem not implemented yet\n")
 }