python-mander/src/pywrapper.cpp at master · gerrymandr/python-mander · GitHub

History

151 lines (132 loc) · 5.29 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

#include <compactnesslib/compactnesslib.hpp>

#include <pybind11.h>

#include <stl.h>

#include <vector>

namespace py = pybind11;

//Tutorials

//http://www.benjack.io/2017/06/12/python-cpp-tests.html

//https://pybind11.readthedocs.io/en/stable/classes.html

//http://people.duke.edu/~ccc14/sta-663-2016/18G_C++_Python_pybind11.html

std::string getUnboundedScoresForGeoJSON(const std::string &geojson, const std::string id, const std::vector<std::string> &score_list){

auto gc = complib::ReadGeoJSON(geojson);

complib::CalculateListOfUnboundedScores(gc, score_list);

return complib::OutScoreJSON(gc, id);

}

void augmentShapefileWithUnboundedScores(const std::string filename, const std::vector<std::string> &score_list){

auto gc = complib::ReadShapefile(filename);

complib::CalculateListOfUnboundedScores(gc, score_list);

complib::WriteShapeScores(gc, filename);

}

void addUnboundedScoresToNewShapefile(const std::string in_filename, const std::string out_filename, const std::vector<std::string> &score_list){

auto gc = complib::ReadShapefile(in_filename);

complib::CalculateListOfUnboundedScores(gc, score_list);

complib::WriteShapefile(gc,out_filename);

}

std::string getBoundedScoresForGeoJSON(

const std::string &gj_subunit,

const std::string &gj_superunit,

const std::string id,

const std::string join_on,

const std::vector<std::string> &score_list

){

auto gc_sub = complib::ReadGeoJSON(gj_subunit);

auto gc_sup = complib::ReadGeoJSON(gj_superunit);

complib::CalculateListOfBoundedScores(gc_sub, gc_sup, join_on, score_list);

complib::CalculateListOfUnboundedScores(gc_sub, score_list);

return complib::OutScoreJSON(gc_sub, id);

}

void augmentShapefileWithBoundedScores(

const std::string subunit_filename,

const std::string superunit_filename,

const std::string join_on,

const std::vector<std::string> &score_list

){

auto gc_sub = complib::ReadShapefile(subunit_filename);

auto gc_sup = complib::ReadShapefile(superunit_filename);

complib::CalculateListOfBoundedScores(gc_sub, gc_sup, join_on, score_list);

complib::CalculateListOfUnboundedScores(gc_sub, score_list);

complib::WriteShapeScores(gc_sub, subunit_filename);

}

void addBoundedScoresToNewShapefile(

const std::string in_subunit_filename,

const std::string in_superunit_filename,

const std::string out_filename,

const std::string join_on,

const std::vector<std::string> &score_list

){

auto gc_sub = complib::ReadShapefile(in_subunit_filename);

auto gc_sup = complib::ReadShapefile(in_superunit_filename);

complib::CalculateListOfBoundedScores(gc_sub, gc_sup, join_on, score_list);

complib::CalculateListOfUnboundedScores(gc_sub, score_list);

complib::WriteShapefile(gc_sub,out_filename);

}

PYBIND11_MODULE(_mander, m) {

m.doc() = "Internal library used by python-mander to calculate scores";

m.def(

"prepGeoJSON",

&complib::PrepGeoJSON,

"Load GeoJSON into memory for quick retrieval later."

);

m.def(

"getListOfUnboundedScores",

&complib::getListOfUnboundedScores,

"Returns a list of unbounded compactness scores."

);

m.def(

"getListOfBoundedScores",

&complib::getListOfBoundedScores,

"Returns a list of bounded compactness scores."

);

m.def(

"getUnboundedScoresForGeoJSON",

&getUnboundedScoresForGeoJSON,

"Takes a GeoJSON string as input, calculates all scores, return a JSON dictionary of the scores keyed to values identified 'id' which are expected to be properties of the GeoJSON objects. If id='', then the object's 0-indexed order is used.",

py::arg("geojson"),

py::arg("id")="",

py::arg("score_list")=""

);

m.def(

"getBoundedScoresForGeoJSON",

&getBoundedScoresForGeoJSON,

py::arg("gj_subunit"),

py::arg("gj_superunit"),

py::arg("join_on")="",

py::arg("join_id")="",

py::arg("score_list")=""

);

m.def(

"augmentShapefileWithUnboundedScores",

&augmentShapefileWithUnboundedScores,

"Given the filename of a shapefile containing subunits, and another of superunits, this function calculates the scores of its shapes and adds them to the file.",

py::arg("filename"),

py::arg("score_list")=""

);

m.def(

"addUnboundedScoresToNewShapefile",

&addUnboundedScoresToNewShapefile,

"Reads a shapefile, calculates its scores, and outputs a new shapefile.",

py::arg("in_filename")="",

py::arg("out_filename")="",

py::arg("score_list")=""

);

m.def(

"augmentShapefileWithBoundedScores",

&augmentShapefileWithBoundedScores,

"Given the filename of a shapefile containing subunits, and another of superunits, this function calculates the scores of its shapes and adds them to the file.",

py::arg("subunit_filename"),

py::arg("superunit_filename"),

py::arg("join_id")="",

py::arg("score_list")=""

);

m.def(

"addBoundedScoresToNewShapefile",

&addBoundedScoresToNewShapefile,

"Reads a shapefile, calculates its scores, and outputs a new shapefile.",

py::arg("in_subunit_filename")="",

py::arg("in_superunit_filename")="",

py::arg("out_filename")="",

py::arg("join_on")="",

py::arg("score_list")=""

);

}

python-mander/src/pywrapper.cpp at master · gerrymandr/python-mander · GitHub

Expand file tree

Breadcrumbs

pywrapper.cpp

Latest commit

History

Breadcrumbs

pywrapper.cpp

File metadata and controls