112 lines
3.6 KiB
TypeScript
112 lines
3.6 KiB
TypeScript
import { useMemo, useRef } from 'react'
|
|
import { last, sortBy } from 'lodash'
|
|
import { scaleTime, scaleLog, scaleLinear } from 'd3-scale'
|
|
|
|
import { Bet } from 'common/bet'
|
|
import { DAY_MS } from 'common/util/time'
|
|
import { getInitialProbability, getProbability } from 'common/calculate'
|
|
import { formatLargeNumber } from 'common/util/format'
|
|
import { PseudoNumericContract } from 'common/contract'
|
|
import { NUMERIC_GRAPH_COLOR } from 'common/numeric-constants'
|
|
import { useIsMobile } from 'web/hooks/use-is-mobile'
|
|
import {
|
|
TooltipProps,
|
|
MARGIN_X,
|
|
MARGIN_Y,
|
|
getDateRange,
|
|
getRightmostVisibleDate,
|
|
formatDateInRange,
|
|
} from '../helpers'
|
|
import { HistoryPoint, SingleValueHistoryChart } from '../generic-charts'
|
|
import { useElementWidth } from 'web/hooks/use-element-width'
|
|
import { Row } from 'web/components/layout/row'
|
|
import { Avatar } from 'web/components/avatar'
|
|
|
|
// mqp: note that we have an idiosyncratic version of 'log scale'
|
|
// contracts. the values are stored "linearly" and can include zero.
|
|
// as a result, we have to do some weird-looking stuff in this code
|
|
|
|
const getScaleP = (min: number, max: number, isLogScale: boolean) => {
|
|
return (p: number) =>
|
|
isLogScale
|
|
? 10 ** (p * Math.log10(max - min + 1)) + min - 1
|
|
: p * (max - min) + min
|
|
}
|
|
|
|
const getBetPoints = (bets: Bet[], scaleP: (p: number) => number) => {
|
|
return sortBy(bets, (b) => b.createdTime).map((b) => ({
|
|
x: new Date(b.createdTime),
|
|
y: scaleP(b.probAfter),
|
|
datum: b,
|
|
}))
|
|
}
|
|
|
|
const PseudoNumericChartTooltip = (props: TooltipProps<HistoryPoint<Bet>>) => {
|
|
const { p, xScale } = props
|
|
const { x, y, datum } = p
|
|
const [start, end] = xScale.domain()
|
|
return (
|
|
<Row className="items-center gap-2 text-sm">
|
|
{datum && <Avatar size="xs" avatarUrl={datum.userAvatarUrl} />}
|
|
<strong>{formatLargeNumber(y)}</strong>
|
|
<span>{formatDateInRange(x, start, end)}</span>
|
|
</Row>
|
|
)
|
|
}
|
|
|
|
export const PseudoNumericContractChart = (props: {
|
|
contract: PseudoNumericContract
|
|
bets: Bet[]
|
|
height?: number
|
|
}) => {
|
|
const { contract, bets } = props
|
|
const { min, max, isLogScale } = contract
|
|
const [startDate, endDate] = getDateRange(contract)
|
|
const scaleP = useMemo(
|
|
() => getScaleP(min, max, isLogScale),
|
|
[min, max, isLogScale]
|
|
)
|
|
const startP = scaleP(getInitialProbability(contract))
|
|
const endP = scaleP(getProbability(contract))
|
|
const betPoints = useMemo(() => getBetPoints(bets, scaleP), [bets, scaleP])
|
|
const data = useMemo(
|
|
() => [
|
|
{ x: startDate, y: startP },
|
|
...betPoints,
|
|
{ x: endDate ?? new Date(Date.now() + DAY_MS), y: endP },
|
|
],
|
|
[betPoints, startDate, startP, endDate, endP]
|
|
)
|
|
const rightmostDate = getRightmostVisibleDate(
|
|
endDate,
|
|
last(betPoints)?.x,
|
|
new Date(Date.now())
|
|
)
|
|
const visibleRange = [startDate, rightmostDate]
|
|
const isMobile = useIsMobile(800)
|
|
const containerRef = useRef<HTMLDivElement>(null)
|
|
const width = useElementWidth(containerRef) ?? 0
|
|
const height = props.height ?? (isMobile ? 150 : 250)
|
|
const xScale = scaleTime(visibleRange, [0, width - MARGIN_X])
|
|
// clamp log scale to make sure zeroes go to the bottom
|
|
const yScale = isLogScale
|
|
? scaleLog([Math.max(min, 1), max], [height - MARGIN_Y, 0]).clamp(true)
|
|
: scaleLinear([min, max], [height - MARGIN_Y, 0])
|
|
|
|
return (
|
|
<div ref={containerRef}>
|
|
{width > 0 && (
|
|
<SingleValueHistoryChart
|
|
w={width}
|
|
h={height}
|
|
xScale={xScale}
|
|
yScale={yScale}
|
|
data={data}
|
|
Tooltip={PseudoNumericChartTooltip}
|
|
color={NUMERIC_GRAPH_COLOR}
|
|
/>
|
|
)}
|
|
</div>
|
|
)
|
|
}
|