-- Id: 7350
-- More information about this indicator can be found at:
-- http://fxcodebase.com/code/viewtopic.php?f=17&t=23207

--+------------------------------------------------------------------+
--|                               Copyright © 2018, Gehtsoft USA LLC |
--|                                            http://fxcodebase.com |
--+------------------------------------------------------------------+
--|                                      Developed by : Mario Jemic  |
--|                                          mario.jemic@gmail.com   |
--+------------------------------------------------------------------+
--|                                 Support our efforts by donating  |
--|                                 Patreon : https://goo.gl/GdXWeN  |
--|                                  Paypal : https://goo.gl/9Rj74e  |
--|                    BitCoin : 15VCJTLaz12Amr7adHSBtL9v8XomURo9RF  |
--|               BitCoin Cash : 1BEtS465S3Su438Kc58h2sqvVvHK9Mijtg  |
--|           Ethereum : 0x8C110cD61538fb6d7A2B47858F0c0AaBd663068D  |
--|                   LiteCoin : LLU8PSY2vsq7B9kRELLZQcKf5nJQrdeqwD  |
--+------------------------------------------------------------------+

function Init()
    indicator:name("Aroon Rainbow");
    indicator:description("Aroon Rainbow");
    indicator:requiredSource(core.Bar);
    indicator:type(core.Oscillator);
    indicator.parameters:addGroup("Calculation");
    indicator.parameters:addInteger("Period", "Period", "Period",  50, 3, 1000);   
	indicator.parameters:addBoolean("ON", "Use Averaging", "", false);
 
end

-- Indicator instance initialization routine
-- Processes indicator parameters and creates output streams
-- TODO: Refine the first period calculation for each of the output streams.
-- TODO: Calculate all constants, create instances all subsequent indicators and load all required libraries
-- Parameters block
local Period  ;

local first;
local source = nil;
local MA={};
local Raw={};
-- Streams block
local Data={};
local Out={};
local Color={};
 local ON;
-- Routine
function Prepare(nameOnly)
    Period = instance.parameters.Period
    
    source = instance.source;
    first = source:first();
    ON = instance.parameters.ON;

    local name = profile:id() .. "(" .. source:name() .. ", " .. Period  .. ")";
    instance:name(name);

    if   (nameOnly) then
        return;
    end

		local i;
		for i = 2, Period, 1 do
			MA[i] = core.indicators:create("AROON", source , i);
			first = math.max(first, MA[i].DATA:first());
		end
	   for i = 2, Period, 1 do
	    Raw[i] =instance:addInternalStream(0, 0);
        Data[i] =instance:addInternalStream(0, 0);
		Color[i] =instance:addInternalStream(0, 0); 
		end
    
	
	instance:ownerDrawn(true);
end

-- Indicator calculation routine
-- TODO: Add your code for calculation output values
function Update(period, mode)
    if period < first or not source:hasData(period) then
	return;
	end
	
	local i;	
	
	for i = 2, Period, 1 do
	   MA[i]:update(mode); 
       Data[i][period] =(( MA[i].UP[period]-MA[i].DOWN[period])+100)/2; 
	end	
	
	if period < source:size()-2 
	or period < first 
	then
	return;
	elseif period == source:size()-2  
	then	
		for i = first, source:size()-3 , 1 do
		 Averaging (  1, Period, i);		
		end
	elseif  period == source:size()-1 then 
	   
	   for i = source:size()-2, source:size()-1 , 1 do
	  	Averaging (  1, Period, i);		
		end 
	
	end
	
	
end

function Averaging ( FIRST, LAST, period)


   local i;	
   FIRST=FIRST+1;
  
	
	for i = FIRST, LAST , 1 do
	local count = 0;
	   
	            Raw[i][period] =  Data[i][period];   --5
				count=count+1;
			 
			 
		if ON then	 
				if i > FIRST then
				Raw[i][period] = Data[i-1][period];  -- 2
				count=count+1;
				end
				
				if i < LAST then
				Raw[i][period] = Data[i+1][period];  --8
				count=count+1;
				end
		
				if period < source:size()-1 then
				
					   Raw[i][period] =  Raw[i][period] + Data[i][period+1];   --6
						count=count+1;
					 
						if i > FIRST then
						Raw[i][period] =Raw[i][period] + Data[i-1][period+1];  -- 3
						count=count+1;
						end
						
						if i < LAST then
						Raw[i][period] =Raw[i][period] + Data[i+1][period+1];  --9
						count=count+1;
						end
				end
				if period > first then
				
						 Raw[i][period] =  Raw[i][period] + Data[i][period-1];   --4
						count=count+1;
					 
						if i > FIRST then
						Raw[i][period] =Raw[i][period] + Data[i-1][period-1];  -- 1
						count=count+1;
						end
						
						if i < LAST then
						Raw[i][period] =Raw[i][period] + Data[i+1][period-1];  --7
						count=count+1;
						end
				end
	    
		  Raw[i][period] = Raw[i][period] / count;
		
		end
		  if Raw[i][period] == nil then
		  return;
		  end
		  
		  Color[i][period] = Coloring ((Raw[i][period]), 50);	         
		   
	end


end

function Draw(stage, context)
    if stage~= 2 then
	return;
	end
	 
	local i;	
	
	local First= math.max(first, context:firstBar ());
	local Last= math.min(source:size()-1, context:lastBar ());
	
 
	local yCell = (context:bottom () -context:top ()) /Period
	
	for i = 2, Period, 1 do
	    y1= context:bottom () -(i)*yCell;
		y2= context:bottom () -(i-1)*yCell;
	    for period = First, Last, 1 do		   
        
		    x, x1, x2 = context:positionOfBar (period);		
          		
		    color1 = Color[i][period]
			color2 =Color[i][period]
			color3 = Color[i][period]
			color4 = Color[i][period]
		 
		    context:drawGradientRectangle (x1, y1, color1, x2, y1, color2, x2, y2, color3, x1, y2, color4);
		end
	end
	
end



function Coloring (value, mid)

 local color;

 if value <= mid then
 color = core.rgb(255 * (value / mid), 255, 0) 
 else 
 color = core.rgb(255, 255 - 255 * ((value - mid) / mid), 0)
 end
 

return  color;

end