RNApolis 0.0.16__py3-none-any.whl → 0.1.1__py3-none-any.whl

{RNApolis-0.0.16.dist-info → RNApolis-0.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: RNApolis
-Version: 0.0.16
+Version: 0.1.1
 Summary: A Python library containing RNA-related bioinformatics functions and classes
 Home-page: https://github.com/tzok/rnapolis-py
 Author: Tomasz Zok

RNApolis-0.1.1.dist-info/RECORD

@@ -0,0 +1,15 @@
+rnapolis/annotator.py,sha256=CLdwuq0Y42unPd3W-NKIqy4JMCSanzUj6GmP9VnfRb4,20766
+rnapolis/clashfinder.py,sha256=jD3s_UovygWi01NUbQNeAeRRFkARTSRraLXUV43UbAA,8514
+rnapolis/common.py,sha256=o6I51SkBuHxiP9G91Uy42h8_JpCVc2Sd8lGLviCTKAc,24518
+rnapolis/metareader.py,sha256=4qtMKRvww2sUStLeV8WVrLEt-ScydHUv4Gxx96tnf-M,1683
+rnapolis/motif_extractor.py,sha256=5gUSNGeT_jBVzpj3gCaZJaRA4XPUpxJ9NteWns6UMAY,961
+rnapolis/parser.py,sha256=Z3Zd_IuRyOP45x5BStgu7UgoyHthhw55fT3udHUhAE4,11905
+rnapolis/tertiary.py,sha256=fA7ml3zlcWz0xsq4z73m-HGQh3KAkwzv7vVM5ivfgkg,18293
+rnapolis/transformer.py,sha256=V9nOQvdq4-p7yUWo0vQg0CDQMpmyxz9t4TMSRVEKHnw,1817
+rnapolis/util.py,sha256=IdquFO3PV1_KDqodjupzm0Rqvgy0CeSzxGHaGEHYXVU,543
+RNApolis-0.1.1.dist-info/LICENSE,sha256=ZGRu12MzCgbYA-Lt8MyBlmjvPZh7xfiD5u5wBx0enq4,1066
+RNApolis-0.1.1.dist-info/METADATA,sha256=0MspfjX3C2UlAe1c1eHVncrtXagp_DlMqA09ABKiS1E,1124
+RNApolis-0.1.1.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
+RNApolis-0.1.1.dist-info/entry_points.txt,sha256=113HwzaWSIoCvcPxkg_j2TIbnezP4_7akpTNr5n3Zjg,220
+RNApolis-0.1.1.dist-info/top_level.txt,sha256=LcO18koxZcWoJ21KDRRRo_tyIbmXL5z61dPitZpy8yc,9
+RNApolis-0.1.1.dist-info/RECORD,,

rnapolis/annotator.py

@@ -19,6 +19,7 @@ from rnapolis.common import (
     BasePhosphate,
     BaseRibose,
     BPh,
+    BpSeq,
     LeontisWesthof,
     Residue,
     Saenger,
@@ -471,11 +472,26 @@ def find_stackings(structure: Structure3D, model: int = 1) -> List[Stacking]:
 
 
 def extract_secondary_structure(
-    tertiary_structure: Structure3D, model: int = 1
+    tertiary_structure: Structure3D, model: int = 1, find_gaps: bool = False
 ) -> Structure2D:
     base_pairs, base_phosphate, base_ribose = find_pairs(tertiary_structure, model)
     stackings = find_stackings(tertiary_structure, model)
-    return Structure2D(base_pairs, stackings, base_ribose, base_phosphate, [])
+    mapping = Mapping2D3D(tertiary_structure, base_pairs, stackings, find_gaps)
+    stems, single_strands, hairpins, loops = mapping.bpseq.elements
+    return Structure2D(
+        base_pairs,
+        stackings,
+        base_ribose,
+        base_phosphate,
+        [],
+        str(mapping.bpseq),
+        mapping.dot_bracket,
+        mapping.extended_dot_bracket,
+        stems,
+        single_strands,
+        hairpins,
+        loops,
+    )
 
 
 def write_json(path: str, structure2d: Structure2D):
@@ -541,9 +557,9 @@ def write_csv(path: str, structure2d: Structure2D):
             )
 
 
-def write_bpseq(path: str, mapping: Mapping2D3D):
+def write_bpseq(path: str, bpseq: BpSeq):
     with open(path, "w") as f:
-        f.write(str(mapping.bpseq))
+        f.write(str(bpseq))
 
 
 def main():
@@ -563,14 +579,14 @@ def main():
     parser.add_argument(
         "--find-gaps",
         action="store_true",
-        help=f"(optional) if set, the program will detect gaps and break the PDB chain into two or more strands; "
-        "the gap is defined as O3'-P distance greater then {1.5 * AVERAGE_OXYGEN_PHOSPHORUS_DISTANCE_COVALENT}",
+        help="(optional) if set, the program will detect gaps and break the PDB chain into two or more strands; "
+        f"the gap is defined as O3'-P distance greater then {1.5 * AVERAGE_OXYGEN_PHOSPHORUS_DISTANCE_COVALENT}",
     )
     args = parser.parse_args()
 
     file = handle_input_file(args.input)
     structure3d = read_3d_structure(file, 1)
-    structure2d = extract_secondary_structure(structure3d)
+    structure2d = extract_secondary_structure(structure3d, 1, args.find_gaps)
 
     if args.csv:
         write_csv(args.csv, structure2d)
@@ -578,15 +594,13 @@ def main():
     if args.json:
         write_json(args.json, structure2d)
 
-    mapping = Mapping2D3D(structure3d, structure2d, args.find_gaps)
-
     if args.bpseq:
-        write_bpseq(args.bpseq, mapping)
+        write_bpseq(args.bpseq, structure2d.bpseq)
 
     if args.extended:
-        print(mapping.extended_dot_bracket)
+        print(structure2d.extendedDotBracket)
     else:
-        print(mapping.dot_bracket)
+        print(structure2d.dotBracket)
 
 
 if __name__ == "__main__":

rnapolis/common.py

@@ -321,15 +321,6 @@ class OtherInteraction(Interaction):
     pass
 
 
-@dataclass(frozen=True, order=True)
-class Structure2D:
-    basePairs: List[BasePair]
-    stackings: List[Stacking]
-    baseRiboseInteractions: List[BaseRibose]
-    basePhosphateInteractions: List[BasePhosphate]
-    otherInteractions: List[OtherInteraction]
-
-
 @dataclass
 class Entry(Sequence):
     index_: int
@@ -380,12 +371,15 @@ class Strand:
         return f"{self.first}-{self.sequence}-{self.last}"
 
 
-@dataclass(frozen=True)
+@dataclass
 class SingleStrand:
     strand: Strand
     is5p: bool
     is3p: bool
 
+    def __post_init__(self):
+        self.description = str(self)
+
     def __str__(self):
         if self.is5p:
             return f"SingleStrand5p {self.strand.first} {self.strand.last} {self.strand.sequence} {self.strand.structure}"
@@ -394,7 +388,7 @@ class SingleStrand:
         return f"SingleStrand {self.strand.first} {self.strand.last} {self.strand.sequence} {self.strand.structure}"
 
 
-@dataclass(frozen=True)
+@dataclass
 class Stem:
     strand5p: Strand
     strand3p: Strand
@@ -410,22 +404,31 @@ class Stem:
             Strand.from_bpseq_entries(strand3p_entries, dotbracket),
         )
 
