import sys
import signal
import asyncio
from bleak import BleakClient

import paho.mqtt.client as mqtt
import time
import json
import struct

async with BleakClient(address) as client:
    print(f"Connected: {client.is_connected}")
    await client.start_notify(char_uuid9, notification_handler9)
    try:
        while True:
            await asyncio.sleep(1.0)
    except KeyboardInterrupt:
        print("Stop!! Key board Interrupt!!");
        await client.stop_notify(char_uuid9)
    finally:
        await client.stop_notify(char_uuid9)

def notification_handler9(sender, data: bytearray):
"""Simple notification handler which sends MP34DT05 PDM data"""
    data1 = bytes_to_long(data)
    sdata['sensor'] = 'MP34DT05'
    sdata['func'] = 'PDM'
    sdata['time'] = time.time()
    sdata['data'] = data1
    publish(sdata)

def bytes_to_long(data):
    Data = [0]*2
    for i in range(0, len(data), 4):
        # 4バイトずつ読み込み、それをlong型に変換
        long_data = struct.unpack('I', data[i:i+4])[0]
        # 上位と下位の16bitに分ける(符号付き)
        Data[1] = struct.unpack('h', struct.pack('H', (long_data >> 16) & 0xFFFF))[0]
        Data[0] = struct.unpack('h', struct.pack('H', long_data & 0xFFFF))[0]
    return Data

# MQTT ブローカーの設定
broker_address = "localhost"
port = 1883
topic = "pdm_pic"
# MQTT クライアントを作成
mqtt_client = mqtt.Client()

 # MQTT ブローカーに接続
mqtt_client.connect(broker_address, port)
print(f"MQTT connected: {broker_address}")

 # MQTT クライアントを切断
mqtt_client.disconnect()
print(f"MQTT disconnected: {broker_address}")

def publish(sdata):
# Publish on MQTT
json_data = json.dumps(sdata)
mqtt_client.publish(topic, json_data)

import paho.mqtt.client as mqtt
import numpy as np
import matplotlib.pyplot as plt
import json
import time

# MQTT ブローカーの設定
broker_address = "localhost"
port = 1883
topic = "fft_data"

def main():
    # MQTT クライアントを作成
    mqtt_client = mqtt.Client()

    mqtt_client.on_connect = on_connect
    mqtt_client.on_message = on_message

    # MQTT ブローカーに接続
    mqtt_client.connect(broker_address, port)
    print(f"MQTT connected: {broker_address}")

    try:
        mqtt_client.loop_forever()
    except KeyboardInterrupt:
        # MQTT クライアントを切断
        mqtt_client.disconnect()
        print(f"MQTT disconnected: {broker_address}")

def on_message(client, userdata, msg):
dict_data = json.loads(msg.payload)
y = np.array(dict_data["data"])

# FFT Configulation
fs = 16000       # sampling frequency
N  = 1024       # number of samples
dt = 1/fs       # sampling period

# Window function
window = np.hanning(N)

    y = y * window

    # FFTを実行
    y_fft = np.fft.fft(y)               # 離散フーリエ変換
    freq = np.fft.fftfreq(N, d=dt)      # 周波数を割り当てる
    Amp = abs(y_fft/(N/2))              # 音の大きさ（振幅の大きさ）

    # 窓補正
    acf=1/(sum(window)/N)
    Amp = acf*Amp

output_FN = "/home/pi/Documents/pdm_imu/pictures/"+dict_data["datetime"]+".png"

### 音波のスペクトル ###
plt.plot(freq[1:int(N/2)], Amp[1:int(N/2)]) # A-f グラフのプロット
plt.xlabel("frequency [Hz]")
plt.ylabel("amplitude [V/rtHz]")
plt.xscale("log")                           # 横軸を対数軸にセット
plt.yscale("log")                           # 縦軸を対数軸にセット
plt.savefig(output_FN, dpi=300)
plt.clf()

    # NumPy配列をPythonリストに変換
    freq_list = freq[1:int(N/2)].tolist()
    amp_list = Amp[1:int(N/2)].tolist()

    # PythonリストをJSON形式に変換
    out_data = {"Num":N, "datatime":dict_data["datetime"], "freq": freq_list, "amp": amp_list}
    data_json = json.dumps(out_data)

    outFile = "/home/pi/Documents/pdm_imu/fftResult/"+dict_data["datetime"]+".json"
    # JSONデータをファイルに出力
    with open(outFile, "w") as f:
        f.write(data_json)

