/*
- Copyright (C) 2012 Carl Hetherington <cth@carlh.net>
+ Copyright (C) 2012-2015 Carl Hetherington <cth@carlh.net>
- This program is free software; you can redistribute it and/or modify
+ This file is part of DCP-o-matic.
+
+ DCP-o-matic is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.
- This program is distributed in the hope that it will be useful,
+ DCP-o-matic is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
- along with this program; if not, write to the Free Software
- Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+ along with DCP-o-matic. If not, see <http://www.gnu.org/licenses/>.
*/
-#ifndef DVDOMATIC_ENCODER_H
-#define DVDOMATIC_ENCODER_H
+#ifndef DCPOMATIC_ENCODER_H
+#define DCPOMATIC_ENCODER_H
-/** @file src/encoder.h
- * @brief Parent class for classes which can encode video and audio frames.
+/** @file src/encoder.h
+ * @brief Encoder class.
*/
+#include "util.h"
+#include "cross.h"
+#include "exception_store.h"
#include <boost/shared_ptr.hpp>
#include <boost/thread/mutex.hpp>
#include <boost/thread/condition.hpp>
#include <boost/thread.hpp>
#include <boost/optional.hpp>
+#include <boost/signals2.hpp>
#include <list>
#include <stdint.h>
-extern "C" {
-#include <libavutil/samplefmt.h>
-}
-#ifdef HAVE_SWRESAMPLE
-extern "C" {
-#include <libswresample/swresample.h>
-}
-#endif
-#include <sndfile.h>
-#include "util.h"
-#include "video_sink.h"
-#include "audio_sink.h"
-class Image;
-class Subtitle;
-class AudioBuffers;
class Film;
-class ServerDescription;
-class DCPVideoFrame;
-class EncodedData;
+class EncodeServerDescription;
+class DCPVideo;
class Writer;
+class Job;
+class PlayerVideo;
/** @class Encoder
- * @brief Encoder to J2K and WAV for DCP.
+ * @brief Class to manage encoding to JPEG2000.
*
- * Video is supplied to process_video as YUV frames, and audio
- * is supplied as uncompressed PCM in blocks of various sizes.
+ * This class keeps a queue of frames to be encoded and distributes
+ * the work around threads and encoding servers.
*/
-class Encoder : public VideoSink, public AudioSink
+class Encoder : public boost::noncopyable, public ExceptionStore
{
public:
- Encoder (boost::shared_ptr<const Film> f);
+ Encoder (boost::shared_ptr<const Film>, boost::shared_ptr<Writer>);
virtual ~Encoder ();
/** Called to indicate that a processing run is about to begin */
- virtual void process_begin ();
+ void begin ();
- /** Call with a frame of video.
- * @param i Video frame image.
- * @param same true if i is the same as the last time we were called.
- * @param s A subtitle that should be on this frame, or 0.
+ /** Called to pass in zero or more bits of video to be encoded
+ * as the next DCP frame.
*/
- void process_video (boost::shared_ptr<Image> i, bool same, boost::shared_ptr<Subtitle> s);
-
- /** Call with some audio data */
- void process_audio (boost::shared_ptr<AudioBuffers>);
+ void encode (std::list<boost::shared_ptr<PlayerVideo> > f);
/** Called when a processing run has finished */
- virtual void process_end ();
+ void end ();
- float current_frames_per_second () const;
- bool skipping () const;
+ float current_encoding_rate () const;
int video_frames_out () const;
private:
-
+
+ void enqueue (boost::shared_ptr<PlayerVideo> f);
void frame_done ();
- void frame_skipped ();
-
- void close_sound_files ();
- void write_audio (boost::shared_ptr<const AudioBuffers> audio);
- void encoder_thread (ServerDescription *);
- void terminate_worker_threads ();
+ void encoder_thread (boost::optional<EncodeServerDescription>);
+ void terminate_threads ();
+ void servers_list_changed ();
/** Film that we are encoding */
boost::shared_ptr<const Film> _film;
- /** Mutex for _time_history, _just_skipped and _last_frame */
- mutable boost::mutex _history_mutex;
+ /** Mutex for _time_history and _video_frames_enqueued */
+ mutable boost::mutex _state_mutex;
/** List of the times of completion of the last _history_size frames;
first is the most recently completed.
*/
std::list<struct timeval> _time_history;
/** Number of frames that we should keep history for */
static int const _history_size;
- /** true if the last frame we processed was skipped (because it was already done) */
- bool _just_skipped;
-
- /** Number of video frames received so far */
- SourceFrame _video_frames_in;
- /** Number of audio frames received so far */
- int64_t _audio_frames_in;
- /** Number of video frames written for the DCP so far */
- int _video_frames_out;
- /** Number of audio frames written for the DCP so far */
- int64_t _audio_frames_out;
-
-#if HAVE_SWRESAMPLE
- SwrContext* _swr_context;
-#endif
+ /** Current DCP frame index */
+ Frame _position;
+
+ /* XXX: probably should be atomic */
+ bool _terminate_enqueue;
+ bool _terminate_encoding;
+ /** Mutex for _threads */
+ mutable boost::mutex _threads_mutex;
+ std::list<boost::thread *> _threads;
+ mutable boost::mutex _queue_mutex;
+ std::list<boost::shared_ptr<DCPVideo> > _queue;
+ /** condition to manage thread wakeups when we have nothing to do */
+ boost::condition _empty_condition;
+ /** condition to manage thread wakeups when we have too much to do */
+ boost::condition _full_condition;
- std::vector<SNDFILE*> _sound_files;
+ boost::shared_ptr<Writer> _writer;
+ Waker _waker;
- bool _have_a_real_frame;
- bool _terminate_encoder;
- std::list<boost::shared_ptr<DCPVideoFrame> > _encode_queue;
- std::list<boost::thread *> _worker_threads;
- mutable boost::mutex _worker_mutex;
- boost::condition _worker_condition;
+ boost::shared_ptr<PlayerVideo> _last_player_video;
- boost::shared_ptr<Writer> _writer;
+ boost::signals2::scoped_connection _server_found_connection;
};
#endif