X-Git-Url: https://main.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Flib%2Fimage.h;h=46633df679d00b9f6eceee34be126711c0fd5d85;hb=f515b8daea9d28200be803bb64ff17e9f30343c4;hp=1d7d75dfc9a32361541dd70d81d86feb59034dcc;hpb=afc495f722f89fea0bcc579046d1a5d362e36f69;p=dcpomatic.git diff --git a/src/lib/image.h b/src/lib/image.h index 1d7d75dfc..46633df67 100644 --- a/src/lib/image.h +++ b/src/lib/image.h @@ -1,19 +1,20 @@ /* - Copyright (C) 2012 Carl Hetherington + Copyright (C) 2012-2016 Carl Hetherington - 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 . */ @@ -24,146 +25,86 @@ #ifndef DCPOMATIC_IMAGE_H #define DCPOMATIC_IMAGE_H -#include -#include -#include +#include "position.h" +#include "position_image.h" +#include "types.h" extern "C" { -#include -#include +#include } -#include "util.h" -#include "ffmpeg_compatibility.h" - -class Scaler; -class RGBFrameImage; -class SimpleImage; - -/** @class Image - * @brief Parent class for wrappers of some image, in some format, that - * can present a set of components and a size in pixels. - * - * This class also has some conversion / processing methods. - * - * The main point of this class (and its subclasses) is to abstract - * details of FFmpeg's memory management and varying data formats. - */ -class Image -{ -public: - Image (AVPixelFormat p) - : _pixel_format (p) - {} - - virtual ~Image () {} - - /** @return Array of pointers to arrays of the component data */ - virtual uint8_t ** data () const = 0; - - /** @return Array of sizes of the data in each line, in bytes (without any alignment padding bytes) */ - virtual int * line_size () const = 0; +#include +#include +#include - /** @return Array of strides for each line (including any alignment padding bytes) */ - virtual int * stride () const = 0; +struct AVFrame; +class Socket; - /** @return Size of the image, in pixels */ - virtual libdcp::Size size () const = 0; +class Image : public boost::enable_shared_from_this +{ +public: + Image (AVPixelFormat p, dcp::Size s, bool aligned); + explicit Image (AVFrame *); + explicit Image (Image const &); + Image (boost::shared_ptr, bool); + Image& operator= (Image const &); + ~Image (); + + uint8_t * const * data () const; + int const * line_size () const; + int const * stride () const; + dcp::Size size () const; + bool aligned () const; - virtual bool aligned () const = 0; + int planes () const; + int vertical_factor (int) const; + int horizontal_factor (int) const; + dcp::Size sample_size (int) const; + float bytes_per_pixel (int) const; - int components () const; - int lines (int) const; + boost::shared_ptr convert_pixel_format (dcp::YUVToRGB yuv_to_rgb, AVPixelFormat out_format, bool aligned, bool fast) const; + boost::shared_ptr scale (dcp::Size out_size, dcp::YUVToRGB yuv_to_rgb, AVPixelFormat out_format, bool aligned, bool fast) const; + boost::shared_ptr crop_scale_window ( + Crop crop, dcp::Size inter_size, dcp::Size out_size, dcp::YUVToRGB yuv_to_rgb, VideoRange video_range, AVPixelFormat out_format, bool aligned, bool fast + ) const; - boost::shared_ptr scale_and_convert_to_rgb (libdcp::Size out_size, int padding, Scaler const * scaler, bool aligned) const; - boost::shared_ptr scale (libdcp::Size, Scaler const *, bool aligned) const; - boost::shared_ptr post_process (std::string, bool aligned) const; - void alpha_blend (boost::shared_ptr image, Position pos); - boost::shared_ptr crop (Crop c, bool aligned) const; - void make_black (); + void make_transparent (); + void alpha_blend (boost::shared_ptr image, Position pos); + void copy (boost::shared_ptr image, Position pos); + void fade (float); void read_from_socket (boost::shared_ptr); void write_to_socket (boost::shared_ptr) const; - + AVPixelFormat pixel_format () const { return _pixel_format; } -protected: - virtual void swap (Image &); - float bytes_per_pixel (int) const; + size_t memory_used () const; -private: - void yuv_16_black (uint16_t); - static uint16_t swap_16 (uint16_t); - - AVPixelFormat _pixel_format; ///< FFmpeg's way of describing the pixel format of this Image -}; + dcp::Data as_png () const; -/** @class FilterBufferImage - * @brief An Image that is held in an AVFilterBufferRef. - */ -class FilterBufferImage : public Image -{ -public: - FilterBufferImage (AVPixelFormat, AVFilterBufferRef *); - ~FilterBufferImage (); + void png_error (char const * message); - uint8_t ** data () const; - int * line_size () const; - int * stride () const; - libdcp::Size size () const; - bool aligned () const; + static boost::shared_ptr ensure_aligned (boost::shared_ptr image); private: - /* Not allowed */ - FilterBufferImage (FilterBufferImage const &); - FilterBufferImage& operator= (FilterBufferImage const &); - - AVFilterBufferRef* _buffer; - int* _line_size; -}; - -/** @class SimpleImage - * @brief An Image for which memory is allocated using a `simple' av_malloc(). - */ -class SimpleImage : public Image -{ -public: - SimpleImage (AVPixelFormat, libdcp::Size, bool); - SimpleImage (SimpleImage const &); - SimpleImage& operator= (SimpleImage const &); - ~SimpleImage (); - - uint8_t ** data () const; - int * line_size () const; - int * stride () const; - libdcp::Size size () const; - bool aligned () const; + friend struct pixel_formats_test; -protected: void allocate (); - void swap (SimpleImage &); - -private: - libdcp::Size _size; ///< size in pixels + void swap (Image &); + void make_part_black (int x, int w); + void yuv_16_black (uint16_t, bool); + static uint16_t swap_16 (uint16_t); + + dcp::Size _size; + AVPixelFormat _pixel_format; ///< FFmpeg's way of describing the pixel format of this Image uint8_t** _data; ///< array of pointers to components - int* _line_size; ///< array of sizes of the data in each line, in pixels (without any alignment padding bytes) - int* _stride; ///< array of strides for each line (including any alignment padding bytes) + int* _line_size; ///< array of sizes of the data in each line, in bytes (without any alignment padding bytes) + int* _stride; ///< array of strides for each line, in bytes (including any alignment padding bytes) bool _aligned; }; -class RGBPlusAlphaImage : public SimpleImage -{ -public: - RGBPlusAlphaImage (boost::shared_ptr); - ~RGBPlusAlphaImage (); - - uint8_t* alpha () const { - return _alpha; - } - -private: - uint8_t* _alpha; -}; +extern PositionImage merge (std::list images); +extern bool operator== (Image const & a, Image const & b); #endif