# -*- coding: utf-8 -*-
#!/usr/bin/pythonimport pyvisa
import xlwings as xw
import os
import matplotlib 
import visa
import time
import paramiko#author_jzk 2020-7-22
#class OSA，仪器的指令不大好记，获取WDM模式下的波长，功率以及OSNR值class OSA():"""docstring for OSA"""'''初始化OSA，连接'''def __init__(self):osa_GPIB='GPIB0::25::INSTR'									#GPIB口self.rm=visa.ResourceManager()								#open visaself.instrument=self.rm.open_resource(osa_GPIB)				#connect agilent osaprint('----------------> the OSA is connected successful n')def close(self):self.instrument.close()print('the osa connect is closed......')def wdm_AutoScan(self):'''进入WDM扫描光谱模式'''self.instrument.write('inst:sel "WDM_Autoscan"')		print('----------------> now the OSA is using sub-system ----WDM Spectrum n')print('wait the sub-systems end for 20s ')time.sleep(20)def wdm_sweep(self):'''WDm auto test operation'''self.instrument.write('INITiate:IMMediate')print('----------------> auto-scan is running now, wait for 15s n')time.sleep(14)def wdm_wav_get(self):'''get channel wavelength'''wve=[]try:print('----------------> wavelength is calculating---')str_wve=self.instrument.query('CALCulate:DATA:CWAVelengths?').split(',')for i in str_wve:wve.append(float(i))time.sleep(1)return wveexcept Exception as e:print('------->None data is calculated, please check the optical input')raise edef wdm_OSNR_get(self):'''get osnr '''osnr=[]try:print('----------------> OSNR is calculating---')str_osnr=self.instrument.query('CALCulate:DATA:CSNR?').split(',')for i in str_osnr:osnr.append(float(i))time.sleep(1)return osnrexcept Exception as e:print('------->None data is calculated, please check the optical input')raise e# 获取对应信号的功率def wdm_Power_get(self):power = []try:print('----------------> the power is calculated')str_power = self.instrument.query('CALCulate:DATA:CPOWers?').split(',')for i in str_power:power.append(float(i))time.sleep(0.5)return powerexcept Exception as e:print('------> None data is get, please check the optical input...')raise e#single sweep一遍'''def sweep(self):self.instrument.write('init:imm')print('take a sweep for osa,please wait for 2 seconds until it is ended')time.sleep(0.5)