source: branches/OKB/HeuristicLab.Clients.OKB-3.3/Views/AlgorithmDataView.cs @ 4481

#region License Information
/* HeuristicLab
 * Copyright (C) 2002-2010 Heuristic and Evolutionary Algorithms Laboratory (HEAL)
 *
 * This file is part of HeuristicLab.
 *
 * HeuristicLab 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 3 of the License, or
 * (at your option) any later version.
 *
 * HeuristicLab 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 HeuristicLab. If not, see <http://www.gnu.org/licenses/>.
 */
#endregion

using System;
using System.IO;
using System.Linq;
using System.Windows.Forms;
using HeuristicLab.Common;
using HeuristicLab.Core.Views;
using HeuristicLab.MainForm;
using HeuristicLab.MainForm.WindowsForms;
using HeuristicLab.Optimization;
using HeuristicLab.Persistence.Default.Xml;
using HeuristicLab.PluginInfrastructure;

namespace HeuristicLab.Clients.OKB {
  [View("AlgorithmData View")]
  [Content(typeof(AlgorithmData), true)]
  public partial class AlgorithmDataView : AsynchronousContentView {
    public long AlgorithmId { get; set; }

    public new AlgorithmData Content {
      get { return (AlgorithmData)base.Content; }
      set { base.Content = value; }
    }

    public AlgorithmDataView() {
      InitializeComponent();
    }

    protected override void OnInitialized(System.EventArgs e) {
      base.OnInitialized(e);
      dataTypeComboBox.DataSource = Administrator.Instance.DataTypes.ToList();
      dataTypeComboBox.SelectedIndex = -1;
    }

    protected override void OnContentChanged() {
      base.OnContentChanged();
      if (Content == null) {
        dataTypeComboBox.SelectedIndex = -1;
        viewHost.Content = null;
        noViewAvailableLabel.Visible = false;
      } else {
        dataTypeComboBox.SelectedItem = Administrator.Instance.DataTypes.FirstOrDefault(d => d.Id == Content.DataTypeId);

        using (MemoryStream stream = new MemoryStream(Content.Data)) {
          try {
            viewHost.Content = XmlParser.Deserialize<IContent>(stream);
            noViewAvailableLabel.Visible = false;
          }
          catch (Exception) {
            viewHost.Content = null;
            noViewAvailableLabel.Visible = true;
          }
          stream.Close();
        }
      }
      fileTextBox.Text = "-";
    }

    protected override void SetEnabledStateOfControls() {
      base.SetEnabledStateOfControls();
      storeDataButton.Enabled = saveFileButton.Enabled = Content != null;
      dataTypeComboBox.Enabled = Content != null && viewHost.Content == null;
      fileTextBox.Enabled = Content != null;
      groupBox.Enabled = Content != null;
    }

