/*
 * Copyright 2011 Google Inc.
 *
 * Use of this source code is governed by a BSD-style license that can be
 * found in the LICENSE file.
 */

#include "SkCanvas.h"
#include "SkDrawLooper.h"
#include "Test.h"

/*
 *  Subclass of looper that just draws once, with an offset in X.
 */
class TestLooper : public SkDrawLooper {
public:
    bool fOnce;

    virtual void init(SkCanvas*) SK_OVERRIDE {
        fOnce = true;
    }

    virtual bool next(SkCanvas* canvas, SkPaint*) SK_OVERRIDE {
        if (fOnce) {
            fOnce = false;
            canvas->translate(SkIntToScalar(10), 0);
            return true;
        }
        return false;
    }

#ifdef SK_DEVELOPER
    virtual void toString(SkString* str) const SK_OVERRIDE {
        str->append("TestLooper:");
    }
#endif

    SK_DECLARE_UNFLATTENABLE_OBJECT()
};

static void test_drawBitmap(skiatest::Reporter* reporter) {
    SkBitmap src;
    src.setConfig(SkBitmap::kARGB_8888_Config, 10, 10);
    src.allocPixels();
    src.eraseColor(SK_ColorWHITE);

    SkBitmap dst;
    dst.setConfig(SkBitmap::kARGB_8888_Config, 10, 10);
    dst.allocPixels();
    dst.eraseColor(SK_ColorTRANSPARENT);

    SkCanvas canvas(dst);
    SkPaint  paint;

    // we are initially transparent
    REPORTER_ASSERT(reporter, 0 == *dst.getAddr32(5, 5));

    // we see the bitmap drawn
    canvas.drawBitmap(src, 0, 0, &paint);
    REPORTER_ASSERT(reporter, 0xFFFFFFFF == *dst.getAddr32(5, 5));

    // reverify we are clear again
    dst.eraseColor(SK_ColorTRANSPARENT);
    REPORTER_ASSERT(reporter, 0 == *dst.getAddr32(5, 5));

    // if the bitmap is clipped out, we don't draw it
    canvas.drawBitmap(src, SkIntToScalar(-10), 0, &paint);
    REPORTER_ASSERT(reporter, 0 == *dst.getAddr32(5, 5));

    // now install our looper, which will draw, since it internally translates
    // to the left. The test is to ensure that canvas' quickReject machinary
    // allows us through, even though sans-looper we would look like we should
    // be clipped out.
    paint.setLooper(new TestLooper)->unref();
    canvas.drawBitmap(src, SkIntToScalar(-10), 0, &paint);
    REPORTER_ASSERT(reporter, 0xFFFFFFFF == *dst.getAddr32(5, 5));
}

static void test(skiatest::Reporter* reporter) {
    test_drawBitmap(reporter);
}

#include "TestClassDef.h"
DEFINE_TESTCLASS("QuickReject", QuickRejectClass, test)
