Beispiel #1
0
	def load_wifi_data_new(self, dir_path, save_path="wifi_floor_temp.npz", ref_list=[]):
		load_in = LoadWifiData()
		file_list = os.listdir(dir_path)
		floor_data = {}
		for floor in file_list:
			print floor
			fl_path = os.path.join(dir_path, floor)
			if os.path.isfile(fl_path):
				continue
			floor_file_list = os.listdir(fl_path)
			for file_name in floor_file_list:
				if file_name.endswith("wifi"):
					name, ext = os.path.splitext(file_name)
					print name
					wifi_file = name + ".wifi"
					wifi_file = os.path.join(fl_path, wifi_file)
					floor_data[floor] = load_in.extract_wifi(wifi_file, ref_list)
		t_bind = []
		for x in floor_data:
			t_bind += floor_data[x].wifi_list
		wifi_bld_list = list(set(t_bind))

		TD = 0

		for x in floor_data:
			D = floor_data[x].wifi_matrix.shape[0]
			TD += D
			L = len(wifi_bld_list)
			t_mat = zeros((D, L))
			for i, j in enumerate(floor_data[x].wifi_list):
				ind = wifi_bld_list.index(j)
				t_mat[:, ind] = floor_data[x].wifi_matrix[:, i]
			floor_data[x].wifi_bld_mat = t_mat

		savez(save_path, floor=floor_data, wifilist=wifi_bld_list, entry_num=TD)
		return floor_data, wifi_bld_list, TD