from qolab.hardware.lockin import LockinSCPI

from qolab.data.trace import Trace
import re
import numpy as np

class SRS_SR865A(LockinSCPI):
    """ SRS SR865A lockin """
    def __init__(self, resource):
        super().__init__(resource)
        self.config['Device model']='SRS SR865A'
        self.resource.read_termination='\n'

    def getFreqInt(self):
        qstr = f'FreqInt?'
        rstr = self.query(qstr)
        return(float(rstr))

    def setFreqInt(self, freq):
        cstr = f'FreqInt {freq}'
        rstr = self.write(cstr)


if __name__ == '__main__': 
    import pyvisa
    print("testing")
    rm = pyvisa.ResourceManager()
    print(rm.list_resources())    
    instr=rm.open_resource('TCPIP::192.168.0.51::INSTR')
    lockin = SRS_SR865A(instr)
    print('------ Header start -------------')
    print(str.join('\n', lockin.getHeader()))
    print('------ Header ends  -------------')