namespace QuantConnect
{
    public class BootCampTask : QCAlgorithm
    {
    	int period = 125;
    	int totalPairsToHold = 6; //Must be an even number, half will be held long, half short.
    	Dictionary<string, MomentumPercent> indicators = new Dictionary<string, MomentumPercent>();
    	decimal leverage = 5m;
    	
        public override void Initialize()
        {
            SetStartDate(2016, 6, 1);
            SetEndDate(2021, 6, 1);
            SetCash(100000);
            
            string[] tickers = new string [18] { "USDCAD","EURJPY","EURUSD","EURCHF","USDCHF","EURGBP",
												 "GBPUSD","AUDCAD","NZDUSD","GBPCHF","AUDUSD","GBPJPY",
												 "USDJPY","CHFJPY","EURCAD","AUDJPY","EURAUD","AUDNZD" };
                 
            //SetBrokerageModel(BrokerageName.FxcmBrokerage);       
            foreach (string ticker in tickers)
            {
            	AddForex(ticker, Resolution.Daily, Market.FXCM);
	            indicators.Add(ticker, MOMP(ticker, period, Resolution.Daily));
	            Securities[ticker].FeeModel = new ConstantFeeModel(0);
            }
            
            SetWarmup(period);
        }

        public override void OnData(Slice data)
        {
        	if (IsWarmingUp) return;
        	
        	List<string> gainers = indicators.Keys.OrderByDescending(x => indicators[x]).Take(totalPairsToHold / 2).ToList();
			List<string> losers = indicators.Keys.OrderBy(x => indicators[x]).Take(totalPairsToHold / 2).ToList();
			
			foreach (string ticker in indicators.Keys)
			{
				if (!gainers.Contains(ticker) && !losers.Contains(ticker))
				{
					if (Portfolio[ticker].Invested)
					{
						Liquidate(ticker);
					}
				}
			}
			
			foreach (string ticker in gainers)
			{
				if (!Portfolio[ticker].Invested)
				{
					SetHoldings(ticker, leverage / (decimal)totalPairsToHold);
				}
			}
			foreach (string ticker in losers)
			{
				if (!Portfolio[ticker].Invested)
				{
					SetHoldings(ticker, -leverage / (decimal)totalPairsToHold);
				}
			}
        }
    }
}