feat: config impl

Cargo.toml

@@ -5,3 +5,5 @@ edition = "2021"
 
 [dependencies]
 rand = "0.8.4"
+serde = { version = "1.0", features = ["derive"] }
+toml = "0.8.19"

src/config.rs

@@ -0,0 +1,47 @@
+use serde::Deserialize;
+use std::fs;
+use toml;
+
+#[derive(Debug, Deserialize)]
+pub struct Config {
+    pub problem: Problem,
+    pub algorithm: Algorithm,
+    pub test: Test,
+}
+
+#[derive(Debug, Deserialize)]
+pub struct Problem {
+    pub size: i32,
+}
+
+#[derive(Debug, Deserialize)]
+pub struct Algorithm {
+    pub max_gen: i32,
+    pub n_neighbors: i32,
+}
+
+#[derive(Debug, Deserialize)]
+pub struct Test {
+    pub count: i32,
+}
+
+impl Default for Config {
+    fn default() -> Self {
+        Config {
+            problem: Problem { size: 20 },
+            algorithm: Algorithm {
+                max_gen: 1000,
+                n_neighbors: 10,
+            },
+            test: Test { count: 100 },
+        }
+    }
+}
+
+impl Config {
+    pub fn get() -> Result<Config, Box<dyn std::error::Error>> {
+        let config_content = fs::read_to_string("config.toml")?;
+        let config: Config = toml::from_str(&config_content)?;
+        Ok(config)
+    }
+}

src/lib.rs

@@ -1,4 +1,5 @@
 pub mod case;
+pub mod config;
 pub mod debug;
 pub mod qts;
 pub mod quantum;

src/main.rs

@@ -1,13 +1,12 @@
 use qts_rust::case::case_1;
+use qts_rust::config::Config;
 use qts_rust::qts::qts;
 use qts_rust::types::{Items, Problem};
 use std::sync::{Arc, Mutex};
 use std::thread;
 
 fn main() {
-    let test_times: i32 = 1000;
-    let max_gen: i32 = 1000;
-    let n_neighbors: i32 = 10;
+    let config = Config::get().unwrap();
 
     let problem = Problem::default();
     let items = Arc::new(Mutex::new(Vec::new()));
@@ -18,11 +17,16 @@ fn main() {
 
     let mut handles = vec![];
 
-    for i in 0..test_times {
+    for i in 0..config.test.count {
         let items = Arc::clone(&items);
         let handle = thread::spawn(move || {
             let items = items.lock().unwrap();
-            qts(&items, capacity, max_gen, n_neighbors);
+            qts(
+                &items,
+                capacity,
+                config.algorithm.max_gen,
+                config.algorithm.n_neighbors,
+            );
             println!("thread[{}] done", i);
         });
         handles.push(handle);