You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

62 lines
1.3 KiB

/*
* Copyright 2018 Google Inc.
*
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
#ifndef sk_pixel_iter_DEFINED
#define sk_pixel_iter_DEFINED
#include "SkPixmap.h"
#include "SkSurface.h"
namespace sk_tool_utils {
class PixelIter {
public:
PixelIter();
PixelIter(SkSurface* surf) {
SkPixmap pm;
if (!surf->peekPixels(&pm)) {
pm.reset();
}
this->reset(pm);
}
void reset(const SkPixmap& pm) {
fPM = pm;
fLoc = { -1, 0 };
}
void* next(SkIPoint* loc = nullptr) {
if (!fPM.addr()) {
return nullptr;
}
fLoc.fX += 1;
if (fLoc.fX >= fPM.width()) {
fLoc.fX = 0;
if (++fLoc.fY >= fPM.height()) {
this->setDone();
return nullptr;
}
}
if (loc) {
*loc = fLoc;
}
return fPM.writable_addr(fLoc.fX, fLoc.fY);
}
void setDone() {
fPM.reset();
}
private:
SkPixmap fPM;
SkIPoint fLoc;
};
} // namespace sk_tool_utils
#endif // sk_tool_utils_DEFINED