Initial commit

unilabos/devices/raman_uv/home_made_raman.py

@@ -0,0 +1,198 @@
+import json
+import serial
+import struct
+import crcmod
+import tkinter as tk
+from tkinter import messagebox
+import matplotlib.pyplot as plt
+from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg
+import time
+
+class RamanObj:
+    def __init__(self, port_laser,port_ccd, baudrate_laser=9600, baudrate_ccd=921600):
+        self.port_laser = port_laser
+        self.port_ccd = port_ccd
+
+        self.baudrate_laser = baudrate_laser
+        self.baudrate_ccd = baudrate_ccd
+        self.success    = False
+        self.status     = "None"
+        self.ser_laser = serial.Serial( self.port_laser, 
+                                        self.baudrate_laser,
+                                        bytesize=serial.EIGHTBITS,
+                                        parity=serial.PARITY_NONE,
+                                        stopbits=serial.STOPBITS_ONE, 
+                                        timeout=1)
+        
+        self.ser_ccd = serial.Serial(   self.port_ccd, 
+                                        self.baudrate_ccd,
+                                        bytesize=serial.EIGHTBITS,
+                                        parity=serial.PARITY_NONE,
+                                        stopbits=serial.STOPBITS_ONE, 
+                                        timeout=1)
+        
+    def laser_on_power(self, output_voltage_laser):
+
+        def calculate_crc(data):
+            """
+            计算Modbus CRC校验码
+            """
+            crc16 = crcmod.predefined.mkCrcFun('modbus')
+            return crc16(data)
+
+
+        device_id = 0x01  # 从机地址
+        register_address = 0x0298  # 寄存器地址
+        output_voltage = int(output_voltage_laser)  # 输出值
+
+        # 将输出值转换为需要发送的格式
+        output_value = int(output_voltage)  # 确保是整数
+        high_byte = (output_value >> 8) & 0xFF
+        low_byte = output_value & 0xFF
+
+        # 构造Modbus RTU数据包
+        function_code = 0x06  # 写单个寄存器
+        message = struct.pack('>BBHH', device_id, function_code, register_address, output_value)
+        crc = calculate_crc(message)
+        crc_bytes = struct.pack('<H', crc)
+        full_message = message + crc_bytes
+
+        try:
+            # 打开串口
+            self.ser_laser.write(full_message)
+
+            # 读取响应
+            response = self.ser_laser.read(8)  # Modbus响应通常为8字节
+            print(f"接收到的响应: {response.hex().upper()}")
+
+            # 校验响应CRC
+            if len(response) >= 5:
+                response_crc = calculate_crc(response[:-2])
+                received_crc = struct.unpack('<H', response[-2:])[0]
+                if response_crc == received_crc:
+                    print("响应CRC校验成功")
+                else:
+                    print("响应CRC校验失败")
+            else:
+                print("响应长度不足，无法校验")
+
+        except Exception as e:
+            print(f"发生错误: {e}")
+
+        return
+
+    def ccd_time(self, int_time):
+        # 发送命令
+        def send_command(ser, command):
+            try:
+                ser.write(command)
+                print(f"\u53d1\u9001: {command.hex()}")
+            except Exception as e:
+                print(f"\u53d1\u9001\u5931\u8d25: {e}")
+
+        # 接收数据
+        def receive_data(ser, expected_bytes):
+            try:
+                data = ser.read(expected_bytes)
+                print(f"\u63a5\u6536: {data.hex()}")
+                return data
+            except Exception as e:
+                print(f"\u63a5\u6536\u5931\u8d25: {e}")
+                return None
+
+        # 读取CCD数据并绘图
+        def read_and_plot(serial_port, int_time):
+
+            # 发送数据清空指令
+            clear_command = bytes([0xAA, 0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x55])
+            send_command(serial_port, clear_command)
+            time.sleep(0.1)
+            clear_command = bytes([0xAA, 0x10, 0x00, 0x00, 0x00, 0x00, 0x10, 0x55])
+            send_command(serial_port, clear_command)
+            receive_data(serial_port, 4136)
+            time.sleep(0.1)
+            clear_command = bytes([0xAA, 0xFF, 0x00, 0x00, 0x00, 0x00, 0xFF, 0x55])
+            send_command(serial_port, clear_command)
+            time.sleep(0.1)
+
+            # 发送数据读取指令
+            clear_command = bytes([0xAA, 0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x55])
+            send_command(serial_port, clear_command)
+            int_time = int(int_time)
+            time.sleep(int_time)
+            read_command = bytes([0xAA, 0x10, 0x00, 0x00, 0x00, 0x00, 0x10, 0x55])
+            send_command(serial_port, read_command)
+
+            # 接收 4136 字节的数据
+            data = receive_data(serial_port, 4136)
+
+            if not data or len(data) != 4136:
+                messagebox.showerror("\u9519\u8bef", "\u63a5\u6536\u6570\u636e\u5931\u8d25\u6216\u6570\u636e\u957f\u5ea6\u4e0d\u6b63\u786e")
+                return
+
+            # 将接收到的字节数据转换为 16 位无符号整型数组
+            values = [struct.unpack(">H", data[i:i + 2])[0] for i in range(0, len(data), 2)]
+            return values
+        
+        try:
+            ser = self.ser_ccd
+            values = read_and_plot(ser, int_time)  # 修正传递serial对象
+            print(f"\u8fd4\u56de\u503c: {values}")
+            return values
+        except Exception as e:
+            messagebox.showerror("\u9519\u8bef", f"\u4e32\u53e3\u521d\u59cb\u5316\u5931\u8d25: {e}")
+            return None
+
+    def raman_without_background(self,int_time, laser_power):
+        self.laser_on_power(0)
+        time.sleep(0.1)
+        ccd_data_background = self.ccd_time(int_time)
+        time.sleep(0.1)
+        self.laser_on_power(laser_power)
+        time.sleep(0.2)
+        ccd_data_total = self.ccd_time(int_time)
+        self.laser_on_power(0)
+        ccd_data = [x - y for x, y in zip(ccd_data_total, ccd_data_background)]
+        return ccd_data
+
+    def raman_without_background_average(self, sample_name , int_time, laser_power, average):
+        ccd_data = [0] * 4136
+        for i in range(average):
+            ccd_data_temp = self.raman_without_background(int_time, laser_power)
+            ccd_data = [x + y for x, y in zip(ccd_data, ccd_data_temp)]
+        ccd_data = [x / average for x in ccd_data]
+        #保存数据 用时间命名
+        t = time.strftime("%Y-%m-%d-%H-%M-%S-%MS", time.localtime())
+        folder_path = r"C:\auto\raman_data"
+        with open(f'{folder_path}/{sample_name}_{t}.txt', 'w') as f:
+            for i in ccd_data:
+                f.write(str(i) + '\n')
+        return ccd_data
+
+    def raman_cmd(self, command:str):
+        self.success = False
+        try:
+            cmd_dict = json.loads(command)
+            ccd_data = self.raman_without_background_average(**cmd_dict)
+            self.success = True
+
+            # return ccd_data
+        except Exception as e:
+            # return None
+            raise f"error: {e}"
+
+if __name__ == "__main__":
+    raman = RamanObj(port_laser='COM20', port_ccd='COM2')
+    ccd_data = raman.raman_without_background_average('test44',0.5,3000,1)
+
+    plt.plot(ccd_data)
+    plt.xlabel("Pixel")
+    plt.ylabel("Intensity")
+    plt.title("Raman Spectrum")
+    plt.show()
+
+
+
+
+
+