aboutsummaryrefslogtreecommitdiffstats
path: root/library/cpp/yt/coding/varint.h
blob: c5399f8b069cd4f91441f4f04189ed7a539b834e (plain) (blame)
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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
#pragma once

#include <library/cpp/yt/exception/exception.h>

#include <util/system/defaults.h>

#include <util/stream/input.h>
#include <util/stream/output.h>

namespace NYT {

////////////////////////////////////////////////////////////////////////////////

constexpr size_t MaxVarInt64Size = (8 * sizeof(ui64) - 1) / 7 + 1;
constexpr size_t MaxVarUint64Size = (8 * sizeof(ui64) - 1) / 7 + 1;

constexpr size_t MaxVarInt32Size = (8 * sizeof(ui32) - 1) / 7 + 1;
constexpr size_t MaxVarUint32Size = (8 * sizeof(ui32) - 1) / 7 + 1;

// Various functions to read/write varints.

// Returns the number of bytes written.
int WriteVarUint64(IOutputStream* output, ui64 value);
int WriteVarUint32(IOutputStream* output, ui32 value);
int WriteVarInt32(IOutputStream* output, i32 value);
int WriteVarInt64(IOutputStream* output, i64 value);

int WriteVarUint64(char* output, ui64 value);
int WriteVarUint32(char* output, ui32 value);
int WriteVarInt32(char* output, i32 value);
int WriteVarInt64(char* output, i64 value);

// Returns the number of bytes read.
int ReadVarUint64(IInputStream* input, ui64* value);
int ReadVarUint32(IInputStream* input, ui32* value);
int ReadVarInt32(IInputStream* input, i32* value);
int ReadVarInt64(IInputStream* input, i64* value);

int ReadVarUint64(const char* input, ui64* value);
int ReadVarUint32(const char* input, ui32* value);
int ReadVarInt32(const char* input, i32* value);
int ReadVarInt64(const char* input, i64* value);

// Throws exception if integer is not complete when `end' is reached.
int ReadVarUint64(const char* input, const char* end, ui64* value);
int ReadVarUint32(const char* input, const char* end, ui32* value);
int ReadVarInt32(const char* input, const char* end, i32* value);
int ReadVarInt64(const char* input, const char* end, i64* value);

////////////////////////////////////////////////////////////////////////////////

} // namespace NYT

#define VARINT_INL_H_
#include "varint-inl.h"
#undef VARINT_INL_H_