from typing import Any, Dict, List, Optional

from .base import ProcessorMixin
from .canny import CannyProcessor
from .clip import CLIPPooledProcessor
from .glm import CogView4GLMProcessor
from .llama import LlamaProcessor
from .t5 import T5Processor
from .text import CaptionEmbeddingDropoutProcessor, CaptionTextDropoutProcessor


class CopyProcessor(ProcessorMixin):
    r"""Processor that copies the input data unconditionally to the output."""

    def __init__(self, output_names: List[str] = None, input_names: Optional[Dict[str, Any]] = None):
        super().__init__()

        self.output_names = output_names
        self.input_names = input_names
        assert len(output_names) == 1

    def forward(self, input: Any) -> Any:
        return {self.output_names[0]: input}