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


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


function Init()
    indicator:name("Arbitrage triangular calculator");
    indicator:description("Arbitrage triangular calculator");
    indicator:requiredSource(core.Bar);
    indicator:type(core.Oscillator);
    
    for i=1,3,1 do
     AddSymbol(i);
    end

    indicator.parameters:addGroup("Display");
    indicator.parameters:addColor("clrIndex", "Index Color", "", core.rgb(255, 0, 0));
    indicator.parameters:addInteger("widthLinReg", "Line width", "Line width", 1, 1, 5);
    indicator.parameters:addInteger("styleLinReg", "Line style", "Line style", core.LINE_SOLID);
    indicator.parameters:setFlag("styleLinReg", core.FLAG_LINE_STYLE);
end

local source;
local barSize;
local SS;
local loading = false;
local offset;
local weekoffset;
local w={};

-- indexes for the instruments in the data table
local SymbolCount=0;
local last = nil;
-- data table
local data = {};
local SymbolsName;

function AddSymbol(ParName)
 indicator.parameters:addString("Symbol" .. ParName, "Symbol" .. ParName, "", "");
 indicator.parameters:setFlag("Symbol" .. ParName, core.FLAG_INSTRUMENTS);
end

-- add a new item into the data table
-- index        - is the index of the item in the table
-- instrument   - the instrument name
-- weight       - the weigth of the instrument
function AddCollectionItem(index, instrument, weight)
    local t, coll, from, to, tmp;
    t = {};
    t.instrument = instrument;
    t.data = nil;
    t.loading = false;
    t.weight = weight;
    t.rqfrom = nil;
    t.rqto = nil;
    data[index] = t;

    if first == nil or first > index then
        first = index;
    end
    if last == nil or last < index then
        last = index;
    end
end

function GetCurrencies(Symb)
 local Pos=string.find(Symb,"/");
 assert(Pos~=nil, "You may use Forex symbols only");
 local Cur1=string.sub(Symb,1,Pos-1);
 local Cur2=string.sub(Symb,Pos+1);
 return Cur1, Cur2;
end

function CheckSymbols()
 local i;
 local C1, C2, C1_, C2_;
 local S={};
 for i=1,3,1 do
  local Symb=instance.parameters:getString("Symbol" .. i);
  assert(Symb~="- Empty -", "All symbols must be specified");
  S[i]=Symb;
 end
 w[1]=1;
 C1, C2=GetCurrencies(S[1]);
 C1_, C2_=GetCurrencies(S[2]);
 w[2]=0;
 if C1==C1_ and C2~=C2_ then
  w[2]=-1;
  C1=C2_;
 elseif C1==C2_ and C2~=C1_ then
  w[2]=1;
  C1=C1_;
 elseif C2==C1_ and C1~=C2_ then
  w[2]=1;
  C2=C2_;
 elseif C2==C2_ and C1~=C1_ then
  w[2]=-1;
  C2=C1_;
 end
 assert(w[2]~=0, "Pairs do not a triangle");

 C1_, C2_=GetCurrencies(S[3]);
 w[3]=0;
 if C1==C1_ and C2==C2_ then
  w[3]=-1;
 elseif C1==C2_ and C2==C1_ then
  w[3]=1;
 end
 assert(w[3]~=0, "Pairs do not a triangle");
end

-- initialize the collection of the instruments
function InitCollection()
    SymbolCount=0;
    SymbolsName="";
    CheckSymbols();
    for i=1,3,1 do
     SymbName=instance.parameters:getString("Symbol" .. i);
     SymbWeight=w[i];
     if SymbName~="- Empty -" and SymbWeight~=0 then
      SymbolCount=SymbolCount+1;
      if SymbolsName~="" then
       SymbolsName=SymbolsName .. "*";
      end
      SymbolsName=SymbolsName .. SymbName .. "^(" .. SymbWeight .. ")";
      AddCollectionItem(SymbolCount, SymbName, SymbWeight);
     end
    end
    
end

-- prepare the indicator
function Prepare(nameOnly)  
    source = instance.source;
    host = core.host;
    barSize = source:barSize();
    
	
	InitCollection();
	
	
    local name = profile:id() .. "(" .. SymbolsName .. ")";
    instance:name(name);
	
	if   (nameOnly) then
        return;
    end 
	
	offset = host:execute("getTradingDayOffset");
    weekoffset = host:execute("getTradingWeekOffset");

     
    SS = instance:addStream("SS", core.Line, name .. ".SS", "SS", instance.parameters.clrIndex, 0);
    SS:setPrecision(math.max(2, instance.source:getPrecision()));
    SS:setWidth(instance.parameters.widthLinReg);
    SS:setStyle(instance.parameters.styleLinReg);
end

local p = {};
local w = {};

-- get the price of the specified instrument
function GetPrice(index, date)
    local t;

    local from, to, tmp;

    t = data[index];

    assert(t ~= nil, "internal error!");

    if t.data == nil then
        -- data is not loaded yet at all
        if source:isAlive() then
            to = 0;
        else
            to = source:date(source:size() - 1);
        end
        from = source:date(source:first());
        t.data = host:execute("getHistory", index, t.instrument, barSize, from, to, source:isBid());
        t.rqfrom = from;
        t.rqto = to;
        t.loading = true;
        loading = true;
        return 0, 0;
    elseif date < t.rqfrom then
        -- requested date is before the first item of the collection
        -- we have ever requested
        from = date;
        to = t.data:date(0);
        host:execute("extendHistory", index, t.data, from, to);
        t.rqfrom = from;
        t.loading = true;
        loading = true;
        return 0, 0;
    elseif not(source:isAlive()) and date > t.rqto then
        -- requested date is after the last item of the collection
        -- we have ever requested
        to = date;
        from = t.data:date(t.data:size() - 1);
        host:execute("extendHistory", index, t.data, from, to);
        t.rqto = to;
        t.loading = true;
        loading = true;
        return 0, 0;
    end

    local p;
    p = core.findDate (t.data, date, false);
    if p < 0 then
        return 0, 0;
    end
    return t.data.close[p], t.weight;
end

local lastdate = nil;

-- the function which is called to calculate the period
function Update(period, mode)

    if loading or period <= source:first() then
        return ;
    end

    -- do not calculate for the floating candle
    period = period - 1;

    if lastdate ~= nil and source:date(period) == lastdate then
        return ;
    end

    lastdate = source:date(period);

    local i, x, absent, a, b;
    absent = false;
    for i = first, last, 1 do
        a, b = GetPrice(i, lastdate);
        if a == 0 then
            absent = true;
        end
        p[i] = a;
        w[i] = b;
    end

    if loading then
        SS:setBookmark(1, period);
        return ;
    end

    if absent then
        if SS:hasData(period - 1) then
            SS[period] = SS[period - 1];
        end
    else
        x = 1;
        for i = first, last, 1 do
            x = x * math.pow(p[i], w[i]);
        end
        SS[period] = x;
    end

    period = period + 1;

    if period > 0 and period == source:size() - 1 then
        SS[period] = SS[period - 1];
    end
end

function AsyncOperationFinished(cookie)

    local t;
    t = data[cookie];
    t.loading = false;
    for i = first, last, 1 do
        if data[i].loading then
            return ;
        end
    end
    loading = false;

    local period;
    period = SS:getBookmark(1);

    if (period < 0) then
        period = 0;
    end
    instance:updateFrom(period);
end