VideoPicture.h

/*******************************************************************************
 * Copyright (c) 2024, 2026, Olivier Ayache.  All rights reserved.
 *
 * This file is part of AVPKit.
 *
 * AVPKit is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * AVPKit 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 Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with AVPKit.  If not, see <http://www.gnu.org/licenses/>.
 *******************************************************************************/
 
#ifndef VIDEOPICTURE_H_
#define VIDEOPICTURE_H_
 
#include <com/avpkit/ferry/RefPointer.h>
#include <com/avpkit/core/IVideoPicture.h>
#include <com/avpkit/core/ICodec.h>
#include <com/avpkit/core/FfmpegIncludes.h>
#include <com/avpkit/ferry/IBuffer.h>
#include <com/avpkit/core/IRational.h>
 
namespace com { namespace avpkit { namespace core
{
 
  class VideoPicture : public IVideoPicture
  {
    VS_JNIUTILS_REFCOUNTED_OBJECT_PRIVATE_MAKE(VideoPicture);
  public:
    // IMediaData
    virtual int64_t getTimeStamp() { return getPts(); }
    virtual void setTimeStamp(int64_t aTimeStamp) { setPts(aTimeStamp); }
    virtual bool isKey() { return isKeyFrame(); }
    virtual IRational* getTimeBase() { return mTimeBase.get(); }
    virtual void setTimeBase(IRational *aBase) { mTimeBase.reset(aBase, true); }
 
    // IVideoPicture Implementation
    virtual bool isKeyFrame();
    virtual void setKeyFrame(bool aIsKey);
    virtual bool isComplete() { return mIsComplete; }
    virtual int getWidth() { return mFrame->width; }
    virtual int getHeight() { return mFrame->height; }
    virtual IPixelFormat::Type getPixelType() { return (IPixelFormat::Type) mFrame->format; }
    virtual int64_t getPts();
    virtual void setPts(int64_t);
 
    virtual int getQuality();
    virtual void setQuality(int newQuality);    
    virtual int32_t getSize();
    virtual com::avpkit::ferry::IBuffer* getData();
    virtual int getDataLineSize(int lineNo);
    virtual void setComplete(bool aIsComplete, IPixelFormat::Type format,
        int width, int height, int64_t pts);
    virtual bool copy(IVideoPicture* srcFrame);
    virtual void setData(com::avpkit::ferry::IBuffer* buffer);
    virtual void render(bool drop, int64_t timeStamp);
    virtual void* getOpaqueData();
 
    // Not for calling from Java
    void fillAVFrame(AVFrame *frame);
 
    void copyAVFrame(AVFrame *frame, IPixelFormat::Type pixel,
        int32_t width, int32_t height);
    
    VS_API_AVPKIT AVFrame *getAVFrame();
 
    static VideoPicture* make(IPixelFormat::Type format, int width, int height);
    
    virtual IVideoPicture::PictType getPictureType();
    virtual void setPictureType(IVideoPicture::PictType type);
    
    virtual void setSideData(IVideoPicture::FrameDataType type, com::avpkit::ferry::IBuffer* buffer);
 
    static VideoPicture* make(com::avpkit::ferry::IBuffer* buffer,
        IPixelFormat::Type format, int width, int height);
 
  protected:
    VideoPicture();
    virtual ~VideoPicture();
    
  private:
    void allocInternalFrameBuffer();
    
    // This is where frame information is kept
    // about a decoded frame.
    AVFrame * mFrame;
    bool mIsComplete;
    
#ifdef __APPLE__
    CMSampleBufferRef mCMSampleBuffer;
#endif
 
    com::avpkit::ferry::RefPointer<com::avpkit::ferry::IBuffer> mBuffer;
    com::avpkit::ferry::RefPointer<IRational> mTimeBase;
  };
 
}}}
 
#endif /*VIDEOPICTURE_H_*/