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.

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