if (context.get("fileName") === undefined) {
    context.set("fileName", "newfile.png")
}
if (context.fileName == msg.payload){
    context.fileName = msg.payload;
    return msg;
} else {
    context.fileName = msg.payload;
}

/*
 * Arduino LSM9DS1 : send sensordata on BLE
 * Arduino MP34DT05 : Digital Microphone
 */
 
#include 
#include 

#define localNAME "peri_pdm"
#define DeviceNAME "PDM"

// UUID for MP34DT05(Digital MicroPhone)
#define MP34DT05_SERVICE_UUID "9997c274-9b20-4181-808f-ef2de31032a8"
#define MP34DT05_PDM_Characteristic_UUID "9997c274-9b21-4181-808f-ef2de31032a8"

// send data size
#define MAX_SAMPLES 1024
short sendData[MAX_SAMPLES];

//Variables for MP34DT05
short PDM_MP34DT05 = 0;
short sampleBuffer[256];
short sampleData;
short sampleDataBuffer[256];
unsigned long theTime;
volatile int samplesRead;
int cnt = 0;

// BLE Service
BLEService Sensor_MP34DT05_Service(MP34DT05_SERVICE_UUID);

// BLE Characteristic
BLELongCharacteristic MP34DT05_PDM(MP34DT05_PDM_Characteristic_UUID, BLERead | BLENotify);

void setup() {
  Serial.begin(115200);
  //while (!Serial);
  Serial.println("Started");

  // initialize the built-in LED pin to indicate when a central is connected
  pinMode(LED_BUILTIN, OUTPUT); 
  
  // MP34DT05 begin initialization
  PDM.onReceive(onPDMdata);

  // optionally set the gain, defaults to 20
  // PDM.setGain(30);

  // initialize PDM with: one channel (mono mode) & 16 kHz sample rate
  if (!PDM.begin(1, 16000)) {
    Serial.println("Failed to start PDM!");
    while (1);
  }

  // BLE begin initialization
  if (!BLE.begin()) {
    Serial.println("starting BLE failed!");
    while (1);
  }
  BLE.setLocalName(localNAME);
  BLE.setDeviceName(DeviceNAME);

  // Initialize MP34DT05 service
  set_MP34DT05();

  // start advertising
  BLE.advertise();
  Serial.println("Bluetooth device active, waiting for connections...");
}

void loop() {
  // wait for a BLE central
  BLEDevice central = BLE.central();
 
  // if a central is connected to the peripheral:
  if (central) {
    Serial.print("Connected to central: ");
    // print the central's BT address:
    Serial.println(central.address());
    // turn on the LED to indicate the connection:
    digitalWrite(LED_BUILTIN, HIGH);
 
    // while the central is connected:
    while (central.connected()) {
      if (cnt == 0){
        Read_PDM_Data();   // reads 1024 data from MP34DT05
        //IncDataGen();      // Incremental data generation for sending test
      }
      notify_PDM();       // SEND_BLE
      counter();
      delay(50);
    }

    // when the central disconnects
    digitalWrite(LED_BUILTIN, LOW);
    Serial.print("Disconnected from central: ");
    Serial.println(central.address());
  } else {
      readMP34DT05_nonBLE();    
  }
}

#include 
#include 

void onPDMdata() {
  // query the number of bytes available
  int bytesAvailable = PDM.available();

  // read into the sample buffer
  PDM.read(sampleBuffer, bytesAvailable);

  // 16-bit, 2 bytes per sample
  samplesRead = bytesAvailable / 2;
}

void set_MP34DT05(){
  // add the service UUID
  BLE.setAdvertisedService(Sensor_MP34DT05_Service);

  // add characteristic
  Sensor_MP34DT05_Service.addCharacteristic(MP34DT05_PDM);

  // Add service
  BLE.addService(Sensor_MP34DT05_Service);
}

