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 | }
|
---|