Ticket #3126: patch

programs/mythbackend/mainserver.cpp

@@ -627 +627 @@
     {
         VERBOSE(VB_IMPORTANT, "Got 'OK' out of sequence.");
     }
+    else if (command == "START_TRANSCODE_DISK" || command == "STOP_TRANSCODE_DISK")
+    {
+        preHandleTranscodeDisk(command, tokens, pbs->getSocket());
+    }
     else if (command == "UNKNOWN_COMMAND")
     {
         VERBOSE(VB_IMPORTANT, "Got 'UNKNOWN_COMMAND' out of sequence.");
@@ -3653 +3657 @@
     gContext->RefreshBackendConfig();
 
     QStringList retlist = "OK";
-    SendResponse(socket, retlist);    
+    SendResponse(socket, retlist);
 }
 
+int MainServer::HandleTranscodeDisk(int kind, QString chanid,\
+                                         QDateTime starttime, QString profile)
+{
+    if (!JobQueue::IsJobQueuedOrRunning(JOB_TRANSCODE, chanid, starttime))
+    {
+        if (kind == 0) 
+        {
+            QString jobHost = "";
+            if (gContext->GetNumSetting("JobsRunOnRecordHost", 0))
+                jobHost = gContext->GetHostName(); //set as local hostname
+
+            if ( SetTranscodeProfile(profile, chanid, starttime) == 0 )
+                JobQueue::QueueJob(JOB_TRANSCODE, chanid, starttime,
+                                            "", "", "", JOB_USE_CUTLIST);
+            else return -1;
+        }
+    }
+    else
+        if (kind == 1)
+            //Cancell job (kind == 1)
+            JobQueue::ChangeJobCmds(JOB_TRANSCODE, chanid,
+                                    starttime, JOB_STOP);
+
+    return 0;
+}
+
+void MainServer::preHandleTranscodeDisk(QString command, QStringList tokens, MythSocket *socket)
+{
+    QStringList retlist;
+    // 3 = command, from, to
+    // 2 = command, from (then it'll overwrite the original one)
+    if ( ( (tokens.size() < 3) && (command == "START_TRANSCODE_DISK") ) ||
+            ( (tokens.size() != 2) && (command == "STOP_TRANSCODE_DISK") ) )
+        VERBOSE(VB_IMPORTANT, "Bad Transcode query");
+    else
+    {
+        QStringList from = QStringList::split("_", tokens[1]);
+        QStringList date = QStringList::split(".", from[1]);
+        QString chanid = from[0];
+
+        //create the right date format
+        char* temp = (char*)malloc(sizeof(char) * date[0].length());
+        strncpy(temp, date[0].ascii(), date[0].length());
+
+        char* temp1 = (char*)malloc(sizeof(char) * 20);
+        snprintf(temp1, 20, "%.4s-%.2s-%.2sT%.2s:%.2s:%.2s\n", temp, temp+4,\
+                    temp+6, temp+8, temp+10, temp+12);
+        temp1[19] = '\0';
+
+        QString date_iso = QString(temp1);
+        QDateTime starttime = QDateTime::fromString(date_iso, Qt::ISODate);
+        free(temp);
+        free(temp1);
+
+        int result;
+        QStringList strlist;
+
+        if (command == "START_TRANSCODE_DISK")
+        {
+            //tokens[2] = PROFILE FOR TRANSCODING
+            VERBOSE(VB_IMPORTANT,"Starting transcode of file on disk.");
+            QString profile = QString("");
+            uint i;
+            for (i=2; i <= tokens.size(); i++)
+            {
+                profile.append(tokens[i]);
+                if (i < (tokens.size()-1) )
+                    profile.append(" ");
+            }
+
+            result = HandleTranscodeDisk(0, chanid, starttime, profile);
+        }
+        else if (command == "STOP_TRANSCODE_DISK")
+        {
+            VERBOSE(VB_IMPORTANT,"Stop transcode of file on disk.");
+            result = HandleTranscodeDisk(1, chanid, starttime, NULL);
+        }
+
+        if (result == 0)
+            strlist << "OK";
+        else
+            strlist << "ERROR";
+
+        SendResponse(socket, strlist);
+        return;
+    }
+}
+
+
 void MainServer::HandleBlockShutdown(bool blockShutdown, PlaybackSock *pbs)
 {            
     pbs->setBlockShutdown(blockShutdown);

programs/mythbackend/mainserver.h

@@ -67 +67 @@
     } DeleteStruct;
 
     void ProcessRequestWork(MythSocket *sock);
+    void preHandleTranscodeDisk(QString command, QStringList tokens,\
+                                MythSocket *socket);
+    int HandleTranscodeDisk(int kind, QString chanid,\
+                                 QDateTime starttime, QString profile);
     void HandleAnnounce(QStringList &slist, QStringList commands, 
                         MythSocket *socket);
     void HandleDone(MythSocket *socket);

programs/mythbackend/backendutil.cpp

@@ -287 +287 @@
     }
 }
 
+int SetTranscodeProfile(QString profile, QString chanid, QDateTime recstartts)
+{
+    if(profile == "Default") // use whatever is already in the transcoder
+        return 0;
+
+    MSqlQuery query(MSqlQuery::InitCon());
+
+    if(profile == "Autodetect")
+    {
+        query.prepare("UPDATE recorded "
+                      "SET transcoder = 0 "
+                      "WHERE chanid = :CHANID "
+                      "AND starttime = :START");
+        query.bindValue(":CHANID",  chanid);
+        query.bindValue(":START",  recstartts);
+
+        if (!query.exec())
+        {
+            MythContext::DBError("MythBackendUtil: unable to update transcoder "
+                                 "in recorded table", query);
+            return -1;
+        }
+    }
+    else
+    {
+        MSqlQuery pidquery(MSqlQuery::InitCon());
+        pidquery.prepare("SELECT r.id "
+                         "FROM recordingprofiles r, profilegroups p "
+                         "WHERE r.profilegroup = p.id "
+                             "AND p.name = 'Transcoders' "
+                             "AND r.name = :PROFILE ");
+        pidquery.bindValue(":PROFILE",  profile);
+
+        if (pidquery.exec() && pidquery.isActive() && pidquery.next())
+        {
+            query.prepare("UPDATE recorded "
+                          "SET transcoder = :TRANSCODER "
+                          "WHERE chanid = :CHANID "
+                              "AND starttime = :START");
+            query.bindValue(":TRANSCODER", pidquery.value(0).toInt());
+            query.bindValue(":CHANID",  chanid);
+            query.bindValue(":START",  recstartts);
+
+            if (!query.exec())
+            {
+                MythContext::DBError("MythBackendUtil: unable to update transcoder "
+                                 "in recorded table", query);
+                return -1;
+            }
+        }
+        else
+        {
+            MythContext::DBError("MythBackend: unable to query transcoder "
+                                 "profile ID", query);
+            return -1;
+        }
+    }
+
+    return 0;
+}
+
 /* vim: set expandtab tabstop=4 shiftwidth=4: */

programs/mythbackend/backendutil.h

@@ -13 +13 @@
 void GetFilesystemInfos(QMap<int, EncoderLink*> *tvList,
                         vector <FileSystemInfo> &fsInfos);
 
+int SetTranscodeProfile(QString profile, QString chanid, QDateTime recstartts);
+
 #endif
 
 /* vim: set expandtab tabstop=4 shiftwidth=4: */