| OLD | NEW |
| 1 #ifndef DMWriteTask_DEFINED | 1 #ifndef DMWriteTask_DEFINED |
| 2 #define DMWriteTask_DEFINED | 2 #define DMWriteTask_DEFINED |
| 3 | 3 |
| 4 #include "DMExpectations.h" | 4 #include "DMExpectations.h" |
| 5 #include "DMTask.h" | 5 #include "DMTask.h" |
| 6 #include "SkBitmap.h" | 6 #include "SkBitmap.h" |
| 7 #include "SkString.h" | 7 #include "SkString.h" |
| 8 #include "SkTArray.h" | 8 #include "SkTArray.h" |
| 9 | 9 |
| 10 | 10 |
| 11 // Writes a bitmap to a file. | 11 // Writes a bitmap to a file. |
| 12 | 12 |
| 13 namespace DM { | 13 namespace DM { |
| 14 | 14 |
| 15 class WriteTask : public CpuTask { | 15 class WriteTask : public CpuTask { |
| 16 | 16 |
| 17 public: | 17 public: |
| 18 WriteTask(const Task& parent, // WriteTask must be a child Task. Pass its
parent here. | 18 enum Mode { |
| 19 SkBitmap bitmap); // Bitmap to write. | 19 kParseName_Mode, // Parse the parent's name into directories by undersc
ores. |
| 20 kVerbatim_Mode, // Don't parse the name at all. |
| 21 }; |
| 22 WriteTask(const Task& parent, // WriteTask must be a child Task. Pass it
s parent here. |
| 23 SkBitmap bitmap, // Bitmap to write. |
| 24 Mode = kParseName_Mode); |
| 20 | 25 |
| 21 virtual void draw() SK_OVERRIDE; | 26 virtual void draw() SK_OVERRIDE; |
| 22 virtual bool shouldSkip() const SK_OVERRIDE; | 27 virtual bool shouldSkip() const SK_OVERRIDE; |
| 23 virtual SkString name() const SK_OVERRIDE; | 28 virtual SkString name() const SK_OVERRIDE; |
| 24 | 29 |
| 25 // Reads image files WriteTask wrote under root and compares them with bitma
p. | 30 // Reads image files WriteTask wrote under root and compares them with bitma
p. |
| 26 class Expectations : public DM::Expectations { | 31 class Expectations : public DM::Expectations { |
| 27 public: | 32 public: |
| 28 explicit Expectations(const char* root) : fRoot(root) {} | 33 explicit Expectations(const char* root) : fRoot(root) {} |
| 29 | 34 |
| 30 bool check(const Task& task, SkBitmap bitmap) const SK_OVERRIDE; | 35 bool check(const Task& task, SkBitmap bitmap) const SK_OVERRIDE; |
| 31 private: | 36 private: |
| 32 const char* fRoot; | 37 const char* fRoot; |
| 33 }; | 38 }; |
| 34 | 39 |
| 35 private: | 40 private: |
| 36 SkTArray<SkString> fSuffixes; | 41 SkTArray<SkString> fSuffixes; |
| 37 SkString fGmName; | 42 SkString fGmName; |
| 38 const SkBitmap fBitmap; | 43 const SkBitmap fBitmap; |
| 39 | 44 |
| 40 void makeDirOrFail(SkString dir); | 45 void makeDirOrFail(SkString dir); |
| 41 }; | 46 }; |
| 42 | 47 |
| 43 } // namespace DM | 48 } // namespace DM |
| 44 | 49 |
| 45 #endif // DMWriteTask_DEFINED | 50 #endif // DMWriteTask_DEFINED |
| OLD | NEW |