Latenight experimenting

frequency_power.jl

 using Pkg; Pkg.activate(".");
 using FFTW, DSP, Plots, DelimitedFiles
 
-function plot_spectogram(s, n=div(length(s),8), noverlap=div(n,2))
+function plot_spectrogram(s, n=div(length(s),8), noverlap=div(n,2))
 	spec = spectrogram(s, n, noverlap)
-	heatmap(spec.time, spec.freq, spec.power)
+	n = 300
+	spec = spectrogram(s, n, n-1)
+	power = spec.power./sum(spec.power, dims=1)
+	heatmap(spec.time, spec.freq, power)
 end
-
-function read_data(dataname, gyro=false)
-	if !gyro
-		return readdlm("data/$(dataname)_acc.txt", Float64)
-	else
-		return (readdlm("data/$(dataname)_acc.txt", Float64),
-			readdlm("data/$(dataname)_gyr.txt", Float64))
+function plot_periodogram(s)
+	peri = periodogram(s)
+	plot(peri.freq, peri.power)
 end
+
+function read_data(dataname, sensor)
+	return readdlm("data/$(dataname)_$(sensor).txt", Float64)
 end
+
+read_acc(dataname) = read_data(dataname, "acc")
+read_gyr(dataname) = read_data(dataname, "gyr")
+
+xyz(data) = data[:,2:4]
+ypr(data) = data[:,2:4]
+
+normalize_data(A) = sqrt.(sum(A.^2, dims=2)) # TODO squareroot?
+mean(v) = sum(v)/length(v)
+detrend(v) = v .- mean(v)
+
+acc_data(dataname) = read_acc(dataname) |> xyz |> normalize_data |> vec |> detrend
+acc_spectogram(dataname) = acc_data(dataname) |> plot_spectrogram
+acc_periodogram(dataname) = acc_data(dataname) |> plot_periodogram
+
+gyr_data(dataname) = read_gyr(dataname) |> ypr |> normalize_data |> vec |> detrend
+gyr_spectogram(dataname) = gyr_data(dataname) |> plot_spectrogram
+gyr_periodogram(dataname) = gyr_data(dataname) |> plot_periodogram