from .Tektronix import Tektronix
import uuid, os
import numpy as np
import time
from matplotlib import pyplot as plt

class TBS(Tektronix):
    __version__ = 'TBS-1.0.0'

    modelDict = {
        "1064": (0x0699, 0x03b3),
        "1052B-EDU": {0x0699, 0x0368},
    }

    def __init__(self, device = None, ident = (None, None), model= None):
        if model is None:
            super().__init__(device, ident)
            return
        if model in self.modelDict:
            ident = ident = self.modelDict[model]
            super().__init__(device, ident)
            if model == "1052B-EDU":
                self.instr.max_transfer_size = 50 
            return

        print('Liste des modèles de '+self.__class__.__name__+' référencés')
        print(self.modelDict)
        raise NameError("le modèle {} n'est pour le moment pas référencé".format(model))
