logos-blockchain/logos-blockchain-simulations
1
0
Fork 0
mirror of https://github.com/logos-blockchain/logos-blockchain-simulations.git synced 2026-02-18 12:13:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: use the correct path for saving coeff.csv

Browse Source
This commit is contained in:
Youngjoon Lee 2024-09-09 14:30:02 +09:00
parent 2bcfdc6bad
commit d43f08d2be
No known key found for this signature in database
GPG Key ID: 167546E2D1712F8C
1 changed files with 23 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
mixnet-rs/ordering/src/bin/coeff.rs
View File

@ -235,12 +235,6 @@ fn main() {
}
fn calculate_coeffs(path: &str) {
let mut senders: Vec<u64> = Vec::new();
let mut receivers: Vec<u64> = Vec::new();
let mut strongs: Vec<u64> = Vec::new();
let mut casuals: Vec<u64> = Vec::new();
let mut weaks: Vec<u64> = Vec::new();
for entry in WalkDir::new(path)
.into_iter()
.filter_map(|e| e.ok())
@ -248,6 +242,12 @@ fn calculate_coeffs(path: &str) {
{
let dir_name = entry.path().file_name().unwrap().to_string_lossy();
if dir_name.starts_with("iteration_") {
let mut senders: Vec<u64> = Vec::new();
let mut receivers: Vec<u64> = Vec::new();
let mut strongs: Vec<u64> = Vec::new();
let mut casuals: Vec<u64> = Vec::new();
let mut weaks: Vec<u64> = Vec::new();
for sent_seq_file in glob(&format!("{}/sent_seq_*.csv", entry.path().display()))
.unwrap()
.filter_map(Result::ok)
@ -287,25 +287,25 @@ fn calculate_coeffs(path: &str) {
);
}
}
// Create a Polars DataFrame
let mut df = DataFrame::new(vec![
Series::new("sender", &senders),
Series::new("receiver", &receivers),
Series::new("strong", &strongs),
Series::new("casual", &casuals),
Series::new("weak", &weaks),
])
.unwrap()
.sort(["sender", "receiver"], SortMultipleOptions::default())
.unwrap();
// Write the sorted DataFrame to a CSV file
let outpath = Path::new(entry.path()).join("coeffs.csv");
let mut file = File::create(&outpath).unwrap();
CsvWriter::new(&mut file).finish(&mut df).unwrap();
tracing::info!("Saved {}", outpath.display());
}
}
// Create a Polars DataFrame
let mut df = DataFrame::new(vec![
Series::new("sender", &senders),
Series::new("receiver", &receivers),
Series::new("strong", &strongs),
Series::new("casual", &casuals),
Series::new("weak", &weaks),
])
.unwrap()
.sort(["sender", "receiver"], SortMultipleOptions::default())
.unwrap();
// Write the sorted DataFrame to a CSV file
let outpath = Path::new(path).join("coeffs.csv");
let mut file = File::create(&outpath).unwrap();
CsvWriter::new(&mut file).finish(&mut df).unwrap();
tracing::info!("Saved {}", outpath.display());
}
fn extract_id(filename: &str) -> Option<u8> {