X-Git-Url: https://git.cweiske.de/enigma2.git/blobdiff_plain/d9ee52e4f0fbe9a1ae00d0e66f9e6f0a07fa319f..47c79edd46ef8d5cf7da20dd4e376d70c55b6243:/lib/dvb/db.h

diff --git a/lib/dvb/db.h b/lib/dvb/db.h
index 55998ebf..265c5970 100644
--- a/lib/dvb/db.h
+++ b/lib/dvb/db.h
@@ -1,15 +1,17 @@
 #ifndef __db_h
 #define __db_h
 
+#include <lib/base/eptrlist.h>
 #include <lib/dvb/idvb.h>
 #include <set>
 
-class ServiceDescriptionTable;
+class ServiceDescriptionSection;
 
 class eDVBDB: public iDVBChannelList
 {
-DECLARE_REF;
+DECLARE_REF(eDVBDB);
 	friend class eDVBDBQuery;
+	friend class eDVBDBBouquetQuery;
 private:
 	struct channel
 	{
@@ -19,7 +21,17 @@ private:
 	std::map<eDVBChannelID, channel> m_channels;
 	
 	std::map<eServiceReferenceDVB, ePtr<eDVBService> > m_services;
+	
+	std::map<std::string, eBouquet> m_bouquets;
 public:
+	void load();
+	void save();
+
+	void loadBouquet(const char *path);
+	void saveBouquet(const char *path);
+	void loadBouquets();
+	void saveBouquets();
+
 	eDVBDB();
 	virtual ~eDVBDB();
 	
@@ -31,20 +43,39 @@ public:
 	RESULT addService(const eServiceReferenceDVB &service, eDVBService *service);
 	RESULT getService(const eServiceReferenceDVB &reference, ePtr<eDVBService> &service);
 
-	RESULT startQuery(ePtr<iDVBChannelListQuery> &query, eDVBChannelQuery *query);
+	RESULT startQuery(ePtr<iDVBChannelListQuery> &query, eDVBChannelQuery *query, const eServiceReference &source);
+
+	RESULT getBouquet(const eServiceReference &ref, const eBouquet* &bouquet);
 };
 
 	// we have to add a possibility to invalidate here.
-class eDVBDBQuery: public iDVBChannelListQuery
+class eDVBDBQueryBase: public iDVBChannelListQuery
 {
-DECLARE_REF;
-private:
+DECLARE_REF(eDVBDBQueryBase);
+protected:
 	std::map<eServiceReferenceDVB, ePtr<eDVBService> >::iterator m_cursor;
 	ePtr<eDVBDB> m_db;
 	ePtr<eDVBChannelQuery> m_query;
+	eServiceReference m_source;
+public:
+	eDVBDBQueryBase(eDVBDB *db, const eServiceReference &source, eDVBChannelQuery *query);
+	int compareLessEqual(const eServiceReferenceDVB &a, const eServiceReferenceDVB &b);
+};
+
+class eDVBDBQuery: public eDVBDBQueryBase
+{
+	std::map<eServiceReferenceDVB, ePtr<eDVBService> >::iterator m_cursor;
+public:
+	eDVBDBQuery(eDVBDB *db, const eServiceReference &source, eDVBChannelQuery *query);
+	RESULT getNextResult(eServiceReferenceDVB &ref);
+};
+
+class eDVBDBBouquetQuery: public eDVBDBQueryBase
+{
+	std::list<eServiceReference>::iterator m_cursor;
 public:
-	eDVBDBQuery(eDVBDB *db, eDVBChannelQuery *query);
-	virtual RESULT getNextResult(eServiceReferenceDVB &ref);
+	eDVBDBBouquetQuery(eDVBDB *db, const eServiceReference &source, eDVBChannelQuery *query);
+	RESULT getNextResult(eServiceReferenceDVB &ref);
 };
 
 #endif