Ticket #2416: ticket2416_copy_eit_2_other_source.patch

mythtv/libs/libmythtv/eithelper.cpp

@@ -34 +34 @@
                                 uint atscmajor, uint atscminor);
 static uint get_chan_id_from_db(uint sourceid,  uint serviceid,
                                 uint networkid, uint transportid);
+static uint get_mapped_chan_id_from_db(uint serviceid);
 static void init_fixup(QMap<uint64_t,uint> &fix);
 static int calc_eit_utc_offset(void);
 
@@ -485 +486 @@
             seriesId,  programId);
 
         db_events.enqueue(event);
+
+        uint mapped_chanid = GetMappedChanID(eit->ServiceID());
+
+        if (mapped_chanid > 0)
+        {
+            DBEventEIT *event = new DBEventEIT(mapped_chanid,
+                                         title,     subtitle,      description,
+                                         category,  category_type,
+                                         starttime, endtime,       fix,
+                                         subtitle_type,
+                                         audio_props,
+                                         video_props, stars,
+                                         seriesId,  programId);
+            db_events.enqueue(event);
+        }
     }
 }
 
@@ -692 +708 @@
 
     uint chanid = get_chan_id_from_db(sourceid, serviceid, networkid, tsid);
     if (chanid)
+        {
         srv_to_chanid[key] = chanid;
 
+        uint mapped_chanid = get_mapped_chan_id_from_db(serviceid);
+        if(mapped_chanid)
+            serviceid_to_mapped_chanid[serviceid] = mapped_chanid;
+    }
+
     return chanid;
 }
 
+uint EITHelper::GetMappedChanID(uint serviceid)
+{
+    ServiceIDToMappedChanID::const_iterator it = serviceid_to_mapped_chanid.find(serviceid);
+    if (it != serviceid_to_mapped_chanid.end())
+        return max(*it, 0);
+    return 0;
+} 
+
 static uint get_chan_id_from_db(uint sourceid,
                                 uint atsc_major, uint atsc_minor)
 {
@@ -767 +797 @@
     return useOnAirGuide ? chanid : 0;
 }
 
+static uint get_mapped_chan_id_from_db(uint serviceid)
+{
+    MSqlQuery query(MSqlQuery::InitCon());
+
+    QString qstr =
+        "SELECT mapped_chanid "
+        "FROM eit_mapping "
+        "WHERE serviceid = :SERVICEID";
+
+    query.prepare(qstr);
+    query.bindValue(":SERVICEID", serviceid);
+    if (!query.exec() || !query.isActive())
+        MythDB::DBError("Looking up mapped chanID", query);
+
+    if (query.next())
+    {
+        return query.value(0).toUInt();
+    }
+
+    return 0;
+}
+
 static void init_fixup(QMap<uint64_t,uint> &fix)
 {
     ///////////////////////////////////////////////////////////////////////////

mythtv/libs/libmythtv/eithelper.h

@@ -45 +45 @@
 typedef QMap<uint,EventIDToATSCEvent>      ATSCSRCToEvents;
 typedef QMap<uint,EventIDToETT>            ATSCSRCToETTs;
 typedef QMap<unsigned long long,int>       ServiceToChanID;
+typedef QMap<int,int>                      ServiceIDToMappedChanID;
 
 class DBEventEIT;
 class EITFixUp;
@@ -92 +93 @@
   private:
     uint GetChanID(uint atsc_major, uint atsc_minor);
     uint GetChanID(uint serviceid, uint networkid, uint transportid);
+        uint GetMappedChanID(uint serviceid);
 
     void CompleteEvent(uint atsc_major, uint atsc_minor,
                        const ATSCEvent &event,
@@ -100 +102 @@
         //QListList_Events  eitList;      ///< Event Information Tables List
     mutable QMutex    eitList_lock; ///< EIT List lock
     mutable ServiceToChanID srv_to_chanid;
+    mutable ServiceIDToMappedChanID serviceid_to_mapped_chanid;
 
     EITFixUp               *eitfixup;
     static EITCache        *eitcache;