shared wall implementation and code refactoring.

This commit is contained in:
Guille Gutierrez 2020-05-28 15:22:46 -04:00
parent 539db33b20
commit 9999c8a331
3 changed files with 57 additions and 11 deletions

View File

@ -1,5 +1,7 @@
from city_model_structure.city_object import CityObject from city_model_structure.city_object import CityObject
from typing import List, Union from typing import List, Union
import pyproj
import reverse_geocoder as rg
class City: class City:
@ -10,6 +12,33 @@ class City:
self._upper_corner = upper_corner self._upper_corner = upper_corner
self._city_objects = city_objects self._city_objects = city_objects
self._srs_name = srs_name self._srs_name = srs_name
# todo: right now extracted at city level, in the future should be extracted also at building level if exist
self._location = None
@property
def location(self):
if self._location is None:
gps = pyproj.CRS('EPSG:4326') # LatLon with WGS84 datum used by GPS units and Google Earth
input_reference = None
try:
input_reference = pyproj.CRS(self.srs_name) # Projected coordinate system from input data
except pyproj.exceptions.CRSError:
print('Invalid projection reference system, please check the input data. (e.g. in CityGML files: srs_name)')
quit()
coordinates = pyproj.transform(input_reference, gps, self.lower_corner[0], self.lower_corner[1])
self._location = rg.search(coordinates)
return self._location
@property
def country_code(self):
return self.location[0]['cc']
@property
def name(self):
if self._name is None:
self._name = self.location[0]['name']
return self._name
@property @property
def city_objects(self) -> Union[List[CityObject], None]: def city_objects(self) -> Union[List[CityObject], None]:
@ -43,10 +72,6 @@ class City:
def srs_name(self): def srs_name(self):
return self._srs_name return self._srs_name
@property
def name(self):
return self._name
@name.setter @name.setter
def name(self, value): def name(self, value):
self._name = value self._name = value

View File

@ -77,9 +77,6 @@ class Surface:
self._area = self.polygon.get_area() self._area = self.polygon.get_area()
return self._area return self._area
def _is_shared(self, surface):
return False
def _is_almost_same_terrain(self, terrain_points, ground_points): def _is_almost_same_terrain(self, terrain_points, ground_points):
equal = 0 equal = 0
for t in terrain_points: for t in terrain_points:
@ -145,7 +142,7 @@ class Surface:
def shared(self, surface): def shared(self, surface):
if self.type is not 'Wall': if self.type is not 'Wall':
return return
if self._is_shared(surface): if self._geometry.is_almost_same_surface(self, surface):
self._shared_surfaces.append((100, surface)) self._shared_surfaces.append((100, surface))
surface.shared(self) surface.shared(self)

View File

@ -10,6 +10,30 @@ class Geometry:
delta = math.sqrt(pow((v1[0] - v2[0]), 2) + pow((v1[1] - v2[1]), 2) + pow((v1[2] - v2[2]), 2)) delta = math.sqrt(pow((v1[0] - v2[0]), 2) + pow((v1[1] - v2[1]), 2) + pow((v1[2] - v2[2]), 2))
return delta <= self._delta return delta <= self._delta
def is_almost_same_surface(self, s1, s2):
# delta is grads an need to be converted into radians
delta = np.rad2deg(self._delta)
difference = (s1.inclination - s2.inclination) % math.pi
if abs(difference) > delta:
return False
# s1 and s2 are at least almost parallel surfaces
p1 = s1.polygon.get_parametric()
selected_coefficient = 0
for coefficient in p1:
if coefficient != 0:
break;
selected_coefficient += 1
# calculate distance point to plane
# select surface1 value for the point (X,Y,Z) where two of the values are 0
s1_coefficient = -p1[3] / p1[selected_coefficient]
p2 = s2.polygon.get_parametric()
n2 = s2.normal
parametric = abs(p2[2] * s1_coefficient + p2[3])
distance = parametric / math.sqrt(pow(n2[0], 2) + pow(n2[1], 2) + pow(n2[2], 2))
return distance <= self._delta
@staticmethod @staticmethod
def to_points_matrix(points, remove_last=False): def to_points_matrix(points, remove_last=False):
rows = points.size // 3 rows = points.size // 3