111 lines
2.4 KiB
Go
111 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"log"
|
|
"os/exec"
|
|
|
|
"git.dotya.ml/wanderer/ak9im/p2/stats"
|
|
)
|
|
|
|
var (
|
|
datafile = flag.String("datafile", "", "read data from this file")
|
|
vis = flag.Bool("vis", false, "run 'python visualise.py' to produce visualisations")
|
|
)
|
|
|
|
func run() error {
|
|
flag.Parse()
|
|
|
|
if *datafile != "" {
|
|
data, err := readFile(datafile)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
u := 0
|
|
y := 1
|
|
|
|
meanU := stats.Mean(data[u])
|
|
meanY := stats.Mean(data[y])
|
|
varianceU := stats.Variance(data[u])
|
|
varianceY := stats.Variance(data[y])
|
|
|
|
maxShift := 0.1
|
|
autocorrelationU := stats.Autocorrelate(data[u], maxShift)
|
|
autocorrelationY := stats.Autocorrelate(data[y], maxShift)
|
|
|
|
mutCorrelationUY, err := stats.MutCorrelate(data[u], data[y], maxShift)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
mutCorrelationYU, err := stats.MutCorrelate(data[y], data[u], maxShift)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
cov := stats.Covariance(data[u], data[y])
|
|
|
|
// impulseFunc, err := stats.ImpulseFunction(autocorrelationU, mutCorrelationUY)
|
|
impulseFunc, err := stats.ImpulseFunction(autocorrelationU, autocorrelationY)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
log.Printf("len(data): %d", len(data[u]))
|
|
|
|
log.Printf("means - u: %v, y: %v", meanU, meanY)
|
|
|
|
log.Printf("variance - u: %v, y: %v", varianceU, varianceY)
|
|
|
|
log.Printf("len(autocorrelationU): %d", len(autocorrelationU))
|
|
log.Printf("autocorrelationU: %v", autocorrelationU)
|
|
log.Printf("autocorrelationY: %v", autocorrelationY)
|
|
log.Printf("mutual correlation U,Y: %v", mutCorrelationUY)
|
|
log.Printf("mutual correlation Y,U: %v", mutCorrelationYU)
|
|
log.Printf("covariance U,Y: %v", cov)
|
|
log.Printf("correlation U,Y: %v", stats.Correlation(data[u], data[y]))
|
|
|
|
log.Printf("len(impulseFunc): %d", len(impulseFunc))
|
|
log.Printf("impulseFunc: %v", impulseFunc)
|
|
|
|
err = saveStuff(
|
|
meanU, meanY,
|
|
varianceU, varianceY,
|
|
cov,
|
|
autocorrelationU, autocorrelationY,
|
|
mutCorrelationUY, mutCorrelationYU,
|
|
impulseFunc,
|
|
)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
if *vis {
|
|
err := visualise()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func visualise() error {
|
|
red := "\033[31m"
|
|
cyan := "\033[36m"
|
|
reset := "\033[0m"
|
|
f := "visualise.py"
|
|
|
|
log.Printf("running %s`python %s`%s", cyan, f, reset)
|
|
|
|
out, err := exec.Command("python", f).CombinedOutput()
|
|
if err != nil {
|
|
log.Printf("visualise failed with:\n\n%s%s%s\n", red, out, reset)
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|