netgen-mesher 6.2.2506.post35.dev0__cp314-cp314-win_amd64.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (340) hide show
  1. netgen/NgOCC.py +7 -0
  2. netgen/__init__.py +114 -0
  3. netgen/__init__.pyi +22 -0
  4. netgen/__main__.py +53 -0
  5. netgen/cmake/NetgenConfig.cmake +79 -0
  6. netgen/cmake/netgen-targets-release.cmake +69 -0
  7. netgen/cmake/netgen-targets.cmake +146 -0
  8. netgen/config/__init__.py +1 -0
  9. netgen/config/__init__.pyi +52 -0
  10. netgen/config/__main__.py +4 -0
  11. netgen/config/config.py +68 -0
  12. netgen/config/config.pyi +54 -0
  13. netgen/csg.py +25 -0
  14. netgen/geom2d.py +178 -0
  15. netgen/gui.py +82 -0
  16. netgen/include/core/archive.hpp +1256 -0
  17. netgen/include/core/array.hpp +1760 -0
  18. netgen/include/core/autodiff.hpp +1131 -0
  19. netgen/include/core/autodiffdiff.hpp +733 -0
  20. netgen/include/core/bitarray.hpp +240 -0
  21. netgen/include/core/concurrentqueue.h +3619 -0
  22. netgen/include/core/exception.hpp +145 -0
  23. netgen/include/core/flags.hpp +199 -0
  24. netgen/include/core/hashtable.hpp +1281 -0
  25. netgen/include/core/localheap.hpp +318 -0
  26. netgen/include/core/logging.hpp +117 -0
  27. netgen/include/core/memtracer.hpp +221 -0
  28. netgen/include/core/mpi4py_pycapi.h +245 -0
  29. netgen/include/core/mpi_wrapper.hpp +643 -0
  30. netgen/include/core/ng_mpi.hpp +94 -0
  31. netgen/include/core/ng_mpi_generated_declarations.hpp +155 -0
  32. netgen/include/core/ng_mpi_native.hpp +25 -0
  33. netgen/include/core/ngcore.hpp +32 -0
  34. netgen/include/core/ngcore_api.hpp +152 -0
  35. netgen/include/core/ngstream.hpp +115 -0
  36. netgen/include/core/paje_trace.hpp +279 -0
  37. netgen/include/core/profiler.hpp +382 -0
  38. netgen/include/core/python_ngcore.hpp +457 -0
  39. netgen/include/core/ranges.hpp +109 -0
  40. netgen/include/core/register_archive.hpp +100 -0
  41. netgen/include/core/signal.hpp +82 -0
  42. netgen/include/core/simd.hpp +160 -0
  43. netgen/include/core/simd_arm64.hpp +407 -0
  44. netgen/include/core/simd_avx.hpp +394 -0
  45. netgen/include/core/simd_avx512.hpp +285 -0
  46. netgen/include/core/simd_generic.hpp +1053 -0
  47. netgen/include/core/simd_math.hpp +178 -0
  48. netgen/include/core/simd_sse.hpp +289 -0
  49. netgen/include/core/statushandler.hpp +37 -0
  50. netgen/include/core/symboltable.hpp +153 -0
  51. netgen/include/core/table.hpp +810 -0
  52. netgen/include/core/taskmanager.hpp +1161 -0
  53. netgen/include/core/type_traits.hpp +65 -0
  54. netgen/include/core/utils.hpp +385 -0
  55. netgen/include/core/version.hpp +102 -0
  56. netgen/include/core/xbool.hpp +47 -0
  57. netgen/include/csg/algprim.hpp +563 -0
  58. netgen/include/csg/brick.hpp +150 -0
  59. netgen/include/csg/csg.hpp +43 -0
  60. netgen/include/csg/csgeom.hpp +389 -0
  61. netgen/include/csg/csgparser.hpp +101 -0
  62. netgen/include/csg/curve2d.hpp +67 -0
  63. netgen/include/csg/edgeflw.hpp +112 -0
  64. netgen/include/csg/explicitcurve2d.hpp +113 -0
  65. netgen/include/csg/extrusion.hpp +185 -0
  66. netgen/include/csg/gencyl.hpp +70 -0
  67. netgen/include/csg/geoml.hpp +16 -0
  68. netgen/include/csg/identify.hpp +213 -0
  69. netgen/include/csg/manifold.hpp +29 -0
  70. netgen/include/csg/meshsurf.hpp +46 -0
  71. netgen/include/csg/polyhedra.hpp +121 -0
  72. netgen/include/csg/revolution.hpp +180 -0
  73. netgen/include/csg/singularref.hpp +84 -0
  74. netgen/include/csg/solid.hpp +295 -0
  75. netgen/include/csg/specpoin.hpp +194 -0
  76. netgen/include/csg/spline3d.hpp +99 -0
  77. netgen/include/csg/splinesurface.hpp +85 -0
  78. netgen/include/csg/surface.hpp +394 -0
  79. netgen/include/csg/triapprox.hpp +63 -0
  80. netgen/include/csg/vscsg.hpp +34 -0
  81. netgen/include/general/autodiff.hpp +356 -0
  82. netgen/include/general/autoptr.hpp +39 -0
  83. netgen/include/general/gzstream.h +121 -0
  84. netgen/include/general/hashtabl.hpp +1692 -0
  85. netgen/include/general/myadt.hpp +48 -0
  86. netgen/include/general/mystring.hpp +226 -0
  87. netgen/include/general/netgenout.hpp +205 -0
  88. netgen/include/general/ngarray.hpp +797 -0
  89. netgen/include/general/ngbitarray.hpp +149 -0
  90. netgen/include/general/ngpython.hpp +74 -0
  91. netgen/include/general/optmem.hpp +44 -0
  92. netgen/include/general/parthreads.hpp +138 -0
  93. netgen/include/general/seti.hpp +50 -0
  94. netgen/include/general/sort.hpp +47 -0
  95. netgen/include/general/spbita2d.hpp +59 -0
  96. netgen/include/general/stack.hpp +114 -0
  97. netgen/include/general/table.hpp +280 -0
  98. netgen/include/general/template.hpp +509 -0
  99. netgen/include/geom2d/csg2d.hpp +750 -0
  100. netgen/include/geom2d/geometry2d.hpp +280 -0
  101. netgen/include/geom2d/spline2d.hpp +234 -0
  102. netgen/include/geom2d/vsgeom2d.hpp +28 -0
  103. netgen/include/gprim/adtree.hpp +1392 -0
  104. netgen/include/gprim/geom2d.hpp +858 -0
  105. netgen/include/gprim/geom3d.hpp +749 -0
  106. netgen/include/gprim/geomfuncs.hpp +212 -0
  107. netgen/include/gprim/geomobjects.hpp +544 -0
  108. netgen/include/gprim/geomops.hpp +404 -0
  109. netgen/include/gprim/geomtest3d.hpp +101 -0
  110. netgen/include/gprim/gprim.hpp +33 -0
  111. netgen/include/gprim/spline.hpp +778 -0
  112. netgen/include/gprim/splinegeometry.hpp +73 -0
  113. netgen/include/gprim/transform3d.hpp +216 -0
  114. netgen/include/include/acisgeom.hpp +3 -0
  115. netgen/include/include/csg.hpp +1 -0
  116. netgen/include/include/geometry2d.hpp +1 -0
  117. netgen/include/include/gprim.hpp +1 -0
  118. netgen/include/include/incopengl.hpp +62 -0
  119. netgen/include/include/inctcl.hpp +13 -0
  120. netgen/include/include/incvis.hpp +6 -0
  121. netgen/include/include/linalg.hpp +1 -0
  122. netgen/include/include/meshing.hpp +1 -0
  123. netgen/include/include/myadt.hpp +1 -0
  124. netgen/include/include/mydefs.hpp +70 -0
  125. netgen/include/include/mystdlib.h +59 -0
  126. netgen/include/include/netgen_config.hpp +27 -0
  127. netgen/include/include/netgen_version.hpp +9 -0
  128. netgen/include/include/nginterface_v2_impl.hpp +395 -0
  129. netgen/include/include/ngsimd.hpp +1 -0
  130. netgen/include/include/occgeom.hpp +1 -0
  131. netgen/include/include/opti.hpp +1 -0
  132. netgen/include/include/parallel.hpp +1 -0
  133. netgen/include/include/stlgeom.hpp +1 -0
  134. netgen/include/include/visual.hpp +1 -0
  135. netgen/include/interface/rw_medit.hpp +11 -0
  136. netgen/include/interface/writeuser.hpp +80 -0
  137. netgen/include/linalg/densemat.hpp +414 -0
  138. netgen/include/linalg/linalg.hpp +29 -0
  139. netgen/include/linalg/opti.hpp +142 -0
  140. netgen/include/linalg/polynomial.hpp +47 -0
  141. netgen/include/linalg/vector.hpp +217 -0
  142. netgen/include/meshing/adfront2.hpp +274 -0
  143. netgen/include/meshing/adfront3.hpp +332 -0
  144. netgen/include/meshing/basegeom.hpp +370 -0
  145. netgen/include/meshing/bcfunctions.hpp +53 -0
  146. netgen/include/meshing/bisect.hpp +72 -0
  147. netgen/include/meshing/boundarylayer.hpp +113 -0
  148. netgen/include/meshing/classifyhpel.hpp +1984 -0
  149. netgen/include/meshing/clusters.hpp +46 -0
  150. netgen/include/meshing/curvedelems.hpp +274 -0
  151. netgen/include/meshing/delaunay2d.hpp +73 -0
  152. netgen/include/meshing/fieldlines.hpp +103 -0
  153. netgen/include/meshing/findip.hpp +198 -0
  154. netgen/include/meshing/findip2.hpp +103 -0
  155. netgen/include/meshing/geomsearch.hpp +69 -0
  156. netgen/include/meshing/global.hpp +54 -0
  157. netgen/include/meshing/hpref_hex.hpp +330 -0
  158. netgen/include/meshing/hpref_prism.hpp +3405 -0
  159. netgen/include/meshing/hpref_pyramid.hpp +154 -0
  160. netgen/include/meshing/hpref_quad.hpp +2082 -0
  161. netgen/include/meshing/hpref_segm.hpp +122 -0
  162. netgen/include/meshing/hpref_tet.hpp +4230 -0
  163. netgen/include/meshing/hpref_trig.hpp +848 -0
  164. netgen/include/meshing/hprefinement.hpp +366 -0
  165. netgen/include/meshing/improve2.hpp +178 -0
  166. netgen/include/meshing/improve3.hpp +151 -0
  167. netgen/include/meshing/localh.hpp +223 -0
  168. netgen/include/meshing/meshclass.hpp +1076 -0
  169. netgen/include/meshing/meshfunc.hpp +47 -0
  170. netgen/include/meshing/meshing.hpp +63 -0
  171. netgen/include/meshing/meshing2.hpp +163 -0
  172. netgen/include/meshing/meshing3.hpp +123 -0
  173. netgen/include/meshing/meshtool.hpp +90 -0
  174. netgen/include/meshing/meshtype.hpp +1930 -0
  175. netgen/include/meshing/msghandler.hpp +62 -0
  176. netgen/include/meshing/paralleltop.hpp +172 -0
  177. netgen/include/meshing/python_mesh.hpp +206 -0
  178. netgen/include/meshing/ruler2.hpp +172 -0
  179. netgen/include/meshing/ruler3.hpp +211 -0
  180. netgen/include/meshing/soldata.hpp +141 -0
  181. netgen/include/meshing/specials.hpp +17 -0
  182. netgen/include/meshing/surfacegeom.hpp +73 -0
  183. netgen/include/meshing/topology.hpp +1003 -0
  184. netgen/include/meshing/validate.hpp +21 -0
  185. netgen/include/meshing/visual_interface.hpp +71 -0
  186. netgen/include/mydefs.hpp +70 -0
  187. netgen/include/nginterface.h +474 -0
  188. netgen/include/nginterface_v2.hpp +406 -0
  189. netgen/include/nglib.h +697 -0
  190. netgen/include/nglib_occ.h +50 -0
  191. netgen/include/occ/occ_edge.hpp +47 -0
  192. netgen/include/occ/occ_face.hpp +52 -0
  193. netgen/include/occ/occ_solid.hpp +23 -0
  194. netgen/include/occ/occ_utils.hpp +376 -0
  195. netgen/include/occ/occ_vertex.hpp +30 -0
  196. netgen/include/occ/occgeom.hpp +659 -0
  197. netgen/include/occ/occmeshsurf.hpp +168 -0
  198. netgen/include/occ/vsocc.hpp +33 -0
  199. netgen/include/pybind11/LICENSE +29 -0
  200. netgen/include/pybind11/attr.h +722 -0
  201. netgen/include/pybind11/buffer_info.h +208 -0
  202. netgen/include/pybind11/cast.h +2361 -0
  203. netgen/include/pybind11/chrono.h +228 -0
  204. netgen/include/pybind11/common.h +2 -0
  205. netgen/include/pybind11/complex.h +74 -0
  206. netgen/include/pybind11/conduit/README.txt +15 -0
  207. netgen/include/pybind11/conduit/pybind11_conduit_v1.h +116 -0
  208. netgen/include/pybind11/conduit/pybind11_platform_abi_id.h +87 -0
  209. netgen/include/pybind11/conduit/wrap_include_python_h.h +72 -0
  210. netgen/include/pybind11/critical_section.h +56 -0
  211. netgen/include/pybind11/detail/class.h +823 -0
  212. netgen/include/pybind11/detail/common.h +1348 -0
  213. netgen/include/pybind11/detail/cpp_conduit.h +75 -0
  214. netgen/include/pybind11/detail/descr.h +226 -0
  215. netgen/include/pybind11/detail/dynamic_raw_ptr_cast_if_possible.h +39 -0
  216. netgen/include/pybind11/detail/exception_translation.h +71 -0
  217. netgen/include/pybind11/detail/function_record_pyobject.h +191 -0
  218. netgen/include/pybind11/detail/init.h +538 -0
  219. netgen/include/pybind11/detail/internals.h +799 -0
  220. netgen/include/pybind11/detail/native_enum_data.h +209 -0
  221. netgen/include/pybind11/detail/pybind11_namespace_macros.h +82 -0
  222. netgen/include/pybind11/detail/struct_smart_holder.h +378 -0
  223. netgen/include/pybind11/detail/type_caster_base.h +1591 -0
  224. netgen/include/pybind11/detail/typeid.h +65 -0
  225. netgen/include/pybind11/detail/using_smart_holder.h +22 -0
  226. netgen/include/pybind11/detail/value_and_holder.h +90 -0
  227. netgen/include/pybind11/eigen/common.h +9 -0
  228. netgen/include/pybind11/eigen/matrix.h +723 -0
  229. netgen/include/pybind11/eigen/tensor.h +521 -0
  230. netgen/include/pybind11/eigen.h +12 -0
  231. netgen/include/pybind11/embed.h +320 -0
  232. netgen/include/pybind11/eval.h +161 -0
  233. netgen/include/pybind11/functional.h +147 -0
  234. netgen/include/pybind11/gil.h +199 -0
  235. netgen/include/pybind11/gil_safe_call_once.h +102 -0
  236. netgen/include/pybind11/gil_simple.h +37 -0
  237. netgen/include/pybind11/iostream.h +265 -0
  238. netgen/include/pybind11/native_enum.h +67 -0
  239. netgen/include/pybind11/numpy.h +2312 -0
  240. netgen/include/pybind11/operators.h +202 -0
  241. netgen/include/pybind11/options.h +92 -0
  242. netgen/include/pybind11/pybind11.h +3645 -0
  243. netgen/include/pybind11/pytypes.h +2680 -0
  244. netgen/include/pybind11/stl/filesystem.h +114 -0
  245. netgen/include/pybind11/stl.h +666 -0
  246. netgen/include/pybind11/stl_bind.h +858 -0
  247. netgen/include/pybind11/subinterpreter.h +299 -0
  248. netgen/include/pybind11/trampoline_self_life_support.h +65 -0
  249. netgen/include/pybind11/type_caster_pyobject_ptr.h +61 -0
  250. netgen/include/pybind11/typing.h +298 -0
  251. netgen/include/pybind11/warnings.h +75 -0
  252. netgen/include/stlgeom/meshstlsurface.hpp +67 -0
  253. netgen/include/stlgeom/stlgeom.hpp +491 -0
  254. netgen/include/stlgeom/stlline.hpp +193 -0
  255. netgen/include/stlgeom/stltool.hpp +331 -0
  256. netgen/include/stlgeom/stltopology.hpp +419 -0
  257. netgen/include/stlgeom/vsstl.hpp +58 -0
  258. netgen/include/visualization/meshdoc.hpp +42 -0
  259. netgen/include/visualization/mvdraw.hpp +325 -0
  260. netgen/include/visualization/vispar.hpp +128 -0
  261. netgen/include/visualization/visual.hpp +28 -0
  262. netgen/include/visualization/visual_api.hpp +10 -0
  263. netgen/include/visualization/vssolution.hpp +399 -0
  264. netgen/lib/libnggui.lib +0 -0
  265. netgen/lib/ngcore.lib +0 -0
  266. netgen/lib/nglib.lib +0 -0
  267. netgen/lib/togl.lib +0 -0
  268. netgen/libnggui.dll +0 -0
  269. netgen/libngguipy.lib +0 -0
  270. netgen/libngguipy.pyd +0 -0
  271. netgen/libngpy/_NgOCC.pyi +1545 -0
  272. netgen/libngpy/__init__.pyi +7 -0
  273. netgen/libngpy/_csg.pyi +259 -0
  274. netgen/libngpy/_geom2d.pyi +323 -0
  275. netgen/libngpy/_meshing.pyi +1111 -0
  276. netgen/libngpy/_stl.pyi +131 -0
  277. netgen/libngpy.lib +0 -0
  278. netgen/libngpy.pyd +0 -0
  279. netgen/meshing.py +65 -0
  280. netgen/ngcore.dll +0 -0
  281. netgen/nglib.dll +0 -0
  282. netgen/occ.py +52 -0
  283. netgen/read_gmsh.py +259 -0
  284. netgen/read_meshio.py +22 -0
  285. netgen/stl.py +2 -0
  286. netgen/togl.dll +0 -0
  287. netgen/version.py +2 -0
  288. netgen/webgui.py +529 -0
  289. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/boundarycondition.geo +16 -0
  290. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/boxcyl.geo +32 -0
  291. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/circle_on_cube.geo +27 -0
  292. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/cone.geo +13 -0
  293. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/cube.geo +16 -0
  294. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/cubeandring.geo +55 -0
  295. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/cubeandspheres.geo +21 -0
  296. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/cubemcyl.geo +18 -0
  297. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/cubemsphere.geo +19 -0
  298. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/cylinder.geo +12 -0
  299. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/cylsphere.geo +12 -0
  300. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/doc/ng4.pdf +0 -0
  301. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/ellipsoid.geo +8 -0
  302. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/ellipticcyl.geo +10 -0
  303. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/extrusion.geo +99 -0
  304. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/fichera.geo +24 -0
  305. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/frame.step +11683 -0
  306. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/hinge.stl +8486 -0
  307. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/lshape3d.geo +26 -0
  308. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/manyholes.geo +26 -0
  309. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/manyholes2.geo +26 -0
  310. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/matrix.geo +27 -0
  311. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/ortho.geo +11 -0
  312. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/part1.stl +2662 -0
  313. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/period.geo +33 -0
  314. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/py_tutorials/exportNeutral.py +26 -0
  315. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/py_tutorials/mesh.py +19 -0
  316. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/py_tutorials/shaft.geo +65 -0
  317. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/revolution.geo +18 -0
  318. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/screw.step +1694 -0
  319. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/sculpture.geo +13 -0
  320. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/shaft.geo +65 -0
  321. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/shell.geo +10 -0
  322. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/sphere.geo +8 -0
  323. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/sphereincube.geo +17 -0
  324. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/square.in2d +35 -0
  325. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/squarecircle.in2d +48 -0
  326. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/squarehole.in2d +47 -0
  327. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/torus.geo +8 -0
  328. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/trafo.geo +57 -0
  329. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/twobricks.geo +15 -0
  330. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/twocubes.geo +18 -0
  331. netgen_mesher-6.2.2506.post35.dev0.data/data/share/netgen/twocyl.geo +16 -0
  332. netgen_mesher-6.2.2506.post35.dev0.dist-info/METADATA +15 -0
  333. netgen_mesher-6.2.2506.post35.dev0.dist-info/RECORD +340 -0
  334. netgen_mesher-6.2.2506.post35.dev0.dist-info/WHEEL +5 -0
  335. netgen_mesher-6.2.2506.post35.dev0.dist-info/entry_points.txt +2 -0
  336. netgen_mesher-6.2.2506.post35.dev0.dist-info/licenses/AUTHORS +1 -0
  337. netgen_mesher-6.2.2506.post35.dev0.dist-info/licenses/LICENSE +504 -0
  338. netgen_mesher-6.2.2506.post35.dev0.dist-info/top_level.txt +2 -0
  339. pyngcore/__init__.py +1 -0
  340. pyngcore/pyngcore.cp314-win_amd64.pyd +0 -0
