gecco 0.5.3

Gene Cluster prediction with Conditional Random Fields
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

//! Resolve the GECCO data directory containing HMM, CRF model, and InterPro files.
//!
//! Resolution order:
//! 1. Explicit path (from `--data-dir` CLI flag or library API)
//! 2. `GECCO_DATA_DIR` environment variable
//! 3. `gecco_data/` next to the running binary
//! 4. `gecco_data/` in the current working directory

use std::path::PathBuf;

/// Default data directory name placed next to the binary.
const DEFAULT_DIR_NAME: &str = "gecco_data";

/// Resolve the data directory path.
///
/// If `explicit` is `Some`, returns that path directly.
/// Otherwise checks `GECCO_DATA_DIR` env var, then falls back to
/// `gecco_data/` next to the current executable.
pub fn resolve(explicit: Option<&PathBuf>) -> PathBuf {
    if let Some(dir) = explicit {
        return dir.clone();
    }

    if let Ok(dir) = std::env::var("GECCO_DATA_DIR") {
        return PathBuf::from(dir);
    }

    // Next to the binary
    if let Ok(exe) = std::env::current_exe() {
        if let Some(parent) = exe.parent() {
            let candidate = parent.join(DEFAULT_DIR_NAME);
            if candidate.is_dir() {
                return candidate;
            }
        }
    }

    // Current working directory
    let candidate = PathBuf::from(DEFAULT_DIR_NAME);
    if candidate.is_dir() {
        return candidate;
    }

    PathBuf::from(DEFAULT_DIR_NAME)
}

/// Expected file paths within the data directory.
pub fn hmm_path(data_dir: &std::path::Path) -> PathBuf {
    data_dir.join("Pfam.h3m")
}

pub fn interpro_path(data_dir: &std::path::Path) -> PathBuf {
    data_dir.join("interpro.json")
}

pub fn crf_model_path(data_dir: &std::path::Path) -> PathBuf {
    data_dir.join("model.crfsuite")
}

pub fn type_classifier_path(data_dir: &std::path::Path) -> PathBuf {
    data_dir.join("type_classifier.rf.json")
}