diff options
| author | Chris Dyer <redpony@gmail.com> | 2014-10-13 00:42:37 -0400 |
|---|---|---|
| committer | Chris Dyer <redpony@gmail.com> | 2014-10-13 00:42:37 -0400 |
| commit | b1ed81ef3216b212295afa76c5d20a56fb647204 (patch) | |
| tree | 9633cdc1b8a341dfa58b0b7fec0e2cae44d28835 /klm/util/read_compressed.hh | |
| parent | 1b17f61d359be6e1c3cea29f8c100db3bcdd73a0 (diff) | |
new kenlm
Diffstat (limited to 'klm/util/read_compressed.hh')
| -rw-r--r-- | klm/util/read_compressed.hh | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/klm/util/read_compressed.hh b/klm/util/read_compressed.hh index 8b54c9e8..767ee94b 100644 --- a/klm/util/read_compressed.hh +++ b/klm/util/read_compressed.hh @@ -1,5 +1,5 @@ -#ifndef UTIL_READ_COMPRESSED__ -#define UTIL_READ_COMPRESSED__ +#ifndef UTIL_READ_COMPRESSED_H +#define UTIL_READ_COMPRESSED_H #include "util/exception.hh" #include "util/scoped.hh" @@ -62,6 +62,10 @@ class ReadCompressed { std::size_t Read(void *to, std::size_t amount); + // Repeatedly call read to fill a buffer unless EOF is hit. + // Return number of bytes read. + std::size_t ReadOrEOF(void *const to, std::size_t amount); + uint64_t RawAmount() const { return raw_amount_; } private: @@ -78,4 +82,4 @@ class ReadCompressed { } // namespace util -#endif // UTIL_READ_COMPRESSED__ +#endif // UTIL_READ_COMPRESSED_H |