#include "config.h"
#include "StyleUpdate.h"
#include "ComposedTreeAncestorIterator.h"
#include "Document.h"
#include "Element.h"
#include "NodeRenderStyle.h"
#include "RenderElement.h"
#include "Text.h"
namespace WebCore {
namespace Style {
Update::Update(Document& document)
: m_document(document)
{
}
const ElementUpdates* Update::elementUpdates(const Element& element) const
{
auto it = m_elements.find(&element);
if (it == m_elements.end())
return nullptr;
return &it->value;
}
ElementUpdates* Update::elementUpdates(const Element& element)
{
auto it = m_elements.find(&element);
if (it == m_elements.end())
return nullptr;
return &it->value;
}
const TextUpdate* Update::textUpdate(const Text& text) const
{
auto it = m_texts.find(&text);
if (it == m_texts.end())
return nullptr;
return &it->value;
}
const RenderStyle* Update::elementStyle(const Element& element) const
{
if (auto* updates = elementUpdates(element))
return updates->update.style.get();
auto* renderer = element.renderer();
if (!renderer)
return nullptr;
return &renderer->style();
}
RenderStyle* Update::elementStyle(const Element& element)
{
if (auto* updates = elementUpdates(element))
return updates->update.style.get();
auto* renderer = element.renderer();
if (!renderer)
return nullptr;
return &renderer->mutableStyle();
}
void Update::addElement(Element& element, Element* parent, ElementUpdates&& elementUpdates)
{
ASSERT(!m_elements.contains(&element));
ASSERT(composedTreeAncestors(element).first() == parent);
addPossibleRoot(parent);
m_elements.add(&element, WTFMove(elementUpdates));
}
void Update::addText(Text& text, Element* parent, TextUpdate&& textUpdate)
{
ASSERT(!m_texts.contains(&text));
ASSERT(composedTreeAncestors(text).first() == parent);
addPossibleRoot(parent);
m_texts.add(&text, WTFMove(textUpdate));
}
void Update::addText(Text& text, TextUpdate&& textUpdate)
{
addText(text, composedTreeAncestors(text).first(), WTFMove(textUpdate));
}
void Update::addPossibleRoot(Element* element)
{
if (!element) {
m_roots.add(&m_document);
return;
}
if (m_elements.contains(element))
return;
m_roots.add(element);
}
}
}