3 ?*td<%@sJddlZddlZddlZddlZdZdZ dZ dZ ddZ Gdd d Z dS) Nzassets/new_master_list.csvz1assets/XJETS-master - col_name_attributes_new.csvz#assets/data_downloads/data_info.csvzassets/all_refs.csvc stj|tj|t|t|kr0tddSt|}krVfdd|D}ndg}g}g}xt|D]}|dkr||}n|}||dk r|||dd} nd} | }|j|||g|j| ||g|j|| dqnW||fS)z Generate a color scale for int type categories. Each category will be at the center of a bin. The will generate a discrete color bar but may have unequal bin widths z1Length of values is greater than length of colorsNcsg|]}|qSr).0i)max_valmin_valr/var/www/atlasX/datastore.py sz'generate_colorscale..r)npmaxminlenprintrangeappend) valuescolorsZnvalsZ norm_values colorscale ticklocationsrlbZprev_ubZubr)rrrgenerate_colorscales.    rc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+S), datastorecCsytjt|_|jj|_t|jj|_|jj|_ tjt j d|_ tjt |_ |jj|j ddd|_tjtjdj|_d|_|jjdd|_Wn0tk r}zt|d |_WYdd}~XnXdS) NleftName)howon attributeZgoodT)deepZbad)pdZread_csv master_csv master_dfcolumns orig_columnsrindex total_sources col_namescitations_filefillnarefs_dfdata_availabilitymergecol_attrZ set_indexT col_attr_dfstatecopy endpoint_df ValueErrorr)selfZverrrrr__init__7s    zdatastore.__init__cCs|jS)N)r$)r4rrrget_orig_columnsIszdatastore.get_orig_columnscCs|jS)N)r")r4rrrget_master_dataLszdatastore.get_master_datacCs|j|jddk|_dS)NZbapp3r )r2)r4rrr_filter_super_luminal_sourcesOsz'datastore._filter_super_luminal_sourcescCs|j|jddk|_dS)Nz X-ray firstr)r2)r4rrr_filter_xray_upstream_offsetRsz&datastore._filter_xray_upstream_offsetcCs|j|jddk|_dS)Nz Radio firstr)r2)r4rrr_filter_xray_downstream_offsetUsz(datastore._filter_xray_downstream_offsetcCs(|j|jddk|jddk@|_dS)Nz Radio firstrz X-ray first)r2)r4rrr(_filter_xray_upstream_downstream_offsetsXs z2datastore._filter_xray_upstream_downstream_offsetscCs|jS)N)r2)r4rrrget_filtered_data^szdatastore.get_filtered_datacCs|jjdd|_dS)NT)r)r"r1r2)r4rrr reset_filtersaszdatastore.reset_filterscCs|j|jddkS)N use_as_scaler )r/)r4rrrget_valid_scalesdszdatastore.get_valid_scalescCs|j|jddkS)N use_as_catr )r/)r4rrrget_valid_categoriesgszdatastore.get_valid_categoriescCs|j|jdjddkS)NZ valid_axisintr )r/astype)r4rrrget_valid_axesjszdatastore.get_valid_axescCs|jS)N)r/)r4rrrget_col_attributesmszdatastore.get_col_attributescCs dddgS)NzSuper luminal sourcesr8)labelvaluer)r4rrr get_filterspszdatastore.get_filterscCsD|j|jdjddk}dd|jD}|jddddd|S) Nr>rBr cSs$g|]\}}|d||ddqS) pretty_name)rFrG primaryTextr)rrrowrrrrsz.datastore.get_scaling_vars..rz No scaling)rFrGrJ)r/rCiterrowsinsert)r4Z scales_dfoptionsrrrget_scaling_vars|s   zdatastore.get_scaling_varscCs<|j|jd}d}d|j|d}|jdd|ddS)Ng?(r area)sizeZsizemodeZsizeminsizerefcolor)r2r)r tolist)r4Z scale_usingscalesZmax_derired_marker_sizerTrrrget_marker_scaling_optionssz$datastore.get_marker_scaling_optionscCsD|j|jdjddk}dd|jD}|jddddd|S) Nr@rBr cSs$g|]\}}|d||ddqS)rI)rFrGrJr)rrrKrrrrsz2datastore.get_categorical_vars..rz No Category)rFrGrJ)r/rCrLrM)r4Zcat_dfrNrrrget_categorical_varss    zdatastore.get_categorical_varscs|j|jd}|ddkjrtj|j}tj|t|}|dkrPddiS|dkrt|dkrttj ddd}ntj t |dd }t ||\}}|dkrdg}fd d |j D||j j|d d d|dd |DddSddiSdS)Nrr rU3ZqualZDark2divZSpectralcsg|] }|qSrr)rr)min_unique_valrrrsz+datastore.get_cat_color..rItoparraycSsg|]}tt|qSr)strrB)rrrrrrs)titleZ titlesideZtickmodeZtickvalsZticktext)rUrZcolorbar)r2r)allr sortuniquer rclrWr`rrVr/loc)r4Z cat_usingZ cat_valuesZ unique_valsZ n_unique_valsrrrr)r]r get_cat_colors,  zdatastore.get_cat_colorcs@jjd|kjdjdtjfddjDddS)NrrNAcs6g|].}tjtjjj|dtj|gqS)rI)htmlTrTdr/rf)rcol)rKr4rrrsz.datastore.get_source_table..r) className)r2ilocr)riTabler')r4sourcer)rKr4rget_source_tables   zdatastore.get_source_tablec Csz|jdj|jdj|jdj|jdj|jdj|jdj|jdj|jdjt|jd j|jd S) Nz#resolved X_ray knotsz#knots with offsetsz#knots without offsetsz #HS Xraysz#HS with offsetsz#HS without offsetsz X-ray firstz Radio firstz/X-ray jet terminates before radio after a bend?) Ztotal_resolved_xray_knotsZtotal_knots_with_offsetsZtotal_knots_without_offsetsZtotal_resolved_xray_HSZtotal_HS_with_offsetsZtotal_HS_without_offsetsZxray_first_offsetsZradio_first_offsetsZ#total_xray_jet_terminates_at_a_bendZ total_jets)r"sumrBr&)r4rrrget_summary_datas        zdatastore.get_summary_datac Cstd||jjd|d}|jddkr0dSg}dd}xV|jD]J\}}|jtj||tjd|ddtjtj d |d d d gqFWtd |tj |S)NsrczName2==""rrc Ss\d}|ddkr$dt|dd}|dd|dd|d d |d |d |d  S)Nrnumber()Zauthorz (yearz), Zjournal2z, volumez, p.pages)rB)rnumrrr get_apa_cites z2datastore.get_formatted_refs..get_apa_citeraz[ADS]linkZ_blank)hreftargetZref_l) rr*queryshaperLrriLiBrAOl)r4rprefsZref_listr~r%rKrrrget_formatted_refss   zdatastore.get_formatted_refsN)__name__ __module__ __qualname__r5r6r7r8r9r:r;r<r=r?rArDrErHrOrXrYrgrqrsrrrrrr6s*  0r)pandasr numpyr Z colorloverredash_html_componentsrir!r-r+r(rrrrrrs'