1 #include <lib/service/servicedvbrecord.h>
2 #include <lib/base/eerror.h>
3 #include <lib/dvb/epgcache.h>
7 DEFINE_REF(eDVBServiceRecord);
9 eDVBServiceRecord::eDVBServiceRecord(const eServiceReferenceDVB &ref): m_ref(ref)
11 CONNECT(m_service_handler.serviceEvent, eDVBServiceRecord::serviceEvent);
18 void eDVBServiceRecord::serviceEvent(int event)
20 eDebug("RECORD service event %d", event);
23 case eDVBServicePMTHandler::eventTuned:
27 if (m_state == stateRecording && m_want_record)
31 case eDVBServicePMTHandler::eventNewProgramInfo:
33 if (m_state == stateIdle)
35 else if (m_want_record) /* doRecord can be called from Prepared and Recording state */
42 RESULT eDVBServiceRecord::prepare(const char *filename, time_t begTime, time_t endTime, int eit_event_id)
44 m_filename = filename;
45 if (m_state == stateIdle)
47 int ret = doPrepare();
50 eEPGCache::getInstance()->Lock();
51 const eit_event_struct *event = 0;
52 eServiceReferenceDVB ref = m_ref.getParentServiceReference();
55 if ( eit_event_id != -1 )
57 eDebug("query epg event id %d", eit_event_id);
58 eEPGCache::getInstance()->lookupEventId(ref, eit_event_id, event);
60 if ( !event && (begTime != -1 && endTime != -1) )
62 time_t queryTime = begTime + ((endTime-begTime)/2);
64 localtime_r(&begTime, &beg);
65 localtime_r(&endTime, &end);
66 localtime_r(&queryTime, &query);
67 eDebug("query stime %d:%d:%d, etime %d:%d:%d, qtime %d:%d:%d",
68 beg.tm_hour, beg.tm_min, beg.tm_sec,
69 end.tm_hour, end.tm_min, end.tm_sec,
70 query.tm_hour, query.tm_min, query.tm_sec);
71 eEPGCache::getInstance()->lookupEventTime(ref, queryTime, event);
75 eDebug("found event.. store to disc");
76 std::string fname = filename;
77 fname.erase(fname.length()-2, 2);
79 int fd = open(fname.c_str(), O_CREAT|O_WRONLY, 0777);
82 int evLen=HILO(event->descriptors_loop_length)+12/*EIT_LOOP_SIZE*/;
83 int wr = ::write( fd, (unsigned char*)event, evLen );
85 eDebug("eit write error (%m)");
89 eEPGCache::getInstance()->Unlock();
97 RESULT eDVBServiceRecord::start()
100 /* when tune wasn't yet successfully, doRecord stays in "prepared"-state which is fine. */
105 RESULT eDVBServiceRecord::stop()
107 eDebug("stop recording!!");
108 if (m_state == stateRecording)
112 if (m_target_fd >= 0)
114 ::close(m_target_fd);
117 m_state = statePrepared;
120 if (m_state == statePrepared)
129 int eDVBServiceRecord::doPrepare()
131 /* allocate a ts recorder if we don't already have one. */
132 if (m_state == stateIdle)
134 m_pids_active.clear();
135 m_state = statePrepared;
136 return m_service_handler.tune(m_ref, 0);
141 int eDVBServiceRecord::doRecord()
143 int err = doPrepare();
148 return 0; /* try it again when we are tuned in */
150 if (!m_record && m_tuned)
153 eDebug("Recording to %s...", m_filename.c_str());
154 ::remove(m_filename.c_str());
155 int fd = ::open(m_filename.c_str(), O_WRONLY|O_CREAT|O_LARGEFILE, 0644);
158 eDebug("eDVBServiceRecord - can't open recording file!");
162 /* turn off kernel caching strategies */
163 posix_fadvise(fd, 0, 0, POSIX_FADV_RANDOM);
165 ePtr<iDVBDemux> demux;
166 if (m_service_handler.getDataDemux(demux))
168 eDebug("eDVBServiceRecord - NO DEMUX available!");
171 demux->createTSRecorder(m_record);
174 eDebug("eDVBServiceRecord - no ts recorder available.");
177 m_record->setTargetFD(fd);
178 m_record->setTargetFilename(m_filename.c_str());
181 eDebug("starting recording..");
183 eDVBServicePMTHandler::program program;
184 if (m_service_handler.getProgramInfo(program))
185 eDebug("getting program info failed.");
188 std::set<int> pids_to_record;
190 pids_to_record.insert(0); // PAT
192 if (program.pmtPid != -1)
193 pids_to_record.insert(program.pmtPid); // PMT
197 eDebugNoNewLine("RECORD: have %d video stream(s)", program.videoStreams.size());
198 if (!program.videoStreams.empty())
200 eDebugNoNewLine(" (");
201 for (std::vector<eDVBServicePMTHandler::videoStream>::const_iterator
202 i(program.videoStreams.begin());
203 i != program.videoStreams.end(); ++i)
205 pids_to_record.insert(i->pid);
207 if (timing_pid == -1)
210 if (i != program.videoStreams.begin())
211 eDebugNoNewLine(", ");
212 eDebugNoNewLine("%04x", i->pid);
214 eDebugNoNewLine(")");
216 eDebugNoNewLine(", and %d audio stream(s)", program.audioStreams.size());
217 if (!program.audioStreams.empty())
219 eDebugNoNewLine(" (");
220 for (std::vector<eDVBServicePMTHandler::audioStream>::const_iterator
221 i(program.audioStreams.begin());
222 i != program.audioStreams.end(); ++i)
224 pids_to_record.insert(i->pid);
226 if (timing_pid == -1)
229 if (i != program.audioStreams.begin())
230 eDebugNoNewLine(", ");
231 eDebugNoNewLine("%04x", i->pid);
233 eDebugNoNewLine(")");
235 eDebugNoNewLine(", and the pcr pid is %04x", program.pcrPid);
236 if (program.pcrPid != 0x1fff)
237 pids_to_record.insert(program.pcrPid);
238 eDebug(", and the text pid is %04x", program.textPid);
239 if (program.textPid != -1)
240 pids_to_record.insert(program.textPid); // Videotext
242 /* find out which pids are NEW and which pids are obsolete.. */
243 std::set<int> new_pids, obsolete_pids;
245 std::set_difference(pids_to_record.begin(), pids_to_record.end(),
246 m_pids_active.begin(), m_pids_active.end(),
247 std::inserter(new_pids, new_pids.begin()));
250 m_pids_active.begin(), m_pids_active.end(),
251 pids_to_record.begin(), pids_to_record.end(),
252 std::inserter(new_pids, new_pids.begin())
255 for (std::set<int>::iterator i(new_pids.begin()); i != new_pids.end(); ++i)
257 eDebug("ADD PID: %04x", *i);
258 m_record->addPID(*i);
260 for (std::set<int>::iterator i(obsolete_pids.begin()); i != obsolete_pids.end(); ++i)
262 eDebug("REMOVED PID: %04x", *i);
263 m_record->removePID(*i);
266 if (timing_pid != -1)
267 m_record->setTimingPID(timing_pid);
269 m_pids_active = pids_to_record;
271 if (m_state != stateRecording)
274 m_state = stateRecording;
280 RESULT eDVBServiceRecord::frontendInfo(ePtr<iFrontendInformation> &ptr)