mirror of
https://github.com/ggerganov/ggwave.git
synced 2026-02-08 18:38:05 +08:00
95 lines
2.3 KiB
C++
95 lines
2.3 KiB
C++
/* Author: Mike Lubinets (aka mersinvald)
|
|
* Date: 29.12.15
|
|
*
|
|
* See LICENSE */
|
|
|
|
#ifndef POLY_H
|
|
#define POLY_H
|
|
|
|
#include <stdint.h>
|
|
#include <string.h>
|
|
#include <assert.h>
|
|
|
|
namespace RS {
|
|
|
|
struct Poly {
|
|
Poly()
|
|
: length(0), _memory(NULL) {}
|
|
|
|
Poly(uint8_t id, uint16_t offset, uint8_t size) \
|
|
: length(0), _id(id), _size(size), _offset(offset), _memory(NULL) {}
|
|
|
|
/* @brief Append number at the end of polynomial
|
|
* @param num - number to append
|
|
* @return false if polynomial can't be stretched */
|
|
inline bool Append(uint8_t num) {
|
|
assert(length < _size);
|
|
ptr()[length++] = num;
|
|
return true;
|
|
}
|
|
|
|
/* @brief Polynomial initialization */
|
|
inline void Init(uint8_t id, uint16_t offset, uint8_t size, uint8_t** memory_ptr) {
|
|
this->_id = id;
|
|
this->_offset = offset;
|
|
this->_size = size;
|
|
this->length = 0;
|
|
this->_memory = memory_ptr;
|
|
}
|
|
|
|
/* @brief Polynomial memory zeroing */
|
|
inline void Reset() {
|
|
memset((void*)ptr(), 0, this->_size);
|
|
}
|
|
|
|
/* @brief Copy polynomial to memory
|
|
* @param src - source byte-sequence
|
|
* @param size - size of polynomial
|
|
* @param offset - write offset */
|
|
inline void Set(const uint8_t* src, uint8_t len, uint8_t offset = 0) {
|
|
assert(src && len <= this->_size-offset);
|
|
memcpy(ptr()+offset, src, len * sizeof(uint8_t));
|
|
length = len + offset;
|
|
}
|
|
|
|
#define poly_max(a, b) ((a > b) ? (a) : (b))
|
|
|
|
inline void Copy(const Poly* src) {
|
|
length = poly_max(length, src->length);
|
|
Set(src->ptr(), length);
|
|
}
|
|
|
|
inline uint8_t& at(uint8_t i) const {
|
|
assert(i < _size);
|
|
return ptr()[i];
|
|
}
|
|
|
|
inline uint8_t id() const {
|
|
return _id;
|
|
}
|
|
|
|
inline uint8_t size() const {
|
|
return _size;
|
|
}
|
|
|
|
// Returns pointer to memory of this polynomial
|
|
inline uint8_t* ptr() const {
|
|
assert(_memory && *_memory);
|
|
return (*_memory) + _offset;
|
|
}
|
|
|
|
uint8_t length;
|
|
|
|
protected:
|
|
|
|
uint8_t _id;
|
|
uint8_t _size; // Size of reserved memory for this polynomial
|
|
uint16_t _offset; // Offset in memory
|
|
uint8_t** _memory; // Pointer to pointer to memory
|
|
};
|
|
|
|
|
|
}
|
|
|
|
#endif // POLY_H
|