Brice Colombier / CASCADE

flip_bits.py 750 Bytes
edit raw blame history 
  1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
# Author: Brice Colombier
#         Laboratoire Hubert Curien
#         42000 Saint-Etienne - France
# Contact: b.colombier@univ-st-etienne.fr
# Project: CASCADE
# File: flip_bits.py
# Date : 2016-10-12


def flip_bits(message, indices_to_flip, indices):
    
    """Flip the message bits at specific indices.

    >>> flip_bits([1, 1, 1, 1], [1])
    [1, 0, 1, 1]
    >>> flip_bits([0, 1, 0, 1], [2], [2, 0, 3, 1])
    [1, 1, 0, 1]
    
    """

    if indices:
        for i in indices_to_flip:
            message[indices.index(i)] ^= 1
    else:
        for i in indices_to_flip:
            message[i] ^= 1
    
            
if __name__ == "__main__":
    message = [0, 1, 0, 1]
    print(message)
    flip_bits(message, [2])
    print(message)