-- Класс для отслеживания баланса покупок/продаж
local BalanceTracker = {}
BalanceTracker.__index = BalanceTracker
-- Конструктор
function BalanceTracker:new()
    local obj = setmetatable({}, BalanceTracker)
    obj.isRunning = true
    obj.array5Min = {}
    obj.lastBarSeconds = 0
    obj.t_id = nil
    obj:createTable()
    return obj
end
-- Создание таблицы
function BalanceTracker:createTable()
    self.t_id = AllocTable()
    if self.t_id == nil then
        error("Failed to allocate table")
    end

    AddColumn(self.t_id, 0, "1", true, QTABLE_INT_TYPE, 15)
    AddColumn(self.t_id, 1, "2", true, QTABLE_INT_TYPE, 15)
    AddColumn(self.t_id, 2, "3", true, QTABLE_INT_TYPE, 15)
    AddColumn(self.t_id, 3, "4", true, QTABLE_INT_TYPE, 15)
    AddColumn(self.t_id, 4, "5", true, QTABLE_INT_TYPE, 15)
    CreateWindow(self.t_id)
    SetWindowCaption(self.t_id, "Баланс покупок/продаж")
    InsertRow(self.t_id, -1)
end
-- Установка цвета
function BalanceTracker:setColor(col, color)
    SetColor(self.t_id, 1, col, color, RGB(0,0,0), color, RGB(0,0,0))
end
function BalanceTracker:red(col)
    self:setColor(col, RGB(255,128,128))
end
function BalanceTracker:yellow(col)
    self:setColor(col, RGB(240,240,0))
end
function BalanceTracker:green(col)
    self:setColor(col, RGB(128,255,128))
end
-- Добавление нового бара в массив
function BalanceTracker:addNewBarToArray5Min()
    table.insert(self.array5Min, 0)
    if #self.array5Min > 5 then
        table.remove(self.array5Min, 1)
    end
    local offset = 5 - #self.array5Min
    for i = 1, #self.array5Min do
        self:setBalance(self.array5Min[i], i + offset - 1)
    end
end
-- Установка баланса и цвет ячейки
function BalanceTracker:setBalance(newBalance, numCol)
    SetCell(self.t_id, 1, numCol, tostring(newBalance))
    if numCol == 4 then
        self.array5Min[#self.array5Min] = newBalance
    end
    if newBalance < 0 then
        self:red(numCol)
    elseif newBalance == 0 then
        self:yellow(numCol)
    else
        self:green(numCol)
    end
end
-- Проверка на новую минуту
function BalanceTracker:checkNewMin(dealDateTime)
    dealDateTime.sec = 0
    dealDateTime.mcs = 0
    local seconds = os.time(dealDateTime)
    if seconds > self.lastBarSeconds then
        self.lastBarSeconds = seconds
        return true
    else
        return false
    end
end
-- Обработка сделки
function BalanceTracker:onAllTrade(alltrade)
    if alltrade.sec_code == SEC_CODE then
        local newMinute = self:checkNewMin(alltrade.datetime)
        if newMinute then
            self:addNewBarToArray5Min()
        end

        if bit.test(alltrade.flags, 0) then
            self:setBalance(self.array5Min[#self.array5Min] - alltrade.qty, 4)
        else
            self:setBalance(self.array5Min[#self.array5Min] + alltrade.qty, 4)
        end
    end
end
-- Остановка скрипта
function BalanceTracker:onStop()
    self.isRunning = false
end
-- Главный цикл
function BalanceTracker:main()
    while self.isRunning do
        sleep(100) -- Уменьшить частоту вызова
    end
end
--return BalanceTracker

local CLASS_CODE     = "SPBFUT"     -- Класс бумаги
local SEC_CODE       = "NGN4"       -- Код бумаги

-- Создание объекта класса и функции обратного вызова
local balanceTracker = BalanceTracker:new()
function main()
    balanceTracker:main()
end
function OnAllTrade(alltrade)
    balanceTracker:onAllTrade(alltrade)
end
function OnStop()
    balanceTracker:onStop()
end