Skip to content
Snippets Groups Projects
Commit ac492a32 authored by Rafal Gumienny's avatar Rafal Gumienny
Browse files

fix: SCHWED-3120 Renumber entities

parent 66e60833
No related branches found
No related tags found
No related merge requests found
...@@ -18,6 +18,7 @@ from ost.mol.alg import (qsscoring, Molck, MolckSettings, CleanlDDTReferences, ...@@ -18,6 +18,7 @@ from ost.mol.alg import (qsscoring, Molck, MolckSettings, CleanlDDTReferences,
PreparelDDTGlobalRDMap, lDDTSettings, CheckStructure, PreparelDDTGlobalRDMap, lDDTSettings, CheckStructure,
LocalDistDiffTest) LocalDistDiffTest)
from ost.conop import CompoundLib from ost.conop import CompoundLib
from ost.seq.alg.renumber import Renumber
class _DefaultStereochemicalParamAction(argparse.Action): class _DefaultStereochemicalParamAction(argparse.Action):
...@@ -426,6 +427,7 @@ def _Main(): ...@@ -426,6 +427,7 @@ def _Main():
reference.GetName())) reference.GetName()))
qs_scorer = qsscoring.QSscorer(reference, model) qs_scorer = qsscoring.QSscorer(reference, model)
if opts.qs_score: if opts.qs_score:
ost.LogInfo("Computing QS-score")
try: try:
if opts.chain_mapping is not None: if opts.chain_mapping is not None:
ost.LogInfo( ost.LogInfo(
...@@ -457,6 +459,7 @@ def _Main(): ...@@ -457,6 +459,7 @@ def _Main():
}) })
# Calculate lDDT # Calculate lDDT
if opts.lddt: if opts.lddt:
ost.LogInfo("Computing lDDT")
lddt_settings = lDDTSettings( lddt_settings = lDDTSettings(
bond_tolerance=opts.bond_tolerance, bond_tolerance=opts.bond_tolerance,
angle_tolerance=opts.angle_tolerance, angle_tolerance=opts.angle_tolerance,
...@@ -470,9 +473,16 @@ def _Main(): ...@@ -470,9 +473,16 @@ def _Main():
if opts.verbosity > 3: if opts.verbosity > 3:
lddt_settings.PrintParameters() lddt_settings.PrintParameters()
# Perform scoring # Perform scoring
for mdl_c, ref_c in qs_scorer.chain_mapping.iteritems(): # Get chains from mapped alignments
model_view = model.Select("cname=%s" % mdl_c) for aln in qs_scorer.alignments:
reference_view = reference.Select("cname=%s" % ref_c) # Get chains and renumber according to alignment (for lDDT)
ch_ref = aln.GetSequence(0).GetName()
reference_view = Renumber(
aln.GetSequence(0)).CreateFullView()
ch_mdl = aln.GetSequence(1).GetName()
model_view = Renumber(aln.GetSequence(1)).CreateFullView()
ost.LogInfo(("Computing lDDT between model chain %s and "
"reference chain %s") % (ch_mdl, ch_ref))
_ComputelDDT(model_view, [reference_view], lddt_settings) _ComputelDDT(model_view, [reference_view], lddt_settings)
if opts.output is not None: if opts.output is not None:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment