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

#include "BraggPredictionFactory.h"

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

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