SaveManager.cs 9.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274
  1. using UnityEngine;
  2. using System;
  3. using System.Linq;
  4. using System.Collections.Generic;
  5. using System.Text;
  6. using System.IO;
  7. using Util;
  8. namespace CM3D2.MultipleMaids.Plugin
  9. {
  10. public partial class MultipleMaids
  11. {
  12. private static readonly byte[] pngEnd = Encoding.ASCII.GetBytes("IEND");
  13. private static string sceneData;
  14. private const string baseDirectoryName = "< Base Directory >";
  15. private string saveScenePath = Path.Combine(Path.GetFullPath(".\\"), "Mod\\MultipleMaidsSave");
  16. private List<SavePng> saveScenes = new List<SavePng>(50);
  17. private GUIStyle selectedButtonStyle;
  18. private string[] directoryList;
  19. private Texture2D frame;
  20. private Rect saveManagerRect;
  21. private Rect saveModalRect;
  22. private Rect resizeManagerRect;
  23. private Vector2 saveManagerScrollPos = Vector2.zero;
  24. private Vector2 dirListScrollPos = Vector2.zero;
  25. private bool saveManagerInitialize = false;
  26. private bool loadSaveFlag = false;
  27. private bool overwriteFlag = false;
  28. private bool createSaveFlag = false;
  29. private bool manageSaveFlag = false;
  30. private bool createDirectory = false;
  31. private bool resizeManager = false;
  32. private int selectedSave = 0;
  33. private int selectedDirectory = 0;
  34. private string currentDirectory = "";
  35. private string textFieldValue = "";
  36. public void InitializeSaveManager()
  37. {
  38. frame = MakeTex(2, 2, Color.white);
  39. if (!Directory.Exists(saveScenePath))
  40. {
  41. Directory.CreateDirectory(saveScenePath);
  42. }
  43. selectedButtonStyle = new GUIStyle("button");
  44. selectedButtonStyle.normal.background = MakeTex(1, 1, new Color(0.5f, 0.5f, 0.5f, 0.4f));
  45. selectedButtonStyle.normal.textColor = Color.white;
  46. GetSaveDirectories();
  47. InitializeSaveList();
  48. saveManagerInitialize = true;
  49. }
  50. private void RefreshSaveManager()
  51. {
  52. SwitchDirectory(currentDirectory);
  53. GetSaveDirectories();
  54. InitializeSaveList();
  55. }
  56. private void InitializeSaveList()
  57. {
  58. saveScenes.Clear();
  59. string workingPath = Path.Combine(saveScenePath, currentDirectory);
  60. DirectoryInfo info = new DirectoryInfo(workingPath);
  61. foreach (var save in info.GetFiles("*.png"))
  62. {
  63. Texture2D screenshot = new Texture2D(1, 1, TextureFormat.ARGB32, false);
  64. screenshot.LoadImage(File.ReadAllBytes(save.FullName));
  65. saveScenes.Add(new SavePng(save, screenshot));
  66. }
  67. selectedSave = saveScenes.Count == 0 ? 0 : saveScenes.Count - 1;
  68. saveScenes.Sort((a, b) => a.info.LastWriteTime.CompareTo(b.info.LastWriteTime));
  69. saveManagerScrollPos.y = 0;
  70. }
  71. private void GetSaveDirectories()
  72. {
  73. DirectoryInfo[] directoryInfo = new DirectoryInfo(saveScenePath).GetDirectories();
  74. directoryList = new string[directoryInfo.Length + 1];
  75. directoryList[0] = baseDirectoryName;
  76. for (int i = 0; i < directoryInfo.Length; i++)
  77. {
  78. directoryList[i + 1] = directoryInfo[i].Name;
  79. }
  80. }
  81. private void SwitchDirectory(string target)
  82. {
  83. if (target.Equals(baseDirectoryName)) target = "";
  84. string targetDirectory = Path.Combine(saveScenePath, target);
  85. if (!Directory.Exists(targetDirectory))
  86. {
  87. currentDirectory = "";
  88. selectedDirectory = 0;
  89. GetSaveDirectories();
  90. }
  91. else
  92. {
  93. selectedDirectory = Array.FindIndex(directoryList, d => d.Equals(target, StringComparison.OrdinalIgnoreCase));
  94. selectedDirectory = selectedDirectory == -1 ? 0 : selectedDirectory;
  95. if (target == currentDirectory) return;
  96. currentDirectory = target;
  97. }
  98. InitializeSaveList();
  99. }
  100. private void LoadSave()
  101. {
  102. string filePath = saveScenes[selectedSave].info.FullName;
  103. if (!File.Exists(filePath))
  104. {
  105. InitializeSaveList();
  106. sceneData = null;
  107. return;
  108. }
  109. using (FileStream fileStream = File.OpenRead(filePath))
  110. {
  111. byte[] buffer = new byte[pngEnd.Length];
  112. long position = 0;
  113. while (true)
  114. {
  115. int bytesRead = fileStream.Read(buffer, 0, buffer.Length);
  116. if (bytesRead != pngEnd.Length)
  117. {
  118. sceneData = null;
  119. return;
  120. }
  121. if (BytesEqual(buffer, pngEnd))
  122. {
  123. fileStream.Position += 4;
  124. break;
  125. }
  126. fileStream.Position = ++position;
  127. }
  128. using (MemoryStream sceneStream = LZMA.Decompress(fileStream))
  129. {
  130. sceneData = Encoding.Unicode.GetString(sceneStream.ToArray());
  131. }
  132. }
  133. SavePng temp = saveScenes[selectedSave];
  134. saveScenes.RemoveAt(selectedSave);
  135. saveScenes.Add(temp);
  136. saveManagerScrollPos.y = 0;
  137. }
  138. private void Overwrite()
  139. {
  140. string filePath = saveScenes[selectedSave].info.FullName;
  141. if (!File.Exists(filePath))
  142. {
  143. RefreshSaveManager();
  144. }
  145. else
  146. {
  147. File.Delete(filePath);
  148. saveScenes.RemoveAt(selectedSave);
  149. saveManagerScrollPos.y = 0;
  150. }
  151. }
  152. private void MakeNewDirectory(string directoryName)
  153. {
  154. directoryName = string.Join("", directoryName.Split(Path.GetInvalidFileNameChars()));
  155. string newDirectory = Path.Combine(saveScenePath, directoryName);
  156. if (!Directory.Exists(newDirectory))
  157. {
  158. Directory.CreateDirectory(newDirectory);
  159. }
  160. GetSaveDirectories();
  161. SwitchDirectory(directoryName);
  162. }
  163. private void SaveScene()
  164. {
  165. string saveDirectory = Path.Combine(saveScenePath, currentDirectory);
  166. if (!Directory.Exists(saveDirectory))
  167. {
  168. thum_byte_to_base64_ = "";
  169. thum_file_path_ = "";
  170. RefreshSaveManager();
  171. return;
  172. }
  173. string sceneString = SerializeScene(true);
  174. #region MM GUI stuff
  175. Texture2D screenshot = new Texture2D(1, 1, TextureFormat.ARGB32, false);
  176. screenshot.LoadImage(File.ReadAllBytes(thum_file_path_));
  177. float num2 = screenshot.width / (float)screenshot.height;
  178. Vector2 vector2 = new Vector2(480f, 270f);
  179. int newWidth = screenshot.width;
  180. int newHeight = screenshot.height;
  181. if (vector2.x < (double)screenshot.width && vector2.y < (double)screenshot.height)
  182. {
  183. newWidth = (int)vector2.x;
  184. newHeight = Mathf.RoundToInt(newWidth / num2);
  185. if (vector2.y < (double)newHeight)
  186. {
  187. newHeight = (int)vector2.y;
  188. newWidth = Mathf.RoundToInt(newHeight * num2);
  189. }
  190. }
  191. else if (vector2.x < (double)screenshot.width)
  192. {
  193. newWidth = (int)vector2.x;
  194. newHeight = Mathf.RoundToInt(newWidth / num2);
  195. }
  196. else if (vector2.y < (double)screenshot.height)
  197. {
  198. newHeight = (int)vector2.y;
  199. newWidth = Mathf.RoundToInt(newHeight * num2);
  200. }
  201. TextureScale.Bilinear(screenshot, newWidth, newHeight);
  202. #endregion
  203. thum_byte_to_base64_ = "";
  204. thum_file_path_ = "";
  205. string filePath = Path.Combine(saveDirectory, $"mmsave{DateTime.Now:yyyyMMddHHmmss}.png");
  206. using (FileStream fileStream = File.Create(filePath))
  207. using (MemoryStream sceneStream = new MemoryStream(Encoding.Unicode.GetBytes(sceneString)))
  208. {
  209. byte[] screenshotBuffer = screenshot.EncodeToPNG();
  210. byte[] sceneBuffer = LZMA.Compress(sceneStream);
  211. fileStream.Write(screenshotBuffer, 0, screenshotBuffer.Length);
  212. fileStream.Write(sceneBuffer, 0, sceneBuffer.Length);
  213. }
  214. saveScenes.Add(new SavePng(new FileInfo(filePath), screenshot));
  215. selectedSave = saveScenes.Count - 1;
  216. }
  217. private static bool BytesEqual(byte[] a, byte[] b)
  218. {
  219. if (a.Length != b.Length) return false;
  220. for (int i = 0; i < a.Length; i++)
  221. {
  222. if (a[i] != b[i]) return false;
  223. }
  224. return true;
  225. }
  226. class SavePng
  227. {
  228. public FileInfo info { get; }
  229. public Texture2D screenshot { get; }
  230. public SavePng(FileInfo info, Texture2D screenshot)
  231. {
  232. this.info = info;
  233. this.screenshot = screenshot;
  234. }
  235. }
  236. }
  237. }