本征和巨大的密集2D阵列

时间:2018-06-30 19:48:08

标签: c++ multidimensional-array eigen memory-mapped-files eigen3

我在一个项目中使用2D Eigen::Array,在大型2D阵列的情况下,我希望继续使用它们。

为了避免出现内存问题,我考虑使用内存映射文件来管理(读取/修改/写入)这些数组,但是我找不到有效的示例。

我发现的最接近的示例是基于boost::interprocess的{​​{3}},但是它使用共享内存(尽管我希望拥有持久性存储)。

缺乏示例使我担心是否有更好的主流替代解决方案来解决我的问题。是这样吗一个最小的例子将非常方便。

编辑:

这是在注释中解释我的用例的最小示例:

#include <Eigen/Dense>


int main()
{
    // Order of magnitude of the required arrays
    Eigen::Index rows = 50000;
    Eigen::Index cols = 40000;

    {
        // Array creation (this is where the memory mapped file should be created)
        Eigen::ArrayXXf arr1 = Eigen::ArrayXXf::Zero( rows, cols );

        // Some operations on the array
        for(Eigen::Index i = 0; i < rows; ++i)
        {
            for(Eigen::Index j = 0; j < cols; ++j)
            {
                arr1( i, j ) = float(i * j);
            }
        }

        // The array goes out of scope, but the data are persistently stored in the file
    }

    {
        // This should actually use the data stored in the file
        Eigen::ArrayXXf arr2 = Eigen::ArrayXXf::Zero( rows, cols );

        // Manipulation of the array data
        for(Eigen::Index i = 0; i < rows; ++i)
        {
            for(Eigen::Index j = 0; j < cols; ++j)
            {
                arr2( i, j ) += 1.0f;
            }
        }

        // The array goes out of scope, but the data are persistently stored in the file
    }

}

3 个答案:

答案 0 :(得分:5)

所以我用谷歌搜索

  

提升内存映射文件

,并在第一个结果中遇到boost::iostreams::mapped_file

结合从Eigen::Mapthis comment的链接,我测试了以下内容:

#include <boost/iostreams/device/mapped_file.hpp>
#include <Eigen/Dense>
boost::iostreams::mapped_file file("foo.bin");

const std::size_t rows = 163840;
const std::size_t columns = 163840;
if (rows * columns * sizeof(float) > file.size()) {
    throw std::runtime_error("file of size " + std::to_string(file.size()) + " couldn’t fit float Matrix of " + std::to_string(rows) + "×"  + std::to_string(columns));
}

Eigen::Map<Eigen::MatrixXf> matrix(reinterpret_cast<float*>(file.data()), rows, columns);

std::cout << matrix(0, 0) << ' ' << matrix(rows - 1, columns - 1) << std::endl;
matrix(0, 0) = 0.5;
matrix(rows - 1, columns - 1) = 0.5;

使用

find_package(Boost REQUIRED COMPONENTS iostreams)
find_package(Eigen3 REQUIRED)
target_link_libraries(${PROJECT_NAME} Boost::iostreams Eigen3::Eigen)

然后我用谷歌搜索

  

Windows创建虚拟文件

first result给了我

fsutil file createnew foo.bin 107374182400

运行该程序两次将给出:

  

0 0

     

0.5 0.5

不增加内存使用量。

因此,它就像一种魅力。

答案 1 :(得分:1)

我认为为此编写自己的类并不难。

要首次初始化数组,请创建一个大小为x * y * elem_sizememory map的文件。

您甚至可以添加带有大小,x,y等信息的小标头-这样,如果您重新打开这些标头,便可以获得所需的所有信息。

现在您有一个大内存块,可以使用成员函数elem(x,y)get_elem() / set_elem()或使用[] operator,然后在该函数中计算位置数据元素。

关闭文件或在文件之间进行提交将保存数据。

对于非常大的文件,最好只需要map只使用文件的一部分,以避免创建非常大的页表。

特定于Windows(不确定在Linux中是否可用):

  • 如果不需要将数据保留在磁盘上,则可以使用delete on close标志打开文件。如果内存不可用,这只会(临时)写入磁盘。

  • 对于稀疏数组,可以使用稀疏文件。这些文件仅将磁盘空间用于包含数据的块。所有其他块都是虚拟的,默认为全零。

答案 2 :(得分:0)

基于this comment和以下答案(https://stackoverflow.com/a/51256963/2741329https://stackoverflow.com/a/51256597/2741329),这是我的可行解决方案:

#include <boost/interprocess/file_mapping.hpp>
#include <boost/interprocess/mapped_region.hpp>
#include <Eigen/Dense>
#include <iostream>
#include <fstream>
#include <filesystem>

namespace fs = std::experimental::filesystem;
namespace bi = boost::interprocess;

int main() {

  std::string array_bin_path = "array.bin";
  const int64_t nr_rows = 28000;
  const int64_t nr_cols = 35000;
  const int64_t array_size = nr_rows * nr_cols * sizeof(float);
  std::cout << "array size: " << array_size << std::endl;

  // if the file already exists but the size is different, remove it
  if(fs::exists(array_bin_path))
  {
    int64_t file_size = fs::file_size(array_bin_path);
    std::cout << "file size: " << file_size << std::endl;
    if(array_size != file_size)
    {
      fs::remove(array_bin_path);
    }
  }

  // create a binary file of the required size
  if(!fs::exists(array_bin_path))
  {
    std::ofstream ofs(array_bin_path, std::ios::binary | std::ios::out | std::ios::trunc);
    ofs.seekp(array_size - 1);
    ofs.put(0);
    ofs.close();
  }

  // use boost interprocess to memory map the file
  const bi::file_mapping mapped_file(array_bin_path.c_str(), bi::read_write);
  bi::mapped_region region(mapped_file, bi::read_write);

  // get the address of the mapped region
  void * addr = region.get_address();

  const std::size_t region_size = region.get_size();
  std::cout << "region size: " << region_size << std::endl;

  // map the file content into a Eigen array
  Eigen::Map<Eigen::ArrayXXf> my_array(reinterpret_cast<float*>(addr), nr_rows, nr_cols);

  // modify the content
  std::cout << "initial array(0, 1) value: " << my_array(0, 1) << std::endl;
  my_array(0, 1) += 1.234f;
  std::cout << "final array(0, 1) value: " << my_array(0, 1) << std::endl;

  return 0;
}

它使用:

  • boost::interprocess代替boost::iostreams,因为它是仅标头。另外,如果我想在单个映射文件上存储多个数组,mapped_region很方便。
  • std::fstream创建二进制文件,std::experimental::filesystem进行检查。
  • Eigen::ArrayXXf根据我的问题要求。