Code/Tools/Statoscope/Statoscope/FrameRecordPathCollection.cs

   1 // Copyright 2001-2019 Crytek GmbH / Crytek Group. All rights reserved.
   2
   3 using System;
   4 using System.Collections.Generic;
   5 using System.Linq;
   6 using System.Text;
   7
   8 namespace Statoscope
   9 {
  10         public interface IReadOnlyFRPC
  11         {
  12                 int PathCount { get; }
  13                 string[] Paths { get; }
  14
  15                 bool ContainsPath(string path);
  16                 int TryGetIndexFromPath(string path);
  17     int GetIndexFromPath(string path);
  18
  19                 string GetPathFromIndex(int idx);
  20         }
  21
  22   public class FrameRecordPathCollection : IReadOnlyFRPC
  23   {
  24     private System.Threading.ReaderWriterLockSlim m_lock = new System.Threading.ReaderWriterLockSlim();
  25                 private Dictionary<string, int> m_pathsToIdx = new Dictionary<string, int>();
  26                 private List<string> m_idxToPaths = new List<string>();
  27
  28     public int PathCount
  29     {
  30       get { return m_idxToPaths.Count; }
  31     }
  32
  33                 public string[] Paths
  34                 {
  35                         get
  36                         {
  37                                 m_lock.EnterReadLock();
  38                                 string[] paths = m_idxToPaths.ToArray();
  39                                 m_lock.ExitReadLock();
  40                                 return paths;
  41                         }
  42                 }
  43
  44                 public int AddPath(string path)
  45                 {
  46       m_lock.EnterWriteLock();
  47
  48                         int idx;
  49       if (m_pathsToIdx.TryGetValue(path, out idx))
  50       {
  51         m_lock.ExitWriteLock();
  52         return idx;
  53       }
  54
  55       idx = m_idxToPaths.Count;
  56       m_pathsToIdx.Add(path, idx);
  57       m_idxToPaths.Add(path);
  58       m_lock.ExitWriteLock();
  59
  60       return idx;
  61                 }
  62
  63                 public void AddPaths(IEnumerable<string> paths)
  64                 {
  65                         m_lock.EnterWriteLock();
  66
  67                         foreach (string path in paths)
  68                         {
  69                                 int idx;
  70                                 if (!m_pathsToIdx.TryGetValue(path, out idx))
  71                                 {
  72                                         idx = m_idxToPaths.Count;
  73                                         m_pathsToIdx.Add(path, idx);
  74                                         m_idxToPaths.Add(path);
  75                                 }
  76                         }
  77
  78                         m_lock.ExitWriteLock();
  79                 }
  80
  81                 public bool ContainsPath(string path)
  82                 {
  83                         return TryGetIndexFromPath(path) != -1;
  84                 }
  85
  86                 public int TryGetIndexFromPath(string path)
  87                 {
  88                         m_lock.EnterReadLock();
  89
  90                         int idx;
  91                         if (m_pathsToIdx.TryGetValue(path, out idx))
  92                         {
  93                                 m_lock.ExitReadLock();
  94                                 return idx;
  95                         }
  96
  97                         m_lock.ExitReadLock();
  98                         return -1;
  99                 }
 100
 101     public int GetIndexFromPath(string path)
 102     {
 103                         int idx = TryGetIndexFromPath(path);
 104                         if (idx != -1)
 105                                 return idx;
 106                         else
 107                                 throw new Exception("Path does not exist: " + path);
 108     }
 109
 110     public string GetPathFromIndex(int idx)
 111     {
 112       string res = string.Empty;
 113
 114       m_lock.EnterReadLock();
 115       if (idx < m_idxToPaths.Count)
 116         res = m_idxToPaths[idx];
 117       m_lock.ExitReadLock();
 118
 119       return res;
 120     }
 121   }
 122 }