1 | // =============================================================== // |
---|
2 | // // |
---|
3 | // File : NT_sort.cxx // |
---|
4 | // Purpose : // |
---|
5 | // // |
---|
6 | // Institute of Microbiology (Technical University Munich) // |
---|
7 | // http://www.arb-home.de/ // |
---|
8 | // // |
---|
9 | // =============================================================== // |
---|
10 | |
---|
11 | #include "NT_local.h" |
---|
12 | |
---|
13 | #include <item_sel_list.h> |
---|
14 | #include <aw_awar.hxx> |
---|
15 | #include <arb_progress.h> |
---|
16 | #include <aw_msg.hxx> |
---|
17 | #include <aw_root.hxx> |
---|
18 | #include <arbdbt.h> |
---|
19 | #include <arb_sort.h> |
---|
20 | |
---|
21 | #define NT_RESORT_FILTER (1<<GB_STRING)|(1<<GB_INT)|(1<<GB_FLOAT) |
---|
22 | |
---|
23 | #define CUSTOM_CRITERIA 3 |
---|
24 | |
---|
25 | struct customCriterion { |
---|
26 | char *key; |
---|
27 | bool reverse; |
---|
28 | |
---|
29 | customCriterion() : key(NULL), reverse(false) {} |
---|
30 | customCriterion(const char *key_, bool reverse_) : key(strdup(key_)), reverse(reverse_) {} |
---|
31 | customCriterion(const customCriterion& other) : key(strdup(other.key)), reverse(other.reverse) {} |
---|
32 | DECLARE_ASSIGNMENT_OPERATOR(customCriterion); |
---|
33 | ~customCriterion() { free(key); } |
---|
34 | }; |
---|
35 | |
---|
36 | static int cmpByKey(GBDATA *gbd1, GBDATA *gbd2, const customCriterion& by) { |
---|
37 | GBDATA *gb_field1 = GB_entry(gbd1, by.key); |
---|
38 | GBDATA *gb_field2 = GB_entry(gbd2, by.key); |
---|
39 | |
---|
40 | int cmp; |
---|
41 | if (gb_field1) { |
---|
42 | if (gb_field2) { |
---|
43 | switch (GB_read_type(gb_field1)) { |
---|
44 | case GB_STRING: { |
---|
45 | const char *s1 = GB_read_char_pntr(gb_field1); |
---|
46 | const char *s2 = GB_read_char_pntr(gb_field2); |
---|
47 | |
---|
48 | cmp = strcmp(s1, s2); |
---|
49 | break; |
---|
50 | } |
---|
51 | case GB_FLOAT: { |
---|
52 | double d1 = GB_read_float(gb_field1); |
---|
53 | double d2 = GB_read_float(gb_field2); |
---|
54 | |
---|
55 | cmp = d1<d2 ? -1 : (d1>d2 ? 1 : 0); |
---|
56 | break; |
---|
57 | } |
---|
58 | case GB_INT: { |
---|
59 | int i1 = GB_read_int(gb_field1); |
---|
60 | int i2 = GB_read_int(gb_field2); |
---|
61 | |
---|
62 | cmp = i1-i2; |
---|
63 | break; |
---|
64 | } |
---|
65 | default: |
---|
66 | cmp = 0; // other field type -> no idea how to compare |
---|
67 | break; |
---|
68 | } |
---|
69 | |
---|
70 | if (by.reverse) cmp = -cmp; |
---|
71 | } |
---|
72 | else cmp = -1; // existing < missing! |
---|
73 | } |
---|
74 | else cmp = gb_field2 ? 1 : 0; |
---|
75 | |
---|
76 | return cmp; |
---|
77 | } |
---|
78 | |
---|
79 | static int resort_data_by_customOrder(const void *v1, const void *v2, void *cd_sortBy) { |
---|
80 | GBDATA *gbd1 = (GBDATA*)v1; |
---|
81 | GBDATA *gbd2 = (GBDATA*)v2; |
---|
82 | |
---|
83 | const customCriterion *sortBy = (const customCriterion *)cd_sortBy; |
---|
84 | |
---|
85 | int cmp = 0; |
---|
86 | for (int c = 0; !cmp && c<CUSTOM_CRITERIA; ++c) { |
---|
87 | cmp = cmpByKey(gbd1, gbd2, sortBy[c]); |
---|
88 | } |
---|
89 | return cmp; |
---|
90 | } |
---|
91 | |
---|
92 | |
---|
93 | static GBDATA **gb_resort_data_list; |
---|
94 | static long gb_resort_data_count; |
---|
95 | |
---|
96 | static void NT_resort_data_base_by_tree(GBT_TREE *tree, GBDATA *gb_species_data) { |
---|
97 | if (tree) { |
---|
98 | if (tree->is_leaf) { |
---|
99 | if (tree->gb_node) { |
---|
100 | gb_resort_data_list[gb_resort_data_count++] = tree->gb_node; |
---|
101 | } |
---|
102 | } |
---|
103 | else { |
---|
104 | NT_resort_data_base_by_tree(tree->leftson, gb_species_data); |
---|
105 | NT_resort_data_base_by_tree(tree->rightson, gb_species_data); |
---|
106 | } |
---|
107 | } |
---|
108 | } |
---|
109 | |
---|
110 | |
---|
111 | static GB_ERROR NT_resort_data_base(GBT_TREE *tree, const customCriterion *sortBy) { |
---|
112 | nt_assert(contradicted(tree, sortBy)); |
---|
113 | |
---|
114 | GB_ERROR error = GB_begin_transaction(GLOBAL.gb_main); |
---|
115 | if (!error) { |
---|
116 | GBDATA *gb_sd = GBT_get_species_data(GLOBAL.gb_main); |
---|
117 | if (!gb_sd) error = GB_await_error(); |
---|
118 | else { |
---|
119 | if (tree) { |
---|
120 | gb_resort_data_count = 0; |
---|
121 | gb_resort_data_list = (GBDATA **)calloc(sizeof(GBDATA *), GB_nsons(gb_sd) + 256); |
---|
122 | NT_resort_data_base_by_tree(tree, gb_sd); |
---|
123 | } |
---|
124 | else { |
---|
125 | gb_resort_data_list = GBT_gen_species_array(GLOBAL.gb_main, &gb_resort_data_count); |
---|
126 | GB_sort((void **)gb_resort_data_list, 0, gb_resort_data_count, resort_data_by_customOrder, (void*)sortBy); |
---|
127 | |
---|
128 | } |
---|
129 | error = GB_resort_data_base(GLOBAL.gb_main, gb_resort_data_list, gb_resort_data_count); |
---|
130 | free(gb_resort_data_list); |
---|
131 | } |
---|
132 | } |
---|
133 | return GB_end_transaction(GLOBAL.gb_main, error); |
---|
134 | } |
---|
135 | |
---|
136 | void NT_resort_data_by_phylogeny(AW_window *, AW_CL cl_ntw, AW_CL) { |
---|
137 | arb_progress progress("Sorting data"); |
---|
138 | GB_ERROR error = 0; |
---|
139 | GBT_TREE *tree = NT_get_tree_root_of_canvas((AWT_canvas*)cl_ntw); |
---|
140 | |
---|
141 | if (!tree) error = "Please select/build a tree first"; |
---|
142 | if (!error) error = NT_resort_data_base(tree, NULL); |
---|
143 | if (error) aw_message(error); |
---|
144 | } |
---|
145 | |
---|
146 | #define AWAR_TREE_SORT1 "ad_tree/sort_1" |
---|
147 | #define AWAR_TREE_SORT2 "ad_tree/sort_2" |
---|
148 | #define AWAR_TREE_SORT3 "ad_tree/sort_3" |
---|
149 | |
---|
150 | #define AWAR_TREE_REV1 "ad_tree/rev1" |
---|
151 | #define AWAR_TREE_REV2 "ad_tree/rev2" |
---|
152 | #define AWAR_TREE_REV3 "ad_tree/rev3" |
---|
153 | |
---|
154 | static void NT_resort_data_by_user_criteria(AW_window *aw) { |
---|
155 | arb_progress progress("Sorting data"); |
---|
156 | |
---|
157 | AW_root *aw_root = aw->get_root(); |
---|
158 | |
---|
159 | customCriterion sortBy[CUSTOM_CRITERIA]; |
---|
160 | sortBy[0] = customCriterion(aw_root->awar(AWAR_TREE_SORT1)->read_char_pntr(), aw_root->awar(AWAR_TREE_REV1)->read_int()); |
---|
161 | sortBy[1] = customCriterion(aw_root->awar(AWAR_TREE_SORT2)->read_char_pntr(), aw_root->awar(AWAR_TREE_REV2)->read_int()); |
---|
162 | sortBy[2] = customCriterion(aw_root->awar(AWAR_TREE_SORT3)->read_char_pntr(), aw_root->awar(AWAR_TREE_REV3)->read_int()); |
---|
163 | |
---|
164 | GB_ERROR error = NT_resort_data_base(NULL, sortBy); |
---|
165 | if (error) aw_message(error); |
---|
166 | } |
---|
167 | |
---|
168 | void NT_create_resort_awars(AW_root *awr, AW_default aw_def) { |
---|
169 | awr->awar_string(AWAR_TREE_SORT1, "name", aw_def); |
---|
170 | awr->awar_string(AWAR_TREE_SORT2, "full_name", aw_def); |
---|
171 | awr->awar_string(AWAR_TREE_SORT3, "", aw_def); |
---|
172 | |
---|
173 | awr->awar_int(AWAR_TREE_REV1, 0, aw_def); |
---|
174 | awr->awar_int(AWAR_TREE_REV2, 0, aw_def); |
---|
175 | awr->awar_int(AWAR_TREE_REV3, 0, aw_def); |
---|
176 | } |
---|
177 | |
---|
178 | AW_window *NT_create_resort_window(AW_root *awr) { |
---|
179 | AW_window_simple *aws = new AW_window_simple; |
---|
180 | aws->init(awr, "SORT_DATABASE", "SORT DATABASE"); |
---|
181 | aws->load_xfig("nt_sort.fig"); |
---|
182 | |
---|
183 | aws->callback((AW_CB0)AW_POPDOWN); |
---|
184 | aws->at("close"); |
---|
185 | aws->create_button("CLOSE", "CLOSE", "C"); |
---|
186 | |
---|
187 | aws->callback(makeHelpCallback("sp_sort_fld.hlp")); |
---|
188 | aws->at("help"); |
---|
189 | aws->create_button("HELP", "HELP", "H"); |
---|
190 | |
---|
191 | create_selection_list_on_itemfields(GLOBAL.gb_main, aws, AWAR_TREE_SORT1, true, NT_RESORT_FILTER, "key1", 0, SPECIES_get_selector(), 20, 10, SF_STANDARD, NULL); |
---|
192 | create_selection_list_on_itemfields(GLOBAL.gb_main, aws, AWAR_TREE_SORT2, true, NT_RESORT_FILTER, "key2", 0, SPECIES_get_selector(), 20, 10, SF_STANDARD, NULL); |
---|
193 | create_selection_list_on_itemfields(GLOBAL.gb_main, aws, AWAR_TREE_SORT3, true, NT_RESORT_FILTER, "key3", 0, SPECIES_get_selector(), 20, 10, SF_STANDARD, NULL); |
---|
194 | |
---|
195 | aws->at("rev1"); aws->label("Reverse"); aws->create_toggle(AWAR_TREE_REV1); |
---|
196 | aws->at("rev2"); aws->label("Reverse"); aws->create_toggle(AWAR_TREE_REV2); |
---|
197 | aws->at("rev3"); aws->label("Reverse"); aws->create_toggle(AWAR_TREE_REV3); |
---|
198 | |
---|
199 | aws->at("go"); |
---|
200 | aws->callback((AW_CB0)NT_resort_data_by_user_criteria); |
---|
201 | aws->create_button("GO", "GO", "G"); |
---|
202 | |
---|
203 | return aws; |
---|
204 | } |
---|