from collint import Experiment

class Experiment:
    def __init__(): ...
    # disabled type checking for this experiment
    # if originally enabled before experiment, it will be turned on once experiment is finished
    def disable_type_checking(self): ...
    # enabled type checking (enabled by default, included here for completeness)
    def max_c(self, max_c): ...
    # enabled parallel running
    def parallel(self, show_progress=True): ...
    # set up blackboard method
    def blackboard(self, ms, bs, n, compute_phi=False): ...
    # set up imitative method
    def imitative(self, ms, ps, n): ...
    # run the experiment, and return results
    # if the experiment is non-parallel, what is actually returned is a generator
    # calling list() on this generator will give a list of all the results
    # otherwise you can just treat it as a normal python generator
    def run(self): ...

from collint import Experiment

M = range(7, 10 + 1) # using range, so disabled type checking
P = [0.5, 0.6]
N = 1

results = (
    Experiment()
    .disable_type_checking()
    .max_c(1.0) # chaging max_c from default (10.0), so need to call this method
    .parallel(show_progress=True) # parallel is false by default, but show_progress is true by default
    .imitative(M, P, N)
    .run()
) # Results are a list of either a dictionary or None if we exceed computational cost of 1.0

from collint import Experiment

M = range(7, 10 + 1) # using range, so disabled type checking
B = range(6, 8 + 1)
N = 10

results = (
    Experiment()
    .disable_type_checking()
    .parallel() # show progress is true by default
    .blackboard(M, B, N, compute_phi = True) # compute_phi False by default
    .run()
) # Results are a list of either a dictionary or None if we exceed computational cost of 10.0

from collint import Experiment

M = range(7, 10 + 1) # using range, so disabled type checking
B = range(6, 8 + 1)
N = 10

results = (
    Experiment()
    .disable_type_checking()
    .blackboard(M, B, N, compute_phi = True) # compute_phi False by default
    .run()
) # Results is a generator

# Can use next() since results is a generator
first_result = next(results)

# Can use in a loop
for result in results:
    print(result) # result is either dictionary or None

M = range(7, 10 + 1)
B = range(6, 8 + 1)
N = 10

# Running experiment again...
results = (
    Experiment()
    .disable_type_checking()
    .blackboard(M, B, N, compute_phi = True) # compute_phi False by default
    .run()
)

# To run all the experiments sequentially without using a loop or next...
# ... use list()!

results = list(results) # experiment actually runs here

M = range(7, 10 + 1)
B = range(6, 8 + 1)
N = 10

# Running experiment again...
results = (
    Experiment()
    .disable_type_checking()
    .max_c(0.2) # set to a low value, we will probably have some Nones
    .blackboard(M, B, N, compute_phi = True) # compute_phi False by default
    .run()
)

# Alternatively, to get rid of all the None values...
# ... use list comprehension!!!
results = [r for r in results if r is not None]

from collint.blackboard import blackboard, blackboard_parallel

# returns a dictionary with keys ("m", "b", "t_star", "c"). "phi" is only present if compute_phi == True
# If c exceeds max_c, result will be None
def blackboard(m, b, /, *, max_c = 10.0, compute_phi = False): ...
# returns a list of the dictionaries or None values. Keys are the same as blackboard()
def blackboard_parallel(ms, bs, n, /, *, max_c = 10.0, compute_phi = False, show_progress = True): ...

from collint.blackboard import blackboard, blackboard_parallel

result = blackboard(10, 7, max_c = 9.0, compute_phi = True)
results = blackboard_parallel([8, 9, 10], [5, 6, 7], 10, max_c = 9.0, compute_phi = True, show_progress = False)

from collint.imitative import imitative, imitative_parallel

# returns a dictionary with keys ("m", "p", "t_star", "c")
# If c exceeds max_c, result will be None
def imitative(m, p, /, *, max_c = 10.0): ...
# returns a list of the dictionaries or None values. Keys are the same as imitative()
def imitative_parallel(ms, ps, n, /, *, max_c = 10.0, show_progress = True): ...

from collint.imitative import imitative, imitative_parallel

result = imitative(10, 0.7, max_c = 9.0)
results = imitative_parallel([8, 9, 10], [0.5, 0.6, 0.7], 10, max_c = 9.0, show_progress = False)

from collint.config import (
    is_python_type_checking_enabled,
    disable_python_type_checking,
    enable_python_type_checking
)

def disable_python_type_checking(): ...
def enable_python_type_checking(): ...
def is_python_type_checking_enabled(): ...

# check if type checking is enabled
enabled = is_python_type_checking_enabled()

# if enabled disable, if disabled enable
if enabled:
    disable_python_type_checking()
else:
    enable_python_type_checking()