Robert Phillips | deaf568 | 2017-09-06 13:07:21 -0400 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2017 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 <memory> |
| 9 | |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 10 | #include "tools/mdbviz/Model.h" |
Robert Phillips | deaf568 | 2017-09-06 13:07:21 -0400 | [diff] [blame] | 11 | |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 12 | #include "include/core/SkBitmap.h" |
| 13 | #include "include/core/SkCanvas.h" |
| 14 | #include "include/core/SkPicture.h" |
| 15 | #include "include/core/SkStream.h" |
| 16 | #include "tools/debugger/DebugCanvas.h" |
Robert Phillips | deaf568 | 2017-09-06 13:07:21 -0400 | [diff] [blame] | 17 | |
| 18 | Model::Model() : fCurOp(0) { |
| 19 | SkImageInfo ii = SkImageInfo::MakeN32Premul(1024, 1024); |
| 20 | fBM.allocPixels(ii, 0); |
| 21 | } |
| 22 | |
| 23 | Model::~Model() { |
| 24 | this->resetOpList(); |
| 25 | } |
| 26 | |
| 27 | Model::ErrorCode Model::load(const char* filename) { |
| 28 | std::unique_ptr<SkStream> stream = SkStream::MakeFromFile(filename); |
| 29 | if (!stream) { |
| 30 | return ErrorCode::kCouldntOpenFile; |
| 31 | } |
| 32 | sk_sp<SkPicture> pic(SkPicture::MakeFromStream(stream.get())); |
| 33 | if (!pic) { |
| 34 | return ErrorCode::kCouldntDecodeSKP; |
| 35 | } |
| 36 | |
| 37 | { |
Mike Klein | 8f4e224 | 2019-03-20 11:59:00 -0500 | [diff] [blame] | 38 | std::unique_ptr<DebugCanvas> temp( |
| 39 | new DebugCanvas(SkScalarCeilToInt(pic->cullRect().width()), |
| 40 | SkScalarCeilToInt(pic->cullRect().height()))); |
Robert Phillips | deaf568 | 2017-09-06 13:07:21 -0400 | [diff] [blame] | 41 | |
| 42 | temp->setPicture(pic.get()); |
| 43 | pic->playback(temp.get()); |
| 44 | temp->setPicture(nullptr); |
| 45 | this->resetOpList(); |
| 46 | temp->detachCommands(&fOps); |
| 47 | } |
| 48 | |
| 49 | this->setCurOp(fOps.count()-1); |
| 50 | |
| 51 | return ErrorCode::kOK; |
| 52 | } |
| 53 | |
| 54 | const char* Model::ErrorString(ErrorCode err) { |
| 55 | static const char* kStrings[] = { |
| 56 | "OK", |
| 57 | "Couldn't read file", |
| 58 | "Couldn't decode picture" |
| 59 | }; |
| 60 | |
| 61 | return kStrings[(int)err]; |
| 62 | } |
| 63 | |
Robert Phillips | 276066b | 2017-09-06 17:17:44 -0400 | [diff] [blame] | 64 | const char* Model::getOpName(int index) const { |
Mike Klein | 8f4e224 | 2019-03-20 11:59:00 -0500 | [diff] [blame] | 65 | return DrawCommand::GetCommandString(fOps[index]->getType()); |
Robert Phillips | deaf568 | 2017-09-06 13:07:21 -0400 | [diff] [blame] | 66 | } |
| 67 | |
Robert Phillips | 276066b | 2017-09-06 17:17:44 -0400 | [diff] [blame] | 68 | bool Model::isHierarchyPush(int index) const { |
Mike Klein | 8f4e224 | 2019-03-20 11:59:00 -0500 | [diff] [blame] | 69 | DrawCommand::OpType type = fOps[index]->getType(); |
Robert Phillips | 276066b | 2017-09-06 17:17:44 -0400 | [diff] [blame] | 70 | |
Mike Klein | 8f4e224 | 2019-03-20 11:59:00 -0500 | [diff] [blame] | 71 | return DrawCommand::kSave_OpType == type || DrawCommand::kSaveLayer_OpType == type || |
| 72 | DrawCommand::kBeginDrawPicture_OpType == type; |
Robert Phillips | 276066b | 2017-09-06 17:17:44 -0400 | [diff] [blame] | 73 | } |
| 74 | |
| 75 | bool Model::isHierarchyPop(int index) const { |
Mike Klein | 8f4e224 | 2019-03-20 11:59:00 -0500 | [diff] [blame] | 76 | DrawCommand::OpType type = fOps[index]->getType(); |
Robert Phillips | 276066b | 2017-09-06 17:17:44 -0400 | [diff] [blame] | 77 | |
Mike Klein | 8f4e224 | 2019-03-20 11:59:00 -0500 | [diff] [blame] | 78 | return DrawCommand::kRestore_OpType == type || DrawCommand::kEndDrawPicture_OpType == type; |
Robert Phillips | 276066b | 2017-09-06 17:17:44 -0400 | [diff] [blame] | 79 | } |
| 80 | |
Robert Phillips | deaf568 | 2017-09-06 13:07:21 -0400 | [diff] [blame] | 81 | void Model::setCurOp(int curOp) { |
| 82 | SkASSERT(curOp < fOps.count()); |
| 83 | |
| 84 | if (curOp == fCurOp) { |
| 85 | return; // the render state is already up to date |
| 86 | } |
| 87 | |
| 88 | fCurOp = curOp; |
| 89 | this->drawTo(fCurOp); |
| 90 | } |
| 91 | |
| 92 | void Model::drawTo(int index) { |
| 93 | SkASSERT(index < fOps.count()); |
| 94 | |
| 95 | SkCanvas canvas(fBM); |
| 96 | |
| 97 | int saveCount = canvas.save(); |
| 98 | |
| 99 | for (int i = 0; i <= index; ++i) { |
| 100 | if (fOps[i]->isVisible()) { |
| 101 | fOps[i]->execute(&canvas); |
| 102 | } |
| 103 | } |
| 104 | |
| 105 | canvas.restoreToCount(saveCount); |
| 106 | } |
| 107 | |
| 108 | void Model::resetOpList() { |
| 109 | for (int i = 0; i < fOps.count(); ++i) { |
| 110 | delete fOps[i]; |
| 111 | } |
Robert Phillips | 276066b | 2017-09-06 17:17:44 -0400 | [diff] [blame] | 112 | fOps.reset(); |
Robert Phillips | deaf568 | 2017-09-06 13:07:21 -0400 | [diff] [blame] | 113 | fCurOp = 0; |
| 114 | } |