From 456fbd3f256875c878d151314328bc62c5e300ea Mon Sep 17 00:00:00 2001
From: Youngjoon Lee <5462944+youngjoon-lee@users.noreply.github.com>
Date: Sat, 17 Aug 2024 23:45:17 +0900
Subject: [PATCH] add --from-paramset

---
 mixnet-rs/dissemination/src/main.rs | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/mixnet-rs/dissemination/src/main.rs b/mixnet-rs/dissemination/src/main.rs
index ff5d010..1be8da7 100644
--- a/mixnet-rs/dissemination/src/main.rs
+++ b/mixnet-rs/dissemination/src/main.rs
@@ -30,6 +30,8 @@ struct Args {
     outdir: String,
     #[arg(short, long)]
     num_threads: usize,
+    #[arg(short, long)]
+    from_paramset: Option<u16>,
 }
 
 fn main() {
@@ -43,6 +45,7 @@ fn main() {
         queue_type,
         outdir,
         num_threads,
+        from_paramset,
     } = args;
 
     // Create a directory and initialize a CSV file only with a header
@@ -70,6 +73,11 @@ fn main() {
 
     pool.install(|| {
         paramsets.par_iter().for_each(|paramset| {
+            if paramset.id < from_paramset.unwrap_or(0) {
+                tracing::info!("ParamSet:{} skipped", paramset.id);
+                return;
+            }
+
             let paramset_dir = format!("{outdir}/{subdir}/__WIP__paramset_{}", paramset.id);
             std::fs::create_dir_all(paramset_dir.as_str()).unwrap();
             save_paramset_info(paramset, format!("{paramset_dir}/paramset.csv").as_str()).unwrap();