gemrb/plugins/2DAImporter/2DAImporter.h

   1 /* GemRB - Infinity Engine Emulator
   2  * Copyright (C) 2003 The GemRB Project
   3  *
   4  * This program is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU General Public License
   6  * as published by the Free Software Foundation; either version 2
   7  * of the License, or (at your option) any later version.
   8
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  17  *
  18  *
  19  */
  20
  21 #ifndef P2DAIMPORTER_H
  22 #define P2DAIMPORTER_H
  23
  24 #include "TableMgr.h"
  25
  26 #include "globals.h"
  27
  28 #include <cstring>
  29
  30 typedef std::vector< char*> RowEntry;
  31
  32 class p2DAImporter : public TableMgr {
  33 private:
  34         DataStream* str;
  35         bool autoFree;
  36         std::vector< char*> colNames;
  37         std::vector< char*> rowNames;
  38         std::vector< char*> ptrs;
  39         std::vector< RowEntry> rows;
  40         char defVal[32];
  41 public:
  42         p2DAImporter(void);
  43         ~p2DAImporter(void);
  44         bool Open(DataStream* stream, bool autoFree = true);
  45         /** Returns the actual number of Rows in the Table */
  46         inline ieDword GetRowCount() const
  47         {
  48                 return ( ieDword ) rows.size();
  49         }
  50
  51         inline ieDword GetColNamesCount() const
  52         {
  53                 return (ieDword) colNames.size();
  54         }
  55
  56         /** Returns the actual number of Columns in the Table */
  57         inline ieDword GetColumnCount(unsigned int row = 0) const
  58         {
  59                 if (rows.size() <= row) {
  60                         return 0;
  61                 }
  62                 return ( ieDword ) rows[row].size();
  63         }
  64         /** Returns a pointer to a zero terminated 2da element,
  65                 if it cannot return a value, it returns the default */
  66         inline const char* QueryField(unsigned int row = 0, unsigned int column = 0) const
  67         {
  68                 if (rows.size() <= row) {
  69                         return ( char * ) defVal;
  70                 }
  71                 if (rows[row].size() <= column) {
  72                         return ( char * ) defVal;
  73                 }
  74                 if (rows[row][column][0]=='*' && !rows[row][column][1]) {
  75                         return ( char * ) defVal;
  76                 }
  77                 return rows[row][column];
  78         }
  79         /** Returns a pointer to a zero terminated 2da element,
  80                  uses column name and row name to search the field */
  81         inline const char* QueryField(const char* row, const char* column) const
  82         {
  83                 int rowi, coli;
  84
  85                 rowi = GetRowIndex(row);
  86
  87                 if (rowi < 0) {
  88                         return ( char * ) defVal;
  89                 }
  90
  91                 coli = GetColumnIndex(column);
  92
  93                 if (coli < 0) {
  94                         return ( char * ) defVal;
  95                 }
  96
  97                 return QueryField((unsigned int) rowi, (unsigned int) coli);
  98         }
  99
 100         virtual const char* QueryDefault() const
 101         {
 102                 return defVal;
 103         }
 104
 105         inline int GetRowIndex(const char* string) const
 106         {
 107                 for (unsigned int index = 0; index < rowNames.size(); index++) {
 108                         if (stricmp( rowNames[index], string ) == 0) {
 109                                 return (int) index;
 110                         }
 111                 }
 112                 return -1;
 113         }
 114
 115         inline int GetColumnIndex(const char* string) const
 116         {
 117                 for (unsigned int index = 0; index < colNames.size(); index++) {
 118                         if (stricmp( colNames[index], string ) == 0) {
 119                                 return (int) index;
 120                         }
 121                 }
 122                 return -1;
 123         }
 124
 125         inline const char* GetColumnName(unsigned int index) const
 126         {
 127                 if (index < colNames.size()) {
 128                         return colNames[index];
 129                 }
 130                 return "";
 131         }
 132
 133         inline const char* GetRowName(unsigned int index) const
 134         {
 135                 if (index < rowNames.size()) {
 136                         return rowNames[index];
 137                 }
 138                 return "";
 139         }
 140
 141         inline int FindTableValue(unsigned int col, long val, int start) const
 142         {
 143                 ieDword row, max;
 144
 145                 max = GetRowCount();
 146                 for (row = start; row < max; row++) {
 147                         const char* ret = QueryField( row, col );
 148                         long Value;
 149                         if (valid_number( ret, Value ) && (Value == val) )
 150                                 return (int) row;
 151                 }
 152                 return -1;
 153         }
 154
 155         inline int FindTableValue(unsigned int col, const char* val, int start) const
 156         {
 157                 ieDword row, max;
 158
 159                 max = GetRowCount();
 160                 for (row = start; row < max; row++) {
 161                         const char* ret = QueryField( row, col );
 162                         if (stricmp(ret, val) == 0)
 163                                 return (int) row;
 164                 }
 165                 return -1;
 166         }
 167 };
 168
 169 #endif