mirror of
https://github.com/QuantumPackage/qp2.git
synced 2024-11-03 20:13:43 +01:00
133 lines
4.2 KiB
Python
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()
|