crossDetectionStub.py

#! /usr/bin/env python
#
# LICENSE:
# Copyright (C) 2016  Neal Patwari
#
#    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/>.
# 
# Author: Neal Patwari, neal.patwari@gmail.com
#
#
# Version History:
#
# Version 1.0:  Initial Release 20 Sept 2016
#


import ast
import getopt
import sys

import numpy as np

import rss
import rti


# Default inputs if not provided on command line
linkCombosToPlot = []
numNodes = 10
infile = sys.stdin
fromStdin = True

# Accept command line inputs to list the links to use; the input file
myopts, args = getopt.getopt(sys.argv[1:], "l:n:f:")
for o, a in myopts:
    if o == "-l":
        inputList = ast.literal_eval(a)
        linkCombosToPlot.append(inputList)
    elif o == "-n":
        numNodes = int(a)
    elif o == "-f":
        infile = open(a)
        fromStdin = False

# If no number of nodes given,
if numNodes <= 1:
    sys.stderr.write('Usage: %s -n numberOfNodes -l "[tx,rx,ch]" ' % sys.argv[0])
    sys.stderr.write('numberOfNodes is required and must be greater than 1')

# If no link combination provided, just plot [tx=1,rx=2,ch=0]
if not linkCombosToPlot:
    linkCombosToPlot.append([1, 2, 0])
links = len(linkCombosToPlot)

print("Plotting columns for links:")
print(linkCombosToPlot)
print("Number of nodes:" + str(numNodes))

# Parameters you may change:
#   plotSkip:  Refresh the plot after this many data lines are read
#   buffL:     buffer length, ie, how much data to plot.
#   startSkip: A serial port seems to have a "memory" of several lines, 
#              which were saved from the previous experiment run.  
#              ** Must be greater than 0.
buffL = 4
startSkip = 1

# remove junk from start of file.
for i in range(startSkip):
    line = infile.readline()

# Use the most recent line to determine how many columns (streams) there are.
lineInt = [float(i) for i in line.split()]
columns = len(lineInt)
rss_dB = [int(f) for f in lineInt[:-1]]  # take all columns except for last column
numLinks = len(rss_dB)
numChs = numLinks / ((numNodes - 1) * numNodes)
nodeList = list(range(1, numNodes + 1))
channelList = list(range(numChs))

# Convert the [tx,rx,ch] combos into column numbers
linkNumList = []
for l in linkCombosToPlot:
    linkNumList.append(rss.linkNumForTxRxChLists(l[0], l[1], l[2], nodeList, channelList))

buff = []
for i in range(links):
    buff.append(rti.FixedLenBuffer([0] * buffL))

# Run forever, adding lines as they are available.
counter = 0
keepReading = True
while keepReading:
    print("counter = " + str(counter))
    line = infile.readline()
    # If at the "end of file", keep reading if reading from stdin.
    if not line:
        keepReading = fromStdin
        continue
    while line[-1] != '\n':  # If line is incomplete, add to it.
        line += infile.readline()

    # Get the integers from the line string
    data = [float(i) for i in line.split()]
    rss_dB = [int(f) for f in data[:-1]]  # take columns except for last (time) col

    # Append the queue for each stream with the newest RSS and Timestamps
    for i, linkNum in enumerate(linkNumList):
        # data > -10 indicates no data measured.  Don't include a new value.
        if rss_dB[linkNum] < -10:
            buff[i].append(rss_dB[linkNum])

    # Calculate the variance on all monitored links
    varVec = np.array([a.var() for a in buff])
    print(varVec)

    counter += 1