root/applications/robust/robustlib.cpp @ 1366

Revision 1366, 9.1 kB (checked in by sindj, 13 years ago)

Jupiijahou, snad se podarilo dobojovat s normalizacnim faktorem. Pridany vsechny potrebne logaritmy do log_nc. JS

Line 
1#include "robustlib.h"
2
3
4double polyhedron::triangulate(bool should_integrate)
5        {
6                for(set<simplex*>::iterator t_ref = triangulation.begin();t_ref!=triangulation.end();t_ref++)
7                {
8                        delete (*t_ref);
9                }               
10                triangulation.clear();
11
12                if(should_integrate)
13                {
14                        ((toprow *)this)->probability = 0.0;
15                }               
16               
17                if(vertices.size()==1)
18                {
19                        simplex* vert_simplex = new simplex((vertex*)this);                     
20
21                        triangulation.insert(vert_simplex);
22                }
23
24                for(list<polyhedron*>::iterator child_ref = children.begin();child_ref!=children.end();child_ref++)
25                {                       
26                        for(set<simplex*>::iterator s_ref = (*child_ref)->triangulation.begin();s_ref!=(*child_ref)->triangulation.end();s_ref++)
27                        {
28                                simplex* new_simplex = new simplex((*s_ref)->vertices);                                                         
29
30                                pair<set<vertex*>::iterator,bool> ret_val = new_simplex->vertices.insert(*vertices.begin());
31
32                                if(ret_val.second == true)
33                                {
34                                        double cur_prob = 0;
35
36                                        /*
37                                        if(should_integrate&&new_simplex->vertices.size()!=3)
38                                        {
39                                                cout << "Error: Wrong vertex count for integration!";
40                                        }
41                                        */
42
43                                        if(should_integrate)
44                                        {
45                                                cur_prob = ((toprow *)this)->integrate_simplex(new_simplex, 'S');
46
47                                                ((toprow *)this)->probability += cur_prob;
48                                        }
49
50                                        triangulation.insert(new_simplex);
51                                }
52                                else
53                                {
54                                        delete new_simplex;
55                                }
56                        }       
57                }       
58
59                if(should_integrate)
60                {
61                        return ((toprow *)this)->probability;
62                }
63                else
64                {
65                        return 0.0;
66                }
67               
68                /*
69                for(list<polyhedron*>::iterator child_ref = children.begin();child_ref!=children.end();child_ref++)
70                {                       
71                        for(list<set<vertex*>>::iterator t_ref = (*child_ref)->triangulation.begin();t_ref!=(*child_ref)->triangulation.end();t_ref++)
72                        {
73                                set<vertex*> new_simplex;
74                                new_simplex.insert((*t_ref).begin(),(*t_ref).end());                   
75
76                                for(set<vertex*>::iterator suitable_vert_ref = vertices.begin();*suitable_vert_ref<*(*t_ref).begin();suitable_vert_ref++)
77                                {
78                                        set<vertex*> suitable_simplex;
79                                        suitable_simplex.insert(new_simplex.begin(),new_simplex.end());
80                                       
81                                        suitable_simplex.insert(*suitable_vert_ref);
82
83                                        triangulation.push_back(suitable_simplex);
84
85                                        if(should_integrate)
86                                        {
87                                                ((toprow *)this)->probability += ((toprow *)this)->integrate_simplex(suitable_simplex, 'S');
88                                        }
89                                }                               
90                        }       
91                }*/             
92        }
93
94
95        double toprow::integrate_simplex(simplex* simplex, char c)
96        {
97                // cout << ((toprow*)this)->condition << endl;         
98                int sigma_order = ((toprow*)this)->condition_order-simplex->vertices.size()-1;
99
100                // cout << "C:" << condition_order << "  N:" << my_emlig->number_of_parameters << "  C+N:" << condition_order-my_emlig->number_of_parameters << endl;
101                // pause(0.1);
102
103                if(sigma_order >= 0)
104                {
105
106                        //cout << endl;
107                        //cout << ((toprow*)this)->condition << endl;
108                        //cout << "C:" << condition_order+2 << "  N:" << my_emlig->number_of_parameters << "  C+N:" << condition_order-my_emlig->number_of_parameters+2 << endl;
109                       
110
111                        emlig* current_emlig;
112                        simplex->clear_gammas();
113
114                        if(this->my_emlig!=NULL)
115                        {
116                                current_emlig = this->my_emlig;
117                        }
118                        else
119                        {
120                                throw exception("The statistic of the polyhedron you are trying to integrate over doesn't belong to any emlig!");
121                        }                                               
122
123                        toprow* as_toprow = (toprow*)this;
124
125                        vec cur_condition = as_toprow->condition_sum.get(1,as_toprow->condition_sum.size()-1);
126
127                        // cout << as_toprow->condition << endl;                       
128
129                        int dimension = simplex->vertices.size()-1;
130
131                        mat jacobian(dimension,dimension);                     
132
133                        double a_0;
134                        map<double,int> as;
135                        vertex* base_vertex;
136                        set<vertex*>::iterator base_ref = simplex->vertices.begin();
137                        bool order_correct;
138                                               
139
140                        do
141                        {
142                                order_correct = true;
143                                as.clear();
144
145                                base_vertex = (*base_ref);
146                       
147                                /*
148                                cout << "Base coords: " << base_vertex->get_coordinates() << endl;
149                                cout << "Condition: " << as_toprow->condition_sum << endl;
150                                pause(0.1);
151                                */
152
153                                a_0 = -base_vertex->get_coordinates()*cur_condition+as_toprow->condition_sum[0];               
154                               
155                                int row_count = 0;
156                                for(set<vertex*>::iterator vert_ref = simplex->vertices.begin(); vert_ref!=simplex->vertices.end();vert_ref++)
157                                {
158                                        if(vert_ref != base_ref)
159                                        {
160                                                vec relative_coords = (*vert_ref)->get_coordinates()-base_vertex->get_coordinates();                                           
161
162                                                jacobian.set_row(row_count,relative_coords);
163
164                                                double new_a = -relative_coords*cur_condition;
165
166                                                if(new_a + a_0 == 0 || new_a == 0)
167                                                {
168                                                        base_ref++;
169
170                                                        if(base_ref == simplex->vertices.end()&&simplex->vertices.size()!=2)
171                                                        {
172                                                                throw new exception("Equal local conditions are paired. If this ever occurs, the software has to be modified to include multiplied a_0!!");
173                                                        }
174                                                        /*
175                                                        else if(implex->vertices.size()==2)
176                                                        {
177                                                                return relative_coords[0]*exp
178                                                        }
179                                                        */
180                                                       
181                                                        order_correct = false;                                         
182
183                                                        break;
184                                                }                                               
185                                               
186                                               
187
188                                                if(a_0<current_emlig->min_ll)
189                                                {
190                                                        current_emlig->minimal_vertex = base_vertex;
191                                                        current_emlig->min_ll = a_0;
192                                                }
193
194                                                double a_m = (*vert_ref)->get_coordinates()*cur_condition-as_toprow->condition_sum[0];
195                                                if(a_m<current_emlig->min_ll)
196                                                {
197                                                        current_emlig->minimal_vertex = (*vert_ref);
198                                                        current_emlig->min_ll = a_m;                                           
199                                                }
200
201                                                //cout << "a0:" << a_0 << " a0 coords:" << base_vertex->get_coordinates() << " am:" << a_m << " am coords:" << (*vert_ref)->get_coordinates() << endl;
202
203                                                // cout << "Absolute coords:(V"  << row_count << ")" << (*vert_ref)->get_coordinates() << endl;
204                                                //cout << "Relative coords:(V"  << row_count << ")" << relative_coords << endl;
205
206                                                pair<map<double,int>::iterator,bool> returned = as.insert(pair<double,int>(new_a,1));
207                                                if(returned.second == false)
208                                                {
209                                                        (*returned.first).second++;
210                                                }
211                                                /*
212                                                else
213                                                {
214                                                        cout << "a[" << row_count << "] = " << new_a << endl;
215                                                }
216                                                */
217                                               
218                                                //as.ins(as.size(),new_a);                                                     
219                                               
220                                                row_count++;
221                                        }
222                                }
223                        }
224                        while(!order_correct);
225                       
226                        /*
227                        cout << "a_0: " << a_0 << "    ";
228                        int as_count = 1;
229                        for(map<double,int>::iterator as_ref = as.begin();as_ref!=as.end();as_ref++)
230                        {
231                                cout << "a_" << as_count << ": " << (*as_ref).first << "    ";
232                                as_count++;
233                        }
234                        */
235
236                        double int_value = 0;                   
237
238                        // cout << jacobian << endl;
239
240                        double det_jacobian    = abs(det(jacobian));
241                        double correction_term = det_jacobian;                 
242                        for(map<double,int>::iterator as_ref = as.begin();as_ref!=as.end();as_ref++)
243                        {
244                                double multiplier = det_jacobian;                               
245                               
246                                int as1_order = (*as_ref).second;
247                               
248                                correction_term /= pow(-(*as_ref).first,as1_order);                                     
249                               
250                                current_emlig->set_correction_factors(as1_order);
251
252                                vector<double> factors;
253                                int number_of_factors = 0;                                                             
254                                for(map<double,int>::iterator as2_ref = as.begin();as2_ref!=as.end();as2_ref++)
255                                {
256                                       
257                                        if(as2_ref!=as_ref)
258                                        {                                                                               
259                                                for(int k = 0;k<(*as2_ref).second;k++)
260                                                {
261                                                        factors.push_back((*as_ref).first-(*as2_ref).first);
262                                                }
263
264                                                multiplier        /= pow((*as_ref).first-(*as2_ref).first,(*as2_ref).second);
265                                                number_of_factors += (*as2_ref).second;
266                                        }
267                                        else
268                                        {
269                                                factors.push_back((*as_ref).first);
270
271                                                multiplier        /= (*as_ref).first;
272                                                number_of_factors += 1;
273                                        }                                       
274                                }       
275
276                                double cur_as_ref = (*as_ref).first;                           
277
278                                double gamma_multiplier = -cur_as_ref-a_0;
279
280                                double bracket = fact(as1_order-1)/pow(gamma_multiplier,sigma_order);
281                                                               
282                                simplex->insert_gamma(0,bracket*multiplier,gamma_multiplier);                                                           
283
284                                // bracket *= fact(sigma_order);                               
285
286                                for(int k = 0;k < as1_order-1;k++)
287                                {
288                                        double local_bracket = 0;
289                                        double bracket_factor = 1/fact(as1_order-1-k)/pow(gamma_multiplier,sigma_order-k-1);//pow((double)-1,k+1)
290                                       
291                                        ivec control_vec = ivec();
292                                        control_vec.ins(0,my_emlig->number_of_parameters-as1_order+1);                                 
293                                       
294                                        for(multiset<my_ivec>::iterator combi_ref = this->my_emlig->correction_factors[k].begin();combi_ref!=this->my_emlig->correction_factors[k].upper_bound(my_ivec(control_vec));combi_ref++)
295                                        {
296                                                double bracket_combination = 1;
297                                                for(int j = 0;j<(*combi_ref).size();j++)
298                                                {
299                                                        //cout << "Factor vector:" << (*combi_ref) << endl;
300                                                       
301                                                        bracket_combination /= factors[(*combi_ref)[j]-1];
302                                                }                                               
303                                                                                               
304                                                local_bracket += bracket_factor*bracket_combination;                                                                   
305                                        }
306
307                                        simplex->insert_gamma(k+1,local_bracket*multiplier,gamma_multiplier);
308
309                                        int division_factor = 1;
310                                        for(int s = 0;s<k;s++)
311                                        {
312                                                division_factor *= k-s;
313                                        }
314                                       
315                                        bracket += local_bracket/division_factor; //local_bracket*fact(sigma_order-k);
316                                }                               
317
318                                int_value += multiplier*bracket;
319
320                               
321                                                                                                                                               
322                        }
323
324                        double correction_term_base = correction_term/pow(-a_0,sigma_order);
325
326                        simplex->insert_gamma(0,correction_term_base,-a_0);                     
327
328                        correction_term = correction_term_base;//fact(sigma_order)*correction_term_base;
329
330                        //cout << c << int_value << endl;
331
332                        int_value += correction_term;                   
333
334                        //cout << "Probability:" << int_value << endl;
335                        //pause(0.100);
336
337                        simplex->probability = int_value;
338                       
339                        return int_value;       
340                       
341                }
342                else
343                {
344                        cout << "Improper probability density." << endl;
345                        return 0.0;
346                }
347        }
Note: See TracBrowser for help on using the browser.