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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
type elem = {
num : int;
symb : string;
name : string;
mass : float option;
cpk_color : string option;
e_config : string option;
e_neg : float option;
rad : float option;
ionization_e : float option;
e_affinity : float option;
oxidation : string option;
state : string option;
melting : float option;
boiling : float option;
density : float option;
block : string option;
discovered : int option
}
let float_or_null stmt n = match (Sqlite3.column stmt n) with
| Sqlite3.Data.NULL -> None
| Sqlite3.Data.FLOAT x -> Some x
| _ -> raise (Failure (Printf.sprintf "Column %d has invalid type" n))
let string_or_null stmt n = match (Sqlite3.column stmt n) with
| Sqlite3.Data.NULL -> None
| Sqlite3.Data.TEXT x -> Some x
| _ -> raise (Failure (Printf.sprintf "Column %d has invalid type" n))
let int_or_null stmt n = match (Sqlite3.column stmt n) with
| Sqlite3.Data.NULL -> None
| Sqlite3.Data.INT x -> Some (Int64.to_int x)
| _ -> raise (Failure (Printf.sprintf "Column %d has invalid type" n))
let chk = function
| Sqlite3.Rc.ROW -> ()
| Sqlite3.Rc.OK -> ()
| _ -> raise (Failure "statement failure")
let get_model db typ =
let open Sqlite3
in let stmt = "SELECT dat FROM searchtable WHERE name = ?;"
|> prepare db
in
bind_text stmt 1 typ |> Rc.check;
step stmt |> chk;
let v = column_blob stmt 0
in
finalize stmt |> Rc.check;
Batch_jaro_winkler.build_runtime_model v
let search_model most model s =
let open Batch_jaro_winkler
in let encoding = Encoding.UTF8
in jaro_winkler_distance ~encoding ~n_best_results:most model s
let by_name n db s = search_model n (get_model db "name") s
let name_stmt db s =
let open Sqlite3
in let stmt = "SELECT \
num, symb, name, mass, cpk_color, e_config, e_neg, rad, \
ionization_e, e_affinity, oxidation, state, melting, \
boiling, density, block, discovered \
FROM elements WHERE name = ?;"
|> prepare db
in bind_text stmt 1 s |> Rc.check;
stmt
let get_elem_from_stmt stmt =
let open Sqlite3
in
step stmt |> chk; {
num = column_int stmt 0;
symb = column_text stmt 1;
name = column_text stmt 2;
mass = float_or_null stmt 3;
cpk_color = string_or_null stmt 4;
e_config = string_or_null stmt 5;
e_neg = float_or_null stmt 6;
rad = float_or_null stmt 7;
ionization_e = float_or_null stmt 8;
e_affinity = float_or_null stmt 9;
oxidation = string_or_null stmt 10;
state = string_or_null stmt 11;
melting = float_or_null stmt 12;
boiling = float_or_null stmt 13;
density = float_or_null stmt 14;
block = string_or_null stmt 15;
discovered = int_or_null stmt 16
}
let get_elem db s = name_stmt db s |> get_elem_from_stmt
let print_elem e =
let p_str name = Option.iter (fun x ->
Printf.printf "%s: %s\n" name x)
in let p_flt name = Option.iter (fun x ->
Printf.printf "%s: %f\n" name x)
in let p_int name = Option.iter (fun x ->
Printf.printf "%s: %d\n" name x)
in Printf.printf "'%s' -- '%s' -- '%d'\n" e.name e.symb e.num;
p_flt "Standard Atomic Mass" e.mass;
p_str "CPK Color" e.cpk_color;
p_str "Electron Configuration" e.e_config;
p_flt "Electronegativity (Pauling)" e.e_neg;
p_flt "Van der Walls (pm)" e.rad;
p_flt "Ionization Energy (eV)" e.ionization_e;
p_flt "Electron Affinity (eV)" e.e_affinity;
p_str "Oxidation States" e.oxidation;
p_str "Standard State" e.state;
p_flt "Melting Point (K)" e.melting;
p_flt "Boiling Point (K)" e.boiling;
p_flt "Density (g/cm³)" e.density;
p_str "Block" e.block;
p_int "Year Discovered" e.discovered;
print_newline()
let print_by_name db most l =
let serc = by_name (Some most) db
in let on_match v = (match serc (String.lowercase_ascii v) with
| [] -> Printf.printf "No matches found for '%s'\n" v;
| (m,p)::[] -> if m <> v then
Printf.printf
"Best match '%s' for '%s' (%.0f%%)\n"
m v (p *. 100.0);
get_elem db m |> print_elem
| l ->
Printf.printf "Matches for '%s'\n" v;
List.iter (fun (a,b) -> Printf.printf "%f -- %s\n" b a)
(List.rev l)
)
in List.iter on_match l
|
