wsp 0.1.6

Rust implementation of the WSP space filling algorithm
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64

use std::process;
use structopt::StructOpt;
use wsp::{adaptive_wsp, wsp, PointSet};

/// Set the parameters of the WSP space filling algorithm
#[derive(StructOpt)]
struct Cli {
    /// Output file where the matrix is stored after WSP
    #[structopt(short = "o", long = "output", default_value = "wsp.csv")]
    output_file: String,
    /// Output file where the matrix is stored before WSP
    #[structopt(short = "i", long = "initial")]
    output_file_before: Option<String>,
    /// Algorithm to generate the initial set of candidate points (low impact)
    #[structopt(short = "a", long = "algo", default_value = "random")]
    _initial_algo: String,
    /// Number of points in the initial set of candidate points (major impact)
    #[structopt(short = "n", long = "nb-initial", default_value = "2000")]
    nb_initial: usize,
    /// Minimal distance desired
    #[structopt(short = "d", long = "distance", default_value = "1.0")]
    d_min: f64,
    /// Dimension of the points
    #[structopt(short = "m", long = "dimension", default_value = "20")]
    dim: usize,
    /// Seed for the origin choice and the initialization
    #[structopt(short = "s", long = "seed", default_value = "51")]
    seed: u64,
    /// Use adaptive algorithm instead of distance input to reach <nb-target> active points in the space
    #[structopt(long = "adaptive")]
    nb_target: Option<usize>,
    /// Display debug information. Only for adaptive WSP
    #[structopt(short = "v", long = "verbose")]
    verbose: bool,
    /// Transport the output matrix. Initially the matrix is nb points * nb dims.
    #[structopt(short = "t", long = "transpose")]
    transpose: bool,
}

fn main() {
    let args = Cli::from_args();

    let mut points: PointSet = PointSet::init_from_random(args.nb_initial, args.dim, args.seed);

    if let Some(filename) = args.output_file_before {
        if let Err(err) = points.save_in_csv(&filename, args.transpose) {
            eprintln!("Error writing in CSV: {}", err);
            process::exit(1);
        }
    }

    match args.nb_target {
        Some(obj_nb) => adaptive_wsp(&mut points, obj_nb, args.verbose),
        None => wsp(&mut points, args.d_min),
    }

    if let Err(err) = points.save_in_csv(&args.output_file, args.transpose) {
        eprintln!("Error writing in CSV: {}", err);
        process::exit(1);
    }
    if args.verbose {
        println!("Nb active: {}", points.nb_active);
    }
}