54 lines
1.5 KiB
Python
54 lines
1.5 KiB
Python
#!/usr/bin/env python2
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# TRX Toolkit
|
|
# Power measurement emulation for BB
|
|
#
|
|
# (C) 2017 by Vadim Yanitskiy <axilirator@gmail.com>
|
|
#
|
|
# All Rights Reserved
|
|
#
|
|
# 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 2 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, write to the Free Software Foundation, Inc.,
|
|
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
|
|
from random import randint
|
|
|
|
class FakePM:
|
|
# Freq. list for good power level
|
|
bts_list = []
|
|
|
|
def __init__(self, noise_min, noise_max, bts_min, bts_max):
|
|
# Save power level ranges
|
|
self.noise_min = noise_min
|
|
self.noise_max = noise_max
|
|
self.bts_min = bts_min
|
|
self.bts_max = bts_max
|
|
|
|
def measure(self, bts):
|
|
if bts in self.bts_list:
|
|
return randint(self.bts_min, self.bts_max)
|
|
else:
|
|
return randint(self.noise_min, self.noise_max)
|
|
|
|
def update_bts_list(self, new_list):
|
|
self.bts_list = new_list
|
|
|
|
def add_bts_list(self, add_list):
|
|
self.bts_list += add_list
|
|
|
|
def del_bts_list(self, del_list):
|
|
for item in del_list:
|
|
if item in self.bts_list:
|
|
self.bts_list.remove(item)
|