RNApolis 0.1.5__py3-none-any.whl → 0.2.0__py3-none-any.whl

{RNApolis-0.1.5.dist-info → RNApolis-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: RNApolis
-Version: 0.1.5
+Version: 0.2.0
 Summary: A Python library containing RNA-related bioinformatics functions and classes
 Home-page: https://github.com/tzok/rnapolis-py
 Author: Tomasz Zok
@@ -15,6 +15,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Topic :: Scientific/Engineering :: Bio-Informatics
 Description-Content-Type: text/markdown
 License-File: LICENSE
+Requires-Dist: graphviz
 Requires-Dist: mmcif
 Requires-Dist: numpy
 Requires-Dist: ordered-set

RNApolis-0.2.0.dist-info/RECORD

@@ -0,0 +1,15 @@
+rnapolis/annotator.py,sha256=XtEJrog8c3zZ-pLlGBytJExGahPstGd-nbPLKQLa58A,21228
+rnapolis/clashfinder.py,sha256=jD3s_UovygWi01NUbQNeAeRRFkARTSRraLXUV43UbAA,8514
+rnapolis/common.py,sha256=DPmRpNkMaxuIai3vfLzSlP6IN0zpj6kmT3LoRjnJUWE,26440
+rnapolis/metareader.py,sha256=4qtMKRvww2sUStLeV8WVrLEt-ScydHUv4Gxx96tnf-M,1683
+rnapolis/motif_extractor.py,sha256=duHvpi9Ulcny9K60E6VBpz5RpJZw-KdTB4_Ph0iP478,774
+rnapolis/parser.py,sha256=Z3Zd_IuRyOP45x5BStgu7UgoyHthhw55fT3udHUhAE4,11905
+rnapolis/tertiary.py,sha256=iWMPD9c21rjMPpEdBd7mPCQgds65IbOr4_Fy06s0NoU,18957
+rnapolis/transformer.py,sha256=V9nOQvdq4-p7yUWo0vQg0CDQMpmyxz9t4TMSRVEKHnw,1817
+rnapolis/util.py,sha256=IdquFO3PV1_KDqodjupzm0Rqvgy0CeSzxGHaGEHYXVU,543
+RNApolis-0.2.0.dist-info/LICENSE,sha256=ZGRu12MzCgbYA-Lt8MyBlmjvPZh7xfiD5u5wBx0enq4,1066
+RNApolis-0.2.0.dist-info/METADATA,sha256=-ApE3-kETFzu512tLPAQa7MJjY7tuT-qyHLkPMBhI7A,1148
+RNApolis-0.2.0.dist-info/WHEEL,sha256=Xo9-1PvkuimrydujYJAjF7pCkriuXBpUPEjma1nZyJ0,92
+RNApolis-0.2.0.dist-info/entry_points.txt,sha256=113HwzaWSIoCvcPxkg_j2TIbnezP4_7akpTNr5n3Zjg,220
+RNApolis-0.2.0.dist-info/top_level.txt,sha256=LcO18koxZcWoJ21KDRRRo_tyIbmXL5z61dPitZpy8yc,9
+RNApolis-0.2.0.dist-info/RECORD,,

rnapolis/annotator.py

@@ -5,7 +5,7 @@ import logging
 import math
 import os
 from collections import Counter, defaultdict
-from typing import IO, Dict, List, Optional, Set, Tuple
+from typing import Dict, List, Optional, Set, Tuple
 
 import numpy
 import numpy.typing
@@ -15,6 +15,7 @@ from scipy.spatial import KDTree
 
 from rnapolis.common import (
     BR,
+    BaseInteractions,
     BasePair,
     BasePhosphate,
     BaseRibose,
@@ -471,19 +472,27 @@ def find_stackings(structure: Structure3D, model: int = 1) -> List[Stacking]:
     return stackings
 
 
-def extract_secondary_structure(
-    tertiary_structure: Structure3D, model: int = 1, find_gaps: bool = False
-) -> Structure2D:
+def extract_base_interactions(
+    tertiary_structure: Structure3D, model: int = 1
+) -> BaseInteractions:
     base_pairs, base_phosphate, base_ribose = find_pairs(tertiary_structure, model)
     stackings = find_stackings(tertiary_structure, model)
-    mapping = Mapping2D3D(tertiary_structure, base_pairs, stackings, find_gaps)
+    return BaseInteractions(base_pairs, stackings, base_ribose, base_phosphate, [])
+
+
+def extract_secondary_structure(
+    tertiary_structure: Structure3D, model: int = 1, find_gaps: bool = False
+) -> BaseInteractions:
+    base_interactions = extract_base_interactions(tertiary_structure, model)
+    mapping = Mapping2D3D(
+        tertiary_structure,
+        base_interactions.basePairs,
+        base_interactions.stackings,
+        find_gaps,
+    )
     stems, single_strands, hairpins, loops = mapping.bpseq.elements
     return Structure2D(
-        base_pairs,
-        stackings,
-        base_ribose,
-        base_phosphate,
-        [],
+        base_interactions,
         str(mapping.bpseq),
         mapping.dot_bracket,
         mapping.extended_dot_bracket,
@@ -494,12 +503,12 @@ def extract_secondary_structure(
     )
 
 
-def write_json(path: str, structure2d: Structure2D):
+def write_json(path: str, structure2d: BaseInteractions):
     with open(path, "wb") as f:
         f.write(orjson.dumps(structure2d))
 
 
-def write_csv(path: str, structure2d: Structure2D):
+def write_csv(path: str, structure2d: BaseInteractions):
     with open(path, "w") as f:
         writer = csv.writer(f)
         writer.writerow(["nt1", "nt2", "type", "classification-1", "classification-2"])

rnapolis/common.py

@@ -897,12 +897,17 @@ class DotBracket:
 
 
 @dataclass(frozen=True, order=True)
-class Structure2D:
+class BaseInteractions:
     basePairs: List[BasePair]
     stackings: List[Stacking]
     baseRiboseInteractions: List[BaseRibose]
     basePhosphateInteractions: List[BasePhosphate]
     otherInteractions: List[OtherInteraction]
+
+
+@dataclass(frozen=True, order=True)
+class Structure2D:
+    baseInteractions: BaseInteractions
     bpseq: str
     dotBracket: str
     extendedDotBracket: str

rnapolis/motif_extractor.py

@@ -1,14 +1,8 @@
 #! /usr/bin/env python
 import argparse
 import itertools
-from typing import IO, Dict, List
-
-import orjson
-from mmcif.io.IoAdapterPy import IoAdapterPy
-from mmcif.io.PdbxReader import DataContainer
 
 from rnapolis.common import BpSeq, DotBracket
-from rnapolis.util import handle_input_file
 
 
 def main():

rnapolis/tertiary.py

@@ -1,7 +1,5 @@
-import itertools
 import logging
 import math
-import string
 from collections import defaultdict
 from dataclasses import dataclass, field
 from functools import cached_property, total_ordering
@@ -20,7 +18,6 @@ from rnapolis.common import (
     ResidueAuth,
     ResidueLabel,
     Stacking,
-    Structure2D,
 )
 
 BASE_ATOMS = {

RNApolis-0.1.5.dist-info/RECORD

@@ -1,15 +0,0 @@
-rnapolis/annotator.py,sha256=_J-ydWfoGgkzwxRoQRGPTxs45rkd823y2zyEiBhYK7E,20921
-rnapolis/clashfinder.py,sha256=jD3s_UovygWi01NUbQNeAeRRFkARTSRraLXUV43UbAA,8514
-rnapolis/common.py,sha256=7oDhrlE-_hgzR6YOacL1rnanmMRg90idY-7BCBPZslQ,26339
-rnapolis/metareader.py,sha256=4qtMKRvww2sUStLeV8WVrLEt-ScydHUv4Gxx96tnf-M,1683
-rnapolis/motif_extractor.py,sha256=VJV1jUPschb_YZTwIatoqF3JQ_jc8uygQZGBao8k9wo,958
-rnapolis/parser.py,sha256=Z3Zd_IuRyOP45x5BStgu7UgoyHthhw55fT3udHUhAE4,11905
-rnapolis/tertiary.py,sha256=PuW2TZJuXzrDvYpR4fnFzsvyOvZlRv2b97N9sUPH1vQ,19005
-rnapolis/transformer.py,sha256=V9nOQvdq4-p7yUWo0vQg0CDQMpmyxz9t4TMSRVEKHnw,1817
-rnapolis/util.py,sha256=IdquFO3PV1_KDqodjupzm0Rqvgy0CeSzxGHaGEHYXVU,543
-RNApolis-0.1.5.dist-info/LICENSE,sha256=ZGRu12MzCgbYA-Lt8MyBlmjvPZh7xfiD5u5wBx0enq4,1066
-RNApolis-0.1.5.dist-info/METADATA,sha256=9u0fjRcuuEcM0dB65JdtKyC-9uRtnWw8WDbjOMamIJs,1124
-RNApolis-0.1.5.dist-info/WHEEL,sha256=Xo9-1PvkuimrydujYJAjF7pCkriuXBpUPEjma1nZyJ0,92
-RNApolis-0.1.5.dist-info/entry_points.txt,sha256=113HwzaWSIoCvcPxkg_j2TIbnezP4_7akpTNr5n3Zjg,220
-RNApolis-0.1.5.dist-info/top_level.txt,sha256=LcO18koxZcWoJ21KDRRRo_tyIbmXL5z61dPitZpy8yc,9
-RNApolis-0.1.5.dist-info/RECORD,,