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

//! rust support for
//! [demes](https://popsim-consortium.github.io/demes-spec-docs).
//!
//! # Example of YAML input
//!
//! ```
//!     let yaml = "
//! description: The Gutenkunst et al. (2009) OOA model.
//! doi:
//! - https://doi.org/10.1371/journal.pgen.1000695
//! time_units: years
//! generation_time: 25
//! 
//! demes:
//! - name: ancestral
//!   description: Equilibrium/root population
//!   epochs:
//!   - {end_time: 220e3, start_size: 7300}
//! - name: AMH
//!   description: Anatomically modern humans
//!   ancestors: [ancestral]
//!   epochs:
//!   - {end_time: 140e3, start_size: 12300}
//! - name: OOA
//!   description: Bottleneck out-of-Africa population
//!   ancestors: [AMH]
//!   epochs:
//!   - {end_time: 21.2e3, start_size: 2100}
//! - name: YRI
//!   description: Yoruba in Ibadan, Nigeria
//!   ancestors: [AMH]
//!   epochs:
//!   - start_size: 12300
//! - name: CEU
//!   description: Utah Residents (CEPH) with Northern and Western European Ancestry
//!   ancestors: [OOA]
//!   epochs:
//!   - {start_size: 1000, end_size: 29725}
//! - name: CHB
//!   description: Han Chinese in Beijing, China
//!   ancestors: [OOA]
//!   epochs:
//!   - {start_size: 510, end_size: 54090}
//! 
//! migrations:
//! - {demes: [YRI, OOA], rate: 25e-5}
//! - {demes: [YRI, CEU], rate: 3e-5}
//! - {demes: [YRI, CHB], rate: 1.9e-5}
//! - {demes: [CEU, CHB], rate: 9.6e-5}
//! ";
//!
//! let graph = match demes::loads(yaml) {
//!     Ok(graph) => graph,
//!     Err(e) => panic!("{}", e),
//! };
//!
//! {
//!     // round trip back into yaml
//!     let yaml_from_graph = serde_yaml::to_string(&graph).unwrap();
//!     let roundtripped_graph = demes::loads(&yaml_from_graph).unwrap();
//!     assert_eq!(graph, roundtripped_graph);
//! }
//!
//! 
//!for deme in graph.demes() {
//!    println!("{} {} {} {} {}",
//!         deme.name(),
//!         deme.start_time(),
//!         deme.end_time(),
//!         deme.start_size(),
//!         deme.end_size());
//!     // A HashMap maps ancestor name -> ancestor Deme
//!     for ancestor in deme.ancestors().keys() {
//!         println!("{} is an ancestor of {}", ancestor, deme.name());
//!     }
//!     // Ref<'_, [String]> of ancestor names
//!     for ancestor in deme.ancestor_names().iter() {
//!         println!("{} is an ancestor of {}", ancestor, deme.name());
//!     }
//!}
//!
//!for m in graph.migrations() {
//!    println!("{} -> {} at rate {} from: {}, to: {}",
//!        m.source(),
//!        m.dest(),
//!        m.rate(),
//!        m.start_time(),
//!        m.end_time());
//!}
//! ```

mod macros;

mod error;
pub mod specification;

use std::io::Read;

pub use error::DemesError;

pub fn loads(yaml: &str) -> Result<specification::Graph, DemesError> {
    specification::Graph::new_resolved_from_str(yaml)
}

pub fn load<T: Read>(reader: T) -> Result<specification::Graph, DemesError> {
    specification::Graph::new_resolved_from_reader(reader)
}