xcoll 0.4.0__py3-none-any.whl → 0.5.0__py3-none-any.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.

Potentially problematic release.


This version of xcoll might be problematic. Click here for more details.

Files changed (324) hide show
  1. xcoll/__init__.py +2 -1
  2. xcoll/beam_elements/__init__.py +7 -1
  3. xcoll/beam_elements/absorber.py +2 -2
  4. xcoll/beam_elements/base.py +105 -67
  5. xcoll/beam_elements/blowup.py +198 -0
  6. xcoll/beam_elements/{collimators_src → elements_src}/black_absorber.h +21 -3
  7. xcoll/beam_elements/{collimators_src → elements_src}/black_crystal.h +20 -2
  8. xcoll/beam_elements/elements_src/blowup.h +42 -0
  9. xcoll/beam_elements/elements_src/emittance_monitor.h +109 -0
  10. xcoll/beam_elements/{collimators_src → elements_src}/everest_block.h +19 -2
  11. xcoll/beam_elements/{collimators_src → elements_src}/everest_collimator.h +19 -3
  12. xcoll/beam_elements/{collimators_src → elements_src}/everest_crystal.h +30 -9
  13. xcoll/beam_elements/everest.py +5 -6
  14. xcoll/beam_elements/monitor.py +428 -0
  15. xcoll/colldb.py +103 -74
  16. xcoll/general.py +4 -4
  17. xcoll/initial_distribution.py +18 -6
  18. xcoll/install.py +3 -1
  19. xcoll/interaction_record/interaction_record.py +126 -80
  20. xcoll/interaction_record/interaction_record_src/interaction_record.h +43 -43
  21. xcoll/line_tools.py +8 -9
  22. xcoll/lossmap.py +48 -38
  23. xcoll/scattering_routines/everest/amorphous.h +4 -11
  24. xcoll/scattering_routines/everest/channeling.h +3 -8
  25. xcoll/scattering_routines/everest/everest.h +4 -1
  26. xcoll/scattering_routines/everest/jaw.h +4 -3
  27. xcoll/scattering_routines/everest/materials.py +35 -15
  28. xcoll/scattering_routines/everest/multiple_coulomb_scattering.h +2 -2
  29. xcoll/scattering_routines/everest/nuclear_interaction.h +1 -1
  30. xcoll/scattering_routines/everest/properties.h +6 -1
  31. xcoll/scattering_routines/fluka/flukaio/lib/libFlukaIO64.a +0 -0
  32. xcoll/scattering_routines/geant4/collimasim/.git +1 -0
  33. xcoll/scattering_routines/geant4/collimasim/.gitignore +12 -0
  34. xcoll/scattering_routines/geant4/collimasim/.gitmodules +3 -0
  35. xcoll/scattering_routines/geant4/collimasim/CMakeLists.txt +26 -0
  36. xcoll/scattering_routines/geant4/collimasim/README.md +21 -0
  37. xcoll/scattering_routines/geant4/collimasim/docs/Makefile +20 -0
  38. xcoll/scattering_routines/geant4/collimasim/docs/make.bat +35 -0
  39. xcoll/scattering_routines/geant4/collimasim/docs/source/collimasim.rst +10 -0
  40. xcoll/scattering_routines/geant4/collimasim/docs/source/conf.py +59 -0
  41. xcoll/scattering_routines/geant4/collimasim/docs/source/index.rst +26 -0
  42. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.appveyor.yml +37 -0
  43. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.clang-format +19 -0
  44. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.clang-tidy +65 -0
  45. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.cmake-format.yaml +73 -0
  46. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.git +1 -0
  47. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/CODEOWNERS +9 -0
  48. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/CONTRIBUTING.md +386 -0
  49. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/ISSUE_TEMPLATE/bug-report.yml +45 -0
  50. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/ISSUE_TEMPLATE/config.yml +8 -0
  51. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/dependabot.yml +16 -0
  52. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/labeler.yml +8 -0
  53. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/labeler_merged.yml +3 -0
  54. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/pull_request_template.md +19 -0
  55. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/workflows/ci.yml +969 -0
  56. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/workflows/configure.yml +84 -0
  57. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/workflows/format.yml +48 -0
  58. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/workflows/labeler.yml +16 -0
  59. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/workflows/pip.yml +103 -0
  60. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.gitignore +45 -0
  61. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.pre-commit-config.yaml +151 -0
  62. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.readthedocs.yml +3 -0
  63. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/CMakeLists.txt +297 -0
  64. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/LICENSE +29 -0
  65. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/MANIFEST.in +6 -0
  66. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/README.rst +180 -0
  67. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/Doxyfile +23 -0
  68. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/Makefile +192 -0
  69. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/_static/theme_overrides.css +11 -0
  70. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/chrono.rst +81 -0
  71. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/custom.rst +93 -0
  72. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/eigen.rst +310 -0
  73. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/functional.rst +109 -0
  74. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/index.rst +43 -0
  75. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/overview.rst +171 -0
  76. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/stl.rst +251 -0
  77. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/strings.rst +305 -0
  78. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/classes.rst +1297 -0
  79. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/embedding.rst +262 -0
  80. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/exceptions.rst +396 -0
  81. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/functions.rst +568 -0
  82. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/misc.rst +337 -0
  83. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/pycpp/index.rst +13 -0
  84. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/pycpp/numpy.rst +463 -0
  85. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/pycpp/object.rst +286 -0
  86. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/pycpp/utilities.rst +155 -0
  87. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/smart_ptrs.rst +174 -0
  88. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/basics.rst +308 -0
  89. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/benchmark.py +91 -0
  90. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/benchmark.rst +95 -0
  91. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/changelog.rst +2050 -0
  92. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/classes.rst +542 -0
  93. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/cmake/index.rst +8 -0
  94. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/compiling.rst +648 -0
  95. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/conf.py +381 -0
  96. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/faq.rst +343 -0
  97. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/index.rst +48 -0
  98. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/installing.rst +105 -0
  99. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/limitations.rst +72 -0
  100. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/pybind11-logo.png +0 -0
  101. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/pybind11_vs_boost_python1.png +0 -0
  102. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/pybind11_vs_boost_python1.svg +427 -0
  103. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/pybind11_vs_boost_python2.png +0 -0
  104. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/pybind11_vs_boost_python2.svg +427 -0
  105. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/reference.rst +130 -0
  106. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/release.rst +96 -0
  107. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/requirements.txt +8 -0
  108. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/upgrade.rst +548 -0
  109. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/attr.h +605 -0
  110. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/buffer_info.h +144 -0
  111. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/cast.h +1432 -0
  112. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/chrono.h +213 -0
  113. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/common.h +2 -0
  114. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/complex.h +65 -0
  115. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/class.h +709 -0
  116. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/common.h +1021 -0
  117. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/descr.h +104 -0
  118. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/init.h +346 -0
  119. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/internals.h +467 -0
  120. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/type_caster_base.h +978 -0
  121. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/typeid.h +55 -0
  122. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/eigen.h +606 -0
  123. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/embed.h +284 -0
  124. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/eval.h +163 -0
  125. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/functional.h +121 -0
  126. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/gil.h +193 -0
  127. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/iostream.h +275 -0
  128. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/numpy.h +1741 -0
  129. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/operators.h +163 -0
  130. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/options.h +65 -0
  131. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/pybind11.h +2497 -0
  132. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/pytypes.h +1879 -0
  133. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/stl/filesystem.h +103 -0
  134. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/stl.h +375 -0
  135. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/stl_bind.h +747 -0
  136. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/noxfile.py +88 -0
  137. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/__init__.py +11 -0
  138. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/__main__.py +52 -0
  139. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/_version.py +12 -0
  140. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/_version.pyi +6 -0
  141. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/commands.py +21 -0
  142. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/py.typed +0 -0
  143. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/setup_helpers.py +482 -0
  144. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/setup_helpers.pyi +63 -0
  145. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pyproject.toml +41 -0
  146. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/setup.cfg +56 -0
  147. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/setup.py +155 -0
  148. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/CMakeLists.txt +503 -0
  149. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/conftest.py +208 -0
  150. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/constructor_stats.h +275 -0
  151. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/cross_module_gil_utils.cpp +73 -0
  152. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/env.py +33 -0
  153. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/extra_python_package/pytest.ini +0 -0
  154. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/extra_python_package/test_files.py +279 -0
  155. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/extra_setuptools/pytest.ini +0 -0
  156. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/extra_setuptools/test_setuphelper.py +143 -0
  157. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/local_bindings.h +85 -0
  158. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/object.h +179 -0
  159. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/pybind11_cross_module_tests.cpp +151 -0
  160. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/pybind11_tests.cpp +91 -0
  161. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/pybind11_tests.h +85 -0
  162. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/pytest.ini +19 -0
  163. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/requirements.txt +12 -0
  164. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_async.cpp +26 -0
  165. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_async.py +25 -0
  166. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_buffers.cpp +216 -0
  167. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_buffers.py +163 -0
  168. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_builtin_casters.cpp +286 -0
  169. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_builtin_casters.py +536 -0
  170. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_call_policies.cpp +107 -0
  171. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_call_policies.py +248 -0
  172. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_callbacks.cpp +227 -0
  173. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_callbacks.py +202 -0
  174. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_chrono.cpp +84 -0
  175. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_chrono.py +210 -0
  176. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_class.cpp +550 -0
  177. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_class.py +473 -0
  178. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/CMakeLists.txt +84 -0
  179. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/embed.cpp +21 -0
  180. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/installed_embed/CMakeLists.txt +28 -0
  181. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/installed_function/CMakeLists.txt +39 -0
  182. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/installed_target/CMakeLists.txt +46 -0
  183. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/main.cpp +6 -0
  184. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/subdirectory_embed/CMakeLists.txt +41 -0
  185. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/subdirectory_function/CMakeLists.txt +35 -0
  186. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/subdirectory_target/CMakeLists.txt +41 -0
  187. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/test.py +10 -0
  188. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_constants_and_functions.cpp +165 -0
  189. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_constants_and_functions.py +53 -0
  190. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_copy_move.cpp +238 -0
  191. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_copy_move.py +126 -0
  192. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_custom_type_casters.cpp +141 -0
  193. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_custom_type_casters.py +117 -0
  194. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_custom_type_setup.cpp +41 -0
  195. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_custom_type_setup.py +50 -0
  196. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_docstring_options.cpp +69 -0
  197. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_docstring_options.py +42 -0
  198. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_eigen.cpp +348 -0
  199. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_eigen.py +771 -0
  200. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_embed/CMakeLists.txt +47 -0
  201. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_embed/catch.cpp +22 -0
  202. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_embed/external_module.cpp +23 -0
  203. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_embed/test_interpreter.cpp +326 -0
  204. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_embed/test_interpreter.py +15 -0
  205. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_enum.cpp +148 -0
  206. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_enum.py +272 -0
  207. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_eval.cpp +119 -0
  208. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_eval.py +51 -0
  209. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_eval_call.py +5 -0
  210. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_exceptions.cpp +285 -0
  211. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_exceptions.h +12 -0
  212. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_exceptions.py +265 -0
  213. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_factory_constructors.cpp +397 -0
  214. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_factory_constructors.py +520 -0
  215. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_gil_scoped.cpp +49 -0
  216. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_gil_scoped.py +94 -0
  217. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_iostream.cpp +125 -0
  218. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_iostream.py +331 -0
  219. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_kwargs_and_defaults.cpp +153 -0
  220. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_kwargs_and_defaults.py +284 -0
  221. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_local_bindings.cpp +107 -0
  222. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_local_bindings.py +257 -0
  223. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_methods_and_attributes.cpp +412 -0
  224. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_methods_and_attributes.py +517 -0
  225. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_modules.cpp +102 -0
  226. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_modules.py +92 -0
  227. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_multiple_inheritance.cpp +233 -0
  228. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_multiple_inheritance.py +360 -0
  229. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_array.cpp +472 -0
  230. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_array.py +593 -0
  231. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_dtypes.cpp +524 -0
  232. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_dtypes.py +441 -0
  233. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_vectorize.cpp +103 -0
  234. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_vectorize.py +267 -0
  235. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_opaque_types.cpp +73 -0
  236. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_opaque_types.py +59 -0
  237. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_operator_overloading.cpp +235 -0
  238. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_operator_overloading.py +146 -0
  239. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_pickling.cpp +189 -0
  240. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_pickling.py +82 -0
  241. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_pytypes.cpp +560 -0
  242. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_pytypes.py +651 -0
  243. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_sequences_and_iterators.cpp +500 -0
  244. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_sequences_and_iterators.py +253 -0
  245. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_smart_ptr.cpp +452 -0
  246. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_smart_ptr.py +318 -0
  247. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_stl.cpp +342 -0
  248. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_stl.py +291 -0
  249. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_stl_binders.cpp +131 -0
  250. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_stl_binders.py +318 -0
  251. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_tagbased_polymorphic.cpp +144 -0
  252. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_tagbased_polymorphic.py +29 -0
  253. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_thread.cpp +66 -0
  254. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_thread.py +44 -0
  255. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_union.cpp +22 -0
  256. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_union.py +9 -0
  257. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_virtual_functions.cpp +510 -0
  258. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_virtual_functions.py +408 -0
  259. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/valgrind-numpy-scipy.supp +140 -0
  260. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/valgrind-python.supp +117 -0
  261. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/FindCatch.cmake +70 -0
  262. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/FindEigen3.cmake +86 -0
  263. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/FindPythonLibsNew.cmake +257 -0
  264. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/check-style.sh +44 -0
  265. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/cmake_uninstall.cmake.in +23 -0
  266. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/libsize.py +39 -0
  267. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/make_changelog.py +64 -0
  268. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/pybind11Common.cmake +402 -0
  269. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/pybind11Config.cmake.in +233 -0
  270. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/pybind11NewTools.cmake +276 -0
  271. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/pybind11Tools.cmake +214 -0
  272. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/pyproject.toml +3 -0
  273. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/setup_global.py.in +65 -0
  274. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/setup_main.py.in +41 -0
  275. xcoll/scattering_routines/geant4/collimasim/pyproject.toml +8 -0
  276. xcoll/scattering_routines/geant4/collimasim/setup.py +144 -0
  277. xcoll/scattering_routines/geant4/collimasim/src/collimasim/BDSPyATInterface.cpp +403 -0
  278. xcoll/scattering_routines/geant4/collimasim/src/collimasim/BDSPyATInterface.hh +100 -0
  279. xcoll/scattering_routines/geant4/collimasim/src/collimasim/BDSXtrackInterface.cpp +763 -0
  280. xcoll/scattering_routines/geant4/collimasim/src/collimasim/BDSXtrackInterface.hh +118 -0
  281. xcoll/scattering_routines/geant4/collimasim/src/collimasim/__init__.py +8 -0
  282. xcoll/scattering_routines/geant4/collimasim/src/collimasim/bindings.cpp +63 -0
  283. xcoll/scattering_routines/geant4/collimasim/src/collimasim/pyCollimatorPass.py +142 -0
  284. xcoll/scattering_routines/geant4/collimasim/src/collimasim/xtrack_collimator.py +556 -0
  285. xcoll/scattering_routines/geant4/collimasim/src/collimasim.egg-info/PKG-INFO +6 -0
  286. xcoll/scattering_routines/geant4/collimasim/src/collimasim.egg-info/SOURCES.txt +24 -0
  287. xcoll/scattering_routines/geant4/collimasim/src/collimasim.egg-info/dependency_links.txt +1 -0
  288. xcoll/scattering_routines/geant4/collimasim/src/collimasim.egg-info/not-zip-safe +1 -0
  289. xcoll/scattering_routines/geant4/collimasim/src/collimasim.egg-info/top_level.txt +1 -0
  290. xcoll/scattering_routines/geant4/collimasim/tests/README.md +25 -0
  291. xcoll/scattering_routines/geant4/collimasim/tests/resources/CollDB_forions.dat +25 -0
  292. xcoll/scattering_routines/geant4/collimasim/tests/resources/CollDB_new_example.dat +18 -0
  293. xcoll/scattering_routines/geant4/collimasim/tests/resources/CollDB_old_example.dat +68 -0
  294. xcoll/scattering_routines/geant4/collimasim/tests/resources/CollDB_testing.dat +15 -0
  295. xcoll/scattering_routines/geant4/collimasim/tests/resources/CollDB_yaml_example.yaml +110 -0
  296. xcoll/scattering_routines/geant4/collimasim/tests/resources/collgaps.dat +7 -0
  297. xcoll/scattering_routines/geant4/collimasim/tests/resources/collgaps_pyat_test.dat +3 -0
  298. xcoll/scattering_routines/geant4/collimasim/tests/resources/collonly_twiss_file_example.tfs +54 -0
  299. xcoll/scattering_routines/geant4/collimasim/tests/resources/settings.gmad +3 -0
  300. xcoll/scattering_routines/geant4/collimasim/tests/resources/settings_black_absorber.gmad +3 -0
  301. xcoll/scattering_routines/geant4/collimasim/tests/resources/settings_ions.gmad +5 -0
  302. xcoll/scattering_routines/geant4/collimasim/tests/resources/twiss_file_testing.tfs +51 -0
  303. xcoll/scattering_routines/geant4/collimasim/tests/test_pyat.py +65 -0
  304. xcoll/scattering_routines/geant4/collimasim/tests/test_pyat_passmethod.py +59 -0
  305. xcoll/scattering_routines/geant4/collimasim/tests/test_pyat_tracking.py +102 -0
  306. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack.py +75 -0
  307. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_angle.py +74 -0
  308. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_colldb_load.py +84 -0
  309. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_interaction.py +159 -0
  310. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_interaction_ion.py +99 -0
  311. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_ions.py +78 -0
  312. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_lost_energy.py +88 -0
  313. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_tilt.py +80 -0
  314. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_tracking.py +97 -0
  315. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_tracking_ions.py +96 -0
  316. xcoll/scattering_routines/geometry/collimator_geometry.h +9 -10
  317. xcoll/scattering_routines/geometry/crystal_geometry.h +9 -6
  318. {xcoll-0.4.0.dist-info → xcoll-0.5.0.dist-info}/METADATA +1 -1
  319. xcoll-0.5.0.dist-info/RECORD +413 -0
  320. xcoll/scattering_routines/fluka/build_fluka_input.py +0 -58
  321. xcoll-0.4.0.dist-info/RECORD +0 -126
  322. {xcoll-0.4.0.dist-info → xcoll-0.5.0.dist-info}/LICENSE +0 -0
  323. {xcoll-0.4.0.dist-info → xcoll-0.5.0.dist-info}/NOTICE +0 -0
  324. {xcoll-0.4.0.dist-info → xcoll-0.5.0.dist-info}/WHEEL +0 -0
