Jungfraujoch/image_analysis/bragg_prediction/BraggPredictionFactory.cpp

// SPDX-FileCopyrightText: 2025 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
// SPDX-License-Identifier: GPL-3.0-only

#include "BraggPredictionFactory.h"

#include "BraggPredictionRot.h"

#ifdef JFJOCH_USE_CUDA
#include "../../common/CUDAWrapper.h"
#include "BraggPredictionGPU.h"
#include "BraggPredictionRotGPU.h"
#endif

std::unique_ptr<BraggPrediction> CreateBraggPrediction(bool rotation_indexing, int max_reflections) {
#ifdef JFJOCH_USE_CUDA
    if (get_gpu_count() > 0) {
        if (rotation_indexing)
            return std::make_unique<BraggPredictionRotGPU>(max_reflections);
        return std::make_unique<BraggPredictionGPU>(max_reflections);
    }
#endif
    if (rotation_indexing)
        return std::make_unique<BraggPredictionRot>(max_reflections);
    return std::make_unique<BraggPrediction>(max_reflections);
}