| 1 | using System;
 | 
|---|
| 2 | using System.Diagnostics.CodeAnalysis;
 | 
|---|
| 3 | using System.IO;
 | 
|---|
| 4 | using UnityEngine;
 | 
|---|
| 5 | using UnityEngine.Profiling;
 | 
|---|
| 6 | using Webserver.FileCache;
 | 
|---|
| 7 | using Object = UnityEngine.Object;
 | 
|---|
| 8 | 
 | 
|---|
| 9 | namespace MapRendering {
 | 
|---|
| 10 |         // Special "cache" for map tile folder as both map rendering and webserver access files in there.
 | 
|---|
| 11 |         // Only map rendering tiles are cached. Writing is done by WriteThrough.
 | 
|---|
| 12 |         public class MapTileCache : AbstractCache {
 | 
|---|
| 13 |                 private readonly byte[] transparentTile;
 | 
|---|
| 14 |                 private CurrentZoomFile[] cache;
 | 
|---|
| 15 | 
 | 
|---|
| 16 |                 public MapTileCache (int _tileSize) {
 | 
|---|
| 17 |                         Texture2D tex = new Texture2D (_tileSize, _tileSize);
 | 
|---|
| 18 |                         Color nullColor = new Color (0, 0, 0, 0);
 | 
|---|
| 19 |                         for (int x = 0; x < _tileSize; x++) {
 | 
|---|
| 20 |                                 for (int y = 0; y < _tileSize; y++) {
 | 
|---|
| 21 |                                         tex.SetPixel (x, y, nullColor);
 | 
|---|
| 22 |                                 }
 | 
|---|
| 23 |                         }
 | 
|---|
| 24 | 
 | 
|---|
| 25 |                         transparentTile = tex.EncodeToPNG ();
 | 
|---|
| 26 |                         Object.Destroy (tex);
 | 
|---|
| 27 |                 }
 | 
|---|
| 28 | 
 | 
|---|
| 29 |                 // SetZoomCount only called before processing happens in MapRenderer.ctor, no locking required
 | 
|---|
| 30 |                 [SuppressMessage ("ReSharper", "InconsistentlySynchronizedField")]
 | 
|---|
| 31 |                 public void SetZoomCount (int _count) {
 | 
|---|
| 32 |                         cache = new CurrentZoomFile[_count];
 | 
|---|
| 33 |                         for (int i = 0; i < cache.Length; i++) {
 | 
|---|
| 34 |                                 cache [i] = new CurrentZoomFile ();
 | 
|---|
| 35 |                         }
 | 
|---|
| 36 |                 }
 | 
|---|
| 37 | 
 | 
|---|
| 38 |                 public byte[] LoadTile (int _zoomlevel, string _filename) {
 | 
|---|
| 39 |                         try {
 | 
|---|
| 40 |                                 lock (cache) {
 | 
|---|
| 41 |                                         CurrentZoomFile cacheEntry = cache [_zoomlevel];
 | 
|---|
| 42 | 
 | 
|---|
| 43 |                                         if (cacheEntry.filename != null && cacheEntry.filename.Equals (_filename)) {
 | 
|---|
| 44 |                                                 return cacheEntry.pngData;
 | 
|---|
| 45 |                                         }
 | 
|---|
| 46 | 
 | 
|---|
| 47 |                                         cacheEntry.filename = _filename;
 | 
|---|
| 48 | 
 | 
|---|
| 49 |                                         if (!File.Exists (_filename)) {
 | 
|---|
| 50 |                                                 cacheEntry.pngData = null;
 | 
|---|
| 51 |                                                 return null;
 | 
|---|
| 52 |                                         }
 | 
|---|
| 53 | 
 | 
|---|
| 54 |                                         Profiler.BeginSample ("ReadPng");
 | 
|---|
| 55 |                                         cacheEntry.pngData = ReadAllBytes (_filename);
 | 
|---|
| 56 |                                         Profiler.EndSample ();
 | 
|---|
| 57 | 
 | 
|---|
| 58 |                                         return cacheEntry.pngData;
 | 
|---|
| 59 |                                 }
 | 
|---|
| 60 |                         } catch (Exception e) {
 | 
|---|
| 61 |                                 Log.Warning ($"Error in MapTileCache.LoadTile: {e}");
 | 
|---|
| 62 |                         }
 | 
|---|
| 63 | 
 | 
|---|
| 64 |                         return null;
 | 
|---|
| 65 |                 }
 | 
|---|
| 66 | 
 | 
|---|
| 67 |                 public void SaveTile (int _zoomlevel, byte[] _contentPng) {
 | 
|---|
| 68 |                         try {
 | 
|---|
| 69 |                                 lock (cache) {
 | 
|---|
| 70 |                                         CurrentZoomFile cacheEntry = cache [_zoomlevel];
 | 
|---|
| 71 | 
 | 
|---|
| 72 |                                         string file = cacheEntry.filename;
 | 
|---|
| 73 |                                         if (string.IsNullOrEmpty (file)) {
 | 
|---|
| 74 |                                                 return;
 | 
|---|
| 75 |                                         }
 | 
|---|
| 76 |                                         
 | 
|---|
| 77 |                                         cacheEntry.pngData = _contentPng;
 | 
|---|
| 78 | 
 | 
|---|
| 79 |                                         Profiler.BeginSample ("WritePng");
 | 
|---|
| 80 |                                         using (Stream stream = new FileStream (file, FileMode.Create, FileAccess.ReadWrite, FileShare.None,
 | 
|---|
| 81 |                                                 4096)) {
 | 
|---|
| 82 |                                                 stream.Write (_contentPng, 0, _contentPng.Length);
 | 
|---|
| 83 |                                         }
 | 
|---|
| 84 |                                         Profiler.EndSample ();
 | 
|---|
| 85 |                                 }
 | 
|---|
| 86 |                         } catch (Exception e) {
 | 
|---|
| 87 |                                 Log.Warning ($"Error in MapTileCache.SaveTile: {e}");
 | 
|---|
| 88 |                         }
 | 
|---|
| 89 |                 }
 | 
|---|
| 90 | 
 | 
|---|
| 91 |                 public void ResetTile (int _zoomlevel) {
 | 
|---|
| 92 |                         try {
 | 
|---|
| 93 |                                 lock (cache) {
 | 
|---|
| 94 |                                         cache [_zoomlevel].filename = null;
 | 
|---|
| 95 |                                         cache [_zoomlevel].pngData = null;
 | 
|---|
| 96 |                                 }
 | 
|---|
| 97 |                         } catch (Exception e) {
 | 
|---|
| 98 |                                 Log.Warning ($"Error in MapTileCache.ResetTile: {e}");
 | 
|---|
| 99 |                         }
 | 
|---|
| 100 |                 }
 | 
|---|
| 101 | 
 | 
|---|
| 102 |                 public override byte[] GetFileContent (string _filename) {
 | 
|---|
| 103 |                         try {
 | 
|---|
| 104 |                                 lock (cache) {
 | 
|---|
| 105 |                                         foreach (CurrentZoomFile czf in cache) {
 | 
|---|
| 106 |                                                 if (czf.filename != null && czf.filename.Equals (_filename)) {
 | 
|---|
| 107 |                                                         return czf.pngData;
 | 
|---|
| 108 |                                                 }
 | 
|---|
| 109 |                                         }
 | 
|---|
| 110 | 
 | 
|---|
| 111 |                                         return !File.Exists (_filename) ? transparentTile : ReadAllBytes (_filename);
 | 
|---|
| 112 |                                 }
 | 
|---|
| 113 |                         } catch (Exception e) {
 | 
|---|
| 114 |                                 Log.Warning ($"Error in MapTileCache.GetFileContent: {e}");
 | 
|---|
| 115 |                         }
 | 
|---|
| 116 | 
 | 
|---|
| 117 |                         return null;
 | 
|---|
| 118 |                 }
 | 
|---|
| 119 | 
 | 
|---|
| 120 |                 public override (int, int) Invalidate () {
 | 
|---|
| 121 |                         return (0, 0);
 | 
|---|
| 122 |                 }
 | 
|---|
| 123 | 
 | 
|---|
| 124 |                 private static byte[] ReadAllBytes (string _path) {
 | 
|---|
| 125 |                         using FileStream fileStream = new FileStream(_path, FileMode.Open, FileAccess.Read, FileShare.Read, 4096);
 | 
|---|
| 126 |                         
 | 
|---|
| 127 |                         int bytesRead = 0;
 | 
|---|
| 128 |                         int bytesLeft = (int) fileStream.Length;
 | 
|---|
| 129 |                         byte[] result = new byte[bytesLeft];
 | 
|---|
| 130 |                         while (bytesLeft > 0) {
 | 
|---|
| 131 |                                 int readThisTime = fileStream.Read (result, bytesRead, bytesLeft);
 | 
|---|
| 132 |                                 if (readThisTime == 0) {
 | 
|---|
| 133 |                                         throw new IOException ("Unexpected end of stream");
 | 
|---|
| 134 |                                 }
 | 
|---|
| 135 | 
 | 
|---|
| 136 |                                 bytesRead += readThisTime;
 | 
|---|
| 137 |                                 bytesLeft -= readThisTime;
 | 
|---|
| 138 |                         }
 | 
|---|
| 139 | 
 | 
|---|
| 140 |                         return result;
 | 
|---|
| 141 |                 }
 | 
|---|
| 142 | 
 | 
|---|
| 143 | 
 | 
|---|
| 144 |                 private class CurrentZoomFile {
 | 
|---|
| 145 |                         public string filename;
 | 
|---|
| 146 |                         public byte[] pngData;
 | 
|---|
| 147 |                 }
 | 
|---|
| 148 |         }
 | 
|---|
| 149 | }
 | 
|---|