示例#1
0
    def get_model(self, filename=None):
        """
    Retrieve a stored mmtbx.model.manager object

    Parameters
    ----------
    filename : str
        Optionally specify which model using its filepath

    Returns
    -------
    model
        The mmtbx.model.manager object

    """
        model = self._get(ModelDataManager.datatype, filename)
        if (self.supports('restraint')):
            restraint_objects = list()
            for filename in self.get_restraint_names():
                restraint_objects.append(
                    (filename, self.get_restraint(filename)))
            model.set_restraint_objects(restraint_objects)
        if hasattr(model, 'info'):  # save filename if possible
            if filename is None:
                filename = self.get_default_model_name()
            if filename:
                model.info().full_file_name = os.path.abspath(filename)
                model.info().file_name = os.path.split(filename)[-1]

        return model
示例#2
0
 def get_model(self, filename=None):
   model = self._get(ModelDataManager.datatype, filename)
   if (self.supports('restraint')):
     restraint_objects = list()
     for filename in self.get_restraint_names():
       restraint_objects.append((filename, self.get_restraint(filename)))
     model.set_restraint_objects(restraint_objects)
   return model