1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
#include "caffe2/serialize/istream_adapter.h"
#include <c10/util/Exception.h>
namespace caffe2 {
namespace serialize {
IStreamAdapter::IStreamAdapter(std::istream* istream) : istream_(istream) {}
size_t IStreamAdapter::size() const {
auto prev_pos = istream_->tellg();
validate("getting the current position");
istream_->seekg(0, istream_->end);
validate("seeking to end");
auto result = istream_->tellg();
validate("getting size");
istream_->seekg(prev_pos);
validate("seeking to the original position");
return result;
}
size_t IStreamAdapter::read(uint64_t pos, void* buf, size_t n, const char* what)
const {
istream_->seekg(pos);
validate(what);
istream_->read(static_cast<char*>(buf), n);
validate(what);
return n;
}
void IStreamAdapter::validate(const char* what) const {
if (!*istream_) {
AT_ERROR("istream reader failed: ", what, ".");
}
}
// NOLINTNEXTLINE(modernize-use-equals-default)
IStreamAdapter::~IStreamAdapter() {}
} // namespace serialize
} // namespace caffe2
|
