mpeg2enc: support mjpegtools 2.0
[gstreamer-omap:gst-plugins-bad.git] / ext / mpeg2enc / gstmpeg2encoder.cc
1 /* GStreamer mpeg2enc (mjpegtools) wrapper
2  * (c) 2003 Ronald Bultje <rbultje@ronald.bitfreak.net>
3  * (c) 2006 Mark Nauwelaerts <manauw@skynet.be>
4  *
5  * gstmpeg2encoder.cc: gstreamer/mpeg2enc encoder class
6  *
7  * This library is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Library General Public
9  * License as published by the Free Software Foundation; either
10  * version 2 of the License, or (at your option) any later version.
11  *
12  * This library is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Library General Public License for more details.
16  *
17  * You should have received a copy of the GNU Library General Public
18  * License along with this library; if not, write to the
19  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
20  * Boston, MA 02111-1307, USA.
21  */
22
23 #ifdef HAVE_CONFIG_H
24 #include "config.h"
25 #endif
26
27 #include <mpegconsts.h>
28 #include <quantize.hh>
29 #if GST_MJPEGTOOLS_API >= 10903
30 #include <ontheflyratectlpass1.hh>
31 #include <ontheflyratectlpass2.hh>
32 #elif GST_MJPEGTOOLS_API >= 10900
33 #include <ontheflyratectl.hh>
34 #include <pass1ratectl.hh>
35 #include <pass2ratectl.hh>
36 #else
37 #include <ratectl.hh>
38 #endif
39 #include <seqencoder.hh>
40 #include <mpeg2coder.hh>
41
42 #include "gstmpeg2enc.hh"
43 #include "gstmpeg2encoder.hh"
44
45 /*
46  * Class init stuff.
47  */
48
49 GstMpeg2Encoder::GstMpeg2Encoder (GstMpeg2EncOptions * options, GstElement * in_element, GstCaps * in_caps):
50 MPEG2Encoder (*options)
51 {
52   element = in_element;
53   gst_object_ref (element);
54   caps = in_caps;
55   gst_caps_ref (in_caps);
56   init_done = FALSE;
57 }
58
59 GstMpeg2Encoder::~GstMpeg2Encoder ()
60 {
61   gst_caps_unref (caps);
62   gst_object_unref (element);
63 }
64
65 gboolean GstMpeg2Encoder::setup ()
66 {
67   MPEG2EncInVidParams
68       strm;
69   GstMpeg2enc *
70       enc;
71
72   enc = GST_MPEG2ENC (element);
73
74   /* I/O */
75   reader = new GstMpeg2EncPictureReader (element, caps, &parms);
76   reader->StreamPictureParams (strm);
77 #if GST_MJPEGTOOLS_API == 10800
78   /* chain thread caters for reading, do not need another thread for this */
79   options.allow_parallel_read = FALSE;
80 #endif
81   if (options.SetFormatPresets (strm)) {
82     return FALSE;
83   }
84   writer = new GstMpeg2EncStreamWriter (enc->srcpad, &parms);
85
86   /* encoding internals */
87   quantizer = new Quantizer (parms);
88 #if GST_MJPEGTOOLS_API < 10900
89   bitrate_controller = new OnTheFlyRateCtl (parms);
90 #else
91   pass1ratectl = new OnTheFlyPass1 (parms);
92   pass2ratectl = new OnTheFlyPass2 (parms);
93 #endif
94 #if GST_MJPEGTOOLS_API >= 10900
95   /* sequencer */
96   seqencoder = new SeqEncoder (parms, *reader, *quantizer,
97       *writer, *pass1ratectl, *pass2ratectl);
98 #elif GST_MJPEGTOOLS_API >= 10800
99   /* sequencer */
100   seqencoder = new SeqEncoder (parms, *reader, *quantizer,
101       *writer, *bitrate_controller);
102 #else
103   coder = new MPEG2Coder (parms, *writer);
104   /* sequencer */
105   seqencoder = new SeqEncoder (parms, *reader, *quantizer,
106       *writer, *coder, *bitrate_controller);
107 #endif
108
109   return TRUE;
110 }
111
112 void
113 GstMpeg2Encoder::init ()
114 {
115   if (!init_done) {
116     parms.Init (options);
117     reader->Init ();
118     quantizer->Init ();
119 #if GST_MJPEGTOOLS_API >= 10800
120     seqencoder->Init ();
121 #endif
122     init_done = TRUE;
123   }
124 }
125
126 /*
127  * Process all input provided by the reader until it signals eos.
128  */
129
130 void
131 GstMpeg2Encoder::encode ()
132 {
133   /* hm, this is all... eek! */
134 #if GST_MJPEGTOOLS_API >= 10800
135   seqencoder->EncodeStream ();
136 #else
137   seqencoder->Encode ();
138 #endif
139 }
140
141 /*
142  * Get current output format.
143  */
144
145 GstCaps *
146 GstMpeg2Encoder::getFormat ()
147 {
148   y4m_ratio_t fps = mpeg_framerate (options.frame_rate);
149
150   return gst_caps_new_simple ("video/mpeg",
151       "systemstream", G_TYPE_BOOLEAN, FALSE,
152       "mpegversion", G_TYPE_INT, options.mpeg,
153       "width", G_TYPE_INT, options.in_img_width,
154       "height", G_TYPE_INT, options.in_img_height,
155       "framerate", GST_TYPE_FRACTION, fps.n, fps.d, NULL);
156 }