improve the API for Upgrade()
[ntk/apt.git] / apt-pkg / algorithms.h
1 // -*- mode: cpp; mode: fold -*-
2 // Description /*{{{*/
3 // $Id: algorithms.h,v 1.10 2001/05/22 04:17:41 jgg Exp $
4 /* ######################################################################
5
6 Algorithms - A set of misc algorithms
7
8 This simulate class displays what the ordering code has done and
9 analyses it with a fresh new dependency cache. In this way we can
10 see all of the effects of an upgrade run.
11
12 pkgDistUpgrade computes an upgrade that causes as many packages as
13 possible to move to the newest verison.
14
15 pkgApplyStatus sets the target state based on the content of the status
16 field in the status file. It is important to get proper crash recovery.
17
18 pkgFixBroken corrects a broken system so that it is in a sane state.
19
20 pkgAllUpgrade attempts to upgade as many packages as possible but
21 without installing new packages.
22
23 The problem resolver class contains a number of complex algorithms
24 to try to best-guess an upgrade state. It solves the problem of
25 maximizing the number of install state packages while having no broken
26 packages.
27
28 ##################################################################### */
29 /*}}}*/
30 #ifndef PKGLIB_ALGORITHMS_H
31 #define PKGLIB_ALGORITHMS_H
32
33
34 #include <apt-pkg/packagemanager.h>
35 #include <apt-pkg/depcache.h>
36
37 #include <iostream>
38
39 #include <apt-pkg/macros.h>
40
41 #ifndef APT_8_CLEANER_HEADERS
42 #include <apt-pkg/acquire.h>
43 using std::ostream;
44 #endif
45
46 class pkgAcquireStatus;
47
48 namespace APT {
49 namespace Upgrade {
50 enum UpgradeMode {NO_INSTALL_OR_REMOVE,
51 ALLOW_NEW_INSTALLS,
52 ALLOW_REMOVAL_AND_NEW_INSTALLS};
53 bool Upgrade(pkgDepCache &Cache, UpgradeMode mode);
54 }
55 }
56
57 class pkgSimulate : public pkgPackageManager /*{{{*/
58 {
59 protected:
60
61 class Policy : public pkgDepCache::Policy
62 {
63 pkgDepCache *Cache;
64 public:
65
66 virtual VerIterator GetCandidateVer(PkgIterator const &Pkg)
67 {
68 return (*Cache)[Pkg].CandidateVerIter(*Cache);
69 }
70
71 Policy(pkgDepCache *Cache) : Cache(Cache) {};
72 };
73
74 unsigned char *Flags;
75
76 Policy iPolicy;
77 pkgDepCache Sim;
78 pkgDepCache::ActionGroup group;
79
80 // The Actuall installation implementation
81 virtual bool Install(PkgIterator Pkg,std::string File);
82 virtual bool Configure(PkgIterator Pkg);
83 virtual bool Remove(PkgIterator Pkg,bool Purge);
84
85 private:
86 void ShortBreaks();
87 void Describe(PkgIterator iPkg,std::ostream &out,bool Current,bool Candidate);
88
89 public:
90
91 pkgSimulate(pkgDepCache *Cache);
92 ~pkgSimulate();
93 };
94 /*}}}*/
95 class pkgProblemResolver /*{{{*/
96 {
97 private:
98 /** \brief dpointer placeholder (for later in case we need it) */
99 void *d;
100
101 pkgDepCache &Cache;
102 typedef pkgCache::PkgIterator PkgIterator;
103 typedef pkgCache::VerIterator VerIterator;
104 typedef pkgCache::DepIterator DepIterator;
105 typedef pkgCache::PrvIterator PrvIterator;
106 typedef pkgCache::Version Version;
107 typedef pkgCache::Package Package;
108
109 enum Flags {Protected = (1 << 0), PreInstalled = (1 << 1),
110 Upgradable = (1 << 2), ReInstateTried = (1 << 3),
111 ToRemove = (1 << 4)};
112 int *Scores;
113 unsigned char *Flags;
114 bool Debug;
115
116 // Sort stuff
117 static pkgProblemResolver *This;
118 static int ScoreSort(const void *a,const void *b);
119
120 struct PackageKill
121 {
122 PkgIterator Pkg;
123 DepIterator Dep;
124 };
125
126 void MakeScores();
127 bool DoUpgrade(pkgCache::PkgIterator Pkg);
128
129 bool ResolveInternal(bool const BrokenFix = false);
130 bool ResolveByKeepInternal();
131
132 protected:
133 bool InstOrNewPolicyBroken(pkgCache::PkgIterator Pkg);
134
135 public:
136
137 inline void Protect(pkgCache::PkgIterator Pkg) {Flags[Pkg->ID] |= Protected; Cache.MarkProtected(Pkg);};
138 inline void Remove(pkgCache::PkgIterator Pkg) {Flags[Pkg->ID] |= ToRemove;};
139 inline void Clear(pkgCache::PkgIterator Pkg) {Flags[Pkg->ID] &= ~(Protected | ToRemove);};
140
141 // Try to intelligently resolve problems by installing and removing packages
142 bool Resolve(bool BrokenFix = false);
143
144 // Try to resolve problems only by using keep
145 bool ResolveByKeep();
146
147 __deprecated void InstallProtect();
148
149 pkgProblemResolver(pkgDepCache *Cache);
150 ~pkgProblemResolver();
151 };
152 /*}}}*/
153 bool pkgDistUpgrade(pkgDepCache &Cache);
154 bool pkgApplyStatus(pkgDepCache &Cache);
155 bool pkgFixBroken(pkgDepCache &Cache);
156
157 bool pkgAllUpgrade(pkgDepCache &Cache);
158
159 bool pkgMinimizeUpgrade(pkgDepCache &Cache);
160
161 void pkgPrioSortList(pkgCache &Cache,pkgCache::Version **List);
162
163 bool ListUpdate(pkgAcquireStatus &progress, pkgSourceList &List, int PulseInterval=0);
164 bool AcquireUpdate(pkgAcquire &Fetcher, int const PulseInterval = 0,
165 bool const RunUpdateScripts = true, bool const ListCleanup = true);
166
167 #endif