lost-sia/src/InfoBoxes/AnnoStats.jsx

Single Image Annotation Tool

import React, { useState, useEffect } from "react";
import { List, Label } from "semantic-ui-react";
import InfoBox from "./InfoBox";
import * as colorlut from "../utils/color";
// import actions from '../../../../actions'
// import * as transform from '../utils/transform'
// const { siaShowImgBar } = actions

const AnnoStats = (props) => {
  const [hideList, setHideList] = useState([]);

  useEffect(() => {
    setHideList([]);
  }, [props.imgLoadCount]);

  const onDismiss = () => {
    if (props.onDismiss) {
      props.onDismiss();
    }
  };

  const onLblClick = (lbl) => {
    let hideLbl = false;
    if (hideList.includes(lbl.id)) {
      setHideList(hideList.filter((e) => e !== lbl.id));
      hideLbl = false;
    } else {
      // hideList.push(lbl.id)
      setHideList([...hideList, lbl.id]);
      hideLbl = true;
    }
    if (props.onHideLbl) {
      props.onHideLbl(lbl, hideLbl);
    }
  };

  const renderRow = (s) => {
    const opacity = hideList.includes(s.id) ? 0.5 : 1.0;
    return (
      <List.Item key={s.id}>
        <List.Content>
          <Label
            as="a"
            tag
            style={{ background: s.color, opacity: opacity }}
            onClick={() => onLblClick(s)}
          >
            {s.label}
            <Label.Detail>{s.amount}</Label.Detail>
          </Label>
        </List.Content>
      </List.Item>
    );
  };

  const renderRows = () => {
    let stats = {};
    props.annos.forEach((a) => {
      // // console.log('render rows', a)
      if (a.status !== "deleted") {
        a.labelIds.forEach((lblId) => {
          if (lblId in stats) {
            stats[lblId] += 1;
          } else {
            stats[lblId] = 1;
          }
        });
        if (a.labelIds.length === 0) {
          if ((-1) in stats) {
            stats[-1] += 1;
          } else {
            stats[-1] = 1;
          }
        }
      }
    });
    const res = Object.entries(stats).map(([key, value]) => {
      let lbl = props.possibleLabels.find((e) => {
        return e.id === parseInt(key);
      });
      if (!lbl) {
        lbl = { id: -1, label: "No Label", color: colorlut.getDefaultColor() };
      }

      lbl.amount = value;
      // return renderRow({class:idToLbl[key], amount:value, color:idToColor[key]})
      return renderRow(lbl);
    });
    return res;
  };
  const renderDescription = () => {
    return <List>{renderRows()}</List>;
  };

  return (
    <InfoBox
      header={"Annotations per Label"}
      content={renderDescription()}
      visible={props.visible}
      defaultPos={props.defaultPos}
      onDismiss={(e) => onDismiss()}
    />
  );
};

export default AnnoStats;