check-dim.hpp

Go to the documentation of this file.

#pragma once
 
#include <alpaqa/config/config.hpp>
#include <optional>
#include <stdexcept>
#include <string>
 
namespace alpaqa::util {
 
template <class V>
void check_dim_msg(const V &v, auto sz, std::string msg) {
    if (v.size() != sz) {
        msg += "\n(should be ";
        msg += std::to_string(sz);
        msg += ", got ";
        msg += std::to_string(v.size());
        msg += ")";
        throw std::invalid_argument(msg);
    }
}
 
template <class V>
void check_dim_msg(std::optional<V> &v, auto sz, std::string msg) {
    if (!v) {
        v = V::Zero(sz);
    } else if (v->size() != sz) {
        msg += "\n(should be ";
        msg += std::to_string(sz);
        msg += ", got ";
        msg += std::to_string(v->size());
        msg += ")";
        throw std::invalid_argument(msg);
    }
}
 
template <class V>
void check_dim(std::string name, V &&v, auto sz) {
    name += ": dimension mismatch";
    check_dim_msg(std::forward<V>(v), sz, name);
}
 
template <class M>
void check_dim_msg(const M &m, auto rows, auto cols, std::string msg) {
    if (m.cols() != cols || m.rows() != rows) {
        msg += "\n(should be ";
        msg += std::to_string(rows);
        msg += "×";
        msg += std::to_string(cols);
        msg += ", got ";
        msg += std::to_string(m.rows());
        msg += "×";
        msg += std::to_string(m.cols());
        msg += ")";
        throw std::invalid_argument(msg);
    }
}
 
template <class M>
void check_dim(std::string name, M &&m, auto rows, auto cols) {
    name += ": dimension mismatch";
    check_dim_msg(std::forward<M>(m), rows, cols, name);
}
 
} // namespace alpaqa::util