diff --git a/.gitignore b/.gitignore index ba9d174..91e6dde 100644 --- a/.gitignore +++ b/.gitignore @@ -9,4 +9,5 @@ src/dataset/data/* src/dataset/nclt !src/dataset/data/*.py -__pycache__ +# Ignore pycache +__pycache__/ diff --git a/src/dataset/dataManipulation/GPSmanip.py b/src/dataset/data_manip/GPSmanip.py similarity index 100% rename from src/dataset/dataManipulation/GPSmanip.py rename to src/dataset/data_manip/GPSmanip.py diff --git a/src/dataset/dataManipulation/findBuildingCoord.py b/src/dataset/data_manip/findBuildingCoord.py similarity index 100% rename from src/dataset/dataManipulation/findBuildingCoord.py rename to src/dataset/data_manip/findBuildingCoord.py diff --git a/src/dataset/dataManipulation/pickles/BuildingMappings.pkl b/src/dataset/data_manip/pickles/BuildingMappings.pkl similarity index 100% rename from src/dataset/dataManipulation/pickles/BuildingMappings.pkl rename to src/dataset/data_manip/pickles/BuildingMappings.pkl diff --git a/src/dataset/dataManipulation/pickles/BuildingMappings.txt b/src/dataset/data_manip/pickles/BuildingMappings.txt similarity index 100% rename from src/dataset/dataManipulation/pickles/BuildingMappings.txt rename to src/dataset/data_manip/pickles/BuildingMappings.txt diff --git a/src/dataset/dataManipulation/pickles/pickleManage.py b/src/dataset/data_manip/pickles/pickleManage.py similarity index 100% rename from src/dataset/dataManipulation/pickles/pickleManage.py rename to src/dataset/data_manip/pickles/pickleManage.py diff --git a/src/dataset/dataManipulation/pickles/referenceBuildings4Pickle.txt b/src/dataset/data_manip/pickles/referenceBuildings4Pickle.txt similarity index 100% rename from src/dataset/dataManipulation/pickles/referenceBuildings4Pickle.txt rename to src/dataset/data_manip/pickles/referenceBuildings4Pickle.txt diff --git a/src/dataset/ManageDataset/project_vel_to_cam.py b/src/dataset/read_dataset/project_vel_to_cam.py similarity index 100% rename from src/dataset/ManageDataset/project_vel_to_cam.py rename to src/dataset/read_dataset/project_vel_to_cam.py diff --git a/src/dataset/ManageDataset/read_gps.py b/src/dataset/read_dataset/read_gps.py similarity index 100% rename from src/dataset/ManageDataset/read_gps.py rename to src/dataset/read_dataset/read_gps.py diff --git a/src/dataset/ManageDataset/read_ground_truth.py b/src/dataset/read_dataset/read_ground_truth.py similarity index 100% rename from src/dataset/ManageDataset/read_ground_truth.py rename to src/dataset/read_dataset/read_ground_truth.py diff --git a/src/dataset/ManageDataset/read_hokuyo_30m.py b/src/dataset/read_dataset/read_hokuyo_30m.py similarity index 100% rename from src/dataset/ManageDataset/read_hokuyo_30m.py rename to src/dataset/read_dataset/read_hokuyo_30m.py diff --git a/src/dataset/ManageDataset/read_hokuyo_4m.py b/src/dataset/read_dataset/read_hokuyo_4m.py similarity index 100% rename from src/dataset/ManageDataset/read_hokuyo_4m.py rename to src/dataset/read_dataset/read_hokuyo_4m.py diff --git a/src/dataset/ManageDataset/read_ms25.py b/src/dataset/read_dataset/read_ms25.py similarity index 100% rename from src/dataset/ManageDataset/read_ms25.py rename to src/dataset/read_dataset/read_ms25.py diff --git a/src/dataset/ManageDataset/read_ms25_euler.py b/src/dataset/read_dataset/read_ms25_euler.py similarity index 100% rename from src/dataset/ManageDataset/read_ms25_euler.py rename to src/dataset/read_dataset/read_ms25_euler.py diff --git a/src/dataset/ManageDataset/read_odom.py b/src/dataset/read_dataset/read_odom.py similarity index 100% rename from src/dataset/ManageDataset/read_odom.py rename to src/dataset/read_dataset/read_odom.py diff --git a/src/dataset/read_dataset/read_vel_hits.py b/src/dataset/read_dataset/read_vel_hits.py new file mode 100644 index 0000000..3d683f1 --- /dev/null +++ b/src/dataset/read_dataset/read_vel_hits.py @@ -0,0 +1,89 @@ +# !/usr/bin/python +# +# Example code to go through the velodyne_hits.bin +# file and read timestamps, number of hits, and the +# hits in each packet. +# +# +# To call: +# +# python read_vel_hits.py velodyne.bin +# + +import sys +import struct + +def convert(x_s, y_s, z_s): + + scaling = 0.005 # 5 mm + offset = -100.0 + + x = x_s * scaling + offset + y = y_s * scaling + offset + z = z_s * scaling + offset + + return x, y, z + +def verify_magic(s): + + magic = 44444 + + m = struct.unpack('=4 and m[0] == magic and m[1] == magic and m[2] == magic and m[3] == magic + +def main(args): + + if len(sys.argv) < 2: + print("Please specify input bin file") + return 1 + + f_bin = open(sys.argv[1], "rb") + + total_hits = 0 + first_utime = -1 + last_utime = -1 + + while True: + magic = f_bin.read(8) + if magic == '': # eof + break + + if not verify_magic(magic): + print("Could not verify magic") + + num_hits = struct.unpack('= 3: + print('Writing to ', sys.argv[2]) + f_csv = open(sys.argv[2], "w") + else: + f_csv = None + + hits = [] + + while True: + + x_str = f_bin.read(2) + if x_str == b'': # eof + break + + x = struct.unpack('