1
/* This file is part of the KDE project
2
Copyright (C) 1999 David Faure <faure@kde.org>
3
Copyright (C) 2004 Nicolas GOUTTE <goutte@kde.org>
5
This library is free software; you can redistribute it and/or
6
modify it under the terms of the GNU Library General Public
7
License as published by the Free Software Foundation; either
8
version 2 of the License, or (at your option) any later version.
10
This library is distributed in the hope that it will be useful,
11
but WITHOUT ANY WARRANTY; without even the implied warranty of
12
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13
Library General Public License for more details.
15
You should have received a copy of the GNU Library General Public License
16
along with this library; see the file COPYING.LIB. If not, write to
17
the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
18
Boston, MA 02111-1307, USA.
21
#include <csvimportdialogui.h>
22
#include <csvimportdialog.h>
25
#include <qcheckbox.h>
27
#include <qlineedit.h>
28
#include <qcombobox.h>
30
#include <qtextstream.h>
31
#include <qbuttongroup.h>
32
#include <qpushbutton.h>
33
#include <qradiobutton.h>
34
#include <qtextcodec.h>
36
#include <kapplication.h>
39
#include <kcombobox.h>
40
#include <kmessagebox.h>
41
#include <kcharsets.h>
44
CSVImportDialog::CSVImportDialog(QWidget* parent, QByteArray& fileArray)
45
: KDialogBase(parent, 0, true, QString::null, Ok|Cancel, No, true),
46
m_dialog(new DialogUI(this)),
56
m_fileArray(fileArray),
57
m_codec( QTextCodec::codecForName( "UTF-8" ) )
59
setCaption( i18n( "Import Data" ) );
60
kapp->restoreOverrideCursor();
62
QStringList encodings;
63
encodings << i18n( "Descriptive encoding name", "Recommended ( %1 )" ).arg( "UTF-8" );
64
encodings << i18n( "Descriptive encoding name", "Locale ( %1 )" ).arg( QTextCodec::codecForLocale()->name() );
65
encodings += KGlobal::charsets()->descriptiveEncodingNames();
66
// Add a few non-standard encodings, which might be useful for text files
67
const QString description(i18n("Descriptive encoding name","Other ( %1 )"));
68
encodings << description.arg("Apple Roman"); // Apple
69
encodings << description.arg("IBM 850") << description.arg("IBM 866"); // MS DOS
70
encodings << description.arg("CP 1258"); // Windows
71
m_dialog->comboBoxEncoding->insertStringList(encodings);
73
m_formatList << i18n( "Text" );
74
m_formatList << i18n( "Number" );
75
//m_formatList << i18n( "Currency" );
76
//m_formatList << i18n( "Date" );
77
m_formatList << i18n( "Decimal Comma Number" );
78
m_formatList << i18n( "Decimal Point Number" );
79
m_dialog->m_formatComboBox->insertStringList( m_formatList );
81
m_dialog->m_sheet->setReadOnly( true );
86
resize( 600, 400 ); // Try to show as much as possible of the table view
87
setMainWidget(m_dialog);
89
m_dialog->m_sheet->setSelectionMode( QTable::Multi );
91
connect(m_dialog->m_formatComboBox, SIGNAL(activated( const QString& )),
92
this, SLOT(formatChanged( const QString& )));
93
connect(m_dialog->m_delimiterBox, SIGNAL(clicked(int)),
94
this, SLOT(delimiterClicked(int)));
95
connect(m_dialog->m_delimiterEdit, SIGNAL(returnPressed()),
96
this, SLOT(returnPressed()));
97
connect(m_dialog->m_delimiterEdit, SIGNAL(textChanged ( const QString & )),
98
this, SLOT(formatChanged ( const QString & ) ));
99
connect(m_dialog->m_comboQuote, SIGNAL(activated(const QString &)),
100
this, SLOT(textquoteSelected(const QString &)));
101
connect(m_dialog->m_sheet, SIGNAL(currentChanged(int, int)),
102
this, SLOT(currentCellChanged(int, int)));
103
connect(m_dialog->m_ignoreDuplicates, SIGNAL(stateChanged(int)),
104
this, SLOT(ignoreDuplicatesChanged(int)));
105
connect(m_dialog->m_updateButton, SIGNAL(clicked()),
106
this, SLOT(updateClicked()));
107
connect(m_dialog->comboBoxEncoding, SIGNAL(textChanged ( const QString & )),
108
this, SLOT(encodingChanged ( const QString & ) ));
112
CSVImportDialog::~CSVImportDialog()
114
kapp->setOverrideCursor(Qt::waitCursor);
118
// ----------------------------------------------------------------
122
bool CSVImportDialog::firstRowContainHeaders()
124
return m_dialog->m_firstRowHeader->isChecked();
128
bool CSVImportDialog::firstColContainHeaders()
130
return m_dialog->m_firstColHeader->isChecked();
134
int CSVImportDialog::rows()
136
int rows = m_dialog->m_sheet->numRows();
139
rows = m_endRow - m_startRow + 1;
145
int CSVImportDialog::cols()
147
int cols = m_dialog->m_sheet->numCols();
150
cols = m_endCol - m_startCol + 1;
156
QString CSVImportDialog::text(int row, int col)
158
// Check for overflow.
159
if ( row >= rows() || col >= cols())
160
return QString::null;
162
return m_dialog->m_sheet->text( row - m_startRow, col - m_startCol );
166
// ----------------------------------------------------------------
169
void CSVImportDialog::fillTable( )
172
bool lastCharDelimiter = false;
173
enum { S_START, S_QUOTED_FIELD, S_MAYBE_END_OF_QUOTED_FIELD, S_END_OF_QUOTED_FIELD,
174
S_MAYBE_NORMAL_FIELD, S_NORMAL_FIELD } state = S_START;
179
kapp->setOverrideCursor(Qt::waitCursor);
181
for (row = 0; row < m_dialog->m_sheet->numRows(); ++row)
182
for (column = 0; column < m_dialog->m_sheet->numCols(); ++column)
183
m_dialog->m_sheet->clearCell(row, column);
187
QTextStream inputStream(m_fileArray, IO_ReadOnly);
188
kdDebug(30501) << "Encoding: " << m_codec->name() << endl;
189
inputStream.setCodec( m_codec );
191
bool lastCharWasCr = false; // Last character was a Carriage Return
192
while (!inputStream.atEnd())
194
inputStream >> x; // read one char
196
// ### TODO: we should perhaps skip all other control characters
199
// We have a Carriage Return, assume that its role is the one of a LineFeed
200
lastCharWasCr = true;
201
x = '\n'; // Replace by Line Feed
203
else if ( x == '\n' && lastCharWasCr )
205
// The end of line was already handled by the Carriage Return, so do nothing for this character
206
lastCharWasCr = false;
209
else if ( x == QChar( 0xc ) )
211
// We have a FormFeed, skip it
212
lastCharWasCr = false;
217
lastCharWasCr = false;
220
if ( column > maxColumn )
226
if (x == m_textquote)
228
state = S_QUOTED_FIELD;
230
else if (x == m_delimiter)
232
if ((m_ignoreDups == false) || (lastCharDelimiter == false))
234
lastCharDelimiter = true;
240
if ( row > ( m_endRow - m_startRow ) && m_endRow >= 0 )
246
state = S_MAYBE_NORMAL_FIELD;
249
case S_QUOTED_FIELD :
250
if (x == m_textquote)
252
state = S_MAYBE_END_OF_QUOTED_FIELD;
256
setText(row - m_startRow, column - m_startCol, field);
257
field = QString::null;
261
if ( row > ( m_endRow - m_startRow ) && m_endRow >= 0 )
271
case S_MAYBE_END_OF_QUOTED_FIELD :
272
if (x == m_textquote)
275
state = S_QUOTED_FIELD;
277
else if (x == m_delimiter || x == '\n')
279
setText(row - m_startRow, column - m_startCol, field);
280
field = QString::null;
285
if ( row > ( m_endRow - m_startRow ) && m_endRow >= 0 )
290
if ((m_ignoreDups == false) || (lastCharDelimiter == false))
292
lastCharDelimiter = true;
298
state = S_END_OF_QUOTED_FIELD;
301
case S_END_OF_QUOTED_FIELD :
302
if (x == m_delimiter || x == '\n')
304
setText(row - m_startRow, column - m_startCol, field);
305
field = QString::null;
310
if ( row > ( m_endRow - m_startRow ) && m_endRow >= 0 )
315
if ((m_ignoreDups == false) || (lastCharDelimiter == false))
317
lastCharDelimiter = true;
323
state = S_END_OF_QUOTED_FIELD;
326
case S_MAYBE_NORMAL_FIELD :
327
if (x == m_textquote)
329
field = QString::null;
330
state = S_QUOTED_FIELD;
333
case S_NORMAL_FIELD :
334
if (x == m_delimiter || x == '\n')
336
setText(row - m_startRow, column - m_startCol, field);
337
field = QString::null;
342
if ( row > ( m_endRow - m_startRow ) && m_endRow >= 0 )
347
if ((m_ignoreDups == false) || (lastCharDelimiter == false))
349
lastCharDelimiter = true;
358
if (x != m_delimiter)
359
lastCharDelimiter = false;
362
if ( !field.isEmpty() )
364
// the last line of the file had not any line end
365
setText(row - m_startRow, column - m_startCol, field);
367
field = QString::null;
371
adjustRows( row - m_startRow );
372
adjustCols( maxColumn - m_startCol );
373
m_dialog->m_colEnd->setMaxValue( maxColumn );
374
if ( m_endCol == -1 )
375
m_dialog->m_colEnd->setValue( maxColumn );
378
for (column = 0; column < m_dialog->m_sheet->numCols(); ++column)
380
const QString header = m_dialog->m_sheet->horizontalHeader()->label(column);
381
if ( m_formatList.find( header ) == m_formatList.end() )
382
m_dialog->m_sheet->horizontalHeader()->setLabel(column, i18n("Text"));
384
m_dialog->m_sheet->adjustColumn(column);
388
kapp->restoreOverrideCursor();
391
void CSVImportDialog::fillComboBox()
393
if ( m_endRow == -1 )
394
m_dialog->m_rowEnd->setValue( m_dialog->m_sheet->numRows() );
396
m_dialog->m_rowEnd->setValue( m_endRow );
398
if ( m_endCol == -1 )
399
m_dialog->m_colEnd->setValue( m_dialog->m_sheet->numCols() );
401
m_dialog->m_colEnd->setValue( m_endCol );
403
m_dialog->m_rowEnd->setMinValue( 1 );
404
m_dialog->m_colEnd->setMinValue( 1 );
405
m_dialog->m_rowEnd->setMaxValue( m_dialog->m_sheet->numRows() );
406
m_dialog->m_colEnd->setMaxValue( m_dialog->m_sheet->numCols() );
408
m_dialog->m_rowStart->setMinValue( 1 );
409
m_dialog->m_colStart->setMinValue( 1 );
410
m_dialog->m_rowStart->setMaxValue( m_dialog->m_sheet->numRows() );
411
m_dialog->m_colStart->setMaxValue( m_dialog->m_sheet->numCols() );
414
int CSVImportDialog::headerType(int col)
416
QString header = m_dialog->m_sheet->horizontalHeader()->label(col);
418
if (header == i18n("Text"))
420
else if (header == i18n("Number"))
422
else if (header == i18n("Currency"))
424
else if ( header == i18n( "Date" ) )
426
else if ( header == i18n( "Decimal Comma Number" ) )
428
else if ( header == i18n( "Decimal Point Number" ) )
431
return TEXT; // Should not happen
434
void CSVImportDialog::setText(int row, int col, const QString& text)
436
if ( row < 1 || col < 1 ) // skipped by the user
439
if ( ( row > ( m_endRow - m_startRow ) && m_endRow > 0 ) || ( col > ( m_endCol - m_startCol ) && m_endCol > 0 ) )
442
if ( m_dialog->m_sheet->numRows() < row )
444
m_dialog->m_sheet->setNumRows( row + 5000 ); /* We add 5000 at a time to limit recalculations */
448
if ( m_dialog->m_sheet->numCols() < col )
450
m_dialog->m_sheet->setNumCols( col );
454
m_dialog->m_sheet->setText( row - 1, col - 1, text );
458
* Called after the first fillTable() when number of rows are unknown.
460
void CSVImportDialog::adjustRows(int iRows)
464
m_dialog->m_sheet->setNumRows( iRows );
465
m_adjustRows = false;
469
void CSVImportDialog::adjustCols(int iCols)
473
m_dialog->m_sheet->setNumCols( iCols );
474
m_adjustCols = false;
476
if ( m_endCol == -1 )
478
if ( iCols > ( m_endCol - m_startCol ) )
479
iCols = m_endCol - m_startCol;
481
m_dialog->m_sheet->setNumCols( iCols );
486
void CSVImportDialog::returnPressed()
488
if (m_dialog->m_delimiterBox->id(m_dialog->m_delimiterBox->selected()) != 4)
491
m_delimiter = m_dialog->m_delimiterEdit->text();
495
void CSVImportDialog::textChanged ( const QString & )
497
m_dialog->m_radioOther->setChecked ( true );
498
delimiterClicked(4); // other
501
void CSVImportDialog::formatChanged( const QString& newValue )
503
//kdDebug(30501) << "CSVImportDialog::formatChanged:" << newValue << endl;
504
for ( int i = 0; i < m_dialog->m_sheet->numSelections(); ++i )
506
QTableSelection select ( m_dialog->m_sheet->selection( i ) );
507
for ( int j = select.leftCol(); j <= select.rightCol() ; ++j )
509
m_dialog->m_sheet->horizontalHeader()->setLabel( j, newValue );
515
void CSVImportDialog::delimiterClicked(int id)
523
m_delimiter = m_dialog->m_delimiterEdit->text();
539
void CSVImportDialog::textquoteSelected(const QString& mark)
541
if (mark == i18n("None"))
544
m_textquote = mark[0];
549
void CSVImportDialog::updateClicked()
551
if ( !checkUpdateRange() )
554
m_startRow = m_dialog->m_rowStart->value() - 1;
555
m_endRow = m_dialog->m_rowEnd->value();
557
m_startCol = m_dialog->m_colStart->value() - 1;
558
m_endCol = m_dialog->m_colEnd->value();
563
bool CSVImportDialog::checkUpdateRange()
565
if ( ( m_dialog->m_rowStart->value() > m_dialog->m_rowEnd->value() )
566
|| ( m_dialog->m_colStart->value() > m_dialog->m_colEnd->value() ) )
568
KMessageBox::error( this, i18n( "Please check the ranges you specified. The start value must be lower than the end value." ) );
575
void CSVImportDialog::currentCellChanged(int, int col)
577
const QString header = m_dialog->m_sheet->horizontalHeader()->label(col);
578
m_dialog->m_formatComboBox->setCurrentText( header );
581
void CSVImportDialog::ignoreDuplicatesChanged(int)
583
if (m_dialog->m_ignoreDuplicates->isChecked())
586
m_ignoreDups = false;
590
QTextCodec* CSVImportDialog::getCodec(void) const
592
const QString strCodec( KGlobal::charsets()->encodingForName( m_dialog->comboBoxEncoding->currentText() ) );
593
kdDebug(30502) << "Encoding: " << strCodec << endl;
596
QTextCodec* codec = QTextCodec::codecForName( strCodec.utf8() );
598
// If QTextCodec has not found a valid encoding, so try with KCharsets.
605
codec = KGlobal::charsets()->codecForName( strCodec, ok );
612
kdWarning(30502) << "Cannot find encoding:" << strCodec << endl;
613
// ### TODO: what parent to use?
614
KMessageBox::error( 0, i18n("Cannot find encoding: %1").arg( strCodec ) );
621
void CSVImportDialog::encodingChanged ( const QString & )
623
QTextCodec* codec = getCodec();
633
#include <csvimportdialog.moc>