ptt-sententree/static/js/pttPush.js

init()
var tsvPath = ''
var titlePath = ''
var tsvString
var defaultStartDate
var defaultEndDate
var totalPosts
var startDate
var endDate
var wordPushList
var randId
var globKeyword = ''
var stopwords = []

function init() {
    $.ajax({
        type: 'POST',
        url: 'ptt_push/init',
        dataType: 'json',
        success: function(data) {
            console.log(data)
            tsvString = data.Result.tsv
            wordPushList = JSON.parse(data.Result.json)
            stopwords = data.Result.stopwords
            console.log(wordPushList)
            $('#idBox').val(data.Result.author)
            buildSentetree(tsvString)
            changeMode(2)
        }
    })
    $(document).ready(function() {
        $(window).keydown(function(event) {
            if (event.keyCode == 13) {
                event.preventDefault()
                sendRequest()
            }
        });
    });
    $(window).on('mousemove', function(e) {
        $('#nodeTitle').css({
            left: e.pageX,
            top: e.pageY
        })
    })
    $('#titleListContainer').hover(
        function() { // Run on hover/mouseenter
            $(this).css('overflow', 'auto')
        },
        function() { // Run on mouseleave
            $(this).css('overflow', 'hidden')
        }
    )
    $('#titleListLayer').click(function(e) {
        if ($('#titleListLayer').is(e.target)) {
            hideTitles()
        }
    })
    $('#stopWordEditorLayer').click(function(e) {
        if ($('#stopWordEditorLayer').is(e.target)) {
            hideStopWordEditor()
        }
    })
    $('#idfEditorLayer').click(function(e) {
        if ($('#idfEditorLayer').is(e.target)) {
            hideIdfEditor()
        }
    })
}

function loadTemplate(num) {
    templates = [{
        userId: '',
        aid: '1Vv4iFY6',
        keyword: '',
        mode: 0,
    }, {
        userId: '',
        aid: '1VyJ2vP_',
        keyword: '',
        mode: 0
    }, {
        userId: 'xetherz3',
        aid: '',
        keyword: '',
        mode: 0
    }]
    let chosenTemplate = templates[num]
    $('#idBox').val(chosenTemplate.userId)
    $('#titleBox').val(chosenTemplate.aid)
    $('#keywordBox').val(chosenTemplate.keyword)
    changeMode(0)
    sendRequest()
}

function clearStopWord() {
    stopwords = []
    $('#sweContainer').html('')
}

function scrollIdfList() {
    let targetWord = $('#idfTarget').val()
    let wordList = $("#ieContainer").find("tr").slice(1).map((_, a) => { return $($(a).find("td")[0]).html() }).get()
    console.log({ "targetWord": targetWord, 'list': wordList })
    if (targetWord != '') {
        if (wordList.indexOf(targetWord) >= 0) {

        } else {
            wordList.push(targetWord)
            $('#ieContainer').find('tbody').append($('<tr>').attr('class', 'w3-display-container').append($('<td>').attr('style', 'vertical-align: middle;').append(targetWord)).append($('<td>').attr('class', 'w3-right-align').attr('style', 'vertical-align: middle;').append($('<input>').attr('type', 'number').val(Object.keys(idfTable).indexOf(targetWord) < 0 ? 0 : idfTable[targetWord]))))
        }
        console.log($($('#ieContainer').children('tbody').find('tr')[[wordList.indexOf(targetWord)]]).position())
        $("#ieTableContainer").scrollTop($($('#ieContainer').children('tbody').find('tr')[[wordList.indexOf(targetWord)]]).position().top - 50)
    }
    $('#newStopWord').val('')
}

function addStopWord() {
    newswRaw = $('#newStopWord').val()
    newswList = newswRaw.split(' ')
    for (newsw of newswList) {
        if (newsw != '') {
            if (stopwords.includes(newsw)) {

            } else {
                stopwords.push(newsw)
                $('#sweContainer').append($('<li>').attr('class', 'w3-display-container').append($('<span>').append(newsw)).append($('<span>').attr('class', 'w3-button w3-hover-red w3-transparent w3-display-right').click(function(e) {
                    var index = $(this).parent().index()
                    console.log(stopwords[index])
                    stopwords.splice(index, 1)
                    console.log(stopwords)
                    $('#sweContainer li').eq(index).remove()
                }).append("&times;")))
                console.log(document.getElementById('sweContainer').children[stopwords.indexOf(newsw)])
            }
            document.getElementById("sweContainer").scrollTop = document.getElementById('sweContainer').children[stopwords.indexOf(newsw)].offsetTop
        }
    }
    $('#newStopWord').val('')
}

function changeMode(_mode) {
    for (i = 0; i < 4; i++) {
        if (i == _mode) {
            $('#modeSelector button').eq(i).css("color", "#aaa")
        } else {
            $('#modeSelector button').eq(i).css("color", "#000")
        }
    }
    mode = _mode
    destroyCurrentGraph()
    d3.select('#graph').append('div').attr('id', 'vis')
    buildSentetree(tsvString)
}

