summaryrefslogtreecommitdiff
path: root/src/vytree.ml
blob: ec48848c6d0ababa1cd34cddab51fb65a8717936 (plain)
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
type 'a	t = {
    name: string;
    data: 'a;
    children: 'a t list
}

type position = Before of string | After of string | Default

type node_type = Leaf_single | Leaf_multi | Tag | Other

exception Empty_path
exception Duplicate_child
exception Nonexistent_path
exception Insert_error of string

let make data name = { name = name; data = data; children = [] }

let make_full data name children = { name = name; data = data; children = children }

let name_of_node node = node.name
let data_of_node node = node.data
let children_of_node node = node.children

let insert_immediate node name data =
    let new_node = make data name in
    let children' = new_node :: node.children in
    { node with children = children' }

let delete_immediate node name =
    let children' = Vylist.remove (fun x -> x.name = name) node.children in
    { node with children = children' }

let adopt node child =
    { node with children = child :: node.children }

let replace node child =
    let children = node.children in
    let name = child.name in
    let children' = Vylist.replace (fun x -> x.name = name) child children in
    { node with children = children' }

let find node name =
    Vylist.find (fun x -> x.name = name) node.children

let find_or_fail node name =
    let child = find node name in
    match child with
    | None -> raise Nonexistent_path
    | Some child' -> child'

let list_children node =
    List.map (fun x -> x.name) node.children

let rec do_with_child fn node path =
    match path with
    | [] -> raise Empty_path
    | [name] -> fn node name
    | name :: names ->
        let next_child = find_or_fail node name in
        let new_node = do_with_child fn next_child names in
        replace node new_node

let rec insert node path data_list =
    match (path, data_list) with
    | ([], _) -> raise Empty_path
    | (_ :: _, []) -> raise (Insert_error "No data found")
    | ([name], [datum]) ->
        begin
            (* Check for duplicate item *)
            let last_child = find node name in
            match last_child with
            | None -> insert_immediate node name datum
            | (Some _) -> raise Duplicate_child
        end
    | (name :: names, datum :: data) ->
        let next_child = find node name in
        match next_child with
        | Some next_child' ->
            let new_node = insert next_child' names data in
            replace node new_node
        | None ->
            let next_child' = make datum name in
            let new_node = insert next_child' names data in
            adopt node new_node

let delete node path =
    do_with_child delete_immediate node path

let update node path data =
    let update_data data' node' name =
        let child = find_or_fail node' name in
        let child = { child with data=data' } in
        replace node' child
    in do_with_child (update_data data) node path

let rec get node path =
    match path with
    | [] -> raise Empty_path
    | [name] -> find_or_fail node name
    | name :: names -> get (find_or_fail node name) names