trimming fat; own files; etc

This commit is contained in:
Chris Forbes
2009-12-05 11:56:16 +13:00
parent 0d81683e34
commit c171185034
22 changed files with 218 additions and 223 deletions

View File

@@ -2,15 +2,13 @@
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Net.Sockets;
using System.Threading;
namespace OpenRa.Game
{
class OrderManager
{
Stream savingReplay;
List<OrderSource> sources;
List<IOrderSource> sources;
int frameNumber = 0;
public int FramesAhead = 3;
@@ -30,14 +28,14 @@ namespace OpenRa.Game
public int FrameNumber { get { return frameNumber; } }
public OrderManager( IEnumerable<OrderSource> sources )
public OrderManager( IEnumerable<IOrderSource> sources )
{
this.sources = sources.ToList();
if (!IsNetplay)
StartGame();
}
public OrderManager( IEnumerable<OrderSource> sources, string replayFilename )
public OrderManager( IEnumerable<IOrderSource> sources, string replayFilename )
: this( sources )
{
savingReplay = new FileStream( replayFilename, FileMode.Create );
@@ -88,180 +86,4 @@ namespace OpenRa.Game
throw new InvalidOperationException( "(OrderManager) Frame number too large" );
}
}
interface OrderSource
{
void SendLocalOrders( int localFrame, List<Order> localOrders );
List<Order> OrdersForFrame( int currentFrame );
bool IsReadyForFrame( int frameNumber );
}
class LocalOrderSource : OrderSource
{
Dictionary<int, List<Order>> orders = new Dictionary<int,List<Order>>();
public List<Order> OrdersForFrame( int currentFrame )
{
// TODO: prune `orders` based on currentFrame.
if (!orders.ContainsKey(currentFrame))
return new List<Order>();
return orders[ currentFrame ];
}
public void SendLocalOrders( int localFrame, List<Order> localOrders )
{
if (localFrame == 0) return;
orders[ localFrame ] = localOrders;
}
public bool IsReadyForFrame( int frameNumber )
{
return true;
}
}
class ReplayOrderSource : OrderSource
{
BinaryReader replayReader;
public ReplayOrderSource( string replayFilename )
{
replayReader = new BinaryReader( File.Open( replayFilename, FileMode.Open ) );
}
public void SendLocalOrders( int localFrame, List<Order> localOrders ) { }
public List<Order> OrdersForFrame( int frameNumber )
{
if( frameNumber == 0 )
return new List<Order>();
try
{
var len = replayReader.ReadInt32() - 4;
var frame = replayReader.ReadInt32();
var ret = replayReader.ReadBytes( len ).ToOrderList();
if( frameNumber != frame )
throw new InvalidOperationException( "Attempted time-travel in OrdersForFrame (replay)" );
return ret;
}
catch( EndOfStreamException )
{
return new List<Order>();
}
}
public bool IsReadyForFrame( int frameNumber )
{
return true;
}
}
class NetworkOrderSource : OrderSource
{
TcpClient socket;
Dictionary<int, List<byte[]>> orderBuffers = new Dictionary<int, List<byte[]>>();
Dictionary<int, bool> gotEverything = new Dictionary<int, bool>();
public NetworkOrderSource( TcpClient socket )
{
this.socket = socket;
this.socket.NoDelay = true;
var reader = new BinaryReader( socket.GetStream() );
new Thread( () =>
{
for (; ; )
{
var len = reader.ReadInt32();
var frame = reader.ReadInt32();
var buf = reader.ReadBytes(len - 4);
lock (orderBuffers)
{
if (len == 5 && buf[0] == 0xef) /* got everything marker */
gotEverything[frame] = true;
else
{
/* accumulate this chunk */
if (!orderBuffers.ContainsKey(frame))
orderBuffers[frame] = new List<byte[]> { buf };
else
orderBuffers[frame].Add(buf);
}
}
}
} ) { IsBackground = true }.Start();
}
static List<byte[]> NoOrders = new List<byte[]>();
List<byte[]> ExtractOrders(int frame)
{
lock (orderBuffers)
{
List<byte[]> result;
if (!orderBuffers.TryGetValue(frame, out result))
result = NoOrders;
orderBuffers.Remove(frame);
gotEverything.Remove(frame);
return result;
}
}
public List<Order> OrdersForFrame( int currentFrame )
{
var orderData = ExtractOrders(currentFrame);
return orderData.SelectMany(a => a.ToOrderList()).ToList();
}
public void SendLocalOrders( int localFrame, List<Order> localOrders )
{
socket.GetStream().WriteFrameData( localOrders, localFrame );
}
public bool IsReadyForFrame( int frameNumber )
{
lock( orderBuffers )
return gotEverything.ContainsKey( frameNumber );
}
}
static class StreamExts
{
public static void Write( this Stream s, byte[] buf )
{
s.Write( buf, 0, buf.Length );
}
}
static class OrderIO
{
public static MemoryStream ToMemoryStream( this IEnumerable<Order> orders, int nextLocalOrderFrame )
{
var ms = new MemoryStream();
ms.Write( BitConverter.GetBytes( nextLocalOrderFrame ) );
foreach( var order in orders )
ms.Write( order.Serialize() );
return ms;
}
public static void WriteFrameData( this Stream s, IEnumerable<Order> orders, int frameNumber )
{
var ms = orders.ToMemoryStream( frameNumber );
s.Write( BitConverter.GetBytes( (int)ms.Length ) );
ms.WriteTo( s );
}
public static List<Order> ToOrderList( this byte[] bytes )
{
var ms = new MemoryStream( bytes );
var reader = new BinaryReader( ms );
var ret = new List<Order>();
while( ms.Position < ms.Length )
ret.Add( Order.Deserialize( reader ) );
return ret;
}
}
}