heimoshuiyu/rust-synapse-compress-state
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Wrapped code with pyo3 so that the libary can be used from python (#42)

Browse Source
This commit is contained in:
Azrenbeth
2021-08-03 09:44:51 +01:00
committed by GitHub
parent a887107607
commit becb2937aa
6 changed files with 390 additions and 168 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/compressor.rs
View File

@@ -23,7 +23,7 @@
//!
//! This produces graphs that look roughly like, for two levels:
//!
//! ``` ignored
//! ```ignore
//! L2 <-------------------- L2 <---------- ...
//! ^--- L1 <--- L1 <--- L1 ^--- L1 <--- L1 <--- L1
//! ```
@@ -182,7 +182,7 @@ impl<'a> Compressor<'a> {
///
/// Returns the state map and the actual base state group (if any) used.
fn get_delta(&mut self, prev_sg: Option<i64>, sg: i64) -> (StateMap<Atom>, Option<i64>) {
let state_map = collapse_state_maps(&self.original_state_map, sg);
let state_map = collapse_state_maps(self.original_state_map, sg);
let mut prev_sg = if let Some(prev_sg) = prev_sg {
prev_sg
@@ -194,7 +194,7 @@ impl<'a> Compressor<'a> {
// a valid base for the state group.
let mut prev_state_map;
'outer: loop {
prev_state_map = collapse_state_maps(&self.original_state_map, prev_sg);
prev_state_map = collapse_state_maps(self.original_state_map, prev_sg);
for (t, s) in prev_state_map.keys() {
if !state_map.contains_key(t, s) {
// This is not a valid base as it contains key the new state