You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

886 lines
20KB

  1. /*
  2. * RAW muxer and demuxer
  3. * Copyright (c) 2001 Fabrice Bellard.
  4. * Copyright (c) 2005 Alex Beregszaszi
  5. *
  6. * This file is part of FFmpeg.
  7. *
  8. * FFmpeg is free software; you can redistribute it and/or
  9. * modify it under the terms of the GNU Lesser General Public
  10. * License as published by the Free Software Foundation; either
  11. * version 2.1 of the License, or (at your option) any later version.
  12. *
  13. * FFmpeg is distributed in the hope that it will be useful,
  14. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16. * Lesser General Public License for more details.
  17. *
  18. * You should have received a copy of the GNU Lesser General Public
  19. * License along with FFmpeg; if not, write to the Free Software
  20. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  21. */
  22. #include "avformat.h"
  23. #ifdef CONFIG_MUXERS
  24. /* simple formats */
  25. static int raw_write_header(struct AVFormatContext *s)
  26. {
  27. return 0;
  28. }
  29. static int flac_write_header(struct AVFormatContext *s)
  30. {
  31. static const uint8_t header[8] = {
  32. 0x66, 0x4C, 0x61, 0x43, 0x80, 0x00, 0x00, 0x22
  33. };
  34. uint8_t *streaminfo = s->streams[0]->codec->extradata;
  35. int len = s->streams[0]->codec->extradata_size;
  36. if(streaminfo != NULL && len > 0) {
  37. put_buffer(&s->pb, header, 8);
  38. put_buffer(&s->pb, streaminfo, len);
  39. }
  40. return 0;
  41. }
  42. static int raw_write_packet(struct AVFormatContext *s, AVPacket *pkt)
  43. {
  44. put_buffer(&s->pb, pkt->data, pkt->size);
  45. put_flush_packet(&s->pb);
  46. return 0;
  47. }
  48. static int raw_write_trailer(struct AVFormatContext *s)
  49. {
  50. return 0;
  51. }
  52. #endif //CONFIG_MUXERS
  53. /* raw input */
  54. static int raw_read_header(AVFormatContext *s, AVFormatParameters *ap)
  55. {
  56. AVStream *st;
  57. int id;
  58. st = av_new_stream(s, 0);
  59. if (!st)
  60. return AVERROR_NOMEM;
  61. id = s->iformat->value;
  62. if (id == CODEC_ID_RAWVIDEO) {
  63. st->codec->codec_type = CODEC_TYPE_VIDEO;
  64. } else {
  65. st->codec->codec_type = CODEC_TYPE_AUDIO;
  66. }
  67. st->codec->codec_id = id;
  68. switch(st->codec->codec_type) {
  69. case CODEC_TYPE_AUDIO:
  70. st->codec->sample_rate = ap->sample_rate;
  71. st->codec->channels = ap->channels;
  72. av_set_pts_info(st, 64, 1, st->codec->sample_rate);
  73. break;
  74. case CODEC_TYPE_VIDEO:
  75. av_set_pts_info(st, 64, ap->time_base.num, ap->time_base.den);
  76. st->codec->width = ap->width;
  77. st->codec->height = ap->height;
  78. st->codec->pix_fmt = ap->pix_fmt;
  79. if(st->codec->pix_fmt == PIX_FMT_NONE)
  80. st->codec->pix_fmt= PIX_FMT_YUV420P;
  81. break;
  82. default:
  83. return -1;
  84. }
  85. return 0;
  86. }
  87. #define RAW_PACKET_SIZE 1024
  88. static int raw_read_packet(AVFormatContext *s, AVPacket *pkt)
  89. {
  90. int ret, size;
  91. // AVStream *st = s->streams[0];
  92. size= RAW_PACKET_SIZE;
  93. ret= av_get_packet(&s->pb, pkt, size);
  94. pkt->stream_index = 0;
  95. if (ret <= 0) {
  96. return AVERROR_IO;
  97. }
  98. /* note: we need to modify the packet size here to handle the last
  99. packet */
  100. pkt->size = ret;
  101. return ret;
  102. }
  103. static int raw_read_partial_packet(AVFormatContext *s, AVPacket *pkt)
  104. {
  105. int ret, size;
  106. size = RAW_PACKET_SIZE;
  107. if (av_new_packet(pkt, size) < 0)
  108. return AVERROR_IO;
  109. pkt->pos= url_ftell(&s->pb);
  110. pkt->stream_index = 0;
  111. ret = get_partial_buffer(&s->pb, pkt->data, size);
  112. if (ret <= 0) {
  113. av_free_packet(pkt);
  114. return AVERROR_IO;
  115. }
  116. pkt->size = ret;
  117. return ret;
  118. }
  119. // http://www.artificis.hu/files/texts/ingenient.txt
  120. static int ingenient_read_packet(AVFormatContext *s, AVPacket *pkt)
  121. {
  122. int ret, size, w, h, unk1, unk2;
  123. if (get_le32(&s->pb) != MKTAG('M', 'J', 'P', 'G'))
  124. return AVERROR_IO; // FIXME
  125. size = get_le32(&s->pb);
  126. w = get_le16(&s->pb);
  127. h = get_le16(&s->pb);
  128. url_fskip(&s->pb, 8); // zero + size (padded?)
  129. url_fskip(&s->pb, 2);
  130. unk1 = get_le16(&s->pb);
  131. unk2 = get_le16(&s->pb);
  132. url_fskip(&s->pb, 22); // ascii timestamp
  133. av_log(NULL, AV_LOG_DEBUG, "Ingenient packet: size=%d, width=%d, height=%d, unk1=%d unk2=%d\n",
  134. size, w, h, unk1, unk2);
  135. if (av_new_packet(pkt, size) < 0)
  136. return AVERROR_IO;
  137. pkt->pos = url_ftell(&s->pb);
  138. pkt->stream_index = 0;
  139. ret = get_buffer(&s->pb, pkt->data, size);
  140. if (ret <= 0) {
  141. av_free_packet(pkt);
  142. return AVERROR_IO;
  143. }
  144. pkt->size = ret;
  145. return ret;
  146. }
  147. static int raw_read_close(AVFormatContext *s)
  148. {
  149. return 0;
  150. }
  151. int pcm_read_seek(AVFormatContext *s,
  152. int stream_index, int64_t timestamp, int flags)
  153. {
  154. AVStream *st;
  155. int block_align, byte_rate;
  156. int64_t pos;
  157. st = s->streams[0];
  158. block_align = st->codec->block_align ? st->codec->block_align :
  159. (av_get_bits_per_sample(st->codec->codec_id) * st->codec->channels) >> 3;
  160. byte_rate = st->codec->bit_rate ? st->codec->bit_rate >> 3 :
  161. block_align * st->codec->sample_rate;
  162. if (block_align <= 0 || byte_rate <= 0)
  163. return -1;
  164. /* compute the position by aligning it to block_align */
  165. pos = av_rescale_rnd(timestamp * byte_rate,
  166. st->time_base.num,
  167. st->time_base.den * (int64_t)block_align,
  168. (flags & AVSEEK_FLAG_BACKWARD) ? AV_ROUND_DOWN : AV_ROUND_UP);
  169. pos *= block_align;
  170. /* recompute exact position */
  171. st->cur_dts = av_rescale(pos, st->time_base.den, byte_rate * (int64_t)st->time_base.num);
  172. url_fseek(&s->pb, pos + s->data_offset, SEEK_SET);
  173. return 0;
  174. }
  175. /* ac3 read */
  176. static int ac3_read_header(AVFormatContext *s,
  177. AVFormatParameters *ap)
  178. {
  179. AVStream *st;
  180. st = av_new_stream(s, 0);
  181. if (!st)
  182. return AVERROR_NOMEM;
  183. st->codec->codec_type = CODEC_TYPE_AUDIO;
  184. st->codec->codec_id = CODEC_ID_AC3;
  185. st->need_parsing = 1;
  186. /* the parameters will be extracted from the compressed bitstream */
  187. return 0;
  188. }
  189. static int shorten_read_header(AVFormatContext *s,
  190. AVFormatParameters *ap)
  191. {
  192. AVStream *st;
  193. st = av_new_stream(s, 0);
  194. if (!st)
  195. return AVERROR_NOMEM;
  196. st->codec->codec_type = CODEC_TYPE_AUDIO;
  197. st->codec->codec_id = CODEC_ID_SHORTEN;
  198. st->need_parsing = 1;
  199. /* the parameters will be extracted from the compressed bitstream */
  200. return 0;
  201. }
  202. /* flac read */
  203. static int flac_read_header(AVFormatContext *s,
  204. AVFormatParameters *ap)
  205. {
  206. AVStream *st;
  207. st = av_new_stream(s, 0);
  208. if (!st)
  209. return AVERROR_NOMEM;
  210. st->codec->codec_type = CODEC_TYPE_AUDIO;
  211. st->codec->codec_id = CODEC_ID_FLAC;
  212. st->need_parsing = 1;
  213. /* the parameters will be extracted from the compressed bitstream */
  214. return 0;
  215. }
  216. /* dts read */
  217. static int dts_read_header(AVFormatContext *s,
  218. AVFormatParameters *ap)
  219. {
  220. AVStream *st;
  221. st = av_new_stream(s, 0);
  222. if (!st)
  223. return AVERROR_NOMEM;
  224. st->codec->codec_type = CODEC_TYPE_AUDIO;
  225. st->codec->codec_id = CODEC_ID_DTS;
  226. st->need_parsing = 1;
  227. /* the parameters will be extracted from the compressed bitstream */
  228. return 0;
  229. }
  230. /* aac read */
  231. static int aac_read_header(AVFormatContext *s,
  232. AVFormatParameters *ap)
  233. {
  234. AVStream *st;
  235. st = av_new_stream(s, 0);
  236. if (!st)
  237. return AVERROR_NOMEM;
  238. st->codec->codec_type = CODEC_TYPE_AUDIO;
  239. st->codec->codec_id = CODEC_ID_AAC;
  240. st->need_parsing = 1;
  241. /* the parameters will be extracted from the compressed bitstream */
  242. return 0;
  243. }
  244. /* mpeg1/h263 input */
  245. static int video_read_header(AVFormatContext *s,
  246. AVFormatParameters *ap)
  247. {
  248. AVStream *st;
  249. st = av_new_stream(s, 0);
  250. if (!st)
  251. return AVERROR_NOMEM;
  252. st->codec->codec_type = CODEC_TYPE_VIDEO;
  253. st->codec->codec_id = s->iformat->value;
  254. st->need_parsing = 1;
  255. /* for mjpeg, specify frame rate */
  256. /* for mpeg4 specify it too (most mpeg4 streams dont have the fixed_vop_rate set ...)*/
  257. if (ap->time_base.num) {
  258. av_set_pts_info(st, 64, ap->time_base.num, ap->time_base.den);
  259. } else if ( st->codec->codec_id == CODEC_ID_MJPEG ||
  260. st->codec->codec_id == CODEC_ID_MPEG4 ||
  261. st->codec->codec_id == CODEC_ID_H264) {
  262. av_set_pts_info(st, 64, 1, 25);
  263. }
  264. return 0;
  265. }
  266. #define SEQ_START_CODE 0x000001b3
  267. #define GOP_START_CODE 0x000001b8
  268. #define PICTURE_START_CODE 0x00000100
  269. #define SLICE_START_CODE 0x00000101
  270. #define PACK_START_CODE 0x000001ba
  271. #define VIDEO_ID 0x000001e0
  272. #define AUDIO_ID 0x000001c0
  273. static int mpegvideo_probe(AVProbeData *p)
  274. {
  275. uint32_t code= -1;
  276. int pic=0, seq=0, slice=0, pspack=0, pes=0;
  277. int i;
  278. for(i=0; i<p->buf_size; i++){
  279. code = (code<<8) + p->buf[i];
  280. if ((code & 0xffffff00) == 0x100) {
  281. switch(code){
  282. case SEQ_START_CODE: seq++; break;
  283. case PICTURE_START_CODE: pic++; break;
  284. case SLICE_START_CODE: slice++; break;
  285. case PACK_START_CODE: pspack++; break;
  286. case VIDEO_ID:
  287. case AUDIO_ID: pes++; break;
  288. }
  289. }
  290. }
  291. if(seq && seq*9<=pic*10 && pic*9<=slice*10 && !pspack && !pes)
  292. return AVPROBE_SCORE_MAX/2+1; // +1 for .mpg
  293. return 0;
  294. }
  295. #define VIDEO_OBJECT_START_CODE 0x00000100
  296. #define VIDEO_OBJECT_LAYER_START_CODE 0x00000120
  297. #define VISUAL_OBJECT_START_CODE 0x000001b5
  298. #define VOP_START_CODE 0x000001b6
  299. static int mpeg4video_probe(AVProbeData *probe_packet)
  300. {
  301. uint32_t temp_buffer= -1;
  302. int VO=0, VOL=0, VOP = 0, VISO = 0;
  303. int i;
  304. for(i=0; i<probe_packet->buf_size; i++){
  305. temp_buffer = (temp_buffer<<8) + probe_packet->buf[i];
  306. if ((temp_buffer & 0xffffff00) == 0x100) {
  307. switch(temp_buffer){
  308. case VOP_START_CODE: VOP++; break;
  309. case VISUAL_OBJECT_START_CODE: VISO++; break;
  310. }
  311. switch(temp_buffer & 0xfffffff0){
  312. case VIDEO_OBJECT_START_CODE: VO++; break;
  313. case VIDEO_OBJECT_LAYER_START_CODE: VOL++; break;
  314. }
  315. }
  316. }
  317. if ( VOP >= VISO && VOP >= VOL && VO >= VOL && VOL > 0)
  318. return AVPROBE_SCORE_MAX/2;
  319. return 0;
  320. }
  321. static int h263_probe(AVProbeData *p)
  322. {
  323. int code;
  324. const uint8_t *d;
  325. if (p->buf_size < 6)
  326. return 0;
  327. d = p->buf;
  328. code = (d[0] << 14) | (d[1] << 6) | (d[2] >> 2);
  329. if (code == 0x20) {
  330. return 50;
  331. }
  332. return 0;
  333. }
  334. static int h261_probe(AVProbeData *p)
  335. {
  336. int code;
  337. const uint8_t *d;
  338. if (p->buf_size < 6)
  339. return 0;
  340. d = p->buf;
  341. code = (d[0] << 12) | (d[1] << 4) | (d[2] >> 4);
  342. if (code == 0x10) {
  343. return 50;
  344. }
  345. return 0;
  346. }
  347. static int ac3_probe(AVProbeData *p)
  348. {
  349. int score=0;
  350. if(p->buf_size < 6)
  351. return 0;
  352. if((p->buf[0] == 0x0B) && (p->buf[1] == 0x77) && // sync word
  353. ((p->buf[4] >> 6) != 3) && // fscod
  354. ((p->buf[5] >> 3) <= 16)) { // bsid
  355. score = AVPROBE_SCORE_MAX / 2 + 10;
  356. }
  357. return score;
  358. }
  359. AVInputFormat shorten_demuxer = {
  360. "shn",
  361. "raw shorten",
  362. 0,
  363. NULL,
  364. shorten_read_header,
  365. raw_read_partial_packet,
  366. raw_read_close,
  367. .flags= AVFMT_GENERIC_INDEX,
  368. .extensions = "shn",
  369. };
  370. AVInputFormat flac_demuxer = {
  371. "flac",
  372. "raw flac",
  373. 0,
  374. NULL,
  375. flac_read_header,
  376. raw_read_partial_packet,
  377. raw_read_close,
  378. .flags= AVFMT_GENERIC_INDEX,
  379. .extensions = "flac",
  380. };
  381. #ifdef CONFIG_MUXERS
  382. AVOutputFormat flac_muxer = {
  383. "flac",
  384. "raw flac",
  385. "audio/x-flac",
  386. "flac",
  387. 0,
  388. CODEC_ID_FLAC,
  389. 0,
  390. flac_write_header,
  391. raw_write_packet,
  392. raw_write_trailer,
  393. .flags= AVFMT_NOTIMESTAMPS,
  394. };
  395. #endif //CONFIG_MUXERS
  396. AVInputFormat ac3_demuxer = {
  397. "ac3",
  398. "raw ac3",
  399. 0,
  400. ac3_probe,
  401. ac3_read_header,
  402. raw_read_partial_packet,
  403. raw_read_close,
  404. .flags= AVFMT_GENERIC_INDEX,
  405. .extensions = "ac3",
  406. };
  407. #ifdef CONFIG_MUXERS
  408. AVOutputFormat ac3_muxer = {
  409. "ac3",
  410. "raw ac3",
  411. "audio/x-ac3",
  412. "ac3",
  413. 0,
  414. CODEC_ID_AC3,
  415. 0,
  416. raw_write_header,
  417. raw_write_packet,
  418. raw_write_trailer,
  419. .flags= AVFMT_NOTIMESTAMPS,
  420. };
  421. #endif //CONFIG_MUXERS
  422. AVInputFormat dts_demuxer = {
  423. "dts",
  424. "raw dts",
  425. 0,
  426. NULL,
  427. dts_read_header,
  428. raw_read_partial_packet,
  429. raw_read_close,
  430. .flags= AVFMT_GENERIC_INDEX,
  431. .extensions = "dts",
  432. };
  433. AVInputFormat aac_demuxer = {
  434. "aac",
  435. "ADTS AAC",
  436. 0,
  437. NULL,
  438. aac_read_header,
  439. raw_read_partial_packet,
  440. raw_read_close,
  441. .flags= AVFMT_GENERIC_INDEX,
  442. .extensions = "aac",
  443. };
  444. AVInputFormat h261_demuxer = {
  445. "h261",
  446. "raw h261",
  447. 0,
  448. h261_probe,
  449. video_read_header,
  450. raw_read_partial_packet,
  451. raw_read_close,
  452. .flags= AVFMT_GENERIC_INDEX,
  453. .extensions = "h261",
  454. .value = CODEC_ID_H261,
  455. };
  456. #ifdef CONFIG_MUXERS
  457. AVOutputFormat h261_muxer = {
  458. "h261",
  459. "raw h261",
  460. "video/x-h261",
  461. "h261",
  462. 0,
  463. 0,
  464. CODEC_ID_H261,
  465. raw_write_header,
  466. raw_write_packet,
  467. raw_write_trailer,
  468. .flags= AVFMT_NOTIMESTAMPS,
  469. };
  470. #endif //CONFIG_MUXERS
  471. AVInputFormat h263_demuxer = {
  472. "h263",
  473. "raw h263",
  474. 0,
  475. h263_probe,
  476. video_read_header,
  477. raw_read_partial_packet,
  478. raw_read_close,
  479. .flags= AVFMT_GENERIC_INDEX,
  480. // .extensions = "h263", //FIXME remove after writing mpeg4_probe
  481. .value = CODEC_ID_H263,
  482. };
  483. #ifdef CONFIG_MUXERS
  484. AVOutputFormat h263_muxer = {
  485. "h263",
  486. "raw h263",
  487. "video/x-h263",
  488. "h263",
  489. 0,
  490. 0,
  491. CODEC_ID_H263,
  492. raw_write_header,
  493. raw_write_packet,
  494. raw_write_trailer,
  495. .flags= AVFMT_NOTIMESTAMPS,
  496. };
  497. #endif //CONFIG_MUXERS
  498. AVInputFormat m4v_demuxer = {
  499. "m4v",
  500. "raw MPEG4 video format",
  501. 0,
  502. mpeg4video_probe, /** probing for mpeg4 data */
  503. video_read_header,
  504. raw_read_partial_packet,
  505. raw_read_close,
  506. .flags= AVFMT_GENERIC_INDEX,
  507. .extensions = "m4v", //FIXME remove after writing mpeg4_probe
  508. .value = CODEC_ID_MPEG4,
  509. };
  510. #ifdef CONFIG_MUXERS
  511. AVOutputFormat m4v_muxer = {
  512. "m4v",
  513. "raw MPEG4 video format",
  514. NULL,
  515. "m4v",
  516. 0,
  517. CODEC_ID_NONE,
  518. CODEC_ID_MPEG4,
  519. raw_write_header,
  520. raw_write_packet,
  521. raw_write_trailer,
  522. .flags= AVFMT_NOTIMESTAMPS,
  523. };
  524. #endif //CONFIG_MUXERS
  525. AVInputFormat h264_demuxer = {
  526. "h264",
  527. "raw H264 video format",
  528. 0,
  529. NULL /*mpegvideo_probe*/,
  530. video_read_header,
  531. raw_read_partial_packet,
  532. raw_read_close,
  533. .flags= AVFMT_GENERIC_INDEX,
  534. .extensions = "h26l,h264,264", //FIXME remove after writing mpeg4_probe
  535. .value = CODEC_ID_H264,
  536. };
  537. #ifdef CONFIG_MUXERS
  538. AVOutputFormat h264_muxer = {
  539. "h264",
  540. "raw H264 video format",
  541. NULL,
  542. "h264",
  543. 0,
  544. CODEC_ID_NONE,
  545. CODEC_ID_H264,
  546. raw_write_header,
  547. raw_write_packet,
  548. raw_write_trailer,
  549. .flags= AVFMT_NOTIMESTAMPS,
  550. };
  551. #endif //CONFIG_MUXERS
  552. AVInputFormat mpegvideo_demuxer = {
  553. "mpegvideo",
  554. "MPEG video",
  555. 0,
  556. mpegvideo_probe,
  557. video_read_header,
  558. raw_read_partial_packet,
  559. raw_read_close,
  560. .flags= AVFMT_GENERIC_INDEX,
  561. .value = CODEC_ID_MPEG1VIDEO,
  562. };
  563. #ifdef CONFIG_MUXERS
  564. AVOutputFormat mpeg1video_muxer = {
  565. "mpeg1video",
  566. "MPEG video",
  567. "video/x-mpeg",
  568. "mpg,mpeg,m1v",
  569. 0,
  570. 0,
  571. CODEC_ID_MPEG1VIDEO,
  572. raw_write_header,
  573. raw_write_packet,
  574. raw_write_trailer,
  575. .flags= AVFMT_NOTIMESTAMPS,
  576. };
  577. #endif //CONFIG_MUXERS
  578. #ifdef CONFIG_MUXERS
  579. AVOutputFormat mpeg2video_muxer = {
  580. "mpeg2video",
  581. "MPEG2 video",
  582. NULL,
  583. "m2v",
  584. 0,
  585. 0,
  586. CODEC_ID_MPEG2VIDEO,
  587. raw_write_header,
  588. raw_write_packet,
  589. raw_write_trailer,
  590. .flags= AVFMT_NOTIMESTAMPS,
  591. };
  592. #endif //CONFIG_MUXERS
  593. AVInputFormat mjpeg_demuxer = {
  594. "mjpeg",
  595. "MJPEG video",
  596. 0,
  597. NULL,
  598. video_read_header,
  599. raw_read_partial_packet,
  600. raw_read_close,
  601. .flags= AVFMT_GENERIC_INDEX,
  602. .extensions = "mjpg,mjpeg",
  603. .value = CODEC_ID_MJPEG,
  604. };
  605. AVInputFormat ingenient_demuxer = {
  606. "ingenient",
  607. "Ingenient MJPEG",
  608. 0,
  609. NULL,
  610. video_read_header,
  611. ingenient_read_packet,
  612. raw_read_close,
  613. .flags= AVFMT_GENERIC_INDEX,
  614. .extensions = "cgi", // FIXME
  615. .value = CODEC_ID_MJPEG,
  616. };
  617. #ifdef CONFIG_MUXERS
  618. AVOutputFormat mjpeg_muxer = {
  619. "mjpeg",
  620. "MJPEG video",
  621. "video/x-mjpeg",
  622. "mjpg,mjpeg",
  623. 0,
  624. 0,
  625. CODEC_ID_MJPEG,
  626. raw_write_header,
  627. raw_write_packet,
  628. raw_write_trailer,
  629. .flags= AVFMT_NOTIMESTAMPS,
  630. };
  631. #endif //CONFIG_MUXERS
  632. AVInputFormat vc1_demuxer = {
  633. "vc1",
  634. "raw vc1",
  635. 0,
  636. NULL /* vc1_probe */,
  637. video_read_header,
  638. raw_read_partial_packet,
  639. raw_read_close,
  640. .extensions = "vc1",
  641. .value = CODEC_ID_VC1,
  642. };
  643. /* pcm formats */
  644. #define PCMINPUTDEF(name, long_name, ext, codec) \
  645. AVInputFormat pcm_ ## name ## _demuxer = {\
  646. #name,\
  647. long_name,\
  648. 0,\
  649. NULL,\
  650. raw_read_header,\
  651. raw_read_packet,\
  652. raw_read_close,\
  653. pcm_read_seek,\
  654. .flags= AVFMT_GENERIC_INDEX,\
  655. .extensions = ext,\
  656. .value = codec,\
  657. };
  658. #define PCMOUTPUTDEF(name, long_name, ext, codec) \
  659. AVOutputFormat pcm_ ## name ## _muxer = {\
  660. #name,\
  661. long_name,\
  662. NULL,\
  663. ext,\
  664. 0,\
  665. codec,\
  666. 0,\
  667. raw_write_header,\
  668. raw_write_packet,\
  669. raw_write_trailer,\
  670. .flags= AVFMT_NOTIMESTAMPS,\
  671. };
  672. #if !defined(CONFIG_MUXERS) && defined(CONFIG_DEMUXERS)
  673. #define PCMDEF(name, long_name, ext, codec) \
  674. PCMINPUTDEF(name, long_name, ext, codec)
  675. #elif defined(CONFIG_MUXERS) && !defined(CONFIG_DEMUXERS)
  676. #define PCMDEF(name, long_name, ext, codec) \
  677. PCMOUTPUTDEF(name, long_name, ext, codec)
  678. #elif defined(CONFIG_MUXERS) && defined(CONFIG_DEMUXERS)
  679. #define PCMDEF(name, long_name, ext, codec) \
  680. PCMINPUTDEF(name, long_name, ext, codec)\
  681. PCMOUTPUTDEF(name, long_name, ext, codec)
  682. #else
  683. #define PCMDEF(name, long_name, ext, codec)
  684. #endif
  685. #ifdef WORDS_BIGENDIAN
  686. #define BE_DEF(s) s
  687. #define LE_DEF(s) NULL
  688. #else
  689. #define BE_DEF(s) NULL
  690. #define LE_DEF(s) s
  691. #endif
  692. PCMDEF(s16le, "pcm signed 16 bit little endian format",
  693. LE_DEF("sw"), CODEC_ID_PCM_S16LE)
  694. PCMDEF(s16be, "pcm signed 16 bit big endian format",
  695. BE_DEF("sw"), CODEC_ID_PCM_S16BE)
  696. PCMDEF(u16le, "pcm unsigned 16 bit little endian format",
  697. LE_DEF("uw"), CODEC_ID_PCM_U16LE)
  698. PCMDEF(u16be, "pcm unsigned 16 bit big endian format",
  699. BE_DEF("uw"), CODEC_ID_PCM_U16BE)
  700. PCMDEF(s8, "pcm signed 8 bit format",
  701. "sb", CODEC_ID_PCM_S8)
  702. PCMDEF(u8, "pcm unsigned 8 bit format",
  703. "ub", CODEC_ID_PCM_U8)
  704. PCMDEF(mulaw, "pcm mu law format",
  705. "ul", CODEC_ID_PCM_MULAW)
  706. PCMDEF(alaw, "pcm A law format",
  707. "al", CODEC_ID_PCM_ALAW)
  708. static int rawvideo_read_packet(AVFormatContext *s, AVPacket *pkt)
  709. {
  710. int packet_size, ret, width, height;
  711. AVStream *st = s->streams[0];
  712. width = st->codec->width;
  713. height = st->codec->height;
  714. packet_size = avpicture_get_size(st->codec->pix_fmt, width, height);
  715. if (packet_size < 0)
  716. return -1;
  717. ret= av_get_packet(&s->pb, pkt, packet_size);
  718. pkt->stream_index = 0;
  719. if (ret != packet_size) {
  720. return AVERROR_IO;
  721. } else {
  722. return 0;
  723. }
  724. }
  725. AVInputFormat rawvideo_demuxer = {
  726. "rawvideo",
  727. "raw video format",
  728. 0,
  729. NULL,
  730. raw_read_header,
  731. rawvideo_read_packet,
  732. raw_read_close,
  733. .flags= AVFMT_GENERIC_INDEX,
  734. .extensions = "yuv,cif,qcif",
  735. .value = CODEC_ID_RAWVIDEO,
  736. };
  737. #ifdef CONFIG_MUXERS
  738. AVOutputFormat rawvideo_muxer = {
  739. "rawvideo",
  740. "raw video format",
  741. NULL,
  742. "yuv",
  743. 0,
  744. CODEC_ID_NONE,
  745. CODEC_ID_RAWVIDEO,
  746. raw_write_header,
  747. raw_write_packet,
  748. raw_write_trailer,
  749. .flags= AVFMT_NOTIMESTAMPS,
  750. };
  751. #endif //CONFIG_MUXERS
  752. #ifdef CONFIG_MUXERS
  753. static int null_write_packet(struct AVFormatContext *s, AVPacket *pkt)
  754. {
  755. return 0;
  756. }
  757. AVOutputFormat null_muxer = {
  758. "null",
  759. "null video format",
  760. NULL,
  761. NULL,
  762. 0,
  763. #ifdef WORDS_BIGENDIAN
  764. CODEC_ID_PCM_S16BE,
  765. #else
  766. CODEC_ID_PCM_S16LE,
  767. #endif
  768. CODEC_ID_RAWVIDEO,
  769. raw_write_header,
  770. null_write_packet,
  771. raw_write_trailer,
  772. .flags = AVFMT_NOFILE | AVFMT_RAWPICTURE | AVFMT_NOTIMESTAMPS,
  773. };
  774. #endif //CONFIG_MUXERS