Hesham/room_scan_sim
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

cleanup

Browse Source 
Signed-off-by: HeshamTB <hishaminv@gmail.com>
This commit is contained in:
HeshamTB 2022-01-25 22:45:45 +03:00
parent 80733cbedb
commit d0a2cbb0db
Signed by: Hesham
GPG Key ID: 74876157D199B09E
2 changed files with 5 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
basictypes.py
View File

@ -56,10 +56,9 @@ class Line:
self.m, self.b = self.__find_line(p1, p2) self.m, self.b = self.__find_line(p1, p2)
v = ValueError() v = ValueError()
if not self.__matching_wall_ids(p1, p2): if not self.__matching_wall_ids(p1, p2):
raise WallIDMismatch('Wall ID for points do not match') raise WallIDMismatch('Wall ID for points does not match')
def __find_line(self, p1:PointWithID, p2:PointWithID) -> float: def __find_line(self, p1:PointWithID, p2:PointWithID) -> float:
# m = ((point2[1] - point1[1])/(point2[0] - point1[0]))
m = (p2.get_y() - p1.get_y()) / (p2.get_x() - p1.get_x()) m = (p2.get_y() - p1.get_y()) / (p2.get_x() - p1.get_x())
# Using p1 to find b # Using p1 to find b
b = p1.get_y() - (m*p1.get_x()) b = p1.get_y() - (m*p1.get_x())

17
mklines.py
View File

@ -6,6 +6,7 @@ import csv
import sys import sys
import math import math
from os.path import exists as file_exists from os.path import exists as file_exists
from basictypes import PointWithID, Line
def main(): def main():
@ -29,18 +30,11 @@ def read_csv(filename: str, remove_header=False) -> list:
new_row = list() new_row = list()
if remove_header and i == 0: continue if remove_header and i == 0: continue
for i, val in enumerate(row): for i, val in enumerate(row):
if i == 0: new_row.append(val) if i == 0: new_row.append(val) # Append wall_id as is
else: new_row.append(float(val)) else: new_row.append(float(val))
tmp_data.append(new_row) tmp_data.append(new_row)
return tmp_data return tmp_data
def get_line_eq(point1, point2)-> tuple:
m = ((point2[1] - point1[1])/(point2[0] - point1[0]))
# y = mx + b
# b = y - mx
# Using p1 to find b
b = point1[1] - (m*point1[0])
def valid_data(samples:list) -> bool: def valid_data(samples:list) -> bool:
# Check if we have even number of samples # Check if we have even number of samples
return len(samples) % 2 == 0 return len(samples) % 2 == 0
@ -57,9 +51,6 @@ def polarlist_to_cart(polar_points: list):
p2 = polar_to_cart(p2[0],p2[1]) p2 = polar_to_cart(p2[0],p2[1])
def points_to_straight_line():
pass
def get_filename() -> str: def get_filename() -> str:
try: try:
filename = sys.argv[1] filename = sys.argv[1]
@ -68,8 +59,8 @@ def get_filename() -> str:
perr('Provide target file name') perr('Provide target file name')
return -1 return -1
def perr(msg): def perr(*args):
print(msg, file=sys.stderr) print(args, file=sys.stderr)
if __name__ == '__main__': if __name__ == '__main__':
exit(main()) exit(main())