diff --git a/src/exportdialog.cpp b/src/exportdialog.cpp index 960c030b..7ca1db32 100644 --- a/src/exportdialog.cpp +++ b/src/exportdialog.cpp @@ -1,241 +1,246 @@ /*************************************************************************** copyright : (C) 2007 by Johannes Simon email : johannes.simon@gmail.com ***************************************************************************/ /*************************************************************************** * * * This program is free software; you can redistribute it and/or modify * * it under the terms of the GNU General Public License as published by * * the Free Software Foundation; either version 2 of the License, or * * (at your option) any later version. * * * ***************************************************************************/ #include "exportdialog.h" #include "kalziumutils.h" #include #include #include #include #include static const char HTML_HEADER[] = "" "\n" "\n" "\n" "\n" "\n"; static const char HTML_FOOTER[] = "\n" "\n"; static const char XML_HEADER[] = "\n"; ElementListEntry::ElementListEntry(Element * element) : QListWidgetItem() { m_atomicNum = element->dataAsVariant(ChemicalDataObject::atomicNumber).toInt(); m_name = element->dataAsString(ChemicalDataObject::name); m_element = element; setText(m_name); } ElementListEntry::~ElementListEntry() { } PropertyListEntry::PropertyListEntry(const QString & name, ChemicalDataObject::BlueObelisk type) : QListWidgetItem() { setText(name); m_type = type; } PropertyListEntry::~PropertyListEntry() { } ExportDialog::ExportDialog(QWidget * parent) : KDialog(parent),m_outputStream(nullptr) { qDebug() << "ExportDialog::ExportDialog"; setButtons(Help | User1 | Cancel); qDebug() << "ExportDialog: setButtons"; ui.setupUi(mainWidget()); qDebug() << "ExportDialog: ui.setupUi(mainWidget());"; setButtonGuiItem(User1, KGuiItem(i18n("OK"))); qDebug() << "ExportDialog: setButtonGuiItem(User1, KGuiItem(i18n(\"OK\")));"; ui.targetFile->setMode(KFile::File | KFile::Directory | KFile::LocalOnly); qDebug() << "ui.targetFile->setMode(KFile::File | KFile::Directory | KFile::LocalOnly);"; setCaption(i18n("Export Chemical Data")); qDebug() << "ui.targetFile->setMode(KFile::File | KFile::Directory | KFile::LocalOnly);"; populateElementList(); qDebug() << "ui.targetFile->setMode(KFile::File | KFile::Directory | KFile::LocalOnly);"; ui.formatList->addItem(".html (formatted html document)", "html"); ui.formatList->addItem(".xml (raw element data)", "xml"); ui.formatList->addItem(".csv (comma-separated data)", "csv"); qDebug() << "ui.formatList->addItem(...);"; connect(this, SIGNAL(user1Clicked()), this, SLOT(slotOkClicked())); qDebug() << "connect(this, SIGNAL(user1Clicked()), this, SLOT(slotOkClicked()));"; setHelp(QString(),"kalzium"); qDebug() << "setHelp(QString(),\"kalzium\");"; } ExportDialog::~ExportDialog() { delete m_outputStream; } void ExportDialog::populateElementList() { // Add descriptive headers QListWidgetItem *header1 = new QListWidgetItem(i18n("Elements")); QListWidgetItem *header2 = new QListWidgetItem(i18n("Properties")); header1->setFlags(Qt::ItemIsEnabled); header2->setFlags(Qt::ItemIsEnabled); QFont font; font.setBold(true); header1->setFont(font); header2->setFont(font); ui.elementListWidget->addItem(header1); ui.propertyListWidget->addItem(header2); // Add elements foreach (Element *element, KalziumDataObject::instance()->ElementList) { ElementListEntry *entry = new ElementListEntry(element); ui.elementListWidget->addItem(entry); } ui.propertyListWidget->addItem(new PropertyListEntry(i18n("Atomic Number"), ChemicalDataObject::atomicNumber)); ui.propertyListWidget->addItem(new PropertyListEntry(i18n("Symbol"), ChemicalDataObject::symbol)); //ui.propertyListWidget->addItem(new PropertyListEntry(i18n("Name"), ChemicalDataObject::name)); ui.propertyListWidget->addItem(new PropertyListEntry(i18n("Mass"), ChemicalDataObject::mass)); ui.propertyListWidget->addItem(new PropertyListEntry(i18n("Exact Mass"), ChemicalDataObject::exactMass)); ui.propertyListWidget->addItem(new PropertyListEntry(i18n("Ionization"), ChemicalDataObject::ionization)); ui.propertyListWidget->addItem(new PropertyListEntry(i18n("Electron Affinity"), ChemicalDataObject::electronAffinity)); ui.propertyListWidget->addItem(new PropertyListEntry(i18n("Electronegativity"), ChemicalDataObject::electronegativityPauling)); ui.propertyListWidget->addItem(new PropertyListEntry(i18n("Covalent Radius"), ChemicalDataObject::radiusCovalent)); ui.propertyListWidget->addItem(new PropertyListEntry(i18n("Van der Waals Radius"), ChemicalDataObject::radiusVDW)); ui.propertyListWidget->addItem(new PropertyListEntry(i18n("Melting Point"), ChemicalDataObject::meltingpoint)); ui.propertyListWidget->addItem(new PropertyListEntry(i18n("Boiling Point"), ChemicalDataObject::boilingpoint)); ui.propertyListWidget->addItem(new PropertyListEntry(i18n("Family"), ChemicalDataObject::family)); } void ExportDialog::slotOkClicked() { QString format = ui.formatList->itemData(ui.formatList->currentIndex(), Qt::UserRole).toString(); QString filename = ui.targetFile->url().toLocalFile(); if (!filename.endsWith(format)) { filename += '.' + format; } QFile outputFile(filename); if (outputFile.exists()) { if (KMessageBox::questionYesNo(this, i18n("File already exists. Do you want to overwrite it?")) == KMessageBox::No) { return; } } if (!outputFile.open(QIODevice::WriteOnly)) { KMessageBox::error(this, i18n("Could not open file for writing.")); return; } delete m_outputStream; m_outputStream = new QTextStream(&outputFile); if (format == "html") { exportToHtml(); } else if (format == "xml") { exportToXml(); } else { exportToCsv(); } // close the dialog done(0); } void ExportDialog::exportToHtml() { *m_outputStream << HTML_HEADER << "\n"; foreach (QListWidgetItem *element, ui.elementListWidget->selectedItems()) { + ElementListEntry* element_entry = static_cast(element); *m_outputStream << "\n\n\n"; foreach (QListWidgetItem *property, ui.propertyListWidget->selectedItems()) { + PropertyListEntry* property_entry = static_cast(property); *m_outputStream << "\n\n\n\n"; } } *m_outputStream << "

