X-Git-Url: https://git.cweiske.de/enigma2.git/blobdiff_plain/cc89d218402e01a1dc381a50130c6e3581a3555f..a7c89c1f5d2b670f51552bee927b11116ab7eecb:/lib/dvb/sec.cpp

diff --git a/lib/dvb/sec.cpp b/lib/dvb/sec.cpp
index b003e16d..b995b806 100644
--- a/lib/dvb/sec.cpp
+++ b/lib/dvb/sec.cpp
@@ -2,6 +2,8 @@
 #include <lib/dvb/sec.h>
 #include <lib/dvb/rotor_calc.h>
 
+#include <set>
+
 #if HAVE_DVB_API_VERSION < 3
 #define FREQUENCY Frequency
 #else
@@ -1032,6 +1034,110 @@ RESULT eDVBSatelliteEquipmentControl::setRotorPosNum(int rotor_pos_num)
 	return 0;
 }
 
+struct sat_compare
+{
+	int orb_pos, lofl, lofh;
+	sat_compare(int o, int lofl, int lofh)
+		:orb_pos(o), lofl(lofl), lofh(lofh)
+	{}
+	sat_compare(const sat_compare &x)
+		:orb_pos(x.orb_pos), lofl(x.lofl), lofh(x.lofh)
+	{}
+	bool operator < (const sat_compare & cmp) const
+	{
+		if (orb_pos == cmp.orb_pos)
+		{
+			if ( abs(lofl-cmp.lofl) < 200000 )
+			{
+				if (abs(lofh-cmp.lofh) < 200000)
+					return false;
+				return lofh<cmp.lofh;
+			}
+			return lofl<cmp.lofl;
+		}
+		return orb_pos < cmp.orb_pos;
+	}
+};
+
+PyObject *eDVBSatelliteEquipmentControl::get_exclusive_satellites(int tu1, int tu2)
+{
+	PyObject *ret=0;
+
+	int tu1_mask = 1 << tu1,
+		tu2_mask = 1 << tu2;
+
+	if (tu1 != tu2)
+	{
+		eDVBRegisteredFrontend *p1=NULL, *p2=NULL;
+		int cnt=0;
+		for (eSmartPtrList<eDVBRegisteredFrontend>::iterator it(m_avail_frontends.begin()); it != m_avail_frontends.end(); ++it, ++cnt)
+		{
+			if (cnt == tu1)
+				p1 = *it;
+			else if (cnt == tu2)
+				p2 = *it;
+		}
+		if (p1 && p2)
+		{
+			// check for linked tuners
+			int tmp1, tmp2;
+			p1->m_frontend->getData(7, tmp1);
+			p2->m_frontend->getData(7, tmp2);
+			if ((void*)tmp1 != p2 && (void*)tmp2 != p1)
+			{
+				// check for rotor dependency
+				p1->m_frontend->getData(8, tmp1);
+				p2->m_frontend->getData(8, tmp2);
+				if ((void*)tmp1 != p2 && (void*)tmp2 != p1)
+				{
+					std::set<sat_compare> tu1sats, tu2sats;
+					std::list<sat_compare> tu1difference, tu2difference;
+					std::insert_iterator<std::list<sat_compare> > insert1(tu1difference, tu1difference.begin()),
+						insert2(tu2difference, tu2difference.begin());
+					for (int idx=0; idx <= m_lnbidx; ++idx )
+					{
+						eDVBSatelliteLNBParameters &lnb_param = m_lnbs[idx];
+						for (std::map<int, eDVBSatelliteSwitchParameters>::iterator sit(lnb_param.m_satellites.begin());
+							sit != lnb_param.m_satellites.end(); ++sit)
+						{
+							if ( lnb_param.tuner_mask & tu1_mask )
+								tu1sats.insert(sat_compare(sit->first, lnb_param.m_lof_lo, lnb_param.m_lof_hi));
+							if ( lnb_param.tuner_mask & tu2_mask )
+								tu2sats.insert(sat_compare(sit->first, lnb_param.m_lof_lo, lnb_param.m_lof_hi));
+						}
+					}
+					std::set_difference(tu1sats.begin(), tu1sats.end(),
+						tu2sats.begin(), tu2sats.end(),
+						insert1);
+					std::set_difference(tu2sats.begin(), tu2sats.end(),
+						tu1sats.begin(), tu1sats.end(),
+						insert2);
+					if (!tu1sats.empty() || !tu2sats.empty())
+					{
+						int idx=0;
+						ret = PyList_New(2+tu1difference.size()+tu2difference.size());
+
+						PyList_SET_ITEM(ret, idx++, PyInt_FromLong(tu1difference.size()));
+						for(std::list<sat_compare>::iterator it(tu1difference.begin()); it != tu1difference.end(); ++it)
+							PyList_SET_ITEM(ret, idx++, PyInt_FromLong(it->orb_pos));
+
+						PyList_SET_ITEM(ret, idx++, PyInt_FromLong(tu2difference.size()));
+						for(std::list<sat_compare>::iterator it(tu2difference.begin()); it != tu2difference.end(); ++it)
+							PyList_SET_ITEM(ret, idx++, PyInt_FromLong(it->orb_pos));
+					}
+				}
+			}
+		}
+	}
+	if (!ret)
+	{
+		ret = PyList_New(2);
+		PyList_SET_ITEM(ret, 0, PyInt_FromLong(0));
+		PyList_SET_ITEM(ret, 1, PyInt_FromLong(0));
+	}
+	return ret;
+}
+
 RESULT eDVBSatelliteEquipmentControl::setTunerLinked(int tu1, int tu2)
 {
 	return setDependencyPointers(tu1, tu2, 7);