xquery version "3.0";

module namespace display="http://divvun.no/iskkan/display";

import module namespace templates="http://exist-db.org/xquery/templates" ;
import module namespace config="http://divvun.no/iskkan/config" at "config.xqm";

declare namespace output="http://www.w3.org/2010/xslt-xquery-serialization";

declare function display:edit-dist-table($tp as xs:int*, $el as element()) {
    let $AllEdit1Place := sum($el/sugg1/@editdist1 + $el/sugg1/@editdist2 + $el/sugg1/@editdist3plus),
        $AllEdit1PlacePerc := round-half-to-even((($AllEdit1Place           div $tp) * 100), 2),
        $OneEdit1PlacePerc := round-half-to-even((($el/sugg1/@editdist1     div $tp) * 100), 2),
        $TwoEdit1PlacePerc := round-half-to-even((($el/sugg1/@editdist2     div $tp) * 100), 2),
        $ThrEdit1PlacePerc := round-half-to-even((($el/sugg1/@editdist3plus div $tp) * 100), 2),
        $AllEdit2Place := sum($el/sugg2/@editdist1 + $el/sugg2/@editdist2 + $el/sugg2/@editdist3plus),
        $AllEdit2PlacePerc := round-half-to-even((($AllEdit2Place           div $tp) * 100), 2),
        $OneEdit2PlacePerc := round-half-to-even((($el/sugg2/@editdist1     div $tp) * 100), 2),
        $TwoEdit2PlacePerc := round-half-to-even((($el/sugg2/@editdist2     div $tp) * 100), 2),
        $ThrEdit2PlacePerc := round-half-to-even((($el/sugg2/@editdist3plus div $tp) * 100), 2),
        $AllEdit3Place := sum($el/sugg3/@editdist1 + $el/sugg3/@editdist2 + $el/sugg3/@editdist3plus),
        $AllEdit3PlacePerc := round-half-to-even((($AllEdit3Place           div $tp) * 100), 2),
        $OneEdit3PlacePerc := round-half-to-even((($el/sugg3/@editdist1     div $tp) * 100), 2),
        $TwoEdit3PlacePerc := round-half-to-even((($el/sugg3/@editdist2     div $tp) * 100), 2),
        $ThrEdit3PlacePerc := round-half-to-even((($el/sugg3/@editdist3plus div $tp) * 100), 2),
        $AllEdit4Place := sum($el/sugg4/@editdist1 + $el/sugg4/@editdist2 + $el/sugg4/@editdist3plus),
        $AllEdit4PlacePerc := round-half-to-even((($AllEdit4Place           div $tp) * 100), 2),
        $OneEdit4PlacePerc := round-half-to-even((($el/sugg4/@editdist1     div $tp) * 100), 2),
        $TwoEdit4PlacePerc := round-half-to-even((($el/sugg4/@editdist2     div $tp) * 100), 2),
        $ThrEdit4PlacePerc := round-half-to-even((($el/sugg4/@editdist3plus div $tp) * 100), 2),
        $AllEdit5Place := sum($el/sugg5/@editdist1 + $el/sugg5/@editdist2 + $el/sugg5/@editdist3plus),
        $AllEdit5PlacePerc := round-half-to-even((($AllEdit5Place           div $tp) * 100), 2),
        $OneEdit5PlacePerc := round-half-to-even((($el/sugg5/@editdist1     div $tp) * 100), 2),
        $TwoEdit5PlacePerc := round-half-to-even((($el/sugg5/@editdist2     div $tp) * 100), 2),
        $ThrEdit5PlacePerc := round-half-to-even((($el/sugg5/@editdist3plus div $tp) * 100), 2),
        $AllEdit6Place := sum($el/suggbelow5/@editdist1 + $el/suggbelow5/@editdist2 + $el/suggbelow5/@editdist3plus),
        $AllEdit6PlacePerc := round-half-to-even((($AllEdit6Place                div $tp) * 100), 2),
        $OneEdit6PlacePerc := round-half-to-even((($el/suggbelow5/@editdist1     div $tp) * 100), 2),
        $TwoEdit6PlacePerc := round-half-to-even((($el/suggbelow5/@editdist2     div $tp) * 100), 2),
        $ThrEdit6PlacePerc := round-half-to-even((($el/suggbelow5/@editdist3plus div $tp) * 100), 2),
        $AllEdit0Place := sum($el/nosugg/@editdist1 + $el/nosugg/@editdist2 + $el/nosugg/@editdist3plus),
        $AllEdit0PlacePerc := round-half-to-even((($AllEdit0Place            div $tp) * 100), 2),
        $OneEdit0PlacePerc := round-half-to-even((($el/nosugg/@editdist1     div $tp) * 100), 2),
        $TwoEdit0PlacePerc := round-half-to-even((($el/nosugg/@editdist2     div $tp) * 100), 2),
        $ThrEdit0PlacePerc := round-half-to-even((($el/nosugg/@editdist3plus div $tp) * 100), 2),
        $AllEditXPlace := sum($el/badsuggs/@editdist1 + $el/badsuggs/@editdist2 + $el/badsuggs/@editdist3plus),
        $AllEditXPlacePerc := round-half-to-even((($AllEditXPlace              div $tp) * 100), 2),
        $OneEditXPlacePerc := round-half-to-even((($el/badsuggs/@editdist1     div $tp) * 100), 2),
        $TwoEditXPlacePerc := round-half-to-even((($el/badsuggs/@editdist2     div $tp) * 100), 2),
        $ThrEditXPlacePerc := round-half-to-even((($el/badsuggs/@editdist3plus div $tp) * 100), 2)
    return (
    <div class="suggestionsummary">
        <h4>Suggestion summary</h4>

    <table>
    <colgroup>
        <col/><col/><col/><col/>
    </colgroup>
    <tbody>
        <tr>
<th>Correct suggestion position for true positives, %:</th>
<th class="number">All edit<br/>dists</th>
<th>Edit<br/>dist=1</th>
<th>Edit<br/>dist=2</th>
<th>Edit<br/>dist≥3</th>
</tr>
<!--tr>
<th colspan="2">Suggestion statistics for true positives (=
                4 018 = 100%):</th><td>8 658
                (77,44%)</td><td>1 848
                (16,53%)</td><td>672
                (6,01%)</td>
</tr-->
<tr>
<td>First place:</td>
<td class="number">{$AllEdit1PlacePerc}</td>
<td class="number">{$OneEdit1PlacePerc}</td>
<td class="number">{$TwoEdit1PlacePerc}</td>
<td class="number">{$ThrEdit1PlacePerc}</td>
</tr>
<tr>
<td>Second place:</td>
<td class="number">{$AllEdit2PlacePerc}</td>
<td class="number">{$OneEdit2PlacePerc}</td>
<td class="number">{$TwoEdit2PlacePerc}</td>
<td class="number">{$ThrEdit2PlacePerc}</td>
</tr>
<tr>
<td>Third place:</td>
<td class="number">{$AllEdit3PlacePerc}</td>
<td class="number">{$OneEdit3PlacePerc}</td>
<td class="number">{$TwoEdit3PlacePerc}</td>
<td class="number">{$ThrEdit3PlacePerc}</td>
</tr>
<tr>
<td>Fourth place:</td>
<td class="number">{$AllEdit4PlacePerc}</td>
<td class="number">{$OneEdit4PlacePerc}</td>
<td class="number">{$TwoEdit4PlacePerc}</td>
<td class="number">{$ThrEdit4PlacePerc}</td>
</tr>
<tr>
<td>Fifth place:</td>
<td class="number">{$AllEdit5PlacePerc}</td>
<td class="number">{$OneEdit5PlacePerc}</td>
<td class="number">{$TwoEdit5PlacePerc}</td>
<td class="number">{$ThrEdit5PlacePerc}</td>
</tr>
<tr>
<td><strong>Sum:</strong> spelling errors with <span class="correct">correct suggestion</span> in top 5:</td>
<td class="number">{round-half-to-even((sum(
                        $AllEdit1PlacePerc +
                        $AllEdit2PlacePerc +
                        $AllEdit3PlacePerc +
                        $AllEdit4PlacePerc +
                        $AllEdit5PlacePerc) * 100), 2) div 100}</td>
<td class="number">{round-half-to-even((sum(
                        $OneEdit1PlacePerc +
                        $OneEdit2PlacePerc +
                        $OneEdit3PlacePerc +
                        $OneEdit4PlacePerc +
                        $OneEdit5PlacePerc) * 100), 2) div 100}</td>
<td class="number">{round-half-to-even((sum(
                        $TwoEdit1PlacePerc +
                        $TwoEdit2PlacePerc +
                        $TwoEdit3PlacePerc +
                        $TwoEdit4PlacePerc +
                        $TwoEdit5PlacePerc) * 100), 2) div 100}</td>
<td class="number">{round-half-to-even((sum(
                        $ThrEdit1PlacePerc +
                        $ThrEdit2PlacePerc +
                        $ThrEdit3PlacePerc +
                        $ThrEdit4PlacePerc +
                        $ThrEdit5PlacePerc) * 100), 2) div 100}</td>
</tr>
<tr>
<td>Below top 5:</td>
<td class="number">{$AllEdit6PlacePerc}</td>
<td class="number">{$OneEdit6PlacePerc}</td>
<td class="number">{$TwoEdit6PlacePerc}</td>
<td class="number">{$ThrEdit6PlacePerc}</td>
</tr>
<tr>
<td>No suggestions:</td>
<td class="number">{$AllEdit0PlacePerc}</td>
<td class="number">{$OneEdit0PlacePerc}</td>
<td class="number">{$TwoEdit0PlacePerc}</td>
<td class="number">{$ThrEdit0PlacePerc}</td>
</tr>
<tr>
<td>Only wrong suggestions:</td>
<td class="number">{$AllEditXPlacePerc}</td>
<td class="number">{$OneEditXPlacePerc}</td>
<td class="number">{$TwoEditXPlacePerc}</td>
<td class="number">{$ThrEditXPlacePerc}</td>
</tr>
</tbody>
</table>
</div>
)
};

