94 lines
3.0 KiB
C++
94 lines
3.0 KiB
C++
|
/*
|
||
|
|
* Copyright 2017 Google Inc.
|
||
|
|
*
|
||
|
|
* Use of this source code is governed by a BSD-style license that can be
|
||
|
|
* found in the LICENSE file.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include "include/codec/SkCodec.h"
|
||
|
|
#include "include/core/SkBitmap.h"
|
||
|
|
#include "include/core/SkData.h"
|
||
|
|
#include "include/core/SkImageInfo.h"
|
||
|
|
#include "include/core/SkRefCnt.h"
|
||
|
|
#include "include/core/SkStream.h"
|
||
|
|
#include "include/private/SkTemplates.h"
|
||
|
|
#include "tests/Test.h"
|
||
|
|
#include "tools/Resources.h"
|
||
|
|
|
||
|
|
#include <cstring>
|
||
|
|
#include <initializer_list>
|
||
|
|
#include <memory>
|
||
|
|
#include <utility>
|
||
|
|
|
||
|
|
namespace {
|
||
|
|
// This class wraps another SkStream. It does not own the underlying stream, so
|
||
|
|
// that the underlying stream can be reused starting from where the first
|
||
|
|
// client left off. This mimics Android's JavaInputStreamAdaptor.
|
||
|
|
class UnowningStream : public SkStream {
|
||
|
|
public:
|
||
|
|
explicit UnowningStream(SkStream* stream)
|
||
|
|
: fStream(stream)
|
||
|
|
{}
|
||
|
|
|
||
|
|
size_t read(void* buf, size_t bytes) override {
|
||
|
|
return fStream->read(buf, bytes);
|
||
|
|
}
|
||
|
|
|
||
|
|
bool rewind() override {
|
||
|
|
return fStream->rewind();
|
||
|
|
}
|
||
|
|
|
||
|
|
bool isAtEnd() const override {
|
||
|
|
return fStream->isAtEnd();
|
||
|
|
}
|
||
|
|
private:
|
||
|
|
SkStream* fStream; // Unowned.
|
||
|
|
};
|
||
|
|
} // namespace
|
||
|
|
|
||
|
|
// Test that some SkCodecs do not attempt to read input beyond the logical
|
||
|
|
// end of the data. Some other SkCodecs do, but some Android apps rely on not
|
||
|
|
// doing so for PNGs. Test on other formats that work.
|
||
|
|
DEF_TEST(Codec_end, r) {
|
||
|
|
for (const char* path : { "images/plane.png",
|
||
|
|
"images/yellow_rose.png",
|
||
|
|
"images/plane_interlaced.png",
|
||
|
|
"images/mandrill.wbmp",
|
||
|
|
"images/randPixels.bmp",
|
||
|
|
}) {
|
||
|
|
sk_sp<SkData> data = GetResourceAsData(path);
|
||
|
|
if (!data) {
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
const int kNumImages = 2;
|
||
|
|
const size_t size = data->size();
|
||
|
|
sk_sp<SkData> multiData = SkData::MakeUninitialized(size * kNumImages);
|
||
|
|
void* dst = multiData->writable_data();
|
||
|
|
for (int i = 0; i < kNumImages; i++) {
|
||
|
|
memcpy(SkTAddOffset<void>(dst, size * i), data->data(), size);
|
||
|
|
}
|
||
|
|
data.reset();
|
||
|
|
|
||
|
|
SkMemoryStream stream(std::move(multiData));
|
||
|
|
for (int i = 0; i < kNumImages; ++i) {
|
||
|
|
std::unique_ptr<SkCodec> codec(SkCodec::MakeFromStream(
|
||
|
|
std::make_unique<UnowningStream>(&stream)));
|
||
|
|
if (!codec) {
|
||
|
|
ERRORF(r, "Failed to create a codec from %s, iteration %i", path, i);
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
auto info = codec->getInfo().makeColorType(kN32_SkColorType);
|
||
|
|
SkBitmap bm;
|
||
|
|
bm.allocPixels(info);
|
||
|
|
|
||
|
|
auto result = codec->getPixels(bm.info(), bm.getPixels(), bm.rowBytes());
|
||
|
|
if (result != SkCodec::kSuccess) {
|
||
|
|
ERRORF(r, "Failed to getPixels from %s, iteration %i error %i", path, i, result);
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|