opensim/OpenSim/Region/Environment/Scenes/EntityManager.cs

using System;
using System.Collections;
using System.Collections.Generic;
using OpenMetaverse;

namespace OpenSim.Region.Environment.Scenes
{
    public class EntityManager : IEnumerable<EntityBase>
    {
        private readonly Dictionary<UUID,EntityBase> m_eb_uuid = new Dictionary<UUID, EntityBase>();
        private readonly Dictionary<uint, EntityBase> m_eb_localID = new Dictionary<uint, EntityBase>();
        private readonly Object m_lock = new Object();

        [Obsolete("Use Add() instead.")]
        public void Add(UUID id, EntityBase eb)
        {
            Add(eb);
        }

        public void Add(EntityBase entity)
        {
            lock(m_lock)
            {
                m_eb_uuid.Add(entity.UUID, entity);
                m_eb_localID.Add(entity.LocalId, entity);
            }
        }

        public void InsertOrReplace(EntityBase entity)
        {
            lock(m_lock)
            {
                m_eb_uuid[entity.UUID] = entity;
                m_eb_localID[entity.LocalId] = entity;
            }
        }

        public void Clear()
        {
            lock (m_lock)
            {
                m_eb_uuid.Clear();
                m_eb_localID.Clear();
            }
        }

        public int Count
        {
            get
            {
                lock (m_lock)
                {
                    return m_eb_uuid.Count;
                }
            }
        }

        public bool ContainsKey(UUID id)
        {
            lock(m_lock)
            {
                return m_eb_uuid.ContainsKey(id);
            }
        }

        public bool ContainsKey(uint localID)
        {
            lock (m_lock)
            {
                return m_eb_localID.ContainsKey(localID);
            }
        }

        public bool Remove(uint localID)
        {
            lock(m_lock)
            {
                bool a = m_eb_uuid.Remove(m_eb_localID[localID].UUID);
                bool b =  m_eb_localID.Remove(localID);

                return a && b;
            }
        }

        public bool Remove(UUID id)
        {
            lock(m_lock)
            {
                bool a = m_eb_localID.Remove(m_eb_uuid[id].LocalId);
                bool b = m_eb_uuid.Remove(id);

                return a && b;
            }
        }

        public List<EntityBase> GetAllByType<T>()
        {
            List<EntityBase> tmp = new List<EntityBase>();

            lock(m_lock)
            {
                foreach (KeyValuePair<UUID, EntityBase> pair in m_eb_uuid)
                {
                    if(pair.Value is T)
                    {
                        tmp.Add(pair.Value);
                    }
                }
            }

            return tmp;
        }

        public List<EntityBase> GetEntities()
        {
            lock (m_lock)
            {
                return new List<EntityBase>(m_eb_uuid.Values);
            }
        }

        public EntityBase this[UUID id]
        {
            get
            {
                lock (m_lock)
                {
                    return m_eb_uuid[id];
                }
            }
            set
            {
                InsertOrReplace(value);
            }
        }

        public EntityBase this[uint localID]
        {
            get
            {
                lock (m_lock)
                {
                    return m_eb_localID[localID];
                }
            }
            set
            {
                InsertOrReplace(value);
            }
        }

        public bool TryGetValue(UUID key, out EntityBase obj)
        {
            lock(m_lock)
            {
                return m_eb_uuid.TryGetValue(key, out obj);
            }
        }

        public bool TryGetValue(uint key, out EntityBase obj)
        {
            lock (m_lock)
            {
                return m_eb_localID.TryGetValue(key, out obj);
            }
        }

        /// <summary>
        /// This could be optimised to work on the list 'live' rather than making a safe copy and iterating that.
        /// </summary>
        /// <returns></returns>
        public IEnumerator<EntityBase> GetEnumerator()
        {
            return GetEntities().GetEnumerator();
        }

        IEnumerator IEnumerable.GetEnumerator()
        {
            return GetEnumerator();
        }
    }
}