# -*- coding: utf-8 -*-
"""
Created on Tue Feb  8 09:27:57 2022

@author: Publico
"""


import visa
import numpy as np
import time
import os
import matplotlib.pyplot as plt

#%%
"""Abro la comunicación con el lockin"""


resourcemanager = visa.ResourceManager()
print(resourcemanager.list_resources())

lockin_name = 'GPIB0::8::INSTR'

lockin = resourcemanager.open_resource(lockin_name)
#%%
'''
frec=200
lockin.write(f"FREQ {200}")

lockin.read_ascii_values("FREQ?")

sensit=21
lockin.write(f"SENS {sensit}")

sensit=22
lockin.write(f"SENS {22}")


x,y=lockin.read("SNAP? 1, 2")
print(x)

lockin.write(f"SLVL {0.2}")

'''
lockin.write(f"FREQ {500}")
lockin.read("FREQ?")
lockin.read_ascii_values("FREQ?")

#%%
x,y=lockin.query_ascii_values("SNAP? 1, 2")

print(y)

o = lockin.query_ascii_values(f"LIAS? {2}")
print(o)

#%%
freqvec=np.geomspace(100,80000,200)
rvector=[]
thetavector=[]
xvector=[]
yvector=[]
escalafvector=[]


import time


sensit=10
lockin.write(f"SENS {sensit}")
for f in freqvec:
    lockin.write(f"FREQ {f}")
    time.sleep(.5)
    #lockin.read(f"SENS {sensit}")
    bit=2
    o = lockin.query_ascii_values(f"LIAS? {bit}")
    while o[0] == 1.0:
        sensit= sensit+1
        lockin.write(f"SENS {sensit}")
        time.sleep(1)
        lockin.query_ascii_values(f"LIAS?")
        o = lockin.query_ascii_values(f"LIAS? {bit}")
        
        
    x,y,r,theta=lockin.query_ascii_values("SNAP? 1, 2, 3, 4")
    escalaf=lockin.query_ascii_values("SENS?")
    xvector.append(x)
    yvector.append(y)
    rvector.append(r)
    thetavector.append(theta)
    escalafvector.append(escalaf[0])

#%%
i=0
#os.chdir (r'E:\resistividad')
np.savetxt('10hierrocostado.dat',np.transpose([freqvec[i:],xvector[i:],yvector[i:],rvector[i:],thetavector[i:],escalafvector[i:]]),header='Frecuencias X Y R theta sens')