64 lines
1.5 KiB
Python
64 lines
1.5 KiB
Python
|
#!/usr/bin/python
|
||
|
|
# -*- encoding: utf-8 -*-
|
||
|
|
|
||
|
|
# ______ _
|
||
|
|
# / _____) _ | |
|
||
|
|
# ( (____ _____ ____ _| |_ _____ ____| |__
|
||
|
|
# \____ \| ___ | (_ _) ___ |/ ___) _ \
|
||
|
|
# _____) ) ____| | | || |_| ____( (___| | | |
|
||
|
|
# (______/|_____)_|_|_| \__)_____)\____)_| |_|
|
||
|
|
# (C)2020 Semtech
|
||
|
|
#
|
||
|
|
# Description:
|
||
|
|
# Spectral Scan CSV result file plot - v0.3.0
|
||
|
|
#
|
||
|
|
# License: Revised BSD License, see LICENSE.TXT file include in the project
|
||
|
|
|
||
|
|
#Library importation
|
||
|
|
import pylab as pl
|
||
|
|
import numpy as np
|
||
|
|
import csv
|
||
|
|
import sys
|
||
|
|
|
||
|
|
#Read argument
|
||
|
|
if len(sys.argv) >= 2:
|
||
|
|
filename = sys.argv[1]
|
||
|
|
else:
|
||
|
|
print ("Usage: %s <filename>" %sys.argv[0])
|
||
|
|
sys.exit()
|
||
|
|
|
||
|
|
#Initiate array
|
||
|
|
rssi = []
|
||
|
|
freq = []
|
||
|
|
|
||
|
|
#Process .csv file
|
||
|
|
with open(filename, 'r') as csvfile:
|
||
|
|
reader = csv.reader(csvfile, delimiter=',', quotechar='|')
|
||
|
|
for row in reader:
|
||
|
|
f=int(row[0])//1000 #frequency
|
||
|
|
freq.append(f)
|
||
|
|
rssi_line=[]
|
||
|
|
rssi_val=[]
|
||
|
|
for k in range(1,(len(row)-1)//2):
|
||
|
|
rssi_line.append(int(row[k*2]))
|
||
|
|
rssi_val.append(int(row[k*2-1]))
|
||
|
|
rssi.append(rssi_line)
|
||
|
|
|
||
|
|
#Set x to frequency axis and y to signal level axis
|
||
|
|
A = np.array(rssi).T
|
||
|
|
A = np.flip(A,0)
|
||
|
|
rssi_val =np.flip(rssi_val,0)
|
||
|
|
|
||
|
|
#Find max value to scale heatmap then plot.
|
||
|
|
maxx = max(max(rssi))
|
||
|
|
print(maxx, A)
|
||
|
|
h=pl.imshow(A , cmap='afmhot', vmin=0, vmax=maxx, aspect='auto', origin='lower', interpolation='nearest')
|
||
|
|
pl.xticks(range(0,len(freq),10),freq[::10])
|
||
|
|
pl.yticks(range(0,len(rssi[1]),2),rssi_val[0::2])
|
||
|
|
pl.show()
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|