summaryrefslogtreecommitdiffstats
path: root/dpd/main.py
blob: 19460ddd8d294f40bc84ff586f59887f994d06db (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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# DPD Calculation Engine main file.
#
# http://www.opendigitalradio.org
# Licence: The MIT License, see notice at the end of this file

"""This Python script is the main file for ODR-DabMod's DPD Computation Engine.
This engine calculates and updates the parameter of the digital
predistortion module of ODR-DabMod."""

import datetime
import os

import logging
dt = datetime.datetime.now().isoformat()
logging_path = "/tmp/dpd_{}".format(dt).replace(".","_").replace(":","-")
os.makedirs(logging_path)
logging.basicConfig(format='%(asctime)s - %(module)s - %(levelname)s - %(message)s',
                    datefmt='%Y-%m-%d %H:%M:%S',
                    filename='{}/dpd.log'.format(logging_path),
                    filemode='w',
                    level=logging.DEBUG)

import traceback
import src.Measure as Measure
import src.Model as Model
import src.Adapt as Adapt
import argparse

parser = argparse.ArgumentParser(description="DPD Computation Engine for ODR-DabMod")
parser.add_argument('--port', default='50055',
        help='port of DPD server to connect to (default: 50055)',
        required=False)
parser.add_argument('--rc-port', default='9400',
        help='port of ODR-DabMod ZMQ Remote Control to connect to (default: 9400)',
        required=False)
parser.add_argument('--samplerate', default='8192000',
        help='Sample rate',
        required=False)
parser.add_argument('--coefs', default='poly.coef',
        help='File with DPD coefficients, which will be read by ODR-DabMod',
        required=False)
parser.add_argument('--txgain', default=65,
                    help='TX Gain',
                    required=False,
                    type=int)
parser.add_argument('--rxgain', default=30,
                    help='TX Gain',
                    required=False,
                    type=int)
parser.add_argument('--samps', default='10240',
        help='Number of samples to request from ODR-DabMod',
        required=False)
parser.add_argument('-i', '--iterations', default='1',
        help='Number of iterations to run',
        required=False)
parser.add_argument('-l', '--load-poly',
        help='Load existing polynomial',
        action="store_true")

cli_args = parser.parse_args()

port = int(cli_args.port)
port_rc = int(cli_args.rc_port)
coef_path = cli_args.coefs
txgain = cli_args.txgain
rxgain = cli_args.rxgain
num_req = int(cli_args.samps)
samplerate = int(cli_args.samplerate)
num_iter = int(cli_args.iterations)

meas = Measure.Measure(samplerate, port, num_req)

adapt = Adapt.Adapt(port_rc, coef_path)
coefs_am, coefs_pm = adapt.get_coefs()
if cli_args.load_poly:
    model = Model.Model(coefs_am, coefs_pm)
else:
    model = Model.Model([1, 0, 0, 0, 0], [0, 0, 0, 0, 0])
adapt.set_txgain(txgain)
adapt.set_rxgain(rxgain)

tx_gain   = adapt.get_txgain()
rx_gain   = adapt.get_rxgain()
dpd_coefs_am, dpd_coefs_pm = adapt.get_coefs()
logging.info(
    "TX gain {}, RX gain {}, dpd_coefs_am {}, dpd_coefs_pm {}".format(
        tx_gain, rx_gain, dpd_coefs_am, dpd_coefs_pm
    )
)

for i in range(num_iter):
    try:
        txframe_aligned, tx_ts, rxframe_aligned, rx_ts = meas.get_samples()
        logging.debug("tx_ts {}, rx_ts {}".format(tx_ts, rx_ts))
        coefs_am, coefs_pm = model.get_next_coefs(txframe_aligned, rxframe_aligned)
        adapt.set_coefs(coefs_am, coefs_pm)
    except Exception as e:
        logging.info("Iteration {} failed.".format(i))
        logging.info(traceback.format_exc())

# The MIT License (MIT)
#
# Copyright (c) 2017 Andreas Steger, Matthias P. Braendli
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in all
# copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.