python实现简单函数发生器

最近学校又抽风把我自动化系的苦逼童鞋留下做课设，简直无聊到爆的-->用VB实现函数发生器，（语言不限制）

大伙不知从哪搞来的MATLAB版本，于是几十个人就在这基础上修修改改蒙混过关了，可我实在不想用MATLAB

写（要是被抓了，几十个人0分，那场面->。->）,用VB？ 又要学一门语言。。况且我不太习惯她的编码方式，于是

乎想起了我一直在玩的python，

# ------------------------先上代码---（由于只是应付课设，在编码规范和代码逻辑上没有做优化，只是实现功能）---------

import numpy as np

import matplotlib.pyplot as plt

import matplotlib.animation as animation

import tkinter

from tkinter import messagebox

freq = 1

margin = 1

deviation = 0

wave = 0

ss = []

fig = plt.figure()

axes1 = fig.add_subplot(111)

line, = axes1.plot(np.linspace(-10,10,3000))

def fangbo(data):

lis = []

for i in list(data):

if i//5%2 == 1:

lis.append(1)

else:

lis.append(-1)

return np.array(lis)

def juchi(data):

lis = []

for i in list(data):

lis.append(i%5/2.5)

return np.array(lis)-1

def sanjiao(data):

lis = []

for i in list(data):

if i//5%2 == 1:

lis.append(i%5/2.5)

else:

lis.append((5-i%5)/2.5)

return np.array(lis)-1

def update(data):

line.set_ydata(data)

return line,

def data_gen():

i = 50

res = [0*i for i in range(3000)]

while True:

global freq, margin, deviation, wave, ss

if wave == 0:

lis = list(margin*np.sin(freq*3.1415926/5*np.linspace(1,100,3000))+deviation)

elif wave == 1:

lis = list(margin*juchi(freq*np.linspace(1,100,3000))+deviation)

elif wave == 2:

lis = list(margin*sanjiao(freq*np.linspace(1,100,3000))+deviation)

elif wave == 3:

lis = list(margin*fangbo(freq*np.linspace(1,100,3000))+deviation)

ss = ss+lis[i-50:i]

yield ss+res[len(ss):]

i = i+50

if len(ss) >= 3000:

ss = ss[50:]

if i > 3050:

i = 50

def ssin():

global wave

wave = 0

def jjc():

global wave

wave = 1

def ssj():

global wave

wave = 2

def ffb():

global wave

wave = 3

def simu():

global freq, margin, deviation

try:

fe1 = float(fe.get('0.0','end'))

me1 = float(me.get('0.0','end'))

de1 = float(de.get('0.0','end'))

xe1 = float(xe.get('0.0','end'))

except:

messagebox.showinfo(title='错误',message='属性值必须为浮点数')

else:

freq = fe1

margin = me1

deviation = de1

if __name__ == '__main__':

root = tkinter.Tk()

root.title('波形发生器')

root.maxsize(320,140)

root.minsize(320,140)

lb = tkinter.Label(root,text='波形选择')

lb.pack()

fbt = tkinter.Button(root,text='正弦波',command=ssin)

fbt.place(x=20,y=20)

sbt = tkinter.Button(root,text='三角波',command=ssj)

sbt.place(x=100,y=20)

jbt = tkinter.Button(root,text='锯齿波',command=jjc)

jbt.place(x=180,y=20)

jbt = tkinter.Button(root,text='方波',command=ffb)

jbt.place(x=260,y=20)

ib = tkinter.Label(root,text='属性控制')

ib.place(x=130,y=50)

fl = tkinter.Label(root,text='频率')

fl.place(x=10,y=70)

fe = tkinter.Text(root,width=3,height=1)

fe.insert('end','1')

fe.place(x=50,y=70)

ml = tkinter.Label(root,text='幅值')

ml.place(x=85,y=70)

me = tkinter.Text(root,width=3,height=1)

me.insert('end','1')

me.place(x=125,y=70)

dl = tkinter.Label(root,text='偏移')

dl.place(x=160,y=70)

de = tkinter.Text(root,width=3,height=1)

de.insert('end','0')

de.place(x=200,y=70)

xl = tkinter.Label(root,text='相位')

xl.place(x=235,y=70)

xe = tkinter.Text(root,width=3,height=1)

xe.insert('end','0')

xe.place(x=275,y=70)

mb = tkinter.Button(root,text='仿真',command=simu)

mb.place(x=130,y=95)

ani = animation.FuncAnimation(fig, update, data_gen, interval=10)

plt.show()

root.mainloop()

# -------------------------------结果如下--------------------------------------------------------------