video2x/tools/video2x/include/validators.h

#pragma once

#include <libvideo2x/fsutils.h>
#include <boost/program_options.hpp>

namespace po = boost::program_options;

template <typename T>
void validate_positive(const T &value, const std::string &option_name) {
    if (value < 0) {
        throw po::validation_error(
            po::validation_error::invalid_option_value,
            option_name,
            option_name + " must be positive"
        );
    }
}

template <typename T>
void validate_min(const T &value, const std::string &option_name, const T &min) {
    if (value < min) {
        throw po::validation_error(
            po::validation_error::invalid_option_value,
            option_name,
            option_name + " must be at least " + std::to_string(min)
        );
    }
}

template <typename T>
void validate_max(const T &value, const std::string &option_name, const T &max) {
    if (value > max) {
        throw po::validation_error(
            po::validation_error::invalid_option_value,
            option_name,
            option_name + " must be at most " + std::to_string(max)
        );
    }
}

template <typename T>
void validate_range(const T &value, const std::string &option_name, const T &min, const T &max) {
    if (value < min || value > max) {
        throw po::validation_error(
            po::validation_error::invalid_option_value,
            option_name,
            option_name + " must be in the range [" + std::to_string(min) + ", " +
                std::to_string(max) + "]"
        );
    }
}

template <typename T>
void validate_greater_equal_one(const T &value, const std::string &option_name) {
    if (value < 1) {
        throw po::validation_error(
            po::validation_error::invalid_option_value,
            option_name,
            option_name + " must be greater than or equal to 1"
        );
    }
}

void validate_anime4k_shader_name(const StringType &shader_name);

void validate_realesrgan_model_name(const StringType &model_name);

void validate_rife_model_name(const StringType &model_name);