summaryrefslogtreecommitdiffstats
path: root/src/Client/XLSSheet.cpp
blob: 915234ff5e7de20f2f8b51eb6fdfe68781671ee7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
/*
 * XLSSheet.cpp
 *
 * Copyright (C) 2009 Matthias Schiffer <matthias@gamezock.de>
 *
 * This program is free software: you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as published by the
 * Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License along
 * with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#include "XLSSheet.h"
#include "XLSReader.h"

#include <cstdlib>
#include <sstream>

namespace Mad {
namespace Client {

std::string XLSSheet::getSheetLevelField(const std::string &name) const {
  xmlNodePtr entry = XLSReader::findNode(node, name);
  if(entry)
    return std::string((char*)xmlNodeGetContent(entry));
  else
    return std::string();
}

void XLSSheet::readRows() {
  rows.clear();

  xmlNodePtr rowsNode = XLSReader::findNode(node, "rows");

  xmlNodePtr rowEntry = rowsNode->children;

  if(firstRowColNames) {
    for(; rowEntry != 0; rowEntry = rowEntry->next) {
      if(rowEntry->type != XML_ELEMENT_NODE || xmlStrcmp(rowEntry->name, (xmlChar*)"row"))
        continue;

      for(xmlNodePtr cellEntry = rowEntry->children; cellEntry != 0; cellEntry = cellEntry->next) {
        if(cellEntry->type != XML_ELEMENT_NODE || xmlStrcmp(cellEntry->name, (xmlChar*)"cell"))
          continue;

        xmlChar *colStr = xmlGetProp(cellEntry, (xmlChar*)"col");
        if(!colStr)
          continue;

        unsigned long col = strtoul((char*)colStr, 0, 10);
        if(col >= colCount)
          continue;

        colNames[col] = (char*)xmlNodeGetContent(cellEntry);
      }

      // Skip the first row
      rowEntry = rowEntry->next;
      break;
    }
  }

  for(; rowEntry != 0; rowEntry = rowEntry->next) {
    if(rowEntry->type != XML_ELEMENT_NODE || xmlStrcmp(rowEntry->name, (xmlChar*)"row"))
      continue;

    boost::shared_ptr<std::vector<std::string> > rowVector(new std::vector<std::string>(colCount));
    rows.push_back(rowVector);

    for(xmlNodePtr cellEntry = rowEntry->children; cellEntry != 0; cellEntry = cellEntry->next) {
      if(cellEntry->type != XML_ELEMENT_NODE || xmlStrcmp(cellEntry->name, (xmlChar*)"cell"))
        continue;

      xmlChar *colStr = xmlGetProp(cellEntry, (xmlChar*)"col");
      if(!colStr)
        continue;

      unsigned long col = strtoul((char*)colStr, 0, 10);
      if(col >= colCount)
        continue;

      (*rowVector)[col] = (char*)xmlNodeGetContent(cellEntry);
    }
  }
}

XLSSheet::XLSSheet(boost::shared_ptr<xmlDoc> doc0, xmlNodePtr node0) : doc(doc0), node(node0), firstRowColNames(false) {
  title = getSheetLevelField("pagetitle");
  colCount = std::strtoul(getSheetLevelField("lastcol").c_str(), 0, 10)+1-std::strtoul(getSheetLevelField("firstcol").c_str(), 0, 10);

  for(unsigned col = 0; col < colCount; ++col) {
    std::ostringstream stream;
    stream << col;

    colNames.push_back(stream.str());
  }

  readRows();
}

}
}