/*
 * Copyright (C) 2012 Google Inc. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY
 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR
 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include "config.h"
#include "ImageDecodingStore.h"

#include "ImageDecoder.h"
#include "ImageFrameGenerator.h"
#include "LazyDecodingPixelRef.h"
#include "ScaledImageFragment.h"
#include "SkRect.h"
#include "SkSize.h"

#include "skia/ext/image_operations.h"

#include <wtf/MainThread.h>
#include <wtf/PassOwnPtr.h>

namespace WebCore {

namespace {

// URI label for a lazily decoded SkPixelRef.
const char labelLazyDecoded[] = "lazy";

ImageDecodingStore* s_instanceOnMainThread = 0;

static void setInstanceOnMainThread(ImageDecodingStore* imageDecodingStore)
{
    delete s_instanceOnMainThread;
    s_instanceOnMainThread = imageDecodingStore;
}
} // namespace

ImageDecodingStore::ImageDecodingStore()
{
}

ImageDecodingStore::~ImageDecodingStore()
{
}

ImageDecodingStore* ImageDecodingStore::instanceOnMainThread()
{
    ASSERT(isMainThread());
    return s_instanceOnMainThread;
}

void ImageDecodingStore::initializeOnMainThread()
{
    ASSERT(isMainThread());
    setInstanceOnMainThread(ImageDecodingStore::create().leakPtr());
}

void ImageDecodingStore::shutdown()
{
    ASSERT(isMainThread());
    setInstanceOnMainThread(0);
}

bool ImageDecodingStore::isLazyDecoded(const SkBitmap& bitmap)
{
    return bitmap.pixelRef()
        && bitmap.pixelRef()->getURI()
        && !memcmp(bitmap.pixelRef()->getURI(), labelLazyDecoded, sizeof(labelLazyDecoded));
}

SkBitmap ImageDecodingStore::createLazyDecodedSkBitmap(PassOwnPtr<ImageDecoder> decoder)
{
    ASSERT(calledOnValidThread());

    SkISize fullSize = SkISize::Make(decoder->size().width(), decoder->size().height());
    ASSERT(!fullSize.isEmpty());

    SkIRect fullRect = SkIRect::MakeSize(fullSize);

    // Creates a lazily decoded SkPixelRef that references the entire image without scaling.
    SkBitmap bitmap;
    bitmap.setConfig(SkBitmap::kARGB_8888_Config, fullSize.width(), fullSize.height());
    bitmap.setPixelRef(new LazyDecodingPixelRef(ImageFrameGenerator::create(decoder), fullSize, fullRect))->unref();

    // Use URI to identify this is a lazily decoded SkPixelRef of type LazyDecodingPixelRef.
    // FIXME: Give the actual URI is more useful.
    bitmap.pixelRef()->setURI(labelLazyDecoded);

    // Inform the bitmap that we will never change the pixels. This is a performance hint
    // subsystems that may try to cache this bitmap (e.g. pictures, pipes, gpu, pdf, etc.)
    bitmap.setImmutable();

    // FIXME: Setting bitmap.setIsOpaque() is big performance gain if possible. We can
    // do so safely if the image is fully loaded and it is a JPEG image, or image was
    // decoded before.

    return bitmap;
}

SkBitmap ImageDecodingStore::resizeLazyDecodedSkBitmap(const SkBitmap& bitmap, const SkISize& scaledSize, const SkIRect& scaledSubset)
{
    ASSERT(calledOnValidThread());

    LazyDecodingPixelRef* pixelRef = static_cast<LazyDecodingPixelRef*>(bitmap.pixelRef());
    ASSERT(!pixelRef->isScaled(pixelRef->frameGenerator()->fullSize()) && !pixelRef->isClipped());

    int rowBytes = 0;
    rowBytes = SkBitmap::ComputeRowBytes(SkBitmap::kARGB_8888_Config, scaledSize.width());

    SkBitmap resizedBitmap;
    resizedBitmap.setConfig(SkBitmap::kARGB_8888_Config, scaledSubset.width(), scaledSubset.height(), rowBytes);
    resizedBitmap.setPixelRef(new LazyDecodingPixelRef(pixelRef->frameGenerator(), scaledSize, scaledSubset))->unref();

    // See comments in createLazyDecodedSkBitmap().
    resizedBitmap.setImmutable();
    return resizedBitmap;
}

void ImageDecodingStore::setData(const SkBitmap& bitmap, PassRefPtr<SharedBuffer> data, bool allDataReceived)
{
    ASSERT(calledOnValidThread());
    if (isLazyDecoded(bitmap))
        static_cast<LazyDecodingPixelRef*>(bitmap.pixelRef())->frameGenerator()->setData(data, allDataReceived);
}

void* ImageDecodingStore::lockPixels(PassRefPtr<ImageFrameGenerator> frameGenerator, const SkISize& scaledSize, const SkIRect& scaledSubset)
{
    ASSERT(calledOnValidThread());
    ASSERT(m_lockedSkBitmap.isNull());

    SkISize fullSize = frameGenerator->fullSize();

    // The cache only saves the full size image at the moment.
    ScaledImageFragment* frameCache = lookupFrameCache(frameGenerator->imageId(), fullSize);
    SkBitmap bitmap;

    if (!frameCache) {
        // FIXME: Pixels are owned by the ImageDecoder in the generator. Find
        // a way to transfer ownership to our cache.
        ImageFrame* frame = frameGenerator->decoder()->frameBufferAtIndex(0);

        // Saves the frame in cache if this is complete.
        if (frame->status() == ImageFrame::FrameComplete) {
            m_frameCache.append(ScaledImageFragment::create(frameGenerator->imageId(), fullSize, frame->getSkBitmap(), true));
        } else if (frame->status() == ImageFrame::FrameEmpty) {
            // FIXME: If this is a decoding error. Report this to BitmapImage.
            return 0;
        }

        bitmap = frame->getSkBitmap();
    } else {
        ASSERT(frameCache->isComplete());
        bitmap = frameCache->bitmap();
    }

    if (fullSize != scaledSize) {
        // Image scaling is done on-the-fly and scaled image is destroyed
        // after use.
        // FIXME: Resize just the requested fragment.
        m_lockedSkBitmap = skia::ImageOperations::Resize(bitmap, skia::ImageOperations::RESIZE_LANCZOS3, scaledSize.width(), scaledSize.height());
    } else
        m_lockedSkBitmap = bitmap;
    m_lockedSkBitmap.lockPixels();
    return m_lockedSkBitmap.getAddr(scaledSubset.x(), scaledSubset.y());
}

void ImageDecodingStore::unlockPixels()
{
    ASSERT(calledOnValidThread());

    m_lockedSkBitmap.unlockPixels();
    m_lockedSkBitmap.reset();
}

void ImageDecodingStore::frameGeneratorBeingDestroyed(ImageFrameGenerator* frameGenerator)
{
    ASSERT(calledOnValidThread());

    deleteFrameCache(frameGenerator->imageId());
}

bool ImageDecodingStore::calledOnValidThread() const
{
    return this == instanceOnMainThread() && isMainThread();
}

ScaledImageFragment* ImageDecodingStore::lookupFrameCache(int imageId, const SkISize& scaledSize) const
{
    for (size_t i = 0; i < m_frameCache.size(); ++i) {
        if (m_frameCache[i]->isEqual(imageId, scaledSize))
            return m_frameCache[i].get();
    }
    return 0;
}

void ImageDecodingStore::deleteFrameCache(int imageId)
{
    for (size_t i = 0; i < m_frameCache.size(); ++i) {
        if (m_frameCache[i]->isEqual(imageId)) {
            m_frameCache.remove(i);
            return;
        }
    }
}

} // namespace WebCore