InvokeAI/frontend/src/features/canvas/IAICanvasBrushButtonPopover.tsx

import { createSelector } from '@reduxjs/toolkit';
import {
  currentCanvasSelector,
  outpaintingCanvasSelector,
  setBrushColor,
  setBrushSize,
  setTool,
} from './canvasSlice';
import { useAppDispatch, useAppSelector } from 'app/store';
import { activeTabNameSelector } from 'features/options/optionsSelectors';
import _ from 'lodash';
import IAIIconButton from 'common/components/IAIIconButton';
import { FaPaintBrush } from 'react-icons/fa';
import IAIPopover from 'common/components/IAIPopover';
import IAIColorPicker from 'common/components/IAIColorPicker';
import IAISlider from 'common/components/IAISlider';
import { Flex } from '@chakra-ui/react';
import IAINumberInput from 'common/components/IAINumberInput';

export const selector = createSelector(
  [currentCanvasSelector, outpaintingCanvasSelector, activeTabNameSelector],
  (currentCanvas, outpaintingCanvas, activeTabName) => {
    const {
      layer,
      maskColor,
      brushColor,
      brushSize,
      eraserSize,
      tool,
      shouldDarkenOutsideBoundingBox,
      shouldShowIntermediates,
    } = currentCanvas;

    const { shouldShowGrid, shouldSnapToGrid, shouldAutoSave } =
      outpaintingCanvas;

    return {
      layer,
      tool,
      maskColor,
      brushColor,
      brushSize,
      eraserSize,
      activeTabName,
      shouldShowGrid,
      shouldSnapToGrid,
      shouldAutoSave,
      shouldDarkenOutsideBoundingBox,
      shouldShowIntermediates,
    };
  },
  {
    memoizeOptions: {
      resultEqualityCheck: _.isEqual,
    },
  }
);

const IAICanvasBrushButtonPopover = () => {
  const dispatch = useAppDispatch();
  const { tool, brushColor, brushSize } = useAppSelector(selector);

  return (
    <IAIPopover
      trigger="hover"
      triggerComponent={
        <IAIIconButton
          aria-label="Brush (B)"
          tooltip="Brush (B)"
          icon={<FaPaintBrush />}
          data-selected={tool === 'brush'}
          onClick={() => dispatch(setTool('brush'))}
        />
      }
    >
      <Flex minWidth={'15rem'} direction={'column'} gap={'1rem'} width={'100%'}>
        <Flex gap={'1rem'} justifyContent="space-between">
          <IAISlider
            label="Size"
            value={brushSize}
            withInput
            onChange={(newSize) => dispatch(setBrushSize(newSize))}
          />
        </Flex>
        <IAIColorPicker
          style={{ width: '100%' }}
          color={brushColor}
          onChange={(newColor) => dispatch(setBrushColor(newColor))}
        />
      </Flex>
    </IAIPopover>
  );
};

export default IAICanvasBrushButtonPopover;