// read pdm data and set to the dimensions
void Read_PDM_Data(){
  // Wait for samples to be read
  for (int j = 0; j<(MAX_SAMPLES/256); j++){
    while (samplesRead!=256);
    for (int i = 0; i < samplesRead; i++) {
      sendData[j*256+i] = sampleBuffer[i];
    }
    // Clear the read count
    samplesRead = 0;
  }
}

// Increment data generator for sending test
void IncDataGen(){
  for (int i=0; i<MAX_SAMPLES; i++){
    sendData[i] = i;
  }
}

// counter for send data control
void counter(){
  cnt++;
  if (cnt >= (MAX_SAMPLES/2+1)) cnt = 0;
}

void notify_PDM(){
  int i = 2*(cnt - 1);
  long sending;
  short sendingL, sendingH;
  if (cnt == 0){
    sending=0x7FFFFFFF;
  } else {
    sending=((long)sendData[i+1] << 16) | (sendData[i] & 0xFFFF);
  }
  MP34DT05_PDM.writeValue(sending);
  sendingH = (short)((sending >> 16) & 0xFFFF);
  sendingL = (short)(sending & 0xFFFF);
  Serial.print(cnt);
  Serial.print(": "); 
  Serial.print(sendingH);
  Serial.print(", "); 
  Serial.println(sendingL);
}

void readMP34DT05_nonBLE() {
  // print samples to the serial monitor or plotter
  for (int i = 0; i < samplesRead; i++) {
    sampleData = sampleBuffer[i];
    LED_Control();
  }
}

void LED_Control(){
  if (abs(sampleData)>=0 && abs(sampleData) < 50){
    digitalWrite(LEDR,HIGH);
    digitalWrite(LEDG,HIGH);
    digitalWrite(LEDB,HIGH);
  } else if (abs(sampleData) < 100){
    digitalWrite(LEDR,HIGH);
    digitalWrite(LEDG,HIGH);
    digitalWrite(LEDB,LOW);        
  } else if (abs(sampleData) < 150){
    digitalWrite(LEDR,HIGH);
    digitalWrite(LEDG,LOW);
    digitalWrite(LEDB,HIGH);        
  } else if (abs(sampleData) < 200){
    digitalWrite(LEDR,HIGH);
    digitalWrite(LEDG,LOW);
    digitalWrite(LEDB,LOW);        
  } else if (abs(sampleData) < 250){
    digitalWrite(LEDR,LOW);
    digitalWrite(LEDG,HIGH);
    digitalWrite(LEDB,HIGH);        
  } else if (abs(sampleData) < 300){
    digitalWrite(LEDR,LOW);
    digitalWrite(LEDG,HIGH);
    digitalWrite(LEDB,LOW);        
  } else if (abs(sampleData) < 350){
    digitalWrite(LEDR,LOW);
    digitalWrite(LEDG,LOW);
    digitalWrite(LEDB,HIGH);
  } else {
    digitalWrite(LEDR,LOW);
    digitalWrite(LEDG,LOW);
    digitalWrite(LEDB,LOW);
  }
}

# -*- coding: utf-8 -*-
#####
# Successively receive micro-phone data of int type
# Rev.0.03: 2023/11/16 無限loopにする。
# Rev.0.04: 2023/11/20 short X 2 = Longの転送に対応
#####
import sys
import signal
import asyncio
from bleak import BleakClient

import paho.mqtt.client as mqtt
import time
import json
import struct

# setting for BLE
ADDRESS = (
    "33:07:90:D9:XX:XX" # Arduino Nano 33 BLE sense Mino-2
)

# UUID for MP34DT05
CHARACTERISTIC_UUID9 = "9997c274-9b21-4181-808f-ef2de31032a8" # read & notify

# MQTT ブローカーの設定
broker_address = "localhost"
port = 1883
topic = "pdm_pic"
# MQTT クライアントを作成
mqtt_client = mqtt.Client()

data = [0.0,0.0,0.0]
sdata = {    # 辞書型データ
    'sensor': 'MP34DT05',
    'func': 'none',
    'time': 'time',
    'data': data
}

def publish(sdata):
    # Publish on MQTT
    json_data = json.dumps(sdata)
    mqtt_client.publish(topic, json_data)

