view/aaTreeViewEntity.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 <abstract/aacore.h>
  23
  24 #include "nsCOMPtr.h"
  25 #include "nsStringAPI.h"
  26
  27 /* Unfrozen API */
  28 #include "unstable/nsIArray.h"
  29 #include "unstable/nsArrayUtils.h"
  30 #include "unstable/nsITreeColumns.h"
  31
  32 /* Project includes */
  33 #include <abstract/base/aaIEntity.h>
  34 #include <abstract/storage/aaILoadQuery.h>
  35 #include "aaTreeView.h"
  36 #include "aaTreeViewEntity.h"
  37
  38 aaTreeViewEntity::aaTreeViewEntity()
  39 {
  40 }
  41
  42 aaTreeViewEntity::~aaTreeViewEntity()
  43 {
  44 }
  45
  46 NS_IMPL_ISUPPORTS_INHERITED0(aaTreeViewEntity, aaTreeView)
  47
  48 /* Private functions */
  49 nsresult
  50 aaTreeViewEntity::doGetCellText(PRInt32 row, nsITreeColumn *col,
  51     nsAString & _retval)
  52 {
  53   nsresult rv;
  54   NS_ENSURE_TRUE(col, NS_ERROR_INVALID_ARG);
  55
  56   PRInt32 ci;
  57   rv = col->GetIndex(&ci);
  58   NS_ENSURE_SUCCESS(rv, rv);
  59   if (ci > 0) {
  60     _retval.Truncate();
  61     return NS_OK;
  62   }
  63
  64   nsCOMPtr<aaIEntity> entity(do_QueryElementAt(mDataSet, row, &rv));
  65   NS_ENSURE_SUCCESS(rv, rv);
  66   if (entity)
  67     return entity->GetTag(_retval);
  68
  69   _retval.Truncate();
  70   return NS_OK;
  71 }