hydro_lang 0.16.0

A Rust framework for correct and performant distributed systems
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

//! File saving utilities for Hydro IR graph visualization.

use std::io::Result;

use super::render::{
    HydroWriteConfig, render_hydro_ir_dot, render_hydro_ir_json, render_hydro_ir_mermaid,
};
use crate::compile::ir::HydroRoot;

fn render_with_config<F>(
    roots: &[HydroRoot],
    config: Option<HydroWriteConfig>,
    renderer: F,
) -> String
where
    F: Fn(&[HydroRoot], HydroWriteConfig<'_>) -> String,
{
    renderer(roots, config.unwrap_or_default())
}

/// Saves Hydro IR roots as a Mermaid diagram file.
pub fn save_mermaid(
    roots: &[HydroRoot],
    filename: Option<&str>,
    config: Option<HydroWriteConfig>,
) -> Result<std::path::PathBuf> {
    let content = render_with_config(roots, config, render_hydro_ir_mermaid);
    save_to_file(content, filename, "hydro_graph.mmd")
}

/// Saves Hydro IR roots as a DOT/Graphviz file.
pub fn save_dot(
    roots: &[HydroRoot],
    filename: Option<&str>,
    config: Option<HydroWriteConfig>,
) -> Result<std::path::PathBuf> {
    let content = render_with_config(roots, config, render_hydro_ir_dot);
    save_to_file(content, filename, "hydro_graph.dot")
}

/// Saves Hydro IR roots as a JSON file.
pub fn save_json(
    roots: &[HydroRoot],
    filename: Option<&str>,
    config: Option<HydroWriteConfig>,
) -> Result<std::path::PathBuf> {
    let content = render_with_config(roots, config, render_hydro_ir_json);
    save_to_file(content, filename, "hydro_graph.json")
}

fn save_to_file(
    content: String,
    filename: Option<&str>,
    default_name: &str,
) -> Result<std::path::PathBuf> {
    let path = std::path::PathBuf::from(filename.unwrap_or(default_name));
    std::fs::write(&path, content)?;
    Ok(path)
}