rustling 0.8.0

A blazingly fast library for computational linguistics
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

use std::path::PathBuf;
use std::process::Command;

fn main() {
    let manifest_dir = PathBuf::from(std::env::var("CARGO_MANIFEST_DIR").unwrap());
    let out_dir = PathBuf::from(std::env::var("OUT_DIR").unwrap());

    // (schema path, OUT_DIR subdirectory, pre-generated fallback file)
    let schemas: &[(&str, &str, &str)] = &[
        ("src/hmm/model.fbs", "hmm", "src/hmm/model_generated.rs"),
        (
            "src/perceptron_pos_tagger/model.fbs",
            "perceptron_pos_tagger",
            "src/perceptron_pos_tagger/model_generated.rs",
        ),
        (
            "src/wordseg/longest_string_matching/model.fbs",
            "lsm",
            "src/wordseg/longest_string_matching/model_generated.rs",
        ),
        ("src/lm/model.fbs", "lm", "src/lm/model_generated.rs"),
    ];

    let has_flatc = Command::new("flatc")
        .arg("--version")
        .output()
        .map(|o| o.status.success())
        .unwrap_or(false);

    for (schema_rel, out_subdir, fallback_rel) in schemas {
        let schema_path = manifest_dir.join(schema_rel);
        println!("cargo:rerun-if-changed={}", schema_path.display());

        let out_subpath = out_dir.join(out_subdir);
        std::fs::create_dir_all(&out_subpath).unwrap();

        if has_flatc {
            let status = Command::new("flatc")
                .args(["--rust", "-o"])
                .arg(&out_subpath)
                .arg(&schema_path)
                .status()
                .expect("flatc invocation failed");
            assert!(
                status.success(),
                "flatc failed for {}",
                schema_path.display()
            );
        } else {
            // Fall back to pre-generated file committed in the repository.
            let fallback_path = manifest_dir.join(fallback_rel);
            let dest = out_subpath.join("model_generated.rs");
            std::fs::copy(&fallback_path, &dest).unwrap_or_else(|e| {
                panic!(
                    "flatc not found and fallback file {} is missing: {}",
                    fallback_path.display(),
                    e
                )
            });
        }
    }
}