OLD | NEW |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "cc/test/picture_cache_model.h" | 5 #include "cc/test/picture_cache_model.h" |
6 | 6 |
7 #include "base/logging.h" | 7 #include "base/logging.h" |
8 #include "third_party/skia/include/core/SkData.h" | 8 #include "third_party/skia/include/core/SkData.h" |
9 #include "third_party/skia/include/core/SkPicture.h" | 9 #include "third_party/skia/include/core/SkPicture.h" |
10 #include "third_party/skia/include/core/SkRefCnt.h" | 10 #include "third_party/skia/include/core/SkRefCnt.h" |
11 #include "third_party/skia/include/core/SkStream.h" | |
12 | 11 |
13 class SkPixelSerializer; | 12 class SkPixelSerializer; |
14 | 13 |
15 namespace cc { | 14 namespace cc { |
16 namespace { | 15 namespace { |
17 | 16 |
18 sk_sp<SkPicture> CopySkPicture(const SkPicture* picture) { | 17 sk_sp<SkPicture> CopySkPicture(const SkPicture* picture) { |
19 SkDynamicMemoryWStream write_stream; | 18 sk_sp<SkData> data = picture->serialize(); |
20 picture->serialize(&write_stream, nullptr); | 19 DCHECK_GT(data->size(), 0u); |
21 DCHECK_GT(write_stream.bytesWritten(), 0u); | 20 return SkPicture::MakeFromData(data.get()); |
22 | |
23 sk_sp<SkData> data(write_stream.copyToData()); | |
24 | |
25 SkMemoryStream read_stream(data); | |
26 return SkPicture::MakeFromStream(&read_stream, nullptr); | |
27 } | 21 } |
28 | 22 |
29 } // namespace | 23 } // namespace |
30 | 24 |
31 PictureCacheModel::PictureCacheModel() = default; | 25 PictureCacheModel::PictureCacheModel() = default; |
32 | 26 |
33 PictureCacheModel::~PictureCacheModel() = default; | 27 PictureCacheModel::~PictureCacheModel() = default; |
34 | 28 |
35 void PictureCacheModel::AddPicture(const SkPicture* picture) { | 29 void PictureCacheModel::AddPicture(const SkPicture* picture) { |
36 sk_sp<SkPicture> picture_copy = CopySkPicture(picture); | 30 sk_sp<SkPicture> picture_copy = CopySkPicture(picture); |
37 pictures_.insert(std::make_pair(picture->uniqueID(), picture_copy)); | 31 pictures_.insert(std::make_pair(picture->uniqueID(), picture_copy)); |
38 } | 32 } |
39 | 33 |
40 sk_sp<const SkPicture> PictureCacheModel::GetPicture(uint32_t unique_id) { | 34 sk_sp<const SkPicture> PictureCacheModel::GetPicture(uint32_t unique_id) { |
41 if (pictures_.find(unique_id) == pictures_.end()) | 35 if (pictures_.find(unique_id) == pictures_.end()) |
42 return nullptr; | 36 return nullptr; |
43 | 37 |
44 return pictures_.find(unique_id)->second; | 38 return pictures_.find(unique_id)->second; |
45 } | 39 } |
46 | 40 |
47 } // namespace cc | 41 } // namespace cc |
OLD | NEW |