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

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

function Init()
    indicator:name("Coloured Stochastic Indicator");
    indicator:description(" ");
    indicator:requiredSource(core.Bar);
    indicator:type(core.Oscillator);	

 
	
		
	indicator.parameters:addGroup(" Calculation");
    indicator.parameters:addInteger("k"  , "Number of periods for %K", "", 5, 2, 1000);
    indicator.parameters:addInteger("sd" , "%D slowing periods", "", 3, 2, 1000);
      indicator.parameters:addInteger("d" , "Number of periods for %D", "", 3, 2, 1000);

    indicator.parameters:addString("ks" , "Smoothing type for %K", "", "MVA");
    indicator.parameters:addStringAlternative("ks" , "MVA", "", "MVA");
    indicator.parameters:addStringAlternative("ks" , "EMA","", "EMA");
    indicator.parameters:addStringAlternative("ks" , "MT4","", "FS");
	 
	indicator.parameters:addString("ds" , "Smoothing type for %D", "", "MVA");
    indicator.parameters:addStringAlternative("ds" , "MVA", "", "MVA");
    indicator.parameters:addStringAlternative("ds" , "EMA","", "EMA");
 

	
	indicator.parameters:addGroup("Line Style");
    
	indicator.parameters:addColor("Up", "OB Color", "",  core.rgb(255, 0, 0));
	indicator.parameters:addColor("Down", "OS Color", "",  core.rgb(0, 255, 0));	
	indicator.parameters:addColor("Neutral", "Neutral Color", "",  core.rgb(0, 0, 0));	
	 
    indicator.parameters:addInteger("width1", "K Line Width (in pixels)", "", 1, 1, 5);
    indicator.parameters:addInteger("style1", "K Line Style", "", core.LINE_SOLID);
    indicator.parameters:setFlag("style1", core.FLAG_LEVEL_STYLE);	
	
	indicator.parameters:addInteger("width2", "D Line Width (in pixels)", "", 1, 1, 5);
    indicator.parameters:addInteger("style2", "D Line Style", "", core.LINE_SOLID);
    indicator.parameters:setFlag("style2", core.FLAG_LEVEL_STYLE);	
	
	indicator.parameters:addGroup("Levels Style" ); 
    indicator.parameters:addDouble("overbought", "Overbought Level", "", 80, 0, 100);
    indicator.parameters:addDouble("oversold", "Oversold Level", "", 30, 0, 100);
	
    indicator.parameters:addInteger("level_overboughtsold_width", "Width", "", 1, 1, 5);
    indicator.parameters:addInteger("level_overboughtsold_style", "Sryle", "", core.LINE_SOLID);
	indicator.parameters:setFlag("level_overboughtsold_style", core.FLAG_LEVEL_STYLE);
    indicator.parameters:addColor("level_overboughtsold_color", "Color", "", core.rgb(0, 0, 255));
	
	
end

 

 
local K;
local D;
 
local k, sd, d, ks, ds 
local source;
local Indicator ;

local first;

 local Neutral,Up, Down;
 local OB, OS;

 function Prepare(nameOnly)    
  
   k=instance.parameters.k;
   sd=instance.parameters.sd;
   d=instance.parameters.d;
   ks=instance.parameters.ks;
   ds=instance.parameters.ds;
   
   OS=instance.parameters.oversold;
   OB=instance.parameters.overbought;
   
   Neutral=instance.parameters.Neutral;
   Up=instance.parameters.Up;
   Down=instance.parameters.Down;
    
    source = instance.source;
   
    local name =  profile:id() ..", (" ..   k ..", " .. sd..", " .. d..", " .. ks..", " .. ds..")";
	
	instance:name(name);
	
	  if   (nameOnly) then
        return;
    end
	 
    Indicator = core.indicators:create("STOCHASTIC", source,  k, sd, d, ks, ds);  
	first=   Indicator.D:first();
	
	
	K= instance:addStream("K", core.Line, "K", "K", Neutral, first);
	D= instance:addStream("D", core.Line, "D", "D", Neutral, first);
	    
    K:setPrecision(2);
	D:setPrecision(2);
	
	K:setWidth(instance.parameters.width1);
    K:setStyle(instance.parameters.style1);
	
	D:setWidth(instance.parameters.width2);
    D:setStyle(instance.parameters.style2);
	
	
	K:addLevel(0);
	K:addLevel(OB, instance.parameters.level_overboughtsold_style, instance.parameters.level_overboughtsold_width, instance.parameters.level_overboughtsold_color);
    K:addLevel(OS, instance.parameters.level_overboughtsold_style, instance.parameters.level_overboughtsold_width, instance.parameters.level_overboughtsold_color);
    K:addLevel(100);
	 
end


function Update(period, mode)

    
	Indicator:update(mode); 
	
		if period < first then
        return;
        end
		
 
	K[period]=Indicator.K[period];
	D[period]=Indicator.D[period];
	
	if K[period] > OB then
	K:setColor(period, Up);
	elseif K[period] < OS then
	K:setColor(period, Down);
	else
	K:setColor(period, Neutral);
	end
	
	if D[period] > OB then
	D:setColor(period, Up);
	elseif K[period] < OS then
	D:setColor(period, Down);
	else
	D:setColor(period, Neutral);
	end
	
end