a J5d@sddlmZddlmZddlZddlZddlmZdej eej e e e ej d d d Z ej ej e e d d dZ ej ej e e e dddZdS))ceil)OptionalN) process_inputhttp://localhost:5000 distances)sources destinationsosrm_server_addressosrm_batch_size cost_typereturncCst||\}}|jd}|jd}t||f}t||}t||} t|D]~} | |} t| d||} t| D]V} | |}t| d||}|| | }|||}t||||d|| | ||f<qvqP|S)aCompute distance matrix from sources to destinations using OSRM service Parameters ---------- sources, destinations 2D Arrays of coordinates in the form [lat, lng] for each row Also, if ``destinations`` is `None`, compute the distance between each source in ``sources``. osrm_server_address Base address of the OSRM server instance osrm_batch_size Subset of sources and destinations for each request. This reduces the request size, thus alleviating max table issues, but slows down the search cost_type "distances" to get the street distances and "durations" for the street time Returns ------- Distance (or duration) matrix of size `num_sources x num_destinations` rr)r )rshapenpZzerosrrangemin_get_batch_osrm_distance)r r r r r num_sourcesnum_destinationsZ cost_matrixZ num_batches_iZ num_batches_jiZstart_iZend_ijZstart_jZend_j sources_batchdestinations_batchr^/var/www/html/django/DPS/env/lib/python3.9/site-packages/python_tsp/distances/osrm_distance.pyosrm_distance_matrix s0          r)rrr r cCs2t||||}t|}|t||S)z2Request the OSRM distance matrix for a given batch)_format_osrm_urlrequestsgetraise_for_statusrarrayjson)rrr r urlresprrrrIs  r)rrr r rc Cs|dd}ddd|D}t||rN|jddkrN|d|d |Sdd d|D}|d|}|jd}|jd} dd dt|D} dd dt||| D} |d|d | d| d| S)aiFormat OSRM url string with sources and destinations Notes ----- Consider the N sources in the form (lat_src1, lgn_src1), (lat_src2, lgn_src2), ... and the M destinations in the form (lat_dest1, lgn_dest1), (lat_dest2, lgn_dest2), ... This function converts these properties in a URL of the form {OSRM_SERVER_ADDRESS}/table/v1/driving/ lng_src1,lat_src1;lng_src2,lat_src2;...;lng_srcN,lat_srcN; lng_dest1,lat_dest1;lng_dest2,lat_dest2;...;lng_destM,lng_destM ?sources=0;1;...;N-1 &destinations=N;N+1;...;N+M-1 &annotations=distance In the simpler case when sources == destinations, the URL is simplified to {OSRM_SERVER_ADDRESS}/table/v1/driving/ lng_src1,lat_src1;lng_src2,lat_src2;...;lng_srcN,lat_srcN ?annotations=distance Obs: Replace "distance" with "duration" if a time matrix is required Obs2: The matrix type follows the singular form in the URL (e.g., "distance"), but the returned JSON follows the plural form (e.g., "distances"). Thus, we ignore the last letter of the input type N;css$|]}|dd|dVqdSr,rNr).0sourcerrr }sz#_format_osrm_url..rrz/table/v1/driving/z ?annotations=css$|]}|dd|dVqdSr'r)r) destinationrrrr+scss|]}t|VqdSNstrr)indexrrrr+css|]}t|VqdSr-r.r0rrrr+sz ?sources=z&destinations=z &annotations=)joinrZ array_equalrr) rrr r Z url_cost_typeZ sources_coordZdestinations_coordZlocations_coordrrZsources_indicesZdestinations_indicesrrrrYsD"          r)Nrrr)mathrtypingrnumpyrrZdata_processingrZndarrayr/intrrrrrrrs6    @