Skip to content

Commit fa6087f

Browse files
committed
Add python access to MolZAtom
1 parent de850d3 commit fa6087f

File tree

1 file changed

+19
-0
lines changed

1 file changed

+19
-0
lines changed

src/extensions/molecule_ext.cpp

+19
Original file line numberDiff line numberDiff line change
@@ -637,6 +637,12 @@ double _getQ3(Molecule& m)
637637
return m.GetPar(quatparname(m, 3)).GetValue();
638638
}
639639

640+
const ScatteringPower* getscatteringpowerpointer(const MolZAtom& a)
641+
{
642+
const ScatteringPower* rv =
643+
a.mpPow==0 ? NULL : a.mpPow;
644+
return rv;
645+
}
640646

641647
} // namespace
642648

@@ -841,4 +847,17 @@ void wrap_molecule()
841847
(Molecule* (*)(ZScatterer*)) &ZScatterer2Molecule, bp::arg("zscatt"),
842848
return_value_policy<manage_new_object>());
843849

850+
// Wrap
851+
class_<MolZAtom> ("MolZAtom", init<const MolZAtom&>())
852+
.def("GetScatteringPower", getscatteringpowerpointer,
853+
return_internal_reference<>())
854+
//return_value_policy<copy_const_reference>())
855+
.add_property("bond_atom", &MolZAtom::mBondAtom)
856+
.add_property("bond_angle_atom", &MolZAtom::mBondAngleAtom)
857+
.add_property("dihdral_angle_atom", &MolZAtom::mDihedralAtom)
858+
.add_property("bond_length", &MolZAtom::mBondLength)
859+
.add_property("bond_angle", &MolZAtom::mBondAngle)
860+
.add_property("dihdral_angle", &MolZAtom::mDihedralAngle)
861+
;
862+
844863
}

0 commit comments

Comments
 (0)