flightmatrixbridge


Nameflightmatrixbridge JSON
Version 1.7.0 PyPI version JSON
download
home_pagehttps://github.com/Kawai-Senpai/Py-FlightMatrix-Bridge
SummaryFlightMatrixBridge is a Python library that facilitates seamless inter-process communication between the Flight Matrix software using shared memory. It handles frames, timestamps, and movement commands for efficient data sharing in robotics applications.
upload_time2024-11-12 17:06:33
maintainerNone
docs_urlNone
authorRanit Bhowmick
requires_pythonNone
licenseMIT License with attribution requirement
keywords robotics inter-process communication shared memory data management
VCS
bugtrack_url
requirements numpy ultraprint opencv-python
Travis-CI No Travis.
coveralls test coverage No coveralls.
            # FlightMatrix Bridge • API for Flight Matrix Simulation Software

![FlightMatrix Bridge Cover](https://github.com/Kawai-Senpai/Py-FlightMatrix-Bridge/blob/0d434625dc45a12744eacaa4df21de29f0072612/Assets/FlightMatrix%20Cover.png)

**FlightMatrix Bridge** is a Python-based API designed for controlling and fetching information, frames, and other data from Flight Matrix. This library enables efficient and real-time communication between various processes in a system, primarily designed for interfacing flight simulators, UAV systems, or other robotics platforms. It utilizes the `multiprocessing.shared_memory` module to share data such as frames, sensor data, and movement commands across multiple processes.

Download the software from [Flight Matrix](https://gamejolt.com/games/flightmatrix/933049).

The project is structured around two primary files:

- **`bridge.py`**: Implements the core functionalities of shared memory management, handling frames, sensor data, movement commands, and logging.
- **`utilities.py`**: Provides utility functions to handle timestamps and convert them into human-readable formats. It also includes a `DroneController` class for controlling drones.

## Table of Contents

1. [Introduction](#introduction)
2. [Features](#features)
3. [Controls](#controls)
4. [Installation](#installation)
5. [Usage](#usage)
    1. [Initializing the FlightMatrixBridge](#initializing-the-flightmatrixbridge)
    2. [Units](#units)
    3. [Axis System](#axis-system)
    4. [Logging Functions](#logging-functions)
    5. [Flight Matrix API](#flight-matrix-api)
        - [Getting Frame Data](#getting-frame-data)
        - [Fetching Sensor Data](#fetching-sensor-data)
        - [Sending Movement Commands](#sending-movement-commands)
    6. [Resolution & Noise Configuration](#resolution--noise-configuration)
    7. [Timestamp Utilities](#timestamp-utilities)
6. [Detailed Functionality](#detailed-functionality)
    - [Initialization](#initialization)
    - [Shared Memory Management](#shared-memory-management)
    - [Frame Handling](#frame-handling)
    - [Sensor Data Handling](#sensor-data-handling)
    - [Movement Command Management](#movement-command-management)
    - [Logging](#logging)
    - [Noise Application](#noise-application)
7. [Examples](#examples)
    1. [Example 1: Fetching Frames](#example-1-fetching-frames)
    2. [Example 2: Sending Movement Commands](#example-2-sending-movement-commands)
    3. [Example 3: Fetching Sensor Data](#example-3-fetching-sensor-data)
    4. [Example 4: Drone Controller](#example-4-drone-controller)
    5. [Example 5: Data Recorder](#example-5-data-recorder)
8. [Documentation](#documentation)
    1. [Class: `FlightMatrixBridge`](#class-flightmatrixbridge)
        - [Attributes](#attributes)
        - [Methods](#methods) 
    2. [Utilities](#utilities)
        1. [`timestamp2string`](#timestamp2stringtimestamp)
        2. [`timestamp2datetime`](#timestamp2datetimetimestamp)
        3. [`cartesian_to_gps`](#cartesian_to_gps)
    3. [Class: `DroneController`](#class-dronecontroller)
          - [Attributes](#attributes)
          - [Methods](#methods)
    4. [Class: `DataRecorder`](#class-datarecorder)
          - [Attributes](#attributes)
          - [Methods](#methods)
    5. [Class: `DataStreamer`](#data-streaming-with-datastreamer)
9. [Credits](#credits)

## Introduction

![FlightMatrix Bridge](https://github.com/Kawai-Senpai/Py-FlightMatrix-Bridge/blob/020882b8df9d9452bcb43664452dadc9df753558/Assets/Screenshot%20(4).png)

The **FlightMatrixBridge** system is designed to bridge multiple processes that need to access shared memory for real-time communication. The typical use cases include flight simulators, robotics platforms, autonomous vehicles, and any application where sharing large datasets like frames or sensor readings between processes is essential.

This package provides:
- An interface to retrieve frames and sensor data from shared memory.
- The ability to send movement commands to be processed by another service.
- Real-time noise application to sensor data.
- Utilities to handle timestamps.

## Features

![Controllable Features](https://github.com/Kawai-Senpai/Py-FlightMatrix-Bridge/blob/020882b8df9d9452bcb43664452dadc9df753558/Assets/Screenshot%20(3).png)
*Controllable Features*

![Simulation Environments](https://github.com/Kawai-Senpai/Py-FlightMatrix-Bridge/blob/020882b8df9d9452bcb43664452dadc9df753558/Assets/Screenshot%20(1).png)
*Simulation Environments*

![Graphics Presets](https://github.com/Kawai-Senpai/Py-FlightMatrix-Bridge/blob/020882b8df9d9452bcb43664452dadc9df753558/Assets/Screenshot%20(2).png)
*Graphics Presets*

The **FlightMatrix** software offers a range of features to facilitate real-time communication and data sharing between processes. Key features include:
- **Dual Camera Support**: Flight Matrix is equipped with two cameras—left and right—that operate simultaneously. Each camera is capable of outputting high-quality RGB images, depth passes (z-depth), and segmentation maps, providing a comprehensive view of your simulated environment.
- **Independent Camera Control**: Each camera can be controlled independently, allowing you to position them relative to the drone with precision. Adjust the x, y, z coordinates, as well as yaw, pitch, and roll to achieve the desired perspective.
- **Variable Speed Control**: Control the speed of each axis and the rotation speed of the cameras, ensuring you can fine-tune the responsiveness to suit your simulation needs.
- **Customizable Output**: Turn on and off various output maps as required. Control the resolution of the output frames and adjust the Field of View (FOV) to enhance your visual experience.
- **Graphics Presets**: Choose from various graphics presets tailored for different simulation scenarios. Optimize the software’s performance based on your hardware capabilities and desired visual fidelity.
- **Diverse Simulation Environments**: Flight Matrix features a range of realistic maps, including architectural, natural, and ultra-realistic environments for authentic simulations. Navigate through intricate landscapes and urban settings as if you were flying in the real world.
- **Human-like AI Characters**: Enhance your simulations with beautifully animated AI characters that simulate real crowds and human interactions. Observe how they behave and interact within the environment, adding depth to your scenarios.

The **FlightMatrixBridge** API provides a simple and efficient way to interact with the Flight Matrix simulation software, enabling you to access frames, sensor data, and movement commands in real-time. The API is designed to be easy to use and flexible, allowing you to integrate it into your projects seamlessly.
- **Frame Management**: Retrieve left/right frames, z-depth maps, and segmentation frames in real-time.
- **Sensor Data Access**: Retrieve real-time sensor data such as location, orientation, velocity, acceleration, magnetometer readings, and more.
- **Movement Command Handling**: Send movement commands (position and orientation) for external systems to process.
- **Noise Simulation**: Add configurable levels of noise to sensor data for testing robustness.
- **Flexible Resolution Handling**: Easily set and adjust resolution for frames.
- **Timestamp Management**: Convert timestamps into human-readable formats and handle system-wide timing data.

## Controls

| Action                          | Keyboard/Mouse          |
|---------------------------------|-------------------------|
| Move Forward                    | W                       |
| Move Backward                   | S                       |
| Move Left                       | A                       |
| Move Right                      | D                       |
| Ascend                          | Space Bar               |
| Descend                         | Left Shift              |
| Rotate (Yaw, Pitch, Roll)       | Arrow Keys              |
| Move Left/Right                 | Q / E                   |
| Pause                           | Escape / Pause / P      |
| Exit                            | Escape / Pause / P      |
| Spawn Human AI Character        | H                       |
| Return to Starting Location     | R                       |

## Installation

Download the software from [Flight Matrix](https://gamejolt.com/games/flightmatrix/933049).

To install the **FlightMatrixBridge (API)**, simply use pip:

```bash
pip install flightmatrixbridge
```

Make sure your system has Python 3.8+ and supports the `multiprocessing.shared_memory` module.

## Usage

### Initializing the FlightMatrixBridge

To initialize and start using the **FlightMatrixBridge**, create an instance of the `FlightMatrixBridge` class and specify the resolution of the frames you want to handle:

```python
from flightmatrix.bridge import FlightMatrixBridge

bridge = FlightMatrixBridge(resolution=(1226, 370), noise_level=0.01, apply_noise=False)  # Set frame resolution (width, height), noise level, and noise application
```

### Units

The system uses the following units for sensor data:
- Length: centimeters (cm)
- Angular values: degrees (°)
- Angular velocity/ gyroscope readings: degrees per second (°/s)
- Acceleration/ accelerometer readings: centimeters per second squared (cm/s²) 
- Magnetometer readings: unit vector
- LiDAR data: centimeters (cm)
- Collision detection: centimeters (cm)
- Timestamp: milliseconds (ms)

### Axis System

The axis system differs slightly between the software interface and the API. Below is a detailed explanation for both.

#### Inside the Software

When adjusting camera positions or configuring movement multipliers within the software, the following axis system is used:

| Direction | Axis |
|-----------|------|
| Forward   | Y    |
| Backward  | -Y   |
| Left      | -X   |
| Right     | X    |
| Up        | Z    |
| Down      | -Z   |

Rotation values are in degrees and are labeled roll, pitch, and yaw:

| Rotation | Axis |
|----------|------|
| Roll     | X    |
| Pitch    | Y    |
| Yaw      | Z    |

**Axis Orientation:**
```
       Z (Up)
        |
        |
        |
        O------ X (Right)
       /
      /
    -Y (Backward)
```

#### In the API

The API uses a different axis system for movement commands and sensor data:

| Direction | Axis |
|-----------|------|
| Forward   | X    |
| Backward  | -X   |
| Left      | -Y   |
| Right     | X    |
| Up        | Z    |
| Down      | -Z   |

Rotation values are in degrees and are labeled roll, pitch, and yaw:

| Rotation | Axis |
|----------|------|
| Roll     | X    |
| Pitch    | Y    |
| Yaw      | Z    |

**Axis Orientation:**
```
       Z (Up)
        |
        |
        |
        O------ Y (Right)
       /
      /
    -X (Backward)
```

### Logging Functions

You can configure logging based on your needs. The logging system provides flexibility to output logs either to the console or a file, and supports different log levels (`DEBUG`, `INFO`, `WARNING`, `ERROR`, `SUCCESS`).

```python
# Set log level to 'DEBUG'
bridge.set_log_level('DEBUG')

# Enable logging to file
bridge.set_write_to_file(True)
```

### Flight Matrix API

The core functionalities include retrieving frames, fetching sensor data, and sending movement commands.

#### Getting Frame Data

You can retrieve frames from both the left and right cameras. You also have access to depth and segmentation data.

```python
# Retrieve right camera frame
right_frame = bridge.get_right_frame()

# Retrieve left camera frame
left_frame = bridge.get_left_frame()

# Retrieve z-depth for the right camera
right_zdepth = bridge.get_right_zdepth()

# Retrieve segmentation frame for the left camera
left_seg = bridge.get_left_seg()
```

#### Fetching Sensor Data

The bridge allows real-time access to sensor data from the shared memory block. This data includes location, orientation, velocity, acceleration, and more.

```python
sensor_data = bridge.get_sensor_data()
print(sensor_data)
```

#### Sending Movement Commands

To send movement commands (position and orientation) to a system, use the `send_movement_command` method.

```python
# Send movement command (x, y, z, roll, pitch, yaw)
bridge.send_movement_command(1.0, 2.0, 3.0, 0.1, 0.2, 0.3)
```

### Resolution & Noise Configuration

You can adjust the frame resolution dynamically and control noise levels applied to sensor data.

```python
# Set a new resolution for frames
bridge.set_resolution(1280, 720)

# Set noise level for sensor data
bridge.set_noise_level(0.05)

# Enable or disable noise application
bridge.set_apply_noise(True)
```

### Timestamp Utilities

The `utilities.py` file provides functions to convert timestamps from milliseconds into human-readable formats and to `datetime` objects.

```python
from flightmatrix.utilities import timestamp2string, timestamp2datetime

# Convert timestamp to string
timestamp_string = timestamp2string(1633029600000)
print(timestamp_string)  # Output: '2021-10-01 00:00:00:000'

# Convert timestamp to datetime object
timestamp_dt = timestamp2datetime(1633029600000)
print(timestamp_dt)  # Output: datetime object in UTC
```

## Detailed Functionality

### Initialization

Upon initialization, the `FlightMatrixBridge` class sets up shared memory blocks for frames, sensor data, and movement commands. It also configures the resolution and frame shapes.

### Shared Memory Management

The shared memory blocks are initialized using `multiprocessing.shared_memory.SharedMemory`, providing fast, low-latency access to the data. Each memory block corresponds to specific data types like frames, sensor readings, or movement commands.

The memory block names and their associated data are defined in the `memory_names` dictionary within the `FlightMatrixBridge` class:

- `right_frame`: Stores the right camera frame.
- `left_frame`: Stores the left camera frame.
- `right_zdepth`: Z-depth map for the right camera.
- `left_zdepth`: Z-depth map for the left camera.
- `right_seg`: Segmentation data for the right camera.
- `left_seg`: Segmentation data for the left camera.
- `sensor_data`: Sensor data shared memory.
- `movement_command`: Memory block for sending movement commands.

### Frame Handling

Frames can be retrieved from the shared memory using the `_get_frame` method. The frames are stored as NumPy arrays and can be either 1-channel (grayscale) or 3-channel (RGB).

### Sensor Data Handling

The `get_sensor_data` method retrieves sensor readings from the shared memory. The sensor data includes:

- Location `(x, y, z)` in *centimeters*
- Orientation `(roll, pitch, yaw)` in *degrees*
- gyroscope `(x, y, z)` in *degrees per second*
- accelerometer `(x, y, z)` in *cm/s^2*
- Magnetometer readings `(x, y, z)` in *unit vector*
- LiDAR data `(LiDARForward, LiDARBackward, LiDARLeft, LiDARRight, LiDARBottom) or (Y, -Y, -X, X, -Z)` in *centimeters*
- Collision detection status `(True/False, LocationX, LocationY, LocationZ)` in *centimeters*
- Timestamp in *milliseconds*

### Movement Command Management

Movement commands are written to shared memory using `send_movement_command`. These commands include the position and orientation of the system and are stored as six floating-point values.

### Logging

The logging system is highly configurable and provides essential feedback about the system's operations. You can adjust the verbosity of the logs and decide whether to write them to a file.

### Noise Application

To simulate real-world noise in sensor data, noise can be added using Gaussian distribution. This feature is optional and can be enabled/disabled dynamically.

## Examples

### Example 1: Fetching Frames

```python
import cv2
from flightmatrix.bridge import FlightMatrixBridge
from flightmatrix.utilities import timestamp2string
import ultraprint.common as p

# Initialize the FlightMatrixBridge
bridge = FlightMatrixBridge()

# Start a loop to continuously fetch and display frames
while True:
    # Fetch the left and right frames
    left_frame_data = bridge.get_left_frame()
    right_frame_data = bridge.get_right_frame()

    # Fetch the z-depth frames for both left and right
    left_zdepth_data = bridge.get_left_zdepth()
    right_zdepth_data = bridge.get_right_zdepth()

    # Retrieve the actual frame arrays and timestamps
    left_frame = left_frame_data['frame']
    right_frame = right_frame_data['frame']

    left_zdepth = left_zdepth_data['frame']
    right_zdepth = right_zdepth_data['frame']
    
    left_timestamp = left_frame_data['timestamp']
    right_timestamp = right_frame_data['timestamp']

    # Convert timestamps to human-readable format
    left_timestamp = timestamp2string(left_timestamp)
    right_timestamp = timestamp2string(right_timestamp)

    # Display the frames in OpenCV windows
    cv2.imshow("Left Frame", left_frame)
    cv2.imshow("Right Frame", right_frame)

    cv2.imshow("Left Z-Depth", left_zdepth)
    cv2.imshow("Right Z-Depth", right_zdepth)

    # Print timestamps for each frame (optional)
    p.purple(f"Left Frame Timestamp: {left_timestamp}")
    p.purple(f"Right Frame Timestamp: {right_timestamp}")

    # Print timestamps for z-depth frames (optional)
    p.lgray(f"Left Z-Depth Timestamp: {left_timestamp}")
    p.lgray(f"Right Z-Depth Timestamp: {right_timestamp}")

    # Break the loop when 'q' is pressed
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

# Release OpenCV windows
cv2.destroyAllWindows()
```

### Example 2: Sending Movement Commands

```python
from flightmatrix.bridge import FlightMatrixBridge

# Initialize the bridge
bridge = FlightMatrixBridge()

# Send a movement command (x, y, z, roll, pitch, yaw)
bridge.send_movement_command(0.5, 1.0, 0.8, 0.0, 0.1, 0.2)
```

In order to reset/stop the movement, you can send a command with all zeros:

```python
bridge.send_movement_command(0.0, 0.0, 0.0, 0.0, 0.0, 0.0)
```

### Example 3: Fetching Sensor Data

```python
from flightmatrix.bridge import FlightMatrixBridge

# Initialize the bridge
bridge = FlightMatrixBridge(resolution=(1226, 370), noise_level=0.01, apply_noise=False)  # Set frame resolution (width, height), noise level, and noise application

# Fetch sensor data
sensor_data = bridge.get_sensor_data()

# Check for errors
if sensor_data.get('error'):
    print("Error fetching sensor data:", sensor_data['error'])
else:
    # Extract sensor readings
    location = sensor_data['location']
    orientation = sensor_data['orientation']
    gyroscope = sensor_data['gyroscope']
    accelerometer = sensor_data['accelerometer']
    magnetometer = sensor_data['magnetometer']
    lidar = sensor_data['lidar']
    collision = sensor_data['collision']
    timestamp = sensor_data['timestamp']

    # Display sensor data in a readable format
    print("Sensor Data:")
    print("-----------------------")
    print(f"Timestamp: {timestamp} ms")
    print(f"Location (cm): X={location[0]:.2f}, Y={location[1]:.2f}, Z={location[2]:.2f}")
    print(f"Orientation (degrees): Roll={orientation[0]:.2f}, Pitch={orientation[1]:.2f}, Yaw={orientation[2]:.2f}")
    print(f"Gyroscope (deg/s): X={gyroscope[0]:.2f}, Y={gyroscope[1]:.2f}, Z={gyroscope[2]:.2f}")
    print(f"Accelerometer (cm/s²): X={accelerometer[0]:.2f}, Y={accelerometer[1]:.2f}, Z={accelerometer[2]:.2f}")
    print(f"Magnetometer (unit vector): X={magnetometer[0]:.2f}, Y={magnetometer[1]:.2f}, Z={magnetometer[2]:.2f}")
    print(f"LiDAR Data (cm): Forward={lidar[0]:.2f}, Backward={lidar[1]:.2f}, Left={lidar[2]:.2f}, Right={lidar[3]:.2f}, Bottom={lidar[4]:.2f}")
    print(f"Collision Detection: Status={collision[0]}, Location (cm): X={collision[1]:.2f}, Y={collision[2]:.2f}, Z={collision[3]:.2f}")

```

### Example 4: Drone Controller

```python

from flightmatrix.bridge import FlightMatrixBridge
from flightmatrix.utilities import DroneController

# Example Usage
bridge = FlightMatrixBridge()
drone = DroneController(bridge)

# Move forward by 1.0 (positive y-axis)
drone.move_forward(1.0)

# Ascend by 0.5 (positive z-axis)
drone.ascend(0.5)

# Rotate in yaw by 0.3
drone.rotate_yaw(0.3)

# Stop only rotation (keep movement intact)
drone.stop_rotation()

# Stop all movement and rotation
drone.stop()

# Hover in place and rotate at 0.5 speed for 5 seconds
drone.hover_and_rotate(0.5, 5)
  
```

### Example 5: Data Recorder

```python

from flightmatrix.bridge import FlightMatrixBridge
from flightmatrix.utilities import DataRecorder
import time

# Example usage (Record data each second for 120 seconds)
if __name__ == "__main__":
    bridge = FlightMatrixBridge()
    recorder = DataRecorder(bridge, base_dir="Sample_Recordings", 
                            record_left_frame=True, 
                            record_right_frame=True, 
                            record_left_zdepth=True, 
                            record_right_zdepth=True, 
                            record_left_seg=True, 
                            record_right_seg=True, 
                            record_sensor_data=True,
                            record_sensor_data_interval=1)
    
    recorder.start_recording()

    time.sleep(120)  # Record for 120 seconds

    recorder.stop_recording()
```

## Documentation

#### Class: `FlightMatrixBridge`
This class interfaces with the Flight Matrix system using shared memory for inter-process communication. It manages frames, timestamps, and movement commands, enabling seamless data sharing between processes.

---

##### **Attributes:**

- `width (int)`: The width of the frame, initialized by the resolution provided.
  
- `height (int)`: The height of the frame, initialized by the resolution provided.

- `frame_shape (tuple)`: Tuple representing the shape of the frame as `(height, width)`.

- `frame_shape_3ch (tuple)`: Tuple representing the shape of the frame with 3 channels as `(height, width, 3)`.

- `noise_level (float)`: Specifies the level of noise to be applied. Defaults to `0.01`.

- `apply_noise (bool)`: Boolean flag that determines whether noise should be applied. Defaults to `False`.

- `memory_names (dict)`: Dictionary mapping keys to shared memory block names. Used for storing frame, depth, segmentation, and movement command data.

- `log (Logger)`: A logger instance used for logging events and debugging messages.

- `shm (dict)`: Dictionary storing the shared memory objects for frame data.

- `shm_timestamps (dict)`: Dictionary storing the shared memory objects for timestamps.

- `num_floats (int)`: Number of float values stored in shared memory for movement commands. Defaults to `6`. Do not edit this value.

---

##### **Methods:**

---

###### **`__init__(self, resolution=(1226, 370), noise_level=0.01, apply_noise=False)`**

**Description:**  
Initializes the `FlightMatrixBridge` class by setting up shared memory, logging, and configuring noise settings.

**Args:**  
- `resolution (tuple, optional)`: A tuple specifying the frame's width and height. Defaults to `(1226, 370)`.
- `noise_level (float, optional)`: Specifies the level of noise to be applied to sensor data. Defaults to `0.01`.
- `apply_noise (bool, optional)`: Boolean flag that determines whether noise should be applied to sensor data. Defaults to `False`.

**Example:**
```python
bridge = FlightMatrixBridge(resolution=(800, 600), noise_level=0.05, apply_noise=True)
```

---

###### **`set_log_level(self, log_level='INFO')`**

**Description:**  
Sets the logging level for the logger instance to control the verbosity of log output.

**Args:**  
- `log_level (str)`: Desired log level (`'DEBUG'`, `'INFO'`, `'WARNING'`, `'ERROR'`). Default is `'INFO'`.

**Returns:**  
None.

**Example:**
```python
bridge.set_log_level('DEBUG')
```

---

###### **`set_write_to_file(self, write_to_file)`**

**Description:**  
Sets whether the logging should be written to a file or not.

**Args:**  
- `write_to_file (bool)`: If `True`, log messages will be written to a file; otherwise, they won't.

**Returns:**  
None.

**Example:**
```python
bridge.set_write_to_file(True)
```

---

###### **`_initialize_shared_memory(self)`**

**Description:**  
Initializes shared memory blocks for frames and timestamps based on the keys stored in `memory_names`. If the shared memory block for a specific key is not available, a warning will be logged.

**Raises:**  
- `FileNotFoundError`: If the shared memory block for a key does not exist.

**Returns:**  
None.

**Example:**
```python
bridge._initialize_shared_memory()
```

---

###### **`_initialize_movement_command_memory(self)`**

**Description:**  
Sets up shared memory for movement commands (`x, y, z, roll, pitch, yaw`) and an availability flag. If the shared memory block exists, it will attach to it; otherwise, it will create a new block.

**Raises:**  
- `FileExistsError`: If the shared memory block already exists when trying to create it.

**Returns:**  
None.

**Example:**
```python
bridge._initialize_movement_command_memory()
```

---

###### **`_get_frame(self, key, channels=3)`**

**Description:**  
Retrieves a frame from shared memory. Handles both 3-channel and single-channel frame retrieval.

**Args:**  
- `key (str)`: Key identifying the shared memory segment.
- `channels (int, optional)`: Number of channels in the frame, default is `3`.

**Returns:**  
- `dict`: A dictionary with:
  - `'frame' (np.ndarray or None)`: The retrieved frame or `None` if an error occurred.
  - `'timestamp' (any or None)`: The timestamp associated with the frame or `None` if an error occurred.
  - `'error' (str or None)`: Error message, if any.

**Raises:**  
- `Warning`: If shared memory is not available or if there is a resolution mismatch.

**Example:**
```python
frame_data = bridge._get_frame('right_frame', channels=3)
```

---

###### **`_get_timestamp(self, key)`**

**Description:**  
Retrieves the timestamp associated with the frame stored in shared memory.

**Args:**  
- `key (str)`: Key identifying the shared memory segment for the timestamp.

**Returns:**  
- `int or None`: The timestamp as an integer, or `None` if not available.

**Example:**
```python
timestamp = bridge._get_timestamp('right_frame')
```

---

###### **`add_noise(self, data)`**

**Description:**  
Adds Gaussian noise to the given data based on the configured noise level.

**Args:**  
- `data (np.ndarray)`: The data (typically a frame) to which noise will be added.

**Returns:**  
- `np.ndarray`: The noisy data.

**Example:**
```python
noisy_frame = bridge.add_noise(frame_data)
```

---

###### **`get_sensor_data(self)`**

**Description:**  
Retrieves sensor data from shared memory and returns it as a dictionary.  
If the sensor data is not available in shared memory, a warning is logged,  
and a dictionary with all sensor fields set to None and an error message is returned.  
The sensor data includes:
- location: 3 floats representing the location coordinates.
- orientation: 3 floats representing the orientation.
- gyroscope: 3 floats representing the gyroscope readings.
- accelerometer: 3 floats representing the accelerometer readings.
- magnetometer: 3 floats representing the magnetometer readings.
- lidar: 5 floats representing the lidar readings.
- collision: 4 floats representing the collision data.
- timestamp: The timestamp of the sensor data.

If noise application is enabled, noise is added to the gyroscope, accelerometer,  
magnetometer, and lidar data.

**Returns:**  
- `dict`: A dictionary containing the sensor data or an error message if the data is not available.

**Example:**
```python
sensor_data = bridge.get_sensor_data()
```

---

###### **`send_movement_command(self, x, y, z, roll, pitch, yaw)`**

**Description:**  
Sends movement command values (`x, y, z, roll, pitch, yaw`) to the shared memory block.

**Args:**  
- `x (float)`: Movement in the X-axis.
- `y (float)`: Movement in the Y-axis.
- `z (float)`: Movement in the Z-axis.
- `roll (float)`: Roll rotation.
- `pitch (float)`: Pitch rotation.
- `yaw (float)`: Yaw rotation.

**Returns:**  
None.

**Example:**
```python
bridge.send_movement_command(1.0, 0.5, -1.0, 0.2, 0.1, -0.3)
```

---

###### **`_write_movement_command(self, commands)`**

**Description:**  
Writes the movement commands to shared memory.

**Args:**  
- `commands (list of float)`: List of movement command values (`[x, y, z, roll, pitch, yaw]`).

**Returns:**  
None.

**Example:**
```python
bridge._write_movement_command([1.0, 0.5, -1.0, 0.2, 0.1, -0.3])
```

---

###### **`set_resolution(self, width, height)`**

**Description:**  
Sets the resolution of the frames by updating the `width` and `height` attributes and recalculating the frame shapes.

**Args:**  
- `width (int)`: Width of the frames.
- `height (int)`: Height of the frames.

**Returns:**  
None.

**Example:**
```python
bridge.set_resolution(800, 600)
```

---

###### **`set_noise_level(self, noise_level)`**

**Description:**  
Sets the noise level for the frames.

**Args:**  
- `noise_level (float)`: The level of noise to apply.

**Returns:**  
None.

**Example:**
```python
bridge.set_noise_level(0.05)
```

---

###### **`set_apply_noise(self, apply_noise)`**

**Description:**  
Sets whether noise should be applied to frames.

**Args:**  
- `apply_noise (bool)`: Whether to apply noise (`True` or `False`).

**Returns:**  
None.

**Example:**
```python
bridge.set_apply_noise(True)
```

---

###### **`get_right_frame(self)`**

**Description:**  
Retrieves the right frame from shared memory.

**Returns:**  
- `dict`: A dictionary with:
  - `'frame' (np.ndarray or None)`: The retrieved right frame or `None` if an error occurred.
  - `'timestamp' (int or None)`: The timestamp associated with the right frame or `None` if an error occurred.
  - `'error' (str or None)`: Error message, if any.

**Example:**
```python
right_frame_data = bridge.get_right_frame()
```

---

###### **`get_left_frame(self)`**

**Description:**  
Retrieves the left frame from shared memory.

**Returns:**  
- `dict`: A dictionary with:
  - `'frame' (np.ndarray or None)`: The retrieved left frame or `None` if an error occurred.
  - `'timestamp' (int or None)`: The timestamp associated with the left frame or `None` if an error occurred.
  - `'error' (str or None)`: Error message, if any.

**Example:**
```python
left_frame_data = bridge.get_left_frame()
```

---

###### **`get_right_zdepth(self)`**

**Description:**  
Retrieves the right depth frame from shared memory.

**Returns:**  
- `dict`: A dictionary with:
  - `'frame' (np.ndarray or None)`: The retrieved right depth frame or `None` if an error occurred.
  - `'timestamp' (int or None)`: The timestamp associated with the right depth frame or `None` if an error occurred.
  - `'error' (str or None)`: Error message, if any.

**Example:**
```python
right_zdepth_data = bridge.get_right_zdepth()
```

---

###### **`get_left_zdepth(self)`**

**Description:**  
Retrieves the left depth frame from shared memory.

**Returns:**  
- `dict`: A dictionary with:
  - `'frame' (np.ndarray or None)`: The retrieved left depth frame or `None` if an error occurred.
  - `'timestamp' (int or None)`: The timestamp associated with the left depth frame or `None` if an error occurred.
  - `'error' (str or None)`: Error message, if any.

**Example:**
```python
left_zdepth_data = bridge.get_left_zdepth()
```

---

###### **`get_right_seg(self)`**

**Description:**  
Retrieves the right segmentation frame from shared memory.

**Returns:**  
- `dict`: A dictionary with:
  - `'frame' (np.ndarray or None)`: The retrieved right segmentation frame or `None` if an error occurred.
  - `'timestamp' (int or None)`: The timestamp associated with the right segmentation frame or `None` if an error occurred.
  - `'error' (str or None)`: Error message, if any.

**Example:**
```python
right_segmentation_data = bridge.get_right_seg()
```

---

###### **`get_left_seg(self)`**

**Description:**  
Retrieves the left segmentation frame from shared memory.

**Returns:**  
- `dict`: A dictionary with:
  - `'frame' (np.ndarray or None)`: The retrieved left segmentation frame or `None` if an error occurred.
  - `'timestamp' (int or None)`: The timestamp associated with the left segmentation frame or `None` if an error occurred.
  - `'error' (str or None)`: Error message, if any.

**Example:**
```python
left_segmentation_data = bridge.get_left_seg()
```

---

#### 2. Utilities
   
##### 1. **`timestamp2string`**

**Description:**  
Converts a timestamp in milliseconds to a human-readable string format.

**Args:**  
- `timestamp (int)`: The timestamp in milliseconds.

**Returns:**  
- `str`: Formatted timestamp as a string in the format 'YYYY-MM-DD HH:MM:SS:fff'.

**Example:**
```python
formatted_time = timestamp2string(1609459200000)
# Output: '2021-01-01 00:00:00:000'
```

---

##### 2. **`timestamp2datetime`**

**Description:**  
Converts a timestamp in milliseconds to a `datetime` object in UTC.

**Args:**  
- `timestamp (int)`: The timestamp in milliseconds.

**Returns:**  
- `datetime`: The corresponding `datetime` object in UTC.

**Example:**
```python
datetime_obj = timestamp2datetime(1609459200000)
# Output: datetime(2021, 1, 1, 0, 0, 0, tzinfo=timezone.utc)
```

---

#### 3. `cartesian_to_gps`

**Description:**  
Converts Cartesian coordinates to GPS coordinates (latitude, longitude, altitude).

**Args:**  
- `x (float)`: X coordinate in centimeters.
- `y (float)`: Y coordinate in centimeters.
- `z (float)`: Z coordinate in centimeters.
- `origin_lat (float, optional)`: Latitude of the origin point in degrees. Defaults to 22.583047.
- `origin_lon (float, optional)`: Longitude of the origin point in degrees. Defaults to 88.45859783333334.
- `origin_alt (float, optional)`: Altitude of the origin point in meters. Defaults to 0.
- `add_noise (bool, optional)`: Whether to add noise to the GPS coordinates. Defaults to False.
- `lat_long_noise_amt (float, optional)`: Amount of noise to add to latitude and longitude. Defaults to 0.0001.
- `alt_noise_amt (float, optional)`: Amount of noise to add to altitude. Defaults to 0.1.
- `earth_radius (float, optional)`: Radius of the Earth in meters. Defaults to 6378137 (meters).

**Returns:**  
- `tuple`: A tuple containing the latitude, longitude, and altitude in meters.

**Example:**
```python
latitude, longitude, altitude = cartesian_to_gps(1000, 2000, 300)
```

---

#### Class: `DroneController`
This class provides an interface to control the drone's movements by sending commands to the flight matrix system. It allows the drone to move along the x, y, and z axes and rotate around the roll, pitch, and yaw axes.

---

##### **Attributes:**

- `bridge (FlightMatrixBridge)`: The bridge object used to communicate with the drone.
- `current_x (float)`: Current x-coordinate position, initialized to `0.0`.
- `current_y (float)`: Current y-coordinate position, initialized to `0.0`.
- `current_z (float)`: Current z-coordinate position, initialized to `0.0`.
- `current_roll (float)`: Current roll angle, initialized to `0.0`.
- `current_pitch (float)`: Current pitch angle, initialized to `0.0`.
- `current_yaw (float)`: Current yaw angle, initialized to `0.0`.

---

##### **Methods:**

###### **`__init__(self, bridge_object: FlightMatrixBridge)`**

**Description:**  
Initializes the `DroneController` class by linking it to a `FlightMatrixBridge` object and setting the initial drone movement parameters to zero.

**Args:**  
- `bridge_object (FlightMatrixBridge)`: An instance of `FlightMatrixBridge` for communication with the flight matrix system.

**Example:**  
```python
bridge = FlightMatrixBridge()
drone_controller = DroneController(bridge)
```

---

###### **`_send_command(self)`**

**Description:**  
Sends the current positional and rotational state (x, y, z, roll, pitch, yaw) as movement commands to the drone.

**Returns:**  
None

**Example:**  
```python
drone_controller._send_command()
```

---

###### **`move_x(self, value)`**

**Description:**  
Moves the drone to a specified x-coordinate.

**Args:**  
- `value (float)`: The x-coordinate to move to.

**Returns:**  
None

**Example:**  
```python
drone_controller.move_x(10.5)  # Move drone to x = 10.5
```

---

###### **`move_y(self, value)`**

**Description:**  
Moves the drone to a specified y-coordinate (left or right).

**Args:**  
- `value (float)`: The y-coordinate to move to.

**Returns:**  
None

**Example:**  
```python
drone_controller.move_y(-5.2)  # Move drone to y = -5.2
```

---

###### **`move_z(self, value)`**

**Description:**  
Moves the drone to a specified z-coordinate (up or down).

**Args:**  
- `value (float)`: The z-coordinate to move to.

**Returns:**  
None

**Example:**  
```python
drone_controller.move_z(15.0)  # Move drone up to z = 15.0
```

---

###### **`rotate_roll(self, value)`**

**Description:**  
Rotates the drone to a specified roll angle.

**Args:**  
- `value (float)`: The roll angle to rotate to.

**Returns:**  
None

**Example:**  
```python
drone_controller.rotate_roll(30.0)  # Rotate drone to a roll angle of 30 degrees
```

---

###### **`rotate_pitch(self, value)`**

**Description:**  
Rotates the drone to a specified pitch angle.

**Args:**  
- `value (float)`: The pitch angle to rotate to.

**Returns:**  
None

**Example:**  
```python
drone_controller.rotate_pitch(-15.0)  # Rotate drone to a pitch angle of -15 degrees
```

---

###### **`rotate_yaw(self, value)`**

**Description:**  
Rotates the drone to a specified yaw angle.

**Args:**  
- `value (float)`: The yaw angle to rotate to, in degrees.

**Returns:**  
None

**Example:**  
```python
drone_controller.rotate_yaw(90.0)  # Rotate drone to a yaw angle of 90 degrees
```

---

###### **`ascend(self, value)`**

**Description:**  
Ascends the drone by a specified value, increasing the current altitude.

**Args:**  
- `value (float)`: The amount to increase the altitude.

**Returns:**  
None

**Example:**  
```python
drone_controller.ascend(5.0)  # Ascend drone by 5 units
```

---

###### **`descend(self, value)`**

**Description:**  
Descends the drone by a specified value, decreasing the current altitude.

**Args:**  
- `value (float)`: The amount to decrease the altitude.

**Returns:**  
None

**Example:**  
```python
drone_controller.descend(3.0)  # Descend drone by 3 units
```

---

###### **`move_forward(self, value)`**

**Description:**  
Moves the drone forward by a specified value (positive y-axis).

**Args:**  
- `value (float)`: The amount to move forward.

**Returns:**  
None

**Example:**  
```python
drone_controller.move_forward(10.0)  # Move drone forward by 10 units
```

---

###### **`move_backward(self, value)`**

**Description:**  
Moves the drone backward by a specified value (negative y-axis).

**Args:**  
- `value (float)`: The amount to move backward.

**Returns:**  
None

**Example:**  
```python
drone_controller.move_backward(8.0)  # Move drone backward by 8 units
```

---

###### **`stop_movement(self)`**

**Description:**  
Stops all drone movements on the x, y, and z axes.

**Returns:**  
None

**Example:**  
```python
drone_controller.stop_movement()  # Stop all drone movements
```

---

#### Class: `DataRecorder`
The `DataRecorder` class is designed to record various types of data from a drone or robotic system using the FlightMatrix framework. It can capture visual frames, z-depth images, segmentation frames, and sensor data, all of which are stored in a structured manner for later analysis.

---

##### **Attributes:**

- `bridge (FlightMatrixBridge)`: The bridge object used to interface with the drone or robot's systems.
- `base_dir (str)`: The base directory where all recorded data will be stored.
- `record_left_frame (bool)`: Flag indicating whether to record the left visual frame (default: `False`).
- `record_right_frame (bool)`: Flag indicating whether to record the right visual frame (default: `False`).
- `record_left_zdepth (bool)`: Flag indicating whether to record the left z-depth frame (default: `False`).
- `record_right_zdepth (bool)`: Flag indicating whether to record the right z-depth frame (default: `False`).
- `record_left_seg (bool)`: Flag indicating whether to record the left segmentation frame (default: `False`).
- `record_right_seg (bool)`: Flag indicating whether to record the right segmentation frame (default: `False`).
- `record_sensor_data (bool)`: Flag indicating whether to record sensor data (default: `False`).
- `sensor_data_interval (float)`: The interval at which sensor data is recorded (default: `0.1` seconds).
- `threads (list)`: List to hold the thread objects for recording data.
- `stop_event (Event)`: Event used to signal the threads to stop.

---

##### **Methods:**

###### **`__init__(self, bridge: FlightMatrixBridge, base_dir: str, record_left_frame: bool = False, record_right_frame: bool = False, record_left_zdepth: bool = False, record_right_zdepth: bool = False, record_left_seg: bool = False, record_right_seg: bool = False, record_sensor_data: bool = False, record_sensor_data_interval: float = 0.1)`**

**Description:**  
Initializes the `DataRecorder` class with specified options for recording. It sets up directories for storing the recorded data based on user selections.

**Args:**  
- `bridge (FlightMatrixBridge)`: An instance of `FlightMatrixBridge` used to interact with the drone/robot.
- `base_dir (str)`: The directory to store recorded files.
- `record_left_frame (bool)`: If `True`, records the left visual frame.
- `record_right_frame (bool)`: If `True`, records the right visual frame.
- `record_left_zdepth (bool)`: If `True`, records the left z-depth frame.
- `record_right_zdepth (bool)`: If `True`, records the right z-depth frame.
- `record_left_seg (bool)`: If `True`, records the left segmentation frame.
- `record_right_seg (bool)`: If `True`, records the right segmentation frame.
- `record_sensor_data (bool)`: If `True`, records sensor data.
- `record_sensor_data_interval (float)`: Time interval for recording sensor data in seconds.

---

###### **`record_frames(self)`**

**Description:**  
Continuously captures and saves visual frames, z-depth frames, and segmentation frames until the recording is stopped. Each frame is saved with a timestamped filename.

---

###### **`record_sensors(self)`**

**Description:**  
Records sensor data at specified intervals, saving the readings to a CSV file. It checks for errors in the sensor data and handles them appropriately.

---

###### **`start_recording(self)`**

**Description:**  
Starts the recording process by launching separate threads for recording frames and sensor data, based on the user’s selections.

---

###### **`stop_recording(self)`**

**Description:**  
Stops the recording process by signaling the threads to finish and waits for them to join back.

---

##### **Example Usage:**

```python
if __name__ == "__main__":
    bridge = FlightMatrixBridge()
    recorder = DataRecorder(bridge, base_dir="Sample_Recordings", 
                            record_left_frame=True, 
                            record_right_frame=True, 
                            record_left_zdepth=True, 
                            record_right_zdepth=True, 
                            record_left_seg=True, 
                            record_right_seg=True, 
                            record_sensor_data=True,
                            record_sensor_data_interval=1)
    
    recorder.start_recording()

    time.sleep(10)  # Record for 10 seconds

    recorder.stop_recording()
```

###### **`is_recording_on(self)`**

**Description:**
Checks if the recording process is currently active.

**Returns:**
- `bool`: `True` if recording is active, `False` otherwise.

---

## Data Streaming with DataStreamer

The `DataStreamer` class provides a convenient way to stream data from the Flight Matrix system using callbacks. It allows you to subscribe to specific data streams (e.g., left frame, sensor data) and specify callback functions that are called whenever new data is available. Each data stream runs in its own thread, enabling parallel data fetching without one stream waiting for the other. You can also specify the interval at which data is fetched or set it to zero for the fastest possible streaming.

### Features

- **Subscribe to Data Streams**: Subscribe to specific data types you are interested in.
- **Parallel Data Fetching**: Each data stream runs in its own thread for efficient parallel processing.
- **Custom Callbacks**: Provide your own callback functions to process the data as it arrives.
- **Adjustable Fetch Interval**: Control the rate at which data is fetched by specifying the interval.

### Usage

#### Import and Initialize

```python
from flightmatrix.bridge import FlightMatrixBridge
from flightmatrix.utilities import DataStreamer

# Initialize the FlightMatrixBridge
bridge = FlightMatrixBridge()

# Initialize the DataStreamer
streamer = DataStreamer(bridge)
```

#### Subscribe to Data Streams

```python

import cv2

# Define a callback function for left frame data
def left_frame_callback(frame_data):
    frame = frame_data['frame']
    timestamp = frame_data['timestamp']
    cv2.imshow("Left Frame", frame)
    print("Left Frame Timestamp:", timestamp)

# Subscribe to the left frame data stream
streamer.subscribe("left_frame", left_frame_callback)

# Define a callback function for sensor data
def sensor_data_callback(sensor_data):
    print("Sensor Data:", sensor_data)

# Subscribe to the sensor data stream
streamer.subscribe("sensor_data", sensor_data_callback)
```

### Available Data Streams

- **left_frame**: Left visual frame data.
- **right_frame**: Right visual frame data.
- **left_zdepth**: Left z-depth frame data.
- **right_zdepth**: Right z-depth frame data.
- **left_seg**: Left segmentation frame data.
- **right_seg**: Right segmentation frame data.
- **sensor_data**: Sensor data from the drone or robot.

Use the `subscribe` method to subscribe to the desired data streams and provide a callback function to process the data as it arrives. The callback function will be called with the data as an argument whenever new data is available. for example, the `left_frame_callback` function will be called with the left frame data whenever a new frame is available. 

### Subscribing to Data Streams

```python
import cv2
from flightmatrix.bridge import FlightMatrixBridge
from flightmatrix.utilities import DataStreamer

def left_frame_callback(left_frame_data):
    left_frame = left_frame_data['frame']
    left_timestamp = left_frame_data['timestamp']
    cv2.imshow("Left Frame", left_frame)

    # Break the loop when 'q' is pressed
    if cv2.waitKey(1) & 0xFF == ord('q'):
        cv2.destroyAllWindows()

    print("Left Frame Timestamp:", left_timestamp)

def right_frame_callback(right_frame_data):
    right_frame = right_frame_data['frame']
    right_timestamp = right_frame_data['timestamp']
    cv2.imshow("Right Frame", right_frame)

    # Break the loop when 'q' is pressed
    if cv2.waitKey(1) & 0xFF == ord('q'):
        cv2.destroyAllWindows()

    print("Right Frame Timestamp:", right_timestamp)

def left_zdepth_callback(left_zdepth_data):
    left_zdepth = left_zdepth_data['frame']
    left_timestamp = left_zdepth_data['timestamp']
    cv2.imshow("Left Z-Depth", left_zdepth)

    # Break the loop when 'q' is pressed
    if cv2.waitKey(1) & 0xFF == ord('q'):
        cv2.destroyAllWindows()

    print("Left Z-Depth Timestamp:", left_timestamp)

def right_zdepth_callback(right_zdepth_data):
    right_zdepth = right_zdepth_data['frame']
    right_timestamp = right_zdepth_data['timestamp']
    cv2.imshow("Right Z-Depth", right_zdepth)

    # Break the loop when 'q' is pressed
    if cv2.waitKey(1) & 0xFF == ord('q'):
        cv2.destroyAllWindows()

    print("Right Z-Depth Timestamp:", right_timestamp)

def left_seg_callback(left_seg_data):
    left_seg = left_seg_data['frame']
    left_timestamp = left_seg_data['timestamp']
    cv2.imshow("Left Segmentation", left_seg)

    # Break the loop when 'q' is pressed
    if cv2.waitKey(1) & 0xFF == ord('q'):
        cv2.destroyAllWindows()

    print("Left Segmentation Timestamp:", left_timestamp)

def right_seg_callback(right_seg_data):
    right_seg = right_seg_data['frame']
    right_timestamp = right_seg_data['timestamp']
    cv2.imshow("Right Segmentation", right_seg)

    # Break the loop when 'q' is pressed
    if cv2.waitKey(1) & 0xFF == ord('q'):
        cv2.destroyAllWindows()

    print("Right Segmentation Timestamp:", right_timestamp)

def sensor_data_callback(sensor_data):
    print("Sensor Data:", sensor_data)

# Initialize the FlightMatrixBridge
bridge = FlightMatrixBridge()

# Initialize the DataStreamer
streamer = DataStreamer(bridge)

# Subscribe to the left frame data stream
streamer.subscribe("left_frame", left_frame_callback, interval=0)

# Subscribe to the right frame data stream
streamer.subscribe("right_frame", right_frame_callback, interval=0.1)

# Subscribe to the left z-depth data stream
streamer.subscribe("left_zdepth", left_zdepth_callback, interval=0.1)

# Subscribe to the right z-depth data stream
streamer.subscribe("right_zdepth", right_zdepth_callback, interval=0.1)

# Subscribe to the left segmentation data stream
streamer.subscribe("left_seg", left_seg_callback, interval=0.1)

# Subscribe to the right segmentation data stream
streamer.subscribe("right_seg", right_seg_callback, interval=0.1)

# Subscribe to the sensor data stream
streamer.subscribe("sensor_data", sensor_data_callback, interval=0.1)
```

## Credits

This project was developed and maintained by [Ranit Bhowmick](https://www.linkedin.com/in/ranitbhowmick), a Robotics and Automation engineer with a passion for building innovative solutions in AI, game development, and full-stack projects. Specializing in advanced Python programming, machine learning, and robotics, I’m always open to collaboration and eager to explore new challenges.

I'd like to express my gratitude to the unreal engine community for their support and feedback. I'm always open to suggestions and contributions to improve this project further.

            

Raw data

            {
    "_id": null,
    "home_page": "https://github.com/Kawai-Senpai/Py-FlightMatrix-Bridge",
    "name": "flightmatrixbridge",
    "maintainer": null,
    "docs_url": null,
    "requires_python": null,
    "maintainer_email": null,
    "keywords": "Robotics, Inter-process communication, Shared memory, Data management",
    "author": "Ranit Bhowmick",
    "author_email": "bhowmickranitking@duck.com",
    "download_url": "https://files.pythonhosted.org/packages/28/ad/1078f80f1857e90460dc62d765545d8ebe69244337d4c56082c39ae6526d/flightmatrixbridge-1.7.0.tar.gz",
    "platform": null,
    "description": "# FlightMatrix Bridge \u00e2\u20ac\u00a2 API for Flight Matrix Simulation Software\r\n\r\n![FlightMatrix Bridge Cover](https://github.com/Kawai-Senpai/Py-FlightMatrix-Bridge/blob/0d434625dc45a12744eacaa4df21de29f0072612/Assets/FlightMatrix%20Cover.png)\r\n\r\n**FlightMatrix Bridge** is a Python-based API designed for controlling and fetching information, frames, and other data from Flight Matrix. This library enables efficient and real-time communication between various processes in a system, primarily designed for interfacing flight simulators, UAV systems, or other robotics platforms. It utilizes the `multiprocessing.shared_memory` module to share data such as frames, sensor data, and movement commands across multiple processes.\r\n\r\nDownload the software from [Flight Matrix](https://gamejolt.com/games/flightmatrix/933049).\r\n\r\nThe project is structured around two primary files:\r\n\r\n- **`bridge.py`**: Implements the core functionalities of shared memory management, handling frames, sensor data, movement commands, and logging.\r\n- **`utilities.py`**: Provides utility functions to handle timestamps and convert them into human-readable formats. It also includes a `DroneController` class for controlling drones.\r\n\r\n## Table of Contents\r\n\r\n1. [Introduction](#introduction)\r\n2. [Features](#features)\r\n3. [Controls](#controls)\r\n4. [Installation](#installation)\r\n5. [Usage](#usage)\r\n    1. [Initializing the FlightMatrixBridge](#initializing-the-flightmatrixbridge)\r\n    2. [Units](#units)\r\n    3. [Axis System](#axis-system)\r\n    4. [Logging Functions](#logging-functions)\r\n    5. [Flight Matrix API](#flight-matrix-api)\r\n        - [Getting Frame Data](#getting-frame-data)\r\n        - [Fetching Sensor Data](#fetching-sensor-data)\r\n        - [Sending Movement Commands](#sending-movement-commands)\r\n    6. [Resolution & Noise Configuration](#resolution--noise-configuration)\r\n    7. [Timestamp Utilities](#timestamp-utilities)\r\n6. [Detailed Functionality](#detailed-functionality)\r\n    - [Initialization](#initialization)\r\n    - [Shared Memory Management](#shared-memory-management)\r\n    - [Frame Handling](#frame-handling)\r\n    - [Sensor Data Handling](#sensor-data-handling)\r\n    - [Movement Command Management](#movement-command-management)\r\n    - [Logging](#logging)\r\n    - [Noise Application](#noise-application)\r\n7. [Examples](#examples)\r\n    1. [Example 1: Fetching Frames](#example-1-fetching-frames)\r\n    2. [Example 2: Sending Movement Commands](#example-2-sending-movement-commands)\r\n    3. [Example 3: Fetching Sensor Data](#example-3-fetching-sensor-data)\r\n    4. [Example 4: Drone Controller](#example-4-drone-controller)\r\n    5. [Example 5: Data Recorder](#example-5-data-recorder)\r\n8. [Documentation](#documentation)\r\n    1. [Class: `FlightMatrixBridge`](#class-flightmatrixbridge)\r\n        - [Attributes](#attributes)\r\n        - [Methods](#methods) \r\n    2. [Utilities](#utilities)\r\n        1. [`timestamp2string`](#timestamp2stringtimestamp)\r\n        2. [`timestamp2datetime`](#timestamp2datetimetimestamp)\r\n        3. [`cartesian_to_gps`](#cartesian_to_gps)\r\n    3. [Class: `DroneController`](#class-dronecontroller)\r\n          - [Attributes](#attributes)\r\n          - [Methods](#methods)\r\n    4. [Class: `DataRecorder`](#class-datarecorder)\r\n          - [Attributes](#attributes)\r\n          - [Methods](#methods)\r\n    5. [Class: `DataStreamer`](#data-streaming-with-datastreamer)\r\n9. [Credits](#credits)\r\n\r\n## Introduction\r\n\r\n![FlightMatrix Bridge](https://github.com/Kawai-Senpai/Py-FlightMatrix-Bridge/blob/020882b8df9d9452bcb43664452dadc9df753558/Assets/Screenshot%20(4).png)\r\n\r\nThe **FlightMatrixBridge** system is designed to bridge multiple processes that need to access shared memory for real-time communication. The typical use cases include flight simulators, robotics platforms, autonomous vehicles, and any application where sharing large datasets like frames or sensor readings between processes is essential.\r\n\r\nThis package provides:\r\n- An interface to retrieve frames and sensor data from shared memory.\r\n- The ability to send movement commands to be processed by another service.\r\n- Real-time noise application to sensor data.\r\n- Utilities to handle timestamps.\r\n\r\n## Features\r\n\r\n![Controllable Features](https://github.com/Kawai-Senpai/Py-FlightMatrix-Bridge/blob/020882b8df9d9452bcb43664452dadc9df753558/Assets/Screenshot%20(3).png)\r\n*Controllable Features*\r\n\r\n![Simulation Environments](https://github.com/Kawai-Senpai/Py-FlightMatrix-Bridge/blob/020882b8df9d9452bcb43664452dadc9df753558/Assets/Screenshot%20(1).png)\r\n*Simulation Environments*\r\n\r\n![Graphics Presets](https://github.com/Kawai-Senpai/Py-FlightMatrix-Bridge/blob/020882b8df9d9452bcb43664452dadc9df753558/Assets/Screenshot%20(2).png)\r\n*Graphics Presets*\r\n\r\nThe **FlightMatrix** software offers a range of features to facilitate real-time communication and data sharing between processes. Key features include:\r\n- **Dual Camera Support**: Flight Matrix is equipped with two cameras\u00e2\u20ac\u201dleft and right\u00e2\u20ac\u201dthat operate simultaneously. Each camera is capable of outputting high-quality RGB images, depth passes (z-depth), and segmentation maps, providing a comprehensive view of your simulated environment.\r\n- **Independent Camera Control**: Each camera can be controlled independently, allowing you to position them relative to the drone with precision. Adjust the x, y, z coordinates, as well as yaw, pitch, and roll to achieve the desired perspective.\r\n- **Variable Speed Control**: Control the speed of each axis and the rotation speed of the cameras, ensuring you can fine-tune the responsiveness to suit your simulation needs.\r\n- **Customizable Output**: Turn on and off various output maps as required. Control the resolution of the output frames and adjust the Field of View (FOV) to enhance your visual experience.\r\n- **Graphics Presets**: Choose from various graphics presets tailored for different simulation scenarios. Optimize the software\u00e2\u20ac\u2122s performance based on your hardware capabilities and desired visual fidelity.\r\n- **Diverse Simulation Environments**: Flight Matrix features a range of realistic maps, including architectural, natural, and ultra-realistic environments for authentic simulations. Navigate through intricate landscapes and urban settings as if you were flying in the real world.\r\n- **Human-like AI Characters**: Enhance your simulations with beautifully animated AI characters that simulate real crowds and human interactions. Observe how they behave and interact within the environment, adding depth to your scenarios.\r\n\r\nThe **FlightMatrixBridge** API provides a simple and efficient way to interact with the Flight Matrix simulation software, enabling you to access frames, sensor data, and movement commands in real-time. The API is designed to be easy to use and flexible, allowing you to integrate it into your projects seamlessly.\r\n- **Frame Management**: Retrieve left/right frames, z-depth maps, and segmentation frames in real-time.\r\n- **Sensor Data Access**: Retrieve real-time sensor data such as location, orientation, velocity, acceleration, magnetometer readings, and more.\r\n- **Movement Command Handling**: Send movement commands (position and orientation) for external systems to process.\r\n- **Noise Simulation**: Add configurable levels of noise to sensor data for testing robustness.\r\n- **Flexible Resolution Handling**: Easily set and adjust resolution for frames.\r\n- **Timestamp Management**: Convert timestamps into human-readable formats and handle system-wide timing data.\r\n\r\n## Controls\r\n\r\n| Action                          | Keyboard/Mouse          |\r\n|---------------------------------|-------------------------|\r\n| Move Forward                    | W                       |\r\n| Move Backward                   | S                       |\r\n| Move Left                       | A                       |\r\n| Move Right                      | D                       |\r\n| Ascend                          | Space Bar               |\r\n| Descend                         | Left Shift              |\r\n| Rotate (Yaw, Pitch, Roll)       | Arrow Keys              |\r\n| Move Left/Right                 | Q / E                   |\r\n| Pause                           | Escape / Pause / P      |\r\n| Exit                            | Escape / Pause / P      |\r\n| Spawn Human AI Character        | H                       |\r\n| Return to Starting Location     | R                       |\r\n\r\n## Installation\r\n\r\nDownload the software from [Flight Matrix](https://gamejolt.com/games/flightmatrix/933049).\r\n\r\nTo install the **FlightMatrixBridge (API)**, simply use pip:\r\n\r\n```bash\r\npip install flightmatrixbridge\r\n```\r\n\r\nMake sure your system has Python 3.8+ and supports the `multiprocessing.shared_memory` module.\r\n\r\n## Usage\r\n\r\n### Initializing the FlightMatrixBridge\r\n\r\nTo initialize and start using the **FlightMatrixBridge**, create an instance of the `FlightMatrixBridge` class and specify the resolution of the frames you want to handle:\r\n\r\n```python\r\nfrom flightmatrix.bridge import FlightMatrixBridge\r\n\r\nbridge = FlightMatrixBridge(resolution=(1226, 370), noise_level=0.01, apply_noise=False)  # Set frame resolution (width, height), noise level, and noise application\r\n```\r\n\r\n### Units\r\n\r\nThe system uses the following units for sensor data:\r\n- Length: centimeters (cm)\r\n- Angular values: degrees (\u00c2\u00b0)\r\n- Angular velocity/ gyroscope readings: degrees per second (\u00c2\u00b0/s)\r\n- Acceleration/ accelerometer readings: centimeters per second squared (cm/s\u00c2\u00b2) \r\n- Magnetometer readings: unit vector\r\n- LiDAR data: centimeters (cm)\r\n- Collision detection: centimeters (cm)\r\n- Timestamp: milliseconds (ms)\r\n\r\n### Axis System\r\n\r\nThe axis system differs slightly between the software interface and the API. Below is a detailed explanation for both.\r\n\r\n#### Inside the Software\r\n\r\nWhen adjusting camera positions or configuring movement multipliers within the software, the following axis system is used:\r\n\r\n| Direction | Axis |\r\n|-----------|------|\r\n| Forward   | Y    |\r\n| Backward  | -Y   |\r\n| Left      | -X   |\r\n| Right     | X    |\r\n| Up        | Z    |\r\n| Down      | -Z   |\r\n\r\nRotation values are in degrees and are labeled roll, pitch, and yaw:\r\n\r\n| Rotation | Axis |\r\n|----------|------|\r\n| Roll     | X    |\r\n| Pitch    | Y    |\r\n| Yaw      | Z    |\r\n\r\n**Axis Orientation:**\r\n```\r\n       Z (Up)\r\n        |\r\n        |\r\n        |\r\n        O------ X (Right)\r\n       /\r\n      /\r\n    -Y (Backward)\r\n```\r\n\r\n#### In the API\r\n\r\nThe API uses a different axis system for movement commands and sensor data:\r\n\r\n| Direction | Axis |\r\n|-----------|------|\r\n| Forward   | X    |\r\n| Backward  | -X   |\r\n| Left      | -Y   |\r\n| Right     | X    |\r\n| Up        | Z    |\r\n| Down      | -Z   |\r\n\r\nRotation values are in degrees and are labeled roll, pitch, and yaw:\r\n\r\n| Rotation | Axis |\r\n|----------|------|\r\n| Roll     | X    |\r\n| Pitch    | Y    |\r\n| Yaw      | Z    |\r\n\r\n**Axis Orientation:**\r\n```\r\n       Z (Up)\r\n        |\r\n        |\r\n        |\r\n        O------ Y (Right)\r\n       /\r\n      /\r\n    -X (Backward)\r\n```\r\n\r\n### Logging Functions\r\n\r\nYou can configure logging based on your needs. The logging system provides flexibility to output logs either to the console or a file, and supports different log levels (`DEBUG`, `INFO`, `WARNING`, `ERROR`, `SUCCESS`).\r\n\r\n```python\r\n# Set log level to 'DEBUG'\r\nbridge.set_log_level('DEBUG')\r\n\r\n# Enable logging to file\r\nbridge.set_write_to_file(True)\r\n```\r\n\r\n### Flight Matrix API\r\n\r\nThe core functionalities include retrieving frames, fetching sensor data, and sending movement commands.\r\n\r\n#### Getting Frame Data\r\n\r\nYou can retrieve frames from both the left and right cameras. You also have access to depth and segmentation data.\r\n\r\n```python\r\n# Retrieve right camera frame\r\nright_frame = bridge.get_right_frame()\r\n\r\n# Retrieve left camera frame\r\nleft_frame = bridge.get_left_frame()\r\n\r\n# Retrieve z-depth for the right camera\r\nright_zdepth = bridge.get_right_zdepth()\r\n\r\n# Retrieve segmentation frame for the left camera\r\nleft_seg = bridge.get_left_seg()\r\n```\r\n\r\n#### Fetching Sensor Data\r\n\r\nThe bridge allows real-time access to sensor data from the shared memory block. This data includes location, orientation, velocity, acceleration, and more.\r\n\r\n```python\r\nsensor_data = bridge.get_sensor_data()\r\nprint(sensor_data)\r\n```\r\n\r\n#### Sending Movement Commands\r\n\r\nTo send movement commands (position and orientation) to a system, use the `send_movement_command` method.\r\n\r\n```python\r\n# Send movement command (x, y, z, roll, pitch, yaw)\r\nbridge.send_movement_command(1.0, 2.0, 3.0, 0.1, 0.2, 0.3)\r\n```\r\n\r\n### Resolution & Noise Configuration\r\n\r\nYou can adjust the frame resolution dynamically and control noise levels applied to sensor data.\r\n\r\n```python\r\n# Set a new resolution for frames\r\nbridge.set_resolution(1280, 720)\r\n\r\n# Set noise level for sensor data\r\nbridge.set_noise_level(0.05)\r\n\r\n# Enable or disable noise application\r\nbridge.set_apply_noise(True)\r\n```\r\n\r\n### Timestamp Utilities\r\n\r\nThe `utilities.py` file provides functions to convert timestamps from milliseconds into human-readable formats and to `datetime` objects.\r\n\r\n```python\r\nfrom flightmatrix.utilities import timestamp2string, timestamp2datetime\r\n\r\n# Convert timestamp to string\r\ntimestamp_string = timestamp2string(1633029600000)\r\nprint(timestamp_string)  # Output: '2021-10-01 00:00:00:000'\r\n\r\n# Convert timestamp to datetime object\r\ntimestamp_dt = timestamp2datetime(1633029600000)\r\nprint(timestamp_dt)  # Output: datetime object in UTC\r\n```\r\n\r\n## Detailed Functionality\r\n\r\n### Initialization\r\n\r\nUpon initialization, the `FlightMatrixBridge` class sets up shared memory blocks for frames, sensor data, and movement commands. It also configures the resolution and frame shapes.\r\n\r\n### Shared Memory Management\r\n\r\nThe shared memory blocks are initialized using `multiprocessing.shared_memory.SharedMemory`, providing fast, low-latency access to the data. Each memory block corresponds to specific data types like frames, sensor readings, or movement commands.\r\n\r\nThe memory block names and their associated data are defined in the `memory_names` dictionary within the `FlightMatrixBridge` class:\r\n\r\n- `right_frame`: Stores the right camera frame.\r\n- `left_frame`: Stores the left camera frame.\r\n- `right_zdepth`: Z-depth map for the right camera.\r\n- `left_zdepth`: Z-depth map for the left camera.\r\n- `right_seg`: Segmentation data for the right camera.\r\n- `left_seg`: Segmentation data for the left camera.\r\n- `sensor_data`: Sensor data shared memory.\r\n- `movement_command`: Memory block for sending movement commands.\r\n\r\n### Frame Handling\r\n\r\nFrames can be retrieved from the shared memory using the `_get_frame` method. The frames are stored as NumPy arrays and can be either 1-channel (grayscale) or 3-channel (RGB).\r\n\r\n### Sensor Data Handling\r\n\r\nThe `get_sensor_data` method retrieves sensor readings from the shared memory. The sensor data includes:\r\n\r\n- Location `(x, y, z)` in *centimeters*\r\n- Orientation `(roll, pitch, yaw)` in *degrees*\r\n- gyroscope `(x, y, z)` in *degrees per second*\r\n- accelerometer `(x, y, z)` in *cm/s^2*\r\n- Magnetometer readings `(x, y, z)` in *unit vector*\r\n- LiDAR data `(LiDARForward, LiDARBackward, LiDARLeft, LiDARRight, LiDARBottom) or (Y, -Y, -X, X, -Z)` in *centimeters*\r\n- Collision detection status `(True/False, LocationX, LocationY, LocationZ)` in *centimeters*\r\n- Timestamp in *milliseconds*\r\n\r\n### Movement Command Management\r\n\r\nMovement commands are written to shared memory using `send_movement_command`. These commands include the position and orientation of the system and are stored as six floating-point values.\r\n\r\n### Logging\r\n\r\nThe logging system is highly configurable and provides essential feedback about the system's operations. You can adjust the verbosity of the logs and decide whether to write them to a file.\r\n\r\n### Noise Application\r\n\r\nTo simulate real-world noise in sensor data, noise can be added using Gaussian distribution. This feature is optional and can be enabled/disabled dynamically.\r\n\r\n## Examples\r\n\r\n### Example 1: Fetching Frames\r\n\r\n```python\r\nimport cv2\r\nfrom flightmatrix.bridge import FlightMatrixBridge\r\nfrom flightmatrix.utilities import timestamp2string\r\nimport ultraprint.common as p\r\n\r\n# Initialize the FlightMatrixBridge\r\nbridge = FlightMatrixBridge()\r\n\r\n# Start a loop to continuously fetch and display frames\r\nwhile True:\r\n    # Fetch the left and right frames\r\n    left_frame_data = bridge.get_left_frame()\r\n    right_frame_data = bridge.get_right_frame()\r\n\r\n    # Fetch the z-depth frames for both left and right\r\n    left_zdepth_data = bridge.get_left_zdepth()\r\n    right_zdepth_data = bridge.get_right_zdepth()\r\n\r\n    # Retrieve the actual frame arrays and timestamps\r\n    left_frame = left_frame_data['frame']\r\n    right_frame = right_frame_data['frame']\r\n\r\n    left_zdepth = left_zdepth_data['frame']\r\n    right_zdepth = right_zdepth_data['frame']\r\n    \r\n    left_timestamp = left_frame_data['timestamp']\r\n    right_timestamp = right_frame_data['timestamp']\r\n\r\n    # Convert timestamps to human-readable format\r\n    left_timestamp = timestamp2string(left_timestamp)\r\n    right_timestamp = timestamp2string(right_timestamp)\r\n\r\n    # Display the frames in OpenCV windows\r\n    cv2.imshow(\"Left Frame\", left_frame)\r\n    cv2.imshow(\"Right Frame\", right_frame)\r\n\r\n    cv2.imshow(\"Left Z-Depth\", left_zdepth)\r\n    cv2.imshow(\"Right Z-Depth\", right_zdepth)\r\n\r\n    # Print timestamps for each frame (optional)\r\n    p.purple(f\"Left Frame Timestamp: {left_timestamp}\")\r\n    p.purple(f\"Right Frame Timestamp: {right_timestamp}\")\r\n\r\n    # Print timestamps for z-depth frames (optional)\r\n    p.lgray(f\"Left Z-Depth Timestamp: {left_timestamp}\")\r\n    p.lgray(f\"Right Z-Depth Timestamp: {right_timestamp}\")\r\n\r\n    # Break the loop when 'q' is pressed\r\n    if cv2.waitKey(1) & 0xFF == ord('q'):\r\n        break\r\n\r\n# Release OpenCV windows\r\ncv2.destroyAllWindows()\r\n```\r\n\r\n### Example 2: Sending Movement Commands\r\n\r\n```python\r\nfrom flightmatrix.bridge import FlightMatrixBridge\r\n\r\n# Initialize the bridge\r\nbridge = FlightMatrixBridge()\r\n\r\n# Send a movement command (x, y, z, roll, pitch, yaw)\r\nbridge.send_movement_command(0.5, 1.0, 0.8, 0.0, 0.1, 0.2)\r\n```\r\n\r\nIn order to reset/stop the movement, you can send a command with all zeros:\r\n\r\n```python\r\nbridge.send_movement_command(0.0, 0.0, 0.0, 0.0, 0.0, 0.0)\r\n```\r\n\r\n### Example 3: Fetching Sensor Data\r\n\r\n```python\r\nfrom flightmatrix.bridge import FlightMatrixBridge\r\n\r\n# Initialize the bridge\r\nbridge = FlightMatrixBridge(resolution=(1226, 370), noise_level=0.01, apply_noise=False)  # Set frame resolution (width, height), noise level, and noise application\r\n\r\n# Fetch sensor data\r\nsensor_data = bridge.get_sensor_data()\r\n\r\n# Check for errors\r\nif sensor_data.get('error'):\r\n    print(\"Error fetching sensor data:\", sensor_data['error'])\r\nelse:\r\n    # Extract sensor readings\r\n    location = sensor_data['location']\r\n    orientation = sensor_data['orientation']\r\n    gyroscope = sensor_data['gyroscope']\r\n    accelerometer = sensor_data['accelerometer']\r\n    magnetometer = sensor_data['magnetometer']\r\n    lidar = sensor_data['lidar']\r\n    collision = sensor_data['collision']\r\n    timestamp = sensor_data['timestamp']\r\n\r\n    # Display sensor data in a readable format\r\n    print(\"Sensor Data:\")\r\n    print(\"-----------------------\")\r\n    print(f\"Timestamp: {timestamp} ms\")\r\n    print(f\"Location (cm): X={location[0]:.2f}, Y={location[1]:.2f}, Z={location[2]:.2f}\")\r\n    print(f\"Orientation (degrees): Roll={orientation[0]:.2f}, Pitch={orientation[1]:.2f}, Yaw={orientation[2]:.2f}\")\r\n    print(f\"Gyroscope (deg/s): X={gyroscope[0]:.2f}, Y={gyroscope[1]:.2f}, Z={gyroscope[2]:.2f}\")\r\n    print(f\"Accelerometer (cm/s\u00c2\u00b2): X={accelerometer[0]:.2f}, Y={accelerometer[1]:.2f}, Z={accelerometer[2]:.2f}\")\r\n    print(f\"Magnetometer (unit vector): X={magnetometer[0]:.2f}, Y={magnetometer[1]:.2f}, Z={magnetometer[2]:.2f}\")\r\n    print(f\"LiDAR Data (cm): Forward={lidar[0]:.2f}, Backward={lidar[1]:.2f}, Left={lidar[2]:.2f}, Right={lidar[3]:.2f}, Bottom={lidar[4]:.2f}\")\r\n    print(f\"Collision Detection: Status={collision[0]}, Location (cm): X={collision[1]:.2f}, Y={collision[2]:.2f}, Z={collision[3]:.2f}\")\r\n\r\n```\r\n\r\n### Example 4: Drone Controller\r\n\r\n```python\r\n\r\nfrom flightmatrix.bridge import FlightMatrixBridge\r\nfrom flightmatrix.utilities import DroneController\r\n\r\n# Example Usage\r\nbridge = FlightMatrixBridge()\r\ndrone = DroneController(bridge)\r\n\r\n# Move forward by 1.0 (positive y-axis)\r\ndrone.move_forward(1.0)\r\n\r\n# Ascend by 0.5 (positive z-axis)\r\ndrone.ascend(0.5)\r\n\r\n# Rotate in yaw by 0.3\r\ndrone.rotate_yaw(0.3)\r\n\r\n# Stop only rotation (keep movement intact)\r\ndrone.stop_rotation()\r\n\r\n# Stop all movement and rotation\r\ndrone.stop()\r\n\r\n# Hover in place and rotate at 0.5 speed for 5 seconds\r\ndrone.hover_and_rotate(0.5, 5)\r\n  \r\n```\r\n\r\n### Example 5: Data Recorder\r\n\r\n```python\r\n\r\nfrom flightmatrix.bridge import FlightMatrixBridge\r\nfrom flightmatrix.utilities import DataRecorder\r\nimport time\r\n\r\n# Example usage (Record data each second for 120 seconds)\r\nif __name__ == \"__main__\":\r\n    bridge = FlightMatrixBridge()\r\n    recorder = DataRecorder(bridge, base_dir=\"Sample_Recordings\", \r\n                            record_left_frame=True, \r\n                            record_right_frame=True, \r\n                            record_left_zdepth=True, \r\n                            record_right_zdepth=True, \r\n                            record_left_seg=True, \r\n                            record_right_seg=True, \r\n                            record_sensor_data=True,\r\n                            record_sensor_data_interval=1)\r\n    \r\n    recorder.start_recording()\r\n\r\n    time.sleep(120)  # Record for 120 seconds\r\n\r\n    recorder.stop_recording()\r\n```\r\n\r\n## Documentation\r\n\r\n#### Class: `FlightMatrixBridge`\r\nThis class interfaces with the Flight Matrix system using shared memory for inter-process communication. It manages frames, timestamps, and movement commands, enabling seamless data sharing between processes.\r\n\r\n---\r\n\r\n##### **Attributes:**\r\n\r\n- `width (int)`: The width of the frame, initialized by the resolution provided.\r\n  \r\n- `height (int)`: The height of the frame, initialized by the resolution provided.\r\n\r\n- `frame_shape (tuple)`: Tuple representing the shape of the frame as `(height, width)`.\r\n\r\n- `frame_shape_3ch (tuple)`: Tuple representing the shape of the frame with 3 channels as `(height, width, 3)`.\r\n\r\n- `noise_level (float)`: Specifies the level of noise to be applied. Defaults to `0.01`.\r\n\r\n- `apply_noise (bool)`: Boolean flag that determines whether noise should be applied. Defaults to `False`.\r\n\r\n- `memory_names (dict)`: Dictionary mapping keys to shared memory block names. Used for storing frame, depth, segmentation, and movement command data.\r\n\r\n- `log (Logger)`: A logger instance used for logging events and debugging messages.\r\n\r\n- `shm (dict)`: Dictionary storing the shared memory objects for frame data.\r\n\r\n- `shm_timestamps (dict)`: Dictionary storing the shared memory objects for timestamps.\r\n\r\n- `num_floats (int)`: Number of float values stored in shared memory for movement commands. Defaults to `6`. Do not edit this value.\r\n\r\n---\r\n\r\n##### **Methods:**\r\n\r\n---\r\n\r\n###### **`__init__(self, resolution=(1226, 370), noise_level=0.01, apply_noise=False)`**\r\n\r\n**Description:**  \r\nInitializes the `FlightMatrixBridge` class by setting up shared memory, logging, and configuring noise settings.\r\n\r\n**Args:**  \r\n- `resolution (tuple, optional)`: A tuple specifying the frame's width and height. Defaults to `(1226, 370)`.\r\n- `noise_level (float, optional)`: Specifies the level of noise to be applied to sensor data. Defaults to `0.01`.\r\n- `apply_noise (bool, optional)`: Boolean flag that determines whether noise should be applied to sensor data. Defaults to `False`.\r\n\r\n**Example:**\r\n```python\r\nbridge = FlightMatrixBridge(resolution=(800, 600), noise_level=0.05, apply_noise=True)\r\n```\r\n\r\n---\r\n\r\n###### **`set_log_level(self, log_level='INFO')`**\r\n\r\n**Description:**  \r\nSets the logging level for the logger instance to control the verbosity of log output.\r\n\r\n**Args:**  \r\n- `log_level (str)`: Desired log level (`'DEBUG'`, `'INFO'`, `'WARNING'`, `'ERROR'`). Default is `'INFO'`.\r\n\r\n**Returns:**  \r\nNone.\r\n\r\n**Example:**\r\n```python\r\nbridge.set_log_level('DEBUG')\r\n```\r\n\r\n---\r\n\r\n###### **`set_write_to_file(self, write_to_file)`**\r\n\r\n**Description:**  \r\nSets whether the logging should be written to a file or not.\r\n\r\n**Args:**  \r\n- `write_to_file (bool)`: If `True`, log messages will be written to a file; otherwise, they won't.\r\n\r\n**Returns:**  \r\nNone.\r\n\r\n**Example:**\r\n```python\r\nbridge.set_write_to_file(True)\r\n```\r\n\r\n---\r\n\r\n###### **`_initialize_shared_memory(self)`**\r\n\r\n**Description:**  \r\nInitializes shared memory blocks for frames and timestamps based on the keys stored in `memory_names`. If the shared memory block for a specific key is not available, a warning will be logged.\r\n\r\n**Raises:**  \r\n- `FileNotFoundError`: If the shared memory block for a key does not exist.\r\n\r\n**Returns:**  \r\nNone.\r\n\r\n**Example:**\r\n```python\r\nbridge._initialize_shared_memory()\r\n```\r\n\r\n---\r\n\r\n###### **`_initialize_movement_command_memory(self)`**\r\n\r\n**Description:**  \r\nSets up shared memory for movement commands (`x, y, z, roll, pitch, yaw`) and an availability flag. If the shared memory block exists, it will attach to it; otherwise, it will create a new block.\r\n\r\n**Raises:**  \r\n- `FileExistsError`: If the shared memory block already exists when trying to create it.\r\n\r\n**Returns:**  \r\nNone.\r\n\r\n**Example:**\r\n```python\r\nbridge._initialize_movement_command_memory()\r\n```\r\n\r\n---\r\n\r\n###### **`_get_frame(self, key, channels=3)`**\r\n\r\n**Description:**  \r\nRetrieves a frame from shared memory. Handles both 3-channel and single-channel frame retrieval.\r\n\r\n**Args:**  \r\n- `key (str)`: Key identifying the shared memory segment.\r\n- `channels (int, optional)`: Number of channels in the frame, default is `3`.\r\n\r\n**Returns:**  \r\n- `dict`: A dictionary with:\r\n  - `'frame' (np.ndarray or None)`: The retrieved frame or `None` if an error occurred.\r\n  - `'timestamp' (any or None)`: The timestamp associated with the frame or `None` if an error occurred.\r\n  - `'error' (str or None)`: Error message, if any.\r\n\r\n**Raises:**  \r\n- `Warning`: If shared memory is not available or if there is a resolution mismatch.\r\n\r\n**Example:**\r\n```python\r\nframe_data = bridge._get_frame('right_frame', channels=3)\r\n```\r\n\r\n---\r\n\r\n###### **`_get_timestamp(self, key)`**\r\n\r\n**Description:**  \r\nRetrieves the timestamp associated with the frame stored in shared memory.\r\n\r\n**Args:**  \r\n- `key (str)`: Key identifying the shared memory segment for the timestamp.\r\n\r\n**Returns:**  \r\n- `int or None`: The timestamp as an integer, or `None` if not available.\r\n\r\n**Example:**\r\n```python\r\ntimestamp = bridge._get_timestamp('right_frame')\r\n```\r\n\r\n---\r\n\r\n###### **`add_noise(self, data)`**\r\n\r\n**Description:**  \r\nAdds Gaussian noise to the given data based on the configured noise level.\r\n\r\n**Args:**  \r\n- `data (np.ndarray)`: The data (typically a frame) to which noise will be added.\r\n\r\n**Returns:**  \r\n- `np.ndarray`: The noisy data.\r\n\r\n**Example:**\r\n```python\r\nnoisy_frame = bridge.add_noise(frame_data)\r\n```\r\n\r\n---\r\n\r\n###### **`get_sensor_data(self)`**\r\n\r\n**Description:**  \r\nRetrieves sensor data from shared memory and returns it as a dictionary.  \r\nIf the sensor data is not available in shared memory, a warning is logged,  \r\nand a dictionary with all sensor fields set to None and an error message is returned.  \r\nThe sensor data includes:\r\n- location: 3 floats representing the location coordinates.\r\n- orientation: 3 floats representing the orientation.\r\n- gyroscope: 3 floats representing the gyroscope readings.\r\n- accelerometer: 3 floats representing the accelerometer readings.\r\n- magnetometer: 3 floats representing the magnetometer readings.\r\n- lidar: 5 floats representing the lidar readings.\r\n- collision: 4 floats representing the collision data.\r\n- timestamp: The timestamp of the sensor data.\r\n\r\nIf noise application is enabled, noise is added to the gyroscope, accelerometer,  \r\nmagnetometer, and lidar data.\r\n\r\n**Returns:**  \r\n- `dict`: A dictionary containing the sensor data or an error message if the data is not available.\r\n\r\n**Example:**\r\n```python\r\nsensor_data = bridge.get_sensor_data()\r\n```\r\n\r\n---\r\n\r\n###### **`send_movement_command(self, x, y, z, roll, pitch, yaw)`**\r\n\r\n**Description:**  \r\nSends movement command values (`x, y, z, roll, pitch, yaw`) to the shared memory block.\r\n\r\n**Args:**  \r\n- `x (float)`: Movement in the X-axis.\r\n- `y (float)`: Movement in the Y-axis.\r\n- `z (float)`: Movement in the Z-axis.\r\n- `roll (float)`: Roll rotation.\r\n- `pitch (float)`: Pitch rotation.\r\n- `yaw (float)`: Yaw rotation.\r\n\r\n**Returns:**  \r\nNone.\r\n\r\n**Example:**\r\n```python\r\nbridge.send_movement_command(1.0, 0.5, -1.0, 0.2, 0.1, -0.3)\r\n```\r\n\r\n---\r\n\r\n###### **`_write_movement_command(self, commands)`**\r\n\r\n**Description:**  \r\nWrites the movement commands to shared memory.\r\n\r\n**Args:**  \r\n- `commands (list of float)`: List of movement command values (`[x, y, z, roll, pitch, yaw]`).\r\n\r\n**Returns:**  \r\nNone.\r\n\r\n**Example:**\r\n```python\r\nbridge._write_movement_command([1.0, 0.5, -1.0, 0.2, 0.1, -0.3])\r\n```\r\n\r\n---\r\n\r\n###### **`set_resolution(self, width, height)`**\r\n\r\n**Description:**  \r\nSets the resolution of the frames by updating the `width` and `height` attributes and recalculating the frame shapes.\r\n\r\n**Args:**  \r\n- `width (int)`: Width of the frames.\r\n- `height (int)`: Height of the frames.\r\n\r\n**Returns:**  \r\nNone.\r\n\r\n**Example:**\r\n```python\r\nbridge.set_resolution(800, 600)\r\n```\r\n\r\n---\r\n\r\n###### **`set_noise_level(self, noise_level)`**\r\n\r\n**Description:**  \r\nSets the noise level for the frames.\r\n\r\n**Args:**  \r\n- `noise_level (float)`: The level of noise to apply.\r\n\r\n**Returns:**  \r\nNone.\r\n\r\n**Example:**\r\n```python\r\nbridge.set_noise_level(0.05)\r\n```\r\n\r\n---\r\n\r\n###### **`set_apply_noise(self, apply_noise)`**\r\n\r\n**Description:**  \r\nSets whether noise should be applied to frames.\r\n\r\n**Args:**  \r\n- `apply_noise (bool)`: Whether to apply noise (`True` or `False`).\r\n\r\n**Returns:**  \r\nNone.\r\n\r\n**Example:**\r\n```python\r\nbridge.set_apply_noise(True)\r\n```\r\n\r\n---\r\n\r\n###### **`get_right_frame(self)`**\r\n\r\n**Description:**  \r\nRetrieves the right frame from shared memory.\r\n\r\n**Returns:**  \r\n- `dict`: A dictionary with:\r\n  - `'frame' (np.ndarray or None)`: The retrieved right frame or `None` if an error occurred.\r\n  - `'timestamp' (int or None)`: The timestamp associated with the right frame or `None` if an error occurred.\r\n  - `'error' (str or None)`: Error message, if any.\r\n\r\n**Example:**\r\n```python\r\nright_frame_data = bridge.get_right_frame()\r\n```\r\n\r\n---\r\n\r\n###### **`get_left_frame(self)`**\r\n\r\n**Description:**  \r\nRetrieves the left frame from shared memory.\r\n\r\n**Returns:**  \r\n- `dict`: A dictionary with:\r\n  - `'frame' (np.ndarray or None)`: The retrieved left frame or `None` if an error occurred.\r\n  - `'timestamp' (int or None)`: The timestamp associated with the left frame or `None` if an error occurred.\r\n  - `'error' (str or None)`: Error message, if any.\r\n\r\n**Example:**\r\n```python\r\nleft_frame_data = bridge.get_left_frame()\r\n```\r\n\r\n---\r\n\r\n###### **`get_right_zdepth(self)`**\r\n\r\n**Description:**  \r\nRetrieves the right depth frame from shared memory.\r\n\r\n**Returns:**  \r\n- `dict`: A dictionary with:\r\n  - `'frame' (np.ndarray or None)`: The retrieved right depth frame or `None` if an error occurred.\r\n  - `'timestamp' (int or None)`: The timestamp associated with the right depth frame or `None` if an error occurred.\r\n  - `'error' (str or None)`: Error message, if any.\r\n\r\n**Example:**\r\n```python\r\nright_zdepth_data = bridge.get_right_zdepth()\r\n```\r\n\r\n---\r\n\r\n###### **`get_left_zdepth(self)`**\r\n\r\n**Description:**  \r\nRetrieves the left depth frame from shared memory.\r\n\r\n**Returns:**  \r\n- `dict`: A dictionary with:\r\n  - `'frame' (np.ndarray or None)`: The retrieved left depth frame or `None` if an error occurred.\r\n  - `'timestamp' (int or None)`: The timestamp associated with the left depth frame or `None` if an error occurred.\r\n  - `'error' (str or None)`: Error message, if any.\r\n\r\n**Example:**\r\n```python\r\nleft_zdepth_data = bridge.get_left_zdepth()\r\n```\r\n\r\n---\r\n\r\n###### **`get_right_seg(self)`**\r\n\r\n**Description:**  \r\nRetrieves the right segmentation frame from shared memory.\r\n\r\n**Returns:**  \r\n- `dict`: A dictionary with:\r\n  - `'frame' (np.ndarray or None)`: The retrieved right segmentation frame or `None` if an error occurred.\r\n  - `'timestamp' (int or None)`: The timestamp associated with the right segmentation frame or `None` if an error occurred.\r\n  - `'error' (str or None)`: Error message, if any.\r\n\r\n**Example:**\r\n```python\r\nright_segmentation_data = bridge.get_right_seg()\r\n```\r\n\r\n---\r\n\r\n###### **`get_left_seg(self)`**\r\n\r\n**Description:**  \r\nRetrieves the left segmentation frame from shared memory.\r\n\r\n**Returns:**  \r\n- `dict`: A dictionary with:\r\n  - `'frame' (np.ndarray or None)`: The retrieved left segmentation frame or `None` if an error occurred.\r\n  - `'timestamp' (int or None)`: The timestamp associated with the left segmentation frame or `None` if an error occurred.\r\n  - `'error' (str or None)`: Error message, if any.\r\n\r\n**Example:**\r\n```python\r\nleft_segmentation_data = bridge.get_left_seg()\r\n```\r\n\r\n---\r\n\r\n#### 2. Utilities\r\n   \r\n##### 1. **`timestamp2string`**\r\n\r\n**Description:**  \r\nConverts a timestamp in milliseconds to a human-readable string format.\r\n\r\n**Args:**  \r\n- `timestamp (int)`: The timestamp in milliseconds.\r\n\r\n**Returns:**  \r\n- `str`: Formatted timestamp as a string in the format 'YYYY-MM-DD HH:MM:SS:fff'.\r\n\r\n**Example:**\r\n```python\r\nformatted_time = timestamp2string(1609459200000)\r\n# Output: '2021-01-01 00:00:00:000'\r\n```\r\n\r\n---\r\n\r\n##### 2. **`timestamp2datetime`**\r\n\r\n**Description:**  \r\nConverts a timestamp in milliseconds to a `datetime` object in UTC.\r\n\r\n**Args:**  \r\n- `timestamp (int)`: The timestamp in milliseconds.\r\n\r\n**Returns:**  \r\n- `datetime`: The corresponding `datetime` object in UTC.\r\n\r\n**Example:**\r\n```python\r\ndatetime_obj = timestamp2datetime(1609459200000)\r\n# Output: datetime(2021, 1, 1, 0, 0, 0, tzinfo=timezone.utc)\r\n```\r\n\r\n---\r\n\r\n#### 3. `cartesian_to_gps`\r\n\r\n**Description:**  \r\nConverts Cartesian coordinates to GPS coordinates (latitude, longitude, altitude).\r\n\r\n**Args:**  \r\n- `x (float)`: X coordinate in centimeters.\r\n- `y (float)`: Y coordinate in centimeters.\r\n- `z (float)`: Z coordinate in centimeters.\r\n- `origin_lat (float, optional)`: Latitude of the origin point in degrees. Defaults to 22.583047.\r\n- `origin_lon (float, optional)`: Longitude of the origin point in degrees. Defaults to 88.45859783333334.\r\n- `origin_alt (float, optional)`: Altitude of the origin point in meters. Defaults to 0.\r\n- `add_noise (bool, optional)`: Whether to add noise to the GPS coordinates. Defaults to False.\r\n- `lat_long_noise_amt (float, optional)`: Amount of noise to add to latitude and longitude. Defaults to 0.0001.\r\n- `alt_noise_amt (float, optional)`: Amount of noise to add to altitude. Defaults to 0.1.\r\n- `earth_radius (float, optional)`: Radius of the Earth in meters. Defaults to 6378137 (meters).\r\n\r\n**Returns:**  \r\n- `tuple`: A tuple containing the latitude, longitude, and altitude in meters.\r\n\r\n**Example:**\r\n```python\r\nlatitude, longitude, altitude = cartesian_to_gps(1000, 2000, 300)\r\n```\r\n\r\n---\r\n\r\n#### Class: `DroneController`\r\nThis class provides an interface to control the drone's movements by sending commands to the flight matrix system. It allows the drone to move along the x, y, and z axes and rotate around the roll, pitch, and yaw axes.\r\n\r\n---\r\n\r\n##### **Attributes:**\r\n\r\n- `bridge (FlightMatrixBridge)`: The bridge object used to communicate with the drone.\r\n- `current_x (float)`: Current x-coordinate position, initialized to `0.0`.\r\n- `current_y (float)`: Current y-coordinate position, initialized to `0.0`.\r\n- `current_z (float)`: Current z-coordinate position, initialized to `0.0`.\r\n- `current_roll (float)`: Current roll angle, initialized to `0.0`.\r\n- `current_pitch (float)`: Current pitch angle, initialized to `0.0`.\r\n- `current_yaw (float)`: Current yaw angle, initialized to `0.0`.\r\n\r\n---\r\n\r\n##### **Methods:**\r\n\r\n###### **`__init__(self, bridge_object: FlightMatrixBridge)`**\r\n\r\n**Description:**  \r\nInitializes the `DroneController` class by linking it to a `FlightMatrixBridge` object and setting the initial drone movement parameters to zero.\r\n\r\n**Args:**  \r\n- `bridge_object (FlightMatrixBridge)`: An instance of `FlightMatrixBridge` for communication with the flight matrix system.\r\n\r\n**Example:**  \r\n```python\r\nbridge = FlightMatrixBridge()\r\ndrone_controller = DroneController(bridge)\r\n```\r\n\r\n---\r\n\r\n###### **`_send_command(self)`**\r\n\r\n**Description:**  \r\nSends the current positional and rotational state (x, y, z, roll, pitch, yaw) as movement commands to the drone.\r\n\r\n**Returns:**  \r\nNone\r\n\r\n**Example:**  \r\n```python\r\ndrone_controller._send_command()\r\n```\r\n\r\n---\r\n\r\n###### **`move_x(self, value)`**\r\n\r\n**Description:**  \r\nMoves the drone to a specified x-coordinate.\r\n\r\n**Args:**  \r\n- `value (float)`: The x-coordinate to move to.\r\n\r\n**Returns:**  \r\nNone\r\n\r\n**Example:**  \r\n```python\r\ndrone_controller.move_x(10.5)  # Move drone to x = 10.5\r\n```\r\n\r\n---\r\n\r\n###### **`move_y(self, value)`**\r\n\r\n**Description:**  \r\nMoves the drone to a specified y-coordinate (left or right).\r\n\r\n**Args:**  \r\n- `value (float)`: The y-coordinate to move to.\r\n\r\n**Returns:**  \r\nNone\r\n\r\n**Example:**  \r\n```python\r\ndrone_controller.move_y(-5.2)  # Move drone to y = -5.2\r\n```\r\n\r\n---\r\n\r\n###### **`move_z(self, value)`**\r\n\r\n**Description:**  \r\nMoves the drone to a specified z-coordinate (up or down).\r\n\r\n**Args:**  \r\n- `value (float)`: The z-coordinate to move to.\r\n\r\n**Returns:**  \r\nNone\r\n\r\n**Example:**  \r\n```python\r\ndrone_controller.move_z(15.0)  # Move drone up to z = 15.0\r\n```\r\n\r\n---\r\n\r\n###### **`rotate_roll(self, value)`**\r\n\r\n**Description:**  \r\nRotates the drone to a specified roll angle.\r\n\r\n**Args:**  \r\n- `value (float)`: The roll angle to rotate to.\r\n\r\n**Returns:**  \r\nNone\r\n\r\n**Example:**  \r\n```python\r\ndrone_controller.rotate_roll(30.0)  # Rotate drone to a roll angle of 30 degrees\r\n```\r\n\r\n---\r\n\r\n###### **`rotate_pitch(self, value)`**\r\n\r\n**Description:**  \r\nRotates the drone to a specified pitch angle.\r\n\r\n**Args:**  \r\n- `value (float)`: The pitch angle to rotate to.\r\n\r\n**Returns:**  \r\nNone\r\n\r\n**Example:**  \r\n```python\r\ndrone_controller.rotate_pitch(-15.0)  # Rotate drone to a pitch angle of -15 degrees\r\n```\r\n\r\n---\r\n\r\n###### **`rotate_yaw(self, value)`**\r\n\r\n**Description:**  \r\nRotates the drone to a specified yaw angle.\r\n\r\n**Args:**  \r\n- `value (float)`: The yaw angle to rotate to, in degrees.\r\n\r\n**Returns:**  \r\nNone\r\n\r\n**Example:**  \r\n```python\r\ndrone_controller.rotate_yaw(90.0)  # Rotate drone to a yaw angle of 90 degrees\r\n```\r\n\r\n---\r\n\r\n###### **`ascend(self, value)`**\r\n\r\n**Description:**  \r\nAscends the drone by a specified value, increasing the current altitude.\r\n\r\n**Args:**  \r\n- `value (float)`: The amount to increase the altitude.\r\n\r\n**Returns:**  \r\nNone\r\n\r\n**Example:**  \r\n```python\r\ndrone_controller.ascend(5.0)  # Ascend drone by 5 units\r\n```\r\n\r\n---\r\n\r\n###### **`descend(self, value)`**\r\n\r\n**Description:**  \r\nDescends the drone by a specified value, decreasing the current altitude.\r\n\r\n**Args:**  \r\n- `value (float)`: The amount to decrease the altitude.\r\n\r\n**Returns:**  \r\nNone\r\n\r\n**Example:**  \r\n```python\r\ndrone_controller.descend(3.0)  # Descend drone by 3 units\r\n```\r\n\r\n---\r\n\r\n###### **`move_forward(self, value)`**\r\n\r\n**Description:**  \r\nMoves the drone forward by a specified value (positive y-axis).\r\n\r\n**Args:**  \r\n- `value (float)`: The amount to move forward.\r\n\r\n**Returns:**  \r\nNone\r\n\r\n**Example:**  \r\n```python\r\ndrone_controller.move_forward(10.0)  # Move drone forward by 10 units\r\n```\r\n\r\n---\r\n\r\n###### **`move_backward(self, value)`**\r\n\r\n**Description:**  \r\nMoves the drone backward by a specified value (negative y-axis).\r\n\r\n**Args:**  \r\n- `value (float)`: The amount to move backward.\r\n\r\n**Returns:**  \r\nNone\r\n\r\n**Example:**  \r\n```python\r\ndrone_controller.move_backward(8.0)  # Move drone backward by 8 units\r\n```\r\n\r\n---\r\n\r\n###### **`stop_movement(self)`**\r\n\r\n**Description:**  \r\nStops all drone movements on the x, y, and z axes.\r\n\r\n**Returns:**  \r\nNone\r\n\r\n**Example:**  \r\n```python\r\ndrone_controller.stop_movement()  # Stop all drone movements\r\n```\r\n\r\n---\r\n\r\n#### Class: `DataRecorder`\r\nThe `DataRecorder` class is designed to record various types of data from a drone or robotic system using the FlightMatrix framework. It can capture visual frames, z-depth images, segmentation frames, and sensor data, all of which are stored in a structured manner for later analysis.\r\n\r\n---\r\n\r\n##### **Attributes:**\r\n\r\n- `bridge (FlightMatrixBridge)`: The bridge object used to interface with the drone or robot's systems.\r\n- `base_dir (str)`: The base directory where all recorded data will be stored.\r\n- `record_left_frame (bool)`: Flag indicating whether to record the left visual frame (default: `False`).\r\n- `record_right_frame (bool)`: Flag indicating whether to record the right visual frame (default: `False`).\r\n- `record_left_zdepth (bool)`: Flag indicating whether to record the left z-depth frame (default: `False`).\r\n- `record_right_zdepth (bool)`: Flag indicating whether to record the right z-depth frame (default: `False`).\r\n- `record_left_seg (bool)`: Flag indicating whether to record the left segmentation frame (default: `False`).\r\n- `record_right_seg (bool)`: Flag indicating whether to record the right segmentation frame (default: `False`).\r\n- `record_sensor_data (bool)`: Flag indicating whether to record sensor data (default: `False`).\r\n- `sensor_data_interval (float)`: The interval at which sensor data is recorded (default: `0.1` seconds).\r\n- `threads (list)`: List to hold the thread objects for recording data.\r\n- `stop_event (Event)`: Event used to signal the threads to stop.\r\n\r\n---\r\n\r\n##### **Methods:**\r\n\r\n###### **`__init__(self, bridge: FlightMatrixBridge, base_dir: str, record_left_frame: bool = False, record_right_frame: bool = False, record_left_zdepth: bool = False, record_right_zdepth: bool = False, record_left_seg: bool = False, record_right_seg: bool = False, record_sensor_data: bool = False, record_sensor_data_interval: float = 0.1)`**\r\n\r\n**Description:**  \r\nInitializes the `DataRecorder` class with specified options for recording. It sets up directories for storing the recorded data based on user selections.\r\n\r\n**Args:**  \r\n- `bridge (FlightMatrixBridge)`: An instance of `FlightMatrixBridge` used to interact with the drone/robot.\r\n- `base_dir (str)`: The directory to store recorded files.\r\n- `record_left_frame (bool)`: If `True`, records the left visual frame.\r\n- `record_right_frame (bool)`: If `True`, records the right visual frame.\r\n- `record_left_zdepth (bool)`: If `True`, records the left z-depth frame.\r\n- `record_right_zdepth (bool)`: If `True`, records the right z-depth frame.\r\n- `record_left_seg (bool)`: If `True`, records the left segmentation frame.\r\n- `record_right_seg (bool)`: If `True`, records the right segmentation frame.\r\n- `record_sensor_data (bool)`: If `True`, records sensor data.\r\n- `record_sensor_data_interval (float)`: Time interval for recording sensor data in seconds.\r\n\r\n---\r\n\r\n###### **`record_frames(self)`**\r\n\r\n**Description:**  \r\nContinuously captures and saves visual frames, z-depth frames, and segmentation frames until the recording is stopped. Each frame is saved with a timestamped filename.\r\n\r\n---\r\n\r\n###### **`record_sensors(self)`**\r\n\r\n**Description:**  \r\nRecords sensor data at specified intervals, saving the readings to a CSV file. It checks for errors in the sensor data and handles them appropriately.\r\n\r\n---\r\n\r\n###### **`start_recording(self)`**\r\n\r\n**Description:**  \r\nStarts the recording process by launching separate threads for recording frames and sensor data, based on the user\u00e2\u20ac\u2122s selections.\r\n\r\n---\r\n\r\n###### **`stop_recording(self)`**\r\n\r\n**Description:**  \r\nStops the recording process by signaling the threads to finish and waits for them to join back.\r\n\r\n---\r\n\r\n##### **Example Usage:**\r\n\r\n```python\r\nif __name__ == \"__main__\":\r\n    bridge = FlightMatrixBridge()\r\n    recorder = DataRecorder(bridge, base_dir=\"Sample_Recordings\", \r\n                            record_left_frame=True, \r\n                            record_right_frame=True, \r\n                            record_left_zdepth=True, \r\n                            record_right_zdepth=True, \r\n                            record_left_seg=True, \r\n                            record_right_seg=True, \r\n                            record_sensor_data=True,\r\n                            record_sensor_data_interval=1)\r\n    \r\n    recorder.start_recording()\r\n\r\n    time.sleep(10)  # Record for 10 seconds\r\n\r\n    recorder.stop_recording()\r\n```\r\n\r\n###### **`is_recording_on(self)`**\r\n\r\n**Description:**\r\nChecks if the recording process is currently active.\r\n\r\n**Returns:**\r\n- `bool`: `True` if recording is active, `False` otherwise.\r\n\r\n---\r\n\r\n## Data Streaming with DataStreamer\r\n\r\nThe `DataStreamer` class provides a convenient way to stream data from the Flight Matrix system using callbacks. It allows you to subscribe to specific data streams (e.g., left frame, sensor data) and specify callback functions that are called whenever new data is available. Each data stream runs in its own thread, enabling parallel data fetching without one stream waiting for the other. You can also specify the interval at which data is fetched or set it to zero for the fastest possible streaming.\r\n\r\n### Features\r\n\r\n- **Subscribe to Data Streams**: Subscribe to specific data types you are interested in.\r\n- **Parallel Data Fetching**: Each data stream runs in its own thread for efficient parallel processing.\r\n- **Custom Callbacks**: Provide your own callback functions to process the data as it arrives.\r\n- **Adjustable Fetch Interval**: Control the rate at which data is fetched by specifying the interval.\r\n\r\n### Usage\r\n\r\n#### Import and Initialize\r\n\r\n```python\r\nfrom flightmatrix.bridge import FlightMatrixBridge\r\nfrom flightmatrix.utilities import DataStreamer\r\n\r\n# Initialize the FlightMatrixBridge\r\nbridge = FlightMatrixBridge()\r\n\r\n# Initialize the DataStreamer\r\nstreamer = DataStreamer(bridge)\r\n```\r\n\r\n#### Subscribe to Data Streams\r\n\r\n```python\r\n\r\nimport cv2\r\n\r\n# Define a callback function for left frame data\r\ndef left_frame_callback(frame_data):\r\n    frame = frame_data['frame']\r\n    timestamp = frame_data['timestamp']\r\n    cv2.imshow(\"Left Frame\", frame)\r\n    print(\"Left Frame Timestamp:\", timestamp)\r\n\r\n# Subscribe to the left frame data stream\r\nstreamer.subscribe(\"left_frame\", left_frame_callback)\r\n\r\n# Define a callback function for sensor data\r\ndef sensor_data_callback(sensor_data):\r\n    print(\"Sensor Data:\", sensor_data)\r\n\r\n# Subscribe to the sensor data stream\r\nstreamer.subscribe(\"sensor_data\", sensor_data_callback)\r\n```\r\n\r\n### Available Data Streams\r\n\r\n- **left_frame**: Left visual frame data.\r\n- **right_frame**: Right visual frame data.\r\n- **left_zdepth**: Left z-depth frame data.\r\n- **right_zdepth**: Right z-depth frame data.\r\n- **left_seg**: Left segmentation frame data.\r\n- **right_seg**: Right segmentation frame data.\r\n- **sensor_data**: Sensor data from the drone or robot.\r\n\r\nUse the `subscribe` method to subscribe to the desired data streams and provide a callback function to process the data as it arrives. The callback function will be called with the data as an argument whenever new data is available. for example, the `left_frame_callback` function will be called with the left frame data whenever a new frame is available. \r\n\r\n### Subscribing to Data Streams\r\n\r\n```python\r\nimport cv2\r\nfrom flightmatrix.bridge import FlightMatrixBridge\r\nfrom flightmatrix.utilities import DataStreamer\r\n\r\ndef left_frame_callback(left_frame_data):\r\n    left_frame = left_frame_data['frame']\r\n    left_timestamp = left_frame_data['timestamp']\r\n    cv2.imshow(\"Left Frame\", left_frame)\r\n\r\n    # Break the loop when 'q' is pressed\r\n    if cv2.waitKey(1) & 0xFF == ord('q'):\r\n        cv2.destroyAllWindows()\r\n\r\n    print(\"Left Frame Timestamp:\", left_timestamp)\r\n\r\ndef right_frame_callback(right_frame_data):\r\n    right_frame = right_frame_data['frame']\r\n    right_timestamp = right_frame_data['timestamp']\r\n    cv2.imshow(\"Right Frame\", right_frame)\r\n\r\n    # Break the loop when 'q' is pressed\r\n    if cv2.waitKey(1) & 0xFF == ord('q'):\r\n        cv2.destroyAllWindows()\r\n\r\n    print(\"Right Frame Timestamp:\", right_timestamp)\r\n\r\ndef left_zdepth_callback(left_zdepth_data):\r\n    left_zdepth = left_zdepth_data['frame']\r\n    left_timestamp = left_zdepth_data['timestamp']\r\n    cv2.imshow(\"Left Z-Depth\", left_zdepth)\r\n\r\n    # Break the loop when 'q' is pressed\r\n    if cv2.waitKey(1) & 0xFF == ord('q'):\r\n        cv2.destroyAllWindows()\r\n\r\n    print(\"Left Z-Depth Timestamp:\", left_timestamp)\r\n\r\ndef right_zdepth_callback(right_zdepth_data):\r\n    right_zdepth = right_zdepth_data['frame']\r\n    right_timestamp = right_zdepth_data['timestamp']\r\n    cv2.imshow(\"Right Z-Depth\", right_zdepth)\r\n\r\n    # Break the loop when 'q' is pressed\r\n    if cv2.waitKey(1) & 0xFF == ord('q'):\r\n        cv2.destroyAllWindows()\r\n\r\n    print(\"Right Z-Depth Timestamp:\", right_timestamp)\r\n\r\ndef left_seg_callback(left_seg_data):\r\n    left_seg = left_seg_data['frame']\r\n    left_timestamp = left_seg_data['timestamp']\r\n    cv2.imshow(\"Left Segmentation\", left_seg)\r\n\r\n    # Break the loop when 'q' is pressed\r\n    if cv2.waitKey(1) & 0xFF == ord('q'):\r\n        cv2.destroyAllWindows()\r\n\r\n    print(\"Left Segmentation Timestamp:\", left_timestamp)\r\n\r\ndef right_seg_callback(right_seg_data):\r\n    right_seg = right_seg_data['frame']\r\n    right_timestamp = right_seg_data['timestamp']\r\n    cv2.imshow(\"Right Segmentation\", right_seg)\r\n\r\n    # Break the loop when 'q' is pressed\r\n    if cv2.waitKey(1) & 0xFF == ord('q'):\r\n        cv2.destroyAllWindows()\r\n\r\n    print(\"Right Segmentation Timestamp:\", right_timestamp)\r\n\r\ndef sensor_data_callback(sensor_data):\r\n    print(\"Sensor Data:\", sensor_data)\r\n\r\n# Initialize the FlightMatrixBridge\r\nbridge = FlightMatrixBridge()\r\n\r\n# Initialize the DataStreamer\r\nstreamer = DataStreamer(bridge)\r\n\r\n# Subscribe to the left frame data stream\r\nstreamer.subscribe(\"left_frame\", left_frame_callback, interval=0)\r\n\r\n# Subscribe to the right frame data stream\r\nstreamer.subscribe(\"right_frame\", right_frame_callback, interval=0.1)\r\n\r\n# Subscribe to the left z-depth data stream\r\nstreamer.subscribe(\"left_zdepth\", left_zdepth_callback, interval=0.1)\r\n\r\n# Subscribe to the right z-depth data stream\r\nstreamer.subscribe(\"right_zdepth\", right_zdepth_callback, interval=0.1)\r\n\r\n# Subscribe to the left segmentation data stream\r\nstreamer.subscribe(\"left_seg\", left_seg_callback, interval=0.1)\r\n\r\n# Subscribe to the right segmentation data stream\r\nstreamer.subscribe(\"right_seg\", right_seg_callback, interval=0.1)\r\n\r\n# Subscribe to the sensor data stream\r\nstreamer.subscribe(\"sensor_data\", sensor_data_callback, interval=0.1)\r\n```\r\n\r\n## Credits\r\n\r\nThis project was developed and maintained by [Ranit Bhowmick](https://www.linkedin.com/in/ranitbhowmick), a Robotics and Automation engineer with a passion for building innovative solutions in AI, game development, and full-stack projects. Specializing in advanced Python programming, machine learning, and robotics, I\u00e2\u20ac\u2122m always open to collaboration and eager to explore new challenges.\r\n\r\nI'd like to express my gratitude to the unreal engine community for their support and feedback. I'm always open to suggestions and contributions to improve this project further.\r\n",
    "bugtrack_url": null,
    "license": "MIT License with attribution requirement",
    "summary": "FlightMatrixBridge is a Python library that facilitates seamless inter-process communication between the Flight Matrix software using shared memory. It handles frames, timestamps, and movement commands for efficient data sharing in robotics applications.",
    "version": "1.7.0",
    "project_urls": {
        "Download": "https://github.com/Kawai-Senpai/Py-FlightMatrix-Bridge",
        "Homepage": "https://github.com/Kawai-Senpai/Py-FlightMatrix-Bridge"
    },
    "split_keywords": [
        "robotics",
        " inter-process communication",
        " shared memory",
        " data management"
    ],
    "urls": [
        {
            "comment_text": "",
            "digests": {
                "blake2b_256": "1dc0a78604efb4288e068b1a586abea9311c50ba644a9ce5c0566f01510ef130",
                "md5": "a58d80227bff86a050bfe7a0897800cf",
                "sha256": "026569d22ee3af8757739c7adabe57cbae1f701e404b47ead82578bd1f32ddd4"
            },
            "downloads": -1,
            "filename": "flightmatrixbridge-1.7.0-py3-none-any.whl",
            "has_sig": false,
            "md5_digest": "a58d80227bff86a050bfe7a0897800cf",
            "packagetype": "bdist_wheel",
            "python_version": "py3",
            "requires_python": null,
            "size": 30423,
            "upload_time": "2024-11-12T17:06:31",
            "upload_time_iso_8601": "2024-11-12T17:06:31.699856Z",
            "url": "https://files.pythonhosted.org/packages/1d/c0/a78604efb4288e068b1a586abea9311c50ba644a9ce5c0566f01510ef130/flightmatrixbridge-1.7.0-py3-none-any.whl",
            "yanked": false,
            "yanked_reason": null
        },
        {
            "comment_text": "",
            "digests": {
                "blake2b_256": "28ad1078f80f1857e90460dc62d765545d8ebe69244337d4c56082c39ae6526d",
                "md5": "c9d7361ffdd39ef7cdb5619b166f9602",
                "sha256": "69ed60bf81ff7cc1d358f6d28c4b3c75ba34b6004d28c6007712d8e150e4d715"
            },
            "downloads": -1,
            "filename": "flightmatrixbridge-1.7.0.tar.gz",
            "has_sig": false,
            "md5_digest": "c9d7361ffdd39ef7cdb5619b166f9602",
            "packagetype": "sdist",
            "python_version": "source",
            "requires_python": null,
            "size": 63609,
            "upload_time": "2024-11-12T17:06:33",
            "upload_time_iso_8601": "2024-11-12T17:06:33.358978Z",
            "url": "https://files.pythonhosted.org/packages/28/ad/1078f80f1857e90460dc62d765545d8ebe69244337d4c56082c39ae6526d/flightmatrixbridge-1.7.0.tar.gz",
            "yanked": false,
            "yanked_reason": null
        }
    ],
    "upload_time": "2024-11-12 17:06:33",
    "github": true,
    "gitlab": false,
    "bitbucket": false,
    "codeberg": false,
    "github_user": "Kawai-Senpai",
    "github_project": "Py-FlightMatrix-Bridge",
    "travis_ci": false,
    "coveralls": false,
    "github_actions": false,
    "requirements": [
        {
            "name": "numpy",
            "specs": [
                [
                    "==",
                    "1.24.2"
                ]
            ]
        },
        {
            "name": "ultraprint",
            "specs": [
                [
                    "==",
                    "3.1.0"
                ]
            ]
        },
        {
            "name": "opencv-python",
            "specs": [
                [
                    "==",
                    "4.6.0.66"
                ]
            ]
        }
    ],
    "lcname": "flightmatrixbridge"
}
        
Elapsed time: 1.65208s