1818 "White_whole" : "red" ,
1919 "Varshney" : "#009e73" ,
2020 "Bentley2016_MA" : "#56b4e9" ,
21- "Bentley2016_PEP" : "blue " ,
21+ "Bentley2016_PEP" : "fuchsia " ,
2222 "Cook2019Herm" : "darkmagenta" ,
23- "Cook2019Male" : "fuchsia " ,
23+ "Cook2019Male" : "blue " ,
2424 "Cook2020" : "darkorchid" ,
2525 "Brittin2021" : "khaki" ,
2626 "Witvliet1" : "#eeee44" ,
@@ -190,6 +190,9 @@ def get_hive_plot_markdown(reader_name, view, connectome, synclass, indent="
190190 if np .sum (connectome .connections [synclass ]) == 0 :
191191 return None
192192
193+ if "brainmap" in view .id .lower ():
194+ return f"\n { indent } Hive plot of that view, { view_id } , is not possible, as it is the BrainMap view\n "
195+
193196 fig = connectome .to_plotly_hive_plot_fig (synclass , view )
194197
195198 if fig is None :
@@ -209,6 +212,43 @@ def get_hive_plot_markdown(reader_name, view, connectome, synclass, indent="
209212 return f'\n { indent } <br/>\n { indent } ```{{.plotly .no-auto-theme}}\n { indent } {{ "file_path": "./{ asset_filename } " }}\n { indent } ```\n '
210213
211214
215+ def get_improved_reader_name (reader_name ):
216+ better_name = (
217+ reader_name .replace ("_" , " " )
218+ .replace ("201" , " 201" )
219+ .replace ("202" , " 202" )
220+ .replace ("Sanchez" , " Sanchez et al. 2023" )
221+ .replace ("tley" , "tley et al." )
222+ .replace ("Cook" , "Cook et al." )
223+ .replace ("Wang" , "Wang et al." )
224+ .replace ("ttin" , "ttin et al." )
225+ .replace ("im" , "im et al." )
226+ .replace ("Herm" , " (herm.)" )
227+ .replace ("Male" , " (male)" )
228+ .replace ("ShortRange" , " (short range)" )
229+ .replace ("liet" , "liet " )
230+ .replace ("MA" , " (monoamin.)" )
231+ .replace ("PEP" , " (peptid.)" )
232+ .replace ("ite A" , "ite et al. 1986 N2U/adult" )
233+ .replace ("ite L4" , "ite et al. 1986 JSU/L4" )
234+ .replace ("ite whole" , "ite et al. 1986 (whole worm)" )
235+ .replace ("Randi" , "Randi et al." )
236+ .replace ("Varshney" , "Varshney et al. 2011" )
237+ .replace ("Witvliet 1" , "Witvliet et al. 2021 1 (L1)" )
238+ .replace ("Witvliet 2" , "Witvliet et al. 2021 2 (L1)" )
239+ .replace ("Witvliet 3" , "Witvliet et al. 2021 3 (L1)" )
240+ .replace ("Witvliet 4" , "Witvliet et al. 2021 4 (L1)" )
241+ .replace ("Witvliet 5" , "Witvliet et al. 2021 5 (L2)" )
242+ .replace ("Witvliet 6" , "Witvliet et al. 2021 6 (L3)" )
243+ .replace ("Witvliet 7" , "Witvliet et al. 2021 7 (adult)" )
244+ .replace ("Witvliet 8" , "Witvliet et al. 2021 8 (adult)" )
245+ .replace ("Gleeson" , "Gleeson et al. 2018" )
246+ .replace ("Olivares" , "Olivares et al. 2021" )
247+ .replace ("Model" , " (model)" )
248+ )
249+ return better_name
250+
251+
212252def generate_comparison_page (
213253 quick : int ,
214254 color_table = True ,
@@ -226,28 +266,30 @@ def generate_comparison_page(
226266 # readers["Wang2024Herm"] = ["cect.Wang2024HermReader", "Wang_2024"]
227267 # readers["Bentley2016_MA"] = ["cect.WormNeuroAtlasMAReader", "Bentley_2016"]
228268 # readers["White_A"] = ["cect.White_A", "White_1986"]
229- # readers["White_whole"] = ["cect.White_whole", "White_1986"]
269+ readers ["White_whole" ] = ["cect.White_whole" , "White_1986" ]
230270 readers ["Test" ] = ["cect.TestDataReader" , None ]
231271
232272 # readers["WormNeuroAtlas"] = ["cect.WormNeuroAtlasReader", "Randi_2023"]
233273
234274 # readers["Randi2023"] = ["cect.WormNeuroAtlasFuncReader", "Randi_2023"]
235275
236- # readers["Brittin2021"] = ["cect.BrittinDataReader", "Brittin_2021"]
237- # readers["Yim2024"] = ["cect.Yim2024DataReader", "Yim_2024"]
276+ readers ["Brittin2021" ] = ["cect.BrittinDataReader" , "Brittin_2021" ]
277+ readers ["Yim2024" ] = ["cect.Yim2024DataReader" , "Yim_2024" ]
238278 # readers["Yim2024NonNorm"] = ["cect.Yim2024NonNormDataReader", "Yim_2024"]
239279
240280 # readers["White_whole"] = ["cect.White_whole", "White_1986"]
241- readers ["GleesonModel" ] = ["cect.GleesonModelReader" , "GleesonModel" ]
242- readers ["OlivaresModel" ] = ["cect.OlivaresModelReader" , "OlivaresModel" ]
281+ # readers["GleesonModel"] = ["cect.GleesonModelReader", "GleesonModel"]
282+ # readers["OlivaresModel"] = ["cect.OlivaresModelReader", "OlivaresModel"]
243283
244284 # readers["Cook2019Herm"] = ["cect.Cook2019HermReader", "Cook_2019"]
245- # readers["Cook2020"] = ["cect.Cook2020DataReader", "Cook_2020"]
285+ # readers["Cook2019Male"] = ["cect.Cook2019MaleReader", "Cook_2019"]
286+ readers ["Cook2020" ] = ["cect.Cook2020DataReader" , "Cook_2020" ]
246287
247- readers ["OpenWormUnified" ] = ["cect.OpenWormUnifiedReader" , "OpenWorm_Unified" ]
288+ # readers["OpenWormUnified"] = ["cect.OpenWormUnifiedReader", "OpenWorm_Unified"]
248289
249290 # readers["Witvliet7"] = ["cect.WitvlietDataReader7", "Witvliet_2021"]
250- readers ["Witvliet8" ] = ["cect.WitvlietDataReader8" , "Witvliet_2021" ]
291+ # readers["Witvliet8"] = ["cect.WitvlietDataReader8", "Witvliet_2021"]
292+ # readers["Wang2024Herm"] = ["cect.Wang2024HermReader", "Wang_2024"]
251293
252294 else :
253295 if not quick :
@@ -286,7 +328,7 @@ def generate_comparison_page(
286328 if not quick :
287329 readers ["WormNeuroAtlas" ] = ["cect.WormNeuroAtlasReader" , "Randi_2023" ]
288330
289- readers ["Randi2023" ] = ["cect.WormNeuroAtlasFuncReader" , "Randi_2023" ]
331+ readers ["Randi2023" ] = ["cect.WormNeuroAtlasFuncReader" , "Randi_2023" ]
290332
291333 if not quick :
292334 readers ["RipollSanchezShortRange" ] = [
@@ -422,7 +464,7 @@ def generate_comparison_page(
422464
423465 f .write (
424466 '---\n title: "Dataset: %s"\n search:\n exclude: true\n ---\n \n '
425- % reader_name
467+ % get_improved_reader_name ( reader_name )
426468 )
427469
428470 desc_full = ""
@@ -670,7 +712,13 @@ def generate_comparison_page(
670712 if c in connectome .nodes :
671713 total_here += 1
672714
673- view_info += f"| Nodes in current view<br/>({ len (view .node_sets )} total)| Num cells in node<br/>({ total_cells } total) | Num in this dataset<br/>({ total_here } total) | Cells |\n | --- | --- | --- | --- |\n "
715+ view_info += (
716+ f'| Nodes in current view<br/>({ len (view .node_sets )} total) <span style="color:#ffffff;">xxxxxxxxxxxxxxxxxxxxxxxxx</span>| '
717+ + f"Num cells in node<br/>({ total_cells } total) | "
718+ + f"Num in this dataset<br/>({ total_here } total) | "
719+ + "Cells in node |\n "
720+ + "| --- | --- | --- | --- |\n "
721+ )
674722
675723 for ns in view .node_sets :
676724 n_in_dataset = np .sum (
@@ -681,6 +729,12 @@ def generate_comparison_page(
681729 )
682730 node_colored = ns .to_markdown ()
683731
732+ node_desc = (
733+ " <br/><i>%s</i>" % ns .description
734+ if ns .description is not None
735+ else ""
736+ )
737+
684738 cells_linked = [
685739 get_cell_internal_link (
686740 c ,
@@ -692,8 +746,9 @@ def generate_comparison_page(
692746 )
693747 for c in sorted (ns .cells )
694748 ]
695- view_info += "|**%s** |%i | %i | %s|\n " % (
749+ view_info += "|**%s**%s |%i | %i | %s|\n " % (
696750 node_colored ,
751+ node_desc ,
697752 len (ns .cells ),
698753 n_in_dataset ,
699754 ", " .join (cells_linked ),
@@ -842,35 +897,7 @@ def generate_comparison_page(
842897
843898 better_names = {}
844899 for reader_name in readers_to_include :
845- better_name = (
846- reader_name .replace ("_" , " " )
847- .replace ("201" , " 201" )
848- .replace ("202" , " 202" )
849- .replace ("Sanchez" , " Sanchez et al. 2023" )
850- .replace ("tley" , "tley et al." )
851- .replace ("Cook" , "Cook et al." )
852- .replace ("Wang" , "Wang et al." )
853- .replace ("ttin" , "ttin et al." )
854- .replace ("im" , "im et al." )
855- .replace ("Herm" , " (herm.)" )
856- .replace ("Male" , " (male)" )
857- .replace ("ShortRange" , " (short range)" )
858- .replace ("liet" , "liet " )
859- .replace ("MA" , " (monoamin.)" )
860- .replace ("PEP" , " (peptid.)" )
861- .replace ("ite A" , "ite et al. 1986 N2U/adult" )
862- .replace ("ite L4" , "ite et al. 1986 JSU/L4" )
863- .replace ("ite whole" , "ite et al. 1986 (whole worm)" )
864- .replace ("Randi" , "Randi et al." )
865- .replace ("Varshney" , "Varshney et al. 2011" )
866- .replace ("Witvliet 1" , "Witvliet et al. 2021 1 (L1)" )
867- .replace ("Witvliet 5" , "Witvliet et al. 2021 5 (L2)" )
868- .replace ("Witvliet 6" , "Witvliet et al. 2021 6 (L3)" )
869- .replace ("Witvliet 8" , "Witvliet et al. 2021 8 (adult)" )
870- .replace ("Gleeson" , "Gleeson et al. 2018" )
871- .replace ("Olivares" , "Olivares et al. 2021" )
872- .replace ("Model" , " (model)" )
873- )
900+ better_name = get_improved_reader_name (reader_name )
874901 better_names [reader_name ] = better_name
875902
876903 # table_html += f' <th style={STYLE}><span style="font-size:150%">{better_name}</span></th>\n'
@@ -946,16 +973,51 @@ def generate_comparison_page(
946973
947974
948975if __name__ == "__main__" :
949- quick = len (sys .argv ) > 1 and eval (sys .argv [1 ])
976+ if "-color" in sys .argv :
977+ from cect .Cells import CORE_ANATOMICAL_CONNECTOMES
978+
979+ import matplotlib .pyplot as plt
980+
981+ plt .figure (figsize = (8 , 6 ))
982+
983+ for conn in CORE_ANATOMICAL_CONNECTOMES :
984+ color = reader_colors [conn ]
985+ print (f"{ conn } : { color } " )
986+ name = get_improved_reader_name (conn )
987+ plt .plot (
988+ [0 ],
989+ [1 ],
990+ label = f"{ name } " ,
991+ color = color ,
992+ linestyle = "None" ,
993+ marker = "o" ,
994+ markersize = 6 if "Cook2019Herm" in conn else 3 ,
995+ )
950996
951- save_to_cache = True
997+ handles , labels = plt .gca ().get_legend_handles_labels ()
998+ ncol = 4
999+ plt .legend (
1000+ handles ,
1001+ labels ,
1002+ loc = "upper center" ,
1003+ bbox_to_anchor = (0.5 , - 0.15 ),
1004+ ncol = ncol ,
1005+ frameon = False ,
1006+ )
1007+ plt .gcf ().subplots_adjust (bottom = 0.25 )
9521008
953- connectomes = generate_comparison_page (
954- quick ,
955- color_table = True ,
956- dataset_pages = False ,
957- save_to_cache = save_to_cache ,
958- load_from_cache = (not save_to_cache ),
959- )
1009+ plt .show ()
1010+ else :
1011+ quick = len (sys .argv ) > 1 and eval (sys .argv [1 ])
1012+
1013+ save_to_cache = True
1014+
1015+ connectomes = generate_comparison_page (
1016+ quick ,
1017+ color_table = True ,
1018+ dataset_pages = False ,
1019+ save_to_cache = save_to_cache ,
1020+ load_from_cache = (not save_to_cache ),
1021+ )
9601022
961- print ("Finished. All loaded connectomes:\n %s" % connectomes )
1023+ print ("Finished. All loaded connectomes:\n %s" % connectomes )
0 commit comments