SnapshotManager.cs

Go to the documentation of this file.

using UnityEngine;
using System.Collections;
using Molecule.Model;
using System;
using System.IO;
using System.Linq;
using UI;

public class SnapshotManager {
    public static string generateSnapshot()
    {
        int size = MoleculeModel.atomsIMDSimulationLocationlist.Count;
        Vector3[] a = new Vector3[size];
        MoleculeModel.atomsIMDSimulationLocationlist.CopyTo(a);
        int c = a.Count();
        
        string snapshot = "";
        
        // snapshot += "MODEL\n";
        
        int j = 0;
        for (int i = 0; i < c; i++)
        {
            j = i + 1;
            snapshot += "ATOM  "
                + string.Format("{0,5}", j)
                    + string.Format("{0,4:###}", MoleculeModel.atomsNamelist[i]) + " "
                    + string.Format("{0,4:###.0}", MoleculeModel.atomsResnamelist[i]) + " " 
                    + string.Format("{0,4:###}", MoleculeModel.residueIds[i])
                    + string.Format("{0,11:###.000}", a[i].x) + " " 
                    + string.Format("{0,8:###.000}", a[i].y) + " " 
                    + string.Format("{0,8:###.000}", a[i].z) + "\n";
        }
        // snapshot += "ENDMDL";
        
        return snapshot;
    }

    public static string getPath()
    {
        return Environment.GetFolderPath(Environment.SpecialFolder.Personal) + "/.unitymol/snapshots/";
    }
    
    public static string saveSnapshot() 
    {
        string molName = GUIDisplay.id;
        string filename = Environment.UserName + "-" + molName + "-" + DateTime.Now.ToString ("yyyy-MM-dd-H-mm-ss") +".pdb";
        string path = getPath();
        System.IO.Directory.CreateDirectory(path);
        StreamWriter sw = new StreamWriter(path + filename);
        sw.Write (generateSnapshot());
        sw.Close();
        
        return filename;
    }
}