95 lines
3.5 KiB
Python
95 lines
3.5 KiB
Python
#!/bin/python3
|
|
|
|
import random
|
|
from bitarray import bitarray
|
|
from bitarray.util import zeros
|
|
|
|
class Block:
|
|
"""This class represents a block in the Ethereum blockchain."""
|
|
|
|
def __init__(self, blockSizeR, blockSizeRK=0, blockSizeC=0, blockSizeCK=0):
|
|
"""Initialize the block with a data array of blocksize^2 zeros.
|
|
BlockSizeR: row size
|
|
BlockSizeRK: original row size, before erasure coding to BlocksSizeR
|
|
BlockSizeC: column size (i.e. number of rows)
|
|
BlockSizeCK: original column size, before erasure coding to BlocksSizeR
|
|
"""
|
|
self.blockSizeR = blockSizeR
|
|
self.blockSizeRK = blockSizeRK if blockSizeRK else blockSizeR/2
|
|
self.blockSizeC = blockSizeC if blockSizeC else blockSizeR
|
|
self.blockSizeCK = blockSizeCK if blockSizeCK else blockSizeRK
|
|
self.data = zeros(self.blockSizeR*self.blockSizeC)
|
|
|
|
def fill(self):
|
|
"""It fills the block data with ones."""
|
|
self.data.setall(1)
|
|
|
|
def merge(self, merged):
|
|
"""It merges (OR) the existing block with the received one."""
|
|
self.data |= merged.data
|
|
|
|
def getSegment(self, rowID, columnID):
|
|
"""Check whether a segment is included"""
|
|
return self.data[rowID*self.blockSizeR + columnID]
|
|
|
|
def setSegment(self, rowID, columnID, value = 1):
|
|
"""Set value for a segment (default 1)"""
|
|
self.data[rowID*self.blockSizeR + columnID] = value
|
|
|
|
def getColumn(self, columnID):
|
|
"""It returns the block column corresponding to columnID."""
|
|
return self.data[columnID::self.blockSizeR]
|
|
|
|
def mergeColumn(self, columnID, column):
|
|
"""It merges (OR) the existing column with the received one."""
|
|
self.data[columnID::self.blockSizeR] |= column
|
|
|
|
def repairColumn(self, id):
|
|
"""It repairs the entire column if it has at least blockSizeCK ones.
|
|
Returns: list of repaired segments
|
|
"""
|
|
line = self.data[id::self.blockSizeR]
|
|
success = line.count(1)
|
|
repairedSamples = 0
|
|
if success >= self.blockSizeCK:
|
|
ret = ~line
|
|
self.data[id::self.blockSizeR] = 1
|
|
repairedSamples = len(line) - success
|
|
else:
|
|
ret = zeros(self.blockSizeC)
|
|
return ret, repairedSamples
|
|
|
|
def getRow(self, rowID):
|
|
"""It returns the block row corresponding to rowID."""
|
|
return self.data[rowID*self.blockSizeR:(rowID+1)*self.blockSizeR]
|
|
|
|
def mergeRow(self, rowID, row):
|
|
"""It merges (OR) the existing row with the received one."""
|
|
self.data[rowID*self.blockSizeR:(rowID+1)*self.blockSizeR] |= row
|
|
|
|
def repairRow(self, id):
|
|
"""It repairs the entire row if it has at least blockSizeRK ones.
|
|
Returns: list of repaired segments.
|
|
"""
|
|
line = self.data[id*self.blockSizeR:(id+1)*self.blockSizeR]
|
|
success = line.count(1)
|
|
repairedSamples = 0
|
|
if success >= self.blockSizeRK:
|
|
ret = ~line
|
|
self.data[id*self.blockSizeR:(id+1)*self.blockSizeR] = 1
|
|
repairedSamples = len(line) - success
|
|
else:
|
|
ret = zeros(self.blockSizeR)
|
|
return ret, repairedSamples
|
|
|
|
def print(self):
|
|
"""It prints the block in the terminal (outside of the logger rules))."""
|
|
dash = "-" * (self.blockSizeR+2)
|
|
print(dash)
|
|
for i in range(self.blockSizeC):
|
|
line = "|"
|
|
for j in range(self.blockSizeR):
|
|
line += "%i" % self.data[(i*self.blockSizeR)+j]
|
|
print(line+"|")
|
|
print(dash)
|