1
0
mirror of https://github.com/danog/ton.git synced 2024-12-04 10:28:27 +01:00
ton/tdfec/td/fec/raptorq/Encoder.h
2019-09-07 14:33:36 +04:00

69 lines
2.0 KiB
C++

/*
This file is part of TON Blockchain Library.
TON Blockchain Library is free software: you can redistribute it and/or modify
it under the terms of the GNU Lesser General Public License as published by
the Free Software Foundation, either version 2 of the License, or
(at your option) any later version.
TON Blockchain Library is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public License
along with TON Blockchain Library. If not, see <http://www.gnu.org/licenses/>.
Copyright 2017-2019 Telegram Systems LLP
*/
#pragma once
#include "td/fec/raptorq/RawEncoder.h"
#include "td/fec/common/SymbolsView.h"
#include "td/utils/optional.h"
#include "td/utils/buffer.h"
#include <atomic>
namespace td {
namespace raptorq {
class Encoder {
public:
struct Parameters {
size_t symbols_count;
size_t symbol_size;
size_t data_size;
};
struct Info {
uint32 symbol_count;
uint32 ready_symbol_count;
};
static Result<std::unique_ptr<Encoder>> create(size_t symbol_size, BufferSlice data);
static std::unique_ptr<Encoder> create(size_t symbol_size, RawEncoder raw_encoder);
Encoder(Rfc::Parameters p, size_t symbol_size, BufferSlice data, optional<RawEncoder> raw_encoder = {});
Status gen_symbol(uint32 id, MutableSlice slice);
Parameters get_parameters() const;
Info get_info() const;
bool has_precalc() const;
// Must be called only once. No concurrent calls are allowed.
// Also it may be and should be called from another thread.
void precalc();
private:
Rfc::Parameters p_;
size_t symbol_size_;
BufferSlice data_;
SymbolsView first_symbols_{p_.K_padded, symbol_size_, data_.as_slice()};
optional<RawEncoder> raw_encoder_;
std::atomic<bool> has_encoder_{false};
};
} // namespace raptorq
} // namespace td