#!/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 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('--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 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(70) adapt.set_rxgain(30) 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): 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) # 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.