th-rust 0.1.0

ThRust is a software framework for thermodynamic and probabilistic computing.
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

/// Plot sweep results
use plotters::prelude::*;

pub struct Plotter {
	data: Vec<u32>,
	buckets: usize,
}

impl Plotter {
	pub fn new(buckets: usize) -> Plotter {
		Plotter {
			data: Vec::new(),
			buckets
		}
	}

	pub fn add_sample(&mut self, config: &Vec<i32>) {
		// Compute number representation
		let number = self.config_to_num(config);
		self.data.push(number);
	}

	pub fn plot(&self, name: &str) -> Result<(), Box<dyn std::error::Error>> {
		// Setup plot
		let path = String::from("out/") + &name;
		let root = BitMapBackend::new(&path, (1024, 768)).into_drawing_area();
		root.fill(&WHITE)?;

		let mut chart = ChartBuilder::on(&root)
		.x_label_area_size(35)
		.y_label_area_size(40)
		.margin(5)
		.caption("MCMC Sweep", ("sans-serif", 50.0))
		.build_cartesian_2d((0u32..(self.buckets - 1) as u32).into_segmented(), 0u32..3000u32)?;

		chart
			.configure_mesh()
			.disable_x_mesh()
			.bold_line_style(&WHITE.mix(0.3))
			.y_desc("Frequency")
			.x_desc("Spin State")
			.axis_desc_style(("sans-serif", 15))
			.draw()?;

		chart.draw_series(
			Histogram::vertical(&chart)
				.style(BLUE.mix(0.5).filled())
				.data(self.data.iter().map(|x: &u32| (*x, 1))),
		).unwrap();

		root.present().expect("Unable to write result to file, please make sure 'plotters-doc-data' dir exists under current dir");
		Ok(())
	}

	fn spin_to_bit(&self, spin: i32) -> u32 {
		if spin == 1 {
			1
		} else {
			0
		}
	}

	fn config_to_num(&self, c: &Vec<i32>) -> u32 {
		let mut power = 1;
		let mut number = 0;

		for val in c.iter().rev() {
			number += self.spin_to_bit(*val) * power;
				power <<= 1;
		}

		number
	}
}