Module pyspecan.model.model

Functions

def GetMode(mode)

Expand source code

def GetMode(mode):
    if mode == _Mode.SWEPT:
        return ModeSwept
    elif mode == _Mode.RT:
        return ModeRT
    raise err.UnknownOption(f"Unknown mode: {mode}")

def GetModel(mode, sink)

Expand source code

def GetModel(mode, sink):
    mode = GetMode(mode)
    sink = GetSink(sink)
    return lambda m=mode, s=sink, **kwargs: Model(m, s, **kwargs)

def GetSink(sink)

Expand source code

def GetSink(sink):
    if sink == _Sink.FILE:
        return SinkFile
    elif sink == _Sink.LIVE:
        return SinkLive
    raise err.UnknownOption(f"Unknown sink: {sink}")

def ModelArgs(mode, sink, parser)

Expand source code

def ModelArgs(mode, sink, parser):
    args_model(parser)
    if mode == _Mode.SWEPT:
        args_swept(parser)
    elif mode == _Mode.RT:
        args_rt(parser)

    if sink == _Sink.FILE:
        args_file(parser)
    elif sink == _Sink.LIVE:
        args_live(parser)