78911f945f6b4e06318827b20d2101ce05e2f50f
[platform/upstream/libsolv.git] / src / problems.c
1 /*
2  * Copyright (c) 2007-2009, Novell Inc.
3  *
4  * This program is licensed under the BSD license, read LICENSE.BSD
5  * for further information
6  */
7
8 /*
9  * problems.c
10  *
11  */
12
13 #include <stdio.h>
14 #include <stdlib.h>
15 #include <unistd.h>
16 #include <string.h>
17 #include <assert.h>
18
19 #include "solver.h"
20 #include "solver_private.h"
21 #include "bitmap.h"
22 #include "pool.h"
23 #include "util.h"
24 #include "evr.h"
25 #include "solverdebug.h"
26
27
28 /**********************************************************************************/
29
30 /* a problem is an item on the solver's problem list. It can either be >0, in that
31  * case it is a update/infarch/dup rule, or it can be <0, which makes it refer to a job
32  * consisting of multiple job rules.
33  */
34
35 void
36 solver_disableproblem(Solver *solv, Id v)
37 {
38   Rule *r;
39   int i;
40   Id *jp;
41
42   if (v > 0)
43     {
44       if (v >= solv->infarchrules && v < solv->infarchrules_end)
45         {
46           Pool *pool = solv->pool;
47           Id name = pool->solvables[-solv->rules[v].p].name;
48           while (v > solv->infarchrules && pool->solvables[-solv->rules[v - 1].p].name == name)
49             v--;
50           for (; v < solv->infarchrules_end && pool->solvables[-solv->rules[v].p].name == name; v++)
51             solver_disablerule(solv, solv->rules + v);
52           return;
53         }
54       if (v >= solv->duprules && v < solv->duprules_end)
55         {
56           Pool *pool = solv->pool;
57           Id name = pool->solvables[-solv->rules[v].p].name;
58           while (v > solv->duprules && pool->solvables[-solv->rules[v - 1].p].name == name)
59             v--;
60           for (; v < solv->duprules_end && pool->solvables[-solv->rules[v].p].name == name; v++)
61             solver_disablerule(solv, solv->rules + v);
62           return;
63         }
64       solver_disablerule(solv, solv->rules + v);
65 #if 0
66       /* XXX: doesn't work */
67       if (v >= solv->updaterules && v < solv->updaterules_end)
68         {
69           /* enable feature rule if we disabled the update rule */
70           r = solv->rules + (v - solv->updaterules + solv->featurerules);
71           if (r->p)
72             solver_enablerule(solv, r);
73         }
74 #endif
75       return;
76     }
77   v = -(v + 1);
78   jp = solv->ruletojob.elements;
79   for (i = solv->jobrules, r = solv->rules + i; i < solv->jobrules_end; i++, r++, jp++)
80     if (*jp == v)
81       solver_disablerule(solv, r);
82 }
83
84 /*-------------------------------------------------------------------
85  * enableproblem
86  */
87
88 void
89 solver_enableproblem(Solver *solv, Id v)
90 {
91   Rule *r;
92   int i;
93   Id *jp;
94
95   if (v > 0)
96     {
97       if (v >= solv->infarchrules && v < solv->infarchrules_end)
98         {
99           Pool *pool = solv->pool;
100           Id name = pool->solvables[-solv->rules[v].p].name;
101           while (v > solv->infarchrules && pool->solvables[-solv->rules[v - 1].p].name == name)
102             v--;
103           for (; v < solv->infarchrules_end && pool->solvables[-solv->rules[v].p].name == name; v++)
104             solver_enablerule(solv, solv->rules + v);
105           return;
106         }
107       if (v >= solv->duprules && v < solv->duprules_end)
108         {
109           Pool *pool = solv->pool;
110           Id name = pool->solvables[-solv->rules[v].p].name;
111           while (v > solv->duprules && pool->solvables[-solv->rules[v - 1].p].name == name)
112             v--;
113           for (; v < solv->duprules_end && pool->solvables[-solv->rules[v].p].name == name; v++)
114             solver_enablerule(solv, solv->rules + v);
115           return;
116         }
117       if (v >= solv->featurerules && v < solv->featurerules_end)
118         {
119           /* do not enable feature rule if update rule is enabled */
120           r = solv->rules + (v - solv->featurerules + solv->updaterules);
121           if (r->d >= 0)
122             return;
123         }
124       solver_enablerule(solv, solv->rules + v);
125       if (v >= solv->updaterules && v < solv->updaterules_end)
126         {
127           /* disable feature rule when enabling update rule */
128           r = solv->rules + (v - solv->updaterules + solv->featurerules);
129           if (r->p)
130             solver_disablerule(solv, r);
131         }
132       return;
133     }
134   v = -(v + 1);
135   jp = solv->ruletojob.elements;
136   for (i = solv->jobrules, r = solv->rules + i; i < solv->jobrules_end; i++, r++, jp++)
137     if (*jp == v)
138       solver_enablerule(solv, r);
139 }
140
141
142 /*-------------------------------------------------------------------
143  * enable weak rules
144  * 
145  * Reenable all disabled weak rules (marked in weakrulemap)
146  * 
147  */
148
149 static void
150 enableweakrules(Solver *solv)
151 {
152   int i;
153   Rule *r;
154
155   for (i = 1, r = solv->rules + i; i < solv->learntrules; i++, r++)
156     {
157       if (r->d >= 0) /* already enabled? */
158         continue;
159       if (!MAPTST(&solv->weakrulemap, i))
160         continue;
161       solver_enablerule(solv, r);
162     }
163 }
164
165
166 /*-------------------------------------------------------------------
167  * 
168  * refine_suggestion
169  * 
170  * at this point, all rules that led to conflicts are disabled.
171  * we re-enable all rules of a problem set but rule "sug", then
172  * continue to disable more rules until there as again a solution.
173  */
174
175 /* FIXME: think about conflicting assertions */
176
177 static void
178 refine_suggestion(Solver *solv, Id *problem, Id sug, Queue *refined, int essentialok)
179 {
180   Pool *pool = solv->pool;
181   int i, j;
182   Id v;
183   Queue disabled;
184   int disabledcnt;
185
186   IF_POOLDEBUG (SOLV_DEBUG_SOLUTIONS)
187     {
188       POOL_DEBUG(SOLV_DEBUG_SOLUTIONS, "refine_suggestion start\n");
189       for (i = 0; problem[i]; i++)
190         {
191           if (problem[i] == sug)
192             POOL_DEBUG(SOLV_DEBUG_SOLUTIONS, "=> ");
193           solver_printproblem(solv, problem[i]);
194         }
195     }
196   queue_empty(refined);
197   if (!essentialok && sug < 0 && (solv->job.elements[-sug - 1] & SOLVER_ESSENTIAL) != 0)
198     return;
199   queue_init(&disabled);
200   queue_push(refined, sug);
201
202   /* re-enable all problem rules with the exception of "sug"(gestion) */
203   solver_reset(solv);
204
205   for (i = 0; problem[i]; i++)
206     if (problem[i] != sug)
207       solver_enableproblem(solv, problem[i]);
208
209   if (sug < 0)
210     solver_reenablepolicyrules(solv, -sug);
211   else if (sug >= solv->updaterules && sug < solv->updaterules_end)
212     {
213       /* enable feature rule */
214       Rule *r = solv->rules + solv->featurerules + (sug - solv->updaterules);
215       if (r->p)
216         solver_enablerule(solv, r);
217     }
218
219   enableweakrules(solv);
220
221   for (;;)
222     {
223       int njob, nfeature, nupdate, pass;
224       queue_empty(&solv->problems);
225       solver_reset(solv);
226
227       if (!solv->problems.count)
228         solver_run_sat(solv, 0, 0);
229
230       if (!solv->problems.count)
231         {
232           POOL_DEBUG(SOLV_DEBUG_SOLUTIONS, "no more problems!\n");
233           break;                /* great, no more problems */
234         }
235       disabledcnt = disabled.count;
236       /* start with 1 to skip over proof index */
237       njob = nfeature = nupdate = 0;
238       for (pass = 0; pass < 2; pass++)
239         {
240           for (i = 1; i < solv->problems.count - 1; i++)
241             {
242               /* ignore solutions in refined */
243               v = solv->problems.elements[i];
244               if (v == 0)
245                 break;  /* end of problem reached */
246               if (sug != v)
247                 {
248                   /* check if v is in the given problems list
249                    * we allow disabling all problem rules *after* sug in
250                    * pass 2, to prevent getting the same solution twice */
251                   for (j = 0; problem[j]; j++)
252                     if (problem[j] == v || (pass && problem[j] == sug))
253                       break;
254                   if (problem[j] == v)
255                     continue;
256                 }
257               if (v >= solv->featurerules && v < solv->featurerules_end)
258                 nfeature++;
259               else if (v > 0)
260                 nupdate++;
261               else
262                 {
263                   if (!essentialok && (solv->job.elements[-v -1] & SOLVER_ESSENTIAL) != 0)
264                     continue;   /* not that one! */
265                   njob++;
266                 }
267               queue_push(&disabled, v);
268             }
269           if (disabled.count != disabledcnt)
270             break;
271         }
272       if (disabled.count == disabledcnt)
273         {
274           /* no solution found, this was an invalid suggestion! */
275           POOL_DEBUG(SOLV_DEBUG_SOLUTIONS, "no solution found!\n");
276           refined->count = 0;
277           break;
278         }
279       if (!njob && nupdate && nfeature)
280         {
281           /* got only update rules, filter out feature rules */
282           POOL_DEBUG(SOLV_DEBUG_SOLUTIONS, "throwing away feature rules\n");
283           for (i = j = disabledcnt; i < disabled.count; i++)
284             {
285               v = disabled.elements[i];
286               if (v < solv->featurerules || v >= solv->featurerules_end)
287                 disabled.elements[j++] = v;
288             }
289           disabled.count = j;
290           nfeature = 0;
291         }
292       if (disabled.count == disabledcnt + 1)
293         {
294           /* just one suggestion, add it to refined list */
295           v = disabled.elements[disabledcnt];
296           if (!nfeature && v != sug)
297             queue_push(refined, v);     /* do not record feature rules */
298           solver_disableproblem(solv, v);
299           if (v >= solv->updaterules && v < solv->updaterules_end)
300             {
301               Rule *r = solv->rules + (v - solv->updaterules + solv->featurerules);
302               if (r->p)
303                 solver_enablerule(solv, r);     /* enable corresponding feature rule */
304             }
305           if (v < 0)
306             solver_reenablepolicyrules(solv, -v);
307         }
308       else
309         {
310           /* more than one solution, disable all */
311           /* do not push anything on refine list, as we do not know which solution to choose */
312           /* thus, the user will get another problem if he selects this solution, where he
313            * can choose the right one */
314           IF_POOLDEBUG (SOLV_DEBUG_SOLUTIONS)
315             {
316               POOL_DEBUG(SOLV_DEBUG_SOLUTIONS, "more than one solution found:\n");
317               for (i = disabledcnt; i < disabled.count; i++)
318                 solver_printproblem(solv, disabled.elements[i]);
319             }
320           for (i = disabledcnt; i < disabled.count; i++)
321             {
322               v = disabled.elements[i];
323               solver_disableproblem(solv, v);
324               if (v >= solv->updaterules && v < solv->updaterules_end)
325                 {
326                   Rule *r = solv->rules + (v - solv->updaterules + solv->featurerules);
327                   if (r->p)
328                     solver_enablerule(solv, r);
329                 }
330             }
331         }
332     }
333   /* all done, get us back into the same state as before */
334   /* enable refined rules again */
335   for (i = 0; i < disabled.count; i++)
336     solver_enableproblem(solv, disabled.elements[i]);
337   queue_free(&disabled);
338   /* reset policy rules */
339   for (i = 0; problem[i]; i++)
340     solver_enableproblem(solv, problem[i]);
341   solver_disablepolicyrules(solv);
342   /* disable problem rules again */
343   for (i = 0; problem[i]; i++)
344     solver_disableproblem(solv, problem[i]);
345   POOL_DEBUG(SOLV_DEBUG_SOLUTIONS, "refine_suggestion end\n");
346 }
347
348
349 /*-------------------------------------------------------------------
350  * sorting helper for problems
351  *
352  * bring update rules before job rules
353  * make essential job rules last
354  */
355
356 static int
357 problems_sortcmp(const void *ap, const void *bp, void *dp)
358 {
359   Queue *job = dp;
360   Id a = *(Id *)ap, b = *(Id *)bp;
361   if (a < 0 && b > 0)
362     return 1;
363   if (a > 0 && b < 0)
364     return -1;
365   if (a < 0 && b < 0)
366     {
367       int af = job->elements[-a - 1] & SOLVER_ESSENTIAL;
368       int bf = job->elements[-b - 1] & SOLVER_ESSENTIAL;
369       int x = af - bf;
370       if (x)
371         return x;
372     }
373   return a - b;
374 }
375
376 /*
377  * convert a solution rule into a job modifier
378  */
379 static void
380 convertsolution(Solver *solv, Id why, Queue *solutionq)
381 {
382   Pool *pool = solv->pool;
383   if (why < 0)
384     {
385       queue_push(solutionq, 0);
386       queue_push(solutionq, -why);
387       return;
388     }
389   if (why >= solv->infarchrules && why < solv->infarchrules_end)
390     {
391       Id p, name;
392       /* infarch rule, find replacement */
393       assert(solv->rules[why].p < 0);
394       name = pool->solvables[-solv->rules[why].p].name;
395       while (why > solv->infarchrules && pool->solvables[-solv->rules[why - 1].p].name == name)
396         why--;
397       p = 0;
398       for (; why < solv->infarchrules_end && pool->solvables[-solv->rules[why].p].name == name; why++)
399         if (solv->decisionmap[-solv->rules[why].p] > 0)
400           {
401             p = -solv->rules[why].p;
402             break;
403           }
404       if (!p)
405         return;         /* false alarm */
406       queue_push(solutionq, SOLVER_SOLUTION_INFARCH);
407       queue_push(solutionq, p);
408       return;
409     }
410   if (why >= solv->duprules && why < solv->duprules_end)
411     {
412       Id p, name;
413       /* dist upgrade rule, find replacement */
414       assert(solv->rules[why].p < 0);
415       name = pool->solvables[-solv->rules[why].p].name;
416       while (why > solv->duprules && pool->solvables[-solv->rules[why - 1].p].name == name)
417         why--;
418       p = 0;
419       for (; why < solv->duprules_end && pool->solvables[-solv->rules[why].p].name == name; why++)
420         if (solv->decisionmap[-solv->rules[why].p] > 0)
421           {
422             p = -solv->rules[why].p;
423             break;
424           }
425       if (!p)
426         return;         /* false alarm */
427       queue_push(solutionq, SOLVER_SOLUTION_DISTUPGRADE);
428       queue_push(solutionq, p);
429       return;
430     }
431   if (why >= solv->updaterules && why < solv->updaterules_end)
432     {
433       /* update rule, find replacement package */
434       Id p, pp, rp = 0;
435       Rule *rr;
436
437       /* check if this is a false positive, i.e. the update rule is fulfilled */
438       rr = solv->rules + why;
439       FOR_RULELITERALS(p, pp, rr)
440         if (p > 0 && solv->decisionmap[p] > 0)
441           return;       /* false alarm */
442
443       p = solv->installed->start + (why - solv->updaterules);
444       if (solv->dupmap_all && solv->rules[why].p != p && solv->decisionmap[p] > 0)
445         {
446           /* distupgrade case, allow to keep old package */
447           queue_push(solutionq, SOLVER_SOLUTION_DISTUPGRADE);
448           queue_push(solutionq, p);
449           return;
450         }
451       if (solv->decisionmap[p] > 0)
452         return;         /* false alarm, turned out we can keep the package */
453       rr = solv->rules + solv->featurerules + (why - solv->updaterules);
454       if (!rr->p)
455         rr = solv->rules + why;
456       if (rr->w2)
457         {
458           int mvrp = 0;         /* multi-version replacement */
459           FOR_RULELITERALS(rp, pp, rr)
460             {
461               if (rp > 0 && solv->decisionmap[rp] > 0 && pool->solvables[rp].repo != solv->installed)
462                 {
463                   mvrp = rp;
464                   if (!(solv->noobsoletes.size && MAPTST(&solv->noobsoletes, rp)))
465                     break;
466                 }
467             }
468           if (!rp && mvrp)
469             {
470               /* found only multi-version replacements */
471               /* have to split solution into two parts */
472               queue_push(solutionq, p);
473               queue_push(solutionq, mvrp);
474             }
475         }
476       queue_push(solutionq, p);
477       queue_push(solutionq, rp);
478       return;
479     }
480   if (why >= solv->bestrules && why < solv->bestrules_end)
481     {
482       int mvrp;
483       Id p, pp, rp = 0;
484       Rule *rr;
485       /* check false positive */
486       rr = solv->rules + why;
487       FOR_RULELITERALS(p, pp, rr)
488         if (p > 0 && solv->decisionmap[p] > 0)
489           return;       /* false alarm */
490       /* check update/feature rule */
491       p = solv->bestrules_pkg[why - solv->bestrules];
492       if (p < 0)
493         {
494           /* install job */
495           queue_push(solutionq, 0);
496           queue_push(solutionq, solv->ruletojob.elements[-p - solv->jobrules] + 1);
497           return;
498         }
499       if (solv->decisionmap[p] > 0)
500         {
501           /* disable best rule by keeping the old package */
502           queue_push(solutionq, SOLVER_SOLUTION_BEST);
503           queue_push(solutionq, p);
504           return;
505         }
506       rr = solv->rules + solv->featurerules + (p - solv->installed->start);
507       if (!rr->p)
508         rr = solv->rules + solv->updaterules + (p - solv->installed->start);
509       mvrp = 0;         /* multi-version replacement */
510       FOR_RULELITERALS(rp, pp, rr)
511         if (rp > 0 && solv->decisionmap[rp] > 0 && pool->solvables[rp].repo != solv->installed)
512           {
513             mvrp = rp;
514             if (!(solv->noobsoletes.size && MAPTST(&solv->noobsoletes, rp)))
515               break;
516           }
517       if (!rp && mvrp)
518         {
519           queue_push(solutionq, SOLVER_SOLUTION_BEST);  /* split, see above */
520           queue_push(solutionq, mvrp);
521           queue_push(solutionq, p);
522           queue_push(solutionq, 0);
523           return;
524         }
525       if (rp)
526         {
527           queue_push(solutionq, SOLVER_SOLUTION_BEST);
528           queue_push(solutionq, rp);
529         }
530       return;
531     }
532 }
533
534 /*
535  * convert problem data into a form usable for refining.
536  * Returns the number of problems.
537  */
538 int
539 solver_prepare_solutions(Solver *solv)
540 {
541   int i, j = 1, idx;
542
543   if (!solv->problems.count)
544     return 0;
545   queue_empty(&solv->solutions);
546   queue_push(&solv->solutions, 0);      /* dummy so idx is always nonzero */
547   idx = solv->solutions.count;
548   queue_push(&solv->solutions, -1);     /* unrefined */
549   /* proofidx stays in position, thus we start with 1 */
550   for (i = 1; i < solv->problems.count; i++) 
551     {   
552       Id p = solv->problems.elements[i];
553       queue_push(&solv->solutions, p); 
554       if (p) 
555         continue;
556       /* end of problem reached */
557       solv->problems.elements[j++] = idx; 
558       if (i + 1 >= solv->problems.count)
559         break;
560       /* start another problem */
561       solv->problems.elements[j++] = solv->problems.elements[++i];  /* copy proofidx */
562       idx = solv->solutions.count;
563       queue_push(&solv->solutions, -1); /* unrefined */
564     }
565   solv->problems.count = j;  
566   return j / 2;
567 }
568
569 /*
570  * refine the simple solution rule list provided by
571  * the solver into multiple lists of job modifiers.
572  */
573 static void
574 create_solutions(Solver *solv, int probnr, int solidx)
575 {
576   Pool *pool = solv->pool;
577   Queue redoq;
578   Queue problem, solution, problems_save;
579   int i, j, nsol;
580   int essentialok;
581   unsigned int now;
582   int oldmistakes = solv->cleandeps_mistakes ? solv->cleandeps_mistakes->count : 0;
583   Id extraflags = -1;
584
585   now = solv_timems(0);
586   queue_init(&redoq);
587   /* save decisionq, decisionq_why, decisionmap */
588   for (i = 0; i < solv->decisionq.count; i++)
589     {
590       Id p = solv->decisionq.elements[i];
591       queue_push(&redoq, p);
592       queue_push(&redoq, solv->decisionq_why.elements[i]);
593       queue_push(&redoq, solv->decisionmap[p > 0 ? p : -p]);
594     }
595   /* save problems queue */
596   problems_save = solv->problems;
597   memset(&solv->problems, 0, sizeof(solv->problems));
598
599   /* extract problem from queue */
600   queue_init(&problem);
601   for (i = solidx + 1; i < solv->solutions.count; i++)
602     {
603       Id v = solv->solutions.elements[i];
604       if (!v)
605         break;
606       queue_push(&problem, v);
607       if (v < 0)
608         extraflags &= solv->job.elements[-v - 1];
609     }
610   if (extraflags == -1)
611     extraflags = 0;
612   if (problem.count > 1)
613     solv_sort(problem.elements, problem.count, sizeof(Id), problems_sortcmp, &solv->job);
614   queue_push(&problem, 0);      /* mark end for refine_suggestion */
615   problem.count--;
616 #if 0
617   for (i = 0; i < problem.count; i++)
618     printf("PP %d %d\n", i, problem.elements[i]);
619 #endif
620
621   /* refine each solution element */
622   nsol = 0;
623   essentialok = 0;
624   queue_init(&solution);
625   for (i = 0; i < problem.count; i++)
626     {
627       int solstart = solv->solutions.count;
628       refine_suggestion(solv, problem.elements, problem.elements[i], &solution, essentialok);
629       queue_push(&solv->solutions, 0);  /* reserve room for number of elements */
630       for (j = 0; j < solution.count; j++)
631         convertsolution(solv, solution.elements[j], &solv->solutions);
632       if (solv->solutions.count == solstart + 1)
633         {
634           solv->solutions.count--;      /* this one did not work out */
635           if (nsol || i + 1 < problem.count)
636             continue;                   /* got one or still hope */
637           if (!essentialok)
638             {
639               /* nothing found, start over */
640               POOL_DEBUG(SOLV_DEBUG_SOLUTIONS, "nothing found, re-run with essentialok = 1\n");
641               essentialok = 1;
642               i = -1;
643               continue;
644             }
645           /* this is bad, we found no solution */
646           /* for now just offer a rule */
647           POOL_DEBUG(SOLV_DEBUG_SOLUTIONS, "nothing found, already did essentialok, fake it\n");
648           queue_push(&solv->solutions, 0);
649           for (j = 0; j < problem.count; j++)
650             {
651               convertsolution(solv, problem.elements[j], &solv->solutions);
652               if (solv->solutions.count > solstart + 1)
653                 break;
654             }
655           if (solv->solutions.count == solstart + 1)
656             {
657               solv->solutions.count--;
658               continue;         /* sorry */
659             }
660         }
661       /* patch in number of solution elements */
662       solv->solutions.elements[solstart] = (solv->solutions.count - (solstart + 1)) / 2;
663       queue_push(&solv->solutions, 0);  /* add end marker */
664       queue_push(&solv->solutions, 0);  /* add end marker */
665       queue_push(&solv->solutions, problem.elements[i]);        /* just for bookkeeping */
666       queue_push(&solv->solutions, extraflags & SOLVER_CLEANDEPS);      /* our extraflags */
667       solv->solutions.elements[solidx + 1 + nsol++] = solstart;
668     }
669   solv->solutions.elements[solidx + 1 + nsol] = 0;      /* end marker */
670   solv->solutions.elements[solidx] = nsol;
671   queue_free(&problem);
672   queue_free(&solution);
673
674   /* restore decisions */
675   memset(solv->decisionmap, 0, pool->nsolvables * sizeof(Id));
676   queue_empty(&solv->decisionq);
677   queue_empty(&solv->decisionq_why);
678   for (i = 0; i < redoq.count; i += 3)
679     {
680       Id p = redoq.elements[i];
681       queue_push(&solv->decisionq, p);
682       queue_push(&solv->decisionq_why, redoq.elements[i + 1]);
683       solv->decisionmap[p > 0 ? p : -p] = redoq.elements[i + 2];
684     }
685   queue_free(&redoq);
686   /* restore problems */
687   queue_free(&solv->problems);
688   solv->problems = problems_save;
689
690   if (solv->cleandeps_mistakes)
691     {
692       if (oldmistakes)
693         queue_truncate(solv->cleandeps_mistakes, oldmistakes);
694       else
695         {
696           queue_free(solv->cleandeps_mistakes);
697           solv->cleandeps_mistakes = solv_free(solv->cleandeps_mistakes);
698         }
699     }
700     
701   POOL_DEBUG(SOLV_DEBUG_STATS, "create_solutions for problem #%d took %d ms\n", probnr, solv_timems(now));
702 }
703
704
705 /**************************************************************************/
706
707 unsigned int
708 solver_problem_count(Solver *solv)
709 {
710   return solv->problems.count / 2;
711 }
712
713 Id
714 solver_next_problem(Solver *solv, Id problem)
715 {
716   if (!problem)
717     return solv->problems.count ? 1 : 0;
718   return (problem + 1) * 2 - 1 < solv->problems.count ? problem + 1 : 0;
719 }
720
721 unsigned int
722 solver_solution_count(Solver *solv, Id problem)
723 {
724   Id solidx = solv->problems.elements[problem * 2 - 1];
725   if (solv->solutions.elements[solidx] < 0)
726     create_solutions(solv, problem, solidx);
727   return solv->solutions.elements[solidx];
728 }
729
730 Id
731 solver_next_solution(Solver *solv, Id problem, Id solution)
732 {
733   Id solidx = solv->problems.elements[problem * 2 - 1];
734   if (solv->solutions.elements[solidx] < 0)
735     create_solutions(solv, problem, solidx);
736   return solv->solutions.elements[solidx + solution + 1] ? solution + 1 : 0;
737 }
738
739 unsigned int
740 solver_solutionelement_count(Solver *solv, Id problem, Id solution)
741 {
742   Id solidx = solv->problems.elements[problem * 2 - 1];
743   solidx = solv->solutions.elements[solidx + solution];
744   return solv->solutions.elements[solidx];
745 }
746
747 Id
748 solver_solutionelement_internalid(Solver *solv, Id problem, Id solution)
749 {
750   Id solidx = solv->problems.elements[problem * 2 - 1];
751   solidx = solv->solutions.elements[solidx + solution];
752   return solv->solutions.elements[solidx + 2 * solv->solutions.elements[solidx] + 3];
753 }
754
755 Id
756 solver_solutionelement_extrajobflags(Solver *solv, Id problem, Id solution)
757 {
758   Id solidx = solv->problems.elements[problem * 2 - 1];
759   solidx = solv->solutions.elements[solidx + solution];
760   return solv->solutions.elements[solidx + 2 * solv->solutions.elements[solidx] + 4];
761 }
762
763
764 /*
765  *  return the next item of the proposed solution
766  *  here are the possibilities for p / rp and what
767  *  the solver expects the application to do:
768  *    p                             rp
769  *  -------------------------------------------------------
770  *    SOLVER_SOLUTION_INFARCH       pkgid
771  *    -> add (SOLVER_INSTALL|SOLVER_SOLVABLE, rp) to the job
772  *    SOLVER_SOLUTION_DISTUPGRADE   pkgid
773  *    -> add (SOLVER_INSTALL|SOLVER_SOLVABLE, rp) to the job
774  *    SOLVER_SOLUTION_BEST          pkgid
775  *    -> add (SOLVER_INSTALL|SOLVER_SOLVABLE, rp) to the job
776  *    SOLVER_SOLUTION_JOB           jobidx
777  *    -> remove job (jobidx - 1, jobidx) from job queue
778  *    pkgid (> 0)                   0
779  *    -> add (SOLVER_ERASE|SOLVER_SOLVABLE, p) to the job
780  *    pkgid (> 0)                   pkgid (> 0)
781  *    -> add (SOLVER_INSTALL|SOLVER_SOLVABLE, rp) to the job
782  *       (this will replace package p)
783  *         
784  * Thus, the solver will either ask the application to remove
785  * a specific job from the job queue, or ask to add an install/erase
786  * job to it.
787  *
788  */
789
790 Id
791 solver_next_solutionelement(Solver *solv, Id problem, Id solution, Id element, Id *p, Id *rp)
792 {
793   Id solidx = solv->problems.elements[problem * 2 - 1];
794   solidx = solv->solutions.elements[solidx + solution];
795   if (!solidx)
796     return 0;
797   solidx += 1 + element * 2;
798   if (!solv->solutions.elements[solidx] && !solv->solutions.elements[solidx + 1])
799     return 0;
800   *p = solv->solutions.elements[solidx];
801   *rp = solv->solutions.elements[solidx + 1];
802   return element + 1;
803 }
804
805 void
806 solver_take_solutionelement(Solver *solv, Id p, Id rp, Id extrajobflags, Queue *job)
807 {
808   int i;
809
810   if (p == SOLVER_SOLUTION_JOB)
811     {
812       job->elements[rp - 1] = SOLVER_NOOP;
813       job->elements[rp] = 0;
814       return;
815     }
816   if (rp <= 0 && p <= 0)
817     return;     /* just in case */
818   if (rp > 0)
819     p = SOLVER_INSTALL|SOLVER_SOLVABLE|extrajobflags;
820   else
821     {
822       rp = p;
823       p = SOLVER_ERASE|SOLVER_SOLVABLE|extrajobflags;
824     }
825   for (i = 0; i < job->count; i += 2)
826     if (job->elements[i] == p && job->elements[i + 1] == rp)
827       return;
828   queue_push2(job, p, rp);
829 }
830
831 void
832 solver_take_solution(Solver *solv, Id problem, Id solution, Queue *job)
833 {
834   Id p, rp, element = 0;
835   Id extrajobflags = solver_solutionelement_extrajobflags(solv, problem, solution);
836   while ((element = solver_next_solutionelement(solv, problem, solution, element, &p, &rp)) != 0)
837     solver_take_solutionelement(solv, p, rp, extrajobflags, job);
838 }
839
840
841 /*-------------------------------------------------------------------
842  * 
843  * find problem rule
844  */
845
846 static void
847 findproblemrule_internal(Solver *solv, Id idx, Id *reqrp, Id *conrp, Id *sysrp, Id *jobrp)
848 {
849   Id rid, d;
850   Id lreqr, lconr, lsysr, ljobr;
851   Rule *r;
852   Id jobassert = 0;
853   int i, reqset = 0;    /* 0: unset, 1: installed, 2: jobassert, 3: assert */
854
855   /* find us a jobassert rule */
856   for (i = idx; (rid = solv->learnt_pool.elements[i]) != 0; i++)
857     {
858       if (rid < solv->jobrules || rid >= solv->jobrules_end)
859         continue;
860       r = solv->rules + rid;
861       d = r->d < 0 ? -r->d - 1 : r->d;
862       if (!d && r->w2 == 0 && r->p > 0)
863         {
864           jobassert = r->p;
865           break;
866         }
867     }
868
869   /* the problem rules are somewhat ordered from "near to the problem" to
870    * "near to the job" */
871   lreqr = lconr = lsysr = ljobr = 0;
872   while ((rid = solv->learnt_pool.elements[idx++]) != 0)
873     {
874       assert(rid > 0);
875       if (rid >= solv->learntrules)
876         findproblemrule_internal(solv, solv->learnt_why.elements[rid - solv->learntrules], &lreqr, &lconr, &lsysr, &ljobr);
877       else if ((rid >= solv->jobrules && rid < solv->jobrules_end) || (rid >= solv->infarchrules && rid < solv->infarchrules_end) || (rid >= solv->duprules && rid < solv->duprules_end) || (rid >= solv->bestrules && rid < solv->bestrules_end))
878         {
879           if (!*jobrp)
880             *jobrp = rid;
881         }
882       else if (rid >= solv->updaterules && rid < solv->updaterules_end)
883         {
884           if (!*sysrp)
885             *sysrp = rid;
886         }
887       else
888         {
889           assert(rid < solv->rpmrules_end);
890           r = solv->rules + rid;
891           d = r->d < 0 ? -r->d - 1 : r->d;
892           if (!d && r->w2 < 0)
893             {
894               if (!*conrp)
895                 *conrp = rid;
896             }
897           else
898             {
899               if (!d && r->w2 == 0 && reqset < 3)
900                 {
901                   if (*reqrp > 0 && r->p < -1)
902                     {
903                       Id op = -solv->rules[*reqrp].p;
904                       if (op > 1 && solv->pool->solvables[op].arch != solv->pool->solvables[-r->p].arch)
905                         continue;       /* different arch, skip */
906                     }
907                   /* prefer assertions */
908                   *reqrp = rid;
909                   reqset = 3;
910                 }
911               else if (jobassert && r->p == -jobassert)
912                 {
913                   /* prefer rules of job assertions */
914                   *reqrp = rid;
915                   reqset = 2;
916                 }
917               else if (solv->installed && r->p < 0 && solv->pool->solvables[-r->p].repo == solv->installed && reqset <= 1)
918                 {
919                   /* prefer rules of job installed package so that the user doesn't get confused by strange packages */
920                   *reqrp = rid;
921                   reqset = 1;
922                 }
923               else if (!*reqrp)
924                 *reqrp = rid;
925             }
926         }
927     }
928   if (!*reqrp && lreqr)
929     *reqrp = lreqr;
930   if (!*conrp && lconr)
931     *conrp = lconr;
932   if (!*jobrp && ljobr)
933     *jobrp = ljobr;
934   if (!*sysrp && lsysr)
935     *sysrp = lsysr;
936 }
937
938 /* 
939  * find problem rule
940  *
941  * search for a rule that describes the problem to the
942  * user. Actually a pretty hopeless task that may leave the user
943  * puzzled. To get all of the needed information use
944  * solver_findallproblemrules() instead.
945  */
946
947 Id
948 solver_findproblemrule(Solver *solv, Id problem)
949 {
950   Id reqr, conr, sysr, jobr;
951   Id idx = solv->problems.elements[2 * problem - 2];
952   reqr = conr = sysr = jobr = 0;
953   findproblemrule_internal(solv, idx, &reqr, &conr, &sysr, &jobr);
954   if (reqr)
955     return reqr;        /* some requires */
956   if (conr)
957     return conr;        /* some conflict */
958   if (sysr)
959     return sysr;        /* an update rule */
960   if (jobr)
961     return jobr;        /* a user request */
962   assert(0);
963   return 0;
964 }
965
966 /*-------------------------------------------------------------------*/
967
968 static void
969 findallproblemrules_internal(Solver *solv, Id idx, Queue *rules)
970 {
971   Id rid;
972   while ((rid = solv->learnt_pool.elements[idx++]) != 0)
973     {
974       if (rid >= solv->learntrules)
975         {
976           findallproblemrules_internal(solv, solv->learnt_why.elements[rid - solv->learntrules], rules);
977           continue;
978         }
979       queue_pushunique(rules, rid);
980     }
981 }
982
983 /*
984  * find all problem rule
985  *
986  * return all rules that lead to the problem. This gives the user
987  * all of the information to understand the problem, but the result
988  * can be a large number of rules.
989  */
990
991 void
992 solver_findallproblemrules(Solver *solv, Id problem, Queue *rules)
993 {
994   queue_empty(rules);
995   findallproblemrules_internal(solv, solv->problems.elements[2 * problem - 2], rules);
996 }
997
998 /* EOF */