SkJpegEncoderImpl.cpp File Reference

#include "src/encode/SkJpegEncoderImpl.h"
#include "include/core/SkAlphaType.h"
#include "include/core/SkBitmap.h"
#include "include/core/SkColorType.h"
#include "include/core/SkData.h"
#include "include/core/SkImageInfo.h"
#include "include/core/SkPixmap.h"
#include "include/core/SkRefCnt.h"
#include "include/core/SkStream.h"
#include "include/core/SkYUVAInfo.h"
#include "include/core/SkYUVAPixmaps.h"
#include "include/encode/SkEncoder.h"
#include "include/encode/SkJpegEncoder.h"
#include "include/private/base/SkAssert.h"
#include "include/private/base/SkNoncopyable.h"
#include "include/private/base/SkTemplates.h"
#include "src/base/SkMSAN.h"
#include "src/codec/SkJpegConstants.h"
#include "src/codec/SkJpegPriv.h"
#include "src/encode/SkImageEncoderFns.h"
#include "src/encode/SkImageEncoderPriv.h"
#include "src/encode/SkJPEGWriteUtility.h"
#include "src/image/SkImage_Base.h"
#include <csetjmp>
#include <cstdint>
#include <cstring>
#include <memory>
#include <utility>
#include "jpeglib.h"

Go to the source code of this file.

Classes
class	SkJpegEncoderMgr

Namespaces
namespace	SkJpegEncoder

Functions
static void	yuva_copy_row (const SkYUVAPixmaps *src, int row, uint8_t *dst)
static std::unique_ptr< SkEncoder >	Make (SkWStream *dst, const SkPixmap *src, const SkYUVAPixmaps *srcYUVA, const SkColorSpace *srcYUVAColorSpace, const SkJpegEncoder::Options &options)
SK_API bool	SkJpegEncoder::Encode (SkWStream *dst, const SkPixmap &src, const Options &options)
SK_API bool	SkJpegEncoder::Encode (SkWStream *dst, const SkYUVAPixmaps &src, const SkColorSpace *srcColorSpace, const Options &options)
SK_API sk_sp< SkData >	SkJpegEncoder::Encode (GrDirectContext *ctx, const SkImage *img, const Options &options)
SK_API std::unique_ptr< SkEncoder >	SkJpegEncoder::Make (SkWStream *dst, const SkPixmap &src, const Options &options)
SK_API std::unique_ptr< SkEncoder >	SkJpegEncoder::Make (SkWStream *dst, const SkYUVAPixmaps &src, const SkColorSpace *srcColorSpace, const Options &options)

Function Documentation

Make()

static std::unique_ptr< SkEncoder > Make ( SkWStream *  dst, const SkPixmap *  src, const SkYUVAPixmaps *  srcYUVA, const SkColorSpace *  srcYUVAColorSpace, const SkJpegEncoder::Options &  options  )

static

