Adding csv writer to help with data recording

Author: TORRENS_PHD_ACCOUNT

CSVWriter/CSVWriter.cs

@@ -0,0 +1,100 @@
+using System;
+using System.IO;
+using System.Text;
+using System.Collections;
+using System.Collections.Generic;
+using UnityEngine;
+using UnityEditor;
+
+[System.Serializable]
+public class CSVWriter
+{
+    public string fileName = null;
+    public string dirName = null;
+    public bool writeUnixTime = true;
+    public bool append_zero_to_filename = false;
+    public List<string> columns;
+
+    [SerializeField, ReadOnly] private string filePath;
+    private StreamWriter eventWriter;
+    private List<string> payload = new List<string>();
+
+    public bool Initialize() {
+        string dname = Application.persistentDataPath;
+        if (dirName != null && dirName.Length > 0) {
+            dname = $"Assets/{dirName}";
+            if (!AssetDatabase.IsValidFolder(dname)) dname = AssetDatabase.GUIDToAssetPath(AssetDatabase.CreateFolder("Assets", dirName));
+        }
+        string fname = (fileName != null && fileName.Length > 0) ? fileName : System.DateTime.Now.ToString("HH-mm-ss");
+        filePath = (append_zero_to_filename) ? Path.Combine(dname, fname+"_0.csv") : Path.Combine(dname, fname+".csv");
+
+        int counter = 1;
+        while(File.Exists(filePath)) {
+            filePath = Path.Combine(dname, fname+$"_{counter}.csv");
+            counter++;
+        }
+
+        eventWriter = new StreamWriter(new FileStream(filePath, FileMode.Create), Encoding.UTF8);
+        // Header Line, if any columns are added to `columns`
+        if (columns.Count > 0) {
+            if (writeUnixTime) columns.Add("unix_ms");
+            eventWriter.WriteLine(String.Join(',', columns));
+        }
+
+        return true;
+    }
+
+
+    public bool WriteLine() {
+        if (payload.Count == 0) return false;
+        WriteLine(String.Join(",",payload));
+        payload = new List<string>();
+        return true;
+    }
+    public bool WriteLine(bool add_unix) {
+        if (payload.Count == 0) return false;
+        WriteLine(String.Join(",",payload), add_unix);
+        payload = new List<string>();
+        return true;
+    }
+    public bool WriteLine(string newLine) {
+        eventWriter.WriteLine(newLine);
+        return true;
+    }
+    public bool WriteLine(string newLine, bool add_unix) {
+        if (add_unix) eventWriter.WriteLine($"{GetUnixTime()},{newLine}");
+        else eventWriter.WriteLine(newLine);
+        return true;
+    }
+
+    public void AddPayload(string to_add) {
+        payload.Add(to_add);
+    }
+    public void AddPayload(int to_add) {
+        payload.Add(to_add.ToString());
+    }
+    public void AddPayload(float to_add) {
+        payload.Add(to_add.ToString());
+    }
+    public void AddPayload(long to_add) {
+        payload.Add($"{to_add}");
+    }
+    public void AddPayload(Vector3 to_add) {
+        payload.Add(to_add.x.ToString());
+        payload.Add(to_add.y.ToString());
+        payload.Add(to_add.z.ToString());
+    }
+    public void AddPayload(Quaternion to_add) {
+        AddPayload(to_add.eulerAngles);
+    }
+
+    public static long GetUnixTime() {
+        DateTime currentTime = DateTime.UtcNow;
+        return ((DateTimeOffset)currentTime).ToUnixTimeMilliseconds();
+    }
+
+    public void Disable() {
+        eventWriter.Flush();
+        eventWriter.Close();
+    }
+}