mirror of
https://pdfium.googlesource.com/pdfium
synced 2026-08-02 20:37:52 +08:00
68f94c8c47
Pass by value and use std::move() when appropriate. Then several local RetainPtrs can be removed. Change-Id: Id23dbdf9fd6b24ecde7b27d74f3d5123fdbaaa0c Reviewed-on: https://pdfium-review.googlesource.com/c/pdfium/+/115413 Reviewed-by: Tom Sepez <tsepez@chromium.org> Commit-Queue: Lei Zhang <thestig@chromium.org>
85 lines
2.7 KiB
C++
85 lines
2.7 KiB
C++
// Copyright 2019 The PDFium Authors
|
|
// Use of this source code is governed by a BSD-style license that can be
|
|
// found in the LICENSE file.
|
|
|
|
#include "public/fpdf_thumbnail.h"
|
|
|
|
#include <utility>
|
|
|
|
#include "core/fpdfapi/page/cpdf_dib.h"
|
|
#include "core/fpdfapi/page/cpdf_page.h"
|
|
#include "core/fpdfapi/parser/cpdf_dictionary.h"
|
|
#include "core/fpdfapi/parser/cpdf_stream.h"
|
|
#include "core/fpdfapi/parser/cpdf_stream_acc.h"
|
|
#include "core/fxge/dib/cfx_dibitmap.h"
|
|
#include "fpdfsdk/cpdfsdk_helpers.h"
|
|
#include "public/fpdfview.h"
|
|
|
|
namespace {
|
|
|
|
RetainPtr<const CPDF_Stream> CPDFStreamForThumbnailFromPage(FPDF_PAGE page) {
|
|
const CPDF_Page* pdf_page = CPDFPageFromFPDFPage(page);
|
|
if (!pdf_page)
|
|
return nullptr;
|
|
|
|
RetainPtr<const CPDF_Dictionary> page_dict = pdf_page->GetDict();
|
|
if (!page_dict->KeyExist("Type"))
|
|
return nullptr;
|
|
|
|
return page_dict->GetStreamFor("Thumb");
|
|
}
|
|
|
|
} // namespace
|
|
|
|
FPDF_EXPORT unsigned long FPDF_CALLCONV
|
|
FPDFPage_GetDecodedThumbnailData(FPDF_PAGE page,
|
|
void* buffer,
|
|
unsigned long buflen) {
|
|
RetainPtr<const CPDF_Stream> thumb_stream =
|
|
CPDFStreamForThumbnailFromPage(page);
|
|
if (!thumb_stream)
|
|
return 0u;
|
|
|
|
return DecodeStreamMaybeCopyAndReturnLength(
|
|
std::move(thumb_stream),
|
|
{static_cast<uint8_t*>(buffer), static_cast<size_t>(buflen)});
|
|
}
|
|
|
|
FPDF_EXPORT unsigned long FPDF_CALLCONV
|
|
FPDFPage_GetRawThumbnailData(FPDF_PAGE page,
|
|
void* buffer,
|
|
unsigned long buflen) {
|
|
RetainPtr<const CPDF_Stream> thumb_stream =
|
|
CPDFStreamForThumbnailFromPage(page);
|
|
if (!thumb_stream)
|
|
return 0u;
|
|
|
|
return GetRawStreamMaybeCopyAndReturnLength(
|
|
std::move(thumb_stream),
|
|
{static_cast<uint8_t*>(buffer), static_cast<size_t>(buflen)});
|
|
}
|
|
|
|
FPDF_EXPORT FPDF_BITMAP FPDF_CALLCONV
|
|
FPDFPage_GetThumbnailAsBitmap(FPDF_PAGE page) {
|
|
RetainPtr<const CPDF_Stream> thumb_stream =
|
|
CPDFStreamForThumbnailFromPage(page);
|
|
if (!thumb_stream)
|
|
return nullptr;
|
|
|
|
const CPDF_Page* pdf_page = CPDFPageFromFPDFPage(page);
|
|
auto dib_source = pdfium::MakeRetain<CPDF_DIB>(pdf_page->GetDocument(),
|
|
std::move(thumb_stream));
|
|
const CPDF_DIB::LoadState start_status = dib_source->StartLoadDIBBase(
|
|
false, nullptr, pdf_page->GetPageResources().Get(), false,
|
|
CPDF_ColorSpace::Family::kUnknown, false, {0, 0});
|
|
if (start_status == CPDF_DIB::LoadState::kFail)
|
|
return nullptr;
|
|
|
|
auto thumb_bitmap = pdfium::MakeRetain<CFX_DIBitmap>();
|
|
if (!thumb_bitmap->Copy(std::move(dib_source))) {
|
|
return nullptr;
|
|
}
|
|
|
|
return FPDFBitmapFromCFXDIBitmap(thumb_bitmap.Leak());
|
|
}
|