File indexing completed on 2024-05-12 15:27:48

 /***************************************************************************
     File                 : ImportDialog.cc
     Project              : LabPlot
     Description          : import file data dialog
     --------------------------------------------------------------------
     Copyright            : (C) 2008-2018 Alexander Semke (alexander.semke@web.de)
     Copyright            : (C) 2008-2015 by Stefan Gerlach (stefan.gerlach@uni.kn)
 
  ***************************************************************************/
 
 /***************************************************************************
  *                                                                         *
  *  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.                                    *
  *                                                                         *
  *  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 General Public License for more details.                           *
  *                                                                         *
  *   You should have received a copy of the GNU General Public License     *
  *   along with this program; if not, write to the Free Software           *
  *   Foundation, Inc., 51 Franklin Street, Fifth Floor,                    *
  *   Boston, MA  02110-1301  USA                                           *
  *                                                                         *
  ***************************************************************************/
 
 #include "ImportDialog.h"
 #include "backend/core/AspectTreeModel.h"
 #include "backend/core/Project.h"
 #include "backend/spreadsheet/Spreadsheet.h"
 #include "backend/matrix/Matrix.h"
 #include "backend/core/Workbook.h"
 #include "backend/lib/macros.h"
 #include "commonfrontend/widgets/TreeViewComboBox.h"
 #include "kdefrontend/MainWin.h"
 
 #include <QDir>
 #include <QGroupBox>
 #include <QInputDialog>
 #include <QLabel>
 #include <QMenu>
 #include <QProgressBar>
 #include <QStatusBar>
 #include <QToolButton>
 #include <QVBoxLayout>
 
 #include <KConfigGroup>
 #include <KLocalizedString>
 #include <KMessageBox>
 #include <KSharedConfig>
 
 /*!
     \class ImportDialog
     \brief Base class for other import dialogs. Provides the "Import to" section of those dialogs.
 
     \ingroup kdefrontend
  */
 ImportDialog::ImportDialog(MainWin* parent) : QDialog(parent),
     vLayout(new QVBoxLayout(this)),
     m_mainWin(parent),
     m_aspectTreeModel(new AspectTreeModel(parent->project())) {
 
     //menu for new data container
     m_newDataContainerMenu = new QMenu(this);
     m_newDataContainerMenu->addAction( QIcon::fromTheme("labplot-workbook-new"), i18n("New Workbook") );
     m_newDataContainerMenu->addAction( QIcon::fromTheme("labplot-spreadsheet-new"), i18n("New Spreadsheet") );
     m_newDataContainerMenu->addAction( QIcon::fromTheme("labplot-matrix-new"), i18n("New Matrix") );
     connect(m_newDataContainerMenu, &QMenu::triggered, this, &ImportDialog::newDataContainer);
 }
 
 ImportDialog::~ImportDialog() {
     if (m_aspectTreeModel)
         delete m_aspectTreeModel;
 
     //save the last used import position for file imports, no need to do this for live data source (cbPosition=0)
     if (cbPosition) {
         KConfigGroup conf(KSharedConfig::openConfig(), "ImportDialog");
         conf.writeEntry("Position", cbPosition->currentIndex());
     }
 }
 
 /*!
     creates widgets for the frame "Import-To" and sets the current model in the "Add to"-combobox.
  */
 void ImportDialog::setModel() {
     //Frame for the "Import To"-Stuff
     frameAddTo = new QGroupBox(this);
     frameAddTo->setTitle(i18n("Import to"));
 
     auto* label = new QLabel(i18n("Data container"));
     label->setToolTip(i18n("Data container where the data has to be imported into"));
 
     auto* grid = new QGridLayout(frameAddTo);
     grid->addWidget(label, 0, 0);
 
     cbAddTo = new TreeViewComboBox();
     cbAddTo->setToolTip(i18n("Data container where the data has to be imported into"));
     cbAddTo->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Fixed);
     grid->addWidget(cbAddTo, 0, 1);
 
     QList<AspectType> list{AspectType::Folder, AspectType::Spreadsheet,
                     AspectType::Matrix, AspectType::Workbook};
     cbAddTo->setTopLevelClasses(list);
 
     list.removeFirst(); // do not allow selection of Folders
     m_aspectTreeModel->setSelectableAspects(list);
 
     cbAddTo->setModel(m_aspectTreeModel);
 
     tbNewDataContainer = new QToolButton(frameAddTo);
     tbNewDataContainer->setText(i18n("New"));
     tbNewDataContainer->setIcon(QIcon::fromTheme("list-add"));
     tbNewDataContainer->setToolButtonStyle(Qt::ToolButtonTextBesideIcon);
     tbNewDataContainer->setToolTip(i18n("Add new data container to the project"));
     grid->addWidget( tbNewDataContainer, 0, 2);
 
     lPosition = new QLabel(i18n("Position"), frameAddTo);
     lPosition->setEnabled(false);
     grid->addWidget(lPosition, 1, 0);
 
     cbPosition = new QComboBox(frameAddTo);
     cbPosition->setEnabled(false);
     cbPosition->addItem(i18n("Append"));
     cbPosition->addItem(i18n("Prepend"));
     cbPosition->addItem(i18n("Replace"));
     KConfigGroup conf(KSharedConfig::openConfig(), "ImportDialog");
     cbPosition->setCurrentIndex(conf.readEntry("Position", 0));
 
     cbPosition->setSizePolicy(QSizePolicy::Preferred, QSizePolicy::Fixed);
     grid->addWidget(cbPosition, 1, 1);
 
     //add the "Import to"-frame to the layout after the first main widget
     vLayout->insertWidget(1, frameAddTo);
 
     connect(tbNewDataContainer, &QToolButton::clicked, this, &ImportDialog::newDataContainerMenu);
     connect(cbAddTo, &TreeViewComboBox::currentModelIndexChanged, this, &ImportDialog::checkOkButton);
 }
 
 void ImportDialog::setCurrentIndex(const QModelIndex& index) {
     DEBUG("ImportFileDialog::setCurrentIndex()");
     QDEBUG(" index =" << index);
     cbAddTo->setCurrentModelIndex(index);
     QDEBUG("cbAddTo->currentModelIndex() =" << cbAddTo->currentModelIndex());
     checkOkButton();
 }
 
 void ImportDialog::newDataContainer(QAction* action) {
     DEBUG("ImportDialog::newDataContainer()");
     QString name = selectedObject();
     QString type = action->iconText().split(' ')[1];
     if (name.isEmpty())
         name = action->iconText();
 
     bool ok;
     // child widgets can't have own icons
     auto* dlg = new QInputDialog(this);
     name = dlg->getText(this, i18n("Add %1", action->iconText()), i18n("%1 name:", type), QLineEdit::Normal, name, &ok);
     if (ok) {
         AbstractAspect* aspect;
         int actionIndex = m_newDataContainerMenu->actions().indexOf(action);
         if (actionIndex == 0)
             aspect = new Workbook(name);
         else if (actionIndex == 1)
             aspect = new Spreadsheet(name);
         else
             aspect = new Matrix(name);
 
         m_mainWin->addAspectToProject(aspect);
         QDEBUG("cbAddTo->setCurrentModelIndex() to " << m_mainWin->model()->modelIndexOfAspect(aspect));
         cbAddTo->setCurrentModelIndex(m_mainWin->model()->modelIndexOfAspect(aspect));
         checkOkButton();
 
         //select "Replace" since this is the most common case when importing into a newly created container
         cbPosition->setCurrentIndex(2);
     }
 
     delete dlg;
 }
 
 void ImportDialog::newDataContainerMenu() {
     m_newDataContainerMenu->exec( tbNewDataContainer->mapToGlobal(tbNewDataContainer->rect().bottomLeft()));
 }