LayoutTreeBuilder.h [plain text]
#pragma once
#if ENABLE(LAYOUT_FORMATTING_CONTEXT)
#include "LayoutContainerBox.h"
#include <wtf/IsoMalloc.h>
#include <wtf/WeakPtr.h>
namespace WebCore {
class RenderBlockFlow;
class RenderBox;
class RenderElement;
class RenderObject;
class RenderTable;
class RenderView;
namespace Layout {
class LayoutState;
class LayoutTreeContent : public CanMakeWeakPtr<LayoutTreeContent> {
WTF_MAKE_ISO_ALLOCATED(LayoutTreeContent);
public:
LayoutTreeContent(const RenderBox&, std::unique_ptr<ContainerBox>);
~LayoutTreeContent();
const ContainerBox& rootLayoutBox() const { return *m_rootLayoutBox; }
ContainerBox& rootLayoutBox() { return *m_rootLayoutBox; }
const RenderBox& rootRenderer() const { return m_rootRenderer; }
void addBox(std::unique_ptr<Box> box)
{
ASSERT(!box->isContainerBox());
m_boxes.add(WTFMove(box));
}
void addContainer(std::unique_ptr<ContainerBox> container) { m_containers.add(WTFMove(container)); }
Box* layoutBoxForRenderer(const RenderObject& renderer) { return m_renderObjectToLayoutBox.get(&renderer); }
const Box* layoutBoxForRenderer(const RenderObject& renderer) const { return m_renderObjectToLayoutBox.get(&renderer); }
const RenderObject* rendererForLayoutBox(const Box& box) const { return m_layoutBoxToRenderObject.get(&box); }
void addLayoutBoxForRenderer(const RenderObject&, Box&);
private:
const RenderBox& m_rootRenderer;
std::unique_ptr<ContainerBox> m_rootLayoutBox;
HashSet<std::unique_ptr<Box>> m_boxes;
HashSet<std::unique_ptr<ContainerBox>> m_containers;
HashMap<const RenderObject*, Box*> m_renderObjectToLayoutBox;
HashMap<const Box*, const RenderObject*> m_layoutBoxToRenderObject;
};
class TreeBuilder {
public:
static std::unique_ptr<Layout::LayoutTreeContent> buildLayoutTree(const RenderView&);
private:
TreeBuilder(LayoutTreeContent&);
void buildTree();
void buildSubTree(const RenderElement& parentRenderer, ContainerBox& parentContainer);
void buildTableStructure(const RenderTable& tableRenderer, ContainerBox& tableWrapperBox);
Box* createLayoutBox(const ContainerBox& parentContainer, const RenderObject& childRenderer);
Box& createReplacedBox(Optional<Box::ElementAttributes>, RenderStyle&&);
Box& createTextBox(String text, bool canUseSimplifiedTextMeasuring, RenderStyle&&);
Box& createLineBreakBox(bool isOptional, RenderStyle&&);
ContainerBox& createContainer(Optional<Box::ElementAttributes>, RenderStyle&&);
LayoutTreeContent& m_layoutTreeContent;
};
#if ENABLE(TREE_DEBUGGING)
void showLayoutTree(const Box&, const LayoutState*);
void showLayoutTree(const Box&);
void printLayoutTreeForLiveDocuments();
#endif
}
}
#endif