Index: dm/DMWriteTask.cpp |
diff --git a/dm/DMWriteTask.cpp b/dm/DMWriteTask.cpp |
index b8ce3189a843ee6ce0be45d04d4b2ee37b917de8..149071276cf581cddcb11215655725b2fb1da37b 100644 |
--- a/dm/DMWriteTask.cpp |
+++ b/dm/DMWriteTask.cpp |
@@ -41,10 +41,10 @@ WriteTask::WriteTask(const Task& parent, SkBitmap bitmap) |
, fData(NULL) |
, fExtension(".png") {} |
-WriteTask::WriteTask(const Task& parent, SkData *data, const char* ext) |
+WriteTask::WriteTask(const Task& parent, SkStreamAsset *data, const char* ext) |
: CpuTask(parent) |
, fGmName(find_gm_name(parent, &fSuffixes)) |
- , fData(SkRef(data)) |
+ , fData(data) |
, fExtension(ext) {} |
void WriteTask::makeDirOrFail(SkString dir) { |
@@ -115,9 +115,10 @@ struct PngAndRaw { |
}; |
// Does not take ownership of data. |
-bool save_data_to_file(const SkData* data, const char* path) { |
+bool save_data_to_file(SkStreamAsset* data, const char* path) { |
+ data->rewind(); |
SkFILEWStream stream(path); |
- if (!stream.isValid() || !stream.write(data->data(), data->size())) { |
+ if (!stream.isValid() || !stream.writeStream(data, data->getLength())) { |
SkDebugf("Can't write %s.\n", path); |
return false; |
} |