math-optim/run.go
leo adfba66f38
All checks were successful
continuous-integration/drone/push Build is passing
mv most of profiling logic to profile.go
2023-02-24 16:00:00 +01:00

125 lines
2.8 KiB
Go

// Copyright 2023 wanderer <a_mirre at utb dot cz>
// SPDX-License-Identifier: GPL-3.0-or-later
package main
import (
"errors"
"flag"
"log"
"runtime/pprof"
"sync"
"git.dotya.ml/wanderer/math-optim/algo"
"git.dotya.ml/wanderer/math-optim/report"
)
var version = "development"
var (
// ref: https://go.dev/blog/pprof.
cpuprofile = flag.String("cpuprofile", "", "write cpu profile to this file")
memprofile = flag.String("memprofile", "", "write memory profile to this file")
doPrint = flag.Bool("printreport", true, "print report.tex to console")
generate = flag.Bool("generate", true, "run algos and generate plot pics/statistical tables (anew)")
rS = flag.Bool("randomsearch", false, "run Random Search algorithm")
sHC = flag.Bool("shc", false, "run Stochastic Hill Climbing algorithm")
n100 = flag.Bool("N100", false, "run the \"100 Neighbours\" variant of SHC")
// TODO(me): add flag for plot output format: -plotout=(svg,eps,pdf).
jDE = flag.Bool("jde", false, "run Differential Evolution algorithm with parameter self adaptation")
// run CEC2020 jDE by default.
c2jDE = flag.Bool("c2jde", true, "run CEC2020 version of the Differential Evolution algorithm with parameter self adaptation")
c2SOMAT3A = flag.Bool("c2somat3a", false, "run CEC2020 version of the SOMA Team-to-Team Adaptive (T3A)")
)
func run() error {
log.Println("starting math-optim version", "'"+version+"'")
flag.Parse()
err := profileCPU(cpuprofile)
if err != nil {
return err
}
// see profile.go on why we're calling this here.
defer pprof.StopCPUProfile()
if *generate {
if !*jDE && !*c2jDE && !*c2SOMAT3A && !*sHC && !*rS {
log.Println("at least one algo needs to be specified, exiting...")
return errors.New("ErrNoAlgoSelected")
}
var wg sync.WaitGroup
var m sync.Mutex
if *jDE {
wg.Add(1)
go algo.DojDE(&wg, &m)
}
if *c2jDE {
wg.Add(1)
go algo.DoCEC2020jDE(&wg, &m)
}
if *c2SOMAT3A {
wg.Add(1)
go algo.DoCEC2020SOMAT3A(&wg, &m)
}
if *rS {
wg.Add(1)
go algo.DoRandomSearch(&wg, &m)
}
if *sHC {
wg.Add(1)
if *n100 {
go algo.DoStochasticHillClimbing100Neigh(&wg, &m)
} else {
go algo.DoStochasticHillClimbing(&wg, &m)
}
}
wg.Wait()
var pL *report.PicList
var benchCount int
if *c2jDE && *c2SOMAT3A {
pL, benchCount = algo.PrepCEC2020ComparisonOfMeans(&wg)
} else {
pL, benchCount = algo.PrepComparisonOfMeans(&wg)
}
report.SaveComparisonOfMeans(*pL, benchCount)
report.SaveTexAllPics()
report.SaveTexAllTables()
}
report.SaveAndPrint(*doPrint)
err = profileMem(memprofile)
if err != nil {
return err
}
log.Println("looks like we're done")
log.Println("run an equivalent of `pdflatex -clean -shell-escape -interaction=nonstopmode ./report.tex` to get a pdf")
return nil
}