rapidyenc/rapidyenc.cc at master · animetosho/rapidyenc · GitHub

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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
#ifdef BUILD_SHARED
# ifdef _MSC_VER
#  define RAPIDYENC_API __declspec(dllexport)
# else
#  define RAPIDYENC_API __attribute__((visibility("default")))
# endif
#endif

#include "rapidyenc.h"

int rapidyenc_version(void) {
	return RAPIDYENC_VERSION;
}

#ifndef RAPIDYENC_DISABLE_ENCODE

#include "src/encoder.h"
void rapidyenc_encode_init(void) {
	static int done = 0;
	if(done) return;
	done = 1;
	RapidYenc::encoder_init();
}

size_t rapidyenc_encode(const void* __restrict src, void* __restrict dest, size_t src_length) {
	return rapidyenc_encode_ex(128, NULL, src, dest, src_length, 1);
}

size_t rapidyenc_encode_ex(int line_size, int* column, const void* __restrict src, void* __restrict dest, size_t src_length, int is_end) {
	int unusedColumn = 0;
	if(!column) column = &unusedColumn;
	return RapidYenc::encode(line_size, column, src, dest, src_length, is_end);
}

int rapidyenc_encode_kernel() {
	return RapidYenc::encode_isa_level();
}

#endif // !defined(RAPIDYENC_DISABLE_ENCODE)

size_t rapidyenc_encode_max_length(size_t length, int line_size) {
	size_t ret = length * 2    /* all characters escaped */
		+ 2 /* allocation for offset and that a newline may occur early */
#if !defined(YENC_DISABLE_AVX256)
		+ 64 /* allocation for YMM overflowing */
#else
		+ 32 /* allocation for XMM overflowing */
#endif
	;
	/* add newlines, considering the possibility of all chars escaped */
	if(line_size == 128) // optimize common case
		return ret + 2 * (length >> 6);
	return ret + 2 * ((length*2) / line_size);
}


#ifndef RAPIDYENC_DISABLE_DECODE

#include "src/decoder.h"
void rapidyenc_decode_init(void) {
	static int done = 0;
	if(done) return;
	done = 1;
	RapidYenc::decoder_init();
}

size_t rapidyenc_decode(const void* src, void* dest, size_t src_length) {
	return rapidyenc_decode_ex(1, src, dest, src_length, NULL);
}

size_t rapidyenc_decode_ex(int is_raw, const void* src, void* dest, size_t src_length, RapidYencDecoderState* state) {
	RapidYencDecoderState unusedState = RYDEC_STATE_CRLF;
	if(!state) state = &unusedState;
	return RapidYenc::decode(is_raw, src, dest, src_length, (RapidYenc::YencDecoderState*)state);
}

RapidYencDecoderEnd rapidyenc_decode_incremental(const void** src, void** dest, size_t src_length, RapidYencDecoderState* state) {
	RapidYencDecoderState unusedState = RYDEC_STATE_CRLF;
	if(!state) state = &unusedState;
	return (RapidYencDecoderEnd)RapidYenc::decode_end(src, dest, src_length, (RapidYenc::YencDecoderState*)state);
}

int rapidyenc_decode_kernel() {
	return RapidYenc::decode_isa_level();
}

#endif // !defined(RAPIDYENC_DISABLE_DECODE)

#ifndef RAPIDYENC_DISABLE_CRC

#include "src/crc.h"
void rapidyenc_crc_init(void) {
	static int done = 0;
	if(done) return;
	done = 1;
	RapidYenc::crc32_init();
}

uint32_t rapidyenc_crc(const void* src, size_t src_length, uint32_t init_crc) {
	return RapidYenc::crc32(src, src_length, init_crc);
}
uint32_t rapidyenc_crc_combine(uint32_t crc1, const uint32_t crc2, uint64_t length2) {
	return RapidYenc::crc32_combine(crc1, crc2, length2);
}
uint32_t rapidyenc_crc_zeros(uint32_t init_crc, uint64_t length) {
	return RapidYenc::crc32_zeros(init_crc, length);
}
uint32_t rapidyenc_crc_unzero(uint32_t init_crc, uint64_t length) {
	return RapidYenc::crc32_unzero(init_crc, length);
}
uint32_t rapidyenc_crc_multiply(uint32_t a, uint32_t b) {
	return RapidYenc::crc32_multiply(a, b);
}
uint32_t rapidyenc_crc_2pow(int64_t n) {
	return RapidYenc::crc32_2pow(n);
}
uint32_t rapidyenc_crc_256pow(uint64_t n) {
	return RapidYenc::crc32_256pow(n);
}

int rapidyenc_crc_kernel() {
	return RapidYenc::crc32_isa_level();
}

#endif // !defined(RAPIDYENC_DISABLE_CRC)