3 s] @s4ddlZddZddZddZdd Zd d ZdS) Ncs|j}|jfdd|D}g}g}x<|D]4}|j|j||jj|j|j||jjq0Wtj|}tj|}|jd}tj|dd}tj|dd} |tj ||df} |tj | |df} tj | | } tj j | \} }}|j | j }| |j | j }||fS)zqRecovers transformation needed to align structure1 with structure2. Modified from http://nghiaho.com/?page_id=671csg|]}|kr|qSr).0num) abs_indices2rq/oak/stanford/groups/akundaje/marinovg/Symbiodinium/2019-09-27-figures/multi_mds/miniMDS-master/linear_algebra.py sz%getTransformation..r)axis)nonzero_abs_indicesappendpointsoffsetposnpmatshapemeantile transposelinalgsvdT)Z structure1Z structure2Z abs_indices1 intersectionabrnZ centroid_aZ centroid_baabbhusvtrtr)rrgetTransformations(     r$cCs@|d|dd|d|dd|d|dddS)z&Euclidean distance between coordinatesrr g?g?r)Zcoord1Zcoord2rrr calcDistance&sr&cs>tj|j}tj|ddtfdd|D}|t|S)Nr)rcsg|]}t|qSr)r&)rcoord)centroidrrr-sz&radius_of_gyration..)rarray getCoordsrsumlen) structurecoordsZdist_sumr)r(rradius_of_gyration*sr/cCst|}tj|tj||f|}|j|dj| d}tjj|\}}tj|ddd}||}|dd|f}tj|dddf|dd|dddf|dd|dddf|dd gj S) zDModified from http://www.nervouscomputer.com/hfs/cmdscale-in-python/r%Nr rg?g?g?g?) r,reyeonesdotreighargsortr)r)distMatrrrevalsZevecsidxrrrcmds0sr9cCs&dddg}t||}||}dddg}dddg}dddg}tjtj||}tj||} t| |} | | } tj|d} tj|d| d} tj|d| d}tj|d| d}tj| d| d|d|dd| || |d| || |gd|| | || d| d|d|dd||| | gd|| | |d||| | | d| d|d|dgg}tj|j}tj|j}tj|j}||}||}||}tj |}xDt t |D]4}||}tj||tj||tj||g||<qW|S)z5Rotate 3-D coords such that vector n aligns to z-axisrr r%) r&rarccosr3crosscossinmatrixr zeros_likeranger,)rr.originZlen_nZn_hatZi_hatZj_hatZk_hatthetarZlen_bb_hatq0q1q2Zq3QZu_hatZv_hatZw_hatZ new_coordsiprrrchange_coordinate_systemEs4        ,rJ)numpyrr$r&r/r9rJrrrrs #