Lab/DFT_Circonv.py at master · hash527/Lab · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
import numpy as np
import matplotlib.pyplot as plt
import math
from scipy import signal

def DFT(x,N):
	j = np.complex(0,1)
	X = []
	N = len(x)
	for k in range(0,N):
		y = 0
		for n in range(0,N):
			y += x[n] * np.exp((-2*j*np.pi*k*n)/N)
		X.append(y)
	#print X
	return X

def IDFT(X):
	j = np.complex(0,1)
	x = []
	N = len(X)
	for n in range(0,N):
		y = 0
		for k in range(0,N):
			y += ( X[k] * np.exp((2*j*np.pi*k*n)/N) ) / N
		x.append(y)
	#print X
	return x

x1 = [1,2,3,4]
x2 = [1,2,3,4]

X1 = DFT(x1,4)
X2 = DFT(x2,4)

X3 = []

X3 = np.dot(X1,X2)

#X3 = IDFT(X3)

def conv(x1,x2):
	N1 = len(x1)
	N2 = len(x2)
	for i in range(N1-1,N1+N2-1):
		x1.append(0)
	for i in range(N2-1,N1+N2-1):
		x2.append(0)
	x3 = np.zeros(N1+N2)
	y = np.zeros(N1)
	for n in range(0,N1+N2-1):
		for k in range(0,N1+N2-1):
			x3[n] = x3[n] + ( x1[k] * x2[n-k])
	for i in range(0,N1):
		y[i] = x3[i] + x3[i+N1]
	return y

#x1x2 = conv(x1,x2)

#print x1x2
print X3