ssa

1
package pmtm_test
2

3
import (
4
	"fmt"
5
	"log"
6
	"os"
7
	"strconv"
8
	"testing"
9

10
	"github.com/RB-PRO/ssa/pkg/oss"
11
	"github.com/RB-PRO/ssa/pkg/pmtm"
12
	"github.com/xuri/excelize/v2"
13
)
14

15
func TestPmtm(t *testing.T) {
16
	x, _ := oss.Make_singnal_xn("pmtm")
17
	y := pmtm.Pmtm(x, 1024)
18
	saveTXT2("test222.txt", y)
19
	safeToXlsx(x, y)
20
}
21

22
// Сохранить в xlsx для дебага
23
func safeToXlsx(x, y []float64) {
24

25
	file_graph := excelize.NewFile()
26
	file_graph.NewSheet("main")
27
	file_graph.NewSheet("golang")
28
	file_graph.DeleteSheet("Sheet1")
29
	lenFor := len(x)
30
	for ind := 0; ind < lenFor; ind++ {
31
		file_graph.SetCellValue("golang", "A"+strconv.Itoa(ind+1), x[ind])
32
	}
33
	lenFor = len(y)
34
	for ind := 0; ind < lenFor; ind++ {
35
		file_graph.SetCellValue("golang", "B"+strconv.Itoa(ind+1), y[ind])
36
	}
37
	if err := file_graph.SaveAs("save_pmtm" + ".xlsx"); err != nil {
38
		fmt.Println(err)
39
	}
40
}
41

42
func saveTXT2(FileName string, data []float64) {
43
	filePW, ErrOpenFile := os.Create(FileName)
44
	if ErrOpenFile != nil {
45
		panic(ErrOpenFile)
46
	}
47
	defer filePW.Close()
48
	for i := range data {
49
		if _, err := filePW.WriteString(fmt.Sprintf("%.8f\n", data[i])); err != nil {
50
			log.Println(err)
51
		}
52
	}
53
}
54