78 lines
2.4 KiB
Go
78 lines
2.4 KiB
Go
|
package geojson_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"b612.me/astro/basic"
|
||
|
|
"b612.me/astro/eclipse"
|
||
|
|
"b612.me/astro/geojson"
|
||
|
|
"b612.me/astro/moon"
|
||
|
|
)
|
||
|
|
|
||
|
|
func BenchmarkStaticMap(b *testing.B) {
|
||
|
|
if err := basic.LoadStarData(); err != nil {
|
||
|
|
b.Fatal(err)
|
||
|
|
}
|
||
|
|
for _, test := range []struct {
|
||
|
|
name string
|
||
|
|
year, month, day int
|
||
|
|
planet moon.OccultationPlanet
|
||
|
|
hr int
|
||
|
|
}{
|
||
|
|
{"saturn", 2025, 1, 5, moon.OccultationSaturn, 0},
|
||
|
|
{"mars", 2025, 1, 14, moon.OccultationMars, 0},
|
||
|
|
{"venus", 2025, 9, 19, moon.OccultationVenus, 0},
|
||
|
|
{"star", 2025, 6, 5, "", 4799},
|
||
|
|
{"partial", 2025, 3, 29, "", 0},
|
||
|
|
{"hybrid", 2023, 4, 20, "", 0},
|
||
|
|
{"annular2012", 2012, 5, 21, "", 0},
|
||
|
|
{"annular2056", 2056, 7, 13, "", 0},
|
||
|
|
} {
|
||
|
|
b.Run(test.name, func(b *testing.B) {
|
||
|
|
zone := time.FixedZone("UTC+8", 8*3600)
|
||
|
|
day := time.Date(test.year, time.Month(test.month), test.day, 0, 0, 0, 0, zone)
|
||
|
|
var star moon.StarCoordinate
|
||
|
|
if test.hr > 0 {
|
||
|
|
data, err := basic.StarDataByHR(test.hr)
|
||
|
|
if err != nil {
|
||
|
|
b.Fatal(err)
|
||
|
|
}
|
||
|
|
star, err = moon.StarCoordinateFromStarData(data)
|
||
|
|
if err != nil {
|
||
|
|
b.Fatal(err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
options := moon.OccultationPathOptions{Step: 20 * time.Minute, TargetSpacingKM: 900, DisableFootprints: true, RiseSetStep: time.Minute}
|
||
|
|
b.ReportAllocs()
|
||
|
|
b.ResetTimer()
|
||
|
|
for i := 0; i < b.N; i++ {
|
||
|
|
var data []byte
|
||
|
|
var err error
|
||
|
|
if test.planet != "" {
|
||
|
|
paths, findErr := moon.FindPlanetOccultationPaths(day, day.Add(24*time.Hour), test.planet, options)
|
||
|
|
if findErr != nil || len(paths) != 1 {
|
||
|
|
b.Fatalf("paths=%d err=%v", len(paths), findErr)
|
||
|
|
}
|
||
|
|
data, err = geojson.MarshalPlanetOccultation(paths[0])
|
||
|
|
} else if test.hr > 0 {
|
||
|
|
paths, findErr := moon.FindStarOccultationPaths(day, day.Add(24*time.Hour), star, options)
|
||
|
|
if findErr != nil || len(paths) != 1 {
|
||
|
|
b.Fatalf("paths=%d err=%v", len(paths), findErr)
|
||
|
|
}
|
||
|
|
data, err = geojson.MarshalStarOccultation(paths[0])
|
||
|
|
} else {
|
||
|
|
path, ok := eclipse.SolarEclipsePartialFootprints(day, eclipse.SolarEclipsePartialFootprintOptions{Step: 5 * time.Minute, BoundaryPoints: 96, CentralShadowStep: 5 * time.Minute, RiseSetStep: 2 * time.Minute})
|
||
|
|
if !ok {
|
||
|
|
b.Fatal("missing eclipse")
|
||
|
|
}
|
||
|
|
data, err = geojson.MarshalSolarEclipse(path, nil)
|
||
|
|
}
|
||
|
|
if err != nil || len(data) == 0 {
|
||
|
|
b.Fatalf("bytes=%d err=%v", len(data), err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|