from tracking.devices.mock_lidar import MockLidar
from rplidar import RPLidar
import tracking.lidar_loader as loader

connection = "TEST"
# connection = '/dev/ttyUSB0'

def get_lidar():
    # Need a way to configure this, maybe with environment variables
    if connection == 'TEST':
        return MockLidar(loader.load_scans_bytes_file("tracking/out.pickle"))
    else:
        return RPLidar(connection)