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

add --sent-seq-limit optional param

Browse Source
This commit is contained in:
Youngjoon Lee 2024-09-10 06:49:47 +09:00
parent d0a9af68d9
commit 1e378c7101
No known key found for this signature in database
GPG Key ID: 167546E2D1712F8C
1 changed files with 19 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -53,13 +53,17 @@ enum OrderingType {
}
fn coeff(seq1: &[Entry], seq2: &[Entry], ordering_type: OrderingType) -> u64 {
let mut coeff = 0;
let mut coeff: u64 = 0;
let mut i = 0;
while i < seq1.len() {
if let Entry::Data(_) = &seq1[i] {
let (c, next_i) = coeff_from(seq1, i, seq2, ordering_type);
coeff += c;
coeff = coeff.saturating_add(c);
if coeff == u64::MAX {
// no need to continue
break;
}
if next_i != i {
i = next_i;
@ -136,9 +140,7 @@ fn strong_coeff_from(
let coeff = if num_consecutive_pairs == 0 {
0
} else {
num_consecutive_pairs
.checked_pow(num_consecutive_pairs.try_into().unwrap())
.unwrap()
num_consecutive_pairs.saturating_pow(num_consecutive_pairs.try_into().unwrap())
};
(coeff, i)
}
@ -219,6 +221,8 @@ struct Args {
path: String,
#[arg(short, long)]
num_threads: usize,
#[arg(short, long)]
sent_seq_limit: Option<usize>,
}
fn main() {
@ -259,6 +263,7 @@ fn calculate_coeffs(args: &Args) {
outpath: entry
.path()
.join(format!("coeffs_{}_{}.csv", sender, receiver)),
sent_seq_limit: args.sent_seq_limit,
};
tasks.push(task);
}
@ -307,6 +312,7 @@ struct Task {
sender: u8,
receiver: u8,
outpath: PathBuf,
sent_seq_limit: Option<usize>,
}
impl Task {
@ -317,8 +323,15 @@ impl Task {
self.recv_seq_file.display()
);
let sent_seq = load_sequence(self.sent_seq_file.to_str().unwrap());
let mut sent_seq = load_sequence(self.sent_seq_file.to_str().unwrap());
let recv_seq = load_sequence(self.recv_seq_file.to_str().unwrap());
if let Some(limit) = self.sent_seq_limit {
if sent_seq.len() > limit {
sent_seq.truncate(limit);
}
}
let strong = coeff(&sent_seq, &recv_seq, OrderingType::Strong);
let casual = coeff(&sent_seq, &recv_seq, OrderingType::Casual);
let weak = coeff(&sent_seq, &recv_seq, OrderingType::Weak);