| | | 1 | | using System.Collections.Generic; |
| | | 2 | | using UnityEngine; |
| | | 3 | | |
| | | 4 | | namespace DCL |
| | | 5 | | { |
| | | 6 | | public class AssetLibrary_RefCounted<AssetType> : AssetLibrary<AssetType> |
| | | 7 | | where AssetType : Asset |
| | | 8 | | { |
| | | 9 | | public class RefCountedAsset |
| | | 10 | | { |
| | | 11 | | public AssetType asset; |
| | | 12 | | public int referenceCount = 0; |
| | | 13 | | } |
| | | 14 | | |
| | 29 | 15 | | public Dictionary<object, RefCountedAsset> masterAssets = new Dictionary<object, RefCountedAsset>(); |
| | 29 | 16 | | public Dictionary<AssetType, RefCountedAsset> assetToRefCountedAsset = new Dictionary<AssetType, RefCountedAsset |
| | | 17 | | |
| | | 18 | | public override bool Add(AssetType asset) |
| | | 19 | | { |
| | 86 | 20 | | if (asset == null || masterAssets.ContainsKey(asset.id)) |
| | 0 | 21 | | return true; |
| | | 22 | | |
| | 86 | 23 | | masterAssets.Add(asset.id, new RefCountedAsset() { asset = asset }); |
| | 86 | 24 | | assetToRefCountedAsset.Add(asset, masterAssets[asset.id]); |
| | | 25 | | |
| | 86 | 26 | | return true; |
| | | 27 | | } |
| | | 28 | | |
| | | 29 | | public override void Cleanup() |
| | | 30 | | { |
| | 1424 | 31 | | foreach (var kvp in masterAssets) |
| | | 32 | | { |
| | 11 | 33 | | kvp.Value.asset.Cleanup(); |
| | | 34 | | } |
| | | 35 | | |
| | 701 | 36 | | masterAssets.Clear(); |
| | 701 | 37 | | assetToRefCountedAsset.Clear(); |
| | 701 | 38 | | } |
| | | 39 | | |
| | 592 | 40 | | public override bool Contains(object id) { return masterAssets.ContainsKey(id); } |
| | | 41 | | |
| | | 42 | | public override bool Contains(AssetType asset) |
| | | 43 | | { |
| | 137 | 44 | | if (asset == null) |
| | | 45 | | { |
| | 4 | 46 | | return false; |
| | | 47 | | } |
| | | 48 | | |
| | 133 | 49 | | bool result = masterAssets.ContainsKey(asset.id); |
| | | 50 | | |
| | 133 | 51 | | return result; |
| | | 52 | | } |
| | | 53 | | |
| | | 54 | | public override AssetType Get(object id) |
| | | 55 | | { |
| | 108 | 56 | | if (!Contains(id)) |
| | 0 | 57 | | return null; |
| | | 58 | | |
| | 108 | 59 | | masterAssets[id].referenceCount++; |
| | 108 | 60 | | return masterAssets[id].asset; |
| | | 61 | | } |
| | | 62 | | |
| | | 63 | | public override void Release(AssetType asset) |
| | | 64 | | { |
| | 71 | 65 | | if (!Contains(asset.id)) |
| | 0 | 66 | | return; |
| | | 67 | | |
| | 71 | 68 | | var refCountedAsset = assetToRefCountedAsset[asset]; |
| | 71 | 69 | | refCountedAsset.referenceCount--; |
| | | 70 | | |
| | 71 | 71 | | if (refCountedAsset.referenceCount > 0) |
| | 0 | 72 | | return; |
| | | 73 | | |
| | 71 | 74 | | asset.Cleanup(); |
| | 71 | 75 | | masterAssets.Remove(asset.id); |
| | 71 | 76 | | assetToRefCountedAsset.Remove(asset); |
| | 71 | 77 | | } |
| | | 78 | | } |
| | | 79 | | } |