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

pub mod orbital;

use proc_macro2::{Delimiter, Group, TokenStream, TokenTree};
use quote::{quote, TokenStreamExt, ToTokens};

use orbital::Orbital;

#[derive(Copy, Clone, Debug, PartialEq, PartialOrd)]
pub struct ElectronData {
    pub electron_configuration: ElectronConfiguration,
    pub ionisation_energies: [f64; 30],
}

/// Representation of electron configuration using StaticVec
#[derive(Copy, Clone, Debug, PartialEq, PartialOrd)]
pub struct ElectronConfiguration(pub(crate) [Orbital; 6]);

impl ToTokens for ElectronData {
    fn to_tokens(&self, tokens: &mut TokenStream) {
        let electron_configuration = &self.electron_configuration;
        let ionisation_energies = self.ionisation_energies.iter();

        let add_tokens = quote! {
            chemistru_elements::raw::ElectronData {
                electron_configuration: #electron_configuration,
                ionisation_energies: staticvec::StaticVec::new_from_slice(&[#(#ionisation_energies),*]),
            }
        };

        let group = TokenTree::Group(Group::new(
            Delimiter::Brace,
            add_tokens,
        ));

        tokens.append(group);
    }
}


impl ToTokens for ElectronConfiguration {
    fn to_tokens(&self, tokens: &mut TokenStream) {
        let orbitals = self.0;

        let add_tokens = quote! {
            chemistru_elements::data::ElectronConfiguration([#(#orbitals),*])
        };

        tokens.append(TokenTree::Group(Group::new(Delimiter::Brace, add_tokens)));
    }
}