namespace Ecng.Trading.Algo.Testing
{
	using System;
	using System.Collections.Generic;

	using Ecng.Trading.BusinessEntities;
	using Ecng.Common;

	/// <summary>
	/// Настройки генератора стакана.
	/// </summary>
	public class MarketDepthGeneratorSettings : GeneratorSettings
	{
		/// <summary>
		/// Создать <see cref="MarketDepthGeneratorSettings"/>.
		/// </summary>
		public MarketDepthGeneratorSettings()
		{
			this.MaxBidsDepth = 10;
			this.MaxAsksDepth = 10;
		}

		/// <summary>
		/// Максимальная глубина бидов.
		/// </summary>
		/// <remarks>
		/// Значение по умолчанию равно 10.
		/// </remarks>
		public int MaxBidsDepth { get; set; }

		/// <summary>
		/// Максимальная глубина офферов.
		/// </summary>
		/// <remarks>
		/// Значение по умолчанию равно 10.
		/// </remarks>
		public int MaxAsksDepth { get; set; }
	}

	/// <summary>
	/// Генератор стакана случайным методом.
	/// </summary>
	public class MarketDepthGenerator
	{
		/// <summary>
		/// Создать <see cref="MarketDepthGenerator"/>.
		/// </summary>
		public MarketDepthGenerator()
		{
			this.Settings = new MarketDepthGeneratorSettings();
		}

		/// <summary>
		/// Настройки генератора.
		/// </summary>
		public MarketDepthGeneratorSettings Settings { get; private set; }

		/// <summary>
		/// Получить стартовую цену, относительно которой будет производится генерация стакана.
		/// </summary>
		/// <param name="security">Инструмент.</param>
		/// <returns>Стартовая цена.</returns>
		public virtual double GetSeedPrice(Security security)
		{
			if (security == null)
				throw new ArgumentNullException("security");

			double seedPrice = 0;

			if (security.LastTrade.Price != 0)
				seedPrice = security.LastTrade.Price;
			else if (security.BestBid.Price != 0)
				seedPrice = security.BestBid.Price + security.BestPair.SpreadPrice / 2;
			else if (security.BestAsk.Price != 0)
				seedPrice = security.BestAsk.Price - security.BestPair.SpreadPrice / 2;
			else if (security.ClosePrice != 0)
				seedPrice = security.ClosePrice;
			else if (security.OpenPrice != 0)
				seedPrice = security.OpenPrice;

			if (seedPrice == 0)
				throw new ArgumentException("Инструмент содержит недостаточно информации для генерации стакана.", "security");

			return seedPrice + RandomGen.GetInt(-this.Settings.MaxPriceStepCount, this.Settings.MaxPriceStepCount) * security.MinStepSize;
		}

		/// <summary>
		/// Сгенерировать стакан.
		/// </summary>
		/// <param name="security">Инструмент, стакан для которого необходимо сгенерировать.</param>
		/// <returns>Сгенерированный стакан.</returns>
		public virtual MarketDepth Generate(Security security)
		{
			var marketDepth = new MarketDepth(security);
			Generate(marketDepth);
			return marketDepth;
		}

		/// <summary>
		/// Сгенерировать стакан.
		/// </summary>
		/// <param name="depth">Стакан с котировками, которые необходимо обновить случайным образом.</param>
		/// <returns>Сгенерированный стакан.</returns>
		public virtual void Generate(MarketDepth depth)
		{
			if (depth == null)
				throw new ArgumentNullException("depth");

			var bids = new List<Quote>();
			var asks = new List<Quote>();

			var seedPrice = GetSeedPrice(depth.Security);

			var lastBidPrice = seedPrice;
			var lastAskPrice = seedPrice;

			for (var i = 0; i < this.Settings.MaxBidsDepth; i++)
			{
				var quote = CreateQuote(lastBidPrice, depth.Security, OrderDirections.Buy);

				if (quote.Price <= 0)
					break;

				bids.Add(quote);
				lastBidPrice = quote.Price;
			}

			for (var i = 0; i < this.Settings.MaxAsksDepth; i++)
			{
				var quote = CreateQuote(lastAskPrice, depth.Security, OrderDirections.Sell);

				if (quote.Price <= 0)
					break;

				asks.Add(quote);
				lastAskPrice = quote.Price;
			}

			depth.Update(bids, asks, true);
		}

		private Quote CreateQuote(double startPrice, Security security, OrderDirections direction)
		{
			if (security == null)
				throw new ArgumentNullException("security");

			if (security.MinStepSize == 0)
				throw new ArgumentException("Минимальный шаг цены не может быть равен 0.", "security");

			return new Quote
			{
				OrderDirection = direction,
				Price = security.ShrinkPrice(startPrice + (direction == OrderDirections.Sell ? 1 : -1) * (RandomGen.GetInt(1, this.Settings.MaxPriceStepCount) * security.MinStepSize)),
				Security = security,
				Volume = RandomGen.GetInt(1, this.Settings.MaxVolume),
			};
		}
	}
}