Frequency Shift Keying in Python

1.2k Views Asked by At

I am looking to take a digital signal and modulate it with 0's and 1's into an analog signal using python. The problem I am having is the wave seems to be going the wrong direction when it shifts. Here is my code so far, I know it isn't pretty.

import numpy as np
import math

import matplotlib.pyplot as plot

def plotPoint():
    current = 0
    amplitude = 1
    frequency = 2000
    X = []
    Y1 = []
    Y2 = []
    signal= [1,0,1,1,0,1,0,1,0,0]


    while current < 10000:
        if current < 1000 and signal[0] == 1:
            Y2.append(1)
        elif current < 2000 and current >= 1000 and signal[1] == 1:
            Y2.append(1)
        elif current < 3000 and current >= 2000 and signal[2] == 1:
            Y2.append(1)
        elif current < 4000 and current >= 3000 and signal[3] == 1:
            Y2.append(1)
        elif current < 5000 and current >= 4000 and signal[4] == 1:
            Y2.append(1)
        elif current < 6000 and current >= 5000 and signal[5] == 1:
            Y2.append(1)
        elif current < 7000 and current >= 6000 and signal[6] == 1:
            Y2.append(1)
        elif current < 8000 and current >= 7000 and signal[7] == 1:
            Y2.append(1)
        elif current < 9000 and current >= 8000 and signal[8] == 1:
            Y2.append(1)
        elif current < 10000 and current >= 9000 and signal[9] == 1:
            Y2.append(1)
        else:
            Y2.append(0)
    
    
        
        if Y2[-1] == 1:
            point2 = amplitude*math.sin((2*np.pi/frequency)*current)
        else:
            point2 = amplitude*math.sin((2*np.pi/(frequency/2)*current)
    
        X.append(current)
        Y1.append(point2)
   
    
        current += 1

    plot.plot(X,Y1)
    plot.plot(X,Y2)


    plot.title('Digital Signal')
    plot.xlabel('Time')
    plot.ylabel('Amplitude')
    plot.grid(True, which='both')
    #plot.axhline(y=0, color='k')

    plot.show()


plotPoint()

You can see the digital and analog signal are both displayed. I used a frequency of 2kHz as the carrier wave and you can see when it shift in this pictureenter image description here What am doing wrong? How can I remedy this?

2

There are 2 best solutions below

0
On

per barny's suggestion this will need small correction of your code:

if Y2[-1] == 1:
    point2 = amplitude*math.sin((2*np.pi/(frequency/2))*current)
else:
    point2 = amplitude*math.sin((2*np.pi/(frequency/4)*current))

enter image description here

Is this what you are looking for?

0
On

This code works with any combination of frequencies and intervals:

amplitude = 1
frequencies = [1200, 4000]
X = []
Y1 = []
Y2 = []
signal= [1,0,1,1,0,1,0,1,0,0]
interval = 1000
phase = 0

for s in signal:
    frequency = frequencies[s]
    for t in range (interval):
        phase += frequency/1000000 * math.pi * 2
        Y2.append(s)
        Y1.append(math.sin(phase) * amplitude)

X = range(len(signal)*interval)
ax, fig = plot.subplots(figsize=(16, 2))
plot.plot(X,Y1)
plot.plot(X,Y2)

plot.title('Digital Signal')
plot.xlabel('Time')
plot.ylabel('Amplitude')
plot.grid(True, which='both')

enter image description here