module Membrane.H264.FFmpeg.Encoder.Native

state_type "State"

type(
  ffmpeg_param :: %Membrane.H264.FFmpeg.Encoder.FFmpegParam{
    key: string,
    value: string
  }
)

spec create(
       width :: int,
       height :: int,
       pix_fmt :: atom,
       preset :: atom,
       tune :: atom,
       profile :: atom,
       max_b_frames :: int,
       gop_size :: int,
       timebase_num :: int,
       timebase_den :: int,
       crf :: int,
       sc_threshold :: int,
       ffmpeg_params :: [ffmpeg_param]
     ) :: {:ok :: label, state} | {:error :: label, reason :: atom}

spec get_frame_size(state) :: {:ok :: label, frame_size :: int} | {:error :: label}

spec encode(payload, pts :: int64, use_shm :: bool, keyframe_requested :: bool, state) ::
       {:ok :: label, dts_list :: [int64], pts_list :: [int64], [payload]}
       | {:error :: label, reason :: atom}

spec flush(use_shm :: bool, state) ::
       {:ok :: label, dts_list :: [int64], pts_list :: [int64], frames :: [payload]}
       | {:error :: label, reason :: atom}

dirty :cpu, [:flush, :encode, :create, :get_frame_size]