ssa

1
package ssa2
2

3
import (
4
	"image/color"
5

6
	"gonum.org/v1/plot"
7
	"gonum.org/v1/plot/plotter"
8
	"gonum.org/v1/plot/vg"
9
)
10

11
func CreateLineChart(y []float64, FileName string) error {
12

13
	p := plot.New()
14

15
	p.Title.Text = "Частоты основного тона периодограммы Томсона spw"
16

17
	// Создание точек данных
18
	pts := make(plotter.XYs, len(y))
19
	xx := make([]float64, len(y))
20
	for i := range xx {
21
		xx[i] = float64(i)
22
	}
23

24
	for i := range pts {
25
		pts[i].X = xx[i]
26
		pts[i].Y = y[i]
27
	}
28

29
	// Создание линейного графика
30
	line, ErrNewLine := plotter.NewLine(pts)
31
	if ErrNewLine != nil {
32
		return ErrNewLine
33
	}
34
	line.LineStyle.Width = vg.Points(1)
35
	line.LineStyle.Color = color.RGBA{B: 255, A: 255}
36

37
	// Добавление графика к графическому контексту
38
	p.Add(line)
39

40
	// Установка названий осей
41
	p.X.Label.Text = "t, s"
42
	p.Y.Label.Text = "fMAX,Гц"
43

44
	// Сохранение графика в файл
45
	if ErrSave := p.Save(6*vg.Inch, 4*vg.Inch, FileName); ErrSave != nil {
46
		return ErrSave
47
	}
48

49
	return nil
50
}
51