Diagram.lua

local SModule = require('StateModule');

local Set = require('common/Set');

local Mapper = require('common/Mapper');

local u = require('common/Utils');

local WManager = require('WindowsManager');

local d = SModule:subclass("Diagram");

local axes = { "x", "y", "z" };

function d:initialize(id)
    SModule.initialize(self, id);
    self._axisMap = Mapper(1, 2);
    self._transform = love.math.newTransform();
    self._sitems = Set();
    self._transform:scale(100);
    self._dirScale = 10;
    self._velScale = 10;
    self._srcColor = { 1, 1, 1, 1 };
    self._dirColor = { 0.9, 0.9, 1 };
    self._velColor = { 0.5, 0.5, 1, 1 }
    WManager:register(self);
end

function d:setScene(s)
    if self._scene ~= s then
        self:StateChanged();
    end
    self._scene = s;
end

function d:addSourceItem(sitem)
    if sitem:isMono() then
        self._sitems:addSingle(sitem);
    end
end

function d:removeSourceItem(sitem)
    self._sitems:removeSingle(sitem);
end

function d:hasSourceItem(sitem)
    return self._sitems:has(sitem);
end

function d:drawAxes()
    local w, h = love.window.getMode();
    w = w / 2;
    h = h / 2;
    love.graphics.line(-w, 0, w, 0);
    love.graphics.line(0, -h, 0, h);
end

function d:drawSourceItem(si)
    local x, y, z = si:getPosition();
    local vol = si:getVolume();
    x, y = self._axisMap:map(x, y, z);
    x, y = self._transform:transformPoint(x, y);
    love.graphics.push();
    love.graphics.translate(x, y);
--     local font = love.graphics.getFont();
    x, y, z = si:getDirection();
    x, y = self._axisMap:map(x, y, z);
    x = x * self._dirScale;
    y = y * self._dirScale;
    local dirAngle = u.VectorAngle(x, y);
    local iangle, oangle, ovol = si:getCone();
    love.graphics.setColor(0.5, 0.5, 0.5, 0.5);
    love.graphics.circle("fill", 0, 0, ovol * 100);
    love.graphics.setColor(0.7, 0.7, 0.7, 1);
--     love.graphics.arc("fill", 0, 0, (ovol + vol) / 2, dirAngle + oangle / 2, dirAngle - oangle / 2);
    love.graphics.arc("fill", 0, 0, (ovol + vol) * 50, dirAngle - oangle / 2 - math.pi / 2, dirAngle + oangle / 2 - math.pi / 2);
    love.graphics.setColor(1, 1, 1, 1);
    love.graphics.arc("fill", 0, 0, vol * 100, dirAngle - iangle / 2 - math.pi / 2, dirAngle + iangle / 2 - math.pi / 2);
    love.graphics.setColor(unpack(self._srcColor));
    love.graphics.circle("fill", 0, 0, 3)
    love.graphics.setColor(unpack(self._dirColor));
    love.graphics.line(0, 0, x, y);
    x, y, z = si:getVelocity();
    x, y = self._axisMap:map(x, y, z);
    x = x * self._velScale;
    y = y * self._velScale;
    love.graphics.setColor(unpack(self._velColor));
    love.graphics.line(0, 0, x, y);
    local angle = u.VectorAngle(x, y);
    love.graphics.push();
    love.graphics.translate(x, y);
    love.graphics.rotate(angle);
    love.graphics.polygon("fill", 0, 0, 5, 10, -5, 10);
    love.graphics.pop();
    love.graphics.setColor(unpack(self._srcColor));
    love.graphics.print(si.id, 5);
    love.graphics.pop();
end

function d:drawListener(l)
    local x, y, z = l:getPosition();
    x, y = self._axisMap:map(x, y, z);
    x, y = self._transform:transformPoint(x, y);
    love.graphics.push();
    love.graphics.translate(x, y);
--     local font = love.graphics.getFont();
    love.graphics.setColor(unpack(self._srcColor));
    love.graphics.circle("fill", 0, 0, 3)
    x, y, z = l:getOrientation();
    x, y = self._axisMap:map(x, y, z);
    x = x * self._dirScale;
    y = y * self._dirScale;
    love.graphics.setColor(unpack(self._dirColor));
    love.graphics.line(0, 0, x, y);
    x, y, z = l:getVelocity();
    x, y = self._axisMap:map(x, y, z);
    x = x * self._velScale;
    y = y * self._velScale;
    love.graphics.setColor(unpack(self._velColor));
    love.graphics.line(0, 0, x, y);
    local sin = x / (x^2 + y^2)^0.5;
    local cos = y / (x^2 + y^2)^0.5;
    local angle = math.asin(sin);
    if cos > 0 then
        angle = math.pi - angle;
    end
    love.graphics.push();
    love.graphics.translate(x, y);
    love.graphics.rotate(angle);
    love.graphics.polygon("fill", 0, 0, 5, 10, -5, 10);
    love.graphics.pop();
    love.graphics.setColor(unpack(self._srcColor));
    love.graphics.print("Listener", 5);
    love.graphics.pop();
end

function d:draw()
    love.graphics.push();
    local w, h = love.window.getMode();
    love.graphics.translate(w / 2, h / 2);
    self:drawAxes();
    if self._scene then
        self:drawListener(self._scene)
    end
    for key, item in pairs(self._sitems:get()) do
        self:drawSourceItem(item);
    end
    love.graphics.pop();
end

function d:windowTitle()
    return "Diagram";
end

function d:windowContent()
    Slab.BeginLayout("DiagramLayout", { Columns = 2 });
    Slab.SetLayoutColumn(1);
    Slab.Text("Diagram");
    Slab.Text("Map x:");
    Slab.Text("Map y:");
    Slab.Text("Direction scale:");
    Slab.Text("Velocity scale:");
    Slab.SetLayoutColumn(2);
    if Slab.Button("Reset") then
        self._axisMap:set(1, 2);
--         item.mouseRecorder.positionScale = 0.01;
--         item.mouseRecorder.velocityScale = 500;
    end
    local maxis = self._axisMap:getSingleReverse(1);
    if Slab.BeginComboBox("MapDiagramX", { Selected = axes[maxis] }) then
        for key, val in ipairs(axes) do
            if key ~= maxis then
                if Slab.TextSelectable(val) then
                    self._axisMap:setSingle(key, 1);
                    self:StateChanged();
                end
            end
        end
        Slab.EndComboBox();
    end
    maxis = self._axisMap:getSingleReverse(2);
    if Slab.BeginComboBox("MapDiagramY", { Selected = axes[maxis] }) then
        for key, val in ipairs(axes) do
            if key ~= maxis then
                if Slab.TextSelectable(val) then
                    self._axisMap:setSingle(key, 2);
                    self:StateChanged();
                end
            end
        end
        Slab.EndComboBox();
    end
    local scale = Slab.PercentageDrag("DirScale", self._dirScale);
    if scale ~= nil then
        self._dirScale = scale;
        self:StateChanged();
    end
    local scale = Slab.PercentageDrag("VelScale", self._velScale);
    if scale ~= nil then
        self._velScale = scale;
        self:StateChanged();
    end
    Slab.EndLayout();

end

return d;