1. bool FileLoader::read()
    2. {
    3.     auto block = MNNMemoryAllocAlign(gCacheSize, MNN_MEMORY_ALIGN_DEFAULT);
    4.     if (nullptr == block)
    5.     {
    6.         MNN_PRINT("Memory Alloc Failed\n");
    7.         return false;
    8.     }
    9.     auto size = fread(block, 1, gCacheSize, mFile);
    10.     mTotalSize = size;
    11.     mBlocks.push_back(std::make_pair(size, block));
    13.     while (size == gCacheSize)
    14.     {
    15.         block = MNNMemoryAllocAlign(gCacheSize, MNN_MEMORY_ALIGN_DEFAULT);
    16.         if (nullptr == block)
    17.         {
    18.             MNN_PRINT("Memory Alloc Failed\n");
    19.             return false;
    20.         }
    21.         size = fread(block, 1, gCacheSize, mFile);
    22.         if (size > gCacheSize)
    23.         {
    24.             MNN_PRINT("Read file Error\n");
    25.             MNNMemoryFreeAlign(block);
    26.             return false;
    27.         }
    28.         mTotalSize += size;
    29.         mBlocks.push_back(std::make_pair(size, block));
    30.     }
    32.     if (ferror(mFile))
    33.     {
    34.         return false;
    35.     }
    36.     return true;
    37. }

    FileLoader按块读取文件，将每块的大小及指针保存在mBlocks中。

    1. bool FileLoader::merge(AutoStorage<uint8_t> &buffer)
    2. {
    3.     buffer.reset((int)mTotalSize);
    4.     if (buffer.get() == nullptr)
    5.     {
    6.         MNN_PRINT("Memory Alloc Failed\n");
    7.         return false;
    8.     }
    9.     auto dst = buffer.get();
    10.     int offset = 0;
    11.     for (auto iter : mBlocks)
    12.     {
    13.         ::memcpy(dst + offset, iter.second, iter.first);
    14.         offset += iter.first;
    15.     }
    16.     return true;
    17. }

    AutoStorage负责进行自我内存存储，通过进行块合并，合并到AutoStorage &buffer中。