@@ -0,0 +1,85 @@
1
+ #pragma once
2
+
3
+ #include <pybind11/pybind11.h>
4
+ #include <pybind11/eval.h>
5
+
6
+ #if defined(_MSC_VER) && _MSC_VER < 1910
7
+ // We get some really long type names here which causes MSVC 2015 to emit warnings
8
+ # pragma warning( \
9
+ disable : 4503) // warning C4503: decorated name length exceeded, name was truncated
10
+ #endif
11
+
12
+ namespace py = pybind11;
13
+ using namespace pybind11::literals;
14
+
15
+ class test_initializer {
16
+ using Initializer = void (*)(py::module_ &);
17
+
18
+ public:
19
+ explicit test_initializer(Initializer init);
20
+ test_initializer(const char *submodule_name, Initializer init);
21
+ };
22
+
23
+ #define TEST_SUBMODULE(name, variable) \
24
+ void test_submodule_##name(py::module_ &); \
25
+ test_initializer name(#name, test_submodule_##name); \
26
+ void test_submodule_##name(py::module_ &(variable))
27
+
28
+ /// Dummy type which is not exported anywhere -- something to trigger a conversion error
29
+ struct UnregisteredType { };
30
+
31
+ /// A user-defined type which is exported and can be used by any test
32
+ class UserType {
33
+ public:
34
+ UserType() = default;
35
+ explicit UserType(int i) : i(i) { }
36
+
37
+ int value() const { return i; }
38
+ void set(int set) { i = set; }
39
+
40
+ private:
41
+ int i = -1;
42
+ };
43
+
44
+ /// Like UserType, but increments `value` on copy for quick reference vs. copy tests
45
+ class IncType : public UserType {
46
+ public:
47
+ using UserType::UserType;
48
+ IncType() = default;
49
+ IncType(const IncType &other) : IncType(other.value() + 1) { }
50
+ IncType(IncType &&) = delete;
51
+ IncType &operator=(const IncType &) = delete;
52
+ IncType &operator=(IncType &&) = delete;
53
+ };
54
+
55
+ /// A simple union for basic testing
56
+ union IntFloat {
57
+ int i;
58
+ float f;
59
+ };
60
+
61
+ /// Custom cast-only type that casts to a string "rvalue" or "lvalue" depending on the cast context.
62
+ /// Used to test recursive casters (e.g. std::tuple, stl containers).
63
+ struct RValueCaster {};
64
+ PYBIND11_NAMESPACE_BEGIN(pybind11)
65
+ PYBIND11_NAMESPACE_BEGIN(detail)
66
+ template<> class type_caster<RValueCaster> {
67
+ public:
68
+ PYBIND11_TYPE_CASTER(RValueCaster, _("RValueCaster"));
69
+ static handle cast(RValueCaster &&, return_value_policy, handle) { return py::str("rvalue").release(); }
70
+ static handle cast(const RValueCaster &, return_value_policy, handle) { return py::str("lvalue").release(); }
71
+ };
72
+ PYBIND11_NAMESPACE_END(detail)
73
+ PYBIND11_NAMESPACE_END(pybind11)
74
+
75
+ template <typename F>
76
+ void ignoreOldStyleInitWarnings(F &&body) {
77
+ py::exec(R"(
78
+ message = "pybind11-bound class '.+' is using an old-style placement-new '(?:__init__|__setstate__)' which has been deprecated"
79
+
80
+ import warnings
81
+ with warnings.catch_warnings():
82
+ warnings.filterwarnings("ignore", message=message, category=FutureWarning)
83
+ body()
84
+ )", py::dict(py::arg("body") = py::cpp_function(body)));
85
+ }
@@ -0,0 +1,19 @@
1
+ [pytest]
2
+ minversion = 3.1
3
+ norecursedirs = test_* extra_*
4
+ xfail_strict = True
5
+ addopts =
6
+ # show summary of skipped tests
7
+ -rs
8
+ # capture only Python print and C++ py::print, but not C output (low-level Python errors)
9
+ --capture=sys
10
+ filterwarnings =
11
+ # make warnings into errors but ignore certain third-party extension issues
12
+ error
13
+ # somehow, some DeprecationWarnings do not get turned into errors
14
+ always::DeprecationWarning
15
+ # importing scipy submodules on some version of Python
16
+ ignore::ImportWarning
17
+ # bogus numpy ABI warning (see numpy/#432)
18
+ ignore:.*numpy.dtype size changed.*:RuntimeWarning
19
+ ignore:.*numpy.ufunc size changed.*:RuntimeWarning
@@ -0,0 +1,12 @@
1
+ --extra-index-url https://antocuni.github.io/pypy-wheels/manylinux2010/
2
+ numpy==1.16.6; python_version<"3.6" and sys_platform!="win32"
3
+ numpy==1.18.0; platform_python_implementation=="PyPy" and sys_platform=="darwin" and python_version>="3.6"
4
+ numpy==1.19.3; (platform_python_implementation!="PyPy" or sys_platform=="linux") and python_version=="3.6"
5
+ numpy==1.21.2; (platform_python_implementation!="PyPy" or sys_platform=="linux") and python_version>="3.7" and python_version<"3.10"
6
+ numpy==1.21.2; platform_python_implementation!="PyPy" and sys_platform=="linux" and python_version=="3.10"
7
+ pytest==4.6.9; python_version<"3.5"
8
+ pytest==6.1.2; python_version=="3.5"
9
+ pytest==6.2.4; python_version>="3.6"
10
+ pytest-timeout
11
+ scipy==1.2.3; (platform_python_implementation!="PyPy" or sys_platform=="linux") and python_version<"3.6"
12
+ scipy==1.5.4; (platform_python_implementation!="PyPy" or sys_platform=="linux") and python_version>="3.6" and python_version<"3.10"
@@ -0,0 +1,26 @@
1
+ /*
2
+ tests/test_async.cpp -- __await__ support
3
+
4
+ Copyright (c) 2019 Google Inc.
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
+ #include "pybind11_tests.h"
11
+
12
+ TEST_SUBMODULE(async_module, m) {
13
+ struct DoesNotSupportAsync {};
14
+ py::class_<DoesNotSupportAsync>(m, "DoesNotSupportAsync")
15
+ .def(py::init<>());
16
+ struct SupportsAsync {};
17
+ py::class_<SupportsAsync>(m, "SupportsAsync")
18
+ .def(py::init<>())
19
+ .def("__await__", [](const SupportsAsync& self) -> py::object {
20
+ static_cast<void>(self);
21
+ py::object loop = py::module_::import("asyncio.events").attr("get_event_loop")();
22
+ py::object f = loop.attr("create_future")();
23
+ f.attr("set_result")(5);
24
+ return f.attr("__await__")();
25
+ });
26
+ }
@@ -0,0 +1,25 @@
1
+ # -*- coding: utf-8 -*-
2
+ import pytest
3
+
4
+ asyncio = pytest.importorskip("asyncio")
5
+ m = pytest.importorskip("pybind11_tests.async_module")
6
+
7
+
8
+ @pytest.fixture
9
+ def event_loop():
10
+ loop = asyncio.new_event_loop()
11
+ yield loop
12
+ loop.close()
13
+
14
+
15
+ async def get_await_result(x):
16
+ return await x
17
+
18
+
19
+ def test_await(event_loop):
20
+ assert 5 == event_loop.run_until_complete(get_await_result(m.SupportsAsync()))
21
+
22
+
23
+ def test_await_missing(event_loop):
24
+ with pytest.raises(TypeError):
25
+ event_loop.run_until_complete(get_await_result(m.DoesNotSupportAsync()))
@@ -0,0 +1,216 @@
1
+ /*
2
+ tests/test_buffers.cpp -- supporting Pythons' buffer protocol
3
+
4
+ Copyright (c) 2016 Wenzel Jakob <wenzel.jakob@epfl.ch>
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
+ #include "pybind11_tests.h"
11
+ #include "constructor_stats.h"
12
+ #include <pybind11/stl.h>
13
+
14
+ TEST_SUBMODULE(buffers, m) {
15
+ // test_from_python / test_to_python:
16
+ class Matrix {
17
+ public:
18
+ Matrix(py::ssize_t rows, py::ssize_t cols) : m_rows(rows), m_cols(cols) {
19
+ print_created(this, std::to_string(m_rows) + "x" + std::to_string(m_cols) + " matrix");
20
+ m_data = new float[(size_t) (rows*cols)];
21
+ memset(m_data, 0, sizeof(float) * (size_t) (rows * cols));
22
+ }
23
+
24
+ Matrix(const Matrix &s) : m_rows(s.m_rows), m_cols(s.m_cols) {
25
+ print_copy_created(this, std::to_string(m_rows) + "x" + std::to_string(m_cols) + " matrix");
26
+ m_data = new float[(size_t) (m_rows * m_cols)];
27
+ memcpy(m_data, s.m_data, sizeof(float) * (size_t) (m_rows * m_cols));
28
+ }
29
+
30
+ Matrix(Matrix &&s) noexcept : m_rows(s.m_rows), m_cols(s.m_cols), m_data(s.m_data) {
31
+ print_move_created(this);
32
+ s.m_rows = 0;
33
+ s.m_cols = 0;
34
+ s.m_data = nullptr;
35
+ }
36
+
37
+ ~Matrix() {
38
+ print_destroyed(this, std::to_string(m_rows) + "x" + std::to_string(m_cols) + " matrix");
39
+ delete[] m_data;
40
+ }
41
+
42
+ Matrix &operator=(const Matrix &s) {
43
+ if (this == &s) {
44
+ return *this;
45
+ }
46
+ print_copy_assigned(this,
47
+ std::to_string(m_rows) + "x" + std::to_string(m_cols) + " matrix");
48
+ delete[] m_data;
49
+ m_rows = s.m_rows;
50
+ m_cols = s.m_cols;
51
+ m_data = new float[(size_t) (m_rows * m_cols)];
52
+ memcpy(m_data, s.m_data, sizeof(float) * (size_t) (m_rows * m_cols));
53
+ return *this;
54
+ }
55
+
56
+ Matrix &operator=(Matrix &&s) noexcept {
57
+ print_move_assigned(this, std::to_string(m_rows) + "x" + std::to_string(m_cols) + " matrix");
58
+ if (&s != this) {
59
+ delete[] m_data;
60
+ m_rows = s.m_rows; m_cols = s.m_cols; m_data = s.m_data;
61
+ s.m_rows = 0; s.m_cols = 0; s.m_data = nullptr;
62
+ }
63
+ return *this;
64
+ }
65
+
66
+ float operator()(py::ssize_t i, py::ssize_t j) const {
67
+ return m_data[(size_t) (i*m_cols + j)];
68
+ }
69
+
70
+ float &operator()(py::ssize_t i, py::ssize_t j) {
71
+ return m_data[(size_t) (i*m_cols + j)];
72
+ }
73
+
74
+ float *data() { return m_data; }
75
+
76
+ py::ssize_t rows() const { return m_rows; }
77
+ py::ssize_t cols() const { return m_cols; }
78
+ private:
79
+ py::ssize_t m_rows;
80
+ py::ssize_t m_cols;
81
+ float *m_data;
82
+ };
83
+ py::class_<Matrix>(m, "Matrix", py::buffer_protocol())
84
+ .def(py::init<py::ssize_t, py::ssize_t>())
85
+ /// Construct from a buffer
86
+ .def(py::init([](const py::buffer &b) {
87
+ py::buffer_info info = b.request();
88
+ if (info.format != py::format_descriptor<float>::format() || info.ndim != 2)
89
+ throw std::runtime_error("Incompatible buffer format!");
90
+
91
+ auto v = new Matrix(info.shape[0], info.shape[1]);
92
+ memcpy(v->data(), info.ptr, sizeof(float) * (size_t) (v->rows() * v->cols()));
93
+ return v;
94
+ }))
95
+
96
+ .def("rows", &Matrix::rows)
97
+ .def("cols", &Matrix::cols)
98
+
99
+ /// Bare bones interface
100
+ .def("__getitem__",
101
+ [](const Matrix &m, std::pair<py::ssize_t, py::ssize_t> i) {
102
+ if (i.first >= m.rows() || i.second >= m.cols())
103
+ throw py::index_error();
104
+ return m(i.first, i.second);
105
+ })
106
+ .def("__setitem__",
107
+ [](Matrix &m, std::pair<py::ssize_t, py::ssize_t> i, float v) {
108
+ if (i.first >= m.rows() || i.second >= m.cols())
109
+ throw py::index_error();
110
+ m(i.first, i.second) = v;
111
+ })
112
+ /// Provide buffer access
113
+ .def_buffer([](Matrix &m) -> py::buffer_info {
114
+ return py::buffer_info(
115
+ m.data(), /* Pointer to buffer */
116
+ { m.rows(), m.cols() }, /* Buffer dimensions */
117
+ { sizeof(float) * size_t(m.cols()), /* Strides (in bytes) for each index */
118
+ sizeof(float) }
119
+ );
120
+ });
121
+
122
+ // test_inherited_protocol
123
+ class SquareMatrix : public Matrix {
124
+ public:
125
+ explicit SquareMatrix(py::ssize_t n) : Matrix(n, n) {}
126
+ };
127
+ // Derived classes inherit the buffer protocol and the buffer access function
128
+ py::class_<SquareMatrix, Matrix>(m, "SquareMatrix")
129
+ .def(py::init<py::ssize_t>());
130
+
131
+
132
+ // test_pointer_to_member_fn
133
+ // Tests that passing a pointer to member to the base class works in
134
+ // the derived class.
135
+ struct Buffer {
136
+ int32_t value = 0;
137
+
138
+ py::buffer_info get_buffer_info() {
139
+ return py::buffer_info(&value, sizeof(value),
140
+ py::format_descriptor<int32_t>::format(), 1);
141
+ }
142
+ };
143
+ py::class_<Buffer>(m, "Buffer", py::buffer_protocol())
144
+ .def(py::init<>())
145
+ .def_readwrite("value", &Buffer::value)
146
+ .def_buffer(&Buffer::get_buffer_info);
147
+
148
+
149
+ class ConstBuffer {
150
+ std::unique_ptr<int32_t> value;
151
+
152
+ public:
153
+ int32_t get_value() const { return *value; }
154
+ void set_value(int32_t v) { *value = v; }
155
+
156
+ py::buffer_info get_buffer_info() const {
157
+ return py::buffer_info(value.get(), sizeof(*value),
158
+ py::format_descriptor<int32_t>::format(), 1);
159
+ }
160
+
161
+ ConstBuffer() : value(new int32_t{0}) {}
162
+ };
163
+ py::class_<ConstBuffer>(m, "ConstBuffer", py::buffer_protocol())
164
+ .def(py::init<>())
165
+ .def_property("value", &ConstBuffer::get_value, &ConstBuffer::set_value)
166
+ .def_buffer(&ConstBuffer::get_buffer_info);
167
+
168
+ struct DerivedBuffer : public Buffer { };
169
+ py::class_<DerivedBuffer>(m, "DerivedBuffer", py::buffer_protocol())
170
+ .def(py::init<>())
171
+ .def_readwrite("value", (int32_t DerivedBuffer::*) &DerivedBuffer::value)
172
+ .def_buffer(&DerivedBuffer::get_buffer_info);
173
+
174
+ struct BufferReadOnly {
175
+ const uint8_t value = 0;
176
+ explicit BufferReadOnly(uint8_t value) : value(value) {}
177
+
178
+ py::buffer_info get_buffer_info() {
179
+ return py::buffer_info(&value, 1);
180
+ }
181
+ };
182
+ py::class_<BufferReadOnly>(m, "BufferReadOnly", py::buffer_protocol())
183
+ .def(py::init<uint8_t>())
184
+ .def_buffer(&BufferReadOnly::get_buffer_info);
185
+
186
+ struct BufferReadOnlySelect {
187
+ uint8_t value = 0;
188
+ bool readonly = false;
189
+
190
+ py::buffer_info get_buffer_info() {
191
+ return py::buffer_info(&value, 1, readonly);
192
+ }
193
+ };
194
+ py::class_<BufferReadOnlySelect>(m, "BufferReadOnlySelect", py::buffer_protocol())
195
+ .def(py::init<>())
196
+ .def_readwrite("value", &BufferReadOnlySelect::value)
197
+ .def_readwrite("readonly", &BufferReadOnlySelect::readonly)
198
+ .def_buffer(&BufferReadOnlySelect::get_buffer_info);
199
+
200
+ // Expose buffer_info for testing.
201
+ py::class_<py::buffer_info>(m, "buffer_info")
202
+ .def(py::init<>())
203
+ .def_readonly("itemsize", &py::buffer_info::itemsize)
204
+ .def_readonly("size", &py::buffer_info::size)
205
+ .def_readonly("format", &py::buffer_info::format)
206
+ .def_readonly("ndim", &py::buffer_info::ndim)
207
+ .def_readonly("shape", &py::buffer_info::shape)
208
+ .def_readonly("strides", &py::buffer_info::strides)
209
+ .def_readonly("readonly", &py::buffer_info::readonly)
210
+ .def("__repr__", [](py::handle self) {
211
+ return py::str("itemsize={0.itemsize!r}, size={0.size!r}, format={0.format!r}, ndim={0.ndim!r}, shape={0.shape!r}, strides={0.strides!r}, readonly={0.readonly!r}").format(self);
212
+ })
213
+ ;
214
+
215
+ m.def("get_buffer_info", [](const py::buffer &buffer) { return buffer.request(); });
216
+ }
@@ -0,0 +1,163 @@
1
+ # -*- coding: utf-8 -*-
2
+ import ctypes
3
+ import io
4
+ import struct
5
+
6
+ import pytest
7
+
8
+ import env
9
+ from pybind11_tests import ConstructorStats
10
+ from pybind11_tests import buffers as m
11
+
12
+ np = pytest.importorskip("numpy")
13
+
14
+
15
+ def test_from_python():
16
+ with pytest.raises(RuntimeError) as excinfo:
17
+ m.Matrix(np.array([1, 2, 3])) # trying to assign a 1D array
18
+ assert str(excinfo.value) == "Incompatible buffer format!"
19
+
20
+ m3 = np.array([[1, 2, 3], [4, 5, 6]]).astype(np.float32)
21
+ m4 = m.Matrix(m3)
22
+
23
+ for i in range(m4.rows()):
24
+ for j in range(m4.cols()):
25
+ assert m3[i, j] == m4[i, j]
26
+
27
+ cstats = ConstructorStats.get(m.Matrix)
28
+ assert cstats.alive() == 1
29
+ del m3, m4
30
+ assert cstats.alive() == 0
31
+ assert cstats.values() == ["2x3 matrix"]
32
+ assert cstats.copy_constructions == 0
33
+ # assert cstats.move_constructions >= 0 # Don't invoke any
34
+ assert cstats.copy_assignments == 0
35
+ assert cstats.move_assignments == 0
36
+
37
+
38
+ # https://foss.heptapod.net/pypy/pypy/-/issues/2444
39
+ def test_to_python():
40
+ mat = m.Matrix(5, 4)
41
+ assert memoryview(mat).shape == (5, 4)
42
+
43
+ assert mat[2, 3] == 0
44
+ mat[2, 3] = 4.0
45
+ mat[3, 2] = 7.0
46
+ assert mat[2, 3] == 4
47
+ assert mat[3, 2] == 7
48
+ assert struct.unpack_from("f", mat, (3 * 4 + 2) * 4) == (7,)
49
+ assert struct.unpack_from("f", mat, (2 * 4 + 3) * 4) == (4,)
50
+
51
+ mat2 = np.array(mat, copy=False)
52
+ assert mat2.shape == (5, 4)
53
+ assert abs(mat2).sum() == 11
54
+ assert mat2[2, 3] == 4 and mat2[3, 2] == 7
55
+ mat2[2, 3] = 5
56
+ assert mat2[2, 3] == 5
57
+
58
+ cstats = ConstructorStats.get(m.Matrix)
59
+ assert cstats.alive() == 1
60
+ del mat
61
+ pytest.gc_collect()
62
+ assert cstats.alive() == 1
63
+ del mat2 # holds a mat reference
64
+ pytest.gc_collect()
65
+ assert cstats.alive() == 0
66
+ assert cstats.values() == ["5x4 matrix"]
67
+ assert cstats.copy_constructions == 0
68
+ # assert cstats.move_constructions >= 0 # Don't invoke any
69
+ assert cstats.copy_assignments == 0
70
+ assert cstats.move_assignments == 0
71
+
72
+
73
+ def test_inherited_protocol():
74
+ """SquareMatrix is derived from Matrix and inherits the buffer protocol"""
75
+
76
+ matrix = m.SquareMatrix(5)
77
+ assert memoryview(matrix).shape == (5, 5)
78
+ assert np.asarray(matrix).shape == (5, 5)
79
+
80
+
81
+ def test_pointer_to_member_fn():
82
+ for cls in [m.Buffer, m.ConstBuffer, m.DerivedBuffer]:
83
+ buf = cls()
84
+ buf.value = 0x12345678
85
+ value = struct.unpack("i", bytearray(buf))[0]
86
+ assert value == 0x12345678
87
+
88
+
89
+ def test_readonly_buffer():
90
+ buf = m.BufferReadOnly(0x64)
91
+ view = memoryview(buf)
92
+ assert view[0] == b"d" if env.PY2 else 0x64
93
+ assert view.readonly
94
+ with pytest.raises(TypeError):
95
+ view[0] = b"\0" if env.PY2 else 0
96
+
97
+
98
+ def test_selective_readonly_buffer():
99
+ buf = m.BufferReadOnlySelect()
100
+
101
+ memoryview(buf)[0] = b"d" if env.PY2 else 0x64
102
+ assert buf.value == 0x64
103
+
104
+ io.BytesIO(b"A").readinto(buf)
105
+ assert buf.value == ord(b"A")
106
+
107
+ buf.readonly = True
108
+ with pytest.raises(TypeError):
109
+ memoryview(buf)[0] = b"\0" if env.PY2 else 0
110
+ with pytest.raises(TypeError):
111
+ io.BytesIO(b"1").readinto(buf)
112
+
113
+
114
+ def test_ctypes_array_1d():
115
+ char1d = (ctypes.c_char * 10)()
116
+ int1d = (ctypes.c_int * 15)()
117
+ long1d = (ctypes.c_long * 7)()
118
+
119
+ for carray in (char1d, int1d, long1d):
120
+ info = m.get_buffer_info(carray)
121
+ assert info.itemsize == ctypes.sizeof(carray._type_)
122
+ assert info.size == len(carray)
123
+ assert info.ndim == 1
124
+ assert info.shape == [info.size]
125
+ assert info.strides == [info.itemsize]
126
+ assert not info.readonly
127
+
128
+
129
+ def test_ctypes_array_2d():
130
+ char2d = ((ctypes.c_char * 10) * 4)()
131
+ int2d = ((ctypes.c_int * 15) * 3)()
132
+ long2d = ((ctypes.c_long * 7) * 2)()
133
+
134
+ for carray in (char2d, int2d, long2d):
135
+ info = m.get_buffer_info(carray)
136
+ assert info.itemsize == ctypes.sizeof(carray[0]._type_)
137
+ assert info.size == len(carray) * len(carray[0])
138
+ assert info.ndim == 2
139
+ assert info.shape == [len(carray), len(carray[0])]
140
+ assert info.strides == [info.itemsize * len(carray[0]), info.itemsize]
141
+ assert not info.readonly
142
+
143
+
144
+ @pytest.mark.skipif(
145
+ "env.PYPY and env.PY2", reason="PyPy2 bytes buffer not reported as readonly"
146
+ )
147
+ def test_ctypes_from_buffer():
148
+ test_pystr = b"0123456789"
149
+ for pyarray in (test_pystr, bytearray(test_pystr)):
150
+ pyinfo = m.get_buffer_info(pyarray)
151
+
152
+ if pyinfo.readonly:
153
+ cbytes = (ctypes.c_char * len(pyarray)).from_buffer_copy(pyarray)
154
+ cinfo = m.get_buffer_info(cbytes)
155
+ else:
156
+ cbytes = (ctypes.c_char * len(pyarray)).from_buffer(pyarray)
157
+ cinfo = m.get_buffer_info(cbytes)
158
+
159
+ assert cinfo.size == pyinfo.size
160
+ assert cinfo.ndim == pyinfo.ndim
161
+ assert cinfo.shape == pyinfo.shape
162
+ assert cinfo.strides == pyinfo.strides
163
+ assert not cinfo.readonly