| 1273 | } |
| 1274 | |
| 1275 | void addUtilsTypes(py::Module& module) |
| 1276 | { |
| 1277 | module.addType(py::define<VocabObject>("_VocabDict", "_UtilsVocabDict", Py_TPFLAGS_BASETYPE) |
| 1278 | .sqLen<&VocabObject::len>() |
| 1279 | .sqGetItem<&VocabObject::getitem>() |
| 1280 | .method<&VocabObject::getstate>("__getstate__") |
| 1281 | .method<&VocabObject::setstate>("__setstate__") |
| 1282 | ); |
| 1283 | |
| 1284 | module.addType(py::define<CorpusObject>("tomotopy._UtilsCorpus", "_UtilsCorpus", Py_TPFLAGS_BASETYPE) |
| 1285 | .mpLen<&CorpusObject::len>() |
| 1286 | .mpGetItem<&CorpusObject::getitem>() |
| 1287 | .method<&CorpusObject::getstate>("__getstate__") |
| 1288 | .method<&CorpusObject::setstate>("__setstate__") |
| 1289 | .method<&CorpusObject::addDoc>("add_doc") |
| 1290 | .method<&CorpusObject::addDocs>("add_docs") |
| 1291 | .method<&CorpusObject::extractNgrams>("extract_ngrams") |
| 1292 | .method<&CorpusObject::concatNgrams>("concat_ngrams")); |
| 1293 | |
| 1294 | module.addType(py::define<CorpusIterObject>("tomotopy._UtilsCorpusIter", "_UtilsCorpusIter", Py_TPFLAGS_DEFAULT) |
| 1295 | .iter<&CorpusIterObject::iter>() |
| 1296 | .iternext<&CorpusIterObject::iternext>()); |
| 1297 | |
| 1298 | module.addType(py::define<DocumentObject>("tomotopy._Document", "_Document", Py_TPFLAGS_BASETYPE) |
| 1299 | .sqLen<&DocumentObject::len>() |
| 1300 | .sqGetItem<&DocumentObject::getitem>() |
| 1301 | .getAttrO<&DocumentObject::getattro>() |
| 1302 | .repr<&DocumentObject::repr>() |
| 1303 | .method<&DocumentObject::getTopics>("get_topics") |
| 1304 | .method<&DocumentObject::getTopicDist>("get_topic_dist") |
| 1305 | .method<&DocumentObject::getSubTopics>("get_sub_topics") |
| 1306 | .method<&DocumentObject::getSubTopicDist>("get_sub_topic_dist") |
| 1307 | .method<&DocumentObject::getWords>("get_words") |
| 1308 | .method<&DocumentObject::getCountVector>("get_count_vector") |
| 1309 | .method<&DocumentObject::getLL>("get_ll") |
| 1310 | .property<&DocumentObject::getAllWords>("_words") |
| 1311 | .property<&DocumentObject::getWeight>("_weight") |
| 1312 | .property<&DocumentObject::getZ>("_topics") |
| 1313 | .property<&DocumentObject::getUid>("_uid") |
| 1314 | .property<&DocumentObject::getRaw>("_raw") |
| 1315 | .property<&DocumentObject::getSpan>("_span") |
| 1316 | .property<&DocumentObject::getMetadata>("_metadata") |
| 1317 | .property<&DocumentObject::getMultiMetadata>("_multi_metadata") |
| 1318 | .property<&DocumentObject::getNumericMetadata>("_numeric_metadata") |
| 1319 | .property<&DocumentObject::getZ2>("_subtopics") |
| 1320 | .property<&DocumentObject::getWindows>("_windows") |
| 1321 | .property<&DocumentObject::getPath>("_path") |
| 1322 | .property<&DocumentObject::getBeta>("_beta") |
| 1323 | .property<&DocumentObject::getY>("_vars") |
| 1324 | .property<&DocumentObject::getLabels>("_labels") |
| 1325 | .property<&DocumentObject::getEta>("_eta") |
| 1326 | .property<&DocumentObject::getTimepoint>("_timepoint") |
| 1327 | .property<&DocumentObject::getPseudoDocId>("_pseudo_doc_id")); |
| 1328 | |
| 1329 | module.addType(py::define<PhraserObject>("tomotopy._Phraser", "_Phraser", Py_TPFLAGS_BASETYPE) |
| 1330 | .call<&PhraserObject::call>() |
| 1331 | .method<&PhraserObject::findall>("findall") |
| 1332 | .method<&PhraserObject::save>("save") |