Ticket #6614: livetv_inputgroups.patch

mythtv/programs/mythbackend/mainserver.cpp

@@ -25 +25 @@
 
 #include <list>
 #include <iostream>
+#include <algorithm>
 using namespace std;
 
 #include <sys/stat.h>
@@ -41 +42 @@
 #include "libmyth/mythcontext.h"
 #include "libmyth/util.h"
 #include "libmyth/mythdbcon.h"
+#include "libmythtv/cardutil.h"
+#include "libmythtv/sourceutil.h"
 
 #include "mainserver.h"
 #include "scheduler.h"
@@ -100 +103 @@
 
 };
 
+// This is a copy of is_input_group_busy from channelbase.cpp.
+static bool is_input_group_busy(
+    uint                       inputid,
+    uint                       groupid,
+    const vector<uint>        &excluded_cardids,
+    QMap<uint,bool>           &busygrp,
+    QMap<uint,bool>           &busyrec,
+    QMap<uint,TunedInputInfo> &busyin,
+    uint                      &mplexid_restriction)
+{
+    // If none are busy, we don't need to check further
+    QMap<uint,bool>::const_iterator bit = busygrp.find(groupid);
+    if ((bit != busygrp.end()) && !*bit)
+        return false;
+
+    vector<TunedInputInfo> conflicts;
+    vector<uint> cardids = CardUtil::GetGroupCardIDs(groupid);
+    for (uint i = 0; i < cardids.size(); i++)
+    {
+        if (find(excluded_cardids.begin(),
+                 excluded_cardids.end(), cardids[i]) != excluded_cardids.end())
+        {
+            continue;
+        }
+
+        TunedInputInfo info;
+        QMap<uint,bool>::const_iterator it = busyrec.find(cardids[i]);
+        if (it == busyrec.end())
+        {
+            busyrec[cardids[i]] = RemoteIsBusy(cardids[i], info);
+            it = busyrec.find(cardids[i]);
+            if (*it)
+                busyin[cardids[i]] = info;
+        }
+
+        if (*it)
+            conflicts.push_back(busyin[cardids[i]]);
+    }
+
+    // If none are busy, we don't need to check further
+    busygrp[groupid] = !conflicts.empty();
+    if (conflicts.empty())
+        return false;
+
+    InputInfo in;
+    in.inputid = inputid;
+    if (!CardUtil::GetInputInfo(in))
+        return true;
+
+    // If they aren't using the same source they are definately busy
+    bool is_busy_input = false;
+
+    for (uint i = 0; i < conflicts.size() && !is_busy_input; i++)
+        is_busy_input = (in.sourceid != conflicts[i].sourceid);
+
+    if (is_busy_input)
+        return true;
+
+    // If the source's channels aren't digitally tuned then there is a conflict
+    is_busy_input = !SourceUtil::HasDigitalChannel(in.sourceid);
+    if (!is_busy_input && conflicts[0].chanid)
+    {
+        MSqlQuery query(MSqlQuery::InitCon());
+        query.prepare(
+            "SELECT mplexid "
+            "FROM channel "
+            "WHERE chanid = :CHANID");
+        query.bindValue(":CHANID", conflicts[0].chanid);
+        if (!query.exec())
+            MythContext::DBError("is_input_group_busy", query);
+        else if (query.next())
+        {
+            mplexid_restriction = query.value(0).toUInt();
+            mplexid_restriction = (32767 == mplexid_restriction) ?
+                0 : mplexid_restriction;
+        }
+    }
+
+    return is_busy_input;
+}
+
+// This is a copy of is_input_busy from channelbase.cpp.
+static bool is_input_busy(
+    uint                       inputid,
+    const vector<uint>        &groupids,
+    const vector<uint>        &excluded_cardids,
+    QMap<uint,bool>           &busygrp,
+    QMap<uint,bool>           &busyrec,
+    QMap<uint,TunedInputInfo> &busyin,
+    uint                      &mplexid_restriction)
+{
+    bool is_busy = false;
+    for (uint i = 0; i < groupids.size() && !is_busy; i++)
+    {
+        is_busy |= is_input_group_busy(
+            inputid, groupids[i], excluded_cardids,
+            busygrp, busyrec, busyin, mplexid_restriction);
+    }
+    return is_busy;
+}
+
+static bool is_input_in_busy_group(uint inputid)
+{
+    vector<uint> groupids = CardUtil::GetInputGroups(inputid);
+    vector<uint> excluded_cardids;
+    excluded_cardids.push_back(CardUtil::GetCardID(inputid));
+    QMap<uint,bool>           busygrp;
+    QMap<uint,bool>           busyrec;
+    QMap<uint,TunedInputInfo> busyin;
+    uint mplexid_restriction = 0;
+    return is_input_busy(inputid, groupids, excluded_cardids,
+                         busygrp, busyrec, busyin, mplexid_restriction);
+}
+
+static bool is_card_in_busy_group(uint cardid)
+{
+    vector<uint> inputids = CardUtil::GetInputIDs(cardid);
+    for (uint i = 0; i < inputids.size(); i++)
+    {
+        if (is_input_in_busy_group(inputids[i]))
+            return true;
+    }
+    return false;
+}
+
+
 QMutex MainServer::truncate_and_close_lock;
 
 class ProcessRequestThread : public QThread
@@ -2528 +2657 @@
         if ((enchost == pbshost) &&
             (elink->IsConnected()) &&
             (!elink->IsBusy()) &&
+            !is_card_in_busy_group(elink->GetCardID()) &&
             (!elink->IsTunerLocked()))
         {
             encoder = elink;
@@ -2646 +2776 @@
                 enchost = elink->GetHostName();
 
             if (enchost == pbshost && elink->IsConnected() &&
-                !elink->IsBusy() && !elink->IsTunerLocked())
+                !elink->IsBusy() &&
+                !is_card_in_busy_group(elink->GetCardID()) &&
+                !elink->IsTunerLocked())
             {
                 encoder = elink;
                 retval = iter.key();
@@ -2657 +2789 @@
         }
 
         if ((retval == -1 || lastcard) && elink->IsConnected() &&
-            !elink->IsBusy() && !elink->IsTunerLocked())
+            !elink->IsBusy() &&
+            !is_card_in_busy_group(elink->GetCardID()) &&
+            !elink->IsTunerLocked())
         {
             encoder = elink;
             retval = iter.key();
@@ -2707 +2841 @@
 
         if ((elink->IsConnected()) &&
             (!elink->IsBusy()) &&
+            !is_card_in_busy_group(elink->GetCardID()) &&
             (!elink->IsTunerLocked()))
         {
             count++;
@@ -2731 +2866 @@
 
         if ((elink->IsConnected()) &&
             (!elink->IsBusy()) &&
+            !is_card_in_busy_group(elink->GetCardID()) &&
             (!elink->IsTunerLocked()))
         {
             strlist << QString::number(iter.key());
@@ -2780 +2916 @@
             if ((retval == -1) &&
                 (elink->IsConnected()) &&
                 (!elink->IsBusy()) &&
+                !is_card_in_busy_group(elink->GetCardID()) &&
                 (!elink->IsTunerLocked()))
             {
                 encoder = elink;