projet_court/src/structure.py

import math
import numpy as np
from geom import *

class Structure:
    def __init__(self, res):
        self.res = res

class Turn(Structure):

    def __init__(self, turn_type, res):
        self.turn_type = turn_type
        Structure.res = res
                        
class Bridge(Structure):

    def __init__(self, bridge_type, res_num, res_partner, indices):
        self.bridge_type = bridge_type
        self.res_num = res_num
        self.res_partner = res_partner
        Structure.res = res_num
        self.i = indices[0]
        self.j = indices[1]

class Helix(Structure):

    def __init__(self, residues, res_num, helix_type):
        self.residues = residues
        self.res_num = res_num
        Structure.res = res_num
        self.helix_type = helix_type