chart_theme.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718
  1. package data
  2. import (
  3. "encoding/json"
  4. "errors"
  5. "eta/eta_api/models/data_manage"
  6. "eta/eta_api/models/data_manage/chart_theme"
  7. "eta/eta_api/models/data_manage/chart_theme/request"
  8. "eta/eta_api/utils"
  9. "fmt"
  10. "github.com/shopspring/decimal"
  11. "math"
  12. "time"
  13. )
  14. // GetThemePreviewChartEdbData 获取图表的指标数据
  15. func GetThemePreviewChartEdbData(chartType int, calendar, startDate, endDate string, mappingList []*data_manage.ChartEdbInfoMapping, extraConfigStr string, seasonExtraConfig string) (edbList []*data_manage.ChartEdbInfoMapping, xEdbIdValue []int, yDataList []data_manage.YData, dataResp interface{}, err error, errMsg string) {
  16. edbList = make([]*data_manage.ChartEdbInfoMapping, 0)
  17. xEdbIdValue = make([]int, 0)
  18. yDataList = make([]data_manage.YData, 0)
  19. var extraConfig interface{}
  20. switch chartType {
  21. case 7: // 柱形图
  22. var barConfig data_manage.BarChartInfoReq
  23. if extraConfigStr == `` {
  24. errMsg = "柱方图未配置"
  25. err = errors.New(errMsg)
  26. return
  27. }
  28. err = json.Unmarshal([]byte(extraConfigStr), &barConfig)
  29. if err != nil {
  30. errMsg = "柱方图配置异常"
  31. err = errors.New(errMsg)
  32. return
  33. }
  34. extraConfig = barConfig
  35. case 10: // 截面散点图
  36. var tmpExtraConfig data_manage.SectionScatterReq
  37. if extraConfigStr == `` {
  38. errMsg = "截面散点图未配置"
  39. err = errors.New(errMsg)
  40. return
  41. }
  42. err = json.Unmarshal([]byte(extraConfigStr), &tmpExtraConfig)
  43. if err != nil {
  44. errMsg = "截面散点配置异常"
  45. err = errors.New(errMsg)
  46. return
  47. }
  48. extraConfig = tmpExtraConfig
  49. case utils.CHART_TYPE_RADAR:
  50. var barConfig data_manage.RadarChartInfoReq
  51. if extraConfigStr == `` {
  52. errMsg = "雷达图未配置"
  53. err = errors.New(errMsg)
  54. return
  55. }
  56. err = json.Unmarshal([]byte(extraConfigStr), &barConfig)
  57. if err != nil {
  58. errMsg = "雷达图配置异常"
  59. err = errors.New(errMsg)
  60. return
  61. }
  62. extraConfig = barConfig
  63. default:
  64. xEdbIdValue = make([]int, 0)
  65. yDataList = make([]data_manage.YData, 0)
  66. }
  67. // 指标对应的所有数据
  68. edbDataListMap, edbList, err := getThemePreviewEdbDataMapList(chartType, calendar, startDate, endDate, mappingList, seasonExtraConfig)
  69. if err != nil {
  70. return
  71. }
  72. // 特殊图形数据处理
  73. switch chartType {
  74. case 2: // 季节性图
  75. if seasonExtraConfig != "" {
  76. dataResp, err = SeasonChartData(edbList, seasonExtraConfig)
  77. } else {
  78. // 兼容无配置的老图
  79. dataResp = new(data_manage.SeasonChartResp)
  80. }
  81. case 7: // 柱形图
  82. barChartConf := extraConfig.(data_manage.BarChartInfoReq)
  83. xEdbIdValue, yDataList, err = BarChartData(mappingList, edbDataListMap, barChartConf.DateList, barChartConf.Sort)
  84. for k := range yDataList {
  85. yDataList[k].Unit = barChartConf.Unit
  86. yDataList[k].UnitEn = barChartConf.UnitEn
  87. }
  88. for _, v := range edbList {
  89. // 指标别名
  90. if barChartConf.EdbInfoIdList != nil && len(barChartConf.EdbInfoIdList) > 0 {
  91. for _, reqEdb := range barChartConf.EdbInfoIdList {
  92. if v.EdbInfoId == reqEdb.EdbInfoId {
  93. v.EdbAliasName = reqEdb.Name
  94. }
  95. }
  96. }
  97. }
  98. case 10: // 截面散点图
  99. sectionScatterConf := extraConfig.(data_manage.SectionScatterReq)
  100. xEdbIdValue, dataResp, err = getThemePreviewSectionScatterChartData(mappingList, edbDataListMap, sectionScatterConf)
  101. var tmpExtraConfig data_manage.SectionScatterReq
  102. if extraConfigStr == `` {
  103. errMsg = "截面散点图未配置"
  104. err = errors.New(errMsg)
  105. return
  106. }
  107. err = json.Unmarshal([]byte(extraConfigStr), &tmpExtraConfig)
  108. if err != nil {
  109. errMsg = "截面散点配置异常"
  110. err = errors.New(errMsg)
  111. return
  112. }
  113. // 这个数据没有必要返回给前端
  114. for _, v := range edbList {
  115. v.DataList = nil
  116. }
  117. case utils.CHART_TYPE_RADAR: //雷达图
  118. radarConf := extraConfig.(data_manage.RadarChartInfoReq)
  119. xEdbIdValue, dataResp, err = RadarChartData(mappingList, edbDataListMap, radarConf)
  120. case utils.CHART_TYPE_SECTION_COMBINE: // 截面组合图
  121. var sectionConf data_manage.ChartSectionAllExtraConf
  122. if extraConfigStr == `` {
  123. errMsg = "截面散点图未配置"
  124. err = errors.New(errMsg)
  125. return
  126. }
  127. err = json.Unmarshal([]byte(extraConfigStr), &sectionConf)
  128. if err != nil {
  129. errMsg = "截面散点配置异常"
  130. err = errors.New(errMsg + err.Error())
  131. return
  132. }
  133. // edbIdList 指标展示顺序;x轴的指标顺序
  134. xEdbIdValue = make([]int, 0)
  135. for _, v := range mappingList {
  136. xEdbIdValue = append(xEdbIdValue, v.EdbInfoId)
  137. }
  138. for k, _ := range sectionConf.SeriesList {
  139. sectionConf.SeriesList[k].NoDataEdbIndex = make([]int, 0)
  140. if k == 0 {
  141. sectionConf.SeriesList[k].DataList = []float64{1455, 1076, 2532, 3729}
  142. sectionConf.SeriesList[k].MaxData = 3729
  143. sectionConf.SeriesList[k].MinData = 1076
  144. }
  145. if k == 1 {
  146. sectionConf.SeriesList[k].DataList = []float64{13, 12, 12, 14}
  147. sectionConf.SeriesList[k].MaxData = 14
  148. sectionConf.SeriesList[k].MinData = 12
  149. }
  150. }
  151. dataResp = &data_manage.ChartSectionCombineDataResp{
  152. DateConfList: sectionConf.DateConfList,
  153. IsHeap: sectionConf.IsHeap,
  154. XDataList: sectionConf.XDataList,
  155. UnitList: sectionConf.UnitList,
  156. BaseChartSeriesName: sectionConf.BaseChartSeriesName,
  157. SortType: sectionConf.SortType,
  158. SeriesList: sectionConf.SeriesList,
  159. LeftMin: "0",
  160. LeftMax: "4000",
  161. RightMin: "10",
  162. RightMax: "15",
  163. }
  164. // 这个数据没有必要返回给前端
  165. for _, v := range edbList {
  166. v.DataList = nil
  167. }
  168. }
  169. return
  170. }
  171. func getThemePreviewEdbDataMapList(chartType int, calendar, startDate, endDate string, mappingList []*data_manage.ChartEdbInfoMapping, seasonExtraConfig string) (edbDataListMap map[int][]*data_manage.EdbDataList, edbList []*data_manage.ChartEdbInfoMapping, err error) {
  172. // 指标对应的所有数据
  173. edbDataListMap = make(map[int][]*data_manage.EdbDataList)
  174. for _, v := range mappingList {
  175. //fmt.Println("v:", v.EdbInfoId)
  176. item := new(data_manage.ChartEdbInfoMapping)
  177. item.EdbInfoId = v.EdbInfoId
  178. item.SourceName = v.SourceName
  179. item.Source = v.Source
  180. item.EdbCode = v.EdbCode
  181. item.EdbName = v.EdbName
  182. item.EdbNameEn = v.EdbNameEn
  183. item.Frequency = v.Frequency
  184. item.EdbType = v.EdbType
  185. item.FrequencyEn = GetFrequencyEn(v.Frequency)
  186. if v.Unit != `无` {
  187. item.Unit = v.Unit
  188. }
  189. item.UnitEn = v.UnitEn
  190. item.StartDate = v.StartDate
  191. item.EndDate = v.EndDate
  192. item.ModifyTime = v.ModifyTime
  193. item.EdbInfoCategoryType = v.EdbInfoCategoryType
  194. item.PredictChartColor = v.PredictChartColor
  195. item.ClassifyId = v.ClassifyId
  196. item.IsAxis = 1
  197. item.EdbInfoType = 1
  198. item.LeadValue = v.LeadValue
  199. item.LeadUnit = v.LeadUnit
  200. item.LeadUnitEn = GetLeadUnitEn(v.LeadUnit)
  201. item.ChartEdbMappingId = v.ChartEdbMappingId
  202. item.ChartInfoId = v.ChartInfoId
  203. item.ChartStyle = v.ChartStyle
  204. item.ChartColor = v.ChartColor
  205. item.ChartWidth = 0
  206. item.IsOrder = false
  207. item.MaxData = v.MaxValue
  208. item.MinData = v.MinValue
  209. item.LatestValue = v.LatestValue
  210. item.LatestDate = v.LatestDate
  211. item.UniqueCode = v.UniqueCode
  212. item.MoveLatestDate = v.LatestDate
  213. var startDateReal string
  214. var diffSeconds int64
  215. if chartType == 2 { //季节性图
  216. startDateReal = startDate
  217. if len(mappingList) > 1 {
  218. item.IsAxis = v.IsAxis
  219. }
  220. } else {
  221. if v.EdbInfoType == 0 && v.LeadUnit != "" && v.LeadValue > 0 { //领先指标
  222. var startTimeRealTemp time.Time
  223. startDateParse, _ := time.Parse(utils.FormatDate, startDate)
  224. switch v.LeadUnit {
  225. case "天":
  226. startTimeRealTemp = startDateParse.AddDate(0, 0, -v.LeadValue)
  227. case "月":
  228. startTimeRealTemp = startDateParse.AddDate(0, -v.LeadValue, 0)
  229. case "季":
  230. startTimeRealTemp = startDateParse.AddDate(0, -3*v.LeadValue, 0)
  231. case "周":
  232. startTimeRealTemp = startDateParse.AddDate(0, 0, -7*v.LeadValue)
  233. case "年":
  234. startTimeRealTemp = startDateParse.AddDate(-v.LeadValue, 0, 0)
  235. }
  236. if startTimeRealTemp.Before(startDateParse) {
  237. startDateReal = startTimeRealTemp.Format(utils.FormatDate)
  238. diffSeconds = (int64(startTimeRealTemp.UnixNano()) - int64(startDateParse.UnixNano())) / 1e6
  239. } else {
  240. startDateReal = startDate
  241. diffSeconds = 0
  242. }
  243. // 预测指标的开始日期也要偏移
  244. {
  245. day, tmpErr := utils.GetDaysBetween2Date(utils.FormatDate, startDate, startDateReal)
  246. if tmpErr != nil {
  247. err = tmpErr
  248. return
  249. }
  250. moveLatestDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, item.MoveLatestDate, time.Local)
  251. if tmpErr != nil {
  252. err = tmpErr
  253. return
  254. }
  255. item.MoveLatestDate = moveLatestDateTime.AddDate(0, 0, day).Format(utils.FormatDate)
  256. }
  257. } else {
  258. startDateReal = startDate
  259. }
  260. }
  261. dataList := make([]*data_manage.EdbDataList, 0)
  262. dataList, err = chart_theme.GetChartThemeDefaultDataItemList(v.EdbInfoId, startDateReal)
  263. if err != nil {
  264. return
  265. }
  266. edbDataListMap[v.EdbInfoId] = dataList
  267. if diffSeconds != 0 && v.EdbInfoType == 0 {
  268. dataListLen := len(dataList)
  269. for i := 0; i < dataListLen; i++ {
  270. dataList[i].DataTimestamp = dataList[i].DataTimestamp - diffSeconds
  271. }
  272. }
  273. if chartType == 2 {
  274. latestDate, tmpErr := time.Parse(utils.FormatDate, v.LatestDate)
  275. if tmpErr != nil {
  276. //item.DataList = dataList
  277. item.IsNullData = true
  278. edbList = append(edbList, item)
  279. continue
  280. }
  281. quarterDataList, tErr := GetSeasonEdbInfoDataListByXDate(dataList, latestDate, seasonExtraConfig)
  282. if tErr != nil {
  283. err = errors.New("获取季节性图表数据失败,Err:" + tErr.Error())
  284. return
  285. }
  286. item.DataList = quarterDataList
  287. } else if chartType == 7 || chartType == utils.CHART_TYPE_RADAR { //柱方图
  288. //item.DataList = dataList
  289. } else {
  290. item.DataList = dataList
  291. }
  292. edbList = append(edbList, item)
  293. }
  294. return
  295. }
  296. func getThemePreviewSectionScatterChartData(mappingList []*data_manage.ChartEdbInfoMapping, edbDataListMap map[int][]*data_manage.EdbDataList, extraConfig data_manage.SectionScatterReq) (edbIdList []int, chartDataResp data_manage.SectionScatterInfoResp, err error) {
  297. // 指标数据数组(10086:{"2022-12-02":100.01,"2022-12-01":102.3})
  298. edbDataMap := make(map[int]map[string]float64)
  299. for edbInfoId, edbDataList := range edbDataListMap {
  300. edbDateData := make(map[string]float64)
  301. for _, edbData := range edbDataList {
  302. edbDateData[edbData.DataTime] = edbData.Value
  303. }
  304. edbDataMap[edbInfoId] = edbDateData
  305. }
  306. // edbIdList 指标展示顺序;x轴的指标顺序
  307. edbIdList = make([]int, 0)
  308. edbMappingMap := make(map[int]*data_manage.ChartEdbInfoMapping)
  309. for _, v := range mappingList {
  310. edbIdList = append(edbIdList, v.EdbInfoId)
  311. edbMappingMap[v.EdbInfoId] = v
  312. }
  313. //SectionScatterSeriesInfoResp
  314. dataListResp := make([]data_manage.SectionScatterSeriesItemResp, 0) //y轴的数据列表
  315. for _, seriesItem := range extraConfig.SeriesList {
  316. var maxDate time.Time
  317. // 系列中的指标数据
  318. tmpSeriesEdbInfoList := make([]data_manage.SectionScatterEdbItemResp, 0)
  319. var minXVal, maxXVal, minYVal, maxYVal float64
  320. for _, edbConf := range seriesItem.EdbInfoList {
  321. tmpItem := data_manage.SectionScatterEdbItemResp{
  322. IsShow: edbConf.IsShow,
  323. Name: edbConf.Name,
  324. NameEn: edbConf.NameEn,
  325. } //单个坐标点的数据
  326. //X轴的数据
  327. {
  328. edbInfoId := edbConf.XEdbInfoId //X轴的指标
  329. edbMappingInfo, ok := edbMappingMap[edbInfoId]
  330. if !ok {
  331. continue
  332. }
  333. findDate := edbConf.XDate //需要的日期值
  334. dataList := edbDataListMap[edbInfoId] //指标的所有数据值
  335. if len(dataList) <= 0 {
  336. // 没有数据的指标id
  337. //findDataList = append(findDataList, 0)
  338. continue
  339. }
  340. tmpItem.XEdbInfoId = edbInfoId
  341. tmpItem.XName = edbMappingInfo.EdbName
  342. tmpItem.XNameEn = edbMappingInfo.EdbNameEn
  343. switch edbConf.XDateType {
  344. case 1: //最新值
  345. findDate = dataList[len(dataList)-1].DataTime
  346. case 2: //近期几天
  347. findDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[len(dataList)-1].DataTime, time.Local)
  348. if tmpErr != nil {
  349. err = tmpErr
  350. return
  351. }
  352. findDateTime = findDateTime.AddDate(0, 0, -edbConf.XDateValue)
  353. lenData := len(dataList) - 1
  354. for i := lenData; i >= 0; i-- {
  355. currDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[i].DataTime, time.Local)
  356. if tmpErr != nil {
  357. err = tmpErr
  358. return
  359. }
  360. if currDateTime.Equal(findDateTime) || currDateTime.Before(findDateTime) {
  361. findDate = dataList[i].DataTime
  362. break
  363. }
  364. }
  365. case 3: // 固定日期
  366. //最早的日期
  367. minDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[0].DataTime, time.Local)
  368. if tmpErr != nil {
  369. err = tmpErr
  370. return
  371. }
  372. //寻找固定日期的数据
  373. findDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, edbConf.XDate, time.Local)
  374. if tmpErr != nil {
  375. err = tmpErr
  376. return
  377. }
  378. for tmpDateTime := findDateTime; tmpDateTime.After(minDateTime) || tmpDateTime.Equal(minDateTime); tmpDateTime = tmpDateTime.AddDate(0, 0, -1) {
  379. tmpDate := tmpDateTime.Format(utils.FormatDate)
  380. if _, ok := edbDataMap[edbInfoId][tmpDate]; ok { //如果能找到数据,那么就返回
  381. findDate = tmpDate
  382. break
  383. }
  384. }
  385. default:
  386. err = errors.New(fmt.Sprint("日期类型异常,Type:", edbConf.XDate))
  387. return
  388. }
  389. findDateTime, _ := time.ParseInLocation(utils.FormatDate, findDate, time.Local)
  390. if maxDate.IsZero() {
  391. maxDate = findDateTime
  392. } else {
  393. if findDateTime.After(maxDate) {
  394. maxDate = findDateTime
  395. }
  396. }
  397. if tmpValue, ok := edbDataMap[edbInfoId][findDate]; ok {
  398. tmpItem.XDate = findDate
  399. tmpItem.XValue = tmpValue
  400. } else {
  401. continue
  402. }
  403. }
  404. //Y轴的数据
  405. {
  406. edbInfoId := edbConf.YEdbInfoId //Y轴的指标
  407. edbMappingInfo, ok := edbMappingMap[edbInfoId]
  408. if !ok {
  409. continue
  410. }
  411. findDate := edbConf.YDate //需要的日期值
  412. dataList := edbDataListMap[edbInfoId] //指标的所有数据值
  413. if len(dataList) <= 0 {
  414. // 没有数据的指标id
  415. //findDataList = append(findDataList, 0)
  416. continue
  417. }
  418. tmpItem.YEdbInfoId = edbInfoId
  419. tmpItem.YName = edbMappingInfo.EdbName
  420. tmpItem.YNameEn = edbMappingInfo.EdbNameEn
  421. switch edbConf.YDateType {
  422. case 1: //最新值
  423. findDate = dataList[len(dataList)-1].DataTime
  424. case 2: //近期几天
  425. findDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[len(dataList)-1].DataTime, time.Local)
  426. if tmpErr != nil {
  427. err = tmpErr
  428. return
  429. }
  430. findDateTime = findDateTime.AddDate(0, 0, -edbConf.YDateValue)
  431. lenData := len(dataList) - 1
  432. for i := lenData; i >= 0; i-- {
  433. currDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[i].DataTime, time.Local)
  434. if tmpErr != nil {
  435. err = tmpErr
  436. return
  437. }
  438. if currDateTime.Equal(findDateTime) || currDateTime.Before(findDateTime) {
  439. findDate = dataList[i].DataTime
  440. break
  441. }
  442. }
  443. case 3: // 固定日期
  444. //最早的日期
  445. minDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[0].DataTime, time.Local)
  446. if tmpErr != nil {
  447. err = tmpErr
  448. return
  449. }
  450. //寻找固定日期的数据
  451. findDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, edbConf.YDate, time.Local)
  452. if tmpErr != nil {
  453. err = tmpErr
  454. return
  455. }
  456. for tmpDateTime := findDateTime; tmpDateTime.After(minDateTime) || tmpDateTime.Equal(minDateTime); tmpDateTime = tmpDateTime.AddDate(0, 0, -1) {
  457. tmpDate := tmpDateTime.Format(utils.FormatDate)
  458. if _, ok := edbDataMap[edbInfoId][tmpDate]; ok { //如果能找到数据,那么就返回
  459. findDate = tmpDate
  460. break
  461. }
  462. }
  463. default:
  464. err = errors.New(fmt.Sprint("日期类型异常,Type:", edbConf.YDate))
  465. return
  466. }
  467. findDateTime, _ := time.ParseInLocation(utils.FormatDate, findDate, time.Local)
  468. if maxDate.IsZero() {
  469. maxDate = findDateTime
  470. } else {
  471. if findDateTime.After(maxDate) {
  472. maxDate = findDateTime
  473. }
  474. }
  475. if tmpValue, ok := edbDataMap[edbInfoId][findDate]; ok {
  476. tmpItem.YDate = findDate
  477. tmpItem.YValue = tmpValue
  478. } else {
  479. continue
  480. }
  481. }
  482. // 获取当前系列的X轴的最大最小值
  483. {
  484. if tmpItem.XValue < minXVal {
  485. minXVal = tmpItem.XValue
  486. }
  487. if tmpItem.XValue > maxXVal {
  488. maxXVal = tmpItem.XValue
  489. }
  490. if tmpItem.YValue < minYVal {
  491. minYVal = tmpItem.YValue
  492. }
  493. if tmpItem.YValue > maxYVal {
  494. maxYVal = tmpItem.YValue
  495. }
  496. }
  497. tmpSeriesEdbInfoList = append(tmpSeriesEdbInfoList, tmpItem)
  498. }
  499. trendLimitData := make([]data_manage.CoordinatePoint, 0) //趋势线的前后坐标点
  500. var trendLine, rSquare string
  501. // 生成线性方程式
  502. var a, b float64
  503. {
  504. coordinateData := make([]utils.Coordinate, 0)
  505. for _, tmpSeriesEdbInfo := range tmpSeriesEdbInfoList {
  506. tmpCoordinate1 := utils.Coordinate{
  507. X: tmpSeriesEdbInfo.XValue,
  508. Y: tmpSeriesEdbInfo.YValue,
  509. }
  510. coordinateData = append(coordinateData, tmpCoordinate1)
  511. }
  512. // 只有存在两个坐标点的时候,才能去计算线性方程和R平方
  513. if len(coordinateData) >= 2 {
  514. a, b = utils.GetLinearResult(coordinateData)
  515. if !math.IsNaN(a) && !math.IsNaN(b) {
  516. if b > 0 {
  517. trendLine = fmt.Sprintf("y=%sx+%s", utils.SubFloatToString(a, 4), utils.SubFloatToString(b, 4))
  518. } else {
  519. trendLine = fmt.Sprintf("y=%sx%s", utils.SubFloatToString(a, 4), utils.SubFloatToString(b, 4))
  520. }
  521. minYVal, _ = decimal.NewFromFloat(a).Mul(decimal.NewFromFloat(minXVal)).Add(decimal.NewFromFloat(b)).Round(4).Float64()
  522. maxYVal, _ = decimal.NewFromFloat(a).Mul(decimal.NewFromFloat(maxXVal)).Add(decimal.NewFromFloat(b)).Round(4).Float64()
  523. }
  524. // 计算R平方
  525. rSquare = fmt.Sprint(utils.CalculationDecisive(coordinateData))
  526. }
  527. trendLimitData = append(trendLimitData, data_manage.CoordinatePoint{
  528. X: minXVal,
  529. Y: minYVal,
  530. }, data_manage.CoordinatePoint{
  531. X: maxXVal,
  532. Y: maxYVal,
  533. })
  534. }
  535. dataListResp = append(dataListResp, data_manage.SectionScatterSeriesItemResp{
  536. Name: seriesItem.Name,
  537. NameEn: seriesItem.NameEn,
  538. IsNameDefault: seriesItem.IsNameDefault,
  539. Color: seriesItem.Color,
  540. EdbInfoList: tmpSeriesEdbInfoList,
  541. ShowTrendLine: seriesItem.ShowTrendLine,
  542. ShowFitEquation: seriesItem.ShowFitEquation,
  543. ShowRSquare: seriesItem.ShowRSquare,
  544. TrendLine: trendLine,
  545. RSquare: rSquare,
  546. TrendLimitData: trendLimitData,
  547. })
  548. }
  549. // 截面散点图点击详情时自动更新系列名
  550. if len(extraConfig.SeriesList) > 0 {
  551. // 默认名字的时候才自动更新
  552. if extraConfig.SeriesList[0].IsNameDefault {
  553. firstXEdbInfoId := extraConfig.SeriesList[0].EdbInfoList[0].XEdbInfoId
  554. if v, ok := edbMappingMap[firstXEdbInfoId]; ok {
  555. extraConfig.SeriesList[0].Name = v.LatestDate
  556. extraConfig.SeriesList[0].NameEn = v.LatestDate
  557. dataListResp[0].Name = v.LatestDate
  558. dataListResp[0].NameEn = v.LatestDate
  559. }
  560. }
  561. }
  562. chartDataResp = data_manage.SectionScatterInfoResp{
  563. XName: extraConfig.XName,
  564. XNameEn: extraConfig.XNameEn,
  565. XUnitName: extraConfig.XUnitName,
  566. XUnitNameEn: extraConfig.XUnitNameEn,
  567. YName: extraConfig.YName,
  568. YNameEn: extraConfig.YNameEn,
  569. YUnitName: extraConfig.YUnitName,
  570. YUnitNameEn: extraConfig.YUnitNameEn,
  571. XMinValue: extraConfig.XMinValue,
  572. XMaxValue: extraConfig.XMaxValue,
  573. YMinValue: extraConfig.YMinValue,
  574. YMaxValue: extraConfig.YMaxValue,
  575. DataList: dataListResp,
  576. }
  577. return
  578. }
  579. // GetChartThemeConfig
  580. // @Description: 根据主题id获取主题信息,如果获取不到的话,那么就获取默认的主题
  581. // @author: Roc
  582. // @datetime 2023-12-19 14:31:17
  583. // @param chartThemeId int
  584. // @param chartType int
  585. // @param source int
  586. // @return chartTheme *chart_theme.ChartTheme
  587. // @return err error
  588. func GetChartThemeConfig(chartThemeId, source, chartType int) (chartTheme *chart_theme.ChartTheme, err error) {
  589. chartTheme, err = chart_theme.GetChartThemeId(chartThemeId)
  590. if err != nil && err.Error() != utils.ErrNoRow() {
  591. return
  592. }
  593. err = nil
  594. // 如果找到了,那么就返回
  595. if chartTheme != nil {
  596. // 兼容历史数据,加入新字段LineOptionList
  597. newConfig, e := ConvertOldChartOptions(chartTheme.Config)
  598. if e == nil {
  599. chartTheme.Config = newConfig
  600. }
  601. return
  602. }
  603. // 没有找到的话,那么就找默认的主题
  604. // 查找主题类型id
  605. chartThemeType, err := chart_theme.GetChartThemeTypeByChartTypeAndSource(chartType, source)
  606. if err != nil {
  607. return
  608. }
  609. // 寻找默认的主题id
  610. chartTheme, err = chart_theme.GetChartThemeId(chartThemeType.DefaultChartThemeId)
  611. if err != nil && err.Error() != utils.ErrNoRow() {
  612. return
  613. }
  614. err = nil
  615. // 如果找到了,那么就返回
  616. if chartTheme != nil {
  617. // 兼容历史数据,加入新字段LineOptionList
  618. newConfig, e := ConvertOldChartOptions(chartTheme.Config)
  619. if e == nil {
  620. chartTheme.Config = newConfig
  621. }
  622. return
  623. }
  624. // 如果还是没找到,那就系统的主题id
  625. chartTheme, err = chart_theme.GetSystemChartTheme(chartThemeType.ChartThemeTypeId)
  626. // 兼容历史数据,加入新字段LineOptionList
  627. newConfig, e := ConvertOldChartOptions(chartTheme.Config)
  628. if e == nil {
  629. chartTheme.Config = newConfig
  630. }
  631. return
  632. }
  633. // 兼容历史数据,加入新字段LineOptionList
  634. func ConvertOldChartOptions(config string) (newConfig string, err error) {
  635. var oldTheme request.OldChartOptions
  636. var newTheme request.NewChartOptions
  637. err = json.Unmarshal([]byte(config), &oldTheme)
  638. if err != nil {
  639. return
  640. }
  641. if oldTheme.LineOptionList != nil && len(oldTheme.LineOptionList) > 0 {
  642. newConfig = config
  643. return
  644. }
  645. newTheme.OldChartOptions = oldTheme
  646. for i := 0; i < 10; i++ {
  647. newLineOption := request.NewLineOptions{
  648. LineOptions: oldTheme.LineOptions,
  649. Color: oldTheme.ColorsOptions[i],
  650. DataMark: "none",
  651. MarkType: "square",
  652. MarkSize: 5,
  653. MarkColor: oldTheme.ColorsOptions[i],
  654. }
  655. newTheme.LineOptionList = append(newTheme.LineOptionList, newLineOption)
  656. }
  657. newThemeStr, _ := json.Marshal(newTheme)
  658. newConfig = string(newThemeStr)
  659. return
  660. }