You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
68 lines
1.5 KiB
68 lines
1.5 KiB
#ifndef MARISA_GRIMOIRE_IO_MAPPER_H_
|
|
#define MARISA_GRIMOIRE_IO_MAPPER_H_
|
|
|
|
#include <cstdio>
|
|
|
|
#include "marisa/base.h"
|
|
|
|
namespace marisa {
|
|
namespace grimoire {
|
|
namespace io {
|
|
|
|
class Mapper {
|
|
public:
|
|
Mapper();
|
|
~Mapper();
|
|
|
|
void open(const char *filename);
|
|
void open(const void *ptr, std::size_t size);
|
|
|
|
template <typename T>
|
|
void map(T *obj) {
|
|
MARISA_THROW_IF(obj == NULL, MARISA_NULL_ERROR);
|
|
*obj = *static_cast<const T *>(map_data(sizeof(T)));
|
|
}
|
|
|
|
template <typename T>
|
|
void map(const T **objs, std::size_t num_objs) {
|
|
MARISA_THROW_IF((objs == NULL) && (num_objs != 0), MARISA_NULL_ERROR);
|
|
MARISA_THROW_IF(num_objs > (MARISA_SIZE_MAX / sizeof(T)),
|
|
MARISA_SIZE_ERROR);
|
|
*objs = static_cast<const T *>(map_data(sizeof(T) * num_objs));
|
|
}
|
|
|
|
void seek(std::size_t size);
|
|
|
|
bool is_open() const;
|
|
|
|
void clear();
|
|
void swap(Mapper &rhs);
|
|
|
|
private:
|
|
const void *ptr_;
|
|
void *origin_;
|
|
std::size_t avail_;
|
|
std::size_t size_;
|
|
#if (defined _WIN32) || (defined _WIN64)
|
|
void *file_;
|
|
void *map_;
|
|
#else // (defined _WIN32) || (defined _WIN64)
|
|
int fd_;
|
|
#endif // (defined _WIN32) || (defined _WIN64)
|
|
|
|
void open_(const char *filename);
|
|
void open_(const void *ptr, std::size_t size);
|
|
|
|
const void *map_data(std::size_t size);
|
|
|
|
// Disallows copy and assignment.
|
|
Mapper(const Mapper &);
|
|
Mapper &operator=(const Mapper &);
|
|
};
|
|
|
|
} // namespace io
|
|
} // namespace grimoire
|
|
} // namespace marisa
|
|
|
|
#endif // MARISA_GRIMOIRE_IO_MAPPER_H_
|