simian 0.2.1

A command-line tool for exploring and implementing Machine Learning algorithms in Rust.
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

use anyhow::Result;
use plotly::Plot;

pub fn handle_plot_html(mut html: String, out_path: Option<&str>) -> Result<()> {
  html = html.replace(
    "<head>",
    "<head>\n    <style>html, body { height: 100%; margin: 0; padding: 0; overflow: hidden; } .plotly-graph-div { height: 100vh !important; width: 100vw !important; }</style>"
  );
  if std::env::var("SIMIAN_STUDIO_DIR").is_ok() {
    crate::paper::render(crate::paper::RenderFormat::Html, &html);
  } else {
    if let Some(p) = out_path {
      std::fs::write(p, html)?;
      println!("Saved plot to {}", p);
      let _ = std::process::Command::new("open").arg(p).spawn();
    }
  }

  Ok(())
}

pub fn generate_plot_html(plot: &Plot) -> String {
  let mut html = plot.to_html();

  let css_injection = "<style>html, body { height: 100%; margin: 0; padding: 0; overflow: hidden; } .plotly-graph-div { height: 100vh !important; width: 100vw !important; }</style>";

  let js_injection = r#"
<script>
    window.addEventListener('load', function() {
        var graph = document.getElementsByClassName('plotly-graph-div')[0];
        if (graph && typeof Plotly !== 'undefined') {
            if (window.location.hash) {
                try {
                    var state = JSON.parse(atob(window.location.hash.slice(1)));
                    Plotly.relayout(graph, state);
                } catch(e) { console.error('Failed to restore plot state', e); }
            }
            var timeoutId = null;
            graph.on('plotly_relayout', function(eventData) {
                if (timeoutId) clearTimeout(timeoutId);
                timeoutId = setTimeout(function() {
                    window.parent.postMessage({ type: 'SIMIAN_PLOT_STATE', state: eventData }, '*');
                }, 500);
            });
        }
    });
</script>
"#;

  if html.contains("</head>") {
    html = html.replace("</head>", &format!("{}\n</head>", css_injection));
  } else {
    html = format!("{}\n{}", css_injection, html);
  }

  if html.contains("</body>") {
    html = html.replace("</body>", &format!("{}\n</body>", js_injection));
  } else {
    html = format!("{}\n{}", html, js_injection);
  }

  html
}