data/__init__.py

"""
Data Package
=============
Data handling, schema definitions, and dataset generation for
the Traffic Accident Reconstruction System.
"""

from .schema import (
    # Enums
    VehicleType,
    Direction,
    VehicleAction,
    WeatherCondition,
    RoadCondition,
    AccidentType,
    Severity,
    RoadType,
    
    # Data classes
    Location,
    Conditions,
    VehicleData,
    AccidentDetails,
    ScenarioMetrics,
    Scenario,
    FaultAssessment,
    TimelineEvent,
    AccidentRecord,
    AnalysisResult,
    
    # Feature schema
    FEATURE_SCHEMA,
    
    # Validation functions
    validate_speed,
    validate_coordinates,
    validate_path,
    validate_probability
)

from .synthetic_dataset_generator import (
    generate_dataset,
    generate_single_accident,
    generate_training_features,
    save_dataset,
    print_dataset_statistics
)

__all__ = [
    # Enums
    'VehicleType',
    'Direction', 
    'VehicleAction',
    'WeatherCondition',
    'RoadCondition',
    'AccidentType',
    'Severity',
    'RoadType',
    
    # Data classes
    'Location',
    'Conditions',
    'VehicleData',
    'AccidentDetails',
    'ScenarioMetrics',
    'Scenario',
    'FaultAssessment',
    'TimelineEvent',
    'AccidentRecord',
    'AnalysisResult',
    
    # Feature schema
    'FEATURE_SCHEMA',
    
    # Validation
    'validate_speed',
    'validate_coordinates',
    'validate_path',
    'validate_probability',
    
    # Dataset generation
    'generate_dataset',
    'generate_single_accident',
    'generate_training_features',
    'save_dataset',
    'print_dataset_statistics'
]