From patchwork Wed May 27 15:20:42 2020 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Paul B Mahol X-Patchwork-Id: 19904 Return-Path: X-Original-To: patchwork@ffaux-bg.ffmpeg.org Delivered-To: patchwork@ffaux-bg.ffmpeg.org Received: from ffbox0-bg.mplayerhq.hu (ffbox0-bg.ffmpeg.org [79.124.17.100]) by ffaux.localdomain (Postfix) with ESMTP id B023C4496AC for ; Wed, 27 May 2020 18:28:15 +0300 (EEST) Received: from [127.0.1.1] (localhost [127.0.0.1]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTP id 875AC68AB98; Wed, 27 May 2020 18:28:15 +0300 (EEST) X-Original-To: ffmpeg-devel@ffmpeg.org Delivered-To: ffmpeg-devel@ffmpeg.org Received: from mail-wr1-f68.google.com (mail-wr1-f68.google.com [209.85.221.68]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTPS id CBEF568AB91 for ; Wed, 27 May 2020 18:28:08 +0300 (EEST) Received: by mail-wr1-f68.google.com with SMTP id s8so24374324wrt.9 for ; Wed, 27 May 2020 08:28:08 -0700 (PDT) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=gmail.com; s=20161025; h=from:to:subject:date:message-id; bh=y9CHfvsIevnV387I8DiiNptjVcLZAZcJGcilrD3McVc=; b=XCoN6EqKUI2bsjY/NJ7/enT4xXVCP5y+/Z+M048E5KmzUy8F2WZzJGWLyvb2wpCsC9 FWbXSwd8DXTkYNOqk8PaMOEBIzC+H+hSu0tQQAfjpRvnDqBRcg/b/oxEZXgSIxbaGvHJ go8qS6FLBrDPtQOmH1TJ24BX5m9W6qQ56oadHXk0kHR6cioSPQZ5VwCFmZ+CAP+RaxLK eUbTKMILB9eE3xFQS2EPY7J895PDsLY7tAdBCo0MmCvoO4ka8o4cXFwLXPKPm4n9wbZw JgAO37FtH/MfF1l48wIrKsfOGmfsT509xGmJm4Q8qZyzrqKQ9ciEPBBEOB9fUjkHyHND Lz8Q== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20161025; h=x-gm-message-state:from:to:subject:date:message-id; bh=y9CHfvsIevnV387I8DiiNptjVcLZAZcJGcilrD3McVc=; b=gEJXHPFVKSQ280+SCSX5WOjNru1Xd3aDhvzj3WP7JVtM0BR9xYntfTzqLtf/EmCY7i FhaBOpiT3ex/rsYHFEXxMTnw+iK+5CfMGnVZDjOK5e5PxlcVMCigUtqWbPQ//dt0g1eg pYgV0UQ9N0OhjYfCNGM2NSdzCBE0r4UPfoG00dGvRpqyjb0B2LkrUjODSh6PRKEEK1VA Np4BW6sMlo3M/AggbIHM5qTEAMvsvplTSprpL5Wobf2foCix6Xz5Zi5tbVLpUoLaJEEY TM9SWD9GkUB1aKpzfDUCUKm/tDpKx0Ghk5L7ZIzZrtzuy9PCUcVPJBNr+GjgqMHVMO4o AwlA== X-Gm-Message-State: AOAM531pUTY5LHj3siMZyPq6hIHr3SSN5hUT1hZ/I79PhoQrkXNRAanl f3fUH82FnDTbTNSOk4Pjo7+w0Hl2RWA= X-Google-Smtp-Source: ABdhPJzsrzEvH1EVHOIJ5Ybl0JVVWZ6EitpRx8nrhLNDE80H5os+9jSKB3IOQMp/MFpeJIZc6A7HpA== X-Received: by 2002:aa7:d312:: with SMTP id p18mr24451881edq.88.1590592855790; Wed, 27 May 2020 08:20:55 -0700 (PDT) Received: from localhost.localdomain ([37.244.239.188]) by smtp.gmail.com with ESMTPSA id dt7sm3002030ejb.33.2020.05.27.08.20.54 for (version=TLS1_3 cipher=TLS_AES_256_GCM_SHA384 bits=256/256); Wed, 27 May 2020 08:20:55 -0700 (PDT) From: Paul B Mahol To: ffmpeg-devel@ffmpeg.org Date: Wed, 27 May 2020 17:20:42 +0200 Message-Id: <20200527152042.23373-1-onemda@gmail.com> X-Mailer: git-send-email 2.17.1 Subject: [FFmpeg-devel] [PATCH] avcodec: add PFM image decoder X-BeenThere: ffmpeg-devel@ffmpeg.org X-Mailman-Version: 2.1.20 Precedence: list List-Id: FFmpeg development discussions and patches List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , Reply-To: FFmpeg development discussions and patches MIME-Version: 1.0 Errors-To: ffmpeg-devel-bounces@ffmpeg.org Sender: "ffmpeg-devel" Signed-off-by: Paul B Mahol --- libavcodec/Makefile | 1 + libavcodec/allcodecs.c | 1 + libavcodec/codec_desc.c | 7 ++++++ libavcodec/codec_id.h | 1 + libavcodec/pnm.c | 21 +++++++++++++--- libavcodec/pnm.h | 2 ++ libavcodec/pnmdec.c | 55 +++++++++++++++++++++++++++++++++++++++++ libavformat/img2.c | 1 + 8 files changed, 85 insertions(+), 4 deletions(-) diff --git a/libavcodec/Makefile b/libavcodec/Makefile index 6d816308c7..fa7bda4c5a 100644 --- a/libavcodec/Makefile +++ b/libavcodec/Makefile @@ -530,6 +530,7 @@ OBJS-$(CONFIG_PBM_DECODER) += pnmdec.o pnm.o OBJS-$(CONFIG_PBM_ENCODER) += pnmenc.o OBJS-$(CONFIG_PCX_DECODER) += pcx.o OBJS-$(CONFIG_PCX_ENCODER) += pcxenc.o +OBJS-$(CONFIG_PFM_DECODER) += pnmdec.o pnm.o OBJS-$(CONFIG_PGM_DECODER) += pnmdec.o pnm.o OBJS-$(CONFIG_PGM_ENCODER) += pnmenc.o OBJS-$(CONFIG_PGMYUV_DECODER) += pnmdec.o pnm.o diff --git a/libavcodec/allcodecs.c b/libavcodec/allcodecs.c index eb5044f80e..f34ce35b42 100644 --- a/libavcodec/allcodecs.c +++ b/libavcodec/allcodecs.c @@ -233,6 +233,7 @@ extern AVCodec ff_pbm_encoder; extern AVCodec ff_pbm_decoder; extern AVCodec ff_pcx_encoder; extern AVCodec ff_pcx_decoder; +extern AVCodec ff_pfm_decoder; extern AVCodec ff_pgm_encoder; extern AVCodec ff_pgm_decoder; extern AVCodec ff_pgmyuv_encoder; diff --git a/libavcodec/codec_desc.c b/libavcodec/codec_desc.c index b94ad0b1e6..9f8847544f 100644 --- a/libavcodec/codec_desc.c +++ b/libavcodec/codec_desc.c @@ -1770,6 +1770,13 @@ static const AVCodecDescriptor codec_descriptors[] = { .long_name = NULL_IF_CONFIG_SMALL("NotchLC"), .props = AV_CODEC_PROP_INTRA_ONLY | AV_CODEC_PROP_LOSSY, }, + { + .id = AV_CODEC_ID_PFM, + .type = AVMEDIA_TYPE_VIDEO, + .name = "pfm", + .long_name = NULL_IF_CONFIG_SMALL("PFM (Portable FloatMap) image"), + .props = AV_CODEC_PROP_INTRA_ONLY | AV_CODEC_PROP_LOSSLESS, + }, /* various PCM "codecs" */ { diff --git a/libavcodec/codec_id.h b/libavcodec/codec_id.h index f7cb0a6056..d885962c9c 100644 --- a/libavcodec/codec_id.h +++ b/libavcodec/codec_id.h @@ -294,6 +294,7 @@ enum AVCodecID { AV_CODEC_ID_CDTOONS, AV_CODEC_ID_MV30, AV_CODEC_ID_NOTCHLC, + AV_CODEC_ID_PFM, /* various PCM "codecs" */ AV_CODEC_ID_FIRST_AUDIO = 0x10000, ///< A dummy id pointing at the start of audio codecs diff --git a/libavcodec/pnm.c b/libavcodec/pnm.c index b5c2881948..a6ae01b494 100644 --- a/libavcodec/pnm.c +++ b/libavcodec/pnm.c @@ -24,6 +24,7 @@ #include "libavutil/avassert.h" #include "libavutil/imgutils.h" +#include "libavutil/avstring.h" #include "avcodec.h" #include "internal.h" #include "pnm.h" @@ -69,8 +70,9 @@ int ff_pnm_decode_header(AVCodecContext *avctx, PNMContext * const s) if (s->bytestream_end - s->bytestream < 3 || s->bytestream[0] != 'P' || - s->bytestream[1] < '1' || - s->bytestream[1] > '7') { + (s->bytestream[1] < '1' || + s->bytestream[1] > '7' && + s->bytestream[1] != 'F')) { s->bytestream += s->bytestream_end > s->bytestream; s->bytestream += s->bytestream_end > s->bytestream; return AVERROR_INVALIDDATA; @@ -78,7 +80,9 @@ int ff_pnm_decode_header(AVCodecContext *avctx, PNMContext * const s) pnm_get(s, buf1, sizeof(buf1)); s->type= buf1[1]-'0'; - if (s->type==1 || s->type==4) { + if (buf1[1] == 'F') { + avctx->pix_fmt = AV_PIX_FMT_GBRPF32; + } else if (s->type==1 || s->type==4) { avctx->pix_fmt = AV_PIX_FMT_MONOWHITE; } else if (s->type==2 || s->type==5) { if (avctx->codec_id == AV_CODEC_ID_PGMYUV) @@ -173,7 +177,16 @@ int ff_pnm_decode_header(AVCodecContext *avctx, PNMContext * const s) if (ret < 0) return ret; - if (avctx->pix_fmt != AV_PIX_FMT_MONOWHITE && avctx->pix_fmt != AV_PIX_FMT_MONOBLACK) { + if (avctx->pix_fmt == AV_PIX_FMT_GBRPF32) { + pnm_get(s, buf1, sizeof(buf1)); + if (av_sscanf(buf1, "%f", &s->scale) != 1) { + av_log(avctx, AV_LOG_ERROR, "Invalid scale.\n"); + return AVERROR_INVALIDDATA; + } + s->endian = s->scale < 0.f; + s->scale = fabsf(s->scale); + s->maxval = (1ULL << 32) - 1; + } else if (avctx->pix_fmt != AV_PIX_FMT_MONOWHITE && avctx->pix_fmt != AV_PIX_FMT_MONOBLACK) { pnm_get(s, buf1, sizeof(buf1)); s->maxval = atoi(buf1); if (s->maxval <= 0 || s->maxval > UINT16_MAX) { diff --git a/libavcodec/pnm.h b/libavcodec/pnm.h index 5bc0aad29f..89c3b5a2da 100644 --- a/libavcodec/pnm.h +++ b/libavcodec/pnm.h @@ -30,6 +30,8 @@ typedef struct PNMContext { uint8_t *bytestream_end; int maxval; ///< maximum value of a pixel int type; + int endian; + float scale; } PNMContext; int ff_pnm_decode_header(AVCodecContext *avctx, PNMContext * const s); diff --git a/libavcodec/pnmdec.c b/libavcodec/pnmdec.c index e03e52297f..05bd11b147 100644 --- a/libavcodec/pnmdec.c +++ b/libavcodec/pnmdec.c @@ -46,6 +46,7 @@ static int pnm_decode_frame(AVCodecContext *avctx, void *data, int i, j, k, n, linesize, h, upgrade = 0, is_mono = 0; unsigned char *ptr; int components, sample_len, ret; + float scale; s->bytestream_start = s->bytestream = (uint8_t *)buf; @@ -254,6 +255,48 @@ static int pnm_decode_frame(AVCodecContext *avctx, void *data, } } break; + case AV_PIX_FMT_GBRPF32: + if (avctx->width * avctx->height * 12 > s->bytestream_end - s->bytestream) + return AVERROR_INVALIDDATA; + scale = 1.f / s->scale; + if (s->endian) { + float *r, *g, *b; + + r = (float *)p->data[2]; + g = (float *)p->data[0]; + b = (float *)p->data[1]; + for (int i = 0; i < avctx->height; i++) { + for (int j = 0; j < avctx->width; j++) { + r[j] = av_int2float(av_le2ne32(((uint32_t *)s->bytestream)[0])) * scale; + g[j] = av_int2float(av_le2ne32(((uint32_t *)s->bytestream)[4])) * scale; + b[j] = av_int2float(av_le2ne32(((uint32_t *)s->bytestream)[8])) * scale; + s->bytestream += 12; + } + + r += p->linesize[2] / 4; + g += p->linesize[0] / 4; + b += p->linesize[1] / 4; + } + } else { + float *r, *g, *b; + + r = (float *)p->data[2]; + g = (float *)p->data[0]; + b = (float *)p->data[1]; + for (int i = 0; i < avctx->height; i++) { + for (int j = 0; j < avctx->width; j++) { + r[j] = av_int2float(av_be2ne32(((uint32_t *)s->bytestream)[0])) * scale; + g[j] = av_int2float(av_be2ne32(((uint32_t *)s->bytestream)[4])) * scale; + b[j] = av_int2float(av_be2ne32(((uint32_t *)s->bytestream)[8])) * scale; + s->bytestream += 12; + } + + r += p->linesize[2] / 4; + g += p->linesize[0] / 4; + b += p->linesize[1] / 4; + } + } + break; } *got_frame = 1; @@ -320,3 +363,15 @@ AVCodec ff_pam_decoder = { .capabilities = AV_CODEC_CAP_DR1, }; #endif + +#if CONFIG_PFM_DECODER +AVCodec ff_pfm_decoder = { + .name = "pfm", + .long_name = NULL_IF_CONFIG_SMALL("PFM (Portable FloatMap) image"), + .type = AVMEDIA_TYPE_VIDEO, + .id = AV_CODEC_ID_PFM, + .priv_data_size = sizeof(PNMContext), + .decode = pnm_decode_frame, + .capabilities = AV_CODEC_CAP_DR1, +}; +#endif diff --git a/libavformat/img2.c b/libavformat/img2.c index 16bc9d2abd..d243d6c125 100644 --- a/libavformat/img2.c +++ b/libavformat/img2.c @@ -40,6 +40,7 @@ const IdStrMap ff_img_tags[] = { { AV_CODEC_ID_PGMYUV, "pgmyuv" }, { AV_CODEC_ID_PBM, "pbm" }, { AV_CODEC_ID_PAM, "pam" }, + { AV_CODEC_ID_PFM, "pfm" }, { AV_CODEC_ID_ALIAS_PIX, "pix" }, { AV_CODEC_ID_DDS, "dds" }, { AV_CODEC_ID_MPEG1VIDEO, "mpg1-img" },