report/aaDiffCounter.cpp

   1 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim:set ts=2 sw=2 sts=2 et cindent tw=79 ft=cpp: */
   3 /*
   4  * Copyright (C) 2007 Sergey Yanovich <ynvich@gmail.com>
   5  *
   6  * This program is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU General Public License as
   8  * published by the Free Software Foundation; either version 2 of the
   9  * License, or (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public
  17  * License along with this program; if not, write to the
  18  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  19  * Boston, MA 02111-1307, USA.
  20  */
  21
  22 #include "xpcom-config.h"
  23
  24 #include "nsCOMPtr.h"
  25 #include "nsWeakReference.h"
  26
  27 /* Unfrozen API */
  28
  29 /* Project includes */
  30 #include "aaIFlow.h"
  31 #include "aaIFact.h"
  32 #include "aaIBalance.h"
  33 #include "aaDiffCounter.h"
  34
  35 NS_IMPL_QUERY_INTERFACE2(aaDiffCounter,
  36                          nsISupportsWeakReference,
  37                          aaILoadObserver)
  38
  39 NS_IMETHODIMP_(nsrefcnt)
  40 aaDiffCounter::AddRef(void)
  41 {
  42   return 1;
  43 }
  44
  45 NS_IMETHODIMP_(nsrefcnt)
  46 aaDiffCounter::Release(void)
  47 {
  48   return 1;
  49 }
  50
  51 NS_IMETHODIMP
  52 aaDiffCounter::ObserveLoad(aaIDataNode *aNode)
  53 {
  54   nsresult rv;
  55   nsCOMPtr<aaIBalance> balance = do_QueryInterface(aNode, &rv);
  56   NS_ENSURE_SUCCESS(rv, rv);
  57
  58   if (balance->PickSide()) {
  59     mDebits += balance->PickDiff();
  60   } else {
  61     mCredits += balance->PickDiff();
  62   }
  63
  64   return NS_OK;
  65 }