1 | // =============================================================== // |
---|
2 | // // |
---|
3 | // File : MG_trees.cxx // |
---|
4 | // Purpose : Merge trees between databases // |
---|
5 | // // |
---|
6 | // Institute of Microbiology (Technical University Munich) // |
---|
7 | // http://www.arb-home.de/ // |
---|
8 | // // |
---|
9 | // =============================================================== // |
---|
10 | |
---|
11 | #include "merge.hxx" |
---|
12 | #include <TreeAdmin.h> |
---|
13 | #include <awt_sel_boxes.hxx> |
---|
14 | #include <aw_root.hxx> |
---|
15 | #include <aw_awar.hxx> |
---|
16 | #include <aw_msg.hxx> |
---|
17 | #include <arbdbt.h> |
---|
18 | #include <arb_global_defs.h> |
---|
19 | |
---|
20 | #define AWAR_TREE_NAME_SRC AWAR_MERGE_TMP_SRC "tree_name" |
---|
21 | #define AWAR_TREE_NAME_DST AWAR_MERGE_TMP_DST "tree_name" |
---|
22 | |
---|
23 | #define AWAR_TREE_XFER_WHAT AWAR_MERGE_TMP "xfer_what" |
---|
24 | #define AWAR_TREE_OVERWRITE AWAR_MERGE_TMP "overwrite" |
---|
25 | |
---|
26 | enum TREE_XFER_MODE { |
---|
27 | XFER_SELECTED, |
---|
28 | XFER_ALL, |
---|
29 | XFER_MISSING, |
---|
30 | XFER_EXISTING, |
---|
31 | }; |
---|
32 | |
---|
33 | void MG_create_trees_awar(AW_root *aw_root, AW_default aw_def) { |
---|
34 | aw_root->awar_string(AWAR_TREE_NAME_SRC, NO_TREE_SELECTED, aw_def); |
---|
35 | aw_root->awar_string(AWAR_TREE_NAME_DST, NO_TREE_SELECTED, aw_def); |
---|
36 | |
---|
37 | aw_root->awar_int(AWAR_TREE_XFER_WHAT, XFER_SELECTED, aw_def); |
---|
38 | aw_root->awar_int(AWAR_TREE_OVERWRITE, 0, aw_def); |
---|
39 | |
---|
40 | TreeAdmin::create_awars(aw_root, aw_def, false); |
---|
41 | } |
---|
42 | |
---|
43 | static GB_ERROR transfer_tree(const char *tree_name, bool overwrite, const char *behind_name) { |
---|
44 | // transfer tree 'tree_name' from DB1->DB2. |
---|
45 | // |
---|
46 | // if 'overwrite' is true and tree already exists, overwrite tree w/o changing its position. |
---|
47 | // |
---|
48 | // if tree does not exist, insert it behind tree 'behind_name'. |
---|
49 | // if tree 'behind_name' does not exist or if 'behind_name' is NULp, insert at end. |
---|
50 | |
---|
51 | |
---|
52 | GB_ERROR error = NULp; |
---|
53 | GBDATA *gb_tree = GBT_find_tree(GLOBAL_gb_src, tree_name); |
---|
54 | if (!gb_tree) { |
---|
55 | if (strcmp(tree_name, NO_TREE_SELECTED) == 0) { |
---|
56 | error = "No tree selected in source DB"; |
---|
57 | } |
---|
58 | else { |
---|
59 | error = GBS_global_string("No tree '%s' in source DB", tree_name); |
---|
60 | } |
---|
61 | } |
---|
62 | else { |
---|
63 | GBDATA *gb_at_tree = NULp; |
---|
64 | GBT_ORDER_MODE next_to = GBT_BEHIND; |
---|
65 | |
---|
66 | if (behind_name && strcmp(behind_name, NO_TREE_SELECTED) != 0) { |
---|
67 | gb_at_tree = GBT_find_tree(GLOBAL_gb_dst, behind_name); |
---|
68 | if (!gb_at_tree) { |
---|
69 | error = GBS_global_string("Can't position tree behind '%s' (no such tree)", behind_name); |
---|
70 | } |
---|
71 | } |
---|
72 | |
---|
73 | if (!error) { |
---|
74 | GBDATA *gb_dest_tree = GBT_find_tree(GLOBAL_gb_dst, tree_name); |
---|
75 | if (gb_dest_tree) { |
---|
76 | if (!overwrite) { |
---|
77 | error = GBS_global_string("Tree '%s' already exists in destination DB", tree_name); |
---|
78 | } |
---|
79 | else { |
---|
80 | // keep position of existing tree |
---|
81 | GBDATA *gb_tree_infrontof = GBT_tree_infrontof(gb_dest_tree); |
---|
82 | if (gb_tree_infrontof) { |
---|
83 | next_to = GBT_BEHIND; |
---|
84 | gb_at_tree = gb_tree_infrontof; |
---|
85 | } |
---|
86 | else { |
---|
87 | GBDATA *gb_tree_behind = GBT_tree_behind(gb_dest_tree); |
---|
88 | next_to = GBT_INFRONTOF; |
---|
89 | gb_at_tree = gb_tree_behind; |
---|
90 | } |
---|
91 | |
---|
92 | error = GB_delete(gb_dest_tree); |
---|
93 | } |
---|
94 | } |
---|
95 | } |
---|
96 | |
---|
97 | if (!error) { |
---|
98 | GBDATA *gb_dest_tree_data = GBT_get_tree_data(GLOBAL_gb_dst); |
---|
99 | if (!gb_dest_tree_data) error = GB_await_error(); |
---|
100 | else { |
---|
101 | GBDATA *gb_dest_tree = GB_create_container(gb_dest_tree_data, tree_name); |
---|
102 | if (!gb_dest_tree) error = GB_await_error(); |
---|
103 | else { |
---|
104 | error = GB_copy_dropProtectMarksAndTempstate(gb_dest_tree, gb_tree); |
---|
105 | if (!error) { |
---|
106 | if (!gb_at_tree) { |
---|
107 | gb_at_tree = GBT_find_bottom_tree(GLOBAL_gb_dst); |
---|
108 | next_to = GBT_BEHIND; |
---|
109 | } |
---|
110 | if (gb_at_tree) { |
---|
111 | error = GBT_move_tree(gb_dest_tree, next_to, gb_at_tree); |
---|
112 | } |
---|
113 | } |
---|
114 | } |
---|
115 | } |
---|
116 | } |
---|
117 | } |
---|
118 | return error; |
---|
119 | } |
---|
120 | |
---|
121 | static void MG_transfer_tree(AW_window *aww) { |
---|
122 | AW_root *awr = aww->get_root(); |
---|
123 | |
---|
124 | TREE_XFER_MODE what = (TREE_XFER_MODE)awr->awar(AWAR_TREE_XFER_WHAT)->read_int(); |
---|
125 | |
---|
126 | AW_awar *awar_tree_source = awr->awar(AWAR_TREE_NAME_SRC); |
---|
127 | AW_awar *awar_tree_dest = awr->awar(AWAR_TREE_NAME_DST); |
---|
128 | |
---|
129 | bool overwrite = awr->awar(AWAR_TREE_OVERWRITE)->read_int(); |
---|
130 | |
---|
131 | char *source_name = awar_tree_source->read_string(); |
---|
132 | char *behind_name = awar_tree_dest->read_string(); |
---|
133 | |
---|
134 | char *select_dst = NULp; |
---|
135 | |
---|
136 | GB_ERROR error = GB_begin_transaction(GLOBAL_gb_dst); |
---|
137 | if (!error) error = GB_begin_transaction(GLOBAL_gb_src); |
---|
138 | |
---|
139 | int xferd_missing = 0; |
---|
140 | int xferd_existing = 0; |
---|
141 | |
---|
142 | if (!error) { |
---|
143 | switch (what) { |
---|
144 | case XFER_SELECTED: |
---|
145 | error = transfer_tree(source_name, overwrite, behind_name); |
---|
146 | if (!error) select_dst = ARB_strdup(source_name); |
---|
147 | break; |
---|
148 | |
---|
149 | case XFER_ALL: |
---|
150 | case XFER_EXISTING: |
---|
151 | for (GBDATA *gb_tree = GBT_find_top_tree(GLOBAL_gb_src); gb_tree && !error; gb_tree = GBT_tree_behind(gb_tree)) { |
---|
152 | const char *tree_name = GBT_get_tree_name(gb_tree); |
---|
153 | GBDATA *gb_exists = GBT_find_tree(GLOBAL_gb_dst, tree_name); |
---|
154 | |
---|
155 | if (gb_exists) { |
---|
156 | xferd_existing++; |
---|
157 | error = transfer_tree(tree_name, overwrite, NULp); |
---|
158 | } |
---|
159 | } |
---|
160 | if (what == XFER_EXISTING) break; |
---|
161 | FALLTHROUGH; // for XFER_ALL |
---|
162 | case XFER_MISSING: |
---|
163 | for (GBDATA *gb_tree = GBT_find_top_tree(GLOBAL_gb_src); gb_tree && !error; gb_tree = GBT_tree_behind(gb_tree)) { |
---|
164 | const char *tree_name = GBT_get_tree_name(gb_tree); |
---|
165 | GBDATA *gb_exists = GBT_find_tree(GLOBAL_gb_dst, tree_name); |
---|
166 | |
---|
167 | if (!gb_exists) { |
---|
168 | error = transfer_tree(tree_name, false, behind_name); |
---|
169 | xferd_missing++; |
---|
170 | if (!select_dst) select_dst = ARB_strdup(tree_name); // select first missing in dest box |
---|
171 | freedup(behind_name, tree_name); |
---|
172 | } |
---|
173 | } |
---|
174 | break; |
---|
175 | } |
---|
176 | } |
---|
177 | |
---|
178 | error = GB_end_transaction(GLOBAL_gb_dst, error); |
---|
179 | GB_end_transaction_show_error(GLOBAL_gb_src, error, aw_message); |
---|
180 | |
---|
181 | if (!error) { |
---|
182 | if (select_dst) awar_tree_dest->write_string(select_dst); |
---|
183 | |
---|
184 | if (what == XFER_SELECTED) { |
---|
185 | GB_transaction ta(GLOBAL_gb_src); |
---|
186 | GBDATA *gb_next = GBT_find_next_tree(GBT_find_tree(GLOBAL_gb_src, source_name)); |
---|
187 | awar_tree_source->write_string(gb_next ? GBT_get_tree_name(gb_next) : NO_TREE_SELECTED); |
---|
188 | } |
---|
189 | |
---|
190 | if (xferd_existing) { |
---|
191 | if (xferd_missing) aw_message(GBS_global_string("Transferred %i existing and %i missing trees", xferd_existing, xferd_missing)); |
---|
192 | else aw_message(GBS_global_string("Transferred %i existing trees", xferd_existing)); |
---|
193 | } |
---|
194 | else { |
---|
195 | if (xferd_missing) aw_message(GBS_global_string("Transferred %i missing trees", xferd_missing)); |
---|
196 | else { |
---|
197 | if (what != XFER_SELECTED) aw_message("No trees have been transferred"); |
---|
198 | } |
---|
199 | } |
---|
200 | } |
---|
201 | |
---|
202 | free(select_dst); |
---|
203 | free(behind_name); |
---|
204 | free(source_name); |
---|
205 | } |
---|
206 | |
---|
207 | AW_window *MG_create_merge_trees_window(AW_root *awr) { |
---|
208 | GB_ERROR error = MG_expect_renamed(); |
---|
209 | if (error) { |
---|
210 | aw_message(error); |
---|
211 | return NULp; // deny to open window before user has renamed species |
---|
212 | } |
---|
213 | |
---|
214 | AW_window_simple *aws = new AW_window_simple; |
---|
215 | |
---|
216 | aws->init(awr, "MERGE_TREES", "MERGE TREES"); |
---|
217 | aws->load_xfig("merge/trees.fig"); |
---|
218 | |
---|
219 | aws->button_length(7); |
---|
220 | |
---|
221 | aws->at("close"); |
---|
222 | aws->callback(AW_POPDOWN); |
---|
223 | aws->create_button("CLOSE", "CLOSE", "C"); |
---|
224 | |
---|
225 | aws->at("help"); |
---|
226 | aws->callback(makeHelpCallback("mg_trees.hlp")); |
---|
227 | aws->create_button("HELP", "HELP", "H"); |
---|
228 | |
---|
229 | aws->at("trees1"); |
---|
230 | awt_create_TREE_selection_list(GLOBAL_gb_src, aws, AWAR_TREE_NAME_SRC); |
---|
231 | |
---|
232 | aws->at("trees2"); |
---|
233 | awt_create_TREE_selection_list(GLOBAL_gb_dst, aws, AWAR_TREE_NAME_DST); |
---|
234 | |
---|
235 | static TreeAdmin::Spec src_spec(GLOBAL_gb_src, AWAR_TREE_NAME_SRC); |
---|
236 | static TreeAdmin::Spec dst_spec(GLOBAL_gb_dst, AWAR_TREE_NAME_DST); |
---|
237 | |
---|
238 | aws->button_length(15); |
---|
239 | |
---|
240 | aws->at("delete1"); |
---|
241 | aws->callback(makeWindowCallback(TreeAdmin::delete_tree_cb, &src_spec)); |
---|
242 | aws->create_button("DELETE TREE_DB1", "Delete Tree"); |
---|
243 | |
---|
244 | aws->at("delete2"); |
---|
245 | aws->callback(makeWindowCallback(TreeAdmin::delete_tree_cb, &dst_spec)); |
---|
246 | aws->create_button("DELETE_TREE_DB2", "Delete Tree"); |
---|
247 | |
---|
248 | aws->at("rename1"); |
---|
249 | aws->callback(makeCreateWindowCallback(TreeAdmin::create_rename_window, &src_spec)); |
---|
250 | aws->create_button("RENAME_TREE_DB1", "Rename Tree"); |
---|
251 | |
---|
252 | aws->at("rename2"); |
---|
253 | aws->callback(makeCreateWindowCallback(TreeAdmin::create_rename_window, &dst_spec)); |
---|
254 | aws->create_button("RENAME_TREE_DB2", "Rename Tree"); |
---|
255 | |
---|
256 | aws->at("transfer"); |
---|
257 | aws->callback(MG_transfer_tree); |
---|
258 | aws->create_autosize_button("TRANSFER_TREE", "Transfer"); |
---|
259 | |
---|
260 | aws->at("xfer_what"); |
---|
261 | aws->create_option_menu(AWAR_TREE_XFER_WHAT); |
---|
262 | aws->insert_default_option("selected tree", "s", XFER_SELECTED); |
---|
263 | aws->insert_option ("all trees", "a", XFER_ALL); |
---|
264 | aws->insert_option ("missing trees", "m", XFER_MISSING); |
---|
265 | aws->insert_option ("existing trees", "e", XFER_EXISTING); |
---|
266 | aws->update_option_menu(); |
---|
267 | |
---|
268 | aws->at("overwrite"); |
---|
269 | aws->label("Overwrite trees?"); |
---|
270 | aws->create_toggle(AWAR_TREE_OVERWRITE); |
---|
271 | |
---|
272 | aws->button_length(0); |
---|
273 | aws->shadow_width(1); |
---|
274 | aws->at("icon"); |
---|
275 | aws->callback(makeHelpCallback("mg_trees.hlp")); |
---|
276 | aws->create_button("HELP_MERGE", "#merge/icon.xpm"); |
---|
277 | |
---|
278 | return aws; |
---|
279 | } |
---|
280 | |
---|