1 /*
2 * Copyright 2014 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
8 #include "include/core/SkBitmap.h"
9 #include "include/core/SkColorSpace.h"
10 #include "include/core/SkData.h"
11 #include "include/core/SkImage.h"
12 #include "include/core/SkImageGenerator.h"
13 #include "include/core/SkStream.h"
14 #include "include/core/SkTypeface.h"
15 #include "src/core/SkOSFile.h"
16 #include "src/utils/SkOSPath.h"
17 #include "tools/ResourceFactory.h"
18 #include "tools/Resources.h"
19 #include "tools/flags/CommandLineFlags.h"
20
21 static DEFINE_string2(resourcePath, i, "resources",
22 "Directory with test resources: images, fonts, etc.");
23
24 sk_sp<SkData> (*gResourceFactory)(const char*) = nullptr;
25
GetResourcePath(const char * resource)26 SkString GetResourcePath(const char* resource) {
27 return SkOSPath::Join(FLAGS_resourcePath[0], resource);
28 }
29
SetResourcePath(const char * resource)30 void SetResourcePath(const char* resource) {
31 FLAGS_resourcePath.set(0, resource);
32 }
33
DecodeDataToBitmap(sk_sp<SkData> data,SkBitmap * dst)34 bool DecodeDataToBitmap(sk_sp<SkData> data, SkBitmap* dst) {
35 std::unique_ptr<SkImageGenerator> gen(SkImageGenerator::MakeFromEncoded(std::move(data)));
36 return gen && dst->tryAllocPixels(gen->getInfo()) &&
37 gen->getPixels(gen->getInfo().makeColorSpace(nullptr), dst->getPixels(), dst->rowBytes());
38 }
39
GetResourceAsStream(const char * resource,bool useFileStream)40 std::unique_ptr<SkStreamAsset> GetResourceAsStream(const char* resource, bool useFileStream) {
41 if (useFileStream) {
42 auto path = GetResourcePath(resource);
43 return SkFILEStream::Make(path.c_str());
44 } else {
45 auto data = GetResourceAsData(resource);
46 return data ? std::unique_ptr<SkStreamAsset>(new SkMemoryStream(std::move(data)))
47 : nullptr;
48 }
49 }
50
GetResourceAsData(const char * resource)51 sk_sp<SkData> GetResourceAsData(const char* resource) {
52 if (sk_sp<SkData> data = gResourceFactory
53 ? gResourceFactory(resource)
54 : SkData::MakeFromFileName(GetResourcePath(resource).c_str())) {
55 return data;
56 }
57 SkDebugf("Resource \"%s\" not found.\n", GetResourcePath(resource).c_str());
58 #ifdef SK_TOOLS_REQUIRE_RESOURCES
59 SK_ABORT("missing resource");
60 #endif
61 return nullptr;
62 }
63
MakeResourceAsTypeface(const char * resource,int ttcIndex)64 sk_sp<SkTypeface> MakeResourceAsTypeface(const char* resource, int ttcIndex) {
65 return SkTypeface::MakeFromStream(GetResourceAsStream(resource), ttcIndex);
66 }
67