Add gen-only flag to only build solver without solving (#43)

examples/heat_1d_ap_fft.rs

@@ -38,6 +38,10 @@ fn main() {
         dot_path.push("plan.dot");
         solver.to_dot_file(&dot_path);
     }
+    if args.gen_only {
+        args.save_wisdom();
+        std::process::exit(0);
+    }
 
     let mut img = None;
     if args.write_image {

examples/heat_2d_ap_fft.rs

@@ -36,6 +36,10 @@ fn main() {
         d_path.push("scratch.txt");
         solver.scratch_descriptor_file(&d_path);
     }
+    if args.gen_only {
+        args.save_wisdom();
+        std::process::exit(0);
+    }
 
     // Create domains
     let mut buffer_1 = OwnedDomain::new(grid_bound);

examples/heat_3d_ap_fft.rs

@@ -34,6 +34,10 @@ fn main() {
         dot_path.push("plan.dot");
         solver.to_dot_file(&dot_path);
     }
+    if args.gen_only {
+        args.save_wisdom();
+        std::process::exit(0);
+    }
 
     // Create domains
     let mut buffer_1 = OwnedDomain::new(grid_bound);

examples/time_varying_2d.rs

@@ -93,6 +93,10 @@ fn main() {
         d_path.push("scratch.txt");
         solver.scratch_descriptor_file(&d_path);
     }
+    if args.gen_only {
+        args.save_wisdom();
+        std::process::exit(0);
+    }
 
     // Create domains
     let mut buffer_1 = OwnedDomain::new(grid_bound);

src/image_1d_example.rs

@@ -60,6 +60,10 @@ pub struct Args {
     /// Cutoff for fft solves
     #[arg(long, default_value = "40")]
     pub cutoff: i32,
+
+    /// Generate solver only, do not solve
+    #[arg(long)]
+    pub gen_only: bool,
 }
 
 impl Args {

src/image_2d_example.rs

@@ -60,6 +60,10 @@ pub struct Args {
     /// Cutoff for fft solves
     #[arg(long, default_value = "40")]
     pub cutoff: i32,
+
+    /// Generate solver only, do not solve
+    #[arg(long)]
+    pub gen_only: bool,
 }
 
 impl Args {

src/image_3d_example.rs

@@ -60,6 +60,10 @@ pub struct Args {
     /// Cutoff for fft solves
     #[arg(long, default_value = "40")]
     pub cutoff: i32,
+
+    /// Generate solver only, do not solve
+    #[arg(long)]
+    pub gen_only: bool,
 }
 
 impl Args {