# -*- coding: utf-8 -*-
# Author: ximing
# Description: misc
# Copyright (c) 2023, XiMing Xing.
# License: MPL-2.0 License

from datetime import datetime
import random
import pathlib
from typing import Any, List, Dict, Union

import omegaconf

"""Add Type"""
AnyPath = Union[str, pathlib.Path, 'os.PathLike']
AnyList = Union[omegaconf.ListConfig, List]
AnyDict = Union[omegaconf.DictConfig, Dict]


def render_batch_wrap(cfg: omegaconf.DictConfig,
                      seed_range: List,
                      pipeline: Any,
                      **pipe_args):
    start_time = datetime.now()
    for idx, seed in enumerate(seed_range):
        cfg.seed = seed  # update seed
        print(f"\n-> [{idx}/{len(seed_range)}], "
              f"current seed: {seed}, "
              f"current time: {datetime.now() - start_time}\n")
        pipe = pipeline(cfg)
        pipe.painterly_rendering(**pipe_args)


def get_seed_range(srange: AnyList):
    # random sampling without specifying a range
    start_, end_ = 1, 1000000
    if srange is not None:  # specify range sequential sampling
        seed_range_ = list(srange)
        assert len(seed_range_) == 2 and int(seed_range_[1]) > int(seed_range_[0])
        start_, end_ = int(seed_range_[0]), int(seed_range_[1])
        seed_range = [i for i in range(start_, end_)]
    else:
        # a list of lengths 1000 sampled from the range start_ to end_ (e.g.: [1, 1000000])
        numbers = list(range(start_, end_))
        seed_range = random.sample(numbers, k=1000)
    return seed_range