fix

control_case/controllers.go

@@ -1,7 +1,17 @@
 package control_case
 
 import (
+	"encoding/csv"
+	"fmt"
+	"net/http"
+	"os"
+	"path/filepath"
+	"strconv"
+	"strings"
+
+	"control/analize"
 	"github.com/che4web/go4rest"
+	"github.com/gin-gonic/gin"
 	"gorm.io/gorm"
 )
 
@@ -11,8 +21,11 @@ type ControlCaseController struct {
 }
 
 func NewControlCaseController(db *gorm.DB) *ControlCaseController {
+	viewSet := go4rest.NewViewSet[ControlCase](db)
+	viewSet.PreloadField = []string{"Params"}
+
 	return &ControlCaseController{
-		ViewSet: go4rest.NewViewSet[ControlCase](db),
+		ViewSet: viewSet,
 		db:      db,
 	}
 }
@@ -28,3 +41,151 @@ func NewParamsController(db *gorm.DB) *ParamsController {
 		db:      db,
 	}
 }
+
+type LaunchControlCaseRequest struct {
+	ParamsID uint   `json:"params_id" binding:"required"`
+	Name     string `json:"name"`
+}
+
+func (c *ControlCaseController) Launch(ctx *gin.Context) {
+	var req LaunchControlCaseRequest
+	if err := ctx.ShouldBindJSON(&req); err != nil {
+		ctx.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+		return
+	}
+
+	var params Params
+	if err := c.db.First(&params, req.ParamsID).Error; err != nil {
+		ctx.JSON(http.StatusNotFound, gin.H{"error": "params not found"})
+		return
+	}
+
+	name := req.Name
+	if strings.TrimSpace(name) == "" {
+		name = fmt.Sprintf("case-%d", params.ID)
+	}
+
+	controlCase := ControlCase{
+		Name:     name,
+		ParamsID: params.ID,
+		Status:   "N",
+	}
+	if err := c.db.Create(&controlCase).Error; err != nil {
+		ctx.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+		return
+	}
+
+	if err := c.db.Preload("Params").First(&controlCase, controlCase.ID).Error; err != nil {
+		ctx.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+		return
+	}
+
+	ctx.JSON(http.StatusCreated, controlCase)
+}
+
+type CSVSeriesResponse struct {
+	Columns []string        `json:"columns"`
+	Rows    [][]interface{} `json:"rows"`
+}
+
+func (c *ControlCaseController) ChartData(ctx *gin.Context) {
+	id, err := strconv.ParseUint(ctx.Param("id"), 10, 32)
+	if err != nil {
+		ctx.JSON(http.StatusBadRequest, gin.H{"error": "invalid ID"})
+		return
+	}
+
+	var controlCase ControlCase
+	if err := c.db.Preload("Params").First(&controlCase, id).Error; err != nil {
+		ctx.JSON(http.StatusNotFound, gin.H{"error": "record not found"})
+		return
+	}
+
+	csvPath := filepath.Join(".", strconv.FormatUint(uint64(controlCase.ParamsID), 10), "foo.csv")
+	file, err := os.Open(csvPath)
+	if err != nil {
+		ctx.JSON(http.StatusNotFound, gin.H{"error": "csv file not found"})
+		return
+	}
+	defer file.Close()
+
+	reader := csv.NewReader(file)
+	records, err := reader.ReadAll()
+	if err != nil {
+		ctx.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+		return
+	}
+	if len(records) == 0 {
+		ctx.JSON(http.StatusOK, CSVSeriesResponse{Columns: []string{}, Rows: [][]interface{}{}})
+		return
+	}
+
+	columns := records[0]
+	rows := make([][]interface{}, 0, len(records)-1)
+	for _, record := range records[1:] {
+		row := make([]interface{}, 0, len(record))
+		for _, value := range record {
+			trimmed := strings.TrimSpace(value)
+			if f, err := strconv.ParseFloat(trimmed, 64); err == nil {
+				row = append(row, f)
+			} else {
+				row = append(row, trimmed)
+			}
+		}
+		rows = append(rows, row)
+	}
+
+	ctx.JSON(http.StatusOK, CSVSeriesResponse{Columns: columns, Rows: rows})
+}
+
+func (c *ControlCaseController) RecalculateAnalysis(ctx *gin.Context) {
+	id, err := strconv.ParseUint(ctx.Param("id"), 10, 32)
+	if err != nil {
+		ctx.JSON(http.StatusBadRequest, gin.H{"error": "invalid ID"})
+		return
+	}
+
+	var controlCase ControlCase
+	if err := c.db.First(&controlCase, id).Error; err != nil {
+		ctx.JSON(http.StatusNotFound, gin.H{"error": "record not found"})
+		return
+	}
+
+	csvPath := filepath.Join(".", strconv.FormatUint(uint64(controlCase.ParamsID), 10), "foo.csv")
+	result, err := analize.AnalyzeCSV(c.db, controlCase.ID, controlCase.Name, csvPath)
+	if err != nil {
+		ctx.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+		return
+	}
+
+	ctx.JSON(http.StatusOK, result)
+}
+
+func (c *ControlCaseController) FieldMap(ctx *gin.Context) {
+	id, err := strconv.ParseUint(ctx.Param("id"), 10, 32)
+	if err != nil {
+		ctx.JSON(http.StatusBadRequest, gin.H{"error": "invalid ID"})
+		return
+	}
+
+	timeValue, err := strconv.ParseFloat(ctx.Query("time"), 64)
+	if err != nil {
+		ctx.JSON(http.StatusBadRequest, gin.H{"error": "invalid time"})
+		return
+	}
+
+	var controlCase ControlCase
+	if err := c.db.First(&controlCase, id).Error; err != nil {
+		ctx.JSON(http.StatusNotFound, gin.H{"error": "record not found"})
+		return
+	}
+
+	h5Path := filepath.Join(".", strconv.FormatUint(uint64(controlCase.ParamsID), 10), "storage.h5")
+	result, err := ReadNearestFieldMap(h5Path, timeValue)
+	if err != nil {
+		ctx.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+		return
+	}
+
+	ctx.JSON(http.StatusOK, result)
+}