function showIdfEditor() {
    $(window).unbind('keydown')
    $(window).keydown(function(event) {
        if (event.keyCode == 13) {
            scrollIdfList()
        }
    })
    $('#ieContainer').empty().append(
            $('<thead>').append($('<tr>')
                .append($('<th>')
                    .attr('style', 'position: sticky; top: 0; background: white;')
                    .append('單詞'))
                .append($('<th>')
                    .attr('class', 'w3-center-align')
                    .attr('style', 'position: sticky; top: 0; background: white;')
                    .append('操作'))
                .append($('<th>').attr('class', 'w3-right-align')
                    .attr('style', 'position: sticky; top: 0; background: white;')
                    .append('單詞頻率')
                )
            )
        )
        .append($('<tbody>'))
    for (word of Object.entries(idfTable).sort((a, b) => { return (b[1] - a[1]) }).map((a) => { return a[0] }).slice(0, 1000)) {
        $('#ieContainer').find('tbody')
            .append($('<tr>')
                .attr('class', 'w3-display-container')
                .append($('<td>')
                    .attr('style', 'vertical-align: middle;')
                    .append(word))
                .append($('<td>')
                    .attr('class', 'w3-right-align')
                    .append($('<button>')
                        .attr('class', 'general-button')
                        .html('設為最小')
                        .click(function() {
                            $(this)
                                .parent()
                                .parent()
                                .find('input').val(0)
                        })
                    )
                    .append($('<button>')
                        .attr('class', 'general-button')
                        .html('設為最大')
                        .click(function() {
                            $(this)
                                .parent()
                                .parent()
                                .find('input').val(Object.values(idfTableOrig).reduce((a, b) => {
                                    return a > b ? a : b
                                }))
                        })
                    )
                    .append($('<button>')
                        .attr('class', 'general-button')
                        .html('重設')
                        .click(function() {
                            var _word = $($(this)
                                .parent()
                                .parent()
                                .find('td')[0]).html()
                            $(this)
                                .parent()
                                .parent()
                                .find('input').val(idfTableOrig[_word])
                        })
                    )
                )
                .append($('<td>')
                    .attr('class', 'w3-right-align')
                    .attr('style', 'vertical-align: middle;')
                    .append($('<input>')
                        .attr('type', 'number')
                        .val(idfTable[word])
                    )
                )
            )
    }
    $('#idfEditorLayer').removeClass('hidden')
}

function hideIdfEditor() {
    $(window).unbind('keydown')
    $(window).keydown(function(event) {
        if (event.keyCode == 13) {
            event.preventDefault()
            sendRequest()
        }
    })
    $('#idfEditorLayer').addClass('hidden')
}

function showStopwordEditor() {
    console.log(stopwords)
    $(window).unbind('keydown')
    $(window).keydown(function(event) {
        if (event.keyCode == 13) {
            addStopWord()
        }
    })
    $('#sweContainer').empty()
    for (word of stopwords) {
        $('#sweContainer').append($('<li>').attr('class', 'w3-display-container').append($('<span>').append(word)).append($('<span>').attr('class', 'w3-button w3-hover-red w3-transparent w3-display-right').click(function(e) {
            var index = $(this).parent().index()
            console.log(stopwords[index])
            stopwords.splice(index, 1)
            console.log(stopwords)
            $('#sweContainer li').eq(index).remove()
        }).append("&times;")))
    }
    $('#stopWordEditorLayer').removeClass('hidden')
}

function hideStopWordEditor() {
    $(window).unbind('keydown')
    $(window).keydown(function(event) {
        if (event.keyCode == 13) {
            event.preventDefault()
            sendRequest()
        }
    })
    $('#stopWordEditorLayer').addClass('hidden')
}

function downloadStopWord() {
    stopWordString = stopwords.join('\n')
    download(stopWordString, 'stopwords.txt', 'text/plain')
}


function hidePopup() {
    $('#infoWindowLayer').toggleClass('hidden')
    $('#progressInfo').html('')
    $('#progBarInner').css('width', 0 + '%')
    closeEventListner()
}

function setDate(_startDate, _endDate) {
    document.getElementById('startDate').value = _startDate
    document.getElementById("endDate").value = _endDate
    startDate = _startDate
    endDate = _endDate
}

function getProgressing(event) {
    data = JSON.parse(event.data)
    $('#progressInfo').html(data.comment)
    $('#progBarInner').css('width', data.progress + '%')
}

function getProgressFinished(event) {
    data = JSON.parse(event.data)
    changeGraph(data)
    hidePopup()
}

function closeEventListner() {
    progListener.removeEventListener('progressing' + randId, getProgressing)
    progListener.removeEventListener('progressFinished' + randId, getProgressFinished)
}

