git.gag.com Git - debian/gnuradio/blob - gnuradio-core/src/python/gnuradio/blks2impl/psk.py

   1 #
   2 # Copyright 2005,2006 Free Software Foundation, Inc.
   3 #
   4 # This file is part of GNU Radio
   5 #
   6 # GNU Radio is free software; you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation; either version 3, or (at your option)
   9 # any later version.
  10 #
  11 # GNU Radio is distributed in the hope that it will be useful,
  12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 # GNU General Public License for more details.
  15 #
  16 # You should have received a copy of the GNU General Public License
  17 # along with GNU Radio; see the file COPYING.  If not, write to
  18 # the Free Software Foundation, Inc., 51 Franklin Street,
  19 # Boston, MA 02110-1301, USA.
  20 #
  21
  22 from math import pi, sqrt, log10
  23 import math, cmath
  24
  25 # The following algorithm generates Gray coded constellations for M-PSK for M=[2,4,8]
  26 def make_gray_constellation(m):
  27     # number of bits/symbol (log2(M))
  28     k = int(log10(m) / log10(2.0))
  29
  30     coeff = 1
  31     const_map = []
  32     bits = [0]*3
  33     for i in range(m):
  34         # get a vector of the k bits to use in this mapping
  35         bits[3-k:3] = [((i&(0x01 << k-j-1)) >> k-j-1) for j in range(k)]
  36
  37         theta = -(2*bits[0]-1)*(2*pi/m)*(bits[0]+abs(bits[1]-bits[2])+2*bits[1])
  38         re = math.cos(theta)
  39         im = math.sin(theta)
  40         const_map.append(complex(re, im))   # plug it into the constellation
  41
  42     # return the constellation; by default, it is normalized
  43     return const_map
  44
  45 # This makes a constellation that increments around the unit circle
  46 def make_constellation(m):
  47     return [cmath.exp(i * 2 * pi / m * 1j) for i in range(m)]
  48
  49 # Common definition of constellations for Tx and Rx
  50 constellation = {
  51     2 : make_constellation(2),           # BPSK
  52     4 : make_constellation(4),           # QPSK
  53     8 : make_constellation(8)            # 8PSK
  54     }
  55
  56 gray_constellation = {
  57     2 : make_gray_constellation(2),           # BPSK
  58     4 : make_gray_constellation(4),           # QPSK
  59     8 : make_gray_constellation(8)            # 8PSK
  60     }
  61
  62 # -----------------------
  63 # Do Gray code
  64 # -----------------------
  65 # binary to gray coding -- constellation does Gray coding
  66 binary_to_gray = {
  67     2 : range(2),
  68     4 : [0,1,3,2],
  69     8 : [0, 1, 3, 2, 7, 6, 4, 5]
  70     }
  71
  72 # gray to binary
  73 gray_to_binary = {
  74     2 : range(2),
  75     4 : [0,1,3,2],
  76     8 : [0, 1, 3, 2, 6, 7, 5, 4]
  77     }
  78
  79 # -----------------------
  80 # Don't Gray code
  81 # -----------------------
  82 # identity mapping
  83 binary_to_ungray = {
  84     2 : range(2),
  85     4 : range(4),
  86     8 : range(8)
  87     }
  88
  89 # identity mapping
  90 ungray_to_binary = {
  91     2 : range(2),
  92     4 : range(4),
  93     8 : range(8)
  94     }