add export single to raw/to file

OnexTale · Jul 20, 2017 · 7135ff6 · 7135ff6
1 parent c4ff6d0
commit 7135ff6
Show file tree

Hide file tree

Showing 4 changed files with 60 additions and 2 deletions.
diff --git a/Source/Ui/OnexTreeImage.cpp b/Source/Ui/OnexTreeImage.cpp
@@ -31,3 +31,19 @@ void OnexTreeImage::onExportAll()
  QMessageBox msgBox(QMessageBox::Information, tr("End of operation"), text);
  msgBox.exec();
 }
+
+void OnexTreeImage::onExportSingle()
+{
+ QString fileName = getSaveDirectory("PNG Image (*png)");
+
+ if (fileName.isEmpty())
+ return;
+
+ if (!fileName.endsWith(".png"))
+ fileName += ".png";
+
+ if (this->getImage().save(fileName, "PNG", 100))
+ QMessageBox::information(NULL, "Yeah", "Image exported");
+ else
+ QMessageBox::critical(NULL, "Woops", "Couldn't export that image");
+}
diff --git a/Source/Ui/OnexTreeImage.h b/Source/Ui/OnexTreeImage.h
@@ -22,6 +22,7 @@ class OnexTreeImage : public OnexTreeZlibItem
 
 private slots:
  virtual void onExportAll();
+ virtual void onExportSingle();
 
 };
 

diff --git a/Source/Ui/OnexTreeItem.cpp b/Source/Ui/OnexTreeItem.cpp
@@ -16,6 +16,11 @@ QString OnexTreeItem::getSelectedDirectory()
  return dir + "/";
 }
 
+QString OnexTreeItem::getSaveDirectory(QString filter)
+{
+ return QFileDialog::getSaveFileName(0, tr("Save as..."), "", filter);
+}
+
 QByteArray OnexTreeItem::getContent()
 {
  return content;
@@ -45,6 +50,16 @@ QMenu *OnexTreeItem::getContextMenu()
  contextMenu->addAction(exportAllAction);
  QObject::connect(exportAllAction, SIGNAL(triggered(bool)), this, SLOT(onExportAll()));
  }
+ else
+ {
+ QAction* exportSingleAction = new QAction(QObject::tr("Export"), contextMenu);
+ contextMenu->addAction(exportSingleAction);
+ QObject::connect(exportSingleAction, SIGNAL(triggered(bool)), this, SLOT(onExportSingle()));
+
+ QAction* exportSingleToRawAction = new QAction(QObject::tr("Export to raw"), contextMenu);
+ contextMenu->addAction(exportSingleToRawAction);
+ QObject::connect(exportSingleToRawAction, SIGNAL(triggered(bool)), this, SLOT(onExporSingleRaw()));
+ }
 
  return contextMenu;
 }
@@ -56,6 +71,28 @@ OnexTreeItem::~OnexTreeItem()
 
 void OnexTreeItem::onExportAll()
 {
- QMessageBox msgBox(QMessageBox::Warning, QObject::tr("Not yet"), QObject::tr("This isn't implemented yet"));
- msgBox.exec();
+ QMessageBox::warning(NULL, tr("Not yet"), tr("This isn't implemented yet"));
+}
+
+void OnexTreeItem::onExportSingle()
+{
+ QMessageBox::warning(NULL, tr("Not yet"), tr("This isn't implemented yet"));
+}
+
+void OnexTreeItem::onExporSingleRaw()
+{
+ QString fileName = getSaveDirectory("Raw data (*.rawdata)");
+ if (fileName.isEmpty())
+ return;
+
+ if (!fileName.endsWith(tr(".rawdata")))
+ fileName += ".rawdata";
+
+ QFile file(fileName);
+ file.open(QIODevice::WriteOnly);
+ if (file.write(this->getContent()) == -1)
+ QMessageBox::critical(NULL, tr("Woops"), tr("Couldn't save that file"));
+ else
+ QMessageBox::information(NULL, tr("Yeah"), tr("File exported"));
+ file.close();
 }
diff --git a/Source/Ui/OnexTreeItem.h b/Source/Ui/OnexTreeItem.h
@@ -15,6 +15,8 @@ class OnexTreeItem : public QObject, public QTreeWidgetItem
  QString name;
 
  QString getSelectedDirectory();
+ QString getSaveDirectory(QString filter);
+ QMessageBox getMsgBox(QString title, QString message, QMessageBox::Icon icon);
 public:
  OnexTreeItem(QString name, QByteArray content = QByteArray());
  QByteArray getContent();
@@ -28,6 +30,8 @@ class OnexTreeItem : public QObject, public QTreeWidgetItem
 
 private slots:
  virtual void onExportAll();
+ virtual void onExportSingle();
+ virtual void onExporSingleRaw();
 };
 
 #endif // ONEXTREEITEM_H