xcoll 0.3.6__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 (350) hide show
  1. xcoll/__init__.py +13 -4
  2. xcoll/beam_elements/__init__.py +14 -6
  3. xcoll/beam_elements/absorber.py +41 -7
  4. xcoll/beam_elements/base.py +1202 -247
  5. xcoll/beam_elements/blowup.py +198 -0
  6. xcoll/beam_elements/elements_src/black_absorber.h +136 -0
  7. xcoll/beam_elements/elements_src/black_crystal.h +129 -0
  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 +59 -30
  11. xcoll/beam_elements/elements_src/everest_collimator.h +237 -0
  12. xcoll/beam_elements/elements_src/everest_crystal.h +280 -0
  13. xcoll/beam_elements/everest.py +65 -119
  14. xcoll/beam_elements/monitor.py +428 -0
  15. xcoll/colldb.py +276 -747
  16. xcoll/general.py +5 -5
  17. xcoll/headers/checks.h +1 -1
  18. xcoll/headers/particle_states.h +2 -2
  19. xcoll/initial_distribution.py +207 -0
  20. xcoll/install.py +179 -0
  21. xcoll/interaction_record/__init__.py +1 -0
  22. xcoll/interaction_record/interaction_record.py +298 -0
  23. xcoll/interaction_record/interaction_record_src/interaction_record.h +98 -0
  24. xcoll/{impacts → interaction_record}/interaction_types.py +11 -4
  25. xcoll/line_tools.py +82 -0
  26. xcoll/lossmap.py +219 -0
  27. xcoll/manager.py +2 -937
  28. xcoll/rf_sweep.py +1 -1
  29. xcoll/scattering_routines/everest/amorphous.h +232 -0
  30. xcoll/scattering_routines/everest/channeling.h +240 -0
  31. xcoll/scattering_routines/everest/crystal_parameters.h +137 -0
  32. xcoll/scattering_routines/everest/everest.h +11 -30
  33. xcoll/scattering_routines/everest/everest.py +13 -10
  34. xcoll/scattering_routines/everest/jaw.h +28 -197
  35. xcoll/scattering_routines/everest/materials.py +37 -15
  36. xcoll/scattering_routines/everest/multiple_coulomb_scattering.h +31 -10
  37. xcoll/scattering_routines/everest/nuclear_interaction.h +86 -0
  38. xcoll/scattering_routines/everest/properties.h +6 -1
  39. xcoll/scattering_routines/fluka/flukaio/lib/libFlukaIO64.a +0 -0
  40. xcoll/scattering_routines/geant4/collimasim/.git +1 -0
  41. xcoll/scattering_routines/geant4/collimasim/.gitignore +12 -0
  42. xcoll/scattering_routines/geant4/collimasim/.gitmodules +3 -0
  43. xcoll/scattering_routines/geant4/collimasim/CMakeLists.txt +26 -0
  44. xcoll/scattering_routines/geant4/collimasim/README.md +21 -0
  45. xcoll/scattering_routines/geant4/collimasim/docs/Makefile +20 -0
  46. xcoll/scattering_routines/geant4/collimasim/docs/make.bat +35 -0
  47. xcoll/scattering_routines/geant4/collimasim/docs/source/collimasim.rst +10 -0
  48. xcoll/scattering_routines/geant4/collimasim/docs/source/conf.py +59 -0
  49. xcoll/scattering_routines/geant4/collimasim/docs/source/index.rst +26 -0
  50. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.appveyor.yml +37 -0
  51. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.clang-format +19 -0
  52. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.clang-tidy +65 -0
  53. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.cmake-format.yaml +73 -0
  54. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.git +1 -0
  55. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/CODEOWNERS +9 -0
  56. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/CONTRIBUTING.md +386 -0
  57. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/ISSUE_TEMPLATE/bug-report.yml +45 -0
  58. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/ISSUE_TEMPLATE/config.yml +8 -0
  59. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/dependabot.yml +16 -0
  60. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/labeler.yml +8 -0
  61. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/labeler_merged.yml +3 -0
  62. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/pull_request_template.md +19 -0
  63. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/workflows/ci.yml +969 -0
  64. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/workflows/configure.yml +84 -0
  65. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/workflows/format.yml +48 -0
  66. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/workflows/labeler.yml +16 -0
  67. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/workflows/pip.yml +103 -0
  68. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.gitignore +45 -0
  69. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.pre-commit-config.yaml +151 -0
  70. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.readthedocs.yml +3 -0
  71. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/CMakeLists.txt +297 -0
  72. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/LICENSE +29 -0
  73. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/MANIFEST.in +6 -0
  74. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/README.rst +180 -0
  75. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/Doxyfile +23 -0
  76. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/Makefile +192 -0
  77. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/_static/theme_overrides.css +11 -0
  78. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/chrono.rst +81 -0
  79. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/custom.rst +93 -0
  80. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/eigen.rst +310 -0
  81. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/functional.rst +109 -0
  82. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/index.rst +43 -0
  83. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/overview.rst +171 -0
  84. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/stl.rst +251 -0
  85. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/strings.rst +305 -0
  86. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/classes.rst +1297 -0
  87. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/embedding.rst +262 -0
  88. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/exceptions.rst +396 -0
  89. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/functions.rst +568 -0
  90. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/misc.rst +337 -0
  91. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/pycpp/index.rst +13 -0
  92. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/pycpp/numpy.rst +463 -0
  93. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/pycpp/object.rst +286 -0
  94. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/pycpp/utilities.rst +155 -0
  95. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/smart_ptrs.rst +174 -0
  96. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/basics.rst +308 -0
  97. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/benchmark.py +91 -0
  98. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/benchmark.rst +95 -0
  99. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/changelog.rst +2050 -0
  100. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/classes.rst +542 -0
  101. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/cmake/index.rst +8 -0
  102. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/compiling.rst +648 -0
  103. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/conf.py +381 -0
  104. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/faq.rst +343 -0
  105. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/index.rst +48 -0
  106. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/installing.rst +105 -0
  107. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/limitations.rst +72 -0
  108. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/pybind11-logo.png +0 -0
  109. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/pybind11_vs_boost_python1.png +0 -0
  110. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/pybind11_vs_boost_python1.svg +427 -0
  111. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/pybind11_vs_boost_python2.png +0 -0
  112. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/pybind11_vs_boost_python2.svg +427 -0
  113. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/reference.rst +130 -0
  114. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/release.rst +96 -0
  115. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/requirements.txt +8 -0
  116. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/upgrade.rst +548 -0
  117. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/attr.h +605 -0
  118. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/buffer_info.h +144 -0
  119. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/cast.h +1432 -0
  120. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/chrono.h +213 -0
  121. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/common.h +2 -0
  122. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/complex.h +65 -0
  123. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/class.h +709 -0
  124. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/common.h +1021 -0
  125. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/descr.h +104 -0
  126. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/init.h +346 -0
  127. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/internals.h +467 -0
  128. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/type_caster_base.h +978 -0
  129. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/typeid.h +55 -0
  130. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/eigen.h +606 -0
  131. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/embed.h +284 -0
  132. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/eval.h +163 -0
  133. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/functional.h +121 -0
  134. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/gil.h +193 -0
  135. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/iostream.h +275 -0
  136. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/numpy.h +1741 -0
  137. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/operators.h +163 -0
  138. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/options.h +65 -0
  139. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/pybind11.h +2497 -0
  140. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/pytypes.h +1879 -0
  141. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/stl/filesystem.h +103 -0
  142. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/stl.h +375 -0
  143. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/stl_bind.h +747 -0
  144. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/noxfile.py +88 -0
  145. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/__init__.py +11 -0
  146. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/__main__.py +52 -0
  147. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/_version.py +12 -0
  148. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/_version.pyi +6 -0
  149. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/commands.py +21 -0
  150. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/py.typed +0 -0
  151. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/setup_helpers.py +482 -0
  152. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/setup_helpers.pyi +63 -0
  153. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pyproject.toml +41 -0
  154. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/setup.cfg +56 -0
  155. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/setup.py +155 -0
  156. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/CMakeLists.txt +503 -0
  157. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/conftest.py +208 -0
  158. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/constructor_stats.h +275 -0
  159. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/cross_module_gil_utils.cpp +73 -0
  160. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/env.py +33 -0
  161. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/extra_python_package/pytest.ini +0 -0
  162. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/extra_python_package/test_files.py +279 -0
  163. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/extra_setuptools/pytest.ini +0 -0
  164. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/extra_setuptools/test_setuphelper.py +143 -0
  165. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/local_bindings.h +85 -0
  166. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/object.h +179 -0
  167. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/pybind11_cross_module_tests.cpp +151 -0
  168. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/pybind11_tests.cpp +91 -0
  169. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/pybind11_tests.h +85 -0
  170. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/pytest.ini +19 -0
  171. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/requirements.txt +12 -0
  172. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_async.cpp +26 -0
  173. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_async.py +25 -0
  174. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_buffers.cpp +216 -0
  175. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_buffers.py +163 -0
  176. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_builtin_casters.cpp +286 -0
  177. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_builtin_casters.py +536 -0
  178. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_call_policies.cpp +107 -0
  179. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_call_policies.py +248 -0
  180. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_callbacks.cpp +227 -0
  181. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_callbacks.py +202 -0
  182. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_chrono.cpp +84 -0
  183. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_chrono.py +210 -0
  184. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_class.cpp +550 -0
  185. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_class.py +473 -0
  186. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/CMakeLists.txt +84 -0
  187. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/embed.cpp +21 -0
  188. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/installed_embed/CMakeLists.txt +28 -0
  189. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/installed_function/CMakeLists.txt +39 -0
  190. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/installed_target/CMakeLists.txt +46 -0
  191. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/main.cpp +6 -0
  192. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/subdirectory_embed/CMakeLists.txt +41 -0
  193. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/subdirectory_function/CMakeLists.txt +35 -0
  194. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/subdirectory_target/CMakeLists.txt +41 -0
  195. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/test.py +10 -0
  196. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_constants_and_functions.cpp +165 -0
  197. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_constants_and_functions.py +53 -0
  198. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_copy_move.cpp +238 -0
  199. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_copy_move.py +126 -0
  200. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_custom_type_casters.cpp +141 -0
  201. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_custom_type_casters.py +117 -0
  202. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_custom_type_setup.cpp +41 -0
  203. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_custom_type_setup.py +50 -0
  204. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_docstring_options.cpp +69 -0
  205. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_docstring_options.py +42 -0
  206. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_eigen.cpp +348 -0
  207. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_eigen.py +771 -0
  208. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_embed/CMakeLists.txt +47 -0
  209. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_embed/catch.cpp +22 -0
  210. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_embed/external_module.cpp +23 -0
  211. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_embed/test_interpreter.cpp +326 -0
  212. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_embed/test_interpreter.py +15 -0
  213. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_enum.cpp +148 -0
  214. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_enum.py +272 -0
  215. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_eval.cpp +119 -0
  216. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_eval.py +51 -0
  217. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_eval_call.py +5 -0
  218. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_exceptions.cpp +285 -0
  219. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_exceptions.h +12 -0
  220. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_exceptions.py +265 -0
  221. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_factory_constructors.cpp +397 -0
  222. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_factory_constructors.py +520 -0
  223. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_gil_scoped.cpp +49 -0
  224. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_gil_scoped.py +94 -0
  225. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_iostream.cpp +125 -0
  226. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_iostream.py +331 -0
  227. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_kwargs_and_defaults.cpp +153 -0
  228. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_kwargs_and_defaults.py +284 -0
  229. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_local_bindings.cpp +107 -0
  230. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_local_bindings.py +257 -0
  231. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_methods_and_attributes.cpp +412 -0
  232. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_methods_and_attributes.py +517 -0
  233. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_modules.cpp +102 -0
  234. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_modules.py +92 -0
  235. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_multiple_inheritance.cpp +233 -0
  236. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_multiple_inheritance.py +360 -0
  237. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_array.cpp +472 -0
  238. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_array.py +593 -0
  239. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_dtypes.cpp +524 -0
  240. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_dtypes.py +441 -0
  241. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_vectorize.cpp +103 -0
  242. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_vectorize.py +267 -0
  243. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_opaque_types.cpp +73 -0
  244. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_opaque_types.py +59 -0
  245. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_operator_overloading.cpp +235 -0
  246. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_operator_overloading.py +146 -0
  247. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_pickling.cpp +189 -0
  248. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_pickling.py +82 -0
  249. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_pytypes.cpp +560 -0
  250. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_pytypes.py +651 -0
  251. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_sequences_and_iterators.cpp +500 -0
  252. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_sequences_and_iterators.py +253 -0
  253. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_smart_ptr.cpp +452 -0
  254. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_smart_ptr.py +318 -0
  255. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_stl.cpp +342 -0
  256. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_stl.py +291 -0
  257. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_stl_binders.cpp +131 -0
  258. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_stl_binders.py +318 -0
  259. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_tagbased_polymorphic.cpp +144 -0
  260. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_tagbased_polymorphic.py +29 -0
  261. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_thread.cpp +66 -0
  262. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_thread.py +44 -0
  263. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_union.cpp +22 -0
  264. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_union.py +9 -0
  265. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_virtual_functions.cpp +510 -0
  266. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_virtual_functions.py +408 -0
  267. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/valgrind-numpy-scipy.supp +140 -0
  268. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/valgrind-python.supp +117 -0
  269. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/FindCatch.cmake +70 -0
  270. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/FindEigen3.cmake +86 -0
  271. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/FindPythonLibsNew.cmake +257 -0
  272. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/check-style.sh +44 -0
  273. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/cmake_uninstall.cmake.in +23 -0
  274. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/libsize.py +39 -0
  275. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/make_changelog.py +64 -0
  276. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/pybind11Common.cmake +402 -0
  277. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/pybind11Config.cmake.in +233 -0
  278. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/pybind11NewTools.cmake +276 -0
  279. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/pybind11Tools.cmake +214 -0
  280. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/pyproject.toml +3 -0
  281. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/setup_global.py.in +65 -0
  282. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/setup_main.py.in +41 -0
  283. xcoll/scattering_routines/geant4/collimasim/pyproject.toml +8 -0
  284. xcoll/scattering_routines/geant4/collimasim/setup.py +144 -0
  285. xcoll/scattering_routines/geant4/collimasim/src/collimasim/BDSPyATInterface.cpp +403 -0
  286. xcoll/scattering_routines/geant4/collimasim/src/collimasim/BDSPyATInterface.hh +100 -0
  287. xcoll/scattering_routines/geant4/collimasim/src/collimasim/BDSXtrackInterface.cpp +763 -0
  288. xcoll/scattering_routines/geant4/collimasim/src/collimasim/BDSXtrackInterface.hh +118 -0
  289. xcoll/scattering_routines/geant4/collimasim/src/collimasim/__init__.py +8 -0
  290. xcoll/scattering_routines/geant4/collimasim/src/collimasim/bindings.cpp +63 -0
  291. xcoll/scattering_routines/geant4/collimasim/src/collimasim/pyCollimatorPass.py +142 -0
  292. xcoll/scattering_routines/geant4/collimasim/src/collimasim/xtrack_collimator.py +556 -0
  293. xcoll/scattering_routines/geant4/collimasim/src/collimasim.egg-info/PKG-INFO +6 -0
  294. xcoll/scattering_routines/geant4/collimasim/src/collimasim.egg-info/SOURCES.txt +24 -0
  295. xcoll/scattering_routines/geant4/collimasim/src/collimasim.egg-info/dependency_links.txt +1 -0
  296. xcoll/scattering_routines/geant4/collimasim/src/collimasim.egg-info/not-zip-safe +1 -0
  297. xcoll/scattering_routines/geant4/collimasim/src/collimasim.egg-info/top_level.txt +1 -0
  298. xcoll/scattering_routines/geant4/collimasim/tests/README.md +25 -0
  299. xcoll/scattering_routines/geant4/collimasim/tests/resources/CollDB_forions.dat +25 -0
  300. xcoll/scattering_routines/geant4/collimasim/tests/resources/CollDB_new_example.dat +18 -0
  301. xcoll/scattering_routines/geant4/collimasim/tests/resources/CollDB_old_example.dat +68 -0
  302. xcoll/scattering_routines/geant4/collimasim/tests/resources/CollDB_testing.dat +15 -0
  303. xcoll/scattering_routines/geant4/collimasim/tests/resources/CollDB_yaml_example.yaml +110 -0
  304. xcoll/scattering_routines/geant4/collimasim/tests/resources/collgaps.dat +7 -0
  305. xcoll/scattering_routines/geant4/collimasim/tests/resources/collgaps_pyat_test.dat +3 -0
  306. xcoll/scattering_routines/geant4/collimasim/tests/resources/collonly_twiss_file_example.tfs +54 -0
  307. xcoll/scattering_routines/geant4/collimasim/tests/resources/settings.gmad +3 -0
  308. xcoll/scattering_routines/geant4/collimasim/tests/resources/settings_black_absorber.gmad +3 -0
  309. xcoll/scattering_routines/geant4/collimasim/tests/resources/settings_ions.gmad +5 -0
  310. xcoll/scattering_routines/geant4/collimasim/tests/resources/twiss_file_testing.tfs +51 -0
  311. xcoll/scattering_routines/geant4/collimasim/tests/test_pyat.py +65 -0
  312. xcoll/scattering_routines/geant4/collimasim/tests/test_pyat_passmethod.py +59 -0
  313. xcoll/scattering_routines/geant4/collimasim/tests/test_pyat_tracking.py +102 -0
  314. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack.py +75 -0
  315. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_angle.py +74 -0
  316. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_colldb_load.py +84 -0
  317. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_interaction.py +159 -0
  318. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_interaction_ion.py +99 -0
  319. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_ions.py +78 -0
  320. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_lost_energy.py +88 -0
  321. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_tilt.py +80 -0
  322. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_tracking.py +97 -0
  323. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_tracking_ions.py +96 -0
  324. xcoll/scattering_routines/geometry/__init__.py +6 -0
  325. xcoll/scattering_routines/geometry/collimator_geometry.h +218 -0
  326. xcoll/scattering_routines/geometry/crystal_geometry.h +153 -0
  327. xcoll/scattering_routines/geometry/geometry.py +26 -0
  328. xcoll/scattering_routines/geometry/get_s.h +92 -0
  329. xcoll/scattering_routines/geometry/methods.h +111 -0
  330. xcoll/scattering_routines/geometry/objects.h +154 -0
  331. xcoll/scattering_routines/geometry/rotation.h +23 -0
  332. xcoll/scattering_routines/geometry/segments.h +226 -0
  333. xcoll/scattering_routines/geometry/sort.h +184 -0
  334. {xcoll-0.3.6.dist-info → xcoll-0.5.0.dist-info}/METADATA +1 -1
  335. xcoll-0.5.0.dist-info/RECORD +413 -0
  336. xcoll/beam_elements/collimators_src/absorber.h +0 -141
  337. xcoll/beam_elements/collimators_src/everest_collimator.h +0 -142
  338. xcoll/beam_elements/collimators_src/everest_crystal.h +0 -115
  339. xcoll/collimator_settings.py +0 -457
  340. xcoll/impacts/__init__.py +0 -1
  341. xcoll/impacts/impacts.py +0 -102
  342. xcoll/impacts/impacts_src/impacts.h +0 -99
  343. xcoll/scattering_routines/everest/crystal.h +0 -1302
  344. xcoll/scattering_routines/everest/scatter.h +0 -169
  345. xcoll/scattering_routines/everest/scatter_crystal.h +0 -260
  346. xcoll/scattering_routines/fluka/build_fluka_input.py +0 -58
  347. xcoll-0.3.6.dist-info/RECORD +0 -111
  348. {xcoll-0.3.6.dist-info → xcoll-0.5.0.dist-info}/LICENSE +0 -0
  349. {xcoll-0.3.6.dist-info → xcoll-0.5.0.dist-info}/NOTICE +0 -0
  350. {xcoll-0.3.6.dist-info → xcoll-0.5.0.dist-info}/WHEEL +0 -0