+    def __post_init__(self):
+        self.description = str(self)
+
     def __str__(self):
         return f"Stem {self.strand5p.first} {self.strand5p.last} {self.strand5p.sequence} {self.strand5p.structure} {self.strand3p.first} {self.strand3p.last} {self.strand3p.sequence} {self.strand3p.structure}"
 
 
-@dataclass(frozen=True)
+@dataclass
 class Hairpin:
     strand: Strand
 
+    def __post_init__(self):
+        self.description = str(self)
+
     def __str__(self):
         return f"Hairpin {self.strand.first} {self.strand.last} {self.strand.sequence} {self.strand.structure}"
 
 
-@dataclass(frozen=True)
+@dataclass
 class Loop:
     strands: List[Strand]
 
+    def __post_init__(self):
+        self.description = str(self)
+
     def __str__(self):
         desc = " ".join(
             [
@@ -516,30 +519,32 @@ class BpSeq:
         return stems
 
     @cached_property
-    def elements(self):
+    def elements(
+        self,
+    ) -> Tuple[List[Stem], List[SingleStrand], List[Hairpin], List[Loop]]:
         if not self.__stems_entries:
-            return []
+            return [], [], [], []
 
-        elements = []
-        stops = set()
+        stems, single_strands, hairpins, loops = [], [], [], []
+        stopset = set()
 
         # stems
         for stem_entries in self.__stems_entries:
             stem = Stem.from_bpseq_entries(
                 stem_entries, self.entries, self.to_dot_bracket.structure
             )
-            elements.append(stem)
-            stops.add(stem.strand5p.first - 1)
-            stops.add(stem.strand5p.last - 1)
-            stops.add(stem.strand3p.first - 1)
-            stops.add(stem.strand3p.last - 1)
+            stems.append(stem)
+            stopset.add(stem.strand5p.first - 1)
+            stopset.add(stem.strand5p.last - 1)
+            stopset.add(stem.strand3p.first - 1)
+            stopset.add(stem.strand3p.last - 1)
 
-        stops = sorted(stops)
+        stops = sorted(stopset)
         loop_candidates = []
 
         # 5' single strand
         if stops[0] > 0:
-            elements.append(
+            single_strands.append(
                 SingleStrand(
                     Strand.from_bpseq_entries(
                         self.entries[: stops[0] + 1],
@@ -555,7 +560,7 @@ class BpSeq:
             candidate = self.entries[stops[i - 1] : stops[i] + 1]
             if all([entry.pair == 0 for entry in candidate[1:-1]]):
                 if candidate[0].pair == candidate[-1].index_:
-                    elements.append(
+                    hairpins.append(
                         Hairpin(
                             Strand.from_bpseq_entries(
                                 candidate, self.to_dot_bracket.structure
@@ -571,7 +576,7 @@ class BpSeq:
 
         # 3' single strand
         if stops[-1] < len(self.entries) - 1:
-            elements.append(
+            single_strands.append(
                 SingleStrand(
                     Strand.from_bpseq_entries(
                         self.entries[stops[-1] :], self.to_dot_bracket.structure
@@ -613,13 +618,13 @@ class BpSeq:
 
             if self.entries[loop[0].first - 1].pair == loop[-1].last:
                 if not all([strand.last - strand.first <= 1 for strand in loop]):
-                    elements.append(Loop(loop))
+                    loops.append(Loop(loop))
 
         for i in range(len(loop_candidates)):
             if i not in used:
-                elements.append(SingleStrand(loop_candidates[i]))
+                single_strands.append(SingleStrand(loop_candidates[i], False, False))
 
-        return elements
+        return stems, single_strands, hairpins, loops
 
     @cached_property
     def __regions(self) -> List[Tuple[int, int, int]]:
@@ -675,14 +680,14 @@ class BpSeq:
                 vars_by_region[i].append(variable)
                 vars_by_order[j].append(variable)
                 var_by_region_order[(i, j)] = variable
-                region_by_var[variable] = i
+                region_by_var[variable] = regions[i]
 
         # define objective function terms
         terms = []
 
         for order, vars in vars_by_order.items():
             for var in vars:
-                length = len(regions[region_by_var[var]])
+                length = region_by_var[var][2]
                 if order == 0:
                     terms.append(var * length)
                 else:
@@ -707,6 +712,7 @@ class BpSeq:
 
         # solve the problem
         try:
+            logging.debug(f"POA: problem formulation\n{problem}")
             problem.solve(solver)
         except pulp.PulpSolverError:
             logging.warning(
@@ -720,7 +726,7 @@ class BpSeq:
             return self.fcfs()
 
         # log solver time statistics
-        logging.info(
+        logging.debug(
             f"POA: solver {solver.name} took {round(problem.solutionTime, 2)} seconds"
         )
 
@@ -833,3 +839,19 @@ class DotBracket:
 
     def __str__(self):
         return f"{self.sequence}\n{self.structure}"
+
+
+@dataclass(frozen=True, order=True)
+class Structure2D:
+    basePairs: List[BasePair]
+    stackings: List[Stacking]
+    baseRiboseInteractions: List[BaseRibose]
+    basePhosphateInteractions: List[BasePhosphate]
+    otherInteractions: List[OtherInteraction]
+    bpseq: str
+    dotBracket: str
+    extendedDotBracket: str
+    stems: List[Stem]
+    singleStrands: List[SingleStrand]
+    hairpins: List[Hairpin]
+    loops: List[Loop]

rnapolis/motif_extractor.py

@@ -1,5 +1,6 @@
 #! /usr/bin/env python
 import argparse
+import itertools
 from typing import IO, Dict, List
 
 import orjson
@@ -24,7 +25,10 @@ def main():
         parser.print_help()
         return
 
-    for element in bpseq.elements:
+    print(f"Full dot-bracket:\n{bpseq.to_dot_bracket}")
+    stems, single_strands, hairpins, loops = bpseq.elements
+
+    for element in itertools.chain(stems, single_strands, hairpins, loops):
         print(element)
 
 

rnapolis/tertiary.py

@@ -388,14 +388,15 @@ class Structure3D:
 @dataclass
 class Mapping2D3D:
     structure3d: Structure3D
-    structure2d: Structure2D
+    base_pairs2d: List[BasePair]
+    stackings2d: List[Stacking]
     find_gaps: bool
 
     @cached_property
     def base_pairs(self) -> List[BasePair3D]:
         result = []
         used = set()
-        for base_pair in self.structure2d.basePairs:
+        for base_pair in self.base_pairs2d:
             nt1 = self.structure3d.find_residue(base_pair.nt1.label, base_pair.nt1.auth)
             nt2 = self.structure3d.find_residue(base_pair.nt2.label, base_pair.nt2.auth)
             if nt1 is not None and nt2 is not None:
@@ -439,7 +440,7 @@ class Mapping2D3D:
     def stackings(self) -> List[Stacking3D]:
         result = []
         used = set()
-        for stacking in self.structure2d.stackings:
+        for stacking in self.stackings2d:
             nt1 = self.structure3d.find_residue(stacking.nt1.label, stacking.nt1.auth)
             nt2 = self.structure3d.find_residue(stacking.nt2.label, stacking.nt2.auth)
             if nt1 is not None and nt2 is not None:

RNApolis-0.0.16.dist-info/RECORD

@@ -1,15 +0,0 @@
-rnapolis/annotator.py,sha256=aaPb1gMbzVv8Jx1iSJ12nW9R-aWE-lq95gZks9RZDfk,20419
-rnapolis/clashfinder.py,sha256=jD3s_UovygWi01NUbQNeAeRRFkARTSRraLXUV43UbAA,8514
-rnapolis/common.py,sha256=_ylHz3KdB75cugX_7wTzqbPQChAUKlnuThsvMLuxsfM,23859
-rnapolis/metareader.py,sha256=4qtMKRvww2sUStLeV8WVrLEt-ScydHUv4Gxx96tnf-M,1683
-rnapolis/motif_extractor.py,sha256=nz3aenyhPBZNuWzdHOCpFG95SLqf5yLhQQi1Bx-i2H8,786
-rnapolis/parser.py,sha256=Z3Zd_IuRyOP45x5BStgu7UgoyHthhw55fT3udHUhAE4,11905
-rnapolis/tertiary.py,sha256=DcY4OPRg9d05L9Fr9G_1KTbMwkNVTyEPnLkJOPODlAM,18276
-rnapolis/transformer.py,sha256=V9nOQvdq4-p7yUWo0vQg0CDQMpmyxz9t4TMSRVEKHnw,1817
-rnapolis/util.py,sha256=IdquFO3PV1_KDqodjupzm0Rqvgy0CeSzxGHaGEHYXVU,543
-RNApolis-0.0.16.dist-info/LICENSE,sha256=ZGRu12MzCgbYA-Lt8MyBlmjvPZh7xfiD5u5wBx0enq4,1066
-RNApolis-0.0.16.dist-info/METADATA,sha256=ODBaX0GLTPv7WaCiFdJErR9dCdeOV50N7sjkf1U1sFs,1125
-RNApolis-0.0.16.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
-RNApolis-0.0.16.dist-info/entry_points.txt,sha256=113HwzaWSIoCvcPxkg_j2TIbnezP4_7akpTNr5n3Zjg,220
-RNApolis-0.0.16.dist-info/top_level.txt,sha256=LcO18koxZcWoJ21KDRRRo_tyIbmXL5z61dPitZpy8yc,9
-RNApolis-0.0.16.dist-info/RECORD,,