a J5d@sddlZddlZddlmZgdZejfddZejfddZejfd d Zejd fd d Z ejfddZ dddejfddZ eeeeeeej eej de e e ej e ddddd ZdS)N)utils)TYPES euclidean manhattanmaximum geographicalpseudo_euclideanxraycCsDt|t|krtdtddt||D}t|}||S)aReturn the Euclidean distance between start and end. This is capable of performing distance calculations for EUC_2D and EUC_3D problems. If ``round=math.ceil`` is passed, this is suitable for CEIL_2D problems as well. :param tuple start: *n*-dimensional coordinate :param tuple end: *n*-dimensional coordinate :param callable round: function to use to round the result :return: rounded distance (dimension mismatch between start and endcss|]}||VqdSN.0dr r N/var/www/html/django/DPS/env/lib/python3.9/site-packages/tsplib95/distances.py "zeuclidean..len ValueErrorsumrdeltasmathsqrt)startendroundZsquare_distancedistancer r rrs  rcCs:t|t|krtdtddt||D}||S)aXReturn the Manhattan distance between start and end. This is capable of performing distance calculations for MAN_2D and MAN_3D problems. :param tuple start: *n*-dimensional coordinate :param tuple end: *n*-dimensional coordinate :param callable round: function to use to round the result :return: rounded distance r css|]}t|VqdSr absrr r rr6rzmanhattan..)rrrrrrrrrr r rr(s rcCs:t|t|krtdtddt||D}||S)aVReturn the Maximum distance between start and end. This is capable of performing distance calculations for MAX_2D and MAX_3D problems. :param tuple start: *n*-dimensional coordinate :param tuple end: *n*-dimensional coordinate :param callable round: function to use to round the result :return: rounded distance r css|]}t|VqdSr rrr r rrIrzmaximum..)rrmaxrrr!r r rr;s rgSc@cCst|t|krtdt|}t|}t|j|j}t|j|j}t|j|j}|tdd||d||d}t |S)azReturn the geographical distance between start and end. This is capable of performing distance calculations for GEO problems. :param tuple start: *n*-dimensional coordinate :param tuple end: *n*-dimensional coordinate :param callable round: function to use to round the result :param float radius: the radius of the Earth :return: rounded distance r g?r) rrrZ RadianGeorcosZlngZlatacosint)rrrZradiusq1q2Zq3rr r rrNs   *rcCs\t|t|krtdtddt||D}t|d}||}||krX|d7}|S)aMReturn the pseudo-Euclidean distance between start and end. This is capable of performing distance calculations for ATT problems. :param tuple start: *n*-dimensional coordinate :param tuple end: *n*-dimensional coordinate :param callable round: function to use to round the result :return: rounded distance r css|]}||VqdSr r rr r rrtrz#pseudo_euclidean.. rr)rrrZ square_sumvaluerr r rr gs r g?c Cst|t|kst|dkr$tdt|d|d}t|t|d}t|d|d}t|d|d}t||||||} |d| S)aReturn x-ray crystallography distance. This is capable of performing distance calculations for xray crystallography problems. As is, it is suitable for XRAY1 problems. However, using ``sx=1.25``, ``sy=1.5``, and ``sz=1.15`` makes it suitable for XRAY2 problems. :param tuple start: 3-dimensional coordinate :param tuple end: 3-dimensional coordinate :param float sx: x motor speed :param float sy: y motor speed :param float sz: z motor speed :return: distance z#start and end must be 3-dimensionalrihrgY@)rrr minr") rrsxsyszrZdxZdyZdzrr r rr ~sr )rg?g?gffffff?)r-r.r/) ZEUC_2DZEUC_3DZMAX_2DZMAX_3DZMAN_2DZMAN_3DZCEIL_2DZGEOZATTZXRAY1ZXRAY2) functoolsrr__all__Znintrrrrr r partialceilrr r r rs*