Skip to content
FileIO.hpp 1.12 KiB
Newer Older
#pragma once

#include <cstddef>
#include <string>
#include "MpiEnvironment.hpp"
Thomas Steinreiter's avatar
Thomas Steinreiter committed
#include "MpiSubarray.hpp"
#include "State.hpp"
#include "Util.hpp"
struct HeaderInfo {
	Size GlobalSize;
	std::size_t HeaderLength;
};

struct FileIO {
	static HeaderInfo ReadHeader(const std::string& path);
	static void WriteHeader(const HeaderInfo& header, const std::string& path,
	                        const MpiEnvironment& env);
	static TileInfo GetTileInfo(Size globalSize, Size procsSize,
	                            std::size_t rank);
	// helper class to share commonly used data for reading and writing
	class Tile {
		static constexpr std::size_t LF = 1; // linefeed chars
		const std::string& _path;
		const std::size_t _headerLength;
		const Size _srcSize;
		State* _buf;
		const TileInfo _tileInfo;
		const Size _tileSize;
		const MpiSubarray _tileType;
		const MpiSubarray _bufType;
		const std::size_t _displ;

	  public:
		Tile(const std::string& path, HeaderInfo header, Size procsSize,
		     std::size_t rank, State* buf);
		void Read();
		void Write() const;