如何泛化Eigen矩阵存储函数以支持Vector和Array类型？

核心问题分析 #

原函数仅支持Eigen::Matrix类型，Eigen::Vector、Eigen::Array等稠密类型无法直接传入；若用Eigen::EigenBase作为参数类型，又因基类未定义data()方法无法访问底层数据。

解决方案1：使用Eigen::DenseBase作为模板参数 #

Eigen::DenseBase<Derived>是所有Eigen稠密数据结构（包括Matrix、Vector、Array，无论动态/固定大小）的共同基类，它提供data()方法获取底层指针，以及size()方法返回总元素数，完美解决兼容性问题。同时修复原函数的字节数计算错误（需乘以元素类型的大小）：

#include <Eigen/Core>
#include <fstream>

template<typename Derived>
void WriteEigenDense(const Eigen::DenseBase<Derived>& dense_data, const char* fileName)
{
    std::ofstream outFile(fileName, std::ios_base::out | std::ios_base::binary);
    // 计算总字节数：元素总数 × 单个元素字节大小
    const std::streamsize total_bytes = dense_data.size() * sizeof(typename Derived::Scalar);
    outFile.write(reinterpret_cast<const char*>(dense_data.data()), total_bytes);
    outFile.close();
}

该函数可直接接收以下类型：

• Eigen::MatrixXd/Eigen::Matrix3f等矩阵类型
• Eigen::VectorXd/Eigen::Vector4i等向量类型
• Eigen::ArrayXXd/Eigen::Array33f等数组类型

解决方案2：C++20概念实现通用约束（可选） #

若需要支持非Eigen的稠密数据结构，可使用C++20的概念约束模板参数，只要类型提供data()、size()方法并定义Scalar类型别名即可：

#include <concepts>
#include <fstream>

template<typename DenseType>
requires requires(const DenseType& data) {
    { data.data() } -> std::convertible_to<const void*>;
    { data.size() } -> std::convertible_to<std::streamsize>;
    typename DenseType::Scalar;
}
void WriteDenseData(const DenseType& data, const char* fileName)
{
    using Scalar = typename DenseType::Scalar;
    std::ofstream outFile(fileName, std::ios_base::out | std::ios_base::binary);
    const std::streamsize total_bytes = data.size() * sizeof(Scalar);
    outFile.write(reinterpret_cast<const char*>(data.data()), total_bytes);
    outFile.close();
}

关键注意事项 #

• 原函数未考虑元素类型的字节大小，直接传入元素总数会导致写入数据不完整，必须乘以sizeof(Scalar)修正。
• 若处理稀疏矩阵，需使用Eigen::SparseBase而非DenseBase，但稀疏矩阵的二进制写入逻辑需单独设计。

内容的提问来源于stack exchange，提问作者Soo