1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124
/*
* AquaVM Workflow Engine
*
* Copyright (C) 2024 Fluence DAO
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation version 3 of the
* License.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
use air_interpreter_data::TraceLen;
use super::ExecutedState;
use super::ExecutionTrace;
use super::KeeperError::*;
use super::KeeperResult;
use crate::TracePos;
type SeenElements = u32;
/// This slider is intended to slide on a subtrace inside provided trace. This subtrace
/// is identified by position and len.
// TODO: check for overflow
#[derive(Debug, Default, Clone, PartialEq, Eq)]
pub struct TraceSlider {
/// Trace that slider slide on.
trace: ExecutionTrace,
/// Position of current subtrace inside trace.
position: TracePos,
/// Length of a current subtrace.
subtrace_len: TraceLen,
/// Count of seen elements since the last position update.
seen_elements: SeenElements,
}
impl TraceSlider {
pub(crate) fn new(trace: impl Into<ExecutionTrace>) -> Self {
let trace = trace.into();
let subtrace_len = trace.trace_states_count();
Self {
trace,
subtrace_len,
..<_>::default()
}
}
/// Returns the next state if current interval length hasn't been reached
/// and None otherwise.
#[allow(clippy::suspicious_operation_groupings)]
pub(crate) fn next_state(&mut self) -> Option<ExecutedState> {
if self.seen_elements >= self.subtrace_len || self.position >= self.trace.trace_states_count().into() {
return None;
}
let result = self.trace[self.position].clone();
self.position += 1;
self.seen_elements += 1;
Some(result)
}
pub(crate) fn set_position_and_len(&mut self, position: TracePos, subtrace_len: TraceLen) -> KeeperResult<()> {
// it's possible to set empty subtrace_len and inconsistent position
if subtrace_len != 0 && position + subtrace_len > self.trace.trace_states_count().into() {
return Err(SetSubtraceLenAndPosFailed {
requested_pos: position,
requested_subtrace_len: subtrace_len,
trace_len: self.trace.trace_states_count(),
});
}
self.position = position;
self.subtrace_len = subtrace_len;
self.seen_elements = 0;
Ok(())
}
pub(crate) fn set_subtrace_len(&mut self, subtrace_len: TraceLen) -> KeeperResult<()> {
let trace_remainder: TraceLen = (TracePos::from(self.trace_len()) - self.position).into();
if trace_remainder < subtrace_len {
return Err(SetSubtraceLenFailed {
requested_subtrace_len: subtrace_len,
trace_position: self.position,
trace_len: self.trace.trace_states_count(),
});
}
self.seen_elements = 0;
self.subtrace_len = subtrace_len;
Ok(())
}
pub(crate) fn position(&self) -> TracePos {
self.position
}
pub(crate) fn subtrace_len(&self) -> TraceLen {
debug_assert!(self.subtrace_len >= self.seen_elements);
self.subtrace_len - self.seen_elements
}
pub(crate) fn state_at_position(&self, position: TracePos) -> Option<&ExecutedState> {
// it would be nice to have the `impl SliceIndex for TracePos`, but it is unstable
self.trace.get(position)
}
pub(super) fn trace_len(&self) -> TraceLen {
self.trace.trace_states_count()
}
}