#!/usr/bin/python3
# -*- coding: utf-8 -*-

import sys
import pyvisa
from pyvisa.constants import *

import importlib


class AFG3000():
    def __init__(self):
        self.rm = pyvisa.ResourceManager()

    def init(self):
        #importlib.reload(sys)
        res = self.rm.list_resources()
        return res;
        # self.mydev = self.rm.open_resource(res[0])

        # print(mydev.query("*OPT?"))

    def SetAmpl(self, ampl):
        self.mydev.write("VOLTAGE:AMPLITUDE {0}".format(ampl))
        #self.mydev.write("*SAV 1")
        #self.mydev.write("*RCL 1")

    def SetFrequency(self, Fre):
        self.mydev.write("FREQUENCY {0}".format(Fre))
        self.mydev.write("*SAV 1")
        #self.mydev.write("*RCL 1")

    def Open(self):
        return self.mydev.write("OUTPut1:STATe ON")

    def Connect(self,dev):
        self.mydev = self.rm.open_resource(dev)

    def Close(self):
        self.mydev.write("OUTPut1:STATe OFF")


    def query(self, Param):
        res = self.mydev.query(str(Param))
        res = float(res)
        return res