dom/xslt/xpath/txResultRecycler.cpp

   1 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* This Source Code Form is subject to the terms of the Mozilla Public
   3  * License, v. 2.0. If a copy of the MPL was not distributed with this
   4  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   5
   6 #include "txResultRecycler.h"
   7 #include "txExprResult.h"
   8 #include "txNodeSet.h"
   9
  10 txResultRecycler::txResultRecycler()
  11     : mEmptyStringResult(new StringResult(nullptr)),
  12       mTrueResult(new BooleanResult(true)),
  13       mFalseResult(new BooleanResult(false)) {}
  14
  15 txResultRecycler::~txResultRecycler() {
  16   txStackIterator stringIter(&mStringResults);
  17   while (stringIter.hasNext()) {
  18     delete static_cast<StringResult*>(stringIter.next());
  19   }
  20   txStackIterator nodesetIter(&mNodeSetResults);
  21   while (nodesetIter.hasNext()) {
  22     delete static_cast<txNodeSet*>(nodesetIter.next());
  23   }
  24   txStackIterator numberIter(&mNumberResults);
  25   while (numberIter.hasNext()) {
  26     delete static_cast<NumberResult*>(numberIter.next());
  27   }
  28 }
  29
  30 void txResultRecycler::recycle(txAExprResult* aResult) {
  31   NS_ASSERTION(aResult->mRefCnt == 0, "In-use txAExprResult recycled");
  32   RefPtr<txResultRecycler> kungFuDeathGrip;
  33   aResult->mRecycler.swap(kungFuDeathGrip);
  34
  35   switch (aResult->getResultType()) {
  36     case txAExprResult::STRING: {
  37       mStringResults.push(static_cast<StringResult*>(aResult));
  38       return;
  39     }
  40     case txAExprResult::NODESET: {
  41       static_cast<txNodeSet*>(aResult)->clear();
  42       mNodeSetResults.push(static_cast<txNodeSet*>(aResult));
  43       return;
  44     }
  45     case txAExprResult::NUMBER: {
  46       mNumberResults.push(static_cast<NumberResult*>(aResult));
  47       return;
  48     }
  49     default: {
  50       delete aResult;
  51     }
  52   }
  53 }
  54
  55 nsresult txResultRecycler::getStringResult(StringResult** aResult) {
  56   if (mStringResults.isEmpty()) {
  57     *aResult = new StringResult(this);
  58   } else {
  59     *aResult = static_cast<StringResult*>(mStringResults.pop());
  60     (*aResult)->mValue.Truncate();
  61     (*aResult)->mRecycler = this;
  62   }
  63   NS_ADDREF(*aResult);
  64
  65   return NS_OK;
  66 }
  67
  68 nsresult txResultRecycler::getStringResult(const nsAString& aValue,
  69                                            txAExprResult** aResult) {
  70   if (mStringResults.isEmpty()) {
  71     *aResult = new StringResult(aValue, this);
  72   } else {
  73     StringResult* strRes = static_cast<StringResult*>(mStringResults.pop());
  74     strRes->mValue = aValue;
  75     strRes->mRecycler = this;
  76     *aResult = strRes;
  77   }
  78   NS_ADDREF(*aResult);
  79
  80   return NS_OK;
  81 }
  82
  83 void txResultRecycler::getEmptyStringResult(txAExprResult** aResult) {
  84   *aResult = mEmptyStringResult;
  85   NS_ADDREF(*aResult);
  86 }
  87
  88 nsresult txResultRecycler::getNodeSet(txNodeSet** aResult) {
  89   if (mNodeSetResults.isEmpty()) {
  90     *aResult = new txNodeSet(this);
  91   } else {
  92     *aResult = static_cast<txNodeSet*>(mNodeSetResults.pop());
  93     (*aResult)->mRecycler = this;
  94   }
  95   NS_ADDREF(*aResult);
  96
  97   return NS_OK;
  98 }
  99
 100 nsresult txResultRecycler::getNodeSet(txNodeSet* aNodeSet,
 101                                       txNodeSet** aResult) {
 102   if (mNodeSetResults.isEmpty()) {
 103     *aResult = new txNodeSet(*aNodeSet, this);
 104   } else {
 105     *aResult = static_cast<txNodeSet*>(mNodeSetResults.pop());
 106     (*aResult)->append(*aNodeSet);
 107     (*aResult)->mRecycler = this;
 108   }
 109   NS_ADDREF(*aResult);
 110
 111   return NS_OK;
 112 }
 113
 114 nsresult txResultRecycler::getNodeSet(const txXPathNode& aNode,
 115                                       txAExprResult** aResult) {
 116   if (mNodeSetResults.isEmpty()) {
 117     *aResult = new txNodeSet(aNode, this);
 118   } else {
 119     txNodeSet* nodes = static_cast<txNodeSet*>(mNodeSetResults.pop());
 120     nodes->append(aNode);
 121     nodes->mRecycler = this;
 122     *aResult = nodes;
 123   }
 124   NS_ADDREF(*aResult);
 125
 126   return NS_OK;
 127 }
 128
 129 nsresult txResultRecycler::getNumberResult(double aValue,
 130                                            txAExprResult** aResult) {
 131   if (mNumberResults.isEmpty()) {
 132     *aResult = new NumberResult(aValue, this);
 133   } else {
 134     NumberResult* numRes = static_cast<NumberResult*>(mNumberResults.pop());
 135     numRes->value = aValue;
 136     numRes->mRecycler = this;
 137     *aResult = numRes;
 138   }
 139   NS_ADDREF(*aResult);
 140
 141   return NS_OK;
 142 }
 143
 144 void txResultRecycler::getBoolResult(bool aValue, txAExprResult** aResult) {
 145   *aResult = aValue ? mTrueResult : mFalseResult;
 146   NS_ADDREF(*aResult);
 147 }
 148
 149 nsresult txResultRecycler::getNonSharedNodeSet(txNodeSet* aNodeSet,
 150                                                txNodeSet** aResult) {
 151   if (aNodeSet->mRefCnt > 1) {
 152     return getNodeSet(aNodeSet, aResult);
 153   }
 154
 155   *aResult = aNodeSet;
 156   NS_ADDREF(*aResult);
 157
 158   return NS_OK;
 159 }
 160
 161 void txAExprResult::Release() {
 162   --mRefCnt;
 163   NS_LOG_RELEASE(this, mRefCnt, "txAExprResult");
 164   if (mRefCnt == 0) {
 165     if (mRecycler) {
 166       mRecycler->recycle(this);
 167     } else {
 168       delete this;
 169     }
 170   }
 171 }