function sendRequest() {
    if ($('#idBox').val() == '' && $('#titleBox').val() == '') {
        window.alert('請至少填寫一個鄉民id或是')
    }
    content = JSON.stringify({
        author: $('#idBox').val(),
        aid: $('#titleBox').val(),
        keyword: $('#keywordBox').val(),
        stopwords: stopwords,
        pos: {
            noun: $('#noun').is(':checked'),
            verb: $('#verb').is(':checked'),
            adj: $('#adj').is(':checked'),
            adv: $('#adv').is(':checked'),
            pron: $('#pron').is(':checked'),
            aux: $('#aux').is(':checked'),
            other: $('#other').is(':checked')
        }
    })
    console.log(content)
    $.ajax({
        type: 'POST',
        url: 'ptt_push/addRequest',
        data: content,
        contentType: 'application/json',
        success: function(data) {
            console.log(data)
            tsvString = data.Result.tsv
            wordPushList = JSON.parse(data.Result.json)
            console.log(wordPushList)
            changeGraph(data.Result)
        }
    })
}

function changeGraph(data) {
    console.log(data)
    tsvString = data.tsv
    let json = JSON.parse(data.json)
    destroyCurrentGraph()
    d3.select('#graph').append('div').attr('id', 'vis')
    buildSentetree(tsvString)
}

function destroyCurrentGraph() {
    d3.selectAll('#vis').remove()
}

function hideTitles() {
    $('#titleListLayer').addClass('hidden')
}

function buildSentetree(tsvString) {
    console.log("Build.")
    var model;
    var tree;
    var data;
    if (typeof tsvString === 'undefined') {
        d3.tsv(tsvPath, buildTree)
    } else {
        data = d3.tsvParse(tsvString)
        buildTree(_, data)
    }

    function buildTree(error, rawdata) {
        const data = rawdata.map(d => Object.assign({}, d, { count: +d.count }));
        model = new SentenTree.SentenTreeBuilder()
            .tokenize(SentenTree.tokenizer.tokenizeBySpace)
            .transformToken(token => (/score(d|s)?/.test(token) ? 'score' : token))
            .buildModel(data, {
                maxSupportRatio: 0.8,
                minSupportRatio: 0.001
            });
        tree = new SentenTree.SentenTreeVis('#vis', {
            fontSize: [15, 40],
            gapBetweenGraph: 10
        });
        tree.data(model.getRenderedGraphs(5))
            .on('nodeClick', node => {
                $("#keywordBox").val(node.data.entity)
                $('#titleListLayer').removeClass('hidden')
                seqList = node.data.seq.DBs.map(function(n) {
                    return n.rawText
                })
                seqList = seqList.filter(function(v, i) {
                    return seqList.indexOf(v) == i
                })
                titleList = []
                console.log(seqList)
                for (s of seqList) {
                    titleTemp = wordPushList.filter(function(n) {
                        return n.part == s
                    })
                    titleList = titleList.concat(titleTemp)
                }
                console.log(titleList)
                info = wordPushList[node.data.entity]
                $('#titleListKeyword').html(node.data.entity)
                $('#titleListKeywordInfo').html('')

                $('#titleListContainer').empty()
                for (i of titleList) {
                    let link = $('<a>').append(
                        $('<h4>').html(i.title)
                    )
                    for (p of i.pushes) {
                        link.append(
                            $('<span>').attr('style', 'margin: 0px 10px').html((['推', '噓', '→'])[p.type - 1] + ' ' + p.author + ': ' + p.content + '<br>')
                        )
                    }
                    $('#titleListContainer').append(
                        $('<li>').attr('class', 'w3-panel').append(
                            link
                        )
                    )
                }
            })
            .on('nodeMouseenter', node => {
                console.log(node)
                let titles = []
                node.data.topEntries.forEach(function(x) {
                    console.log(x)
                    let result = wordPushList.filter(function(y) {
                        return y.part == x.rawText
                    })
                    for (r of result) {
                        if (titles.indexOf(r) < 0 && titles.length < 5) {
                            titles.push(r)
                        }
                    }
                })
                console.log(titles)
                infoStr = ''
                for (index in titles) {
                    if (index == 0) {
                        infoStr += titles[index].title + '<br>'
                    } else {
                        if (titles[index].title != titles[index - 1].title) {
                            infoStr += titles[index].title + '<br>'
                        }
                    }
                    pos = titles[index].part.indexOf(node.data.entity)
                    infoStr += titles[index].pushes.filter(function(x) {
                        return x.content.includes(node.data.entity)
                    })[0].content + '<br>'
                }
                $(nodeTitleContent).html(infoStr)
                $('#nodeTitle').removeClass('hidden')
                tree.highlightNeighbors(node)
            })
            .on('nodeMouseleave', node => {
                $('#nodeTitle').addClass('hidden')
                tree.clearHighlightNeighbors()
            }).on('layoutStart', layout => {
                console.log(layout)
            }).on('linkMouseenter', link => {
                console.log(link)
            })
        new ResizeSensor(jQuery('#d3kitRoot'), function() {
            var scale, origin;
            scale = Math.min(2, ($('#graph').outerWidth()) / ($('#d3kitRoot').outerWidth() + 60))

            $('#vis').css({
                transform: "scale(" + scale + ")",
                'transform-origin': 'top left'
            });
        })
    }
}