package dto

import (
	"google.golang.org/protobuf/proto"
	"joylink.club/bj-rtsts-server/dto/common_proto"
	"joylink.club/bj-rtsts-server/dto/data_proto"
)

type PageTrainInfoReqDto struct {
	PageQueryDto
	Name string `json:"name" form:"name"`
}

type TrainInfoReqDto struct {
	Name string `json:"name" form:"name"`
	Pid  int32  `json:"pid" form:"pid"`
}

type TrainInfoDto struct {
	Id             int32  `json:"id" form:"id"`
	Name           string `json:"name" form:"name"`
	TrainModel     int32  `json:"train_model" form:"train_model"`
	CarriageLength int32  `json:"carriage_length" form:"carriage_length"`
	TotalLength    int32  `json:"total_length" form:"total_length"`
	//MinDiameter     int32            `json:"min_diameter" form:"min_diameter"`
	//MaxDiameter     int32            `json:"max_diameter" form:"max_diameter"`
	TrainSets       string           `json:"train_sets" form:"train_sets"`
	Description     string           `json:"description" form:"description"`
	TrainConfigData *ConfigTrainData `json:"trainConfigData" form:"trainConfigData"`
}

func ConvertToTrainDto(trailInfos []*PublishedDto) []*TrainInfoDto {
	var result []*TrainInfoDto
	for _, t := range trailInfos {
		result = append(result, ConvertDtoFromTrain(t))
	}
	return result
}
func TrainConfigToDtoConvert(dc *common_proto.TrainDynamicConfig) *ConfigTrainData {
	if dc != nil {
		return &ConfigTrainData{
			DavisParamA: dc.DavisParamA, DavisParamB: dc.DavisParamB, DavisParamC: dc.DavisParamC,
			CurveResistanceParamR1: dc.CurveResistanceParamR1, CurveResistanceParamR2: dc.CurveResistanceParamR2, CurveResistanceParamR3: dc.CurveResistanceParamR3,
			CurveResistanceParamR4: dc.CurveResistanceParamR4, RevolvingMassParam: dc.RevolvingMassParam,
			Jump: dc.Jump, Slip: dc.Slip, Slide: dc.Slide, Idling: dc.Idling, StopSign: dc.StopSign,
		}
	}
	return nil
}
func ConvertDtoFromTrain(t *PublishedDto) *TrainInfoDto {
	message := &data_proto.Train{}
	proto.Unmarshal(t.Proto, message)
	return &TrainInfoDto{
		Id:             t.ID,
		Name:           t.Name,
		Description:    t.Note,
		TrainModel:     int32(message.TrainModel),
		CarriageLength: message.CarriageLength,
		TotalLength:    message.TotalLength,
		//MinDiameter:     message.MinDiameter,
		//MaxDiameter:     message.MaxDiameter,
		TrainSets:       message.TrainSets,
		TrainConfigData: TrainConfigToDtoConvert(message.DynamicConfig),
	}
}