diff --git a/src/structure.py b/src/structure.py index 7cf5a7c..45f11ef 100644 --- a/src/structure.py +++ b/src/structure.py @@ -1,13 +1,14 @@ class Turn: - def __init__(self, turn_type): + def __init__(self, turn_type, res_num): self.turn_type = turn_type class Bridge: - def __init__(self, bridge_type): + def __init__(self, bridge_type, res_num): self.bridge_type = bridge_type + self.res_num = res_num class Helix: @@ -20,11 +21,13 @@ def get_turns(residues): for j in range(3,6): if(i+j0): + return(bridges) + else: + return(False) -def get_helix(turns): - pass +def get_helix(residues, turns): + for i in range(residues): + for j in range(i+1,i+5): + if