Changeset 2286 for branches/CEDMA-Exporter-715/tools/CedmaExporter/CedmaExporterForm.cs

Timestamp:

08/14/09 11:13:52 (15 years ago)

Author:

gkronber

Message:

Fixed a bug in the s-expression exporter. Improved exporter UI. #715 & #719

File:

• branches/CEDMA-Exporter-715/tools/CedmaExporter/CedmaExporterForm.cs (modified) (1 diff)

branches/CEDMA-Exporter-715/tools/CedmaExporter/CedmaExporterForm.cs

@@ -7 +7 @@
 using System.Text;
 using System.Windows.Forms;
+using System.IO;
+using SemWeb;
+using HeuristicLab.Core;
+using HeuristicLab.DataAnalysis;
 
 namespace CedmaExporter {
   public partial class CedmaExporterForm : Form {
-
+    private BackgroundWorker worker;
     public CedmaExporterForm() {
       InitializeComponent();
+    }
 
+    private void exportButton_Click(object sender, EventArgs e) {
       var dialog = new OpenFileDialog();
       DialogResult result;
-      do {
-        result = dialog.ShowDialog();
-        if (result == DialogResult.OK) {
-          string fileName = dialog.FileName;
-          CedmaExporter.Export(fileName, fileName.Replace(".db3", "") + ".txt");
+      result = dialog.ShowDialog();
+      if (result == DialogResult.OK) {
+        string fileName = dialog.FileName;
+        exportButton.Enabled = false;
+        cancelButton.Enabled = true;
+        worker = new BackgroundWorker();
+        worker.WorkerReportsProgress = true;
+        worker.WorkerSupportsCancellation = true;
+        worker.DoWork += CreateWorkerDelegate(worker, fileName);
+        worker.ProgressChanged += new ProgressChangedEventHandler(worker_ProgressChanged);
+        worker.RunWorkerCompleted += new RunWorkerCompletedEventHandler(worker_RunWorkerCompleted);
+        worker.RunWorkerAsync();
+      }
+    }
+
+    private DoWorkEventHandler CreateWorkerDelegate(BackgroundWorker worker, string fileName) {
+      return (sender, args) => {
+        string inputFileName = fileName;
+        string outputFileName = fileName.Replace(".db3", "") + ".txt";
+        string rdfConnectionString = "sqlite:rdf:Data Source=\"" + inputFileName + "\"";
+        using (StreamWriter writer = File.CreateText(outputFileName)) {
+          using (Store store = Store.Create(rdfConnectionString)) {
+            var dsEntities = store.SelectSubjects(Ontology.InstanceOf, Ontology.TypeDataSet);
+            HeuristicLab.CEDMA.Core.Problem prob = (HeuristicLab.CEDMA.Core.Problem)PersistenceManager.RestoreFromGZip(Convert.FromBase64String(((Literal)store.SelectObjects(dsEntities[0], Ontology.SerializedData)[0]).Value));
+            Dataset ds = prob.Dataset;
+            CedmaExporter.WriteColumnHeaders(writer);
+            List<string> inputVariables = CedmaExporter.WriteVariableImpactHeaders(store, writer);
+            writer.WriteLine();
+            var subjects = store.SelectSubjects(new Entity(Ontology.InstanceOf.Uri), new Entity(Ontology.TypeModel.Uri));
+            int i = 0;
+            var exporter = new ModelExporter(ds, Path.GetDirectoryName(outputFileName), false);
+            foreach (var model in subjects) {
+              CedmaExporter.WriteModel(model, ++i, store, writer, inputVariables, exporter);
+              worker.ReportProgress((i * 100) / subjects.Count());
+              if (worker.CancellationPending) return;
+            }
+          }
         }
-      } while (result != DialogResult.Cancel);
-      Application.Exit();
+      };
+    }
+
+    void worker_RunWorkerCompleted(object sender, RunWorkerCompletedEventArgs e) {
+      exportButton.Enabled = true;
+      cancelButton.Enabled = false;
+      progressBar.Value = 0;
+    }
+
+    void worker_ProgressChanged(object sender, ProgressChangedEventArgs e) {
+      progressBar.Value = e.ProgressPercentage;
+    }
+
+    private void cancelButton_Click(object sender, EventArgs e) {
+      worker.CancelAsync();
     }
   }