    private void refreshDataButton_Click(object sender, EventArgs e) {
      AlgorithmData data = Administrator.Instance.GetAlgorithmData(AlgorithmId);
      Content = data;
    }
    private void storeDataButton_Click(object sender, EventArgs e) {
      if (viewHost.Content != null) {
        try {
          using (MemoryStream stream = new MemoryStream()) {
            IAlgorithm algorithm = viewHost.Content as IAlgorithm;
            algorithm.Prepare(true);
            algorithm.Problem = null;
            XmlGenerator.Serialize(algorithm, stream);
            stream.Close();
            Content.Data = stream.ToArray();
          }
        }
        catch (Exception ex) {
          ErrorHandling.ShowErrorDialog(this, ex);
        }
      }
      Administrator.Instance.UpdateAlgorithmData(Content);
    }
    private void newDataButton_Click(object sender, EventArgs e) {
      using (TypeSelectorDialog dialog = new TypeSelectorDialog()) {
        dialog.Caption = "Select Algorithm";
        dialog.TypeSelector.Caption = "Available Algorithms";
        dialog.TypeSelector.Configure(typeof(IAlgorithm), false, true);
        if (dialog.ShowDialog(this) == DialogResult.OK) {
          try {
            if (Content == null) Content = new AlgorithmData { AlgorithmId = AlgorithmId, Data = new byte[0] };
            viewHost.Content = (IContent)dialog.TypeSelector.CreateInstanceOfSelectedType();
            DataType dataType = Administrator.Instance.DataTypes.FirstOrDefault(d => d.Name == viewHost.Content.GetType().AssemblyQualifiedName);
            if (dataType == null) {
              dataType = new DataType { Name = viewHost.Content.GetType().AssemblyQualifiedName, SqlName = "Blob" };
              dataType.PlatformId = Administrator.Instance.Platforms.FirstOrDefault(p => p.Name == "HeuristicLab 3.3").Id;

              Administrator.Instance.Store(dataType);
              Administrator.Instance.DataTypes.Add(dataType);
              dataTypeComboBox.DataSource = Administrator.Instance.DataTypes.OrderBy(d => d.Name).ToList();
            }
            dataTypeComboBox.SelectedItem = dataType;
            dataTypeComboBox.Enabled = false;
            fileTextBox.Text = "-";
            noViewAvailableLabel.Visible = false;
          }
          catch (Exception ex) {
            ErrorHandling.ShowErrorDialog(this, ex);
          }
        }
      }
    }
    private void openFileButton_Click(object sender, EventArgs e) {
      if (openFileDialog.ShowDialog(this) == DialogResult.OK) {
        try {
          if (Content == null) Content = new AlgorithmData { AlgorithmId = AlgorithmId, Data = new byte[0] };
          IAlgorithm algorithm = null;
          try {
            algorithm = XmlParser.Deserialize<IAlgorithm>(openFileDialog.FileName);
          }
          catch (Exception) { }

          if (algorithm != null) {
            viewHost.Content = algorithm;
            noViewAvailableLabel.Visible = false;
            DataType dataType = Administrator.Instance.DataTypes.FirstOrDefault(d => d.Name == viewHost.Content.GetType().AssemblyQualifiedName);
            if (dataType == null) {
              dataType = new DataType { Name = viewHost.Content.GetType().AssemblyQualifiedName, SqlName = "Blob" };
              dataType.PlatformId = Administrator.Instance.Platforms.FirstOrDefault(p => p.Name == "HeuristicLab 3.3").Id;

              Administrator.Instance.Store(dataType);
              Administrator.Instance.DataTypes.Add(dataType);
              dataTypeComboBox.DataSource = Administrator.Instance.DataTypes.OrderBy(d => d.Name).ToList();
            }
            dataTypeComboBox.SelectedItem = dataType;
          } else {
            viewHost.Content = null;
            noViewAvailableLabel.Visible = true;
            using (FileStream stream = new FileStream(openFileDialog.FileName, FileMode.Open, FileAccess.Read)) {
              byte[] bytes = new byte[stream.Length];
              stream.Read(bytes, 0, bytes.Length);
              stream.Close();
              Content.Data = bytes;
            }
          }
          dataTypeComboBox.Enabled = viewHost.Content == null;
          fileTextBox.Text = openFileDialog.FileName;
        }
        catch (Exception ex) {
          ErrorHandling.ShowErrorDialog(this, ex);
        }
      }
    }
    private void saveFileButton_Click(object sender, EventArgs e) {
      if (saveFileDialog.ShowDialog(this) == DialogResult.OK) {
        try {
          if (viewHost.Content != null) {
            XmlGenerator.Serialize(viewHost.Content, saveFileDialog.FileName);
          } else {
            using (FileStream stream = new FileStream(saveFileDialog.FileName, FileMode.Create, FileAccess.Write)) {
              stream.Write(Content.Data, 0, Content.Data.Length);
              stream.Close();
            }
          }
          fileTextBox.Text = saveFileDialog.FileName;
        }
        catch (Exception ex) {
          ErrorHandling.ShowErrorDialog(this, ex);
        }
      }
    }
    private void dataTypeComboBox_SelectedIndexChanged(object sender, EventArgs e) {
      if (Content != null) Content.DataTypeId = ((DataType)dataTypeComboBox.SelectedItem).Id;
    }
  }
}