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.

936 lines
22KB

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