From: Zooko O'Whielacronx zooko@zooko.com Date: Sat, 14 Apr 2007 17:52:45 +0000 (+0530) Subject: pyfec: add easyfec.Decoder(), which is easier to use than fec.Decoder() X-Git-Url: https://git.rkrishnan.org/somewhere?a=commitdiff_plain;h=5b696e2244a57f9893fca90afa0dd664ed58cd86;p=tahoe-lafs%2Fzfec.git pyfec: add easyfec.Decoder(), which is easier to use than fec.Decoder() darcs-hash:95bf3656157b44e5c7eddfae3430d4db0204ef30 --- diff --git a/pyfec/fec/easyfec.py b/pyfec/fec/easyfec.py index de7e8b9..8fc1506 100644 --- a/pyfec/fec/easyfec.py +++ b/pyfec/fec/easyfec.py @@ -7,7 +7,6 @@ def div_ceil(n, d): """ return (n/d) + (n%d != 0) - class Encoder(object): def __init__(self, k, m): self.fec = fec.Encoder(k, m) @@ -22,8 +21,17 @@ class Encoder(object): # padding if len(l[-1]) != len(l[0]): l[-1] = l[-1] + ('\x00'*(len(l[0])-len(l[-1]))) - return self.fec.encode(l) - - def decode(self, blocks): - return self.fec.decode(blocks) + res = self.fec.encode(l) + return res +class Decoder(object): + def __init__(self, k, m): + self.fec = fec.Decoder(k, m) + + def decode(self, blocks, sharenums, padlen=0): + blocks = self.fec.decode(blocks, sharenums) + data = ''.join(blocks) + if padlen: + data = data[:-padlen] + return data +