transforms.js

const Transform = require("../../src/Transform").Transform;
const identityFunction = require("../../src/Transform").identityFunction;

var teamLogoTransform = new Transform(
    "select * from teams;",
    "nba",
    function(row) {
        var id = parseInt(row[0]);
        var y = Math.floor(id / 6);
        var x = id - y * 6;
        var ret = [];
        ret.push(row[0]);
        ret.push((x * 2 + 1) * 80);
        ret.push((y * 2 + 1) * 80 + 100);
        ret.push(row[1]);
        ret.push(row[2]);
        ret.push(row[3]);
        ret.push(row[4]);

        return Java.to(ret, "java.lang.String[]");
    },
    ["id", "x", "y", "team_id", "city", "name", "abbr"],
    true
);

var teamTimelineTransform = new Transform(
    "select game_id, year, month, day, home_team, away_team, home_score, away_score, 1 from games;",
    "nba",
    function(row, width, height, renderParams) {
        var ret = [];
        // id
        ret.push(row[0]);
        // x
        var curDate = new Date(row[1], row[2] - 1, row[3]);
        ret.push(
            d3
                .scaleTime()
                .domain([new Date(2017, 9, 17), new Date(2018, 3, 11)])
                .range([82, width - 82])(curDate)
        );
        // y
        var beginDate = new Date(2000, 0, 1);
        var oneDay = 24 * 60 * 60 * 1000;
        var daysPassed = Math.round(
            Math.abs((curDate.getTime() - beginDate.getTime()) / oneDay)
        );
        ret.push(
            daysPassed % 2 == 0
                ? renderParams.timelineUpperY
                : renderParams.timelineLowerY
        );

        // rest of the attributes
        for (var i = 1; i <= 8; i++) ret.push(row[i]);

        return Java.to(ret, "java.lang.String[]");
    },
    [
        "game_id",
        "x",
        "y",
        "year",
        "month",
        "day",
        "home_team",
        "away_team",
        "home_score",
        "away_score",
        "timeline"
    ],
    true,
    // uncomment below to enable updates on the timeline canvas

    {
        cx: function(oldRow, width, height) {
            let newRow = oldRow;
            let x = newRow["cx"];
            let reverseDate = d3
                .scaleLinear()
                .domain([82, width - 82])
                .range([new Date(2017, 9, 17), new Date(2018, 3, 11)])(x);
            let y = newRow["cy"];
            if (Math.abs(y - 510) > Math.abs(y - 740)) y = 740;
            else y = 510;
            newRow["cy"] = y;
            let month = reverseDate.getUTCMonth() + 1;
            let day = reverseDate.getUTCDate();
            let year = reverseDate.getUTCFullYear();
            newRow["month"] = month;
            newRow["day"] = day;
            newRow["year"] = year;
            return newRow;
        },
        cy: identityFunction,
        year: identityFunction,
        month: identityFunction,
        day: identityFunction,
        home_score: identityFunction,
        away_score: identityFunction
    }
);

var teamTimelineStaticTransform = new Transform(
    "select city, name, abbr from teams;",
    "nba",
    "",
    [],
    true
);

var playByPlayTransform = new Transform(
    "select games.game_id, period, qtr_time, score, margin, home_desc, away_desc, home_team, away_team, play_id, h_player_id, a_player_id" +
        " from plays, games" +
        " where plays.game_id = games.game_id;",
    "nba",
    function(row) {
        var ret = [];
        // game_id
        ret.push(row[0]);

        // y
        ret.push((+row[9] + 1) * 160);

        // period & qtr_time
        ret.push(row[1]);
        ret.push(row[2]);

        // reverse score
        if (row[3] == "None") ret.push("");
        else {
            var scores = row[3].split("-");
            ret.push(
                scores[1].replace(/\s+/, "") +
                    " - " +
                    scores[0].replace(/\s+/, "")
            );
        }

        // rest of the attributes
        for (var i = 4; i <= 8; i++) ret.push(row[i]);
        ret.push(row[10]);
        ret.push(row[11]);

        return Java.to(ret, "java.lang.String[]");
    },
    [
        "game_id",
        "y",
        "period",
        "qtr_time",
        "score",
        "margin",
        "home_desc",
        "away_desc",
        "home_team",
        "away_team",
        "h_player_id",
        "a_player_id"
    ],
    true
);

var playByPlayStaticTransform = new Transform(
    "select team1.abbr as abbr1, team2.abbr as abbr2 from teams as team1, teams as team2;",
    "nba",
    "",
    [],
    true
);

var boxscoreTransform = new Transform(
    "select * from player_boxscore;",
    "nba",
    "",
    [],
    true
);

module.exports = {
    teamLogoTransform,
    teamTimelineTransform,
    teamTimelineStaticTransform,
    playByPlayTransform,
    playByPlayStaticTransform,
    boxscoreTransform
};