#include "DMRecordTask.h"
#include "DMUtil.h"
#include "DMWriteTask.h"
#include "SkCommandLineFlags.h"
#include "SkRecord.h"
#include "SkRecordDraw.h"
#include "SkRecordOpts.h"
#include "SkRecorder.h"

DEFINE_bool(skr, true, "If true, run SKR tests.");

namespace DM {

RecordTask::RecordTask(const Task& parent, skiagm::GM* gm, SkBitmap reference, Mode mode)
    : CpuTask(parent)
    , fOptimize(mode == kOptimize_Mode)
    , fName(UnderJoin(parent.name().c_str(), fOptimize ? "skr" : "skr-noopt"))
    , fGM(gm)
    , fReference(reference)
    {}

RecordTask::RecordTask(const Task& parent, SkPicture* pic, SkBitmap reference, Mode mode)
    : CpuTask(parent)
    , fOptimize(mode == kOptimize_Mode)
    , fName(UnderJoin(parent.name().c_str(), fOptimize ? "skr" : "skr-noopt"))
    , fPicture(SkRef(pic))
    , fReference(reference)
    {}

void RecordTask::draw() {
    // Record into an SkRecord.
    SkRecord record;
    SkRecorder recorder(SkRecorder::kWriteOnly_Mode, &record,
                        fReference.width(), fReference.height());

    if (fGM.get()) {
        recorder.concat(fGM->getInitialTransform());
        fGM->draw(&recorder);
    } else {
        fPicture->draw(&recorder);
    }


    if (fOptimize) {
        SkRecordOptimize(&record);
    }

    // Draw the SkRecord back into a bitmap.
    SkBitmap bitmap;
    AllocatePixels(fReference, &bitmap);
    SkCanvas target(bitmap);
    SkRecordDraw(record, &target);

    if (!BitmapsEqual(bitmap, fReference)) {
        this->fail();
        this->spawnChild(SkNEW_ARGS(WriteTask, (*this, bitmap)));
    }
}

bool RecordTask::shouldSkip() const {
    return !FLAGS_skr;
}

}  // namespace DM