1 #include <lib/service/servicedvbrecord.h>
2 #include <lib/base/eerror.h>
3 #include <lib/dvb/epgcache.h>
4 #include <lib/dvb/metaparser.h>
9 #include <netinet/in.h>
12 #error no byte order defined!
15 DEFINE_REF(eDVBServiceRecord);
17 eDVBServiceRecord::eDVBServiceRecord(const eServiceReferenceDVB &ref): m_ref(ref)
19 CONNECT(m_service_handler.serviceEvent, eDVBServiceRecord::serviceEvent);
20 CONNECT(m_event_handler.m_eit_changed, eDVBServiceRecord::gotNewEvent);
31 void eDVBServiceRecord::serviceEvent(int event)
33 eDebug("RECORD service event %d", event);
36 case eDVBServicePMTHandler::eventTuned:
41 /* start feeding EIT updates */
42 ePtr<iDVBDemux> m_demux;
43 if (!m_service_handler.getDataDemux(m_demux))
45 eServiceReferenceDVB &ref = (eServiceReferenceDVB&) m_ref;
46 int sid = ref.getParentServiceID().get();
48 sid = ref.getServiceID().get();
49 if ( ref.getParentTransportStreamID().get() &&
50 ref.getParentTransportStreamID() != ref.getTransportStreamID() )
51 m_event_handler.startOther(m_demux, sid);
53 m_event_handler.start(m_demux, sid);
56 if (m_state == stateRecording && m_want_record)
58 m_event((iRecordableService*)this, evTunedIn);
61 case eDVBServicePMTHandler::eventTuneFailed:
63 eDebug("record failed to tune");
64 m_event((iRecordableService*)this, evTuneFailed);
67 case eDVBServicePMTHandler::eventNewProgramInfo:
69 if (m_state == stateIdle)
71 else if (m_want_record) /* doRecord can be called from Prepared and Recording state */
73 m_event((iRecordableService*)this, evNewProgramInfo);
76 case eDVBServicePMTHandler::eventMisconfiguration:
77 m_error = errMisconfiguration;
78 m_event((iRecordableService*)this, evTuneFailed);
80 case eDVBServicePMTHandler::eventNoResources:
81 m_error = errNoResources;
82 m_event((iRecordableService*)this, evTuneFailed);
87 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)
89 m_filename = filename;
92 if (m_state == stateIdle)
94 int ret = doPrepare();
97 eEPGCache::getInstance()->Lock();
98 const eit_event_struct *event = 0;
99 eServiceReferenceDVB ref = m_ref.getParentServiceReference();
102 if ( eit_event_id != -1 )
104 eDebug("query epg event id %d", eit_event_id);
105 eEPGCache::getInstance()->lookupEventId(ref, eit_event_id, event);
107 if ( !event && (begTime != -1 && endTime != -1) )
109 time_t queryTime = begTime + ((endTime-begTime)/2);
111 localtime_r(&begTime, &beg);
112 localtime_r(&endTime, &end);
113 localtime_r(&queryTime, &query);
114 eDebug("query stime %d:%d:%d, etime %d:%d:%d, qtime %d:%d:%d",
115 beg.tm_hour, beg.tm_min, beg.tm_sec,
116 end.tm_hour, end.tm_min, end.tm_sec,
117 query.tm_hour, query.tm_min, query.tm_sec);
118 eEPGCache::getInstance()->lookupEventTime(ref, queryTime, event);
122 eDebug("found event.. store to disc");
123 std::string fname = filename;
124 fname.erase(fname.length()-2, 2);
126 int fd = open(fname.c_str(), O_CREAT|O_WRONLY, 0777);
129 int evLen=HILO(event->descriptors_loop_length)+12/*EIT_LOOP_SIZE*/;
130 int wr = ::write( fd, (unsigned char*)event, evLen );
132 eDebug("eit write error (%m)");
136 eEPGCache::getInstance()->Unlock();
144 meta.m_time_create = begTime;
150 meta.m_description = descr;
151 ret = meta.updateMeta(filename) ? -255 : 0;
158 RESULT eDVBServiceRecord::prepareStreaming()
162 if (m_state == stateIdle)
167 RESULT eDVBServiceRecord::start(bool simulate)
169 m_simulate = simulate;
171 /* when tune wasn't yet successfully, doRecord stays in "prepared"-state which is fine. */
172 m_event((iRecordableService*)this, evStart);
176 RESULT eDVBServiceRecord::stop()
179 eDebug("stop recording!");
180 if (m_state == stateRecording)
184 if (m_target_fd >= 0)
186 ::close(m_target_fd);
192 m_state = statePrepared;
193 } else if (!m_simulate)
194 eDebug("(was not recording)");
195 if (m_state == statePrepared)
200 m_event((iRecordableService*)this, evRecordStopped);
204 int eDVBServiceRecord::doPrepare()
206 /* allocate a ts recorder if we don't already have one. */
207 if (m_state == stateIdle)
209 m_pids_active.clear();
210 m_state = statePrepared;
211 return m_service_handler.tune(m_ref, 0, 0, m_simulate);
216 int eDVBServiceRecord::doRecord()
218 int err = doPrepare();
221 m_error = errTuneFailed;
222 m_event((iRecordableService*)this, evRecordFailed);
227 return 0; /* try it again when we are tuned in */
229 if (!m_record && m_tuned && !m_streaming && !m_simulate)
231 eDebug("Recording to %s...", m_filename.c_str());
232 ::remove(m_filename.c_str());
233 int fd = ::open(m_filename.c_str(), O_WRONLY|O_CREAT|O_LARGEFILE, 0644);
236 eDebug("eDVBServiceRecord - can't open recording file!");
237 m_error = errOpenRecordFile;
238 m_event((iRecordableService*)this, evRecordFailed);
239 return errOpenRecordFile;
242 /* turn off kernel caching strategies */
243 posix_fadvise(fd, 0, 0, POSIX_FADV_RANDOM);
245 ePtr<iDVBDemux> demux;
246 if (m_service_handler.getDataDemux(demux))
248 eDebug("eDVBServiceRecord - NO DEMUX available!");
249 m_error = errNoDemuxAvailable;
250 m_event((iRecordableService*)this, evRecordFailed);
251 return errNoDemuxAvailable;
253 demux->createTSRecorder(m_record);
256 eDebug("eDVBServiceRecord - no ts recorder available.");
257 m_error = errNoTsRecorderAvailable;
258 m_event((iRecordableService*)this, evRecordFailed);
259 return errNoTsRecorderAvailable;
261 m_record->setTargetFD(fd);
262 m_record->setTargetFilename(m_filename.c_str());
263 m_record->connectEvent(slot(*this, &eDVBServiceRecord::recordEvent), m_con_record_event);
270 m_state = stateRecording;
271 eDebug("start streaming...");
274 eDebug("start recording...");
276 eDVBServicePMTHandler::program program;
277 if (m_service_handler.getProgramInfo(program))
278 eDebug("getting program info failed.");
281 std::set<int> pids_to_record;
283 pids_to_record.insert(0); // PAT
285 if (program.pmtPid != -1)
286 pids_to_record.insert(program.pmtPid); // PMT
288 int timing_pid = -1, timing_pid_type = -1;
290 eDebugNoNewLine("RECORD: have %d video stream(s)", program.videoStreams.size());
291 if (!program.videoStreams.empty())
293 eDebugNoNewLine(" (");
294 for (std::vector<eDVBServicePMTHandler::videoStream>::const_iterator
295 i(program.videoStreams.begin());
296 i != program.videoStreams.end(); ++i)
298 pids_to_record.insert(i->pid);
300 if (timing_pid == -1)
303 timing_pid_type = i->type;
306 if (i != program.videoStreams.begin())
307 eDebugNoNewLine(", ");
308 eDebugNoNewLine("%04x", i->pid);
310 eDebugNoNewLine(")");
312 eDebugNoNewLine(", and %d audio stream(s)", program.audioStreams.size());
313 if (!program.audioStreams.empty())
315 eDebugNoNewLine(" (");
316 for (std::vector<eDVBServicePMTHandler::audioStream>::const_iterator
317 i(program.audioStreams.begin());
318 i != program.audioStreams.end(); ++i)
320 pids_to_record.insert(i->pid);
322 if (timing_pid == -1)
325 timing_pid_type = -1;
328 if (i != program.audioStreams.begin())
329 eDebugNoNewLine(", ");
330 eDebugNoNewLine("%04x", i->pid);
332 eDebugNoNewLine(")");
334 if (!program.subtitleStreams.empty())
336 eDebugNoNewLine(" (");
337 for (std::vector<eDVBServicePMTHandler::subtitleStream>::const_iterator
338 i(program.subtitleStreams.begin());
339 i != program.subtitleStreams.end(); ++i)
341 pids_to_record.insert(i->pid);
343 if (i != program.subtitleStreams.begin())
344 eDebugNoNewLine(", ");
345 eDebugNoNewLine("%04x", i->pid);
347 eDebugNoNewLine(")");
349 eDebugNoNewLine(", and the pcr pid is %04x", program.pcrPid);
350 if (program.pcrPid != 0x1fff)
351 pids_to_record.insert(program.pcrPid);
352 eDebug(", and the text pid is %04x", program.textPid);
353 if (program.textPid != -1)
354 pids_to_record.insert(program.textPid); // Videotext
356 /* find out which pids are NEW and which pids are obsolete.. */
357 std::set<int> new_pids, obsolete_pids;
359 std::set_difference(pids_to_record.begin(), pids_to_record.end(),
360 m_pids_active.begin(), m_pids_active.end(),
361 std::inserter(new_pids, new_pids.begin()));
364 m_pids_active.begin(), m_pids_active.end(),
365 pids_to_record.begin(), pids_to_record.end(),
366 std::inserter(new_pids, new_pids.begin())
369 for (std::set<int>::iterator i(new_pids.begin()); i != new_pids.end(); ++i)
371 eDebug("ADD PID: %04x", *i);
372 m_record->addPID(*i);
375 for (std::set<int>::iterator i(obsolete_pids.begin()); i != obsolete_pids.end(); ++i)
377 eDebug("REMOVED PID: %04x", *i);
378 m_record->removePID(*i);
381 if (timing_pid != -1)
382 m_record->setTimingPID(timing_pid, timing_pid_type);
384 m_pids_active = pids_to_record;
386 if (m_state != stateRecording)
389 m_state = stateRecording;
394 m_event((iRecordableService*)this, evRecordRunning);
398 RESULT eDVBServiceRecord::frontendInfo(ePtr<iFrontendInformation> &ptr)
404 RESULT eDVBServiceRecord::connectEvent(const Slot2<void,iRecordableService*,int> &event, ePtr<eConnection> &connection)
406 connection = new eConnection((iRecordableService*)this, m_event.connect(event));
410 RESULT eDVBServiceRecord::stream(ePtr<iStreamableService> &ptr)
416 extern void PutToDict(ePyObject &dict, const char*key, long val); // defined in dvb/frontend.cpp
418 PyObject *eDVBServiceRecord::getStreamingData()
420 eDVBServicePMTHandler::program program;
421 if (!m_tuned || m_service_handler.getProgramInfo(program))
426 ePyObject r = program.createPythonObject();
427 ePtr<iDVBDemux> demux;
428 if (!m_service_handler.getDataDemux(demux))
431 if (!demux->getCADemuxID(demux_id))
432 PutToDict(r, "demux", demux_id);
438 void eDVBServiceRecord::recordEvent(int event)
442 case iDVBTSRecorder::eventWriteError:
443 eWarning("[eDVBServiceRecord] record write error");
445 m_event((iRecordableService*)this, evRecordWriteError);
448 eDebug("unhandled record event %d", event);
452 void eDVBServiceRecord::gotNewEvent()
454 ePtr<eServiceEvent> event_now;
455 m_event_handler.getEvent(event_now, 0);
460 int event_id = event_now->getEventId();
466 if (m_record->getCurrentPCR(p))
467 eDebug("getting PCR failed!");
471 m_event_timestamps[/* event_id*/ ++i] = p;
472 eDebug("pcr of eit change: %llx", p);
476 if (event_id != m_last_event_id)
477 eDebug("[eDVBServiceRecord] now running: %s (%d seconds)", event_now->getEventName().c_str(), event_now->getDuration());
479 m_last_event_id = event_id;
482 void eDVBServiceRecord::saveCutlist()
484 /* XXX: dupe of eDVBServicePlay::saveCuesheet, refactor plz */
485 std::string filename = m_filename + ".cuts";
489 if (tstools.openFile(m_filename.c_str()))
491 eDebug("[eDVBServiceRecord] saving cutlist failed because tstools failed");
495 FILE *f = fopen(filename.c_str(), "wb");
499 unsigned long long where;
502 for (std::map<int,pts_t>::iterator i(m_event_timestamps.begin()); i != m_event_timestamps.end(); ++i)
505 off_t offset = 0; // fixme, we need to note down both
506 if (tstools.fixupPTS(offset, p))
508 eDebug("[eDVBServiceRecord] fixing up PTS failed, not saving");
511 eDebug("fixed up %llx to %llx (offset %llx)", i->second, p, offset);
512 #if BYTE_ORDER == BIG_ENDIAN
517 what = htonl(2); /* mark */
518 fwrite(&where, sizeof(where), 1, f);
519 fwrite(&what, sizeof(what), 1, f);