85 lines
2.4 KiB
C++
85 lines
2.4 KiB
C++
|
#include "DMPipeTask.h"
|
||
|
#include "DMUtil.h"
|
||
|
#include "DMWriteTask.h"
|
||
|
|
||
|
#include "SamplePipeControllers.h"
|
||
|
#include "SkCommandLineFlags.h"
|
||
|
#include "SkGPipe.h"
|
||
|
|
||
|
DEFINE_bool(pipe, false, "If true, check several pipe variants against the reference bitmap.");
|
||
|
|
||
|
namespace DM {
|
||
|
|
||
|
static uint32_t get_flags(bool crossProcess, bool sharedAddressSpace) {
|
||
|
SkASSERT(!(!crossProcess && sharedAddressSpace));
|
||
|
uint32_t flags = 0;
|
||
|
if (crossProcess) {
|
||
|
flags |= SkGPipeWriter::kCrossProcess_Flag;
|
||
|
if (sharedAddressSpace) {
|
||
|
flags |= SkGPipeWriter::kSharedAddressSpace_Flag;
|
||
|
}
|
||
|
}
|
||
|
return flags;
|
||
|
}
|
||
|
|
||
|
static const char* get_name(const uint32_t flags) {
|
||
|
if (flags & SkGPipeWriter::kCrossProcess_Flag &&
|
||
|
flags & SkGPipeWriter::kSharedAddressSpace_Flag) {
|
||
|
return "cross_process_shared_address_space_pipe";
|
||
|
} else if (flags & SkGPipeWriter::kCrossProcess_Flag) {
|
||
|
return "cross_process_pipe";
|
||
|
} else {
|
||
|
return "pipe";
|
||
|
}
|
||
|
}
|
||
|
|
||
|
PipeTask::PipeTask(const Task& parent,
|
||
|
skiagm::GM* gm,
|
||
|
SkBitmap reference,
|
||
|
bool crossProcess,
|
||
|
bool sharedAddressSpace)
|
||
|
: Task(parent)
|
||
|
, fFlags(get_flags(crossProcess, sharedAddressSpace))
|
||
|
, fName(UnderJoin(parent.name().c_str(), get_name(fFlags)))
|
||
|
, fGM(gm)
|
||
|
, fReference(reference)
|
||
|
{}
|
||
|
|
||
|
void PipeTask::draw() {
|
||
|
SkBitmap bitmap;
|
||
|
SetupBitmap(fReference.config(), fGM.get(), &bitmap);
|
||
|
|
||
|
SkCanvas canvas(bitmap);
|
||
|
PipeController pipeController(&canvas, &SkImageDecoder::DecodeMemory);
|
||
|
SkGPipeWriter writer;
|
||
|
|
||
|
SkCanvas* pipeCanvas = writer.startRecording(&pipeController,
|
||
|
fFlags,
|
||
|
bitmap.width(),
|
||
|
bitmap.height());
|
||
|
pipeCanvas->concat(fGM->getInitialTransform());
|
||
|
fGM->draw(pipeCanvas);
|
||
|
writer.endRecording();
|
||
|
|
||
|
if (!BitmapsEqual(bitmap, fReference)) {
|
||
|
this->fail();
|
||
|
this->spawnChild(SkNEW_ARGS(WriteTask, (*this, bitmap)));
|
||
|
}
|
||
|
}
|
||
|
|
||
|
bool PipeTask::shouldSkip() const {
|
||
|
if (!FLAGS_pipe) {
|
||
|
return true;
|
||
|
}
|
||
|
if (fGM->getFlags() & skiagm::GM::kSkipPipe_Flag) {
|
||
|
return true;
|
||
|
}
|
||
|
if (fFlags == SkGPipeWriter::kCrossProcess_Flag &&
|
||
|
fGM->getFlags() & skiagm::GM::kSkipPipeCrossProcess_Flag) {
|
||
|
return true;
|
||
|
}
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
} // namespace DM
|