zsb/fidc-backtest-engine
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

去重股票状态rolling计算

Browse Source
This commit is contained in:
boris
2026-06-21 00:54:33 +08:00
parent a270e368c8
commit fd27429713
1 changed files with 26 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/fidc-core/src/platform_expr_strategy.rs
+26 -57
View File
@@ -2264,63 +2264,32 @@ impl PlatformExprStrategy {
let factor = ctx.data.require_factor(factor_date, symbol)?;
let candidate = ctx.data.require_candidate(date, symbol)?;
let instrument = ctx.data.instrument(symbol);
let stock_ma_short = self
.stock_decision_rolling_mean(
ctx,
date,
symbol,
&factor.extra_factors,
"close",
self.config.stock_short_ma_days,
)
.unwrap_or(f64::NAN);
let stock_ma_mid = self
.stock_decision_rolling_mean(
ctx,
date,
symbol,
&factor.extra_factors,
"close",
self.config.stock_mid_ma_days,
)
.unwrap_or(f64::NAN);
let stock_ma_long = self
.stock_decision_rolling_mean(
ctx,
date,
symbol,
&factor.extra_factors,
"close",
self.config.stock_long_ma_days,
)
.unwrap_or(f64::NAN);
let stock_ma5 = self
.stock_decision_rolling_mean(ctx, date, symbol, &factor.extra_factors, "close", 5)
.unwrap_or(f64::NAN);
let stock_ma10 = self
.stock_decision_rolling_mean(ctx, date, symbol, &factor.extra_factors, "close", 10)
.unwrap_or(f64::NAN);
let stock_ma20 = self
.stock_decision_rolling_mean(ctx, date, symbol, &factor.extra_factors, "close", 20)
.unwrap_or(f64::NAN);
let stock_ma30 = self
.stock_decision_rolling_mean(ctx, date, symbol, &factor.extra_factors, "close", 30)
.unwrap_or(f64::NAN);
let stock_volume_ma5 = self
.stock_decision_rolling_mean(ctx, date, symbol, &factor.extra_factors, "volume", 5)
.unwrap_or(f64::NAN);
let stock_volume_ma10 = self
.stock_decision_rolling_mean(ctx, date, symbol, &factor.extra_factors, "volume", 10)
.unwrap_or(f64::NAN);
let stock_volume_ma20 = self
.stock_decision_rolling_mean(ctx, date, symbol, &factor.extra_factors, "volume", 20)
.unwrap_or(f64::NAN);
let stock_volume_ma60 = self
.stock_decision_rolling_mean(ctx, date, symbol, &factor.extra_factors, "volume", 60)
.unwrap_or(f64::NAN);
let stock_volume_ma100 = self
.stock_decision_rolling_mean(ctx, date, symbol, &factor.extra_factors, "volume", 100)
.unwrap_or(f64::NAN);
let mut rolling_values = HashMap::<(&'static str, usize), f64>::new();
let mut rolling = |field: &'static str, lookback: usize| -> f64 {
*rolling_values.entry((field, lookback)).or_insert_with(|| {
self.stock_decision_rolling_mean(
ctx,
date,
symbol,
&factor.extra_factors,
field,
lookback,
)
.unwrap_or(f64::NAN)
})
};
let stock_ma_short = rolling("close", self.config.stock_short_ma_days);
let stock_ma_mid = rolling("close", self.config.stock_mid_ma_days);
let stock_ma_long = rolling("close", self.config.stock_long_ma_days);
let stock_ma5 = rolling("close", 5);
let stock_ma10 = rolling("close", 10);
let stock_ma20 = rolling("close", 20);
let stock_ma30 = rolling("close", 30);
let stock_volume_ma5 = rolling("volume", 5);
let stock_volume_ma10 = rolling("volume", 10);
let stock_volume_ma20 = rolling("volume", 20);
let stock_volume_ma60 = rolling("volume", 60);
let stock_volume_ma100 = rolling("volume", 100);
let touched_upper_limit = !market.paused
&& (market.is_at_upper_limit_price(market.close)
|| market.is_at_upper_limit_price(market.open)