unlab-gpu 0.1.0

Micro scripting language for neural networks that uses unmtx-gpu.
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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114

//
// Copyright (c) 2025-2026 Łukasz Szpakowski
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at https://mozilla.org/MPL/2.0/.
//
use std::ffi::OsString;
use std::path::PathBuf;
use std::process::exit;
use std::sync::Arc;
use std::sync::RwLock;
use clap::Parser;
use unlab_gpu::Home;
use unlab_gpu::ModNode;
use unlab_gpu::Value;
use unlab_gpu::add_std_builtin_funs;
use unlab_gpu::finalize_backend;
use unlab_gpu::initialize_backend;
use unlab_gpu::main_loop;

#[derive(Parser, Debug)]
#[command(version)]
struct Args
{
    /// Unlab-gpu home directory
    #[arg(short = 'H', long)]
    home_dir: Option<String>,
    /// Binary path
    #[arg(short = 'B', long)]
    bin_path: Option<String>,
    /// Library path
    #[arg(short = 'L', long)]
    lib_path: Option<String>,
    /// Documentation path
    #[arg(short = 'D', long)]
    doc_path: Option<String>,
    /// Add directory to binary path
    #[arg(short, long)]
    bin_dir: Vec<String>,
    /// Add directory to library path
    #[arg(short, long)]
    lib_dir: Vec<String>,
    /// Add directory to documentation path
    #[arg(short, long)]
    doc_dir: Vec<String>,
    /// Don't handle CTRL-C
    #[arg(short, long)]
    no_ctrl_c: bool,
    /// Don't show plotter windows
    #[arg(short = 'p', long)]
    no_plotter_windows: bool,
    /// Script file
    script_file: Option<String>,
    /// Arguments
    args: Vec<String>,
}

fn main()
{
    let args = Args::parse();
    let mut home = match Home::new(&args.home_dir, &args.bin_path, &args.lib_path, &args.doc_path, false) {
        Some(tmp_home) => tmp_home,
        None => {
            eprintln!("no unlab-gpu home directory");
            exit(1);
        },
    };
    match home.add_dirs_to_bin_path(args.bin_dir.as_slice()) {
        Ok(()) => (),
        Err(err) => {
            eprintln!("{}", err);
            exit(1);
        },
    }
    match home.add_dirs_to_lib_path(args.lib_dir.as_slice()) {
        Ok(()) => (),
        Err(err) => {
            eprintln!("{}", err);
            exit(1);
        },
    }
    match home.add_dirs_to_doc_path(args.doc_dir.as_slice()) {
        Ok(()) => (),
        Err(err) => {
            eprintln!("{}", err);
            exit(1);
        },
    }
    match initialize_backend(home.backend_config_file()) {
        Ok(()) => (),
        Err(err) => {
            eprintln!("{}", err);
            exit(1);
        },
    }
    let exit_code = {
        let mut root_mod: ModNode<Value, ()> = ModNode::new(());
        add_std_builtin_funs(&mut root_mod);
        let root_mod_arc = Arc::new(RwLock::new(root_mod));
        main_loop(args.script_file.clone(), args.args.clone(), PathBuf::from(home.history_file()), root_mod_arc, OsString::from(home.lib_path()), OsString::from(home.doc_path()), !args.no_ctrl_c, !args.no_plotter_windows)
    };
    match finalize_backend() {
        Ok(()) => (),
        Err(err) => {
            eprintln!("{}", err);
            exit(1);
        },
    }
    match exit_code {
        Some(exit_code) => exit(exit_code),
        None => (),
    }
}