declare function display:table($invocation as map) {
    <table>
        <caption>Precision and recall</caption>
        <thead>
            <tr>
                <th>Reality</th>
                <th>Real no</th>
                <th>Positive</th>
                <th>Negative</th>
            </tr>
        </thead>
    </table>
};

declare function display:quality($el as element()) {
    <div class="suggestionsummary">
        <h4>Suggestion summary</h4>

        <table class="quality-table">
            <tr>
                <th>Suggestion<br/>Position</th>
                <th>Edit dist=1</th>
                <th>Edit dist=2</th>
                <th>Edit dist>=3</th>
            </tr> {
                for $p in 1 to 6
                return
                <tr>
                    <td data-th="Suggestion Position">{$p}</td>{
                      for $dist in 1 to 3
                      return
                      <td data-th="Edit Dist {$dist}">{$el/words[@position=$p][@edit_dist=$dist]}</td>
                }</tr>
            }
        </table>
        {
            transform:transform($el, doc("/db/apps/iskkan/resources/xslt/bar-chart-svg.xsl"), ()) 
        }
    </div>
};

declare function display:summary($invocation as map) {
    <div class="precrecallsummary">
        <h4>Precision and Recall</h4>

        <svg viewBox="0 0 900 100" width="900px" heigth="100px">
            <g id="pra-chart" font-size="16" transform="translate(0,0)">
                <g id="precision" transform="translate(0, 0)">
                    <text x="0" y="15">Precision (tp/(tp+fp))</text>
                    <rect x="200" y="0" width="{$invocation('precision')}" height="20" fill="lightgray"/>
                    <text x="{$invocation('precision') + 205}" y="15">{$invocation('precision')}%</text>
                </g>
                <g id="recall" transform="translate(0, 30)">
                    <text x="0" y="15">Recall (tp/(tp+fn))</text>
                    <rect x="200" y="0" width="{$invocation('recall')}" height="20" fill="lightgray"/>
                    <text x="{$invocation('recall') + 205}" y="15">{$invocation('recall')}%</text>
                </g>
                <g id="accuracy" transform="translate(0, 60)">
                    <text x="0" y="15">Accuracy ((tp+tn)/words)</text>
                    <rect x="200" y="0" width="{$invocation('accuracy')}" height="20" fill="lightgray"/>
                    <text x="{$invocation('accuracy') + 205}" y="15">{$invocation('accuracy')}%</text>
                </g>
            </g>
        </svg>

        <h4>Data for precision and recall</h4>

        <p>Nº of input words: <strong>{$invocation('words')}</strong></p>
        <table class="summary-table">
            
            <tbody>
                <tr>
                    <td colspan="2" rowspan="2"></td><th colspan="2">Speller view</th>
                </tr>
                
                <tr>
                    <td>Speller Positive<br/>
                              (Nº of flagged words):<br/>
                        <strong>{$invocation('speller')('error')}</strong>
                    </td>
                    <td>Speller Negative<br/>
                              (Nº of accepted words):<br/>
                        <strong>{$invocation('speller')('correct')}</strong>
                    </td>
                </tr>
                
                <tr>
                    <th rowspan="2">Reality</th>
                    <td>Nº of real errors:<br/>
                        <strong>{$invocation('original')('error')}</strong>
                    </td>
                    <td data-th="True Positive">Nº of true positives<br/>
                              (detected real errors):<br/>
                        <strong>{$invocation('positives')('true')}</strong>
                    </td>
                    <td data-th="False Negative">Nº of false negatives<br/>
                              (unflagged spelling errors):<br/>
                        <strong>{$invocation('negatives')('false')}</strong>
                    </td>
                </tr>
                
                <tr>
                    <td>Nº of real correct words:<br/>
                        <strong>{$invocation('original')('correct')}</strong>
                    </td>
                    <td data-th="False Positive">Nº of false positives<br/>
                              (incorrectly flagged words):<br/>
                        <strong>{$invocation('positives')('false')}</strong>
                    </td>
                    <td data-th="True Negative">Nº of true negatives<br/>
                              (unflagged correct words):<br/>
                        <strong>{$invocation('negatives')('true')}</strong>
                    </td>
                </tr>
            </tbody>
        </table>
    </div>
};