weijuesen
70 lines
1.9 KiB
Go
70 lines
1.9 KiB
Go
package service
|
||
|
||
import "math"
|
||
|
||
// RiskInput 风险评分输入（各系数取值 0~1）
|
||
type RiskInput struct {
|
||
AI float64 // AI 识别置信度（权重 0.5）
|
||
Env float64 // 环境风险系数（权重 0.2）
|
||
Stage float64 // 饲养阶段风险系数（权重 0.15）
|
||
Uniformity float64 // 群体整齐度偏离度（权重 0.15）
|
||
}
|
||
|
||
// ComputeRiskScore 按规格书 3.1.4 公式计算 0~100 风险分：
|
||
// 风险分 = 0.5×AI置信度 + 0.2×环境系数 + 0.15×阶段系数 + 0.15×整齐度偏离度
|
||
func ComputeRiskScore(in RiskInput) float64 {
|
||
score := 0.5*in.AI + 0.2*in.Env + 0.15*in.Stage + 0.15*in.Uniformity
|
||
score = math.Max(0, math.Min(1, score))
|
||
return score * 100
|
||
}
|
||
|
||
// RiskLevel 按规格书 3.1.4 分级：绿 0-30 / 黄 31-60 / 橙 61-80 / 红 81-100
|
||
func RiskLevel(score float64) string {
|
||
switch {
|
||
case score <= 30:
|
||
return "green"
|
||
case score <= 60:
|
||
return "yellow"
|
||
case score <= 80:
|
||
return "orange"
|
||
default:
|
||
return "red"
|
||
}
|
||
}
|
||
|
||
// StageCoefficient 蚕房阶段 → 阶段风险系数（Room.Stage 粗粒度映射；
|
||
// 待 #7 蚕匾/批次管理的龄期字段落地后细化）
|
||
func StageCoefficient(stage string) float64 {
|
||
switch stage {
|
||
case "pupa":
|
||
return 0.5 // 核型多角体病 5龄后期至蛹期高发
|
||
case "larva":
|
||
return 0.4 // 软化病 5 龄集中暴发等
|
||
case "moth":
|
||
return 0.2
|
||
case "egg":
|
||
return 0.1
|
||
default:
|
||
return 0
|
||
}
|
||
}
|
||
|
||
// EnvCoefficient 由最新温湿度计算环境风险系数（规则取自规格书 3.2.3）
|
||
func EnvCoefficient(temp, humidity *float64) float64 {
|
||
coef := 0.0
|
||
if humidity != nil {
|
||
switch {
|
||
case *humidity >= 80:
|
||
coef = math.Max(coef, 0.8) // 白僵病等真菌病高湿条件
|
||
case *humidity >= 75:
|
||
coef = math.Max(coef, 0.5)
|
||
}
|
||
}
|
||
if temp != nil {
|
||
if *temp > 30 || *temp < 20 {
|
||
coef = math.Max(coef, 0.5) // 温度突变诱发核型多角体病
|
||
}
|
||
}
|
||
return coef
|
||
}
|