HTMLImageLoader.cpp [plain text]
#include "config.h"
#include "HTMLImageLoader.h"
#include "CSSHelper.h"
#include "CachedImage.h"
#include "DocLoader.h"
#include "Document.h"
#include "Element.h"
#include "EventNames.h"
#include "HTMLNames.h"
#include "RenderImage.h"
#include "Frame.h"
#include "Page.h"
using namespace std;
namespace WebCore {
using namespace EventNames;
using namespace HTMLNames;
HTMLImageLoader::HTMLImageLoader(Element* elt)
: m_element(elt)
, m_image(0)
, m_firedLoad(true)
, m_imageComplete(true)
, m_loadManually(false)
{
}
HTMLImageLoader::~HTMLImageLoader()
{
if (m_image && m_image->isLoaded())
m_element->document()->topDocument()->decrementTotalImageDataSize(m_image);
if (m_image)
m_image->deref(this);
m_element->document()->removeImage(this);
}
void HTMLImageLoader::setImage(CachedImage *newImage)
{
CachedImage *oldImage = m_image;
if (newImage != oldImage) {
if (oldImage && oldImage->isLoaded())
m_element->document()->topDocument()->decrementTotalImageDataSize(oldImage);
if (newImage && newImage->isLoaded())
m_element->document()->topDocument()->incrementTotalImageDataSize(newImage);
setLoadingImage(newImage);
m_firedLoad = true;
m_imageComplete = true;
if (newImage)
newImage->ref(this);
if (oldImage)
oldImage->deref(this);
}
if (RenderObject* renderer = element()->renderer())
if (renderer->isImage())
static_cast<RenderImage*>(renderer)->resetAnimation();
}
void HTMLImageLoader::setLoadingImage(CachedImage *loadingImage)
{
m_firedLoad = false;
m_imageComplete = false;
m_image = loadingImage;
}
void HTMLImageLoader::updateFromElement()
{
Element* elem = element();
Document* doc = elem->document();
if (!doc->renderer())
return;
bool ok;
int width = elem->getAttribute(widthAttr).toInt(&ok);
if (!ok)
width = 0;
int height = elem->getAttribute(heightAttr).toInt(&ok);
if (!ok)
height = 0;
if (width * height * 4 > doc->frame()->maximumImageSize()) return;
if (memoryLimitReached()) return;
AtomicString attr = elem->getAttribute(elem->imageSourceAttributeName());
CachedImage *newImage = 0;
if (!attr.isEmpty()) {
if (m_loadManually) {
doc->docLoader()->setAutoLoadImages(false);
newImage = new CachedImage(doc->docLoader(), parseURL(attr), false );
newImage->setLoading(true);
newImage->setDocLoader(doc->docLoader());
doc->docLoader()->m_docResources.set(newImage->url(), newImage);
} else
newImage = doc->docLoader()->requestImage(parseURL(attr));
}
CachedImage *oldImage = m_image;
if (newImage != oldImage) {
#ifdef INSTRUMENT_LAYOUT_SCHEDULING
if (!doc->ownerElement() && newImage)
printf("Image requested at %d\n", doc->elapsedTime());
#endif
if (oldImage && oldImage->isLoaded())
m_element->document()->topDocument()->decrementTotalImageDataSize(oldImage);
if (newImage && newImage->isLoaded())
m_element->document()->topDocument()->incrementTotalImageDataSize(newImage);
setLoadingImage(newImage);
if (newImage)
newImage->ref(this);
if (oldImage)
oldImage->deref(this);
}
if (RenderObject* renderer = elem->renderer())
if (renderer->isImage())
static_cast<RenderImage*>(renderer)->resetAnimation();
}
void HTMLImageLoader::dispatchLoadEvent()
{
if (!haveFiredLoadEvent() && image()) {
setHaveFiredLoadEvent(true);
element()->dispatchHTMLEvent(image()->errorOccurred() ? errorEvent : loadEvent, false, false);
}
}
void HTMLImageLoader::notifyFinished(CachedResource *image)
{
m_imageComplete = true;
Element* elem = element();
Document* doc = elem->document();
doc->dispatchImageLoadEventSoon(this);
Document* mainFrameDocument = doc->topDocument();
mainFrameDocument->incrementTotalImageDataSize(m_image);
unsigned animatedImageSize = m_image->animatedImageSize();
if (animatedImageSize) {
mainFrameDocument->incrementAnimatedImageDataCount(animatedImageSize);
if (mainFrameDocument->animatedImageDataCount() > 2 * 1024 * 1024)
m_image->stopAnimatedImage();
}
#ifdef INSTRUMENT_LAYOUT_SCHEDULING
if (!doc->ownerElement())
printf("Image loaded at %d\n", doc->elapsedTime());
#endif
if (RenderObject* renderer = elem->renderer())
if (renderer->isImage())
static_cast<RenderImage*>(renderer)->setCachedImage(m_image);
}
bool HTMLImageLoader::memoryLimitReached()
{
if (!element() || !element()->document() || !element()->document()->page() || !element()->document()->page()->mainFrame() || !element()->document()->page()->mainFrame()->document())
return false;
Frame *mainFrame = element()->document()->page()->mainFrame();
Document *mainFrameDocument = mainFrame->document();
static unsigned long maxImageSize = mainFrame->maximumImageSize() / 2; return mainFrameDocument->totalImageDataSize() > maxImageSize;
}
}