fix a bug

Zentopian · Sep 6, 2016 · bd1002d · bd1002d
1 parent 99d98b2
commit bd1002d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/mylibs/ECC.py b/mylibs/ECC.py
@@ -42,7 +42,8 @@ def encode(ver, ecl, data_codewords):
     en = ecc_num_per_block[ver-1][lindex[ecl]]
     ecc = []
     for dc in data_codewords:
-        ecc.append(get_ecc(dc, en))
+        test = get_ecc(dc, en)
+        ecc.append(test)
     return ecc
 
 def get_ecc(dc, ecc_num):
@@ -62,7 +63,7 @@ def divide(MP, *GP):
             GP[i] = po2[GP[i]]
         return XOR(GP, *MP)
     else:
-        return MP[1:]
+        return XOR([0]*len(GP), *MP)
 
 
 def XOR(GP, *MP):