1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
// Copyright 2015 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef NthIndexCache_h
#define NthIndexCache_h
#include "core/CoreExport.h"
#include "core/dom/Element.h"
#include "platform/heap/Handle.h"
#include "wtf/HashMap.h"
namespace blink {
class Document;
class CORE_EXPORT NthIndexData final : public GarbageCollected<NthIndexData> {
WTF_MAKE_NONCOPYABLE(NthIndexData);
public:
NthIndexData(ContainerNode&);
NthIndexData(ContainerNode&, const QualifiedName& type);
unsigned nthIndex(Element&) const;
unsigned nthLastIndex(Element&) const;
unsigned nthOfTypeIndex(Element&) const;
unsigned nthLastOfTypeIndex(Element&) const;
private:
HeapHashMap<Member<Element>, unsigned> m_elementIndexMap;
unsigned m_count = 0;
DECLARE_TRACE();
};
class CORE_EXPORT NthIndexCache final {
STACK_ALLOCATED();
WTF_MAKE_NONCOPYABLE(NthIndexCache);
public:
explicit NthIndexCache(Document&);
~NthIndexCache();
static unsigned nthChildIndex(Element&);
static unsigned nthLastChildIndex(Element&);
static unsigned nthOfTypeIndex(Element&);
static unsigned nthLastOfTypeIndex(Element&);
private:
using IndexByType = HeapHashMap<String, Member<NthIndexData>>;
using ParentMap = HeapHashMap<Member<Node>, Member<NthIndexData>>;
using ParentMapForType = HeapHashMap<Member<Node>, Member<IndexByType>>;
void cacheNthIndexDataForParent(Element&);
void cacheNthOfTypeIndexDataForParent(Element&);
IndexByType& ensureTypeIndexMap(ContainerNode&);
NthIndexData* nthTypeIndexDataForParent(Element&) const;
Member<Document> m_document;
Member<ParentMap> m_parentMap;
Member<ParentMapForType> m_parentMapForType;
#if DCHECK_IS_ON()
uint64_t m_domTreeVersion;
#endif
};
} // namespace blink
#endif // NthIndexCache_h
|