def bytes_to_long(data):
    Data = [0]*2
    for i in range(0, len(data), 4):
        # 4バイトずつ読み込み、それをlong型に変換
        long_data = struct.unpack('I', data[i:i+4])[0]
    # 上位と下位の16bitに分ける(符号付き)
    Data[1] = struct.unpack('h', struct.pack('H', (long_data >> 16) & 0xFFFF))[0]
    Data[0] = struct.unpack('h', struct.pack('H', long_data & 0xFFFF))[0]
    return Data

def bytes_to_double(data):
    return struct.unpack('d', data)[0]

def bytes_to_signed_int(data):
    return int.from_bytes(data, byteorder='little', signed=True)

def notification_handler9(sender, data: bytearray):
    """Simple notification handler which sends MP34DT05 PDM data"""
    data1 = bytes_to_long(data)
    sdata['sensor'] = 'MP34DT05'
    sdata['func'] = 'PDM'
    sdata['time'] = time.time()
    sdata['data'] = data1
    publish(sdata)

async def main(address):
    char_uuid9 = CHARACTERISTIC_UUID9

    # MQTT ブローカーに接続
    mqtt_client.connect(broker_address, port)
    print(f"MQTT connected: {broker_address}")

    print(address, char_uuid9)
    async with BleakClient(address) as client:
        print(f"Connected: {client.is_connected}")
        await client.start_notify(char_uuid9, notification_handler9)
        try:
            while True:
                await asyncio.sleep(1.0)
        except KeyboardInterrupt:
            print("Stop!! Key board Interrupt!!");
            await client.stop_notify(char_uuid9)
        finally:
            await client.stop_notify(char_uuid9)

    print("Disconnected")
    # MQTT クライアントを切断
    mqtt_client.disconnect()
    print(f"MQTT disconnected: {broker_address}")
        
if __name__ == "__main__":
    asyncio.run(
        main(
            sys.argv[1] if len(sys.argv) > 1 else ADDRESS,
        )
    )

