1 #include <lib/service/servicedvbrecord.h>
2 #include <lib/base/eerror.h>
6 DEFINE_REF(eDVBServiceRecord);
8 eDVBServiceRecord::eDVBServiceRecord(const eServiceReferenceDVB &ref): m_ref(ref), m_service_handler(1)
10 CONNECT(m_service_handler.serviceEvent, eDVBServiceRecord::serviceEvent);
15 void eDVBServiceRecord::serviceEvent(int event)
17 eDebug("RECORD service event %d", event);
20 case eDVBServicePMTHandler::eventTuned:
25 case eDVBServicePMTHandler::eventNewProgramInfo:
27 if (m_state == stateIdle)
29 else if (m_want_record) /* doRecord can be called from Prepared and Recording state */
37 RESULT eDVBServiceRecord::prepare(const char *filename)
39 m_filename = filename;
40 if (m_state == stateIdle)
41 return m_service_handler.tune(m_ref);
46 RESULT eDVBServiceRecord::start()
49 /* when tune wasn't yet successfully, doRecord stays in "prepared"-state which is fine. */
54 RESULT eDVBServiceRecord::stop()
56 eDebug("stop recording!!");
57 if (m_state == stateRecording)
60 m_state = statePrepared;
63 if (m_state == statePrepared)
72 int eDVBServiceRecord::doPrepare()
74 /* allocate a ts recorder if we don't already have one. */
75 if (m_state == stateIdle)
77 eDebug("Recording to %s...", m_filename.c_str());
78 ::remove(m_filename.c_str());
79 int fd = ::open(m_filename.c_str(), O_WRONLY|O_CREAT, 0644);
82 eDebug("eDVBServiceRecord - can't open hardcoded recording file!");
85 ePtr<iDVBDemux> demux;
86 if (m_service_handler.getDemux(demux))
88 eDebug("eDVBServiceRecord - NO DEMUX available!");
91 demux->createTSRecorder(m_record);
94 eDebug("eDVBServiceRecord - no ts recorder available.");
97 m_record->setTargetFD(fd);
98 m_pids_active.clear();
99 m_state = statePrepared;
100 } else if ((m_state == statePrepared) || (m_state == stateRecording))
102 /* when we're already recording, we already have a recorder allocated. */
108 int eDVBServiceRecord::doRecord()
110 int err = doPrepare();
114 eDebug("starting recording..");
116 eDVBServicePMTHandler::program program;
117 if (m_service_handler.getProgramInfo(program))
118 eDebug("getting program info failed.");
121 std::set<int> pids_to_record;
123 pids_to_record.insert(0); // PAT
125 if (program.pmtPid != -1)
126 pids_to_record.insert(program.pmtPid); // PMT
128 eDebugNoNewLine("RECORD: have %d video stream(s)", program.videoStreams.size());
129 if (!program.videoStreams.empty())
131 eDebugNoNewLine(" (");
132 for (std::vector<eDVBServicePMTHandler::videoStream>::const_iterator
133 i(program.videoStreams.begin());
134 i != program.videoStreams.end(); ++i)
136 pids_to_record.insert(i->pid);
137 if (i != program.videoStreams.begin())
138 eDebugNoNewLine(", ");
139 eDebugNoNewLine("%04x", i->pid);
141 eDebugNoNewLine(")");
143 eDebugNoNewLine(", and %d audio stream(s)", program.audioStreams.size());
144 if (!program.audioStreams.empty())
146 eDebugNoNewLine(" (");
147 for (std::vector<eDVBServicePMTHandler::audioStream>::const_iterator
148 i(program.audioStreams.begin());
149 i != program.audioStreams.end(); ++i)
151 pids_to_record.insert(i->pid);
152 if (i != program.audioStreams.begin())
153 eDebugNoNewLine(", ");
154 eDebugNoNewLine("%04x", i->pid);
156 eDebugNoNewLine(")");
158 eDebug(", and the pcr pid is %04x", program.pcrPid);
159 if (program.pcrPid != 0x1fff)
160 pids_to_record.insert(program.pcrPid);
162 /* find out which pids are NEW and which pids are obsolete.. */
163 std::set<int> new_pids, obsolete_pids;
165 std::set_difference(pids_to_record.begin(), pids_to_record.end(),
166 m_pids_active.begin(), m_pids_active.end(),
167 std::inserter(new_pids, new_pids.begin()));
170 m_pids_active.begin(), m_pids_active.end(),
171 pids_to_record.begin(), pids_to_record.end(),
172 std::inserter(new_pids, new_pids.begin())
175 for (std::set<int>::iterator i(new_pids.begin()); i != new_pids.end(); ++i)
177 eDebug("ADD PID: %04x", *i);
178 m_record->addPID(*i);
180 for (std::set<int>::iterator i(obsolete_pids.begin()); i != obsolete_pids.end(); ++i)
182 eDebug("REMOVED PID: %04x", *i);
183 m_record->removePID(*i);
186 if (m_state != stateRecording)
189 m_state = stateRecording;