feat: Complete Rust port of WiFi-DensePose with modular crates
Major changes: - Organized Python v1 implementation into v1/ subdirectory - Created Rust workspace with 9 modular crates: - wifi-densepose-core: Core types, traits, errors - wifi-densepose-signal: CSI processing, phase sanitization, FFT - wifi-densepose-nn: Neural network inference (ONNX/Candle/tch) - wifi-densepose-api: Axum-based REST/WebSocket API - wifi-densepose-db: SQLx database layer - wifi-densepose-config: Configuration management - wifi-densepose-hardware: Hardware abstraction - wifi-densepose-wasm: WebAssembly bindings - wifi-densepose-cli: Command-line interface Documentation: - ADR-001: Workspace structure - ADR-002: Signal processing library selection - ADR-003: Neural network inference strategy - DDD domain model with bounded contexts Testing: - 69 tests passing across all crates - Signal processing: 45 tests - Neural networks: 21 tests - Core: 3 doc tests Performance targets: - 10x faster CSI processing (~0.5ms vs ~5ms) - 5x lower memory usage (~100MB vs ~500MB) - WASM support for browser deployment
This commit is contained in:
Claude
266
v1/src/__init__.py
Normal file
266
v1/src/__init__.py
Normal file
@@ -0,0 +1,266 @@
|
||||
"""
|
||||
WiFi-DensePose API Package
|
||||
==========================
|
||||
|
||||
A comprehensive system for WiFi-based human pose estimation using CSI data
|
||||
and DensePose neural networks.
|
||||
|
||||
This package provides:
|
||||
- Real-time CSI data collection from WiFi routers
|
||||
- Advanced signal processing and phase sanitization
|
||||
- DensePose neural network integration for pose estimation
|
||||
- RESTful API for data access and control
|
||||
- Background task management for data processing
|
||||
- Comprehensive monitoring and logging
|
||||
|
||||
Example usage:
|
||||
>>> from src.app import app
|
||||
>>> from src.config.settings import get_settings
|
||||
>>>
|
||||
>>> settings = get_settings()
|
||||
>>> # Run with: uvicorn src.app:app --host 0.0.0.0 --port 8000
|
||||
|
||||
For CLI usage:
|
||||
$ wifi-densepose start --host 0.0.0.0 --port 8000
|
||||
$ wifi-densepose status
|
||||
$ wifi-densepose stop
|
||||
|
||||
Author: WiFi-DensePose Team
|
||||
License: MIT
|
||||
"""
|
||||
|
||||
__version__ = "1.1.0"
|
||||
__author__ = "WiFi-DensePose Team"
|
||||
__email__ = "team@wifi-densepose.com"
|
||||
__license__ = "MIT"
|
||||
__copyright__ = "Copyright 2024 WiFi-DensePose Team"
|
||||
|
||||
# Package metadata
|
||||
__title__ = "wifi-densepose"
|
||||
__description__ = "WiFi-based human pose estimation using CSI data and DensePose neural networks"
|
||||
__url__ = "https://github.com/wifi-densepose/wifi-densepose"
|
||||
__download_url__ = "https://github.com/wifi-densepose/wifi-densepose/archive/main.zip"
|
||||
|
||||
# Version info tuple
|
||||
__version_info__ = tuple(int(x) for x in __version__.split('.'))
|
||||
|
||||
# Import key components for easy access
|
||||
try:
|
||||
from src.app import app
|
||||
from src.config.settings import get_settings, Settings
|
||||
from src.logger import setup_logging, get_logger
|
||||
|
||||
# Core components
|
||||
from src.core.csi_processor import CSIProcessor
|
||||
from src.core.phase_sanitizer import PhaseSanitizer
|
||||
from src.core.pose_estimator import PoseEstimator
|
||||
from src.core.router_interface import RouterInterface
|
||||
|
||||
# Services
|
||||
from src.services.orchestrator import ServiceOrchestrator
|
||||
from src.services.health_check import HealthCheckService
|
||||
from src.services.metrics import MetricsService
|
||||
|
||||
# Database
|
||||
from src.database.connection import get_database_manager
|
||||
from src.database.models import (
|
||||
Device, Session, CSIData, PoseDetection,
|
||||
SystemMetric, AuditLog
|
||||
)
|
||||
|
||||
__all__ = [
|
||||
# Core app
|
||||
'app',
|
||||
'get_settings',
|
||||
'Settings',
|
||||
'setup_logging',
|
||||
'get_logger',
|
||||
|
||||
# Core processing
|
||||
'CSIProcessor',
|
||||
'PhaseSanitizer',
|
||||
'PoseEstimator',
|
||||
'RouterInterface',
|
||||
|
||||
# Services
|
||||
'ServiceOrchestrator',
|
||||
'HealthCheckService',
|
||||
'MetricsService',
|
||||
|
||||
# Database
|
||||
'get_database_manager',
|
||||
'Device',
|
||||
'Session',
|
||||
'CSIData',
|
||||
'PoseDetection',
|
||||
'SystemMetric',
|
||||
'AuditLog',
|
||||
|
||||
# Metadata
|
||||
'__version__',
|
||||
'__version_info__',
|
||||
'__author__',
|
||||
'__email__',
|
||||
'__license__',
|
||||
'__copyright__',
|
||||
]
|
||||
|
||||
except ImportError as e:
|
||||
# Handle import errors gracefully during package installation
|
||||
import warnings
|
||||
warnings.warn(
|
||||
f"Some components could not be imported: {e}. "
|
||||
"This is normal during package installation.",
|
||||
ImportWarning
|
||||
)
|
||||
|
||||
__all__ = [
|
||||
'__version__',
|
||||
'__version_info__',
|
||||
'__author__',
|
||||
'__email__',
|
||||
'__license__',
|
||||
'__copyright__',
|
||||
]
|
||||
|
||||
|
||||
def get_version():
|
||||
"""Get the package version."""
|
||||
return __version__
|
||||
|
||||
|
||||
def get_version_info():
|
||||
"""Get the package version as a tuple."""
|
||||
return __version_info__
|
||||
|
||||
|
||||
def get_package_info():
|
||||
"""Get comprehensive package information."""
|
||||
return {
|
||||
'name': __title__,
|
||||
'version': __version__,
|
||||
'version_info': __version_info__,
|
||||
'description': __description__,
|
||||
'author': __author__,
|
||||
'author_email': __email__,
|
||||
'license': __license__,
|
||||
'copyright': __copyright__,
|
||||
'url': __url__,
|
||||
'download_url': __download_url__,
|
||||
}
|
||||
|
||||
|
||||
def check_dependencies():
|
||||
"""Check if all required dependencies are available."""
|
||||
missing_deps = []
|
||||
optional_deps = []
|
||||
|
||||
# Core dependencies
|
||||
required_modules = [
|
||||
('fastapi', 'FastAPI'),
|
||||
('uvicorn', 'Uvicorn'),
|
||||
('pydantic', 'Pydantic'),
|
||||
('sqlalchemy', 'SQLAlchemy'),
|
||||
('numpy', 'NumPy'),
|
||||
('torch', 'PyTorch'),
|
||||
('cv2', 'OpenCV'),
|
||||
('scipy', 'SciPy'),
|
||||
('pandas', 'Pandas'),
|
||||
('redis', 'Redis'),
|
||||
('psutil', 'psutil'),
|
||||
('click', 'Click'),
|
||||
]
|
||||
|
||||
for module_name, display_name in required_modules:
|
||||
try:
|
||||
__import__(module_name)
|
||||
except ImportError:
|
||||
missing_deps.append(display_name)
|
||||
|
||||
# Optional dependencies
|
||||
optional_modules = [
|
||||
('scapy', 'Scapy (for network packet capture)'),
|
||||
('paramiko', 'Paramiko (for SSH connections)'),
|
||||
('serial', 'PySerial (for serial communication)'),
|
||||
('matplotlib', 'Matplotlib (for plotting)'),
|
||||
('prometheus_client', 'Prometheus Client (for metrics)'),
|
||||
]
|
||||
|
||||
for module_name, display_name in optional_modules:
|
||||
try:
|
||||
__import__(module_name)
|
||||
except ImportError:
|
||||
optional_deps.append(display_name)
|
||||
|
||||
return {
|
||||
'missing_required': missing_deps,
|
||||
'missing_optional': optional_deps,
|
||||
'all_required_available': len(missing_deps) == 0,
|
||||
}
|
||||
|
||||
|
||||
def print_system_info():
|
||||
"""Print system and package information."""
|
||||
import sys
|
||||
import platform
|
||||
|
||||
info = get_package_info()
|
||||
deps = check_dependencies()
|
||||
|
||||
print(f"WiFi-DensePose v{info['version']}")
|
||||
print(f"Python {sys.version}")
|
||||
print(f"Platform: {platform.platform()}")
|
||||
print(f"Architecture: {platform.architecture()[0]}")
|
||||
print()
|
||||
|
||||
if deps['all_required_available']:
|
||||
print("✅ All required dependencies are available")
|
||||
else:
|
||||
print("❌ Missing required dependencies:")
|
||||
for dep in deps['missing_required']:
|
||||
print(f" - {dep}")
|
||||
|
||||
if deps['missing_optional']:
|
||||
print("\n⚠️ Missing optional dependencies:")
|
||||
for dep in deps['missing_optional']:
|
||||
print(f" - {dep}")
|
||||
|
||||
print(f"\nFor more information, visit: {info['url']}")
|
||||
|
||||
|
||||
# Package-level configuration
|
||||
import logging
|
||||
|
||||
# Set up basic logging configuration
|
||||
logging.getLogger(__name__).addHandler(logging.NullHandler())
|
||||
|
||||
# Suppress some noisy third-party loggers
|
||||
logging.getLogger('urllib3').setLevel(logging.WARNING)
|
||||
logging.getLogger('requests').setLevel(logging.WARNING)
|
||||
logging.getLogger('asyncio').setLevel(logging.WARNING)
|
||||
|
||||
# Package initialization message
|
||||
if __name__ != '__main__':
|
||||
logger = logging.getLogger(__name__)
|
||||
logger.debug(f"WiFi-DensePose package v{__version__} initialized")
|
||||
|
||||
|
||||
# Compatibility aliases for backward compatibility
|
||||
try:
|
||||
WifiDensePose = app # Legacy alias
|
||||
except NameError:
|
||||
WifiDensePose = None # Will be None if app import failed
|
||||
|
||||
try:
|
||||
get_config = get_settings # Legacy alias
|
||||
except NameError:
|
||||
get_config = None # Will be None if get_settings import failed
|
||||
|
||||
|
||||
def main():
|
||||
"""Main entry point for the package when run as a module."""
|
||||
print_system_info()
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
Reference in New Issue
Block a user