[{"id":"b92c72884452607b","type":"tab","label":"PDM_PIC0","disabled":false,"info":"","env":[]},{"id":"eb667179704ae718","type":"change","z":"b92c72884452607b","name":"削除","rules":[{"t":"delete","p":"payload.sensor","pt":"msg"},{"t":"delete","p":"payload.func","pt":"msg"},{"t":"delete","p":"payload.time","pt":"msg"}],"action":"","property":"","from":"","to":"","reg":false,"x":230,"y":340,"wires":[["f2e0b41e61d59d98"]]},{"id":"840928bbe2304943","type":"mqtt in","z":"b92c72884452607b","name":"","topic":"pdm_pic","qos":"2","datatype":"auto-detect","broker":"496bc63a53e5165b","nl":false,"rap":true,"rh":0,"inputs":0,"x":100,"y":340,"wires":[["eb667179704ae718"]]},{"id":"6b39757a0019cb48","type":"debug","z":"b92c72884452607b","name":"debug 73","active":true,"tosidebar":true,"console":false,"tostatus":false,"complete":"false","statusVal":"","statusType":"auto","x":800,"y":480,"wires":[]},{"id":"fcddb3a469f1195c","type":"change","z":"b92c72884452607b","name":"filename","rules":[{"t":"set","p":"filename","pt":"msg","to":"filename","tot":"flow"},{"t":"set","p":"filename","pt":"msg","to":"filename&\".csv\"","tot":"jsonata"}],"action":"","property":"","from":"","to":"","reg":false,"x":680,"y":340,"wires":[["b86e08f6db1ccd43"]]},{"id":"b86e08f6db1ccd43","type":"csv","z":"b92c72884452607b","name":"","sep":",","hdrin":"","hdrout":"none","multi":"one","ret":"\\n","temp":"cycle,count,datetime,data","skip":"0","strings":true,"include_empty_strings":"","include_null_values":"","x":810,"y":340,"wires":[["5f704ff7e909bcf0"]]},{"id":"5f704ff7e909bcf0","type":"file","z":"b92c72884452607b","name":"","filename":"filename","filenameType":"msg","appendNewline":false,"createDir":true,"overwriteFile":"false","encoding":"none","x":940,"y":340,"wires":[[]]},{"id":"f2de7a511eddd62f","type":"comment","z":"b92c72884452607b","name":"送信データ集約","info":"","x":120,"y":220,"wires":[]},{"id":"c6d91d227d1d8218","type":"inject","z":"b92c72884452607b","name":"初期化","props":[{"p":"payload"},{"p":"topic","vt":"str"}],"repeat":"","crontab":"","once":true,"onceDelay":0.1,"topic":"","payload":"","payloadType":"date","x":120,"y":260,"wires":[["c02c5bc558f0cf77","4fb20a31deffaa5d"]]},{"id":"c02c5bc558f0cf77","type":"function","z":"b92c72884452607b","name":"初期化","func":"flow.set('count', 0);\nflow.set('cycle', 0);\nflow.set('count_max', 1024);","outputs":1,"noerr":0,"initialize":"","finalize":"","libs":[],"x":270,"y":240,"wires":[[]]},{"id":"4fb20a31deffaa5d","type":"function","z":"b92c72884452607b","name":"配列作成","func":"// 配列の要素数\nflow.set(\"length\", 1024);\n// 配列作成\n//if (!context.flow.array) {\ncontext.flow.array = new Array(flow.get(\"length\"));\n//}\n// 配列の初期化:データを0で埋める\ncontext.flow.array.fill(0);","outputs":1,"noerr":0,"initialize":"","finalize":"","libs":[],"x":280,"y":280,"wires":[[]]},{"id":"503549393a32743c","type":"inject","z":"b92c72884452607b","name":"受信開始","props":[{"p":"filename","v":"/home/pi/Documents/pdm_imu/receive_","vt":"str"},{"p":"payload"}],"repeat":"","crontab":"","once":false,"onceDelay":0.1,"topic":"","payload":"93:B3:5F:16:CD:A7","payloadType":"str","x":140,"y":80,"wires":[["afa7ca985f556c5d","a7dd8a739a9e0401"]]},{"id":"afa7ca985f556c5d","type":"moment","z":"b92c72884452607b","name":"日時","topic":"","input":"","inputType":"date","inTz":"Asia/Tokyo","adjAmount":0,"adjType":"days","adjDir":"add","format":"YYYYMMDD_HHmmss","locale":"ja-JP","output":"datetime","outputType":"msg","outTz":"Asia/Tokyo","x":290,"y":120,"wires":[["d839b8ea10bbcadd"]]},{"id":"d839b8ea10bbcadd","type":"change","z":"b92c72884452607b","name":"filename","rules":[{"t":"set","p":"filename","pt":"flow","to":"filename&datetime&ext","tot":"jsonata"}],"action":"","property":"","from":"","to":"","reg":false,"x":420,"y":120,"wires":[[]]},{"id":"a7dd8a739a9e0401","type":"exec","z":"b92c72884452607b","command":"python3 /home/pi/source/python/ble/ble_pdh04.py","addpay":"payload","append":"","useSpawn":"false","timer":"","winHide":false,"oldrc":false,"name":"ble_pdh04","x":310,"y":60,"wires":[[],["4b9e541a10ea3b4e"],["4b9e541a10ea3b4e"]]},{"id":"4b9e541a10ea3b4e","type":"debug","z":"b92c72884452607b","name":"debug 74","active":true,"tosidebar":true,"console":false,"tostatus":false,"complete":"false","statusVal":"","statusType":"auto","x":460,"y":60,"wires":[]},{"id":"d03001a5483a7983","type":"mqtt out","z":"b92c72884452607b","name":"","topic":"fft_data","qos":"","retain":"","respTopic":"","contentType":"","userProps":"","correl":"","expiry":"","broker":"496bc63a53e5165b","x":680,"y":380,"wires":[]},{"id":"80e071f702c8e792","type":"function","z":"b92c72884452607b","name":"png file","func":"if (context.get(\"fileName\") === undefined) {\n    context.set(\"fileName\", \"newfile.png\")\n}\nif (context.fileName == msg.payload){\n    context.fileName = msg.payload;\n    return msg;\n} else {\n    context.fileName = msg.payload;\n}\n","outputs":1,"noerr":0,"initialize":"","finalize":"","libs":[],"x":280,"y":480,"wires":[["615fda249ef06afe","b296b5e594363430"]]},{"id":"fad8be86f73b7229","type":"watch","z":"b92c72884452607b","name":"png監視","files":"/home/pi/Documents/pdm_imu/pictures","recursive":"","x":140,"y":480,"wires":[["80e071f702c8e792"]]},{"id":"615fda249ef06afe","type":"image viewer","z":"b92c72884452607b","name":"","width":160,"data":"payload","dataType":"msg","active":true,"x":410,"y":480,"wires":[[]]},{"id":"b296b5e594363430","type":"debug","z":"b92c72884452607b","name":"debug 75","active":false,"tosidebar":true,"console":false,"tostatus":false,"complete":"false","statusVal":"","statusType":"auto","x":420,"y":440,"wires":[]},{"id":"2f636e0a129e1d4e","type":"mqtt in","z":"b92c72884452607b","name":"","topic":"fft_data","qos":"2","datatype":"auto-detect","broker":"496bc63a53e5165b","nl":false,"rap":true,"rh":0,"inputs":0,"x":670,"y":480,"wires":[["6b39757a0019cb48"]]},{"id":"f2e0b41e61d59d98","type":"moment","z":"b92c72884452607b","name":"日時","topic":"","input":"","inputType":"date","inTz":"Asia/Tokyo","adjAmount":0,"adjType":"days","adjDir":"add","format":"YYYY-MM-DD_HH_mm_ss","locale":"ja-JP","output":"payload.datetime","outputType":"msg","outTz":"Asia/Tokyo","x":350,"y":340,"wires":[["35acbef30630d644","e159219189a042dc"]]},{"id":"d2e98d8914ce0525","type":"inject","z":"b92c72884452607b","name":"FFT開始","props":[{"p":"payload"},{"p":"topic","vt":"str"}],"repeat":"","crontab":"","once":false,"onceDelay":0.1,"topic":"","payload":"/home/pi/source/python/ble/fft_calc02.py","payloadType":"str","x":640,"y":60,"wires":[["582cb6604b366206"]]},{"id":"582cb6604b366206","type":"exec","z":"b92c72884452607b","command":"python3 ","addpay":"payload","append":"","useSpawn":"false","timer":"","winHide":false,"oldrc":false,"name":"fft_calc02","x":780,"y":60,"wires":[[],["fd60ec46c1ef8d91"],["fd60ec46c1ef8d91"]]},{"id":"fd60ec46c1ef8d91","type":"debug","z":"b92c72884452607b","name":"debug 76","active":true,"tosidebar":true,"console":false,"tostatus":false,"complete":"payload","targetType":"msg","statusVal":"","statusType":"auto","x":920,"y":60,"wires":[]},{"id":"c3aa19db00daae06","type":"inject","z":"b92c72884452607b","name":"FFT終了","props":[{"p":"kill","v":"","vt":"str"},{"p":"topic","vt":"str"}],"repeat":"","crontab":"","once":false,"onceDelay":0.1,"topic":"","x":640,"y":100,"wires":[["582cb6604b366206"]]},{"id":"48c695c84ea80cb6","type":"inject","z":"b92c72884452607b","name":"受信終了","props":[{"p":"kill","v":"","vt":"str"},{"p":"topic","vt":"str"}],"repeat":"","crontab":"","once":false,"onceDelay":0.1,"topic":"","x":140,"y":120,"wires":[["a7dd8a739a9e0401"]]},{"id":"35acbef30630d644","type":"debug","z":"b92c72884452607b","name":"debug 77","active":false,"tosidebar":true,"console":false,"tostatus":false,"complete":"false","statusVal":"","statusType":"auto","x":480,"y":380,"wires":[]},{"id":"e159219189a042dc","type":"function","z":"b92c72884452607b","name":"集約(ble_pdh04)","func":"var cycle = flow.get('cycle');\nvar count = flow.get('count');\nvar count_max = flow.get('count_max');\nif (msg.payload.data[1] >= (2 ** 15 - 16) || count > 1024){\n    count = 0;\n    context.flow.array.fill(0);\n    node.warn(\"first data detexted(\"+String(msg.payload.data[1]+\")!!\"));\n} else {\n    context.flow.array[count]   = msg.payload.data[0];\n    context.flow.array[count+1] = msg.payload.data[1];\n    msg.payload.count = count;\n    msg.payload.cycle = cycle;\n    count = count+2;\n}\nflow.set('count', count);\nif (count == count_max){\n    msg.payload.data = context.flow.array;\n    cycle++;\n    flow.set('cycle', cycle);\n    return msg;\n}\n","outputs":1,"noerr":0,"initialize":"","finalize":"","libs":[],"x":510,"y":340,"wires":[["fcddb3a469f1195c","d03001a5483a7983"]]},{"id":"5c5bb3b1c0f92cc0","type":"comment","z":"b92c72884452607b","name":"外部プログラム起動","info":"","x":130,"y":40,"wires":[]},{"id":"efe0b5481ecac62e","type":"comment","z":"b92c72884452607b","name":"画像表示","info":"","x":100,"y":440,"wires":[]},{"id":"496bc63a53e5165b","type":"mqtt-broker","name":"","broker":"localhost","port":"1883","clientid":"","autoConnect":true,"usetls":false,"protocolVersion":"4","keepalive":"60","cleansession":true,"birthTopic":"","birthQos":"0","birthPayload":"","birthMsg":{},"closeTopic":"","closeQos":"0","closePayload":"","closeMsg":{},"willTopic":"","willQos":"0","willPayload":"","willMsg":{},"userProps":"","sessionExpiry":""}]

