python-elmo/elmo/utils.py

import numpy as np

### Binary operations
def hweight(n):
    """ Return the Hamming weight of 'n' """
    c = 0
    while n>0:
        c += (n & 1)
        n >>= 1
    return c

def hdistance(x,y):
    """ Return the Hamming distance between 'x' and 'y' """
    return hweight(x^y)

def binary_writing(n, nb_bits=32, with_hamming=False):
    """ Return the binary writing 'w' of 'n' with 'nb_bits'
    If with_hamming is True, return a couple (w, h) with 'h' the Hamming weight of 'n'
    """
    n = np.array(n)
    w, h = np.zeros((nb_bits, len(n))), np.zeros((len(n)))

    for ind in range(nb_bits):
        w[ind] = (n & 1)
        h += w[ind]

        n >>= 1
        ind += 1
    
    return (w, h) if with_hamming else w
    
### Conversion
def to_hex(v, nb_bits=16):
    """ Convert the value 'v' into a hexadecimal string (without the prefix 0x)"""
    try:
        v_hex = v.hex()
    except AttributeError:
        v_hex = hex(v)[2:]
    return '0'*(nb_bits//4-len(v_hex)) + v_hex

def split_octet(hexstr):
    """ Split a hexadecimal string (without the prefix 0x)
    into a list of bytes described with a 2-length hexadecimal string
    """
    return [hexstr[i:i+2] for i in range(0, len(hexstr), 2)]

def to_signed_hex(v, nb_bits=16):
    """ Convert the signed value 'v'
    into a list of bytes described with a 2-length hexadecimal string
    """
    try:
        return split_octet(to_hex(v & ((1<<nb_bits)-1), nb_bits=nb_bits))
    except TypeError as err:
        raise TypeError('Error to transform a <{}> into signed hex.'.format(type(v))) from err

### Write function
def write(_input, uint, nb_bits=16):
    """ Write in '_input' the value 'uint' by writing each byte
    with hexadecimal format in a new line
    """
    uint = to_signed_hex(uint, nb_bits=nb_bits)
    for i in range(nb_bits//8):
        _input.write(uint[i]+'\n')
    
def write_list(_input, uint_list, nb_bits=16):
    """ Write in '_input' the list of values 'uint_list' by writing each byte
    with hexadecimal format in a new line
    """
    for uint in uint_list:
        write(_input, uint, nb_bits=nb_bits)

### Print Utils
class Color:
    """ Color codes to print colored text in stdout """
    HEADER = '\033[95m'
    OKBLUE = '\033[94m'
    OKCYAN = '\033[96m'
    OKGREEN = '\033[92m'
    WARNING = '\033[93m'
    FAIL = '\033[91m'
    ENDC = '\033[0m'
    BOLD = '\033[1m'
    UNDERLINE = '\033[4m'