/*************************************************************************** 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 static const char HTML_HEADER[] = "" "\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(0) { setButtons( Help | User1 | Cancel ); ui.setupUi( mainWidget() ); setButtonGuiItem( User1, KGuiItem( i18n( "Ok" ) ) ); ui.targetFile->setMode( KFile::File | KFile::Directory | KFile::LocalOnly ); setCaption( i18n( "Export Chemical Data" ) ); populateElementList(); ui.formatList->addItem( ".html (formatted html document)", "html" ); ui.formatList->addItem( ".xml (raw element data)", "xml" ); ui.formatList->addItem( ".csv (comma-separated data)", "csv" ); connect( this, SIGNAL( user1Clicked() ), this, SLOT( slotOkClicked() ) ); connect( this, SIGNAL( helpClicked() ), this, SLOT( slotHelpClicked() ) ); 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::exactMass ) ); 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() ) { *m_outputStream << "\n\n\n"; foreach( QListWidgetItem * property, ui.propertyListWidget->selectedItems() ) { *m_outputStream << "\n\n\n\n"; } } *m_outputStream << "

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

\n" << HTML_FOOTER; } void ExportDialog::exportToXml() { *m_outputStream << XML_HEADER << "\n"; foreach( QListWidgetItem * element, ui.elementListWidget->selectedItems() ) { *m_outputStream << " m_element->dataAsString( ChemicalDataObject::name ) << "\">\n"; foreach( QListWidgetItem * property, ui.propertyListWidget->selectedItems() ) { *m_outputStream << " text() << "\">" << KalziumUtils::prettyUnit( ( ( ElementListEntry* )element )->m_element, ( ( PropertyListEntry* )property )->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() << "\""; } *m_outputStream << "\n"; foreach( QListWidgetItem * element, ui.elementListWidget->selectedItems() ) { *m_outputStream << "\"" << ( (ElementListEntry* )element )->m_element->dataAsString( ChemicalDataObject::name ) << "\""; foreach( QListWidgetItem * property, ui.propertyListWidget->selectedItems() ) { *m_outputStream << ", \"" << KalziumUtils::prettyUnit( ( ( ElementListEntry* )element )->m_element, ( ( PropertyListEntry* )property )->m_type ) << "\""; } *m_outputStream << "\n"; } } void ExportDialog::slotHelpClicked() { }