使用live555直播来自v4l2的摄像头图像

结合前面的 采集 v4l2 视频, 使用 live555, 通过 rtsp 发布实时流. capture.h, capture.cpp, vcompress.h, vcompress.cpp 需要参考前面几片文章. 这里仅仅贴出 v4l2_x264_service.cpp [cpp] view plaincopy #includestdio.h #includestdlib.h #includeunistd

结合前面的 采集 v4l2 视频, 使用 live555, 通过 rtsp 发布实时流. capture.h, capture.cpp, vcompress.h, vcompress.cpp 需要参考前面几片文章. 这里仅仅贴出 v4l2_x264_service.cpp

[cpp] view plaincopy

1. #include
2. #include
3. #include
4. #include
6. #include
7. #include
8. #include
10. #include
11. #include
13. #include "capture.h"
14. #include "vcompress.h"
16. static UsageEnvironment *_env = 0;
18. #define SINK_PORT 3030
20. #define VIDEO_WIDTH 320
21. #define VIDEO_HEIGHT 240
22. #define FRAME_PER_SEC 5.0
24. pid_t gettid()
25. {
26. return syscall(SYS_gettid);
27. }
30. // 使用 webcam + x264
31. class WebcamFrameSource : public FramedSource
32. {
33. void *mp_capture, *mp_compress; // v4l2 + x264 encoder
34. int m_started;
35. void *mp_token;
37. public:
38. WebcamFrameSource (UsageEnvironment &env)
39. : FramedSource(env)
40. {
41. fprintf(stderr, "[%d] %s .... calling\n", gettid(), __func__);
42. mp_capture = capture_open("/dev/video0", VIDEO_WIDTH, VIDEO_HEIGHT, PIX_FMT_YUV420P);
43. if (!mp_capture) {
44. fprintf(stderr, "%s: open /dev/video0 err\n", __func__);
45. exit(-1);
46. }
48. mp_compress = vc_open(VIDEO_WIDTH, VIDEO_HEIGHT, FRAME_PER_SEC);
49. if (!mp_compress) {
50. fprintf(stderr, "%s: open x264 err\n", __func__);
51. exit(-1);
52. }
54. m_started = 0;
55. mp_token = 0;
56. }
58. ~WebcamFrameSource ()
59. {
60. fprintf(stderr, "[%d] %s .... calling\n", gettid(), __func__);
62. if (m_started) {
63. envir().taskScheduler().unscheduleDelayedTask(mp_token);
64. }
66. if (mp_compress)
67. vc_close(mp_compress);
68. if (mp_capture)
69. capture_close(mp_capture);
70. }
72. protected:
73. virtual void doGetNextFrame ()
74. {
75. if (m_started) return;
76. m_started = 1;
78. // 根据 fps, 计算等待时间
79. double delay = 1000.0 / FRAME_PER_SEC;
80. int to_delay = delay * 1000; // us
82. mp_token = envir().taskScheduler().scheduleDelayedTask(to_delay,
83. getNextFrame, this);
84. }

[cpp] view plaincopy

1. virtual unsigned maxFrameSize() const // 这个很重要, 如果不设置, 可能导致 getNextFrame() 出现 fMaxSize 小于实际编码帧的情况, 导致图像不完整

[cpp] view plaincopy

1. { return 100*1024; }

[cpp] view plaincopy

1. private:
2. static void getNextFrame (void *ptr)
3. {
4. ((WebcamFrameSource*)ptr)->getNextFrame1();
5. }
7. void getNextFrame1 ()
8. {
9. // capture:
10. Picture pic;
11. if (capture_get_picture(mp_capture, &pic) < 0) {
12. fprintf(stderr, "==== %s: capture_get_picture err\n", __func__);
13. m_started = 0;
14. return;
15. }
17. // compress
18. const void *outbuf;
19. int outlen;
20. if (vc_compress(mp_compress, pic.data, pic.stride, &outbuf, &outlen) < 0) {
21. fprintf(stderr, "==== %s: vc_compress err\n", __func__);
22. m_started = 0;
23. return;
24. }
26. int64_t pts, dts;
27. int key;
28. vc_get_last_frame_info(mp_compress, &key, &pts, &dts);
30. // save outbuf
31. gettimeofday(&fPresentationTime, 0);
32. fFrameSize = outlen;
33. if (fFrameSize > fMaxSize) {
34. fNumTruncatedBytes = fFrameSize - fMaxSize;
35. fFrameSize = fMaxSize;
36. }
37. else {
38. fNumTruncatedBytes = 0;
39. }
41. memmove(fTo, outbuf, fFrameSize);
43. // notify
44. afterGetting(this);
46. m_started = 0;
47. }
48. };
50. class WebcamOndemandMediaSubsession : public OnDemandServerMediaSubsession
51. {
52. public:
53. static WebcamOndemandMediaSubsession *createNew (UsageEnvironment &env, FramedSource *source)
54. {
55. return new WebcamOndemandMediaSubsession(env, source);
56. }
58. protected:
59. WebcamOndemandMediaSubsession (UsageEnvironment &env, FramedSource *source)
60. : OnDemandServerMediaSubsession(env, True) // reuse the first source
61. {
62. fprintf(stderr, "[%d] %s .... calling\n", gettid(), __func__);
63. mp_source = source;
64. mp_sdp_line = 0;
65. }
67. ~WebcamOndemandMediaSubsession ()
68. {
69. fprintf(stderr, "[%d] %s .... calling\n", gettid(), __func__);
70. if (mp_sdp_line) free(mp_sdp_line);
71. }
73. private:
74. static void afterPlayingDummy (void *ptr)
75. {
76. fprintf(stderr, "[%d] %s .... calling\n", gettid(), __func__);
77. // ok
78. WebcamOndemandMediaSubsession *This = (WebcamOndemandMediaSubsession*)ptr;
79. This->m_done = 0xff;
80. }
82. static void chkForAuxSDPLine (void *ptr)
83. {
84. WebcamOndemandMediaSubsession *This = (WebcamOndemandMediaSubsession *)ptr;
85. This->chkForAuxSDPLine1();
86. }
88. void chkForAuxSDPLine1 ()
89. {
90. fprintf(stderr, "[%d] %s .... calling\n", gettid(), __func__);
91. if (mp_dummy_rtpsink->auxSDPLine())
92. m_done = 0xff;
93. else {
94. int delay = 100*1000; // 100ms
95. nextTask() = envir().taskScheduler().scheduleDelayedTask(delay,
96. chkForAuxSDPLine, this);
97. }
98. }
100. protected:
101. virtual const char *getAuxSDPLine (RTPSink *sink, FramedSource *source)
102. {
103. fprintf(stderr, "[%d] %s .... calling\n", gettid(), __func__);
104. if (mp_sdp_line) return mp_sdp_line;
106. mp_dummy_rtpsink = sink;
107. mp_dummy_rtpsink->startPlaying(*source, 0, 0);
108. //mp_dummy_rtpsink->startPlaying(*source, afterPlayingDummy, this);
109. chkForAuxSDPLine(this);
110. m_done = 0;
111. envir().taskScheduler().doEventLoop(&m_done);
112. mp_sdp_line = strdup(mp_dummy_rtpsink->auxSDPLine());
113. mp_dummy_rtpsink->stopPlaying();
115. return mp_sdp_line;
116. }
118. virtual RTPSink *createNewRTPSink(Groupsock *rtpsock, unsigned char type, FramedSource *source)
119. {
120. fprintf(stderr, "[%d] %s .... calling\n", gettid(), __func__);
121. return H264VideoRTPSink::createNew(envir(), rtpsock, type);
122. }
124. virtual FramedSource *createNewStreamSource (unsigned sid, unsigned &bitrate)
125. {
126. fprintf(stderr, "[%d] %s .... calling\n", gettid(), __func__);
127. bitrate = 500;
128. return H264VideoStreamFramer::createNew(envir(), new WebcamFrameSource(envir()));
129. }
131. private:
132. FramedSource *mp_source; // 对应 WebcamFrameSource
133. char *mp_sdp_line;
134. RTPSink *mp_dummy_rtpsink;
135. char m_done;
136. };
138. static void test_task (void *ptr)
139. {
140. fprintf(stderr, "test: task ....\n");
141. _env->taskScheduler().scheduleDelayedTask(100000, test_task, 0);
142. }
144. static void test (UsageEnvironment &env)
145. {
146. fprintf(stderr, "test: begin...\n");
148. char done = 0;
149. int delay = 100 * 1000;
150. env.taskScheduler().scheduleDelayedTask(delay, test_task, 0);
151. env.taskScheduler().doEventLoop(&done);
153. fprintf(stderr, "test: end..\n");
154. }
156. int main (int argc, char **argv)
157. {
158. // env
159. TaskScheduler *scheduler = BasicTaskScheduler::createNew();
160. _env = BasicUsageEnvironment::createNew(*scheduler);
162. // test
163. //test(*_env);
165. // rtsp server
166. RTSPServer *rtspServer = RTSPServer::createNew(*_env, 8554);
167. if (!rtspServer) {
168. fprintf(stderr, "ERR: create RTSPServer err\n");
169. ::exit(-1);
170. }
172. // add live stream
173. do {
174. WebcamFrameSource *webcam_source = 0;
176. ServerMediaSession *sms = ServerMediaSession::createNew(*_env, "webcam", 0, "Session from /dev/video0");
177. sms->addSubsession(WebcamOndemandMediaSubsession::createNew(*_env, webcam_source));
178. rtspServer->addServerMediaSession(sms);
180. char *url = rtspServer->rtspURL(sms);
181. *_env << "using url \"" << url << "\"\n";
182. delete [] url;
183. } while (0);
185. // run loop
186. _env->taskScheduler().doEventLoop();
188. return 1;
189. }

需要 live555 + libavcodec + libswscale + libx264, client 使用 vlc, mplayer, quicktime, .....