rust/land_dump/src/main.rs
author alfadur
Sun, 04 Nov 2018 07:16:34 +0300
changeset 14133 b04dac00e8e2
parent 14130 376a0551b00a
child 14135 ab280be4b617
permissions -rw-r--r--
add command arguments to use a template from file into land_dump
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
14126
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
     1
use png::HasParameters;
14133
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
     2
use std::{
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
     3
    fs::File,
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
     4
    io::{BufWriter, Read},
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
     5
    path::{Path, PathBuf}
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
     6
};
14126
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
     7
use structopt::StructOpt;
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
     8
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
     9
use integral_geometry::{Point, Rect, Size};
14133
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    10
use landgen::{
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    11
    outline_template::OutlineTemplate,
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    12
    template_based::TemplatedLandGenerator,
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    13
    LandGenerationParameters,
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    14
    LandGenerator
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    15
};
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    16
use mapgen::MapGenerator;
14126
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    17
use lfprng::LaggedFibonacciPRNG;
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    18
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    19
#[derive(StructOpt, Debug)]
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    20
#[structopt(name = "basic")]
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    21
struct Opt {
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    22
    #[structopt(short = "s", long = "seed", default_value = "TEST_SEED")]
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    23
    seed: String,
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    24
    #[structopt(short = "d", long = "dump-before-distort")]
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    25
    dump_before_distort: bool,
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    26
    #[structopt(short = "b", long = "dump-before-bezierize")]
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    27
    dump_before_bezierize: bool,
14130
376a0551b00a - Add distance-divider option to land_dump
unc0rr
parents: 14126
diff changeset
    28
    #[structopt(short = "f", long = "distance-divisor", default_value = "100")]
376a0551b00a - Add distance-divider option to land_dump
unc0rr
parents: 14126
diff changeset
    29
    distance_divisor: u32,
14133
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    30
    #[structopt(short = "i", long = "templates-file")]
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    31
    templates_file: Option<String>,
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    32
    #[structopt(short = "t", long = "template-type")]
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    33
    template_type: Option<String>
14126
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    34
}
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    35
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    36
fn template() -> OutlineTemplate {
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    37
    let mut template = OutlineTemplate::new(Size::new(4096, 2048));
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    38
    template.islands = vec![vec![
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    39
        Rect::new(100, 2050, 1, 1),
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    40
        Rect::new(100, 500, 400, 1200),
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    41
        Rect::new(3600, 500, 400, 1200),
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    42
        Rect::new(3900, 2050, 1, 1),
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    43
    ]];
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    44
    template.fill_points = vec![Point::new(2047, 2047)];
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    45
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    46
    template
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    47
}
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    48
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    49
fn dump(
14133
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    50
    template: &OutlineTemplate,
14126
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    51
    seed: &[u8],
14130
376a0551b00a - Add distance-divider option to land_dump
unc0rr
parents: 14126
diff changeset
    52
    distance_divisor: u32,
14126
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    53
    skip_distort: bool,
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    54
    skip_bezier: bool,
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    55
    file_name: &Path,
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    56
) -> std::io::Result<()> {
14130
376a0551b00a - Add distance-divider option to land_dump
unc0rr
parents: 14126
diff changeset
    57
    let params = LandGenerationParameters::new(0 as u8, 255, distance_divisor, skip_distort, skip_bezier);
14133
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    58
    let landgen = TemplatedLandGenerator::new(template.clone());
14126
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    59
    let mut prng = LaggedFibonacciPRNG::new(seed);
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    60
    let land = landgen.generate_land(&params, &mut prng);
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    61
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    62
    let file = File::create(file_name)?;
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    63
    let ref mut w = BufWriter::new(file);
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    64
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    65
    let mut encoder = png::Encoder::new(w, land.width() as u32, land.height() as u32); // Width is 2 pixels and height is 1.
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    66
    encoder
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    67
        .set(png::ColorType::Grayscale)
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    68
        .set(png::BitDepth::Eight);
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    69
    let mut writer = encoder.write_header()?;
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    70
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    71
    writer.write_image_data(land.raw_pixels()).unwrap();
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    72
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    73
    Ok(())
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    74
}
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    75
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    76
fn main() {
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    77
    let opt = Opt::from_args();
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    78
    println!("{:?}", opt);
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
    79
14133
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    80
    let template =
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    81
        if let Some(path) = opt.templates_file {
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    82
            let mut result = String::new();
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    83
            File::open(path)
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    84
                .expect("Unable to read templates file")
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    85
                .read_to_string(&mut result);
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    86
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    87
            let mut generator = MapGenerator::new();
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    88
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    89
            let bom = b"\xEF\xBB\xBF";
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    90
            let source = if &result.as_bytes()[..bom.len()] == &bom[..] {
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    91
                &result[bom.len()..]
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    92
            } else {
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    93
                &result[..]
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    94
            };
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    95
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    96
            generator.import_yaml_templates(source);
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    97
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    98
            let template_type = &opt.template_type
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
    99
                .expect("No template type specified");
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
   100
            generator.get_template(template_type)
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
   101
                .expect(&format!("Template type {} not found", template_type))
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
   102
                .clone()
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
   103
        } else {
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
   104
            template()
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
   105
        };
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
   106
14126
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
   107
    if opt.dump_before_distort {
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
   108
        dump(
14133
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
   109
            &template,
14126
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
   110
            opt.seed.as_str().as_bytes(),
14130
376a0551b00a - Add distance-divider option to land_dump
unc0rr
parents: 14126
diff changeset
   111
            opt.distance_divisor,
14126
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
   112
            true,
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
   113
            true,
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
   114
            Path::new("out.before_distort.png"),
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
   115
        )
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
   116
        .unwrap();
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
   117
    }
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
   118
    if opt.dump_before_bezierize {
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
   119
        dump(
14133
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
   120
            &template,
14126
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
   121
            opt.seed.as_str().as_bytes(),
14130
376a0551b00a - Add distance-divider option to land_dump
unc0rr
parents: 14126
diff changeset
   122
            opt.distance_divisor,
14126
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
   123
            false,
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
   124
            true,
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
   125
            Path::new("out.before_bezier.png"),
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
   126
        )
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
   127
        .unwrap();
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
   128
    }
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
   129
    dump(
14133
b04dac00e8e2 add command arguments to use a template from file into land_dump
alfadur
parents: 14130
diff changeset
   130
        &template,
14126
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
   131
        opt.seed.as_str().as_bytes(),
14130
376a0551b00a - Add distance-divider option to land_dump
unc0rr
parents: 14126
diff changeset
   132
        opt.distance_divisor,
14126
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
   133
        false,
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
   134
        true,
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
   135
        Path::new("out.full.png"),
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
   136
    )
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
   137
    .unwrap();
69db1d2e4cec land_dump app for testing templated landgen
unc0rr
parents:
diff changeset
   138
}