Make a tooltip to restrict users from log scales

This commit is contained in:
Sam Nolan 2022-05-02 20:40:15 +00:00
parent cc70047904
commit 7e05d63009

View File

@ -1,7 +1,7 @@
import * as React from "react"; import * as React from "react";
import _ from "lodash"; import _ from "lodash";
import type { Distribution } from "@quri/squiggle-lang"; import type { Distribution } from "@quri/squiggle-lang";
import { distributionErrorToString, result, shape } from "@quri/squiggle-lang"; import { distributionErrorToString } from "@quri/squiggle-lang";
import { Vega, VisualizationSpec } from "react-vega"; import { Vega, VisualizationSpec } from "react-vega";
import * as chartSpecification from "../vega-specs/spec-distributions.json"; import * as chartSpecification from "../vega-specs/spec-distributions.json";
import { ErrorBox } from "./ErrorBox"; import { ErrorBox } from "./ErrorBox";
@ -12,6 +12,7 @@ import {
linearYScale, linearYScale,
expYScale, expYScale,
} from "./DistributionVegaScales"; } from "./DistributionVegaScales";
import styled from "styled-components";
type DistributionChartProps = { type DistributionChartProps = {
distribution: Distribution; distribution: Distribution;
@ -26,28 +27,29 @@ export const DistributionChart: React.FC<DistributionChartProps> = ({
}: DistributionChartProps) => { }: DistributionChartProps) => {
let [isLogX, setLogX] = React.useState(false); let [isLogX, setLogX] = React.useState(false);
let [isExpY, setExpY] = React.useState(false); let [isExpY, setExpY] = React.useState(false);
let shape = distribution.pointSet();
const [sized, _] = useSize((size) => { const [sized, _] = useSize((size) => {
let shape = distribution.pointSet(); var disableLog = false;
if (shape.tag === "Ok") { if (shape.tag === "Ok") {
let spec = buildSpec(isLogX, isExpY, shape.value); let massBelow0 =
if (spec.tag == "Ok") { shape.value.continuous.some((x) => x.x <= 0) ||
let widthProp = width ? width - 20 : size.width - 10; shape.value.discrete.some((x) => x.x <= 0);
var result = ( if (massBelow0) {
<div> disableLog = true;
<Vega
spec={spec.value}
data={{ con: shape.value.continuous, dis: shape.value.discrete }}
width={widthProp}
height={height}
actions={false}
/>
</div>
);
} else {
var result = (
<ErrorBox heading={spec.value.heading}>{spec.value.error}</ErrorBox>
);
} }
let spec = buildVegaSpec(isLogX, isExpY);
let widthProp = width ? width - 20 : size.width - 10;
var result = (
<div>
<Vega
spec={spec}
data={{ con: shape.value.continuous, dis: shape.value.discrete }}
width={widthProp}
height={height}
actions={false}
/>
</div>
);
} else { } else {
var result = ( var result = (
<ErrorBox heading="Distribution Error"> <ErrorBox heading="Distribution Error">
@ -59,7 +61,19 @@ export const DistributionChart: React.FC<DistributionChartProps> = ({
<> <>
{result} {result}
<div> <div>
<CheckBox label="Log X scale" value={isLogX} onChange={setLogX} /> {disableLog ? (
<CheckBox
label="Log X scale"
value={isLogX}
onChange={setLogX}
disabled={true}
tooltip={
"Your distribution has mass lower than or equal to 0, cannot use log scale"
}
/>
) : (
<CheckBox label="Log X scale" value={isLogX} onChange={setLogX} />
)}
<CheckBox label="Exp Y scale" value={isExpY} onChange={setExpY} /> <CheckBox label="Exp Y scale" value={isExpY} onChange={setExpY} />
</div> </div>
</> </>
@ -68,44 +82,44 @@ export const DistributionChart: React.FC<DistributionChartProps> = ({
return sized; return sized;
}; };
type ViewError = { heading: string; error: string }; function buildVegaSpec(isLogX: boolean, isExpY: boolean): VisualizationSpec {
return {
function buildSpec( ...chartSpecification,
isLogX: boolean, scales: [
isExpY: boolean, isLogX ? logXScale : linearXScale,
shape: shape isExpY ? expYScale : linearYScale,
): result<VisualizationSpec, ViewError> { ],
let someBelow0 = } as VisualizationSpec;
shape.continuous.some((x) => x.x <= 0) ||
shape.discrete.some((x) => x.x <= 0);
if (!(isLogX && someBelow0)) {
return {
tag: "Ok",
value: {
...chartSpecification,
scales: [
isLogX ? logXScale : linearXScale,
isExpY ? expYScale : linearYScale,
],
} as VisualizationSpec,
};
} else {
return {
tag: "Error",
value: {
heading: "Log Viewing error",
error:
"Distribution contains values lower than or equal to 0. Cannot view",
},
};
}
} }
export const CheckBox = ({ label, onChange, value }) => { interface CheckBoxProps {
label: string;
onChange: (x: boolean) => void;
value: boolean;
disabled?: boolean;
tooltip?: string;
}
const Label = styled.label<{ disabled: boolean }>`
${(props) => props.disabled && "color: #999;"}
`;
export const CheckBox = ({
label,
onChange,
value,
disabled = false,
tooltip,
}: CheckBoxProps) => {
return ( return (
<span> <span title={tooltip}>
<input type="checkbox" value={value} onChange={() => onChange(!value)} /> <input
<label>{label}</label> type="checkbox"
value={value + ""}
onChange={() => onChange(!value)}
disabled={disabled}
/>
<Label disabled={disabled}>{label}</Label>
</span> </span>
); );
}; };