video_encoder.h

/********************************************************************
 * Copyright (C) 2015 Liangliang Nan <liangliang.nan@gmail.com>
 * https://3d.bk.tudelft.nl/liangliang/
 *
 * This file is part of Easy3D. If it is useful in your research/work,
 * I would be grateful if you show your appreciation by citing it:
 * ------------------------------------------------------------------
 *      Liangliang Nan.
 *      Easy3D: a lightweight, easy-to-use, and efficient C++ library
 *      for processing and rendering 3D data.
 *      Journal of Open Source Software, 6(64), 3255, 2021.
 * ------------------------------------------------------------------
 *
 * Easy3D is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License Version 3
 * as published by the Free Software Foundation.
 *
 * Easy3D 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, see <http://www.gnu.org/licenses/>.
 ********************************************************************/
 
#ifndef EASY3D_VIDEO_ENCODER_H
#define EASY3D_VIDEO_ENCODER_H
 
#include <string>
 
 
namespace internal {
    class VideoEncoderImpl;
}
 
 
namespace easy3d {

    class VideoEncoder
    {
    public:
        explicit VideoEncoder(const std::string& file_name, int framerate = 30, int bitrate = 8 * 1024 * 1024);

        ~VideoEncoder();

        enum PixelFormat {
            PIX_FMT_RGB_888,    
            PIX_FMT_BGR_888,    
            PIX_FMT_RGBA_8888,  
            PIX_FMT_BGRA_8888   
        };

        bool encode(const unsigned char* image_data, int width, int height, PixelFormat pixel_format);

        static bool is_size_acceptable(int width, int height) { return (width % 8) == 0 && (height % 8) == 0; }
 
    private:
        internal::VideoEncoderImpl* encoder_;
 
        const std::string file_name_;
        const int framerate_;
        const int bitrate_;
    };
 
}
 
#endif  // EASY3D_VIDEO_ENCODER_H