#include "IStochasticProblem.hpp"
#include "generators/CirculantEmbedding.hpp"
#include "generators/HybridFluxGenerator.hpp"


SampleGenerator *IStochasticProblem::createGenerator(GeneratorName genName,
                                                     Meshes &meshes) {
    if (genName == "DummyGenerator")
        return new DummyGenerator(meshes);
//    if (genName == "UniformRandomNumberGenerator")
//        return new UniformRandomNumberGenerator(meshes);
//    if (genName == "NormalRandomNumberGenerator")
//        return new NormalRandomNumberGenerator(meshes);

    if (genName == "CirculantEmbedding")
        return new MultilevelCirculantEmbedding(meshes);
    if (genName == "HybridFluxGenerator")
        return new MultilevelHybridFluxGenerator(meshes);
    Exit("Generator not implemented")
}