dom/svg/SVGElementFactory.cpp

   1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim: set ts=8 sts=2 et sw=2 tw=80: */
   3 /* This Source Code Form is subject to the terms of the Mozilla Public
   4  * License, v. 2.0. If a copy of the MPL was not distributed with this
   5  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #include "SVGElementFactory.h"
   8 #include "nsGkAtoms.h"
   9 #include "nsIContent.h"
  10 #include "mozilla/dom/NodeInfo.h"
  11 #include "mozilla/dom/Element.h"
  12 #include "mozilla/dom/FromParser.h"
  13 #include "mozilla/StaticPtr.h"
  14 #include "nsTHashMap.h"
  15 #include "nsHashKeys.h"
  16
  17 using namespace mozilla;
  18 using namespace mozilla::dom;
  19
  20 // Hash table that maps nsAtom* SVG tags to a SVGContentCreatorFunction.
  21 using TagAtomTable =
  22     nsTHashMap<nsPtrHashKey<nsAtom>, SVGContentCreatorFunction>;
  23 StaticAutoPtr<TagAtomTable> sTagAtomTable;
  24
  25 #define SVG_TAG(_tag, _classname)                                         \
  26   nsresult NS_NewSVG##_classname##Element(                                \
  27       nsIContent** aResult,                                               \
  28       already_AddRefed<mozilla::dom::NodeInfo>&& aNodeInfo);              \
  29                                                                           \
  30   nsresult NS_NewSVG##_classname##Element(                                \
  31       nsIContent** aResult,                                               \
  32       already_AddRefed<mozilla::dom::NodeInfo>&& aNodeInfo,               \
  33       FromParser aFromParser) {                                           \
  34     return NS_NewSVG##_classname##Element(aResult, std::move(aNodeInfo)); \
  35   }
  36
  37 #define SVG_FROM_PARSER_TAG(_tag, _classname)
  38
  39 #include "SVGTagList.h"
  40 #undef SVG_TAG
  41 #undef SVG_FROM_PARSER_TAG
  42
  43 nsresult NS_NewSVGElement(Element** aResult,
  44                           already_AddRefed<mozilla::dom::NodeInfo>&& aNodeInfo);
  45
  46 enum SVGTag {
  47 #define SVG_TAG(_tag, _classname) eSVGTag_##_tag,
  48 #define SVG_FROM_PARSER_TAG(_tag, _classname) eSVGTag_##_tag,
  49 #include "SVGTagList.h"
  50 #undef SVG_TAG
  51 #undef SVG_FROM_PARSER_TAG
  52   eSVGTag_Count
  53 };
  54
  55 void SVGElementFactory::Init() {
  56   sTagAtomTable = new TagAtomTable(64);
  57
  58 #define SVG_TAG(_tag, _classname) \
  59   sTagAtomTable->InsertOrUpdate(  \
  60       nsGkAtoms::_tag,            \
  61       SVGContentCreatorFunction(NS_NewSVG##_classname##Element));
  62 #define SVG_FROM_PARSER_TAG(_tag, _classname) \
  63   sTagAtomTable->InsertOrUpdate(              \
  64       nsGkAtoms::_tag,                        \
  65       SVGContentCreatorFunction(NS_NewSVG##_classname##Element));
  66 #include "SVGTagList.h"
  67 #undef SVG_TAG
  68 #undef SVG_FROM_PARSER_TAG
  69 }
  70
  71 void SVGElementFactory::Shutdown() { sTagAtomTable = nullptr; }
  72
  73 nsresult NS_NewSVGElement(Element** aResult,
  74                           already_AddRefed<mozilla::dom::NodeInfo>&& aNodeInfo,
  75                           FromParser aFromParser) {
  76   NS_ASSERTION(sTagAtomTable, "no lookup table, needs SVGElementFactory::Init");
  77
  78   RefPtr<mozilla::dom::NodeInfo> ni = aNodeInfo;
  79   nsAtom* name = ni->NameAtom();
  80
  81   NS_ASSERTION(
  82       ni->NamespaceEquals(kNameSpaceID_SVG),
  83       "Trying to create SVG elements that aren't in the SVG namespace");
  84
  85   SVGContentCreatorFunction cb = sTagAtomTable->Get(name);
  86   if (cb) {
  87     nsCOMPtr<nsIContent> content;
  88     nsresult rv = cb(getter_AddRefs(content), ni.forget(), aFromParser);
  89     *aResult = content.forget().take()->AsElement();
  90     return rv;
  91   }
  92
  93   // if we don't know what to create, just create a standard svg element:
  94   return NS_NewSVGElement(aResult, ni.forget());
  95 }
  96
  97 nsresult NS_NewSVGUnknownElement(
  98     nsIContent** aResult, already_AddRefed<mozilla::dom::NodeInfo>&& aNodeInfo,
  99     FromParser aFromParser) {
 100   RefPtr<mozilla::dom::NodeInfo> ni = aNodeInfo;
 101   nsCOMPtr<Element> element;
 102   nsresult rv = NS_NewSVGElement(getter_AddRefs(element), ni.forget());
 103   element.forget(aResult);
 104   return rv;
 105 }