X-Git-Url: https://git.cweiske.de/enigma2.git/blobdiff_plain/c6cb89ad8e6db0985d61717a19cacd9e52c1687e..d43e4676bd757d1de73b8dd669154ffd42192b10:/lib/service/servicedvbrecord.cpp diff --git a/lib/service/servicedvbrecord.cpp b/lib/service/servicedvbrecord.cpp index 2f1dea0e..4c41b5ed 100644 --- a/lib/service/servicedvbrecord.cpp +++ b/lib/service/servicedvbrecord.cpp @@ -1,15 +1,18 @@ #include #include +#include #include DEFINE_REF(eDVBServiceRecord); -eDVBServiceRecord::eDVBServiceRecord(const eServiceReferenceDVB &ref): m_ref(ref), m_service_handler(1) +eDVBServiceRecord::eDVBServiceRecord(const eServiceReferenceDVB &ref): m_ref(ref) { CONNECT(m_service_handler.serviceEvent, eDVBServiceRecord::serviceEvent); m_state = stateIdle; m_want_record = 0; + m_tuned = 0; + m_target_fd = -1; } void eDVBServiceRecord::serviceEvent(int event) @@ -20,6 +23,9 @@ void eDVBServiceRecord::serviceEvent(int event) case eDVBServicePMTHandler::eventTuned: { eDebug("tuned.."); + m_tuned = 1; + if (m_state == stateRecording && m_want_record) + doRecord(); break; } case eDVBServicePMTHandler::eventNewProgramInfo: @@ -33,12 +39,57 @@ void eDVBServiceRecord::serviceEvent(int event) } } - -RESULT eDVBServiceRecord::prepare(const char *filename) +RESULT eDVBServiceRecord::prepare(const char *filename, time_t begTime, time_t endTime, int eit_event_id) { m_filename = filename; if (m_state == stateIdle) - return m_service_handler.tune(m_ref); + { + int ret = doPrepare(); + if (!ret) + { + eEPGCache::getInstance()->Lock(); + const eit_event_struct *event = 0; + eServiceReferenceDVB ref = m_ref.getParentServiceReference(); + if (!ref.valid()) + ref = m_ref; + if ( eit_event_id != -1 ) + { + eDebug("query epg event id %d", eit_event_id); + eEPGCache::getInstance()->lookupEventId(ref, eit_event_id, event); + } + if ( !event && (begTime != -1 && endTime != -1) ) + { + time_t queryTime = begTime + ((endTime-begTime)/2); + tm beg, end, query; + localtime_r(&begTime, &beg); + localtime_r(&endTime, &end); + localtime_r(&queryTime, &query); + eDebug("query stime %d:%d:%d, etime %d:%d:%d, qtime %d:%d:%d", + beg.tm_hour, beg.tm_min, beg.tm_sec, + end.tm_hour, end.tm_min, end.tm_sec, + query.tm_hour, query.tm_min, query.tm_sec); + eEPGCache::getInstance()->lookupEventTime(ref, queryTime, event); + } + if ( event ) + { + eDebug("found event.. store to disc"); + std::string fname = filename; + fname.erase(fname.length()-2, 2); + fname+="eit"; + int fd = open(fname.c_str(), O_CREAT|O_WRONLY, 0777); + if (fd>-1) + { + int evLen=HILO(event->descriptors_loop_length)+12/*EIT_LOOP_SIZE*/; + int wr = ::write( fd, (unsigned char*)event, evLen ); + if ( wr != evLen ) + eDebug("eit write error (%m)"); + ::close(fd); + } + } + eEPGCache::getInstance()->Unlock(); + } + return ret; + } else return -1; } @@ -56,7 +107,13 @@ RESULT eDVBServiceRecord::stop() eDebug("stop recording!!"); if (m_state == stateRecording) { - m_record->stop(); + if (m_record) + m_record->stop(); + if (m_target_fd >= 0) + { + ::close(m_target_fd); + m_target_fd = -1; + } m_state = statePrepared; } @@ -74,16 +131,39 @@ int eDVBServiceRecord::doPrepare() /* allocate a ts recorder if we don't already have one. */ if (m_state == stateIdle) { + m_pids_active.clear(); + m_state = statePrepared; + return m_service_handler.tune(m_ref, 0); + } + return 0; +} + +int eDVBServiceRecord::doRecord() +{ + int err = doPrepare(); + if (err) + return err; + + if (!m_tuned) + return 0; /* try it again when we are tuned in */ + + if (!m_record && m_tuned) + { + eDebug("Recording to %s...", m_filename.c_str()); ::remove(m_filename.c_str()); - int fd = ::open(m_filename.c_str(), O_WRONLY|O_CREAT, 0644); + int fd = ::open(m_filename.c_str(), O_WRONLY|O_CREAT|O_LARGEFILE, 0644); if (fd == -1) { - eDebug("eDVBServiceRecord - can't open hardcoded recording file!"); + eDebug("eDVBServiceRecord - can't open recording file!"); return -1; } + + /* turn off kernel caching strategies */ + posix_fadvise(fd, 0, 0, POSIX_FADV_RANDOM); + ePtr demux; - if (m_service_handler.getDemux(demux)) + if (m_service_handler.getDataDemux(demux)) { eDebug("eDVBServiceRecord - NO DEMUX available!"); return -2; @@ -95,22 +175,9 @@ int eDVBServiceRecord::doPrepare() return -3; } m_record->setTargetFD(fd); - m_pids_active.clear(); - m_state = statePrepared; - } else if ((m_state == statePrepared) || (m_state == stateRecording)) - { - /* when we're already recording, we already have a recorder allocated. */ - assert(m_record); + m_record->setTargetFilename(m_filename.c_str()); + m_target_fd = fd; } - return 0; -} - -int eDVBServiceRecord::doRecord() -{ - int err = doPrepare(); - if (err) - return err; - eDebug("starting recording.."); eDVBServicePMTHandler::program program; @@ -125,6 +192,8 @@ int eDVBServiceRecord::doRecord() if (program.pmtPid != -1) pids_to_record.insert(program.pmtPid); // PMT + int timing_pid = -1; + eDebugNoNewLine("RECORD: have %d video stream(s)", program.videoStreams.size()); if (!program.videoStreams.empty()) { @@ -134,6 +203,10 @@ int eDVBServiceRecord::doRecord() i != program.videoStreams.end(); ++i) { pids_to_record.insert(i->pid); + + if (timing_pid == -1) + timing_pid = i->pid; + if (i != program.videoStreams.begin()) eDebugNoNewLine(", "); eDebugNoNewLine("%04x", i->pid); @@ -149,16 +222,23 @@ int eDVBServiceRecord::doRecord() i != program.audioStreams.end(); ++i) { pids_to_record.insert(i->pid); + + if (timing_pid == -1) + timing_pid = i->pid; + if (i != program.audioStreams.begin()) eDebugNoNewLine(", "); eDebugNoNewLine("%04x", i->pid); } eDebugNoNewLine(")"); } - eDebug(", and the pcr pid is %04x", program.pcrPid); + eDebugNoNewLine(", and the pcr pid is %04x", program.pcrPid); if (program.pcrPid != 0x1fff) pids_to_record.insert(program.pcrPid); - + eDebug(", and the text pid is %04x", program.textPid); + if (program.textPid != -1) + pids_to_record.insert(program.textPid); // Videotext + /* find out which pids are NEW and which pids are obsolete.. */ std::set new_pids, obsolete_pids; @@ -183,6 +263,11 @@ int eDVBServiceRecord::doRecord() m_record->removePID(*i); } + if (timing_pid != -1) + m_record->setTimingPID(timing_pid); + + m_pids_active = pids_to_record; + if (m_state != stateRecording) { m_record->start(); @@ -191,3 +276,9 @@ int eDVBServiceRecord::doRecord() } return 0; } + +RESULT eDVBServiceRecord::frontendInfo(ePtr &ptr) +{ + ptr = this; + return 0; +}