decoder_test.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "import numpy as np\n",
    "from playsound import playsound\n",
    "from matplotlib import pyplot as plt\n",
    "from scipy.fft import fft, fftfreq, rfft, rfftfreq\n",
    "from scipy.io.wavfile import write, read \n",
    "from scipy.io import wavfile\n",
    "from time import sleep\n",
    "from math import floor\n",
    "import pyaudio\n",
    "import wave\n",
    "import random\n",
    "import pandas as pd\n",
    "import tensorflow as tf\n",
    "import keras\n",
    "import warnings\n",
    "warnings.filterwarnings(\"ignore\")\n",
    "import os\n",
    "import sys\n",
    "from time import sleep\n",
    "from project import decoder, modulator, wav_manager\n",
    "print(\"Press 1 to start searching for incoming signal or press 2 to raise an error for whatever reason\")\n",
    "inp = str(1)\n",
    "if inp == \"2\":\n",
    "    raise TypeError(\"why?\")\n",
    "if inp == \"1\":\n",
    "  dec = decoder()\n",
    "  model_x = tf.keras.models.load_model(\"model_june_new9.h5\")\n",
    "  model_param = tf.keras.models.load_model(\"threshold_v1.h5\")\n",
    "  dec.set_models(model_x, model_param)\n",
    "  dec.set_freqses(1000, 1500)\n",
    "  dec.set_change_time(0.1)\n",
    "  dec.set_max_param(8)\n",
    "  dec.set_print_amp(True)\n",
    "  dec.set_draw_graph(False)\n",
    "  dec.set_amplitude_threshold(3)\n",
    "  dec.set_predict_amp(True)\n",
    "  print(\"Choose method (only_AI, no_AI, both)\")\n",
    "  method = \"only_AI\"\n",
    "  dec.set_method(method)\n",
    "  sample_rate = 44100\n",
    "  i = 0\n",
    "  while True:\n",
    "    filename = str(i)\n",
    "    chunk = 1024\n",
    "    FORMAT = pyaudio.paInt16\n",
    "    channels = 1\n",
    "    sample_rate = 44100\n",
    "    record_seconds = 0.2\n",
    "    p = pyaudio.PyAudio()\n",
    "    stream = p.open(format=FORMAT,\n",
    "                channels=channels,\n",
    "                rate=sample_rate,\n",
    "                input=True,\n",
    "                output=True,\n",
    "                frames_per_buffer=chunk)\n",
    "    frames = []\n",
    "    print(\"Recording\")\n",
    "    for i in range(int(sample_rate / chunk * record_seconds)):\n",
    "      data = stream.read(chunk)\n",
    "      frames.append(data)\n",
    "    stream.stop_stream()\n",
    "    stream.close()\n",
    "    p.terminate()\n",
    "    wf = wave.open(filename, \"wb\")\n",
    "    wf.setnchannels(channels)\n",
    "    wf.setsampwidth(p.get_sample_size(FORMAT))\n",
    "    wf.setframerate(sample_rate)\n",
    "    wf.writeframes(b\"\".join(frames))\n",
    "    wf.close()\n",
    "    data = list(b\"\".join(frames))\n",
    "    wava = wav_manager()\n",
    "    wava.set_recorded_name(filename)\n",
    "    data, _ = wava.get_data_sample_rate()\n",
    "    xf, yf = dec.fourier_abs(data)\n",
    "    yf = dec.normalize_data_array(dec.secure_split(yf, 2), xf, 1000, 1500)\n",
    "    amp1, amp2 = dec.freq_sum(yf[0], xf, 1000), dec.freq_sum(yf[0], xf, 1500)\n",
    "    threshold = 7\n",
    "    koef = max(amp1, amp2) / min(amp1, amp2)\n",
    "    print(amp2)\n",
    "    if koef > threshold:\n",
    "       print(\"Отправляем сигнал чтобы начать запись\")\n",
    "       break\n",
    "    amp1, amp2 = dec.freq_sum(yf[1], xf, dec.freq0), dec.freq_sum(yf[1], xf, dec.freq1)\n",
    "    koef = max(amp1, amp2) / min(amp1, amp2)\n",
    "    print(koef, threshold)\n",
    "    if koef > threshold:\n",
    "       print(\"Отправляем сигнал чтобы начать запись\")\n",
    "       break\n",
    "  mod = modulator()\n",
    "  mod.set_text(\"11\")\n",
    "  _, signal = mod.signal_modulation()\n",
    "  wava.set_writing_name(\"Affirmative.wav\")\n",
    "  wava.write_sound(sample_rate, signal)\n",
    "  mod.set_text(\"00\")\n",
    "  _, signal = mod.signal_modulation()\n",
    "  wava.set_writing_name(\"Negative.wav\")\n",
    "  wava.write_sound(sample_rate, signal)\n",
    "  sleep(2)\n",
    "  playsound(\"Affirmative.wav\")\n",
    "  sleep(0.5)\n",
    "Continue = True\n",
    "while Continue:\n",
    "  print(\"RECORDING SIGNAL\")\n",
    "  record_seconds = 7\n",
    "  p = pyaudio.PyAudio()\n",
    "  stream = p.open(format=FORMAT,\n",
    "                channels=channels,\n",
    "                rate=sample_rate,\n",
    "                input=True,\n",
    "                output=True,\n",
    "                frames_per_buffer=chunk)\n",
    "  frames = []\n",
    "  for i in range(int(sample_rate / chunk * record_seconds)):\n",
    "    data = stream.read(chunk)\n",
    "    frames.append(data)\n",
    "  stream.stop_stream()\n",
    "  stream.close()\n",
    "  p.terminate()\n",
    "  print(\"Stopped recording, attemting to decode\")\n",
    "  data = list(b\"\".join(frames))\n",
    "  wf = wave.open(\"SIGNAL\", \"wb\")\n",
    "  wf.setnchannels(channels)\n",
    "  wf.setsampwidth(p.get_sample_size(FORMAT))\n",
    "  wf.setframerate(sample_rate)\n",
    "  wf.writeframes(b\"\".join(frames))\n",
    "  wf.close()\n",
    "  _, code1, code = dec.decoder_1(data)\n",
    "  wava.set_recorded_name(\"SIGNAL\")\n",
    "  data, _ = wava.get_data_sample_rate()\n",
    "  cnt = 0\n",
    "  while (code and code1 == \"WRONG\") or code != \"rofl\" or code1 != \"rofl\" or code1 != \"0\":\n",
    "   print(code1, code)\n",
    "   chunk = 1024\n",
    "   FORMAT = pyaudio.paInt16\n",
    "   channels = 1\n",
    "   sample_rate = 44100\n",
    "   playsound(\"Negative.wav\")\n",
    "   sleep(1)\n",
    "   p = pyaudio.PyAudio()\n",
    "   stream = p.open(format=FORMAT,\n",
    "                channels=channels,\n",
    "                rate=sample_rate,\n",
    "                input=True,\n",
    "                output=True,\n",
    "                frames_per_buffer=chunk)\n",
    "   frames = []\n",
    "   print(\"ERROR IN DECODING, RE-RECORDING\")\n",
    "   for i in range(int(sample_rate / chunk * record_seconds)):\n",
    "     data = stream.read(chunk)\n",
    "     frames.append(data)\n",
    "   stream.stop_stream()\n",
    "   stream.close()\n",
    "   p.terminate()\n",
    "   print(\"Stopped recording, attemting to decode\")\n",
    "   data = list(b\"\".join(frames))\n",
    "   wf = wave.open(\"SIGNAL\", \"wb\")\n",
    "   wf.setnchannels(channels)\n",
    "   wf.setsampwidth(p.get_sample_size(FORMAT))\n",
    "   wf.setframerate(sample_rate)\n",
    "   wf.writeframes(b\"\".join(frames))\n",
    "   wf.close()\n",
    "   wava.set_recorded_name(\"SIGNAL\")\n",
    "   data, _ = wava.get_data_sample_rate()\n",
    "   _, code1, code = dec.decoder_1(data)\n",
    "   if method == \"only_AI\":\n",
    "     code = \"1\"\n",
    "   if method == \"no_AI\":\n",
    "     code1 = \"0\"\n",
    "   if cnt >= 10:\n",
    "     Continue = False\n",
    "     break\n",
    "   cnt += 1\n",
    "  actual_code = code1\n",
    "  print(actual_code)\n",
    "  playsound(\"Affirmative.wav\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "import os\n",
    "import sys\n",
    "os.close(sys.stderr.fileno())"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "    yf = dec.normalize_data_array(dec.secure_split(yf, 2), xf, 1000, 1500)\n",
    "amp1, amp2 = dec.freq_sum(yf[1], xf, dec.freq0), dec.freq_sum(yf[1], xf, dec.freq1)\n",
    "    koef = max(amp1, amp2) / min(amp1, amp2)\n",
    "    print(koef, threshold)\n",
    "    if koef > threshold:\n",
    "       print(\"Отправляем сигнал чтобы начать запись\")\n",
    "       break"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.10.6"
  },
  "orig_nbformat": 4
 },
 "nbformat": 4,
 "nbformat_minor": 2
}