9
1
mirror of https://github.com/QuantumPackage/qp2.git synced 2024-11-03 20:13:43 +01:00
qp2/scripts/utility/qp_bitmasks.py

133 lines
4.2 KiB
Python

#!/usr/bin/env python3
BIT_KIND_SIZE=64
def int_to_string(s):
"""Transforms any integer to a string representation
>>> print int_to_string(10)
1010
>>> print int_to_string(1024)
10000000000
>>> print int_to_string(123456789)
111010110111100110100010101
>>> print int_to_string(12345678912345678910)
1010101101010100101010011000111110000001011001010010010000111110
>>> print int_to_string(0)
0
>>> print int_to_string(-128)
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "qp_bitmasks.py", line 23, in int_to_string
assert s>=0
AssertionError
"""
assert type(s) in (int, long)
assert s>=0
return '{s:0b}'.format(s=s)
def string_to_bitmask(s,bit_kind_size=BIT_KIND_SIZE):
"""Transforms a string into an bitmask
>>> print string_to_bitmask(10)
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "qp_bitmasks.py", line 30, in string_to_bitmask
assert type(s) == str
AssertionError
>>> print string_to_bitmask('10')
['0000000000000000000000000000000000000000000000000000000000000010']
>>> print string_to_bitmask('1010'*64)
['1010101010101010101010101010101010101010101010101010101010101010', '1010101010101010101010101010101010101010101010101010101010101010', '1010101010101010101010101010101010101010101010101010101010101010', '1010101010101010101010101010101010101010101010101010101010101010']
>>> print string_to_bitmask('1010'*64,4)
['1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010', '1010']
"""
assert type(s) == str
assert bit_kind_size > 0
while len(s) % bit_kind_size != 0:
s = '0'+s
return [ s[i:i+bit_kind_size] for i in range(0,len(s),bit_kind_size) ]
def int_to_bitmask(s,bit_kind_size=BIT_KIND_SIZE):
"""Transforms a string into an bitmask
>>> int_to_bitmask(1)
['0000000000000000000000000000000000000000000000000000000000000001']
>>> int_to_bitmask(-1)
['1111111111111111111111111111111111111111111111111111111111111111']
>>> int_to_bitmask(10)
['0000000000000000000000000000000000000000000000000000000000001010']
>>> int_to_bitmask(-10)
['1111111111111111111111111111111111111111111111111111111111110110']
>>>
"""
assert type(s) in (int, long)
if s < 0:
s = s + (1 << bit_kind_size)
return ['{s:0{width}b}'.format(s=s,width=bit_kind_size)]
class BitMask(object):
"""
>>> A = BitMask( [ -127,47 ], bit_kind_size=8)
>>> print A
['10000001', '00101111']
>>> print A[0]
-127
>>> print A[1]
47
>>> A[0] = 127
>>> print A
['01111111', '00101111']
>>> A[1] = '100'
>>> print A
['01111111', '00000100']
>>> A[1] = '10000001'
>>> print A
['01111111', '10000001']
>>> print A[1]
-127
"""
def __init__(self,l=[],bit_kind_size=BIT_KIND_SIZE):
self.bit_kind_size = bit_kind_size
self._data_int = l
@property
def N_int(self):
return len(self._data_int)
def __getitem__(self,i):
return self._data_int[i]
def __setitem__(self,i,value):
if type(value) in (int,long):
self._data_int[i] = value
elif type(value) == str:
s = string_to_bitmask(value,bit_kind_size=self.bit_kind_size)[0]
if s[0] == '0':
self._data_int[i] = int(s,2)
else:
s = s.replace('0','.').replace('1','0').replace('.','1')
self._data_int[i] = -int(s,2)-1
def __repr__(self):
result = []
for i in self._data_int:
result += int_to_bitmask(i,bit_kind_size=self.bit_kind_size)
return str(result)
def excitation_degree(l_a,l_b):
'''
excitation_degree([895],[959])
>> 1
'''
return sum(bin(a ^ b).count("1") for a,b in zip(l_a,l_b) ) // 2
if __name__ == '__main__':
import doctest
doctest.testmod()