1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
// Copyright 2018 Global Phasing Ltd.
#include "gemmi/ccp4.hpp"
#include "common.h"
#include <pybind11/stl.h>
#define GEMMI_READ_MAP_IMPLEMENTATION
#include "gemmi/read_map.hpp" // defines read_ccp4_map, read_ccp4_mask
namespace py = pybind11;
using namespace gemmi;
template<typename T>
py::class_<T> add_ccp4_common(py::module& m, const char* name) {
using Map = Ccp4<T>;
return py::class_<Map, Ccp4Base>(m, name)
.def(py::init<>())
.def_readwrite("grid", &Map::grid)
.def("setup", &Map::setup,
py::arg("default_value"), py::arg("mode")=MapSetup::Full)
.def("update_ccp4_header", &Map::update_ccp4_header,
py::arg("mode")=-1, py::arg("update_stats")=true)
.def("full_cell", &Map::full_cell)
.def("write_ccp4_map", &Map::write_ccp4_map, py::arg("filename"))
.def("set_extent", &Map::set_extent)
.def("__repr__", [=](const Map& self) {
const SpaceGroup* sg = self.grid.spacegroup;
return cat("<gemmi.", name, " with grid ",
self.grid.nu, 'x', self.grid.nv, 'x', self.grid.nw,
" in SG #", sg ? std::to_string(sg->ccp4) : "?", '>');
});
}
void add_ccp4(py::module& m) {
py::enum_<MapSetup>(m, "MapSetup")
.value("Full", MapSetup::Full)
.value("NoSymmetry", MapSetup::NoSymmetry)
.value("ReorderOnly", MapSetup::ReorderOnly);
py::class_<Ccp4Base>(m, "Ccp4Base")
.def("header_i32", &Ccp4Base::header_i32)
.def("header_float", &Ccp4Base::header_float)
.def("header_str", &Ccp4Base::header_str)
.def("set_header_i32", &Ccp4Base::set_header_i32)
.def("set_header_float", &Ccp4Base::set_header_float)
.def("set_header_str", &Ccp4Base::set_header_str)
.def("axis_positions", &Ccp4Base::axis_positions)
.def("get_extent", &Ccp4Base::get_extent)
.def("has_skew_transformation", &Ccp4Base::has_skew_transformation)
.def("get_skew_transformation", &Ccp4Base::get_skew_transformation)
;
add_ccp4_common<float>(m, "Ccp4Map");
add_ccp4_common<int8_t>(m, "Ccp4Mask");
m.def("read_ccp4_map", &read_ccp4_map,
py::arg("path"), py::arg("setup")=false, py::return_value_policy::move,
"Reads a CCP4 file, mode 2 (floating-point data).");
m.def("read_ccp4_mask", &read_ccp4_mask,
py::arg("path"), py::arg("setup")=false, py::return_value_policy::move,
"Reads a CCP4 file, mode 0 (int8_t data, usually 0/1 masks).");
}
|
