1/*
2 * Copyright 2016 Google Inc.
3 *
4 * Use of this source code is governed by a BSD-style license that can be
5 * found in the LICENSE file.
6 */
7#ifndef CodecPriv_DEFINED
8#define CodecPriv_DEFINED
9
10#include "SkBitmap.h"
11#include "SkCodec.h"
12#include "SkData.h"
13
14inline bool decode_memory(const void* mem, size_t size, SkBitmap* bm) {
15    std::unique_ptr<SkCodec> codec(SkCodec::NewFromData(SkData::MakeWithoutCopy(mem, size)));
16    if (!codec) {
17        return false;
18    }
19
20    // Construct a color table for the decode if necessary
21    sk_sp<SkColorTable> colorTable(nullptr);
22    SkPMColor* colorPtr = nullptr;
23    int* colorCountPtr = nullptr;
24    int maxColors = 256;
25    if (kIndex_8_SkColorType == codec->getInfo().colorType()) {
26        SkPMColor colors[256];
27        colorTable.reset(new SkColorTable(colors, maxColors));
28        colorPtr = const_cast<SkPMColor*>(colorTable->readColors());
29        colorCountPtr = &maxColors;
30    }
31
32    bm->allocPixels(codec->getInfo(), nullptr, colorTable.get());
33    const SkCodec::Result result = codec->getPixels(codec->getInfo(), bm->getPixels(),
34            bm->rowBytes(), nullptr, colorPtr, colorCountPtr);
35    return result == SkCodec::kSuccess || result == SkCodec::kIncompleteInput;
36}
37#endif  // CodecPriv_DEFINED
38