import copy

import numpy as np

from ding.envs.common import EnvElementRunner
from ding.envs.env.base_env import BaseEnv
from .gfootball_action import GfootballRawAction


class GfootballRawActionRunner(EnvElementRunner):

    def _init(self, cfg, *args, **kwargs) -> None:
        # set self._core and other state variable
        self._core = GfootballRawAction(cfg)

    def get(self, engine: BaseEnv) -> np.array:
        agent_action = copy.deepcopy(engine.agent_action)
        return agent_action

    def reset(self) -> None:
        pass