Call compute() upon encoder creation

This is not done automatically, but is required when using the RFC version of libRaptorQ.
harmony-one · Dec 7, 2018 · 5459ed8 · 5459ed8
1 parent cbee330
commit 5459ed8
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/pkg/impl/libraptorq/encoder.go b/pkg/impl/libraptorq/encoder.go
@@ -12,7 +12,7 @@ type EncoderFactory struct {
 
 func (*EncoderFactory) New(input []byte, symbolSize uint16, minSubSymbolSize uint16,
  maxSubBlockSize uint32, alignment uint8) (enc raptorq.Encoder, err error) {
- wrapped := swig.NewBytesEncoder(input, minSubSymbolSize, symbolSize,
+ wrapped := swig.InitBytesEncoder(input, minSubSymbolSize, symbolSize,
  int64(maxSubBlockSize))
  if !wrapped.Initialized() {
  swig.DeleteBytesEncoder(wrapped)

diff --git a/pkg/impl/libraptorq/swig/swig.swigcxx b/pkg/impl/libraptorq/swig/swig.swigcxx
@@ -1094,3 +1094,20 @@ public:
 
 %template(BytesEncoder) RFC6330__v1::Impl::Encoder<unsigned char *, unsigned char *>;
 %template(BytesDecoder) RFC6330__v1::Impl::Decoder<unsigned char *, unsigned char *>;
+
+%inline %{
+
+typedef RFC6330__v1::Impl::Encoder<unsigned char *, unsigned char *> BytesEncoder;
+
+BytesEncoder *InitBytesEncoder(
+ unsigned char *SLICEBEGIN, unsigned char *SLICEEND,
+ uint16_t min_subsymbol_size, uint16_t symbol_size, size_t max_sub_block
+) {
+ auto enc = new BytesEncoder(
+ SLICEBEGIN, SLICEEND, min_subsymbol_size, symbol_size, max_sub_block
+ );
+ enc->compute(RaptorQ__v1::Compute::COMPLETE | RaptorQ__v1::Compute::NO_BACKGROUND);
+ return enc;
+}
+
+%}