Definition at line 259 of file SkJpegEncoderImpl.cpp.

                                                                            {
    // Exactly one of |src| or |srcYUVA| should be specified.
    if (srcYUVA) {
        SkASSERT(!src);
        if (!srcYUVA->isValid()) {
            return nullptr;
        }
    } else {
        SkASSERT(src);
        if (!src || !SkPixmapIsValid(*src)) {
            return nullptr;
        }
    }
 
    std::unique_ptr<SkJpegEncoderMgr> encoderMgr = SkJpegEncoderMgr::Make(dst);
 
    skjpeg_error_mgr::AutoPushJmpBuf jmp(encoderMgr->errorMgr());
    if (setjmp(jmp)) {
        return nullptr;
    }
 
    if (srcYUVA) {
        if (!encoderMgr->setParams(srcYUVA->pixmapsInfo(), options)) {
            return nullptr;
        }
    } else {
        if (!encoderMgr->setParams(src->info(), options)) {
            return nullptr;
        }
    }
 
    jpeg_set_quality(encoderMgr->cinfo(), options.fQuality, TRUE);
    jpeg_start_compress(encoderMgr->cinfo(), TRUE);
 
    // Write XMP metadata. This will only write the standard XMP segment.
    // TODO(ccameron): Split this into a standard and extended XMP segment if needed.
    if (options.xmpMetadata) {
        SkDynamicMemoryWStream s;
        s.write(kXMPStandardSig, sizeof(kXMPStandardSig));
        s.write(options.xmpMetadata->data(), options.xmpMetadata->size());
        auto data = s.detachAsData();
        jpeg_write_marker(encoderMgr->cinfo(), kXMPMarker, data->bytes(), data->size());
    }
 
    // Write the ICC profile.
    // TODO(ccameron): This limits ICC profile size to a single segment's parameters (less than
    // 64k). Split larger profiles into more segments.
    sk_sp<SkData> icc = icc_from_color_space(srcYUVA ? srcYUVAColorSpace : src->colorSpace(),
                                             options.fICCProfile,
                                             options.fICCProfileDescription);
    if (icc) {
        // Create a contiguous block of memory with the icc signature followed by the profile.
        sk_sp<SkData> markerData = SkData::MakeUninitialized(kICCMarkerHeaderSize + icc->size());
        uint8_t* ptr = (uint8_t*)markerData->writable_data();
        memcpy(ptr, kICCSig, sizeof(kICCSig));
        ptr += sizeof(kICCSig);
        *ptr++ = 1;  // This is the first marker.
        *ptr++ = 1;  // Out of one total markers.
        memcpy(ptr, icc->data(), icc->size());
 
        jpeg_write_marker(encoderMgr->cinfo(), kICCMarker, markerData->bytes(), markerData->size());
    }
 
    if (srcYUVA) {
        return std::make_unique<SkJpegEncoderImpl>(std::move(encoderMgr), srcYUVA);
    }
    return std::make_unique<SkJpegEncoderImpl>(std::move(encoderMgr), *src);
}

yuva_copy_row()

static void yuva_copy_row ( const SkYUVAPixmaps *  src, int  row, uint8_t *  dst  )

static

Definition at line 183 of file SkJpegEncoderImpl.cpp.

                                                                           {
    int width = src->plane(0).width();
    switch (src->yuvaInfo().planeConfig()) {
        case SkYUVAInfo::PlaneConfig::kY_U_V: {
            auto [ssWidthU, ssHeightU] = src->yuvaInfo().planeSubsamplingFactors(1);
            auto [ssWidthV, ssHeightV] = src->yuvaInfo().planeSubsamplingFactors(2);
            const uint8_t* srcY = reinterpret_cast<const uint8_t*>(src->plane(0).addr(0, row));
            const uint8_t* srcU =
                    reinterpret_cast<const uint8_t*>(src->plane(1).addr(0, row / ssHeightU));
            const uint8_t* srcV =
                    reinterpret_cast<const uint8_t*>(src->plane(2).addr(0, row / ssHeightV));
            for (int col = 0; col < width; ++col) {
                dst[3 * col + 0] = srcY[col];
                dst[3 * col + 1] = srcU[col / ssWidthU];
                dst[3 * col + 2] = srcV[col / ssWidthV];
            }
            break;
        }
        case SkYUVAInfo::PlaneConfig::kY_UV: {
            auto [ssWidthUV, ssHeightUV] = src->yuvaInfo().planeSubsamplingFactors(1);
            const uint8_t* srcY = reinterpret_cast<const uint8_t*>(src->plane(0).addr(0, row));
            const uint8_t* srcUV =
                    reinterpret_cast<const uint8_t*>(src->plane(1).addr(0, row / ssHeightUV));
            for (int col = 0; col < width; ++col) {
                dst[3 * col + 0] = srcY[col];
                dst[3 * col + 1] = srcUV[2 * (col / ssWidthUV) + 0];
                dst[3 * col + 2] = srcUV[2 * (col / ssWidthUV) + 1];
            }
            break;
        }
        default:
            break;
    }
}