--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/WebKit2/WebProcess/WebPage/ChunkedUpdateDrawingArea.cpp	Fri Sep 17 09:02:29 2010 +0300
@@ -0,0 +1,217 @@
+/*
+ * Copyright (C) 2010 Apple 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 INC. AND ITS CONTRIBUTORS ``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 INC. OR ITS 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 "ChunkedUpdateDrawingArea.h"
+
+#include "DrawingAreaMessageKinds.h"
+#include "DrawingAreaProxyMessageKinds.h"
+#include "MessageID.h"
+#include "UpdateChunk.h"
+#include "WebCoreArgumentCoders.h"
+#include "WebPage.h"
+#include "WebProcess.h"
+
+using namespace WebCore;
+
+namespace WebKit {
+
+ChunkedUpdateDrawingArea::ChunkedUpdateDrawingArea(WebPage* webPage)
+    : DrawingArea(ChunkedUpdateDrawingAreaType, webPage)
+    , m_isWaitingForUpdate(false)
+    , m_paintingIsSuspended(false)
+    , m_displayTimer(WebProcess::shared().runLoop(), this, &ChunkedUpdateDrawingArea::display)
+{
+}
+
+ChunkedUpdateDrawingArea::~ChunkedUpdateDrawingArea()
+{
+}
+
+void ChunkedUpdateDrawingArea::invalidateWindow(const IntRect& rect, bool immediate)
+{
+}
+
+void ChunkedUpdateDrawingArea::invalidateContentsAndWindow(const IntRect& rect, bool immediate)
+{
+    setNeedsDisplay(rect);
+}
+
+void ChunkedUpdateDrawingArea::invalidateContentsForSlowScroll(const IntRect& rect, bool immediate)
+{
+    setNeedsDisplay(rect);
+}
+
+void ChunkedUpdateDrawingArea::scroll(const IntSize& scrollDelta, const IntRect& rectToScroll, const IntRect& clipRect)
+{
+    // FIXME: Do something much smarter.
+    setNeedsDisplay(rectToScroll);
+}
+
+void ChunkedUpdateDrawingArea::setNeedsDisplay(const IntRect& rect)
+{
+    // FIXME: Collect a set of rects/region instead of just the union
+    // of all rects.
+    m_dirtyRect.unite(rect);
+    scheduleDisplay();
+}
+
+void ChunkedUpdateDrawingArea::display()
+{
+    ASSERT(!m_isWaitingForUpdate);
+ 
+    if (m_paintingIsSuspended)
+        return;
+
+    if (m_dirtyRect.isEmpty())
+        return;
+
+    // Layout if necessary.
+    m_webPage->layoutIfNeeded();
+ 
+    IntRect dirtyRect = m_dirtyRect;
+    m_dirtyRect = IntRect();
+
+    // Create a new UpdateChunk and paint into it.
+    UpdateChunk updateChunk(dirtyRect);
+    paintIntoUpdateChunk(&updateChunk);
+
+    WebProcess::shared().connection()->send(DrawingAreaProxyMessage::Update, m_webPage->pageID(), CoreIPC::In(updateChunk));
+
+    m_isWaitingForUpdate = true;
+    m_displayTimer.stop();
+}
+
+void ChunkedUpdateDrawingArea::scheduleDisplay()
+{
+    if (m_paintingIsSuspended)
+        return;
+
+    if (m_isWaitingForUpdate)
+        return;
+    
+    if (m_dirtyRect.isEmpty())
+        return;
+
+    if (m_displayTimer.isActive())
+        return;
+
+    m_displayTimer.startOneShot(0);
+}
+
+void ChunkedUpdateDrawingArea::setSize(const IntSize& viewSize)
+{
+    ASSERT_ARG(viewSize, !viewSize.isEmpty());
+
+    // We don't want to wait for an update until we display.
+    m_isWaitingForUpdate = false;
+    
+    m_webPage->setSize(viewSize);
+
+    // Layout if necessary.
+    m_webPage->layoutIfNeeded();
+
+    if (m_paintingIsSuspended) {
+        ASSERT(!m_displayTimer.isActive());
+
+        // Painting is suspended, just send back an empty update chunk.
+        WebProcess::shared().connection()->send(DrawingAreaProxyMessage::DidSetSize, m_webPage->pageID(), CoreIPC::In(UpdateChunk()));
+        return;
+    }
+
+    // Create a new UpdateChunk and paint into it.
+    UpdateChunk updateChunk(IntRect(0, 0, viewSize.width(), viewSize.height()));
+    paintIntoUpdateChunk(&updateChunk);
+
+    m_displayTimer.stop();
+
+    WebProcess::shared().connection()->send(DrawingAreaProxyMessage::DidSetSize, m_webPage->pageID(), CoreIPC::In(updateChunk));
+}
+
+void ChunkedUpdateDrawingArea::suspendPainting()
+{
+    ASSERT(!m_paintingIsSuspended);
+    
+    m_paintingIsSuspended = true;
+    m_displayTimer.stop();
+}
+
+void ChunkedUpdateDrawingArea::resumePainting(bool forceRepaint)
+{
+    ASSERT(m_paintingIsSuspended);
+    
+    m_paintingIsSuspended = false;
+
+    if (forceRepaint) {
+        // Just set the dirty rect to the entire page size.
+        m_dirtyRect = IntRect(IntPoint(0, 0), m_webPage->size());
+    }
+
+    // Schedule a display.
+    scheduleDisplay();
+}
+
+void ChunkedUpdateDrawingArea::didUpdate()
+{
+    m_isWaitingForUpdate = false;
+
+    // Display if needed.
+    display();
+}
+
+void ChunkedUpdateDrawingArea::didReceiveMessage(CoreIPC::Connection*, CoreIPC::MessageID messageID, CoreIPC::ArgumentDecoder& arguments)
+{
+    switch (messageID.get<DrawingAreaMessage::Kind>()) {
+        case DrawingAreaMessage::SetSize: {
+            IntSize size;
+            if (!arguments.decode(CoreIPC::Out(size)))
+                return;
+
+            setSize(size);
+            break;
+        }
+        
+        case DrawingAreaMessage::SuspendPainting:
+            suspendPainting();
+            break;
+
+        case DrawingAreaMessage::ResumePainting: {
+            bool forceRepaint;
+            if (!arguments.decode(CoreIPC::Out(forceRepaint)))
+                return;
+            
+            resumePainting(forceRepaint);
+            break;
+        }
+        case DrawingAreaMessage::DidUpdate:
+            didUpdate();
+            break;
+
+        default:
+            ASSERT_NOT_REACHED();
+            break;
+    }
+}
+
+} // namespace WebKit