#include "config.h"
#include "ImageLoader.h"
#include "CSSHelper.h"
#include "CachedImage.h"
#include "DocLoader.h"
#include "Document.h"
#include "Element.h"
#include "RenderImage.h"
namespace WebCore {
class ImageLoadEventSender {
public:
ImageLoadEventSender();
void dispatchLoadEventSoon(ImageLoader*);
void cancelLoadEvent(ImageLoader*);
void dispatchPendingLoadEvents();
private:
~ImageLoadEventSender();
void timerFired(Timer<ImageLoadEventSender>*);
Timer<ImageLoadEventSender> m_timer;
Vector<ImageLoader*> m_dispatchSoonList;
Vector<ImageLoader*> m_dispatchingList;
};
static ImageLoadEventSender& loadEventSender()
{
DEFINE_STATIC_LOCAL(ImageLoadEventSender, sender, ());
return sender;
}
ImageLoader::ImageLoader(Element* element)
: m_element(element)
, m_image(0)
, m_firedLoad(true)
, m_imageComplete(true)
, m_loadManually(false)
{
}
ImageLoader::~ImageLoader()
{
if (m_image)
m_image->removeClient(this);
loadEventSender().cancelLoadEvent(this);
}
void ImageLoader::setImage(CachedImage* newImage)
{
ASSERT(m_failedLoadURL.isEmpty());
CachedImage* oldImage = m_image.get();
if (newImage != oldImage) {
setLoadingImage(newImage);
m_firedLoad = true;
m_imageComplete = true;
if (newImage)
newImage->addClient(this);
if (oldImage)
oldImage->removeClient(this);
}
if (RenderObject* renderer = m_element->renderer()) {
if (!renderer->isImage())
return;
toRenderImage(renderer)->resetAnimation();
}
}
void ImageLoader::setLoadingImage(CachedImage* loadingImage)
{
m_firedLoad = false;
m_imageComplete = false;
m_image = loadingImage;
}
void ImageLoader::updateFromElement()
{
Document* document = m_element->document();
if (!document->renderer())
return;
AtomicString attr = m_element->getAttribute(m_element->imageSourceAttributeName());
if (attr == m_failedLoadURL)
return;
CachedImage* newImage = 0;
if (!(attr.isNull() || (attr.isEmpty() && document->baseURI().isLocalFile()))) {
if (m_loadManually) {
document->docLoader()->setAutoLoadImages(false);
newImage = new CachedImage(sourceURI(attr));
newImage->setLoading(true);
newImage->setDocLoader(document->docLoader());
document->docLoader()->m_documentResources.set(newImage->url(), newImage);
} else
newImage = document->docLoader()->requestImage(sourceURI(attr));
m_failedLoadURL = !newImage ? attr : AtomicString();
}
CachedImage* oldImage = m_image.get();
if (newImage != oldImage) {
setLoadingImage(newImage);
if (newImage)
newImage->addClient(this);
if (oldImage)
oldImage->removeClient(this);
}
if (RenderObject* renderer = m_element->renderer()) {
if (!renderer->isImage())
return;
toRenderImage(renderer)->resetAnimation();
}
}
void ImageLoader::updateFromElementIgnoringPreviousError()
{
m_failedLoadURL = AtomicString();
updateFromElement();
}
void ImageLoader::notifyFinished(CachedResource*)
{
ASSERT(m_failedLoadURL.isEmpty());
m_imageComplete = true;
loadEventSender().dispatchLoadEventSoon(this);
if (RenderObject* renderer = m_element->renderer()) {
if (!renderer->isImage())
return;
toRenderImage(renderer)->setCachedImage(m_image.get());
}
}
void ImageLoader::dispatchPendingLoadEvent()
{
if (m_firedLoad)
return;
if (!m_image)
return;
if (!m_element->document()->attached())
return;
m_firedLoad = true;
dispatchLoadEvent();
}
void ImageLoader::dispatchPendingLoadEvents()
{
loadEventSender().dispatchPendingLoadEvents();
}
ImageLoadEventSender::ImageLoadEventSender()
: m_timer(this, &ImageLoadEventSender::timerFired)
{
}
void ImageLoadEventSender::dispatchLoadEventSoon(ImageLoader* loader)
{
m_dispatchSoonList.append(loader);
if (!m_timer.isActive())
m_timer.startOneShot(0);
}
void ImageLoadEventSender::cancelLoadEvent(ImageLoader* loader)
{
size_t size = m_dispatchSoonList.size();
for (size_t i = 0; i < size; ++i) {
if (m_dispatchSoonList[i] == loader)
m_dispatchSoonList[i] = 0;
}
size = m_dispatchingList.size();
for (size_t i = 0; i < size; ++i) {
if (m_dispatchingList[i] == loader)
m_dispatchingList[i] = 0;
}
if (m_dispatchSoonList.isEmpty())
m_timer.stop();
}
void ImageLoadEventSender::dispatchPendingLoadEvents()
{
if (!m_dispatchingList.isEmpty())
return;
m_timer.stop();
m_dispatchingList.swap(m_dispatchSoonList);
size_t size = m_dispatchingList.size();
for (size_t i = 0; i < size; ++i) {
if (ImageLoader* loader = m_dispatchingList[i])
loader->dispatchPendingLoadEvent();
}
m_dispatchingList.clear();
}
void ImageLoadEventSender::timerFired(Timer<ImageLoadEventSender>*)
{
dispatchPendingLoadEvents();
}
}