QuantumPackage/scripts/utility/qp_bitmasks.py

#!/usr/bin/env python2

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()