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.

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