From 57c6561fa26f3889bbae817105cd275f6e7d2b61 Mon Sep 17 00:00:00 2001 From: Dominik Werder Date: Wed, 5 Mar 2025 15:23:15 +0100 Subject: [PATCH] Add switch for cnt zero bins --- src/binning/timeweight/timeweight_bins.rs | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/binning/timeweight/timeweight_bins.rs b/src/binning/timeweight/timeweight_bins.rs index 8e2156d..040426e 100644 --- a/src/binning/timeweight/timeweight_bins.rs +++ b/src/binning/timeweight/timeweight_bins.rs @@ -38,6 +38,7 @@ where agg: ::AggregatorTw, non_fnl: bool, out: ContainerBins, + produce_cnt_zero: bool, } impl BinnedBinsTimeweight @@ -60,11 +61,16 @@ where agg: BVT::AggregatorTw::new(), non_fnl: false, out: ContainerBins::new(), + produce_cnt_zero: false, } } + pub fn cnt_zero_enable(&mut self) { + self.produce_cnt_zero = true; + } + fn maybe_emit_active(&mut self) { - if self.cnt != 0 { + if self.cnt != 0 || self.produce_cnt_zero { let ts1 = self.active_beg; let ts2 = self.active_end; let cnt = self.cnt;