blob: de8ffbc93c5a2dda2409a54a443b19ceae145e05 (
plain)
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
#!/usr/bin/env python
#
# Copyright 2011 Free Software Foundation, Inc.
#
# This file is part of GNU Radio
#
# SPDX-License-Identifier: GPL-3.0-or-later
#
#
class GrayCodeGenerator(object):
"""
Generates and caches gray codes.
"""
def __init__(self):
self.gcs = [0, 1]
# The last power of two passed through.
self.lp2 = 2
# The next power of two that will be passed through.
self.np2 = 4
# Current index
self.i = 2
def get_gray_code(self, length):
"""
Returns a list of gray code of given length.
"""
if len(self.gcs) < length:
self.generate_new_gray_code(length)
return self.gcs[:length]
def generate_new_gray_code(self, length):
"""
Generates new gray code and places into cache.
"""
while len(self.gcs) < length:
if self.i == self.lp2:
# if i is a power of two then gray number is of form 1100000...
result = self.i + self.i // 2
else:
# if not we take advantage of the symmetry of all but the last bit
# around a power of two.
result = self.gcs[2 * self.lp2 - 1 - self.i] + self.lp2
self.gcs.append(result)
self.i += 1
if self.i == self.np2:
self.lp2 = self.i
self.np2 = self.i * 2
_gray_code_generator = GrayCodeGenerator()
gray_code = _gray_code_generator.get_gray_code
|
