sam2/demo/frontend/src/common/components/effects/HighlightEffects.tsx

/**
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
import EffectVariantBadge from '@/common/components/effects/EffectVariantBadge';
import ToolbarActionIcon from '@/common/components/toolbar/ToolbarActionIcon';
import ToolbarSection from '@/common/components/toolbar/ToolbarSection';
import useVideoEffect from '@/common/components/video/editor/useVideoEffect';
import {EffectIndex} from '@/common/components/video/effects/Effects';
import {
  activeHighlightEffectAtom,
  activeHighlightEffectGroupAtom,
} from '@/demo/atoms';
import {useAtomValue} from 'jotai';

export default function HighlightEffects() {
  const setEffect = useVideoEffect();
  const activeEffect = useAtomValue(activeHighlightEffectAtom);
  const activeEffectsGroup = useAtomValue(activeHighlightEffectGroupAtom);

  return (
    <ToolbarSection title="Selected Objects" borderBottom={true}>
      {activeEffectsGroup.map(highlightEffect => {
        return (
          <ToolbarActionIcon
            variant="toggle"
            key={highlightEffect.title}
            icon={highlightEffect.Icon}
            title={highlightEffect.title}
            isActive={activeEffect.name === highlightEffect.effectName}
            badge={
              activeEffect.name === highlightEffect.effectName && (
                <EffectVariantBadge
                  label={`${activeEffect.variant + 1}/${activeEffect.numVariants}`}
                />
              )
            }
            onClick={() => {
              if (activeEffect.name === highlightEffect.effectName) {
                setEffect(highlightEffect.effectName, EffectIndex.HIGHLIGHT, {
                  variant:
                    (activeEffect.variant + 1) % activeEffect.numVariants,
                });
              } else {
                setEffect(highlightEffect.effectName, EffectIndex.HIGHLIGHT);
              }
            }}
          />
        );
      })}
    </ToolbarSection>
  );
}