source: binary-improvements/7dtd-server-fixes/src/FileCache/MapTileCache.cs@ 336

Last change on this file since 336 was 331, checked in by alloc, 6 years ago

Major allocation and execution time improvements on the map rendering code

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