osmo-gsm-tester/src/osmo_gsm_tester/obj/rfemu_gnuradio_zmq.py

67 lines
2.3 KiB
Python

# osmo_gsm_tester: class defining a RF emulation object implemented using SRS ENB stdin interface
#
# Copyright (C) 2020 by sysmocom - s.f.m.c. GmbH
#
# Author: Pau Espin Pedrol <pespin@sysmocom.de>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
from ..core import log
from .rfemu import RFemulation
from .gnuradio_zmq_broker import GrBroker
class RFemulationGnuradioZmq(RFemulation):
##############
# PROTECTED
##############
def __init__(self, conf):
super().__init__(conf, 'gnuradio_zmq')
self.broker = None
self.ctrl_port = 5005
self.cell_id = int(conf.get('cell_id'))
if self.cell_id is None:
raise log.Error('No "cell_id" attribute provided in rfemu conf!')
self.enb = conf.get('enb')
if self.enb is None:
raise log.Error('No "srsenb" attribute provided in rfemu conf!')
self.set_name('%s_%s_%d' % (self.name(), self.enb.name(), self.cell_id))
self.testenv = self.enb.testenv
self.configure()
def __del__(self):
if self.broker:
self.broker.unref()
self.broker = None
self.enb = None
self.testenv = None
def configure(self):
self.broker = GrBroker.ref()
#############################
# PUBLIC (test API included)
#############################
def set_attenuation(self, db):
for cell in self.enb.gen_conf['enb']['cell_list']:
if int(cell['cell_id']) == self.cell_id:
# convert negative dB to amplitude
amp = pow(10, -db/20.0)
self.broker.cmd_set_relative_gain_on_local_port(cell['zmq_enb_peer_port'], amp)
break
def get_max_attenuation(self):
return 200
# vim: expandtab tabstop=4 shiftwidth=4