60 lines
1.3 KiB
C#
Executable File
60 lines
1.3 KiB
C#
Executable File
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using IjwFramework.Types;
|
|
using System.Collections;
|
|
|
|
namespace OpenRa.Collections
|
|
{
|
|
public class Set<T> : IEnumerable<T>
|
|
{
|
|
Dictionary<T, bool> data = new Dictionary<T, bool>();
|
|
|
|
public void Add( T obj )
|
|
{
|
|
data.Add( obj, false );
|
|
if( OnAdd != null )
|
|
OnAdd( obj );
|
|
}
|
|
|
|
public void Remove( T obj )
|
|
{
|
|
data.Remove( obj );
|
|
if( OnRemove != null )
|
|
OnRemove( obj );
|
|
}
|
|
|
|
public event Action<T> OnAdd;
|
|
public event Action<T> OnRemove;
|
|
|
|
public IEnumerator<T> GetEnumerator()
|
|
{
|
|
return data.Keys.GetEnumerator();
|
|
}
|
|
|
|
IEnumerator IEnumerable.GetEnumerator()
|
|
{
|
|
return GetEnumerator();
|
|
}
|
|
}
|
|
|
|
public class CachedView<T,U> : Set<U>
|
|
{
|
|
public CachedView( Set<T> set, Func<T, bool> include, Func<T, U> store )
|
|
: this( set, include, x => new[] { store( x ) } )
|
|
{
|
|
}
|
|
|
|
public CachedView( Set<T> set, Func<T,bool> include, Func<T,IEnumerable<U>> store )
|
|
{
|
|
foreach( var t in set )
|
|
if( include( t ) )
|
|
store( t ).Do( x => Add( x ) );
|
|
|
|
set.OnAdd += obj => { if( include( obj ) ) store( obj ).Do( x => Add( x ) ); };
|
|
set.OnRemove += obj => { if( include( obj ) ) store( obj ).Do( x => Remove( x ) ); };
|
|
}
|
|
}
|
|
}
|