diff --git a/libavformat/avienc.c b/libavformat/avienc.c index e36e88335f..2264241d57 100644 --- a/libavformat/avienc.c +++ b/libavformat/avienc.c @@ -28,6 +28,7 @@ #include "config_components.h" #include "riff.h" #include "mpegts.h" +#include "rawutils.h" #include "libavformat/avlanguage.h" #include "libavutil/avstring.h" #include "libavutil/avutil.h" diff --git a/libavformat/internal.h b/libavformat/internal.h index eb18277cea..da9abde637 100644 --- a/libavformat/internal.h +++ b/libavformat/internal.h @@ -901,33 +901,6 @@ int ff_parse_creation_time_metadata(AVFormatContext *s, int64_t *timestamp, int */ int ff_standardize_creation_time(AVFormatContext *s); -#define CONTAINS_PAL 2 -/** - * Reshuffles the lines to use the user specified stride. - * - * @param ppkt input and output packet - * @return negative error code or - * 0 if no new packet was allocated - * non-zero if a new packet was allocated and ppkt has to be freed - * CONTAINS_PAL if in addition to a new packet the old contained a palette - */ -int ff_reshuffle_raw_rgb(AVFormatContext *s, AVPacket **ppkt, AVCodecParameters *par, int expected_stride); - -/** - * Retrieves the palette from a packet, either from side data, or - * appended to the video data in the packet itself (raw video only). - * It is commonly used after a call to ff_reshuffle_raw_rgb(). - * - * Use 0 for the ret parameter to check for side data only. - * - * @param pkt pointer to packet before calling ff_reshuffle_raw_rgb() - * @param ret return value from ff_reshuffle_raw_rgb(), or 0 - * @param palette pointer to palette buffer - * @return negative error code or - * 1 if the packet has a palette, else 0 - */ -int ff_get_packet_palette(AVFormatContext *s, AVPacket *pkt, int ret, uint32_t *palette); - struct AVBPrint; /** * Finalize buf into extradata and set its size appropriately. diff --git a/libavformat/movenc.c b/libavformat/movenc.c index d32328d40c..c720fd742d 100644 --- a/libavformat/movenc.c +++ b/libavformat/movenc.c @@ -62,6 +62,7 @@ #include "mov_chan.h" #include "movenc_ttml.h" #include "mux.h" +#include "rawutils.h" #include "ttmlenc.h" #include "version.h" #include "vpcc.h" diff --git a/libavformat/rawutils.c b/libavformat/rawutils.c index 996412adb9..e44c946d47 100644 --- a/libavformat/rawutils.c +++ b/libavformat/rawutils.c @@ -19,8 +19,10 @@ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ +#include "libavutil/intreadwrite.h" +#include "libavcodec/packet.h" #include "avformat.h" -#include "internal.h" +#include "rawutils.h" int ff_reshuffle_raw_rgb(AVFormatContext *s, AVPacket **ppkt, AVCodecParameters *par, int expected_stride) { @@ -65,3 +67,27 @@ fail: return ret; } + +int ff_get_packet_palette(AVFormatContext *s, AVPacket *pkt, int ret, uint32_t *palette) +{ + uint8_t *side_data; + size_t size; + + side_data = av_packet_get_side_data(pkt, AV_PKT_DATA_PALETTE, &size); + if (side_data) { + if (size != AVPALETTE_SIZE) { + av_log(s, AV_LOG_ERROR, "Invalid palette side data\n"); + return AVERROR_INVALIDDATA; + } + memcpy(palette, side_data, AVPALETTE_SIZE); + return 1; + } + + if (ret == CONTAINS_PAL) { + for (int i = 0; i < AVPALETTE_COUNT; i++) + palette[i] = AV_RL32(pkt->data + pkt->size - AVPALETTE_SIZE + i*4); + return 1; + } + + return 0; +} diff --git a/libavformat/rawutils.h b/libavformat/rawutils.h new file mode 100644 index 0000000000..a91123412b --- /dev/null +++ b/libavformat/rawutils.h @@ -0,0 +1,56 @@ +/* + * copyright (c) 2001 Fabrice Bellard + * + * This file is part of FFmpeg. + * + * FFmpeg 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 2.1 of the License, or (at your option) any later version. + * + * FFmpeg 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 FFmpeg; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA + */ + +#ifndef AVFORMAT_RAWUTILS_H +#define AVFORMAT_RAWUTILS_H + +#include +#include "libavcodec/codec_par.h" +#include "libavcodec/packet.h" +#include "avformat.h" + +#define CONTAINS_PAL 2 +/** + * Reshuffles the lines to use the user specified stride. + * + * @param ppkt input and output packet + * @return negative error code or + * 0 if no new packet was allocated + * non-zero if a new packet was allocated and ppkt has to be freed + * CONTAINS_PAL if in addition to a new packet the old contained a palette + */ +int ff_reshuffle_raw_rgb(AVFormatContext *s, AVPacket **ppkt, AVCodecParameters *par, int expected_stride); + +/** + * Retrieves the palette from a packet, either from side data, or + * appended to the video data in the packet itself (raw video only). + * It is commonly used after a call to ff_reshuffle_raw_rgb(). + * + * Use 0 for the ret parameter to check for side data only. + * + * @param pkt pointer to packet before calling ff_reshuffle_raw_rgb() + * @param ret return value from ff_reshuffle_raw_rgb(), or 0 + * @param palette pointer to palette buffer + * @return negative error code or + * 1 if the packet has a palette, else 0 + */ +int ff_get_packet_palette(AVFormatContext *s, AVPacket *pkt, int ret, uint32_t *palette); + +#endif /* AVFORMAT_RAWUTILS_H */ diff --git a/libavformat/utils.c b/libavformat/utils.c index b53e2aff0c..cc5507a19e 100644 --- a/libavformat/utils.c +++ b/libavformat/utils.c @@ -1715,30 +1715,6 @@ int ff_standardize_creation_time(AVFormatContext *s) return ret; } -int ff_get_packet_palette(AVFormatContext *s, AVPacket *pkt, int ret, uint32_t *palette) -{ - uint8_t *side_data; - size_t size; - - side_data = av_packet_get_side_data(pkt, AV_PKT_DATA_PALETTE, &size); - if (side_data) { - if (size != AVPALETTE_SIZE) { - av_log(s, AV_LOG_ERROR, "Invalid palette side data\n"); - return AVERROR_INVALIDDATA; - } - memcpy(palette, side_data, AVPALETTE_SIZE); - return 1; - } - - if (ret == CONTAINS_PAL) { - for (int i = 0; i < AVPALETTE_COUNT; i++) - palette[i] = AV_RL32(pkt->data + pkt->size - AVPALETTE_SIZE + i*4); - return 1; - } - - return 0; -} - int ff_bprint_to_codecpar_extradata(AVCodecParameters *par, struct AVBPrint *buf) { int ret;