@@ -0,0 +1,126 @@
1
+ # -*- coding: utf-8 -*-
2
+ import pytest
3
+
4
+ from pybind11_tests import copy_move_policies as m
5
+
6
+
7
+ def test_lacking_copy_ctor():
8
+ with pytest.raises(RuntimeError) as excinfo:
9
+ m.lacking_copy_ctor.get_one()
10
+ assert "is non-copyable!" in str(excinfo.value)
11
+
12
+
13
+ def test_lacking_move_ctor():
14
+ with pytest.raises(RuntimeError) as excinfo:
15
+ m.lacking_move_ctor.get_one()
16
+ assert "is neither movable nor copyable!" in str(excinfo.value)
17
+
18
+
19
+ def test_move_and_copy_casts():
20
+ """Cast some values in C++ via custom type casters and count the number of moves/copies."""
21
+
22
+ cstats = m.move_and_copy_cstats()
23
+ c_m, c_mc, c_c = (
24
+ cstats["MoveOnlyInt"],
25
+ cstats["MoveOrCopyInt"],
26
+ cstats["CopyOnlyInt"],
27
+ )
28
+
29
+ # The type move constructions/assignments below each get incremented: the move assignment comes
30
+ # from the type_caster load; the move construction happens when extracting that via a cast or
31
+ # loading into an argument.
32
+ assert m.move_and_copy_casts(3) == 18
33
+ assert c_m.copy_assignments + c_m.copy_constructions == 0
34
+ assert c_m.move_assignments == 2
35
+ assert c_m.move_constructions >= 2
36
+ assert c_mc.alive() == 0
37
+ assert c_mc.copy_assignments + c_mc.copy_constructions == 0
38
+ assert c_mc.move_assignments == 2
39
+ assert c_mc.move_constructions >= 2
40
+ assert c_c.alive() == 0
41
+ assert c_c.copy_assignments == 2
42
+ assert c_c.copy_constructions >= 2
43
+ assert c_m.alive() + c_mc.alive() + c_c.alive() == 0
44
+
45
+
46
+ def test_move_and_copy_loads():
47
+ """Call some functions that load arguments via custom type casters and count the number of
48
+ moves/copies."""
49
+
50
+ cstats = m.move_and_copy_cstats()
51
+ c_m, c_mc, c_c = (
52
+ cstats["MoveOnlyInt"],
53
+ cstats["MoveOrCopyInt"],
54
+ cstats["CopyOnlyInt"],
55
+ )
56
+
57
+ assert m.move_only(10) == 10 # 1 move, c_m
58
+ assert m.move_or_copy(11) == 11 # 1 move, c_mc
59
+ assert m.copy_only(12) == 12 # 1 copy, c_c
60
+ assert m.move_pair((13, 14)) == 27 # 1 c_m move, 1 c_mc move
61
+ assert m.move_tuple((15, 16, 17)) == 48 # 2 c_m moves, 1 c_mc move
62
+ assert m.copy_tuple((18, 19)) == 37 # 2 c_c copies
63
+ # Direct constructions: 2 c_m moves, 2 c_mc moves, 1 c_c copy
64
+ # Extra moves/copies when moving pairs/tuples: 3 c_m, 3 c_mc, 2 c_c
65
+ assert m.move_copy_nested((1, ((2, 3, (4,)), 5))) == 15
66
+
67
+ assert c_m.copy_assignments + c_m.copy_constructions == 0
68
+ assert c_m.move_assignments == 6
69
+ assert c_m.move_constructions == 9
70
+ assert c_mc.copy_assignments + c_mc.copy_constructions == 0
71
+ assert c_mc.move_assignments == 5
72
+ assert c_mc.move_constructions == 8
73
+ assert c_c.copy_assignments == 4
74
+ assert c_c.copy_constructions == 6
75
+ assert c_m.alive() + c_mc.alive() + c_c.alive() == 0
76
+
77
+
78
+ @pytest.mark.skipif(not m.has_optional, reason="no <optional>")
79
+ def test_move_and_copy_load_optional():
80
+ """Tests move/copy loads of std::optional arguments"""
81
+
82
+ cstats = m.move_and_copy_cstats()
83
+ c_m, c_mc, c_c = (
84
+ cstats["MoveOnlyInt"],
85
+ cstats["MoveOrCopyInt"],
86
+ cstats["CopyOnlyInt"],
87
+ )
88
+
89
+ # The extra move/copy constructions below come from the std::optional move (which has to move
90
+ # its arguments):
91
+ assert m.move_optional(10) == 10 # c_m: 1 move assign, 2 move construct
92
+ assert m.move_or_copy_optional(11) == 11 # c_mc: 1 move assign, 2 move construct
93
+ assert m.copy_optional(12) == 12 # c_c: 1 copy assign, 2 copy construct
94
+ # 1 move assign + move construct moves each of c_m, c_mc, 1 c_c copy
95
+ # +1 move/copy construct each from moving the tuple
96
+ # +1 move/copy construct each from moving the optional (which moves the tuple again)
97
+ assert m.move_optional_tuple((3, 4, 5)) == 12
98
+
99
+ assert c_m.copy_assignments + c_m.copy_constructions == 0
100
+ assert c_m.move_assignments == 2
101
+ assert c_m.move_constructions == 5
102
+ assert c_mc.copy_assignments + c_mc.copy_constructions == 0
103
+ assert c_mc.move_assignments == 2
104
+ assert c_mc.move_constructions == 5
105
+ assert c_c.copy_assignments == 2
106
+ assert c_c.copy_constructions == 5
107
+ assert c_m.alive() + c_mc.alive() + c_c.alive() == 0
108
+
109
+
110
+ def test_private_op_new():
111
+ """An object with a private `operator new` cannot be returned by value"""
112
+
113
+ with pytest.raises(RuntimeError) as excinfo:
114
+ m.private_op_new_value()
115
+ assert "is neither movable nor copyable" in str(excinfo.value)
116
+
117
+ assert m.private_op_new_reference().value == 1
118
+
119
+
120
+ def test_move_fallback():
121
+ """#389: rvp::move should fall-through to copy on non-movable objects"""
122
+
123
+ m1 = m.get_moveissue1(1)
124
+ assert m1.value == 1
125
+ m2 = m.get_moveissue2(2)
126
+ assert m2.value == 2
@@ -0,0 +1,141 @@
1
+ /*
2
+ tests/test_custom_type_casters.cpp -- tests type_caster<T>
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
+
13
+
14
+ // py::arg/py::arg_v testing: these arguments just record their argument when invoked
15
+ class ArgInspector1 { public: std::string arg = "(default arg inspector 1)"; };
16
+ class ArgInspector2 { public: std::string arg = "(default arg inspector 2)"; };
17
+ class ArgAlwaysConverts { };
18
+ namespace pybind11 { namespace detail {
19
+ template <> struct type_caster<ArgInspector1> {
20
+ public:
21
+ PYBIND11_TYPE_CASTER(ArgInspector1, _("ArgInspector1"));
22
+
23
+ bool load(handle src, bool convert) {
24
+ value.arg = "loading ArgInspector1 argument " +
25
+ std::string(convert ? "WITH" : "WITHOUT") + " conversion allowed. "
26
+ "Argument value = " + (std::string) str(src);
27
+ return true;
28
+ }
29
+
30
+ static handle cast(const ArgInspector1 &src, return_value_policy, handle) {
31
+ return str(src.arg).release();
32
+ }
33
+ };
34
+ template <> struct type_caster<ArgInspector2> {
35
+ public:
36
+ PYBIND11_TYPE_CASTER(ArgInspector2, _("ArgInspector2"));
37
+
38
+ bool load(handle src, bool convert) {
39
+ value.arg = "loading ArgInspector2 argument " +
40
+ std::string(convert ? "WITH" : "WITHOUT") + " conversion allowed. "
41
+ "Argument value = " + (std::string) str(src);
42
+ return true;
43
+ }
44
+
45
+ static handle cast(const ArgInspector2 &src, return_value_policy, handle) {
46
+ return str(src.arg).release();
47
+ }
48
+ };
49
+ template <> struct type_caster<ArgAlwaysConverts> {
50
+ public:
51
+ PYBIND11_TYPE_CASTER(ArgAlwaysConverts, _("ArgAlwaysConverts"));
52
+
53
+ bool load(handle, bool convert) {
54
+ return convert;
55
+ }
56
+
57
+ static handle cast(const ArgAlwaysConverts &, return_value_policy, handle) {
58
+ return py::none().release();
59
+ }
60
+ };
61
+ } // namespace detail
62
+ } // namespace pybind11
63
+
64
+ // test_custom_caster_destruction
65
+ class DestructionTester {
66
+ public:
67
+ DestructionTester() { print_default_created(this); }
68
+ ~DestructionTester() { print_destroyed(this); }
69
+ DestructionTester(const DestructionTester &) { print_copy_created(this); }
70
+ DestructionTester(DestructionTester &&) noexcept { print_move_created(this); }
71
+ DestructionTester &operator=(const DestructionTester &) { print_copy_assigned(this); return *this; }
72
+ DestructionTester &operator=(DestructionTester &&) noexcept {
73
+ print_move_assigned(this);
74
+ return *this;
75
+ }
76
+ };
77
+ namespace pybind11 { namespace detail {
78
+ template <> struct type_caster<DestructionTester> {
79
+ PYBIND11_TYPE_CASTER(DestructionTester, _("DestructionTester"));
80
+ bool load(handle, bool) { return true; }
81
+
82
+ static handle cast(const DestructionTester &, return_value_policy, handle) {
83
+ return py::bool_(true).release();
84
+ }
85
+ };
86
+ } // namespace detail
87
+ } // namespace pybind11
88
+
89
+ TEST_SUBMODULE(custom_type_casters, m) {
90
+ // test_custom_type_casters
91
+
92
+ // test_noconvert_args
93
+ //
94
+ // Test converting. The ArgAlwaysConverts is just there to make the first no-conversion pass
95
+ // fail so that our call always ends up happening via the second dispatch (the one that allows
96
+ // some conversion).
97
+ class ArgInspector {
98
+ public:
99
+ ArgInspector1 f(ArgInspector1 a, ArgAlwaysConverts) { return a; }
100
+ std::string g(const ArgInspector1 &a,
101
+ const ArgInspector1 &b,
102
+ int c,
103
+ ArgInspector2 *d,
104
+ ArgAlwaysConverts) {
105
+ return a.arg + "\n" + b.arg + "\n" + std::to_string(c) + "\n" + d->arg;
106
+ }
107
+ static ArgInspector2 h(ArgInspector2 a, ArgAlwaysConverts) { return a; }
108
+ };
109
+ // [workaround(intel)] ICC 20/21 breaks with py::arg().stuff, using py::arg{}.stuff works.
110
+ py::class_<ArgInspector>(m, "ArgInspector")
111
+ .def(py::init<>())
112
+ .def("f", &ArgInspector::f, py::arg(), py::arg() = ArgAlwaysConverts())
113
+ .def("g", &ArgInspector::g, "a"_a.noconvert(), "b"_a, "c"_a.noconvert()=13, "d"_a=ArgInspector2(), py::arg() = ArgAlwaysConverts())
114
+ .def_static("h", &ArgInspector::h, py::arg{}.noconvert(), py::arg() = ArgAlwaysConverts())
115
+ ;
116
+ m.def(
117
+ "arg_inspect_func",
118
+ [](const ArgInspector2 &a, const ArgInspector1 &b, ArgAlwaysConverts) {
119
+ return a.arg + "\n" + b.arg;
120
+ },
121
+ py::arg{}.noconvert(false),
122
+ py::arg_v(nullptr, ArgInspector1()).noconvert(true),
123
+ py::arg() = ArgAlwaysConverts());
124
+
125
+ m.def("floats_preferred", [](double f) { return 0.5 * f; }, "f"_a);
126
+ m.def("floats_only", [](double f) { return 0.5 * f; }, "f"_a.noconvert());
127
+ m.def("ints_preferred", [](int i) { return i / 2; }, "i"_a);
128
+ m.def("ints_only", [](int i) { return i / 2; }, "i"_a.noconvert());
129
+
130
+ // test_custom_caster_destruction
131
+ // Test that `take_ownership` works on types with a custom type caster when given a pointer
132
+
133
+ // default policy: don't take ownership:
134
+ m.def("custom_caster_no_destroy", []() { static auto *dt = new DestructionTester(); return dt; });
135
+
136
+ m.def("custom_caster_destroy", []() { return new DestructionTester(); },
137
+ py::return_value_policy::take_ownership); // Takes ownership: destroy when finished
138
+ m.def("custom_caster_destroy_const", []() -> const DestructionTester * { return new DestructionTester(); },
139
+ py::return_value_policy::take_ownership); // Likewise (const doesn't inhibit destruction)
140
+ m.def("destruction_tester_cstats", &ConstructorStats::get<DestructionTester>, py::return_value_policy::reference);
141
+ }
@@ -0,0 +1,117 @@
1
+ # -*- coding: utf-8 -*-
2
+ import pytest
3
+
4
+ from pybind11_tests import custom_type_casters as m
5
+
6
+
7
+ def test_noconvert_args(msg):
8
+ a = m.ArgInspector()
9
+ assert (
10
+ msg(a.f("hi"))
11
+ == """
12
+ loading ArgInspector1 argument WITH conversion allowed. Argument value = hi
13
+ """
14
+ )
15
+ assert (
16
+ msg(a.g("this is a", "this is b"))
17
+ == """
18
+ loading ArgInspector1 argument WITHOUT conversion allowed. Argument value = this is a
19
+ loading ArgInspector1 argument WITH conversion allowed. Argument value = this is b
20
+ 13
21
+ loading ArgInspector2 argument WITH conversion allowed. Argument value = (default arg inspector 2)
22
+ """ # noqa: E501 line too long
23
+ )
24
+ assert (
25
+ msg(a.g("this is a", "this is b", 42))
26
+ == """
27
+ loading ArgInspector1 argument WITHOUT conversion allowed. Argument value = this is a
28
+ loading ArgInspector1 argument WITH conversion allowed. Argument value = this is b
29
+ 42
30
+ loading ArgInspector2 argument WITH conversion allowed. Argument value = (default arg inspector 2)
31
+ """ # noqa: E501 line too long
32
+ )
33
+ assert (
34
+ msg(a.g("this is a", "this is b", 42, "this is d"))
35
+ == """
36
+ loading ArgInspector1 argument WITHOUT conversion allowed. Argument value = this is a
37
+ loading ArgInspector1 argument WITH conversion allowed. Argument value = this is b
38
+ 42
39
+ loading ArgInspector2 argument WITH conversion allowed. Argument value = this is d
40
+ """
41
+ )
42
+ assert (
43
+ a.h("arg 1")
44
+ == "loading ArgInspector2 argument WITHOUT conversion allowed. Argument value = arg 1"
45
+ )
46
+ assert (
47
+ msg(m.arg_inspect_func("A1", "A2"))
48
+ == """
49
+ loading ArgInspector2 argument WITH conversion allowed. Argument value = A1
50
+ loading ArgInspector1 argument WITHOUT conversion allowed. Argument value = A2
51
+ """
52
+ )
53
+
54
+ assert m.floats_preferred(4) == 2.0
55
+ assert m.floats_only(4.0) == 2.0
56
+ with pytest.raises(TypeError) as excinfo:
57
+ m.floats_only(4)
58
+ assert (
59
+ msg(excinfo.value)
60
+ == """
61
+ floats_only(): incompatible function arguments. The following argument types are supported:
62
+ 1. (f: float) -> float
63
+
64
+ Invoked with: 4
65
+ """
66
+ )
67
+
68
+ assert m.ints_preferred(4) == 2
69
+ assert m.ints_preferred(True) == 0
70
+ with pytest.raises(TypeError) as excinfo:
71
+ m.ints_preferred(4.0)
72
+ assert (
73
+ msg(excinfo.value)
74
+ == """
75
+ ints_preferred(): incompatible function arguments. The following argument types are supported:
76
+ 1. (i: int) -> int
77
+
78
+ Invoked with: 4.0
79
+ """ # noqa: E501 line too long
80
+ )
81
+
82
+ assert m.ints_only(4) == 2
83
+ with pytest.raises(TypeError) as excinfo:
84
+ m.ints_only(4.0)
85
+ assert (
86
+ msg(excinfo.value)
87
+ == """
88
+ ints_only(): incompatible function arguments. The following argument types are supported:
89
+ 1. (i: int) -> int
90
+
91
+ Invoked with: 4.0
92
+ """
93
+ )
94
+
95
+
96
+ def test_custom_caster_destruction():
97
+ """Tests that returning a pointer to a type that gets converted with a custom type caster gets
98
+ destroyed when the function has py::return_value_policy::take_ownership policy applied."""
99
+
100
+ cstats = m.destruction_tester_cstats()
101
+ # This one *doesn't* have take_ownership: the pointer should be used but not destroyed:
102
+ z = m.custom_caster_no_destroy()
103
+ assert cstats.alive() == 1 and cstats.default_constructions == 1
104
+ assert z
105
+
106
+ # take_ownership applied: this constructs a new object, casts it, then destroys it:
107
+ z = m.custom_caster_destroy()
108
+ assert z
109
+ assert cstats.default_constructions == 2
110
+
111
+ # Same, but with a const pointer return (which should *not* inhibit destruction):
112
+ z = m.custom_caster_destroy_const()
113
+ assert z
114
+ assert cstats.default_constructions == 3
115
+
116
+ # Make sure we still only have the original object (from ..._no_destroy()) alive:
117
+ assert cstats.alive() == 1
@@ -0,0 +1,41 @@
1
+ /*
2
+ tests/test_custom_type_setup.cpp -- Tests `pybind11::custom_type_setup`
3
+
4
+ Copyright (c) Google LLC
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/pybind11.h>
11
+
12
+ #include "pybind11_tests.h"
13
+
14
+ namespace py = pybind11;
15
+
16
+ namespace {
17
+
18
+ struct OwnsPythonObjects {
19
+ py::object value = py::none();
20
+ };
21
+ } // namespace
22
+
23
+ TEST_SUBMODULE(custom_type_setup, m) {
24
+ py::class_<OwnsPythonObjects> cls(
25
+ m, "OwnsPythonObjects", py::custom_type_setup([](PyHeapTypeObject *heap_type) {
26
+ auto *type = &heap_type->ht_type;
27
+ type->tp_flags |= Py_TPFLAGS_HAVE_GC;
28
+ type->tp_traverse = [](PyObject *self_base, visitproc visit, void *arg) {
29
+ auto &self = py::cast<OwnsPythonObjects &>(py::handle(self_base));
30
+ Py_VISIT(self.value.ptr());
31
+ return 0;
32
+ };
33
+ type->tp_clear = [](PyObject *self_base) {
34
+ auto &self = py::cast<OwnsPythonObjects &>(py::handle(self_base));
35
+ self.value = py::none();
36
+ return 0;
37
+ };
38
+ }));
39
+ cls.def(py::init<>());
40
+ cls.def_readwrite("value", &OwnsPythonObjects::value);
41
+ }
@@ -0,0 +1,50 @@
1
+ # -*- coding: utf-8 -*-
2
+
3
+ import gc
4
+ import weakref
5
+
6
+ import pytest
7
+
8
+ import env # noqa: F401
9
+ from pybind11_tests import custom_type_setup as m
10
+
11
+
12
+ @pytest.fixture
13
+ def gc_tester():
14
+ """Tests that an object is garbage collected.
15
+
16
+ Assumes that any unreferenced objects are fully collected after calling
17
+ `gc.collect()`. That is true on CPython, but does not appear to reliably
18
+ hold on PyPy.
19
+ """
20
+
21
+ weak_refs = []
22
+
23
+ def add_ref(obj):
24
+ # PyPy does not support `gc.is_tracked`.
25
+ if hasattr(gc, "is_tracked"):
26
+ assert gc.is_tracked(obj)
27
+ weak_refs.append(weakref.ref(obj))
28
+
29
+ yield add_ref
30
+
31
+ gc.collect()
32
+ for ref in weak_refs:
33
+ assert ref() is None
34
+
35
+
36
+ # PyPy does not seem to reliably garbage collect.
37
+ @pytest.mark.skipif("env.PYPY")
38
+ def test_self_cycle(gc_tester):
39
+ obj = m.OwnsPythonObjects()
40
+ obj.value = obj
41
+ gc_tester(obj)
42
+
43
+
44
+ # PyPy does not seem to reliably garbage collect.
45
+ @pytest.mark.skipif("env.PYPY")
46
+ def test_indirect_cycle(gc_tester):
47
+ obj = m.OwnsPythonObjects()
48
+ obj_list = [obj]
49
+ obj.value = obj_list
50
+ gc_tester(obj)
@@ -0,0 +1,69 @@
1
+ /*
2
+ tests/test_docstring_options.cpp -- generation of docstrings and signatures
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
+
12
+ TEST_SUBMODULE(docstring_options, m) {
13
+ // test_docstring_options
14
+ {
15
+ py::options options;
16
+ options.disable_function_signatures();
17
+
18
+ m.def("test_function1", [](int, int) {}, py::arg("a"), py::arg("b"));
19
+ m.def("test_function2", [](int, int) {}, py::arg("a"), py::arg("b"), "A custom docstring");
20
+
21
+ m.def("test_overloaded1", [](int) {}, py::arg("i"), "Overload docstring");
22
+ m.def("test_overloaded1", [](double) {}, py::arg("d"));
23
+
24
+ m.def("test_overloaded2", [](int) {}, py::arg("i"), "overload docstring 1");
25
+ m.def("test_overloaded2", [](double) {}, py::arg("d"), "overload docstring 2");
26
+
27
+ m.def("test_overloaded3", [](int) {}, py::arg("i"));
28
+ m.def("test_overloaded3", [](double) {}, py::arg("d"), "Overload docstr");
29
+
30
+ options.enable_function_signatures();
31
+
32
+ m.def("test_function3", [](int, int) {}, py::arg("a"), py::arg("b"));
33
+ m.def("test_function4", [](int, int) {}, py::arg("a"), py::arg("b"), "A custom docstring");
34
+
35
+ options.disable_function_signatures().disable_user_defined_docstrings();
36
+
37
+ m.def("test_function5", [](int, int) {}, py::arg("a"), py::arg("b"), "A custom docstring");
38
+
39
+ {
40
+ py::options nested_options;
41
+ nested_options.enable_user_defined_docstrings();
42
+ m.def("test_function6", [](int, int) {}, py::arg("a"), py::arg("b"), "A custom docstring");
43
+ }
44
+ }
45
+
46
+ m.def("test_function7", [](int, int) {}, py::arg("a"), py::arg("b"), "A custom docstring");
47
+
48
+ {
49
+ py::options options;
50
+ options.disable_user_defined_docstrings();
51
+ options.disable_function_signatures();
52
+
53
+ m.def("test_function8", []() {});
54
+ }
55
+
56
+ {
57
+ py::options options;
58
+ options.disable_user_defined_docstrings();
59
+
60
+ struct DocstringTestFoo {
61
+ int value;
62
+ void setValue(int v) { value = v; }
63
+ int getValue() const { return value; }
64
+ };
65
+ py::class_<DocstringTestFoo>(m, "DocstringTestFoo", "This is a class docstring")
66
+ .def_property("value_prop", &DocstringTestFoo::getValue, &DocstringTestFoo::setValue, "This is a property docstring")
67
+ ;
68
+ }
69
+ }
@@ -0,0 +1,42 @@
1
+ # -*- coding: utf-8 -*-
2
+ from pybind11_tests import docstring_options as m
3
+
4
+
5
+ def test_docstring_options():
6
+ # options.disable_function_signatures()
7
+ assert not m.test_function1.__doc__
8
+
9
+ assert m.test_function2.__doc__ == "A custom docstring"
10
+
11
+ # docstring specified on just the first overload definition:
12
+ assert m.test_overloaded1.__doc__ == "Overload docstring"
13
+
14
+ # docstring on both overloads:
15
+ assert m.test_overloaded2.__doc__ == "overload docstring 1\noverload docstring 2"
16
+
17
+ # docstring on only second overload:
18
+ assert m.test_overloaded3.__doc__ == "Overload docstr"
19
+
20
+ # options.enable_function_signatures()
21
+ assert m.test_function3.__doc__.startswith("test_function3(a: int, b: int) -> None")
22
+
23
+ assert m.test_function4.__doc__.startswith("test_function4(a: int, b: int) -> None")
24
+ assert m.test_function4.__doc__.endswith("A custom docstring\n")
25
+
26
+ # options.disable_function_signatures()
27
+ # options.disable_user_defined_docstrings()
28
+ assert not m.test_function5.__doc__
29
+
30
+ # nested options.enable_user_defined_docstrings()
31
+ assert m.test_function6.__doc__ == "A custom docstring"
32
+
33
+ # RAII destructor
34
+ assert m.test_function7.__doc__.startswith("test_function7(a: int, b: int) -> None")
35
+ assert m.test_function7.__doc__.endswith("A custom docstring\n")
36
+
37
+ # when all options are disabled, no docstring (instead of an empty one) should be generated
38
+ assert m.test_function8.__doc__ is None
39
+
40
+ # Suppression of user-defined docstrings for non-function objects
41
+ assert not m.DocstringTestFoo.__doc__
42
+ assert not m.DocstringTestFoo.value_prop.__doc__