#pragma once
#if ENABLE(LAYOUT_FORMATTING_CONTEXT)
#include "FormattingContext.h"
#include "LayoutState.h"
#include <wtf/IsoMalloc.h>
#include <wtf/WeakPtr.h>
namespace WebCore {
namespace Layout {
class Box;
class FloatingState;
enum class StyleDiff;
class FormattingState {
WTF_MAKE_ISO_ALLOCATED(FormattingState);
public:
FloatingState& floatingState() const { return m_floatingState; }
void markNeedsLayout(const Box&, StyleDiff);
bool needsLayout(const Box&);
void setIntrinsicWidthConstraintsForBox(const Box&, FormattingContext::IntrinsicWidthConstraints);
Optional<FormattingContext::IntrinsicWidthConstraints> intrinsicWidthConstraintsForBox(const Box&) const;
void clearIntrinsicWidthConstraints(const Box&);
void setIntrinsicWidthConstraints(FormattingContext::IntrinsicWidthConstraints intrinsicWidthConstraints) { m_intrinsicWidthConstraints = intrinsicWidthConstraints; }
Optional<FormattingContext::IntrinsicWidthConstraints> intrinsicWidthConstraints() const { return m_intrinsicWidthConstraints; }
bool isBlockFormattingState() const { return m_type == Type::Block; }
bool isInlineFormattingState() const { return m_type == Type::Inline; }
bool isTableFormattingState() const { return m_type == Type::Table; }
LayoutState& layoutState() const { return m_layoutState; }
Display::Box& displayBox(const Box& layoutBox);
using OutOfFlowBoxList = Vector<WeakPtr<const Box>>;
void addOutOfFlowBox(const Box& outOfFlowBox) { m_outOfFlowBoxes.append(makeWeakPtr(outOfFlowBox)); }
void removeOutOfFlowBox(const Box&);
const OutOfFlowBoxList& outOfFlowBoxes() const { return m_outOfFlowBoxes; }
protected:
enum class Type { Block, Inline, Table };
FormattingState(Ref<FloatingState>&&, Type, LayoutState&);
~FormattingState();
private:
LayoutState& m_layoutState;
Ref<FloatingState> m_floatingState;
HashMap<const Box*, FormattingContext::IntrinsicWidthConstraints> m_intrinsicWidthConstraintsForBoxes;
Optional<FormattingContext::IntrinsicWidthConstraints> m_intrinsicWidthConstraints;
OutOfFlowBoxList m_outOfFlowBoxes;
Type m_type;
};
inline void FormattingState::setIntrinsicWidthConstraintsForBox(const Box& layoutBox, FormattingContext::IntrinsicWidthConstraints intrinsicWidthConstraints)
{
ASSERT(!m_intrinsicWidthConstraintsForBoxes.contains(&layoutBox));
ASSERT(&m_layoutState.formattingStateForBox(layoutBox) == this);
m_intrinsicWidthConstraintsForBoxes.set(&layoutBox, intrinsicWidthConstraints);
}
inline void FormattingState::clearIntrinsicWidthConstraints(const Box& layoutBox)
{
m_intrinsicWidthConstraints = { };
m_intrinsicWidthConstraintsForBoxes.remove(&layoutBox);
}
inline Optional<FormattingContext::IntrinsicWidthConstraints> FormattingState::intrinsicWidthConstraintsForBox(const Box& layoutBox) const
{
ASSERT(&m_layoutState.formattingStateForBox(layoutBox) == this);
auto iterator = m_intrinsicWidthConstraintsForBoxes.find(&layoutBox);
if (iterator == m_intrinsicWidthConstraintsForBoxes.end())
return { };
return iterator->value;
}
}
}
#define SPECIALIZE_TYPE_TRAITS_LAYOUT_FORMATTING_STATE(ToValueTypeName, predicate) \
SPECIALIZE_TYPE_TRAITS_BEGIN(WebCore::Layout::ToValueTypeName) \
static bool isType(const WebCore::Layout::FormattingState& formattingState) { return formattingState.predicate; } \
SPECIALIZE_TYPE_TRAITS_END()
#endif