anki/ts/graphs/ease.ts

77 lines
2.4 KiB
TypeScript
Raw Normal View History

2020-06-23 12:43:19 +02:00
// Copyright: Ankitects Pty Ltd and contributors
// License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
/* eslint
@typescript-eslint/no-non-null-assertion: "off",
@typescript-eslint/no-explicit-any: "off",
*/
import type pb from "anki/backend_proto";
2020-08-21 04:58:02 +02:00
import { extent, histogram, sum } from "d3-array";
2020-06-23 12:43:19 +02:00
import { scaleLinear, scaleSequential } from "d3-scale";
import { CardType } from "anki/cards";
import type { HistogramData } from "./histogram-graph";
2020-06-23 12:43:19 +02:00
import { interpolateRdYlGn } from "d3-scale-chromatic";
import type { I18n } from "anki/i18n";
import type { TableDatum } from "./graph-helpers";
2020-06-23 12:43:19 +02:00
export interface GraphData {
eases: number[];
}
export function gatherData(data: pb.BackendProto.GraphsOut): GraphData {
2020-06-24 01:41:07 +02:00
const eases = (data.cards as pb.BackendProto.Card[])
.filter((c) => [CardType.Review, CardType.Relearn].includes(c.ctype))
.map((c) => c.easeFactor / 10);
2020-06-23 12:43:19 +02:00
return { eases };
}
2020-08-21 04:58:02 +02:00
export function prepareData(
data: GraphData,
i18n: I18n
): [HistogramData | null, TableDatum[]] {
2020-06-23 12:43:19 +02:00
// get min/max
const allEases = data.eases;
if (!allEases.length) {
2020-08-21 04:58:02 +02:00
return [null, []];
2020-06-23 12:43:19 +02:00
}
const total = allEases.length;
const [_xMin, origXMax] = extent(allEases);
let xMax = origXMax;
const xMin = 130;
xMax = xMax! + 1;
const desiredBars = 20;
const scale = scaleLinear().domain([130, xMax!]).nice();
const bins = histogram()
.domain(scale.domain() as any)
.thresholds(scale.ticks(desiredBars))(allEases);
const colourScale = scaleSequential(interpolateRdYlGn).domain([xMin, 300]);
2020-06-28 12:52:38 +02:00
function hoverText(data: HistogramData, binIdx: number, _percent: number): string {
const bin = data.bins[binIdx];
const minPct = Math.floor(bin.x0!);
const maxPct = Math.floor(bin.x1!);
const percent = maxPct - minPct <= 10 ? `${bin.x0}%` : `${bin.x0}%-${bin.x1}%`;
return i18n.tr(i18n.TR.STATISTICS_CARD_EASE_TOOLTIP, {
cards: bin.length,
percent,
});
}
2020-08-21 04:58:02 +02:00
const xTickFormat = (num: number): string => `${num.toFixed(0)}%`;
const tableData = [
{
label: i18n.tr(i18n.TR.STATISTICS_AVERAGE_EASE),
value: xTickFormat(sum(allEases) / total),
},
];
return [
{ scale, bins, total, hoverText, colourScale, showArea: false, xTickFormat },
tableData,
];
2020-06-23 12:43:19 +02:00
}