from ample.parsers import _contactfile_parser

class PconscContactParser(_contactfile_parser.ContactfileParser):
    """ Parser for PconsC1/2/3 contact prediction files """

    _RES1 = 0
    _RES2 = 1
    _RAW_SCORE = 2

    _METHOD = "pconsc"

    def __init__(self):
        _contactfile_parser.ContactfileParser.__init__(self)

    def read(self, contactfile):
        with open(contactfile, "r") as fh:
            for line in iter(fh.readline, ''):
                line = line.strip().split()
                
                # Define the contact in a dictionary - use parent method
                contact = self.define_contact(line,
                                              res1_idx=self._RES1,
                                              res2_idx=self._RES2,
                                              raw_score_idx=self._RAW_SCORE,
                                              method=self._METHOD,
                                              file=contactfile)

                self.contacts.append(contact)
        return