1 #include <lib/service/servicedvbrecord.h>
2 #include <lib/base/eerror.h>
3 #include <lib/dvb/epgcache.h>
4 #include <lib/dvb/metaparser.h>
5 #include <lib/base/httpstream.h>
10 #include <netinet/in.h>
13 #error no byte order defined!
16 DEFINE_REF(eDVBServiceRecord);
18 eDVBServiceRecord::eDVBServiceRecord(const eServiceReferenceDVB &ref, bool isstreamclient):
19 m_ref(ref),m_is_stream_client(isstreamclient)
21 CONNECT(m_service_handler.serviceEvent, eDVBServiceRecord::serviceEvent);
22 CONNECT(m_event_handler.m_eit_changed, eDVBServiceRecord::gotNewEvent);
33 void eDVBServiceRecord::serviceEvent(int event)
35 eDebug("RECORD service event %d", event);
38 case eDVBServicePMTHandler::eventTuned:
43 /* start feeding EIT updates */
44 ePtr<iDVBDemux> m_demux;
45 if (!m_service_handler.getDataDemux(m_demux))
47 eServiceReferenceDVB &ref = (eServiceReferenceDVB&) m_ref;
48 int sid = ref.getParentServiceID().get();
50 sid = ref.getServiceID().get();
51 if ( ref.getParentTransportStreamID().get() &&
52 ref.getParentTransportStreamID() != ref.getTransportStreamID() )
53 m_event_handler.startOther(m_demux, sid);
55 m_event_handler.start(m_demux, sid);
58 if (m_state == stateRecording && m_want_record)
60 m_event((iRecordableService*)this, evTunedIn);
63 case eDVBServicePMTHandler::eventTuneFailed:
65 eDebug("record failed to tune");
66 m_event((iRecordableService*)this, evTuneFailed);
69 case eDVBServicePMTHandler::eventNewProgramInfo:
71 if (m_state == stateIdle)
73 else if (m_want_record) /* doRecord can be called from Prepared and Recording state */
75 m_event((iRecordableService*)this, evNewProgramInfo);
78 case eDVBServicePMTHandler::eventMisconfiguration:
79 m_error = errMisconfiguration;
80 m_event((iRecordableService*)this, evTuneFailed);
82 case eDVBServicePMTHandler::eventNoResources:
83 m_error = errNoResources;
84 m_event((iRecordableService*)this, evTuneFailed);
89 RESULT eDVBServiceRecord::prepare(const char *filename, time_t begTime, time_t endTime, int eit_event_id, const char *name, const char *descr, const char *tags)
91 m_filename = filename;
94 if (m_state == stateIdle)
96 int ret = doPrepare();
99 eServiceReferenceDVB ref = m_ref.getParentServiceReference();
100 ePtr<eDVBResourceManager> res_mgr;
102 std::string service_data;
105 if (!eDVBResourceManager::getInstance(res_mgr))
107 ePtr<iDVBChannelList> db;
108 if (!res_mgr->getChannelList(db))
110 ePtr<eDVBService> service;
111 if (!db->getService(ref, service))
114 sprintf(tmp, "f:%x", service->m_flags);
117 for (int x=0; x < eDVBService::cacheMax; ++x)
119 int entry = service->getCacheEntry((eDVBService::cacheID)x);
122 sprintf(tmp, ",c:%02d%04x", x, entry);
129 meta.m_time_create = begTime;
132 meta.m_service_data = service_data;
136 meta.m_description = descr;
139 ret = meta.updateMeta(filename) ? -255 : 0;
142 const eit_event_struct *event = 0;
143 eEPGCache::getInstance()->Lock();
144 if ( eit_event_id != -1 )
146 eDebug("query epg event id %d", eit_event_id);
147 eEPGCache::getInstance()->lookupEventId(ref, eit_event_id, event);
149 if ( !event && (begTime != -1 && endTime != -1) )
151 time_t queryTime = begTime + ((endTime-begTime)/2);
153 localtime_r(&begTime, &beg);
154 localtime_r(&endTime, &end);
155 localtime_r(&queryTime, &query);
156 eDebug("query stime %d:%d:%d, etime %d:%d:%d, qtime %d:%d:%d",
157 beg.tm_hour, beg.tm_min, beg.tm_sec,
158 end.tm_hour, end.tm_min, end.tm_sec,
159 query.tm_hour, query.tm_min, query.tm_sec);
160 eEPGCache::getInstance()->lookupEventTime(ref, queryTime, event);
164 eDebug("found event.. store to disc");
165 std::string fname = filename;
166 fname.erase(fname.length()-2, 2);
168 int fd = open(fname.c_str(), O_CREAT|O_WRONLY, 0777);
171 int evLen=HILO(event->descriptors_loop_length)+12/*EIT_LOOP_SIZE*/;
172 int wr = ::write( fd, (unsigned char*)event, evLen );
174 eDebug("eit write error (%m)");
178 eEPGCache::getInstance()->Unlock();
186 RESULT eDVBServiceRecord::prepareStreaming()
190 if (m_state == stateIdle)
195 RESULT eDVBServiceRecord::start(bool simulate)
197 m_simulate = simulate;
199 /* when tune wasn't yet successfully, doRecord stays in "prepared"-state which is fine. */
200 m_event((iRecordableService*)this, evStart);
204 RESULT eDVBServiceRecord::stop()
207 eDebug("stop recording!");
208 if (m_state == stateRecording)
212 if (m_target_fd >= 0)
214 ::close(m_target_fd);
220 m_state = statePrepared;
221 } else if (!m_simulate)
222 eDebug("(was not recording)");
223 if (m_state == statePrepared)
228 m_event((iRecordableService*)this, evRecordStopped);
232 int eDVBServiceRecord::doPrepare()
234 /* allocate a ts recorder if we don't already have one. */
235 if (m_state == stateIdle)
237 bool isstreamclient = false;
238 m_pids_active.clear();
239 m_state = statePrepared;
240 ePtr<iTsSource> source;
241 if (!m_ref.path.empty())
243 if (m_is_stream_client)
245 isstreamclient = true;
246 eHttpStream *f = new eHttpStream();
247 f->open(m_ref.path.c_str());
248 source = ePtr<iTsSource>(f);
252 /* re-record a recording */
253 eRawFile *f = new eRawFile();
254 f->open(m_ref.path.c_str());
255 source = ePtr<iTsSource>(f);
260 m_event((iRecordableService*)this, evTuneStart);
262 return m_service_handler.tuneExt(m_ref, 0, source, m_ref.path.c_str(), 0, m_simulate, 0, isstreamclient);
267 int eDVBServiceRecord::doRecord()
269 int err = doPrepare();
272 m_error = errTuneFailed;
273 m_event((iRecordableService*)this, evRecordFailed);
278 return 0; /* try it again when we are tuned in */
280 if (!m_record && m_tuned && !m_streaming && !m_simulate)
282 eDebug("Recording to %s...", m_filename.c_str());
283 ::remove(m_filename.c_str());
284 int fd = ::open(m_filename.c_str(), O_WRONLY|O_CREAT|O_LARGEFILE, 0644);
287 eDebug("eDVBServiceRecord - can't open recording file!");
288 m_error = errOpenRecordFile;
289 m_event((iRecordableService*)this, evRecordFailed);
290 return errOpenRecordFile;
293 /* turn off kernel caching strategies */
294 posix_fadvise(fd, 0, 0, POSIX_FADV_RANDOM);
296 ePtr<iDVBDemux> demux;
297 if (m_service_handler.getDataDemux(demux))
299 eDebug("eDVBServiceRecord - NO DEMUX available!");
300 m_error = errNoDemuxAvailable;
301 m_event((iRecordableService*)this, evRecordFailed);
302 return errNoDemuxAvailable;
304 demux->createTSRecorder(m_record);
307 eDebug("eDVBServiceRecord - no ts recorder available.");
308 m_error = errNoTsRecorderAvailable;
309 m_event((iRecordableService*)this, evRecordFailed);
310 return errNoTsRecorderAvailable;
312 m_record->setTargetFD(fd);
313 m_record->setTargetFilename(m_filename.c_str());
314 m_record->connectEvent(slot(*this, &eDVBServiceRecord::recordEvent), m_con_record_event);
321 m_state = stateRecording;
322 eDebug("start streaming...");
325 eDebug("start recording...");
327 eDVBServicePMTHandler::program program;
328 if (m_service_handler.getProgramInfo(program))
329 eDebug("getting program info failed.");
332 std::set<int> pids_to_record;
334 pids_to_record.insert(0); // PAT
336 if (program.pmtPid != -1)
337 pids_to_record.insert(program.pmtPid); // PMT
339 int timing_pid = -1, timing_pid_type = -1;
341 eDebugNoNewLine("RECORD: have %zd video stream(s)", program.videoStreams.size());
342 if (!program.videoStreams.empty())
344 eDebugNoNewLine(" (");
345 for (std::vector<eDVBServicePMTHandler::videoStream>::const_iterator
346 i(program.videoStreams.begin());
347 i != program.videoStreams.end(); ++i)
349 pids_to_record.insert(i->pid);
351 if (timing_pid == -1)
354 timing_pid_type = i->type;
357 if (i != program.videoStreams.begin())
358 eDebugNoNewLine(", ");
359 eDebugNoNewLine("%04x", i->pid);
361 eDebugNoNewLine(")");
363 eDebugNoNewLine(", and %zd audio stream(s)", program.audioStreams.size());
364 if (!program.audioStreams.empty())
366 eDebugNoNewLine(" (");
367 for (std::vector<eDVBServicePMTHandler::audioStream>::const_iterator
368 i(program.audioStreams.begin());
369 i != program.audioStreams.end(); ++i)
371 pids_to_record.insert(i->pid);
373 if (timing_pid == -1)
376 timing_pid_type = -1;
379 if (i != program.audioStreams.begin())
380 eDebugNoNewLine(", ");
381 eDebugNoNewLine("%04x", i->pid);
383 eDebugNoNewLine(")");
385 if (!program.subtitleStreams.empty())
387 eDebugNoNewLine(" (");
388 for (std::vector<eDVBServicePMTHandler::subtitleStream>::const_iterator
389 i(program.subtitleStreams.begin());
390 i != program.subtitleStreams.end(); ++i)
392 pids_to_record.insert(i->pid);
394 if (i != program.subtitleStreams.begin())
395 eDebugNoNewLine(", ");
396 eDebugNoNewLine("%04x", i->pid);
398 eDebugNoNewLine(")");
400 eDebugNoNewLine(", and the pcr pid is %04x", program.pcrPid);
401 if (program.pcrPid != 0x1fff)
402 pids_to_record.insert(program.pcrPid);
403 eDebug(", and the text pid is %04x", program.textPid);
404 if (program.textPid != -1)
405 pids_to_record.insert(program.textPid); // Videotext
407 /* find out which pids are NEW and which pids are obsolete.. */
408 std::set<int> new_pids, obsolete_pids;
410 std::set_difference(pids_to_record.begin(), pids_to_record.end(),
411 m_pids_active.begin(), m_pids_active.end(),
412 std::inserter(new_pids, new_pids.begin()));
415 m_pids_active.begin(), m_pids_active.end(),
416 pids_to_record.begin(), pids_to_record.end(),
417 std::inserter(obsolete_pids, obsolete_pids.begin())
420 for (std::set<int>::iterator i(new_pids.begin()); i != new_pids.end(); ++i)
422 eDebug("ADD PID: %04x", *i);
423 m_record->addPID(*i);
426 for (std::set<int>::iterator i(obsolete_pids.begin()); i != obsolete_pids.end(); ++i)
428 eDebug("REMOVED PID: %04x", *i);
429 m_record->removePID(*i);
432 if (timing_pid != -1)
433 m_record->setTimingPID(timing_pid, timing_pid_type);
435 m_pids_active = pids_to_record;
437 if (m_state != stateRecording)
440 m_state = stateRecording;
445 m_event((iRecordableService*)this, evRecordRunning);
449 RESULT eDVBServiceRecord::frontendInfo(ePtr<iFrontendInformation> &ptr)
455 RESULT eDVBServiceRecord::connectEvent(const Slot2<void,iRecordableService*,int> &event, ePtr<eConnection> &connection)
457 connection = new eConnection((iRecordableService*)this, m_event.connect(event));
461 RESULT eDVBServiceRecord::stream(ePtr<iStreamableService> &ptr)
467 extern void PutToDict(ePyObject &dict, const char*key, long val); // defined in dvb/frontend.cpp
469 PyObject *eDVBServiceRecord::getStreamingData()
471 eDVBServicePMTHandler::program program;
472 if (!m_tuned || m_service_handler.getProgramInfo(program))
477 ePyObject r = program.createPythonObject();
478 ePtr<iDVBDemux> demux;
479 if (!m_service_handler.getDataDemux(demux))
482 if (!demux->getCADemuxID(demux_id))
483 PutToDict(r, "demux", demux_id);
489 void eDVBServiceRecord::recordEvent(int event)
493 case iDVBTSRecorder::eventWriteError:
494 eWarning("[eDVBServiceRecord] record write error");
496 m_event((iRecordableService*)this, evRecordWriteError);
499 eDebug("unhandled record event %d", event);
503 void eDVBServiceRecord::gotNewEvent()
505 ePtr<eServiceEvent> event_now;
506 m_event_handler.getEvent(event_now, 0);
511 int event_id = event_now->getEventId();
517 if (m_record->getCurrentPCR(p))
518 eDebug("getting PCR failed!");
521 m_event_timestamps[event_id] = p;
522 eDebug("pcr of eit change: %llx", p);
526 if (event_id != m_last_event_id)
527 eDebug("[eDVBServiceRecord] now running: %s (%d seconds)", event_now->getEventName().c_str(), event_now->getDuration());
529 m_last_event_id = event_id;
531 m_event((iRecordableService*)this, evNewEventInfo);
534 void eDVBServiceRecord::saveCutlist()
536 /* XXX: dupe of eDVBServicePlay::saveCuesheet, refactor plz */
537 std::string filename = m_filename + ".cuts";
541 if (tstools.openFile(m_filename.c_str()))
543 eDebug("[eDVBServiceRecord] saving cutlist failed because tstools failed");
547 FILE *f = fopen(filename.c_str(), "wb");
551 unsigned long long where;
554 for (std::map<int,pts_t>::iterator i(m_event_timestamps.begin()); i != m_event_timestamps.end(); ++i)
557 off_t offset = 0; // fixme, we need to note down both
558 if (tstools.fixupPTS(offset, p))
560 eDebug("[eDVBServiceRecord] fixing up PTS failed, not saving");
563 eDebug("fixed up %llx to %llx (offset %llx)", i->second, p, offset);
564 #if BYTE_ORDER == BIG_ENDIAN
569 what = htonl(2); /* mark */
570 fwrite(&where, sizeof(where), 1, f);
571 fwrite(&what, sizeof(what), 1, f);
578 RESULT eDVBServiceRecord::subServices(ePtr<iSubserviceList> &ptr)
584 int eDVBServiceRecord::getNumberOfSubservices()
586 ePtr<eServiceEvent> evt;
587 if (!m_event_handler.getEvent(evt, 0))
588 return evt->getNumOfLinkageServices();
592 RESULT eDVBServiceRecord::getSubservice(eServiceReference &sub, unsigned int n)
594 ePtr<eServiceEvent> evt;
595 if (!m_event_handler.getEvent(evt, 0))
597 if (!evt->getLinkageService(sub, m_ref, n))
600 sub.type=eServiceReference::idInvalid;