road_segmentation.py

"""
@author: Bin Xu-Sigurdsson, Fraunhofer ISE
"""

import os 
import pandas as pd
import geopandas as gpd
from shapely.geometry import Point
import osmnx as ox
import warnings
warnings.simplefilter(action='ignore', category=FutureWarning)
import shapely
import numpy as np

#from pyproj import Proj, transform
from collections import Counter

# may need to change this
filepath = r'C:\street_segmentation'

  

def read_csv(file):
    """
    read csv file and find the centriod of polygons
    """
    try:
        if os.path.exists(file):            
            data = pd.read_csv(file,  skiprows=[i for i in range(0,19)],header=None, sep =";", decimal = ",") 
            data.columns = data.iloc[0]
            data = data.loc[2:1790]
            
            data.iloc[:,39:52]= data.iloc[:,39:52].apply(pd.to_numeric, errors='coerce', axis=1)
            data['Longitude'] = [x.replace(',', '.') for x in data['Longitude']]
            data['Longitude'] = data['Longitude'].astype(float)
            data['Latitude'] = [x.replace(',', '.') for x in data['Latitude']]
            data['Latitude'] = data['Latitude'].astype(float)
            data['X-coordinate'] = [x.replace(',', '.') for x in data['X-coordinate']]
            data['X-coordinate'] = data['X-coordinate'].astype(float)
            data['Y-coordinate'] = [x.replace(',', '.') for x in data['Y-coordinate']]
            data['Y-coordinate'] = data['Y-coordinate'].astype(float)
            
            
            #data['Title'] = data3['Title'].astype(int)
            #df = gpd.GeoDataFrame(data, geometry=gpd.points_from_xy(data.Longitude, data.Latitude))
            #data.loc[:,'hd_Specific_space_heating_demand']=data.loc[:,'hd_Specific_space_heating_demand'].replace(0, np.nan)
            #data= data.dropna(subset=['hd_Specific_space_heating_demand'])        
            #data=data.drop_duplicates()
            #data['centriod_coord'] = data['geometry'].apply(lambda x: x.centroid)  
            
            return(data)
            
    except OSError:
            print('Error: Input file is missing %s' %(file))

# def proj2lonlat(x1, y1): 
#     """
    
#     """
#     inProj = Proj(init= goal_crs)
#     outProj = Proj(init='epsg:4326')
#     x2,y2 = transform(inProj,outProj,x1,y1)
#     return(x2,y2)   

def get_osm(outputfile):
    """
    
    get osm street data
    save graph nodes and edges as ESRI shapefiles to disk.
    """