o Uݢg@sUddlmZddlZddlZddlmZddlmZddlm Z e e ej ej edZded<ed ed ed gZed Zed ZedZedZedZedZeegZdZededededgZededeegeZededededgZedededgeZeeZed ed!ed"ed#gZ ed$d%gZ!e"e#Z$d&Z%d'Z&d(Z'dd1d2Z*Gd3d*d*eZ+e,e-e+fZ.d?d6d7Z/d@d:d;Z0dS)A) annotationsN) OrderedDict) TypedDict)GENE_EXPRESSION_LIBRARY_TYPEzchemistry_defs.jsonzdict[str, ChemistryDef]CHEMISTRY_DEFSzSC3Pv3HT-polyAz SC3Pv3HT-CS1ZSC5PHTZSC3Pv3LTzSC-FBZ SPATIAL3Pv1zARC-v1z SC3Pv3-polyAz SC3Pv3-CS1ZSC3Pv4z SC3Pv4-polyAz SC3Pv4-CS1zSC3Pv4-polyA-OCMzSC3Pv4-CS1-OCMZSC3Pv1ZSC3Pv2z SC5P-PE-v3z SC5P-R2-v3zSC5P-R2-OCM-v3z SC5P-R1-v3zSC5P-PEzSC5P-R2zSC5P-R1ZSCVDJzSCVDJ-R2zSCVDJ-v3z SCVDJ-R2-v3)Z threeprimeZ SC3P_auto)Z fiveprimeZ SC5P_autocustom descriptionnamereturn ChemistryDefcsZtfddtD}t|dkrtdt|dkr)td|dS)z5Returns a chemistry definition dict for a given name.csg|] }|dkr|qSr .0nr r c/oak/stanford/groups/akundaje/marinovg/programs/cellranger-9.0.1/lib/python/cellranger/chemistry.py oz!get_chemistry..rzCould not find chemistry named z%Found multiple chemistries with name )CHEMISTRY_ALIASESgetDEFINED_CHEMISTRIESlen ValueError)r chemistriesr r r get_chemistryks   rstrcsNfddtD}t|dkrtdt|dkr#td|dS)zMGiven all the loaded chemistries, return the one with a matching description.csg|] }|tkr|qSr )CHEMISTRY_DESCRIPTION_FIELDrrr rryrz2get_chemistry_from_description..rz*Could not find chemistry with description rz,Found multiple chemistries with description )rrr)rrr rrget_chemistry_from_descriptionws   r None | strcCs<t|}d}||}|durt|dkr|ddSdS)a0Try to load a chemistry whitelist from a description of a chemistry. Args: description: Returns: the whitelist if it's a simple chemistry or None, since we don't enforce that a whitelist exists for all chemistries explicitly in code (it's just a JSON file). barcodeNrr whitelist)rrr)rchemr!Zbarcode_sectionr r r-get_whitelist_name_from_chemistry_descriptions   r$c@sReZdZUdZded<ded<ded<ded<d ed <d ed <ded <ded<dS)r zA chemistry definition.rr rz+list[dict[str, int | str | dict[str, str]]]r!zlist[dict[str, int | str]]Zumizdict[str, int | str | None]rnaz"dict[str, int | str | None] | NoneZrna2Z endednessZ strandednessN)__name__ __module__ __qualname____doc____annotations__r r r rr s  chemistry_defs ChemistryDefsc CsJt|dkrtt|Sz|tWSty$}ztd|d}~ww)a Return the only chemistry definition when there is only one. Return the GEX chemistry definition when there are multiple. Fail when there are multiple chemistry definitions and no GEX. This matches the implementation in the Rust ChemistryDefs type. rz"no gene expression chemistry foundN)rnextitervaluesrKeyErrorr)r+excr r rget_primary_chemistry_defs   r2defsset[str]cCsXt}|D]"}|dd}|ddksJ|dd}t|ts$J||q|S)zEGet all gel bead whitelists from the provided dict of chemistry defs.r!rkindZgel_beadr"r )setr/ isinstanceradd)r3Z whitelistsr#Zbarcode_read_componentr"r r r"get_whitelists_from_chemistry_defss    r9)r r )rrr r )rrr r )r+r,r r )r3r,r r4)1 __future__rjsonos collectionsrtypingrcellranger.rna.libraryrloadopenpathjoindirname__file__rr*HT_CHEMISTRIESCHEMISTRY_SC3P_LTZCHEMISTRY_SC_FBZCHEMISTRY_SPATIAL3P_V1ZCHEMISTRY_ARC_V1ZCHEMISTRY_SC3P_V3_POLYAZCHEMISTRY_SC3P_V3_CS1SC3P_V3_CHEMISTRIESZSC3P_V4_PREFIXSC3P_V4_CHEMISTRIESZSC3P_CHEMISTRIESSC5P_V3_CHEMISTRIESSC5P_CHEMISTRIESZSC_GEMX_CHEMISTRIESZSCVDJ_CHEMISTRIESrlistr/rZCUSTOM_CHEMISTRY_NAMErZCHEMISTRY_NAME_FIELDrrr$r dictrr,r2r9r r r rs