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.

902 lines
21KB

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