129 lines
4.1 KiB
Python
Executable File
129 lines
4.1 KiB
Python
Executable File
#!/usr/bin/python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
|
|
import re
|
|
import os
|
|
import time
|
|
import json
|
|
import argparse
|
|
from numpy import nanstd, nanmedian, sqrt
|
|
from libExtraction import separateMAC
|
|
from libLocalization import deriveLocation, get_distance
|
|
|
|
|
|
def get_known_locs(results, args):
|
|
database = json.load(open(args['json'], 'r'))
|
|
for mac in database.keys():
|
|
if mac not in results:
|
|
del database[mac]
|
|
for mac in results.keys():
|
|
if mac not in database:
|
|
del results[mac]
|
|
return database
|
|
|
|
|
|
def wrapper(args):
|
|
if not os.path.isfile(args['filepath']):
|
|
return
|
|
results, nameLine, startLine, endLine = separateMAC(args['filepath'])
|
|
args['config_entry'] = get_known_locs(results, args)
|
|
# method 1: average and then compute single loc
|
|
final_result = {}
|
|
for mac in results:
|
|
dists = []
|
|
for line in results[mac]:
|
|
try:
|
|
# tmp = float(line.split(',')[4])
|
|
# ugly hack
|
|
# if mac == '34:f6:4b:5e:69:1f':
|
|
# tmp = tmp * 0.7927 + 483.3157 # med school fit for 1f
|
|
# elif mac == '34:f6:4b:5e:69:0b':
|
|
# tmp = tmp * 0.6927 + 400.3157 # med school fit for 0b
|
|
tmp = float(line.split(',')[1]) # already fitted result
|
|
dists.append(tmp)
|
|
except Exception as e:
|
|
print(e)
|
|
final_result[mac] = (nanmedian(dists), nanstd(dists))
|
|
print('est:', final_result)
|
|
loc = deriveLocation(args, final_result)
|
|
print('loc:', loc)
|
|
# # method 2: do localization first for every pair and then average
|
|
# all_locs = {}
|
|
# if results:
|
|
# keys = results.keys()
|
|
# idxs = [0] * len(keys)
|
|
# locs = []
|
|
# while all([idxs[i] < len(results[keys[i]]) for i in range(len(idxs))]):
|
|
# lines = [results[keys[i]][idxs[i]] for i in range(len(idxs))]
|
|
# times = [float(x.split(',')[7]) for x in lines]
|
|
# maxT = max(times)
|
|
# if all([abs(t - maxT) < 0.01 for t in times]):
|
|
# dists = [
|
|
# (float(x.split(',')[1]), sqrt(float(x.split(',')[5])))
|
|
# for x in lines
|
|
# ]
|
|
# loc = deriveLocation(args, dict(zip(keys, dists)))
|
|
# print('{0:.4f},{1:.4f}'.format(loc[0], loc[1]))
|
|
# locs.append(loc)
|
|
# for i in range(len(idxs)):
|
|
# idxs[i] += 1
|
|
# else:
|
|
# for i in range(len(idxs)):
|
|
# if abs(times[i] - maxT) > 0.01:
|
|
# idxs[i] += 1
|
|
# x, y = zip(*locs)
|
|
# loc = (nanmedian(x), nanmedian(y))
|
|
# locstd = (nanstd(x), nanstd(y))
|
|
# print(loc)
|
|
# print(locstd)
|
|
match = re.search(r"static_([0-9.]+)_([0-9.]+)_", args['filepath'])
|
|
if match:
|
|
trueX = float(match.group(1)) * 100
|
|
trueY = float(match.group(2)) * 100
|
|
true_result = {}
|
|
for mac in args['config_entry']:
|
|
mac_loc = args['config_entry'][mac]['location'].split(',')
|
|
mac_loc = (float(mac_loc[0]), float(mac_loc[1]))
|
|
true_result[mac] = (get_distance(mac_loc, (trueX, trueY)), 0)
|
|
print('true:', true_result)
|
|
err = get_distance(loc, (trueX, trueY))
|
|
print('err:', err)
|
|
|
|
|
|
def main():
|
|
p = argparse.ArgumentParser(description='separate data from MAC addr')
|
|
p.add_argument(
|
|
'filepath',
|
|
help="file path"
|
|
)
|
|
p.add_argument(
|
|
'--outfp', '-f',
|
|
default=None,
|
|
help="if set, will write raw fetched data to file"
|
|
)
|
|
p.add_argument(
|
|
'--json', '-j',
|
|
default='config_entry.default',
|
|
help="load a config json file"
|
|
)
|
|
p.add_argument(
|
|
'--verbose', '-v',
|
|
default=False,
|
|
action="store_true",
|
|
help="if set, show detailed messages"
|
|
)
|
|
try:
|
|
args = vars(p.parse_args())
|
|
except Exception as e:
|
|
print(str(e))
|
|
sys.exit()
|
|
args['time_of_exec'] = int(time.time())
|
|
# TODO: manually add bounds
|
|
args['loc_bounds'] = {'y_min': 0}
|
|
wrapper(args)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|