#pragma once

#include <nnwt/packet.h>
#include <libavcodec/avcodec.h>
#include <libavformat/avformat.h>
#include <libavutil/dict.h>

void nn_packet_write_av_codec_parameters(struct nn_packet *packet, AVCodecParameters *codecpar);
void nn_packet_write_av_codec_id(struct nn_packet *packet, enum AVCodecID codec_id);
void nn_packet_write_av_stream(struct nn_packet *packet, AVStream *stream);
void nn_packet_write_av_packet(struct nn_packet *packet, AVPacket *pkt);

void nn_packet_read_av_codec_parameters(struct nn_packet *packet, AVCodecParameters *codecpar);
enum AVCodecID nn_packet_read_av_codec_id(struct nn_packet *packet);
AVStream *nn_packet_read_av_stream(AVFormatContext *format_context, const AVCodec *codec, struct nn_packet *packet);
void nn_packet_read_av_packet(struct nn_packet *packet, AVPacket *pkt);