@@ -0,0 +1,858 @@
1
+ /*
2
+ pybind11/std_bind.h: Binding generators for STL data types
3
+
4
+ Copyright (c) 2016 Sergey Lyskov and Wenzel Jakob
5
+
6
+ All rights reserved. Use of this source code is governed by a
7
+ BSD-style license that can be found in the LICENSE file.
8
+ */
9
+
10
+ #pragma once
11
+
12
+ #include "detail/common.h"
13
+ #include "detail/type_caster_base.h"
14
+ #include "cast.h"
15
+ #include "operators.h"
16
+
17
+ #include <algorithm>
18
+ #include <sstream>
19
+ #include <type_traits>
20
+
21
+ PYBIND11_NAMESPACE_BEGIN(PYBIND11_NAMESPACE)
22
+ PYBIND11_NAMESPACE_BEGIN(detail)
23
+
24
+ /* SFINAE helper class used by 'is_comparable */
25
+ template <typename T>
26
+ struct container_traits {
27
+ template <typename T2>
28
+ static std::true_type
29
+ test_comparable(decltype(std::declval<const T2 &>() == std::declval<const T2 &>()) *);
30
+ template <typename T2>
31
+ static std::false_type test_comparable(...);
32
+ template <typename T2>
33
+ static std::true_type test_value(typename T2::value_type *);
34
+ template <typename T2>
35
+ static std::false_type test_value(...);
36
+ template <typename T2>
37
+ static std::true_type test_pair(typename T2::first_type *, typename T2::second_type *);
38
+ template <typename T2>
39
+ static std::false_type test_pair(...);
40
+
41
+ static constexpr const bool is_comparable
42
+ = std::is_same<std::true_type, decltype(test_comparable<T>(nullptr))>::value;
43
+ static constexpr const bool is_pair
44
+ = std::is_same<std::true_type, decltype(test_pair<T>(nullptr, nullptr))>::value;
45
+ static constexpr const bool is_vector
46
+ = std::is_same<std::true_type, decltype(test_value<T>(nullptr))>::value;
47
+ static constexpr const bool is_element = !is_pair && !is_vector;
48
+ };
49
+
50
+ /* Default: is_comparable -> std::false_type */
51
+ template <typename T, typename SFINAE = void>
52
+ struct is_comparable : std::false_type {};
53
+
54
+ /* For non-map data structures, check whether operator== can be instantiated */
55
+ template <typename T>
56
+ struct is_comparable<
57
+ T,
58
+ enable_if_t<container_traits<T>::is_element && container_traits<T>::is_comparable>>
59
+ : std::true_type {};
60
+
61
+ /* For a vector/map data structure, recursively check the value type
62
+ (which is std::pair for maps) */
63
+ template <typename T>
64
+ struct is_comparable<T, enable_if_t<container_traits<T>::is_vector>>
65
+ : is_comparable<typename recursive_container_traits<T>::type_to_check_recursively> {};
66
+
67
+ template <>
68
+ struct is_comparable<recursive_bottom> : std::true_type {};
69
+
70
+ /* For pairs, recursively check the two data types */
71
+ template <typename T>
72
+ struct is_comparable<T, enable_if_t<container_traits<T>::is_pair>> {
73
+ static constexpr const bool value = is_comparable<typename T::first_type>::value
74
+ && is_comparable<typename T::second_type>::value;
75
+ };
76
+
77
+ /* Fallback functions */
78
+ template <typename, typename, typename... Args>
79
+ void vector_if_copy_constructible(const Args &...) {}
80
+ template <typename, typename, typename... Args>
81
+ void vector_if_equal_operator(const Args &...) {}
82
+ template <typename, typename, typename... Args>
83
+ void vector_if_insertion_operator(const Args &...) {}
84
+ template <typename, typename, typename... Args>
85
+ void vector_modifiers(const Args &...) {}
86
+
87
+ template <typename Vector, typename Class_>
88
+ void vector_if_copy_constructible(enable_if_t<is_copy_constructible<Vector>::value, Class_> &cl) {
89
+ cl.def(init<const Vector &>(), "Copy constructor");
90
+ }
91
+
92
+ template <typename Vector, typename Class_>
93
+ void vector_if_equal_operator(enable_if_t<is_comparable<Vector>::value, Class_> &cl) {
94
+ using T = typename Vector::value_type;
95
+
96
+ cl.def(self == self);
97
+ cl.def(self != self);
98
+
99
+ cl.def(
100
+ "count",
101
+ [](const Vector &v, const T &x) { return std::count(v.begin(), v.end(), x); },
102
+ arg("x"),
103
+ "Return the number of times ``x`` appears in the list");
104
+
105
+ cl.def(
106
+ "remove",
107
+ [](Vector &v, const T &x) {
108
+ auto p = std::find(v.begin(), v.end(), x);
109
+ if (p != v.end()) {
110
+ v.erase(p);
111
+ } else {
112
+ throw value_error();
113
+ }
114
+ },
115
+ arg("x"),
116
+ "Remove the first item from the list whose value is x. "
117
+ "It is an error if there is no such item.");
118
+
119
+ cl.def(
120
+ "__contains__",
121
+ [](const Vector &v, const T &x) { return std::find(v.begin(), v.end(), x) != v.end(); },
122
+ arg("x"),
123
+ "Return true the container contains ``x``");
124
+ }
125
+
126
+ // Vector modifiers -- requires a copyable vector_type:
127
+ // (Technically, some of these (pop and __delitem__) don't actually require copyability, but it
128
+ // seems silly to allow deletion but not insertion, so include them here too.)
129
+ template <typename Vector, typename Class_>
130
+ void vector_modifiers(
131
+ enable_if_t<is_copy_constructible<typename Vector::value_type>::value, Class_> &cl) {
132
+ using T = typename Vector::value_type;
133
+ using SizeType = typename Vector::size_type;
134
+ using DiffType = typename Vector::difference_type;
135
+
136
+ auto wrap_i = [](DiffType i, SizeType n) {
137
+ if (i < 0) {
138
+ i += n;
139
+ }
140
+ if (i < 0 || (SizeType) i >= n) {
141
+ throw index_error();
142
+ }
143
+ return i;
144
+ };
145
+
146
+ cl.def(
147
+ "append",
148
+ [](Vector &v, const T &value) { v.push_back(value); },
149
+ arg("x"),
150
+ "Add an item to the end of the list");
151
+
152
+ cl.def(init([](const iterable &it) {
153
+ auto v = std::unique_ptr<Vector>(new Vector());
154
+ v->reserve(len_hint(it));
155
+ for (handle h : it) {
156
+ v->push_back(h.cast<T>());
157
+ }
158
+ return v.release();
159
+ }));
160
+
161
+ cl.def("clear", [](Vector &v) { v.clear(); }, "Clear the contents");
162
+
163
+ cl.def(
164
+ "extend",
165
+ [](Vector &v, const Vector &src) { v.insert(v.end(), src.begin(), src.end()); },
166
+ arg("L"),
167
+ "Extend the list by appending all the items in the given list");
168
+
169
+ cl.def(
170
+ "extend",
171
+ [](Vector &v, const iterable &it) {
172
+ const size_t old_size = v.size();
173
+ v.reserve(old_size + len_hint(it));
174
+ try {
175
+ for (handle h : it) {
176
+ v.push_back(h.cast<T>());
177
+ }
178
+ } catch (const cast_error &) {
179
+ v.erase(v.begin() + static_cast<typename Vector::difference_type>(old_size),
180
+ v.end());
181
+ try {
182
+ v.shrink_to_fit();
183
+ } catch (const std::exception &) { // NOLINT(bugprone-empty-catch)
184
+ // Do nothing
185
+ }
186
+ throw;
187
+ }
188
+ },
189
+ arg("L"),
190
+ "Extend the list by appending all the items in the given list");
191
+
192
+ cl.def(
193
+ "insert",
194
+ [](Vector &v, DiffType i, const T &x) {
195
+ // Can't use wrap_i; i == v.size() is OK
196
+ if (i < 0) {
197
+ i += v.size();
198
+ }
199
+ if (i < 0 || (SizeType) i > v.size()) {
200
+ throw index_error();
201
+ }
202
+ v.insert(v.begin() + i, x);
203
+ },
204
+ arg("i"),
205
+ arg("x"),
206
+ "Insert an item at a given position.");
207
+
208
+ cl.def(
209
+ "pop",
210
+ [](Vector &v) {
211
+ if (v.empty()) {
212
+ throw index_error();
213
+ }
214
+ T t = std::move(v.back());
215
+ v.pop_back();
216
+ return t;
217
+ },
218
+ "Remove and return the last item");
219
+
220
+ cl.def(
221
+ "pop",
222
+ [wrap_i](Vector &v, DiffType i) {
223
+ i = wrap_i(i, v.size());
224
+ T t = std::move(v[(SizeType) i]);
225
+ v.erase(std::next(v.begin(), i));
226
+ return t;
227
+ },
228
+ arg("i"),
229
+ "Remove and return the item at index ``i``");
230
+
231
+ cl.def("__setitem__", [wrap_i](Vector &v, DiffType i, const T &t) {
232
+ i = wrap_i(i, v.size());
233
+ v[(SizeType) i] = t;
234
+ });
235
+
236
+ /// Slicing protocol
237
+ cl.def(
238
+ "__getitem__",
239
+ [](const Vector &v, const slice &slice) -> Vector * {
240
+ size_t start = 0, stop = 0, step = 0, slicelength = 0;
241
+
242
+ if (!slice.compute(v.size(), &start, &stop, &step, &slicelength)) {
243
+ throw error_already_set();
244
+ }
245
+
246
+ auto *seq = new Vector();
247
+ seq->reserve((size_t) slicelength);
248
+
249
+ for (size_t i = 0; i < slicelength; ++i) {
250
+ seq->push_back(v[start]);
251
+ start += step;
252
+ }
253
+ return seq;
254
+ },
255
+ arg("s"),
256
+ "Retrieve list elements using a slice object");
257
+
258
+ cl.def(
259
+ "__setitem__",
260
+ [](Vector &v, const slice &slice, const Vector &value) {
261
+ size_t start = 0, stop = 0, step = 0, slicelength = 0;
262
+ if (!slice.compute(v.size(), &start, &stop, &step, &slicelength)) {
263
+ throw error_already_set();
264
+ }
265
+
266
+ if (slicelength != value.size()) {
267
+ throw std::runtime_error(
268
+ "Left and right hand size of slice assignment have different sizes!");
269
+ }
270
+
271
+ for (size_t i = 0; i < slicelength; ++i) {
272
+ v[start] = value[i];
273
+ start += step;
274
+ }
275
+ },
276
+ "Assign list elements using a slice object");
277
+
278
+ cl.def(
279
+ "__delitem__",
280
+ [wrap_i](Vector &v, DiffType i) {
281
+ i = wrap_i(i, v.size());
282
+ v.erase(v.begin() + i);
283
+ },
284
+ "Delete the list elements at index ``i``");
285
+
286
+ cl.def(
287
+ "__delitem__",
288
+ [](Vector &v, const slice &slice) {
289
+ size_t start = 0, stop = 0, step = 0, slicelength = 0;
290
+
291
+ if (!slice.compute(v.size(), &start, &stop, &step, &slicelength)) {
292
+ throw error_already_set();
293
+ }
294
+
295
+ if (step == 1 && false) {
296
+ v.erase(v.begin() + (DiffType) start, v.begin() + DiffType(start + slicelength));
297
+ } else {
298
+ for (size_t i = 0; i < slicelength; ++i) {
299
+ v.erase(v.begin() + DiffType(start));
300
+ start += step - 1;
301
+ }
302
+ }
303
+ },
304
+ "Delete list elements using a slice object");
305
+ }
306
+
307
+ // If the type has an operator[] that doesn't return a reference (most notably std::vector<bool>),
308
+ // we have to access by copying; otherwise we return by reference.
309
+ template <typename Vector>
310
+ using vector_needs_copy
311
+ = negation<std::is_same<decltype(std::declval<Vector>()[typename Vector::size_type()]),
312
+ typename Vector::value_type &>>;
313
+
314
+ // The usual case: access and iterate by reference
315
+ template <typename Vector, typename Class_>
316
+ void vector_accessor(enable_if_t<!vector_needs_copy<Vector>::value, Class_> &cl) {
317
+ using T = typename Vector::value_type;
318
+ using SizeType = typename Vector::size_type;
319
+ using DiffType = typename Vector::difference_type;
320
+ using ItType = typename Vector::iterator;
321
+
322
+ auto wrap_i = [](DiffType i, SizeType n) {
323
+ if (i < 0) {
324
+ i += n;
325
+ }
326
+ if (i < 0 || (SizeType) i >= n) {
327
+ throw index_error();
328
+ }
329
+ return i;
330
+ };
331
+
332
+ cl.def(
333
+ "__getitem__",
334
+ [wrap_i](Vector &v, DiffType i) -> T & {
335
+ i = wrap_i(i, v.size());
336
+ return v[(SizeType) i];
337
+ },
338
+ return_value_policy::reference_internal // ref + keepalive
339
+ );
340
+
341
+ cl.def(
342
+ "__iter__",
343
+ [](Vector &v) {
344
+ return make_iterator<return_value_policy::reference_internal, ItType, ItType, T &>(
345
+ v.begin(), v.end());
346
+ },
347
+ keep_alive<0, 1>() /* Essential: keep list alive while iterator exists */
348
+ );
349
+ }
350
+
351
+ // The case for special objects, like std::vector<bool>, that have to be returned-by-copy:
352
+ template <typename Vector, typename Class_>
353
+ void vector_accessor(enable_if_t<vector_needs_copy<Vector>::value, Class_> &cl) {
354
+ using T = typename Vector::value_type;
355
+ using SizeType = typename Vector::size_type;
356
+ using DiffType = typename Vector::difference_type;
357
+ using ItType = typename Vector::iterator;
358
+ cl.def("__getitem__", [](const Vector &v, DiffType i) -> T {
359
+ if (i < 0) {
360
+ i += v.size();
361
+ if (i < 0) {
362
+ throw index_error();
363
+ }
364
+ }
365
+ auto i_st = static_cast<SizeType>(i);
366
+ if (i_st >= v.size()) {
367
+ throw index_error();
368
+ }
369
+ return v[i_st];
370
+ });
371
+
372
+ cl.def(
373
+ "__iter__",
374
+ [](Vector &v) {
375
+ return make_iterator<return_value_policy::copy, ItType, ItType, T>(v.begin(), v.end());
376
+ },
377
+ keep_alive<0, 1>() /* Essential: keep list alive while iterator exists */
378
+ );
379
+ }
380
+
381
+ template <typename Vector, typename Class_>
382
+ auto vector_if_insertion_operator(Class_ &cl, std::string const &name)
383
+ -> decltype(std::declval<std::ostream &>() << std::declval<typename Vector::value_type>(),
384
+ void()) {
385
+ using size_type = typename Vector::size_type;
386
+
387
+ cl.def(
388
+ "__repr__",
389
+ [name](Vector &v) {
390
+ std::ostringstream s;
391
+ s << name << '[';
392
+ for (size_type i = 0; i < v.size(); ++i) {
393
+ s << v[i];
394
+ if (i != v.size() - 1) {
395
+ s << ", ";
396
+ }
397
+ }
398
+ s << ']';
399
+ return s.str();
400
+ },
401
+ "Return the canonical string representation of this list.");
402
+ }
403
+
404
+ // Provide the buffer interface for vectors if we have data() and we have a format for it
405
+ // GCC seems to have "void std::vector<bool>::data()" - doing SFINAE on the existence of data()
406
+ // is insufficient, we need to check it returns an appropriate pointer
407
+ template <typename Vector, typename = void>
408
+ struct vector_has_data_and_format : std::false_type {};
409
+ template <typename Vector>
410
+ struct vector_has_data_and_format<
411
+ Vector,
412
+ enable_if_t<std::is_same<decltype(format_descriptor<typename Vector::value_type>::format(),
413
+ std::declval<Vector>().data()),
414
+ typename Vector::value_type *>::value>> : std::true_type {};
415
+
416
+ // [workaround(intel)] Separate function required here
417
+ // Workaround as the Intel compiler does not compile the enable_if_t part below
418
+ // (tested with icc (ICC) 2021.1 Beta 20200827)
419
+ template <typename... Args>
420
+ constexpr bool args_any_are_buffer() {
421
+ return detail::any_of<std::is_same<Args, buffer_protocol>...>::value;
422
+ }
423
+
424
+ // [workaround(intel)] Separate function required here
425
+ // [workaround(msvc)] Can't use constexpr bool in return type
426
+
427
+ // Add the buffer interface to a vector
428
+ template <typename Vector, typename Class_, typename... Args>
429
+ void vector_buffer_impl(Class_ &cl, std::true_type) {
430
+ using T = typename Vector::value_type;
431
+
432
+ static_assert(vector_has_data_and_format<Vector>::value,
433
+ "There is not an appropriate format descriptor for this vector");
434
+
435
+ // numpy.h declares this for arbitrary types, but it may raise an exception and crash hard
436
+ // at runtime if PYBIND11_NUMPY_DTYPE hasn't been called, so check here
437
+ format_descriptor<T>::format();
438
+
439
+ cl.def_buffer([](Vector &v) -> buffer_info {
440
+ return buffer_info(v.data(),
441
+ static_cast<ssize_t>(sizeof(T)),
442
+ format_descriptor<T>::format(),
443
+ 1,
444
+ {v.size()},
445
+ {sizeof(T)});
446
+ });
447
+
448
+ cl.def(init([](const buffer &buf) {
449
+ auto info = buf.request();
450
+ if (info.ndim != 1 || info.strides[0] % static_cast<ssize_t>(sizeof(T))) {
451
+ throw type_error("Only valid 1D buffers can be copied to a vector");
452
+ }
453
+ if (!detail::compare_buffer_info<T>::compare(info)
454
+ || (ssize_t) sizeof(T) != info.itemsize) {
455
+ throw type_error("Format mismatch (Python: " + info.format
456
+ + " C++: " + format_descriptor<T>::format() + ")");
457
+ }
458
+
459
+ T *p = static_cast<T *>(info.ptr);
460
+ ssize_t step = info.strides[0] / static_cast<ssize_t>(sizeof(T));
461
+ T *end = p + info.shape[0] * step;
462
+ if (step == 1) {
463
+ return Vector(p, end);
464
+ }
465
+ Vector vec;
466
+ vec.reserve((size_t) info.shape[0]);
467
+ for (; p != end; p += step) {
468
+ vec.push_back(*p);
469
+ }
470
+ return vec;
471
+ }));
472
+
473
+ return;
474
+ }
475
+
476
+ template <typename Vector, typename Class_, typename... Args>
477
+ void vector_buffer_impl(Class_ &, std::false_type) {}
478
+
479
+ template <typename Vector, typename Class_, typename... Args>
480
+ void vector_buffer(Class_ &cl) {
481
+ vector_buffer_impl<Vector, Class_, Args...>(
482
+ cl, detail::any_of<std::is_same<Args, buffer_protocol>...>{});
483
+ }
484
+
485
+ PYBIND11_NAMESPACE_END(detail)
486
+
487
+ //
488
+ // std::vector
489
+ //
490
+ template <typename Vector, typename holder_type = default_holder_type<Vector>, typename... Args>
491
+ class_<Vector, holder_type> bind_vector(handle scope, std::string const &name, Args &&...args) {
492
+ using Class_ = class_<Vector, holder_type>;
493
+
494
+ // If the value_type is unregistered (e.g. a converting type) or is itself registered
495
+ // module-local then make the vector binding module-local as well:
496
+ using vtype = typename Vector::value_type;
497
+ auto *vtype_info = detail::get_type_info(typeid(vtype));
498
+ bool local = !vtype_info || vtype_info->module_local;
499
+
500
+ Class_ cl(scope, name.c_str(), pybind11::module_local(local), std::forward<Args>(args)...);
501
+
502
+ // Declare the buffer interface if a buffer_protocol() is passed in
503
+ detail::vector_buffer<Vector, Class_, Args...>(cl);
504
+
505
+ cl.def(init<>());
506
+
507
+ // Register copy constructor (if possible)
508
+ detail::vector_if_copy_constructible<Vector, Class_>(cl);
509
+
510
+ // Register comparison-related operators and functions (if possible)
511
+ detail::vector_if_equal_operator<Vector, Class_>(cl);
512
+
513
+ // Register stream insertion operator (if possible)
514
+ detail::vector_if_insertion_operator<Vector, Class_>(cl, name);
515
+
516
+ // Modifiers require copyable vector value type
517
+ detail::vector_modifiers<Vector, Class_>(cl);
518
+
519
+ // Accessor and iterator; return by value if copyable, otherwise we return by ref + keep-alive
520
+ detail::vector_accessor<Vector, Class_>(cl);
521
+
522
+ cl.def(
523
+ "__bool__",
524
+ [](const Vector &v) -> bool { return !v.empty(); },
525
+ "Check whether the list is nonempty");
526
+
527
+ cl.def("__len__", [](const Vector &vec) { return vec.size(); });
528
+
529
+ #if 0
530
+ // C++ style functions deprecated, leaving it here as an example
531
+ cl.def(init<size_type>());
532
+
533
+ cl.def("resize",
534
+ (void (Vector::*) (size_type count)) & Vector::resize,
535
+ "changes the number of elements stored");
536
+
537
+ cl.def("erase",
538
+ [](Vector &v, SizeType i) {
539
+ if (i >= v.size())
540
+ throw index_error();
541
+ v.erase(v.begin() + i);
542
+ }, "erases element at index ``i``");
543
+
544
+ cl.def("empty", &Vector::empty, "checks whether the container is empty");
545
+ cl.def("size", &Vector::size, "returns the number of elements");
546
+ cl.def("push_back", (void (Vector::*)(const T&)) &Vector::push_back, "adds an element to the end");
547
+ cl.def("pop_back", &Vector::pop_back, "removes the last element");
548
+
549
+ cl.def("max_size", &Vector::max_size, "returns the maximum possible number of elements");
550
+ cl.def("reserve", &Vector::reserve, "reserves storage");
551
+ cl.def("capacity", &Vector::capacity, "returns the number of elements that can be held in currently allocated storage");
552
+ cl.def("shrink_to_fit", &Vector::shrink_to_fit, "reduces memory usage by freeing unused memory");
553
+
554
+ cl.def("clear", &Vector::clear, "clears the contents");
555
+ cl.def("swap", &Vector::swap, "swaps the contents");
556
+
557
+ cl.def("front", [](Vector &v) {
558
+ if (v.size()) return v.front();
559
+ else throw index_error();
560
+ }, "access the first element");
561
+
562
+ cl.def("back", [](Vector &v) {
563
+ if (v.size()) return v.back();
564
+ else throw index_error();
565
+ }, "access the last element ");
566
+
567
+ #endif
568
+
569
+ return cl;
570
+ }
571
+
572
+ //
573
+ // std::map, std::unordered_map
574
+ //
575
+
576
+ PYBIND11_NAMESPACE_BEGIN(detail)
577
+
578
+ /* Fallback functions */
579
+ template <typename, typename, typename... Args>
580
+ void map_if_insertion_operator(const Args &...) {}
581
+ template <typename, typename, typename... Args>
582
+ void map_assignment(const Args &...) {}
583
+
584
+ // Map assignment when copy-assignable: just copy the value
585
+ template <typename Map, typename Class_>
586
+ void map_assignment(
587
+ enable_if_t<is_copy_assignable<typename Map::mapped_type>::value, Class_> &cl) {
588
+ using KeyType = typename Map::key_type;
589
+ using MappedType = typename Map::mapped_type;
590
+
591
+ cl.def("__setitem__", [](Map &m, const KeyType &k, const MappedType &v) {
592
+ auto it = m.find(k);
593
+ if (it != m.end()) {
594
+ it->second = v;
595
+ } else {
596
+ m.emplace(k, v);
597
+ }
598
+ });
599
+ }
600
+
601
+ // Not copy-assignable, but still copy-constructible: we can update the value by erasing and
602
+ // reinserting
603
+ template <typename Map, typename Class_>
604
+ void map_assignment(enable_if_t<!is_copy_assignable<typename Map::mapped_type>::value
605
+ && is_copy_constructible<typename Map::mapped_type>::value,
606
+ Class_> &cl) {
607
+ using KeyType = typename Map::key_type;
608
+ using MappedType = typename Map::mapped_type;
609
+
610
+ cl.def("__setitem__", [](Map &m, const KeyType &k, const MappedType &v) {
611
+ // We can't use m[k] = v; because value type might not be default constructable
612
+ auto r = m.emplace(k, v);
613
+ if (!r.second) {
614
+ // value type is not copy assignable so the only way to insert it is to erase it
615
+ // first...
616
+ m.erase(r.first);
617
+ m.emplace(k, v);
618
+ }
619
+ });
620
+ }
621
+
622
+ template <typename Map, typename Class_>
623
+ auto map_if_insertion_operator(Class_ &cl, std::string const &name)
624
+ -> decltype(std::declval<std::ostream &>() << std::declval<typename Map::key_type>()
625
+ << std::declval<typename Map::mapped_type>(),
626
+ void()) {
627
+
628
+ cl.def(
629
+ "__repr__",
630
+ [name](Map &m) {
631
+ std::ostringstream s;
632
+ s << name << '{';
633
+ bool f = false;
634
+ for (auto const &kv : m) {
635
+ if (f) {
636
+ s << ", ";
637
+ }
638
+ s << kv.first << ": " << kv.second;
639
+ f = true;
640
+ }
641
+ s << '}';
642
+ return s.str();
643
+ },
644
+ "Return the canonical string representation of this map.");
645
+ }
646
+
647
+ struct keys_view {
648
+ virtual size_t len() = 0;
649
+ virtual iterator iter() = 0;
650
+ virtual bool contains(const handle &k) = 0;
651
+ virtual ~keys_view() = default;
652
+ };
653
+
654
+ struct values_view {
655
+ virtual size_t len() = 0;
656
+ virtual iterator iter() = 0;
657
+ virtual ~values_view() = default;
658
+ };
659
+
660
+ struct items_view {
661
+ virtual size_t len() = 0;
662
+ virtual iterator iter() = 0;
663
+ virtual ~items_view() = default;
664
+ };
665
+
666
+ template <typename Map>
667
+ struct KeysViewImpl : public detail::keys_view {
668
+ explicit KeysViewImpl(Map &map) : map(map) {}
669
+ size_t len() override { return map.size(); }
670
+ iterator iter() override { return make_key_iterator(map.begin(), map.end()); }
671
+ bool contains(const handle &k) override {
672
+ try {
673
+ return map.find(k.template cast<typename Map::key_type>()) != map.end();
674
+ } catch (const cast_error &) {
675
+ return false;
676
+ }
677
+ }
678
+ Map &map;
679
+ };
680
+
681
+ template <typename Map>
682
+ struct ValuesViewImpl : public detail::values_view {
683
+ explicit ValuesViewImpl(Map &map) : map(map) {}
684
+ size_t len() override { return map.size(); }
685
+ iterator iter() override { return make_value_iterator(map.begin(), map.end()); }
686
+ Map &map;
687
+ };
688
+
689
+ template <typename Map>
690
+ struct ItemsViewImpl : public detail::items_view {
691
+ explicit ItemsViewImpl(Map &map) : map(map) {}
692
+ size_t len() override { return map.size(); }
693
+ iterator iter() override { return make_iterator(map.begin(), map.end()); }
694
+ Map &map;
695
+ };
696
+
697
+ inline str format_message_key_error_key_object(handle py_key) {
698
+ str message = "pybind11::bind_map key";
699
+ if (!py_key) {
700
+ return message;
701
+ }
702
+ try {
703
+ message = str(py_key);
704
+ } catch (const std::exception &) {
705
+ try {
706
+ message = repr(py_key);
707
+ } catch (const std::exception &) {
708
+ return message;
709
+ }
710
+ }
711
+ const ssize_t cut_length = 100;
712
+ if (len(message) > 2 * cut_length + 3) {
713
+ return str(message[slice(0, cut_length, 1)]) + str("✄✄✄")
714
+ + str(message[slice(-cut_length, static_cast<ssize_t>(len(message)), 1)]);
715
+ }
716
+ return message;
717
+ }
718
+
719
+ template <typename KeyType>
720
+ str format_message_key_error(const KeyType &key) {
721
+ object py_key;
722
+ try {
723
+ py_key = cast(key);
724
+ } catch (const std::exception &) {
725
+ do { // Trick to avoid "empty catch" warning/error.
726
+ } while (false);
727
+ }
728
+ return format_message_key_error_key_object(py_key);
729
+ }
730
+
731
+ PYBIND11_NAMESPACE_END(detail)
732
+
733
+ template <typename Map, typename holder_type = default_holder_type<Map>, typename... Args>
734
+ class_<Map, holder_type> bind_map(handle scope, const std::string &name, Args &&...args) {
735
+ using KeyType = typename Map::key_type;
736
+ using MappedType = typename Map::mapped_type;
737
+ using KeysView = detail::keys_view;
738
+ using ValuesView = detail::values_view;
739
+ using ItemsView = detail::items_view;
740
+ using Class_ = class_<Map, holder_type>;
741
+
742
+ // If either type is a non-module-local bound type then make the map binding non-local as well;
743
+ // otherwise (e.g. both types are either module-local or converting) the map will be
744
+ // module-local.
745
+ auto *tinfo = detail::get_type_info(typeid(MappedType));
746
+ bool local = !tinfo || tinfo->module_local;
747
+ if (local) {
748
+ tinfo = detail::get_type_info(typeid(KeyType));
749
+ local = !tinfo || tinfo->module_local;
750
+ }
751
+
752
+ Class_ cl(scope, name.c_str(), pybind11::module_local(local), std::forward<Args>(args)...);
753
+
754
+ // Wrap KeysView if it wasn't already wrapped
755
+ if (!detail::get_type_info(typeid(KeysView))) {
756
+ class_<KeysView> keys_view(scope, "KeysView", pybind11::module_local(local));
757
+ keys_view.def("__len__", &KeysView::len);
758
+ keys_view.def("__iter__",
759
+ &KeysView::iter,
760
+ keep_alive<0, 1>() /* Essential: keep view alive while iterator exists */
761
+ );
762
+ keys_view.def("__contains__", &KeysView::contains);
763
+ }
764
+ // Similarly for ValuesView:
765
+ if (!detail::get_type_info(typeid(ValuesView))) {
766
+ class_<ValuesView> values_view(scope, "ValuesView", pybind11::module_local(local));
767
+ values_view.def("__len__", &ValuesView::len);
768
+ values_view.def("__iter__",
769
+ &ValuesView::iter,
770
+ keep_alive<0, 1>() /* Essential: keep view alive while iterator exists */
771
+ );
772
+ }
773
+ // Similarly for ItemsView:
774
+ if (!detail::get_type_info(typeid(ItemsView))) {
775
+ class_<ItemsView> items_view(scope, "ItemsView", pybind11::module_local(local));
776
+ items_view.def("__len__", &ItemsView::len);
777
+ items_view.def("__iter__",
778
+ &ItemsView::iter,
779
+ keep_alive<0, 1>() /* Essential: keep view alive while iterator exists */
780
+ );
781
+ }
782
+
783
+ cl.def(init<>());
784
+
785
+ // Register stream insertion operator (if possible)
786
+ detail::map_if_insertion_operator<Map, Class_>(cl, name);
787
+
788
+ cl.def(
789
+ "__bool__",
790
+ [](const Map &m) -> bool { return !m.empty(); },
791
+ "Check whether the map is nonempty");
792
+
793
+ cl.def(
794
+ "__iter__",
795
+ [](Map &m) { return make_key_iterator(m.begin(), m.end()); },
796
+ keep_alive<0, 1>() /* Essential: keep map alive while iterator exists */
797
+ );
798
+
799
+ cl.def(
800
+ "keys",
801
+ [](Map &m) { return std::unique_ptr<KeysView>(new detail::KeysViewImpl<Map>(m)); },
802
+ keep_alive<0, 1>() /* Essential: keep map alive while view exists */
803
+ );
804
+
805
+ cl.def(
806
+ "values",
807
+ [](Map &m) { return std::unique_ptr<ValuesView>(new detail::ValuesViewImpl<Map>(m)); },
808
+ keep_alive<0, 1>() /* Essential: keep map alive while view exists */
809
+ );
810
+
811
+ cl.def(
812
+ "items",
813
+ [](Map &m) { return std::unique_ptr<ItemsView>(new detail::ItemsViewImpl<Map>(m)); },
814
+ keep_alive<0, 1>() /* Essential: keep map alive while view exists */
815
+ );
816
+
817
+ cl.def(
818
+ "__getitem__",
819
+ [](Map &m, const KeyType &k) -> MappedType & {
820
+ auto it = m.find(k);
821
+ if (it == m.end()) {
822
+ set_error(PyExc_KeyError, detail::format_message_key_error(k));
823
+ throw error_already_set();
824
+ }
825
+ return it->second;
826
+ },
827
+ return_value_policy::reference_internal // ref + keepalive
828
+ );
829
+
830
+ cl.def("__contains__", [](Map &m, const KeyType &k) -> bool {
831
+ auto it = m.find(k);
832
+ if (it == m.end()) {
833
+ return false;
834
+ }
835
+ return true;
836
+ });
837
+ // Fallback for when the object is not of the key type
838
+ cl.def("__contains__", [](Map &, const object &) -> bool { return false; });
839
+
840
+ // Assignment provided only if the type is copyable
841
+ detail::map_assignment<Map, Class_>(cl);
842
+
843
+ cl.def("__delitem__", [](Map &m, const KeyType &k) {
844
+ auto it = m.find(k);
845
+ if (it == m.end()) {
846
+ set_error(PyExc_KeyError, detail::format_message_key_error(k));
847
+ throw error_already_set();
848
+ }
849
+ m.erase(it);
850
+ });
851
+
852
+ // Always use a lambda in case of `using` declaration
853
+ cl.def("__len__", [](const Map &m) { return m.size(); });
854
+
855
+ return cl;
856
+ }
857
+
858
+ PYBIND11_NAMESPACE_END(PYBIND11_NAMESPACE)