diff --git a/modules/io/pymod/__init__.py b/modules/io/pymod/__init__.py index 30ea9e3a6066ae64678200582dac595196ac90a4..e5f0a7834083ef5e2ea387d34e007598a2053664 100644 --- a/modules/io/pymod/__init__.py +++ b/modules/io/pymod/__init__.py @@ -204,7 +204,7 @@ def SavePDB(models, filename, dialect=None, pqr=False, profile='DEFAULT'): writer=PDBWriter(filename, profile) writer.SetIsPQR(pqr) if len(models)>1: - writer.SetMultiModel(True) + writer.write_multi_model=True for model in models: writer.Write(model) diff --git a/modules/io/pymod/export_pdb_io.cc b/modules/io/pymod/export_pdb_io.cc index c146557f914008b64d6052ef16238e53bb014795..c057534e0a9e61b030ef89181d5dca5fac80e707 100644 --- a/modules/io/pymod/export_pdb_io.cc +++ b/modules/io/pymod/export_pdb_io.cc @@ -72,6 +72,10 @@ void export_pdb_io() .def("Write", write_a) .def("SetIsPQR", &PDBWriter::SetIsPQR) .def("IsPQR", &PDBWriter::IsPQR) + .def("SetWriteMultiModel", &PDBWriter::SetWriteMultiModel) + .def("GetWriteMultiModel", &PDBWriter::GetWriteMultiModel) + .add_property("write_multi_model", &PDBWriter::GetWriteMultiModel, + &PDBWriter::SetWriteMultiModel) .def("Write", write_b) ; }