← Back to branch summary

~ubuntu-branches/ubuntu/karmic/kdepim/karmic-backports

« back to all changes in this revision

Viewing changes to kcontactmanager/xxport/csv/qcsvreader.cpp

  • Committer: Bazaar Package Importer
  • Author(s): Alessandro Ghersi, Alessandro Ghersi, Harald Sitter
  • Date: 2009-06-27 04:40:05 UTC
  • mfrom: (1.1.39 upstream)
  • Revision ID: james.westby@ubuntu.com-20090627044005-4y2vm9xz7rvmzi4p
Tags: 4:4.2.95svn20090701-0ubuntu1
https://launchpad.net/bugs/378373
[ Alessandro Ghersi ]
* New upstream release
  - Bump build-deps
* Remove akonadi-kde and libmaildir4 packages
  - remove akonadi-kde.install and libmaildir4.install
  - remove libmaildir4 from debian/rules
  - remove akonadi-kde and libmaildir4 from depends
  - remove akonadi-kde and libmaildir4 from installgen
* Update kdepim-dev.install
* Update kpilot.install
* Add akonadi-kde and libmaildir4 transitional packages

[ Harald Sitter ]
* KAddressbook replaces Kontact << 4.2.85 (LP: #378373)

Show diffs side-by-side

added added

removed removed

Lines of Context:
kcontactmanager/xxport/csv/qcsvreader.cpp
1
 
/*
2
 
    Copyright (c) 2009 Tobias Koenig <tokoe@kde.org>
3
 
 
4
 
    This library is free software; you can redistribute it and/or modify it
5
 
    under the terms of the GNU Library General Public License as published by
6
 
    the Free Software Foundation; either version 2 of the License, or (at your
7
 
    option) any later version.
8
 
 
9
 
    This library is distributed in the hope that it will be useful, but WITHOUT
10
 
    ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11
 
    FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Library General Public
12
 
    License for more details.
13
 
 
14
 
    You should have received a copy of the GNU Library General Public License
15
 
    along with this library; see the file COPYING.LIB.  If not, write to the
16
 
    Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
17
 
    02110-1301, USA.
18
 
*/
19
 
 
20
 
#include "qcsvreader.h"
21
 
 
22
 
#include <QtCore/QStringList>
23
 
#include <QtCore/QTextCodec>
24
 
#include <QtCore/QTextStream>
25
 
 
26
 
QCsvBuilderInterface::~QCsvBuilderInterface()
27
 
{
28
 
}
29
 
 
30
 
class QCsvReader::Private
31
 
{
32
 
  public:
33
 
    Private( QCsvBuilderInterface *builder )
34
 
      : mBuilder( builder )
35
 
    {
36
 
      mTextQuote = QChar( '"' );
37
 
      mDelimiter = QChar( ' ' );
38
 
      mStartRow = 0;
39
 
      mCodec = QTextCodec::codecForLocale();
40
 
 
41
 
      mIgnoreDuplicates = true;
42
 
    }
43
 
 
44
 
    void emitBeginLine( uint row );
45
 
    void emitEndLine( uint row );
46
 
    void emitField( const QString &data, int row, int column );
47
 
 
48
 
    QCsvBuilderInterface *mBuilder;
49
 
    QTextCodec *mCodec;
50
 
    QChar mTextQuote;
51
 
    QChar mDelimiter;
52
 
 
53
 
    uint mStartRow;
54
 
    bool mIgnoreDuplicates;
55
 
};
56
 
 
57
 
void QCsvReader::Private::emitBeginLine( uint row )
58
 
{
59
 
  if ( (row - mStartRow) > 0 )
60
 
    mBuilder->beginLine();
61
 
}
62
 
 
63
 
void QCsvReader::Private::emitEndLine( uint row )
64
 
{
65
 
  if ( (row - mStartRow) > 0 )
66
 
    mBuilder->endLine();
67
 
}
68
 
 
69
 
void QCsvReader::Private::emitField( const QString &data, int row, int column )
70
 
{
71
 
  if ( (row - mStartRow) > 0 )
72
 
    mBuilder->field( data, row - mStartRow - 1, column - 1 );
73
 
}
74
 
 
75
 
QCsvReader::QCsvReader( QCsvBuilderInterface *builder )
76
 
  : d( new Private( builder ) )
77
 
{
78
 
  Q_ASSERT( builder );
79
 
}
80
 
 
81
 
QCsvReader::~QCsvReader()
82
 
{
83
 
  delete d;
84
 
}
85
 
 
86
 
bool QCsvReader::read( QIODevice *device )
87
 
{
88
 
  int row, column;
89
 
  bool lastCharDelimiter = false;
90
 
  bool ignoreDups = d->mIgnoreDuplicates;
91
 
  enum { S_START, S_QUOTED_FIELD, S_MAYBE_END_OF_QUOTED_FIELD, S_END_OF_QUOTED_FIELD,
92
 
         S_MAYBE_NORMAL_FIELD, S_NORMAL_FIELD } state = S_START;
93
 
 
94
 
  QChar x;
95
 
  QString field;
96
 
 
97
 
  row = column = 1;
98
 
 
99
 
  d->mBuilder->begin();
100
 
  d->emitBeginLine( row );
101
 
 
102
 
  if ( !device->isOpen() ) {
103
 
    d->mBuilder->error( "Device is not open" );
104
 
    d->emitEndLine( row );
105
 
    d->mBuilder->end();
106
 
    return false;
107
 
  }
108
 
 
109
 
  QTextStream inputStream( device );
110
 
  inputStream.setCodec( d->mCodec );
111
 
 
112
 
  int maxColumn = 0;
113
 
  while ( !inputStream.atEnd() ) {
114
 
    inputStream >> x; // read one char
115
 
 
116
 
    if ( x == '\r' ) inputStream >> x; // eat '\r', to handle DOS/LOSEDOWS files correctly
117
 
 
118
 
    switch ( state ) {
119
 
      case S_START:
120
 
        if ( x == d->mTextQuote ) {
121
 
          state = S_QUOTED_FIELD;
122
 
        } else if ( x == d->mDelimiter ) {
123
 
          if ( ( ignoreDups == false ) || ( lastCharDelimiter == false ) )
124
 
            ++column;
125
 
          lastCharDelimiter = true;
126
 
        } else if ( x == '\n' ) {
127
 
          d->emitEndLine( row );
128
 
          ++row;
129
 
          column = 1;
130
 
          d->emitBeginLine( row );
131
 
        } else {
132
 
          field += x;
133
 
          state = S_MAYBE_NORMAL_FIELD;
134
 
        }
135
 
        break;
136
 
      case S_QUOTED_FIELD:
137
 
        if ( x == d->mTextQuote ) {
138
 
          state = S_MAYBE_END_OF_QUOTED_FIELD;
139
 
        } else if ( x == '\n' &&  d->mTextQuote.isNull() ) {
140
 
          d->emitField( field, row, column );
141
 
          field.clear();
142
 
          if ( x == '\n' ) {
143
 
            d->emitEndLine( row );
144
 
            ++row;
145
 
            column = 1;
146
 
            d->emitBeginLine( row );
147
 
          } else {
148
 
            if ( ( ignoreDups == false ) || ( lastCharDelimiter == false ) )
149
 
              ++column;
150
 
            lastCharDelimiter = true;
151
 
          }
152
 
          state = S_START;
153
 
        } else {
154
 
          field += x;
155
 
        }
156
 
        break;
157
 
      case S_MAYBE_END_OF_QUOTED_FIELD:
158
 
        if ( x == d->mTextQuote ) {
159
 
          field += x;
160
 
          state = S_QUOTED_FIELD;
161
 
        } else if ( x == d->mDelimiter || x == '\n' ) {
162
 
          d->emitField( field, row, column );
163
 
          field.clear();
164
 
          if ( x == '\n' ) {
165
 
            d->emitEndLine( row );
166
 
            ++row;
167
 
            column = 1;
168
 
            d->emitBeginLine( row );
169
 
          } else {
170
 
            if ( ( ignoreDups == false ) || ( lastCharDelimiter == false ) )
171
 
              ++column;
172
 
            lastCharDelimiter = true;
173
 
          }
174
 
          state = S_START;
175
 
        } else {
176
 
          state = S_END_OF_QUOTED_FIELD;
177
 
        }
178
 
        break;
179
 
      case S_END_OF_QUOTED_FIELD:
180
 
        if ( x == d->mDelimiter || x == '\n' ) {
181
 
          d->emitField( field, row, column );
182
 
          field.clear();
183
 
          if ( x == '\n' ) {
184
 
            d->emitEndLine( row );
185
 
            ++row;
186
 
            column = 1;
187
 
            d->emitBeginLine( row );
188
 
          } else {
189
 
            if ( ( ignoreDups == false ) || ( lastCharDelimiter == false ) )
190
 
              ++column;
191
 
            lastCharDelimiter = true;
192
 
          }
193
 
          state = S_START;
194
 
        } else {
195
 
          state = S_END_OF_QUOTED_FIELD;
196
 
        }
197
 
        break;
198
 
      case S_MAYBE_NORMAL_FIELD:
199
 
        if ( x == d->mTextQuote ) {
200
 
          field.clear();
201
 
          state = S_QUOTED_FIELD;
202
 
          break;
203
 
        }
204
 
      case S_NORMAL_FIELD:
205
 
        if ( x == d->mDelimiter || x == '\n' ) {
206
 
          d->emitField( field, row, column );
207
 
          field.clear();
208
 
          if ( x == '\n' ) {
209
 
            d->emitEndLine( row );
210
 
            ++row;
211
 
            column = 1;
212
 
            d->emitBeginLine( row );
213
 
          } else {
214
 
            if ( ( ignoreDups == false ) || ( lastCharDelimiter == false ) )
215
 
              ++column;
216
 
            lastCharDelimiter = true;
217
 
          }
218
 
          state = S_START;
219
 
        } else {
220
 
          field += x;
221
 
        }
222
 
    }
223
 
    if ( x != d->mDelimiter )
224
 
      lastCharDelimiter = false;
225
 
 
226
 
    if ( column > maxColumn )
227
 
      maxColumn = column;
228
 
  }
229
 
 
230
 
  // file with only one line without '\n'
231
 
  if ( field.length() > 0 ) {
232
 
    d->emitField( field, row, column );
233
 
    ++row;
234
 
    field.clear();
235
 
  }
236
 
 
237
 
  d->emitEndLine( row );
238
 
  d->mBuilder->end();
239
 
 
240
 
  return true;
241
 
}
242
 
 
243
 
void QCsvReader::setTextQuote( const QChar &textQuote )
244
 
{
245
 
  d->mTextQuote = textQuote;
246
 
}
247
 
 
248
 
QChar QCsvReader::textQuote() const
249
 
{
250
 
  return d->mTextQuote;
251
 
}
252
 
 
253
 
void QCsvReader::setDelimiter( const QChar &delimiter )
254
 
{
255
 
  d->mDelimiter = delimiter;
256
 
}
257
 
 
258
 
QChar QCsvReader::delimiter() const
259
 
{
260
 
  return d->mDelimiter;
261
 
}
262
 
 
263
 
void QCsvReader::setStartRow( uint startRow )
264
 
{
265
 
  d->mStartRow = startRow;
266
 
}
267
 
 
268
 
uint QCsvReader::startRow() const
269
 
{
270
 
  return d->mStartRow;
271
 
}
272
 
 
273
 
void QCsvReader::setTextCodec( QTextCodec *textCodec )
274
 
{
275
 
  d->mCodec = textCodec;
276
 
}
277
 
 
278
 
QTextCodec *QCsvReader::textCodec() const
279
 
{
280
 
  return d->mCodec;
281
 
}
282
 
 
283
 
 
284
 
class QCsvStandardBuilder::Private
285
 
{
286
 
  public:
287
 
    Private()
288
 
    {
289
 
      init();
290
 
    }
291
 
 
292
 
    void init();
293
 
 
294
 
    QString mLastErrorString;
295
 
    uint mRowCount;
296
 
    uint mColumnCount;
297
 
    QList<QStringList> mRows;
298
 
};
299
 
 
300
 
void QCsvStandardBuilder::Private::init()
301
 
{
302
 
  mRows.clear();
303
 
  mRowCount = 0;
304
 
  mColumnCount = 0;
305
 
  mLastErrorString.clear();
306
 
}
307
 
 
308
 
QCsvStandardBuilder::QCsvStandardBuilder()
309
 
  : d( new Private )
310
 
{
311
 
}
312
 
 
313
 
QCsvStandardBuilder::~QCsvStandardBuilder()
314
 
{
315
 
  delete d;
316
 
}
317
 
 
318
 
QString QCsvStandardBuilder::lastErrorString() const
319
 
{
320
 
  return d->mLastErrorString;
321
 
}
322
 
 
323
 
uint QCsvStandardBuilder::rowCount() const
324
 
{
325
 
  return d->mRowCount;
326
 
}
327
 
 
328
 
uint QCsvStandardBuilder::columnCount() const
329
 
{
330
 
  return d->mColumnCount;
331
 
}
332
 
 
333
 
QString QCsvStandardBuilder::data( uint row, uint column ) const
334
 
{
335
 
  if ( row > d->mRowCount || column > d->mColumnCount || column >= (uint)d->mRows[ row ].count() )
336
 
    return QString();
337
 
 
338
 
  return d->mRows[ row ][ column ];
339
 
}
340
 
 
341
 
void QCsvStandardBuilder::begin()
342
 
{
343
 
  d->init();
344
 
}
345
 
 
346
 
void QCsvStandardBuilder::beginLine()
347
 
{
348
 
  d->mRows.append( QStringList() );
349
 
  d->mRowCount++;
350
 
}
351
 
 
352
 
void QCsvStandardBuilder::field( const QString &data, uint row, uint column )
353
 
{
354
 
  const uint size = d->mRows[ row ].size();
355
 
  if ( column >= size ) {
356
 
    for ( uint i = column; i < size + 1; ++i )
357
 
      d->mRows[ row ].append( QString() );
358
 
  }
359
 
 
360
 
  d->mRows[ row ][ column ] = data;
361
 
 
362
 
  d->mColumnCount = qMax( d->mColumnCount, column + 1 );
363
 
}
364
 
 
365
 
void QCsvStandardBuilder::endLine()
366
 
{
367
 
}
368
 
 
369
 
void QCsvStandardBuilder::end()
370
 
{
371
 
}
372
 
 
373
 
void QCsvStandardBuilder::error( const QString &errorMsg )
374
 
{
375
 
  d->mLastErrorString = errorMsg;
376
 
}
377
 
 
378
 
#include "qcsvreader.moc"