SVG: Move/rename reference management to SVGElement

Source/core/svg/SVGElement.cpp

 /*
  * Copyright (C) 2004, 2005, 2006, 2007, 2008 Nikolas Zimmermann <zimmermann@kde.org>
  * Copyright (C) 2004, 2005, 2006, 2008 Rob Buis <buis@kde.org>
  * Copyright (C) 2008 Apple Inc. All rights reserved.
  * Copyright (C) 2008 Alp Toker <alp@atoker.com>
  * Copyright (C) 2009 Cameron McCormack <cam@mcc.id.au>
  *
  * This library is free software; you can redistribute it and/or
  * modify it under the terms of the GNU Library General Public
  * License as published by the Free Software Foundation; either
@@ skipping 80 matching lines @@
     // The below teardown is all handled by weak pointer processing in oilpan.
 #if !ENABLE(OILPAN)
     if (hasSVGRareData()) {
         if (SVGCursorElement* cursorElement = svgRareData()->cursorElement())
             cursorElement->removeReferencedElement(this);
         if (CSSCursorImageValue* cursorImageValue = svgRareData()->cursorImageValue())
             cursorImageValue->removeReferencedElement(this);
 
         // With Oilpan, either removedFrom has been called or the document is dead
         // as well and there is no reason to clear out the references.
-        document().accessSVGExtensions().rebuildAllElementReferencesForTarget(this);
-        document().accessSVGExtensions().removeAllElementReferencesForTarget(this);
+        rebuildAllIncomingReferences();
+        removeAllIncomingReferences();
     }
 #endif
 }
 
 void SVGElement::detach(const AttachContext& context)
 {
     Element::detach(context);
     if (SVGElement* element = correspondingElement())
         element->removeInstanceMapping(this);
 }
@@ skipping 197 matching lines @@
         }
 
         m_elementsWithRelativeLengths.clear();
     }
 
     ASSERT_WITH_SECURITY_IMPLICATION(!rootParent->isSVGElement() || !toSVGElement(rootParent)->m_elementsWithRelativeLengths.contains(this));
 
     Element::removedFrom(rootParent);
 
     if (wasInDocument) {
-        document().accessSVGExtensions().rebuildAllElementReferencesForTarget(this);
-        document().accessSVGExtensions().removeAllElementReferencesForTarget(this);
+        rebuildAllIncomingReferences();
+        removeAllIncomingReferences();
     }
 
     invalidateInstances();
 }
 
 void SVGElement::childrenChanged(bool changedByParser, Node* beforeChange, Node* afterChange, int childCountDelta)
 {
     Element::childrenChanged(changedByParser, beforeChange, afterChange, childCountDelta);
 
     // Invalidate all instances associated with us.
@@ skipping 558 matching lines @@
 {
     ASSERT_NOT_REACHED();
     return 0;
 }
 
 void SVGElement::attributeChanged(const QualifiedName& name, const AtomicString& newValue, AttributeModificationReason)
 {
     Element::attributeChanged(name, newValue);
 
     if (isIdAttributeName(name))
-        document().accessSVGExtensions().rebuildAllElementReferencesForTarget(this);
+        rebuildAllIncomingReferences();
 
     // Changes to the style attribute are processed lazily (see Element::getAttribute() and related methods),
     // so we don't want changes to the style attribute to result in extra work here.
     if (name != HTMLNames::styleAttr)
         svgAttributeChanged(name);
 }
 
 void SVGElement::svgAttributeChanged(const QualifiedName& attrName)
 {
     CSSPropertyID propId = SVGElement::cssPropertyIdForSVGAttributeName(attrName);
@@ skipping 234 matching lines @@
         animatableAttributes.add(SVGNames::zAttr);
     }
 
     if (name == classAttr)
         return true;
 
     return animatableAttributes.contains(name);
 }
 #endif
 
+SVGElementSet* SVGElement::setOfIncomingReferences() const
+{
+    if (!hasSVGRareData())
+        return 0;
+    return &svgRareData()->incomingReferences();
+}
+
+void SVGElement::addReferenceTo(SVGElement* targetElement)
+{
+    ASSERT(targetElement);
+
+    ensureSVGRareData()->outgoingReferences().add(targetElement);
+    targetElement->ensureSVGRareData()->incomingReferences().add(this);
+}
+
+void SVGElement::rebuildAllIncomingReferences()
+{
+    if (!hasSVGRareData())
+        return;
+
+    const SVGElementSet& incomingReferences = svgRareData()->incomingReferences();
+
+    // Iterate on a snapshot as |incomingReferences| may be altered inside loop.
+    WillBeHeapVector<RawPtrWillBeMember<SVGElement> > incomingReferencesSnapshot;
+    copyToVector(incomingReferences, incomingReferencesSnapshot);
+
+    // Force rebuilding the |sourceElement| so it knows about this change.
+    for (WillBeHeapVector<RawPtrWillBeMember<SVGElement> >::iterator it = incomingReferencesSnapshot.begin(), itEnd = incomingReferencesSnapshot.end(); it != itEnd; ++it) {
+        SVGElement* sourceElement = *it;
+
+        // Before rebuilding |sourceElement| ensure it was not removed from under us.
+        if (incomingReferences.contains(sourceElement))
+            sourceElement->svgAttributeChanged(XLinkNames::hrefAttr);
+    }
+}
+
+void SVGElement::removeAllIncomingReferences()
+{
+    if (!hasSVGRareData())
+        return;
+
+    SVGElementSet& incomingReferences = svgRareData()->incomingReferences();
+    for (SVGElementSet::iterator it = incomingReferences.begin(), itEnd = incomingReferences.end(); it != itEnd; ++it) {
+        SVGElement* sourceElement = *it;
+        ASSERT(sourceElement->hasSVGRareData());
+        sourceElement->ensureSVGRareData()->outgoingReferences().remove(this);
+    }
+    incomingReferences.clear();
+}
+
+void SVGElement::removeAllOutgoingReferences()
+{
+    if (!hasSVGRareData())
+        return;
+
+    SVGElementSet& outgoingReferences = svgRareData()->outgoingReferences();
+    for (SVGElementSet::iterator it = outgoingReferences.begin(), itEnd = outgoingReferences.end(); it != itEnd; ++it) {
+        SVGElement* targetElement = *it;
+        ASSERT(targetElement->hasSVGRareData());
+        targetElement->ensureSVGRareData()->incomingReferences().remove(this);
+    }
+    outgoingReferences.clear();
+}
+
 void SVGElement::trace(Visitor* visitor)
 {
     visitor->trace(m_elementsWithRelativeLengths);
     visitor->trace(m_SVGRareData);
     Element::trace(visitor);
 }
 
 const AtomicString& SVGElement::eventParameterName()
 {
     DEFINE_STATIC_LOCAL(const AtomicString, evtString, ("evt", AtomicString::ConstructFromLiteral));
     return evtString;
 }
 
 }