" - << ((ElementListEntry*)element)->m_element->dataAsString(ChemicalDataObject::name) + << element_entry->m_element->dataAsString(ChemicalDataObject::name) << "
" - << ((PropertyListEntry*) property)->text() + << property_entry->text() << "	" << KalziumUtils::prettyUnit( - ((ElementListEntry) element)->m_element, - ((PropertyListEntry) property)->m_type) + element_entry->m_element, + property_entry->m_type) << "

\n" << HTML_FOOTER; } void ExportDialog::exportToXml() { *m_outputStream << XML_HEADER << "\n"; foreach (QListWidgetItem *element, ui.elementListWidget->selectedItems()) { + ElementListEntry* element_entry = static_cast(element); *m_outputStream << " m_element->dataAsString(ChemicalDataObject::name) + << element_entry->m_element->dataAsString(ChemicalDataObject::name) << "\">\n"; foreach (QListWidgetItem *property, ui.propertyListWidget->selectedItems()) { + PropertyListEntry* property_entry = static_cast(property); *m_outputStream << " text() + << property_entry->text() << "\">" << KalziumUtils::prettyUnit( - ((ElementListEntry*) element)->m_element, - ((PropertyListEntry*) property)->m_type) + element_entry->m_element, + property_entry->m_type) << "\n"; } *m_outputStream << " \n"; } *m_outputStream << "\n"; } void ExportDialog::exportToCsv() { *m_outputStream << "Name"; foreach (QListWidgetItem *property, ui.propertyListWidget->selectedItems()) { *m_outputStream << ", \"" - << ((PropertyListEntry*) property)->text() + << static_cast(property)->text() << "\""; } *m_outputStream << "\n"; foreach (QListWidgetItem *element, ui.elementListWidget->selectedItems()) { + ElementListEntry* element_entry = static_cast(element); *m_outputStream << "\"" - << ((ElementListEntry*) element)->m_element->dataAsString(ChemicalDataObject::name) + << element_entry->m_element->dataAsString(ChemicalDataObject::name) << "\""; foreach (QListWidgetItem *property, ui.propertyListWidget->selectedItems()) { *m_outputStream << ", \"" << KalziumUtils::prettyUnit( - ((ElementListEntry*) element)->m_element, - ((PropertyListEntry*) property)->m_type) + element_entry->m_element, + static_cast(property)->m_type) << "\""; } *m_outputStream << "\n"; } }