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.

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