anki/ts/src/stats/card-counts.ts

152 lines
4.6 KiB
TypeScript
Raw Normal View History

// Copyright: Ankitects Pty Ltd and contributors
// License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
2020-07-04 05:38:46 +02:00
/* eslint
@typescript-eslint/no-non-null-assertion: "off",
@typescript-eslint/no-explicit-any: "off",
*/
import { CardQueue } from "../cards";
2020-07-04 05:38:46 +02:00
import pb from "../backend/proto";
import { schemeGreens, schemeBlues } from "d3-scale-chromatic";
import "d3-transition";
import { select, mouse } from "d3-selection";
import { scaleLinear } from "d3-scale";
import { showTooltip, hideTooltip } from "./tooltip";
import { GraphBounds } from "./graphs";
import { cumsum } from "d3-array";
2020-06-27 13:10:17 +02:00
import { I18n } from "../i18n";
2020-06-27 13:10:17 +02:00
type Count = [string, number];
2020-07-04 05:38:46 +02:00
export interface GraphData {
2020-06-27 13:10:17 +02:00
title: string;
counts: Count[];
2020-07-04 05:38:46 +02:00
totalCards: number;
}
2020-07-04 05:38:46 +02:00
export function gatherData(data: pb.BackendProto.GraphsOut, i18n: I18n): GraphData {
// fixme: handle preview cards
const totalCards = data.cards.length;
let newCards = 0;
let young = 0;
let mature = 0;
let suspended = 0;
let buried = 0;
for (const card of data.cards as pb.BackendProto.Card[]) {
switch (card.queue) {
case CardQueue.New:
newCards += 1;
break;
case CardQueue.Review:
if (card.ivl >= 21) {
mature += 1;
break;
}
// young falls through
case CardQueue.Learn:
case CardQueue.DayLearn:
young += 1;
break;
case CardQueue.Suspended:
suspended += 1;
break;
case CardQueue.SchedBuried:
case CardQueue.UserBuried:
buried += 1;
break;
}
}
2020-06-27 13:10:17 +02:00
const counts = [
[i18n.tr(i18n.TR.STATISTICS_COUNTS_NEW_CARDS), newCards] as Count,
[i18n.tr(i18n.TR.STATISTICS_COUNTS_YOUNG_CARDS), young] as Count,
[i18n.tr(i18n.TR.STATISTICS_COUNTS_MATURE_CARDS), mature] as Count,
[i18n.tr(i18n.TR.STATISTICS_COUNTS_SUSPENDED_CARDS), suspended] as Count,
[i18n.tr(i18n.TR.STATISTICS_COUNTS_BURIED_CARDS), buried] as Count,
];
return {
2020-06-27 13:10:17 +02:00
title: i18n.tr(i18n.TR.STATISTICS_COUNTS_TITLE),
counts,
2020-07-04 05:38:46 +02:00
totalCards,
};
}
interface Reviews {
mature: number;
young: number;
learn: number;
relearn: number;
early: number;
}
export function renderCards(
svgElem: SVGElement,
bounds: GraphBounds,
sourceData: GraphData
): void {
const summed = cumsum(sourceData.counts, (d) => d[1]);
const data = Array.from(summed).map((n, idx) => {
return {
count: sourceData.counts[idx],
idx,
total: n,
};
});
2020-07-06 06:01:49 +02:00
// ensuring a non-zero range makes a better animation
// in the empty data case
const xMax = Math.max(1, summed.slice(-1)[0]);
2020-07-04 05:38:46 +02:00
const x = scaleLinear().domain([0, xMax]);
const svg = select(svgElem);
const trans = svg.transition().duration(600) as any;
2020-07-06 06:01:49 +02:00
x.range([bounds.marginLeft, bounds.width - bounds.marginRight - bounds.marginLeft]);
2020-07-04 05:38:46 +02:00
const tooltipText = (d: any): string => {
const pct = ((d.count[1] / xMax) * 100).toFixed(2);
return `${d.count[0]}: ${d.count[1]} (${pct}%)`;
};
2020-07-04 05:38:46 +02:00
const updateBar = (sel: any): any => {
return sel
.on("mousemove", function (this: any, d: any) {
const [x, y] = mouse(document.body);
showTooltip(tooltipText(d), x, y);
})
.transition(trans)
2020-07-06 06:01:49 +02:00
.attr("width", (d) => x(d.total));
2020-07-04 05:38:46 +02:00
};
data.reverse();
svg.select("g.days")
.selectAll("rect")
.data(data)
.join(
(enter) =>
enter
.append("rect")
.attr("height", 10)
.attr("x", x(0))
.attr("y", bounds.marginTop)
.attr("fill", (d: any): any => {
switch (d.idx) {
case 0:
return schemeBlues[5][2];
case 1:
return schemeGreens[5][2];
case 2:
return schemeGreens[5][3];
case 3:
return "#FFDC41";
case 4:
return "grey";
}
})
.on("mouseout", hideTooltip)
.call((d) => updateBar(d)),
(update) => update.call((d) => updateBar(d))
);
}