import paho.mqtt.client as mqtt
import numpy as np
import matplotlib.pyplot as plt
import json

# MQTT ブローカーの設定
broker_address = "localhost"
port = 1883
topic = "fft_data"

# FFT Configulation
fs = 16000       # sampling frequency
N  = 1024       # number of samples
dt = 1/fs       # sampling period

# Window function
window = np.hanning(N)

def on_connect(client, userdata, flags, rc):
    print("Connected with result code "+str(rc))
    client.subscribe(topic)

def on_message(client, userdata, msg):
    dict_data = json.loads(msg.payload)
    y = np.array(dict_data["data"])
    y = y * window

    # FFTを実行
    y_fft = np.fft.fft(y)               # 離散フーリエ変換
    freq = np.fft.fftfreq(N, d=dt)      # 周波数を割り当てる
    Amp = abs(y_fft/(N/2))              # 音の大きさ（振幅の大きさ）

    # 窓補正
    acf=1/(sum(window)/N)
    Amp = acf*Amp

    output_FN = "/home/pi/Documents/pdm_imu/pictures/"+dict_data["datetime"]+".png"

# スレッドを開始
    ### 音波のスペクトル ###
    plt.plot(freq[1:int(N/2)], Amp[1:int(N/2)]) # A-f グラフのプロット
    plt.xlabel("frequency [Hz]")
    plt.ylabel("amplitude [V/rtHz]")
    plt.xscale("log")                           # 横軸を対数軸にセット
    plt.yscale("log")                           # 縦軸を対数軸にセット
    plt.savefig(output_FN, dpi=300)
    plt.clf()

    # NumPy配列をPythonリストに変換
    freq_list = freq[1:int(N/2)].tolist()
    amp_list = Amp[1:int(N/2)].tolist()

    # PythonリストをJSON形式に変換
    out_data = {"Num":N, "datatime":dict_data["datetime"], "freq": freq_list, "amp": amp_list}
    data_json = json.dumps(out_data)

    outFile = "/home/pi/Documents/pdm_imu/fftResult/"+dict_data["datetime"]+".json"
    # JSONデータをファイルに出力
    with open(outFile, "w") as f:
        f.write(data_json)

def main():
    # MQTT クライアントを作成
    mqtt_client = mqtt.Client()

    mqtt_client.on_connect = on_connect
    mqtt_client.on_message = on_message

    # MQTT ブローカーに接続
    mqtt_client.connect(broker_address, port)
    print(f"MQTT connected: {broker_address}")

    try:
        #while True:
        mqtt_client.loop_forever()
    except KeyboardInterrupt:
        # MQTT クライアントを切断
        mqtt_client.disconnect()
        print(f"MQTT disconnected: {broker_address}")

if __name__ == "__main__":
    main()