xcoll 0.5.11__py3-none-any.whl → 0.6.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 (333) hide show
  1. xcoll/__init__.py +5 -18
  2. xcoll/beam_elements/__init__.py +1 -0
  3. xcoll/beam_elements/absorber.py +12 -2
  4. xcoll/beam_elements/base.py +162 -62
  5. xcoll/beam_elements/blowup.py +1 -0
  6. xcoll/beam_elements/elements_src/black_absorber.h +57 -58
  7. xcoll/beam_elements/elements_src/black_crystal.h +49 -50
  8. xcoll/beam_elements/elements_src/everest_block.h +21 -11
  9. xcoll/beam_elements/elements_src/everest_collimator.h +100 -91
  10. xcoll/beam_elements/elements_src/everest_crystal.h +145 -140
  11. xcoll/beam_elements/elements_src/transparent_collimator.h +126 -0
  12. xcoll/beam_elements/elements_src/transparent_crystal.h +118 -0
  13. xcoll/beam_elements/everest.py +16 -5
  14. xcoll/beam_elements/monitor.py +1 -0
  15. xcoll/beam_elements/transparent.py +83 -0
  16. xcoll/colldb.py +15 -6
  17. xcoll/general.py +1 -1
  18. xcoll/headers/particle_states.py +51 -0
  19. xcoll/initial_distribution.py +129 -103
  20. xcoll/interaction_record/interaction_record.py +2 -1
  21. xcoll/interaction_record/interaction_types.py +2 -2
  22. xcoll/line_tools.py +163 -74
  23. xcoll/lossmap.py +519 -127
  24. xcoll/plot.py +109 -0
  25. xcoll/rf_sweep.py +6 -0
  26. xcoll/scattering_routines/engine.py +600 -0
  27. xcoll/scattering_routines/environment.py +297 -0
  28. xcoll/scattering_routines/everest/amorphous.h +95 -71
  29. xcoll/scattering_routines/everest/{channeling.h → channelling.h} +121 -112
  30. xcoll/scattering_routines/everest/constants.h +1 -1
  31. xcoll/scattering_routines/everest/crystal_parameters.h +9 -9
  32. xcoll/scattering_routines/everest/everest.h +8 -3
  33. xcoll/scattering_routines/everest/everest.py +2 -1
  34. xcoll/scattering_routines/everest/ionisation_loss.h +141 -0
  35. xcoll/scattering_routines/everest/jaw.h +19 -24
  36. xcoll/scattering_routines/everest/materials.py +2 -0
  37. xcoll/scattering_routines/everest/multiple_coulomb_scattering.h +2 -2
  38. xcoll/scattering_routines/everest/nuclear_interaction.h +35 -19
  39. xcoll/scattering_routines/everest/properties.h +4 -73
  40. xcoll/xaux.py +73 -0
  41. {xcoll-0.5.11.dist-info → xcoll-0.6.0.dist-info}/METADATA +5 -5
  42. xcoll-0.6.0.dist-info/RECORD +135 -0
  43. xcoll/_manager.py +0 -22
  44. xcoll/headers/particle_states.h +0 -25
  45. xcoll/install.py +0 -35
  46. xcoll/scattering_routines/geant4/collimasim/.git +0 -1
  47. xcoll/scattering_routines/geant4/collimasim/.gitignore +0 -12
  48. xcoll/scattering_routines/geant4/collimasim/.gitmodules +0 -3
  49. xcoll/scattering_routines/geant4/collimasim/CMakeLists.txt +0 -26
  50. xcoll/scattering_routines/geant4/collimasim/README.md +0 -21
  51. xcoll/scattering_routines/geant4/collimasim/docs/Makefile +0 -20
  52. xcoll/scattering_routines/geant4/collimasim/docs/make.bat +0 -35
  53. xcoll/scattering_routines/geant4/collimasim/docs/source/collimasim.rst +0 -10
  54. xcoll/scattering_routines/geant4/collimasim/docs/source/conf.py +0 -59
  55. xcoll/scattering_routines/geant4/collimasim/docs/source/index.rst +0 -26
  56. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.appveyor.yml +0 -37
  57. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.clang-format +0 -19
  58. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.clang-tidy +0 -65
  59. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.cmake-format.yaml +0 -73
  60. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.git +0 -1
  61. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/CODEOWNERS +0 -9
  62. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/CONTRIBUTING.md +0 -386
  63. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/ISSUE_TEMPLATE/bug-report.yml +0 -45
  64. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/ISSUE_TEMPLATE/config.yml +0 -8
  65. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/dependabot.yml +0 -16
  66. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/labeler.yml +0 -8
  67. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/labeler_merged.yml +0 -3
  68. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/pull_request_template.md +0 -19
  69. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/workflows/ci.yml +0 -969
  70. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/workflows/configure.yml +0 -84
  71. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/workflows/format.yml +0 -48
  72. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/workflows/labeler.yml +0 -16
  73. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/workflows/pip.yml +0 -103
  74. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.gitignore +0 -45
  75. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.pre-commit-config.yaml +0 -151
  76. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.readthedocs.yml +0 -3
  77. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/CMakeLists.txt +0 -297
  78. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/LICENSE +0 -29
  79. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/MANIFEST.in +0 -6
  80. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/README.rst +0 -180
  81. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/Doxyfile +0 -23
  82. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/Makefile +0 -192
  83. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/_static/theme_overrides.css +0 -11
  84. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/chrono.rst +0 -81
  85. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/custom.rst +0 -93
  86. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/eigen.rst +0 -310
  87. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/functional.rst +0 -109
  88. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/index.rst +0 -43
  89. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/overview.rst +0 -171
  90. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/stl.rst +0 -251
  91. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/strings.rst +0 -305
  92. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/classes.rst +0 -1297
  93. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/embedding.rst +0 -262
  94. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/exceptions.rst +0 -396
  95. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/functions.rst +0 -568
  96. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/misc.rst +0 -337
  97. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/pycpp/index.rst +0 -13
  98. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/pycpp/numpy.rst +0 -463
  99. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/pycpp/object.rst +0 -286
  100. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/pycpp/utilities.rst +0 -155
  101. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/smart_ptrs.rst +0 -174
  102. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/basics.rst +0 -308
  103. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/benchmark.py +0 -91
  104. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/benchmark.rst +0 -95
  105. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/changelog.rst +0 -2050
  106. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/classes.rst +0 -542
  107. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/cmake/index.rst +0 -8
  108. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/compiling.rst +0 -648
  109. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/conf.py +0 -381
  110. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/faq.rst +0 -343
  111. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/index.rst +0 -48
  112. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/installing.rst +0 -105
  113. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/limitations.rst +0 -72
  114. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/pybind11-logo.png +0 -0
  115. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/pybind11_vs_boost_python1.png +0 -0
  116. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/pybind11_vs_boost_python1.svg +0 -427
  117. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/pybind11_vs_boost_python2.png +0 -0
  118. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/pybind11_vs_boost_python2.svg +0 -427
  119. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/reference.rst +0 -130
  120. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/release.rst +0 -96
  121. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/requirements.txt +0 -8
  122. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/upgrade.rst +0 -548
  123. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/attr.h +0 -605
  124. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/buffer_info.h +0 -144
  125. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/cast.h +0 -1432
  126. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/chrono.h +0 -213
  127. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/common.h +0 -2
  128. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/complex.h +0 -65
  129. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/class.h +0 -709
  130. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/common.h +0 -1021
  131. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/descr.h +0 -104
  132. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/init.h +0 -346
  133. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/internals.h +0 -467
  134. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/type_caster_base.h +0 -978
  135. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/typeid.h +0 -55
  136. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/eigen.h +0 -606
  137. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/embed.h +0 -284
  138. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/eval.h +0 -163
  139. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/functional.h +0 -121
  140. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/gil.h +0 -193
  141. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/iostream.h +0 -275
  142. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/numpy.h +0 -1741
  143. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/operators.h +0 -163
  144. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/options.h +0 -65
  145. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/pybind11.h +0 -2497
  146. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/pytypes.h +0 -1879
  147. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/stl/filesystem.h +0 -103
  148. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/stl.h +0 -375
  149. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/stl_bind.h +0 -747
  150. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/noxfile.py +0 -88
  151. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/__init__.py +0 -11
  152. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/__main__.py +0 -52
  153. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/_version.py +0 -12
  154. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/_version.pyi +0 -6
  155. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/commands.py +0 -21
  156. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/py.typed +0 -0
  157. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/setup_helpers.py +0 -482
  158. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/setup_helpers.pyi +0 -63
  159. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pyproject.toml +0 -41
  160. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/setup.cfg +0 -56
  161. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/setup.py +0 -155
  162. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/CMakeLists.txt +0 -503
  163. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/conftest.py +0 -208
  164. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/constructor_stats.h +0 -275
  165. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/cross_module_gil_utils.cpp +0 -73
  166. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/env.py +0 -33
  167. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/extra_python_package/pytest.ini +0 -0
  168. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/extra_python_package/test_files.py +0 -279
  169. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/extra_setuptools/pytest.ini +0 -0
  170. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/extra_setuptools/test_setuphelper.py +0 -143
  171. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/local_bindings.h +0 -85
  172. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/object.h +0 -179
  173. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/pybind11_cross_module_tests.cpp +0 -151
  174. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/pybind11_tests.cpp +0 -91
  175. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/pybind11_tests.h +0 -85
  176. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/pytest.ini +0 -19
  177. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/requirements.txt +0 -12
  178. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_async.cpp +0 -26
  179. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_async.py +0 -25
  180. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_buffers.cpp +0 -216
  181. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_buffers.py +0 -163
  182. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_builtin_casters.cpp +0 -286
  183. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_builtin_casters.py +0 -536
  184. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_call_policies.cpp +0 -107
  185. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_call_policies.py +0 -248
  186. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_callbacks.cpp +0 -227
  187. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_callbacks.py +0 -202
  188. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_chrono.cpp +0 -84
  189. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_chrono.py +0 -210
  190. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_class.cpp +0 -550
  191. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_class.py +0 -473
  192. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/CMakeLists.txt +0 -84
  193. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/embed.cpp +0 -21
  194. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/installed_embed/CMakeLists.txt +0 -28
  195. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/installed_function/CMakeLists.txt +0 -39
  196. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/installed_target/CMakeLists.txt +0 -46
  197. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/main.cpp +0 -6
  198. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/subdirectory_embed/CMakeLists.txt +0 -41
  199. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/subdirectory_function/CMakeLists.txt +0 -35
  200. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/subdirectory_target/CMakeLists.txt +0 -41
  201. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/test.py +0 -10
  202. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_constants_and_functions.cpp +0 -165
  203. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_constants_and_functions.py +0 -53
  204. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_copy_move.cpp +0 -238
  205. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_copy_move.py +0 -126
  206. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_custom_type_casters.cpp +0 -141
  207. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_custom_type_casters.py +0 -117
  208. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_custom_type_setup.cpp +0 -41
  209. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_custom_type_setup.py +0 -50
  210. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_docstring_options.cpp +0 -69
  211. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_docstring_options.py +0 -42
  212. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_eigen.cpp +0 -348
  213. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_eigen.py +0 -771
  214. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_embed/CMakeLists.txt +0 -47
  215. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_embed/catch.cpp +0 -22
  216. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_embed/external_module.cpp +0 -23
  217. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_embed/test_interpreter.cpp +0 -326
  218. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_embed/test_interpreter.py +0 -15
  219. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_enum.cpp +0 -148
  220. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_enum.py +0 -272
  221. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_eval.cpp +0 -119
  222. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_eval.py +0 -51
  223. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_eval_call.py +0 -5
  224. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_exceptions.cpp +0 -285
  225. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_exceptions.h +0 -12
  226. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_exceptions.py +0 -265
  227. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_factory_constructors.cpp +0 -397
  228. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_factory_constructors.py +0 -520
  229. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_gil_scoped.cpp +0 -49
  230. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_gil_scoped.py +0 -94
  231. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_iostream.cpp +0 -125
  232. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_iostream.py +0 -331
  233. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_kwargs_and_defaults.cpp +0 -153
  234. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_kwargs_and_defaults.py +0 -284
  235. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_local_bindings.cpp +0 -107
  236. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_local_bindings.py +0 -257
  237. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_methods_and_attributes.cpp +0 -412
  238. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_methods_and_attributes.py +0 -517
  239. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_modules.cpp +0 -102
  240. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_modules.py +0 -92
  241. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_multiple_inheritance.cpp +0 -233
  242. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_multiple_inheritance.py +0 -360
  243. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_array.cpp +0 -472
  244. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_array.py +0 -593
  245. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_dtypes.cpp +0 -524
  246. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_dtypes.py +0 -441
  247. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_vectorize.cpp +0 -103
  248. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_vectorize.py +0 -267
  249. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_opaque_types.cpp +0 -73
  250. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_opaque_types.py +0 -59
  251. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_operator_overloading.cpp +0 -235
  252. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_operator_overloading.py +0 -146
  253. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_pickling.cpp +0 -189
  254. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_pickling.py +0 -82
  255. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_pytypes.cpp +0 -560
  256. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_pytypes.py +0 -651
  257. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_sequences_and_iterators.cpp +0 -500
  258. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_sequences_and_iterators.py +0 -253
  259. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_smart_ptr.cpp +0 -452
  260. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_smart_ptr.py +0 -318
  261. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_stl.cpp +0 -342
  262. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_stl.py +0 -291
  263. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_stl_binders.cpp +0 -131
  264. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_stl_binders.py +0 -318
  265. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_tagbased_polymorphic.cpp +0 -144
  266. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_tagbased_polymorphic.py +0 -29
  267. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_thread.cpp +0 -66
  268. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_thread.py +0 -44
  269. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_union.cpp +0 -22
  270. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_union.py +0 -9
  271. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_virtual_functions.cpp +0 -510
  272. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_virtual_functions.py +0 -408
  273. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/valgrind-numpy-scipy.supp +0 -140
  274. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/valgrind-python.supp +0 -117
  275. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/FindCatch.cmake +0 -70
  276. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/FindEigen3.cmake +0 -86
  277. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/FindPythonLibsNew.cmake +0 -257
  278. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/check-style.sh +0 -44
  279. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/cmake_uninstall.cmake.in +0 -23
  280. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/libsize.py +0 -39
  281. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/make_changelog.py +0 -64
  282. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/pybind11Common.cmake +0 -402
  283. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/pybind11Config.cmake.in +0 -233
  284. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/pybind11NewTools.cmake +0 -276
  285. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/pybind11Tools.cmake +0 -214
  286. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/pyproject.toml +0 -3
  287. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/setup_global.py.in +0 -65
  288. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/setup_main.py.in +0 -41
  289. xcoll/scattering_routines/geant4/collimasim/pyproject.toml +0 -8
  290. xcoll/scattering_routines/geant4/collimasim/setup.py +0 -144
  291. xcoll/scattering_routines/geant4/collimasim/src/collimasim/BDSPyATInterface.cpp +0 -403
  292. xcoll/scattering_routines/geant4/collimasim/src/collimasim/BDSPyATInterface.hh +0 -100
  293. xcoll/scattering_routines/geant4/collimasim/src/collimasim/BDSXtrackInterface.cpp +0 -763
  294. xcoll/scattering_routines/geant4/collimasim/src/collimasim/BDSXtrackInterface.hh +0 -118
  295. xcoll/scattering_routines/geant4/collimasim/src/collimasim/__init__.py +0 -8
  296. xcoll/scattering_routines/geant4/collimasim/src/collimasim/bindings.cpp +0 -63
  297. xcoll/scattering_routines/geant4/collimasim/src/collimasim/pyCollimatorPass.py +0 -142
  298. xcoll/scattering_routines/geant4/collimasim/src/collimasim/xtrack_collimator.py +0 -556
  299. xcoll/scattering_routines/geant4/collimasim/src/collimasim.egg-info/PKG-INFO +0 -6
  300. xcoll/scattering_routines/geant4/collimasim/src/collimasim.egg-info/SOURCES.txt +0 -24
  301. xcoll/scattering_routines/geant4/collimasim/src/collimasim.egg-info/dependency_links.txt +0 -1
  302. xcoll/scattering_routines/geant4/collimasim/src/collimasim.egg-info/not-zip-safe +0 -1
  303. xcoll/scattering_routines/geant4/collimasim/src/collimasim.egg-info/top_level.txt +0 -1
  304. xcoll/scattering_routines/geant4/collimasim/tests/README.md +0 -25
  305. xcoll/scattering_routines/geant4/collimasim/tests/resources/CollDB_forions.dat +0 -25
  306. xcoll/scattering_routines/geant4/collimasim/tests/resources/CollDB_new_example.dat +0 -18
  307. xcoll/scattering_routines/geant4/collimasim/tests/resources/CollDB_old_example.dat +0 -68
  308. xcoll/scattering_routines/geant4/collimasim/tests/resources/CollDB_testing.dat +0 -15
  309. xcoll/scattering_routines/geant4/collimasim/tests/resources/CollDB_yaml_example.yaml +0 -110
  310. xcoll/scattering_routines/geant4/collimasim/tests/resources/collgaps.dat +0 -7
  311. xcoll/scattering_routines/geant4/collimasim/tests/resources/collgaps_pyat_test.dat +0 -3
  312. xcoll/scattering_routines/geant4/collimasim/tests/resources/collonly_twiss_file_example.tfs +0 -54
  313. xcoll/scattering_routines/geant4/collimasim/tests/resources/settings.gmad +0 -3
  314. xcoll/scattering_routines/geant4/collimasim/tests/resources/settings_black_absorber.gmad +0 -3
  315. xcoll/scattering_routines/geant4/collimasim/tests/resources/settings_ions.gmad +0 -5
  316. xcoll/scattering_routines/geant4/collimasim/tests/resources/twiss_file_testing.tfs +0 -51
  317. xcoll/scattering_routines/geant4/collimasim/tests/test_pyat.py +0 -65
  318. xcoll/scattering_routines/geant4/collimasim/tests/test_pyat_passmethod.py +0 -59
  319. xcoll/scattering_routines/geant4/collimasim/tests/test_pyat_tracking.py +0 -102
  320. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack.py +0 -75
  321. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_angle.py +0 -74
  322. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_colldb_load.py +0 -84
  323. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_interaction.py +0 -159
  324. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_interaction_ion.py +0 -99
  325. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_ions.py +0 -78
  326. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_lost_energy.py +0 -88
  327. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_tilt.py +0 -80
  328. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_tracking.py +0 -97
  329. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_tracking_ions.py +0 -96
  330. xcoll-0.5.11.dist-info/RECORD +0 -413
  331. {xcoll-0.5.11.dist-info → xcoll-0.6.0.dist-info}/LICENSE +0 -0
  332. {xcoll-0.5.11.dist-info → xcoll-0.6.0.dist-info}/NOTICE +0 -0
  333. {xcoll-0.5.11.dist-info → xcoll-0.6.0.dist-info}/WHEEL +0 -0
@@ -1,193 +0,0 @@
1
- /*
2
- pybind11/gil.h: RAII helpers for managing the GIL
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
- #pragma once
11
-
12
- #include "detail/common.h"
13
- #include "detail/internals.h"
14
-
15
- PYBIND11_NAMESPACE_BEGIN(PYBIND11_NAMESPACE)
16
-
17
-
18
- PYBIND11_NAMESPACE_BEGIN(detail)
19
-
20
- // forward declarations
21
- PyThreadState *get_thread_state_unchecked();
22
-
23
- PYBIND11_NAMESPACE_END(detail)
24
-
25
-
26
- #if defined(WITH_THREAD) && !defined(PYPY_VERSION)
27
-
28
- /* The functions below essentially reproduce the PyGILState_* API using a RAII
29
- * pattern, but there are a few important differences:
30
- *
31
- * 1. When acquiring the GIL from an non-main thread during the finalization
32
- * phase, the GILState API blindly terminates the calling thread, which
33
- * is often not what is wanted. This API does not do this.
34
- *
35
- * 2. The gil_scoped_release function can optionally cut the relationship
36
- * of a PyThreadState and its associated thread, which allows moving it to
37
- * another thread (this is a fairly rare/advanced use case).
38
- *
39
- * 3. The reference count of an acquired thread state can be controlled. This
40
- * can be handy to prevent cases where callbacks issued from an external
41
- * thread would otherwise constantly construct and destroy thread state data
42
- * structures.
43
- *
44
- * See the Python bindings of NanoGUI (http://github.com/wjakob/nanogui) for an
45
- * example which uses features 2 and 3 to migrate the Python thread of
46
- * execution to another thread (to run the event loop on the original thread,
47
- * in this case).
48
- */
49
-
50
- class gil_scoped_acquire {
51
- public:
52
- PYBIND11_NOINLINE gil_scoped_acquire() {
53
- auto &internals = detail::get_internals();
54
- tstate = (PyThreadState *) PYBIND11_TLS_GET_VALUE(internals.tstate);
55
-
56
- if (!tstate) {
57
- /* Check if the GIL was acquired using the PyGILState_* API instead (e.g. if
58
- calling from a Python thread). Since we use a different key, this ensures
59
- we don't create a new thread state and deadlock in PyEval_AcquireThread
60
- below. Note we don't save this state with internals.tstate, since we don't
61
- create it we would fail to clear it (its reference count should be > 0). */
62
- tstate = PyGILState_GetThisThreadState();
63
- }
64
-
65
- if (!tstate) {
66
- tstate = PyThreadState_New(internals.istate);
67
- #if !defined(NDEBUG)
68
- if (!tstate)
69
- pybind11_fail("scoped_acquire: could not create thread state!");
70
- #endif
71
- tstate->gilstate_counter = 0;
72
- PYBIND11_TLS_REPLACE_VALUE(internals.tstate, tstate);
73
- } else {
74
- release = detail::get_thread_state_unchecked() != tstate;
75
- }
76
-
77
- if (release) {
78
- PyEval_AcquireThread(tstate);
79
- }
80
-
81
- inc_ref();
82
- }
83
-
84
- void inc_ref() {
85
- ++tstate->gilstate_counter;
86
- }
87
-
88
- PYBIND11_NOINLINE void dec_ref() {
89
- --tstate->gilstate_counter;
90
- #if !defined(NDEBUG)
91
- if (detail::get_thread_state_unchecked() != tstate)
92
- pybind11_fail("scoped_acquire::dec_ref(): thread state must be current!");
93
- if (tstate->gilstate_counter < 0)
94
- pybind11_fail("scoped_acquire::dec_ref(): reference count underflow!");
95
- #endif
96
- if (tstate->gilstate_counter == 0) {
97
- #if !defined(NDEBUG)
98
- if (!release)
99
- pybind11_fail("scoped_acquire::dec_ref(): internal error!");
100
- #endif
101
- PyThreadState_Clear(tstate);
102
- if (active)
103
- PyThreadState_DeleteCurrent();
104
- PYBIND11_TLS_DELETE_VALUE(detail::get_internals().tstate);
105
- release = false;
106
- }
107
- }
108
-
109
- /// This method will disable the PyThreadState_DeleteCurrent call and the
110
- /// GIL won't be acquired. This method should be used if the interpreter
111
- /// could be shutting down when this is called, as thread deletion is not
112
- /// allowed during shutdown. Check _Py_IsFinalizing() on Python 3.7+, and
113
- /// protect subsequent code.
114
- PYBIND11_NOINLINE void disarm() {
115
- active = false;
116
- }
117
-
118
- PYBIND11_NOINLINE ~gil_scoped_acquire() {
119
- dec_ref();
120
- if (release)
121
- PyEval_SaveThread();
122
- }
123
- private:
124
- PyThreadState *tstate = nullptr;
125
- bool release = true;
126
- bool active = true;
127
- };
128
-
129
- class gil_scoped_release {
130
- public:
131
- explicit gil_scoped_release(bool disassoc = false) : disassoc(disassoc) {
132
- // `get_internals()` must be called here unconditionally in order to initialize
133
- // `internals.tstate` for subsequent `gil_scoped_acquire` calls. Otherwise, an
134
- // initialization race could occur as multiple threads try `gil_scoped_acquire`.
135
- auto &internals = detail::get_internals();
136
- tstate = PyEval_SaveThread();
137
- if (disassoc) {
138
- auto key = internals.tstate;
139
- PYBIND11_TLS_DELETE_VALUE(key);
140
- }
141
- }
142
-
143
- /// This method will disable the PyThreadState_DeleteCurrent call and the
144
- /// GIL won't be acquired. This method should be used if the interpreter
145
- /// could be shutting down when this is called, as thread deletion is not
146
- /// allowed during shutdown. Check _Py_IsFinalizing() on Python 3.7+, and
147
- /// protect subsequent code.
148
- PYBIND11_NOINLINE void disarm() {
149
- active = false;
150
- }
151
-
152
- ~gil_scoped_release() {
153
- if (!tstate)
154
- return;
155
- // `PyEval_RestoreThread()` should not be called if runtime is finalizing
156
- if (active)
157
- PyEval_RestoreThread(tstate);
158
- if (disassoc) {
159
- auto key = detail::get_internals().tstate;
160
- PYBIND11_TLS_REPLACE_VALUE(key, tstate);
161
- }
162
- }
163
- private:
164
- PyThreadState *tstate;
165
- bool disassoc;
166
- bool active = true;
167
- };
168
- #elif defined(PYPY_VERSION)
169
- class gil_scoped_acquire {
170
- PyGILState_STATE state;
171
- public:
172
- gil_scoped_acquire() { state = PyGILState_Ensure(); }
173
- ~gil_scoped_acquire() { PyGILState_Release(state); }
174
- void disarm() {}
175
- };
176
-
177
- class gil_scoped_release {
178
- PyThreadState *state;
179
- public:
180
- gil_scoped_release() { state = PyEval_SaveThread(); }
181
- ~gil_scoped_release() { PyEval_RestoreThread(state); }
182
- void disarm() {}
183
- };
184
- #else
185
- class gil_scoped_acquire {
186
- void disarm() {}
187
- };
188
- class gil_scoped_release {
189
- void disarm() {}
190
- };
191
- #endif
192
-
193
- PYBIND11_NAMESPACE_END(PYBIND11_NAMESPACE)
@@ -1,275 +0,0 @@
1
- /*
2
- pybind11/iostream.h -- Tools to assist with redirecting cout and cerr to Python
3
-
4
- Copyright (c) 2017 Henry F. Schreiner
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
- WARNING: The implementation in this file is NOT thread safe. Multiple
10
- threads writing to a redirected ostream concurrently cause data races
11
- and potentially buffer overflows. Therefore it is currently a requirement
12
- that all (possibly) concurrent redirected ostream writes are protected by
13
- a mutex.
14
- #HelpAppreciated: Work on iostream.h thread safety.
15
- For more background see the discussions under
16
- https://github.com/pybind/pybind11/pull/2982 and
17
- https://github.com/pybind/pybind11/pull/2995.
18
- */
19
-
20
- #pragma once
21
-
22
- #include "pybind11.h"
23
-
24
- #include <algorithm>
25
- #include <cstring>
26
- #include <iostream>
27
- #include <iterator>
28
- #include <memory>
29
- #include <ostream>
30
- #include <streambuf>
31
- #include <string>
32
- #include <utility>
33
-
34
- PYBIND11_NAMESPACE_BEGIN(PYBIND11_NAMESPACE)
35
- PYBIND11_NAMESPACE_BEGIN(detail)
36
-
37
- // Buffer that writes to Python instead of C++
38
- class pythonbuf : public std::streambuf {
39
- private:
40
- using traits_type = std::streambuf::traits_type;
41
-
42
- const size_t buf_size;
43
- std::unique_ptr<char[]> d_buffer;
44
- object pywrite;
45
- object pyflush;
46
-
47
- int overflow(int c) override {
48
- if (!traits_type::eq_int_type(c, traits_type::eof())) {
49
- *pptr() = traits_type::to_char_type(c);
50
- pbump(1);
51
- }
52
- return sync() == 0 ? traits_type::not_eof(c) : traits_type::eof();
53
- }
54
-
55
- // Computes how many bytes at the end of the buffer are part of an
56
- // incomplete sequence of UTF-8 bytes.
57
- // Precondition: pbase() < pptr()
58
- size_t utf8_remainder() const {
59
- const auto rbase = std::reverse_iterator<char *>(pbase());
60
- const auto rpptr = std::reverse_iterator<char *>(pptr());
61
- auto is_ascii = [](char c) {
62
- return (static_cast<unsigned char>(c) & 0x80) == 0x00;
63
- };
64
- auto is_leading = [](char c) {
65
- return (static_cast<unsigned char>(c) & 0xC0) == 0xC0;
66
- };
67
- auto is_leading_2b = [](char c) {
68
- return static_cast<unsigned char>(c) <= 0xDF;
69
- };
70
- auto is_leading_3b = [](char c) {
71
- return static_cast<unsigned char>(c) <= 0xEF;
72
- };
73
- // If the last character is ASCII, there are no incomplete code points
74
- if (is_ascii(*rpptr))
75
- return 0;
76
- // Otherwise, work back from the end of the buffer and find the first
77
- // UTF-8 leading byte
78
- const auto rpend = rbase - rpptr >= 3 ? rpptr + 3 : rbase;
79
- const auto leading = std::find_if(rpptr, rpend, is_leading);
80
- if (leading == rbase)
81
- return 0;
82
- const auto dist = static_cast<size_t>(leading - rpptr);
83
- size_t remainder = 0;
84
-
85
- if (dist == 0)
86
- remainder = 1; // 1-byte code point is impossible
87
- else if (dist == 1)
88
- remainder = is_leading_2b(*leading) ? 0 : dist + 1;
89
- else if (dist == 2)
90
- remainder = is_leading_3b(*leading) ? 0 : dist + 1;
91
- // else if (dist >= 3), at least 4 bytes before encountering an UTF-8
92
- // leading byte, either no remainder or invalid UTF-8.
93
- // Invalid UTF-8 will cause an exception later when converting
94
- // to a Python string, so that's not handled here.
95
- return remainder;
96
- }
97
-
98
- // This function must be non-virtual to be called in a destructor.
99
- int _sync() {
100
- if (pbase() != pptr()) { // If buffer is not empty
101
- gil_scoped_acquire tmp;
102
- // This subtraction cannot be negative, so dropping the sign.
103
- auto size = static_cast<size_t>(pptr() - pbase());
104
- size_t remainder = utf8_remainder();
105
-
106
- if (size > remainder) {
107
- str line(pbase(), size - remainder);
108
- pywrite(line);
109
- pyflush();
110
- }
111
-
112
- // Copy the remainder at the end of the buffer to the beginning:
113
- if (remainder > 0)
114
- std::memmove(pbase(), pptr() - remainder, remainder);
115
- setp(pbase(), epptr());
116
- pbump(static_cast<int>(remainder));
117
- }
118
- return 0;
119
- }
120
-
121
- int sync() override {
122
- return _sync();
123
- }
124
-
125
- public:
126
- explicit pythonbuf(const object &pyostream, size_t buffer_size = 1024)
127
- : buf_size(buffer_size), d_buffer(new char[buf_size]), pywrite(pyostream.attr("write")),
128
- pyflush(pyostream.attr("flush")) {
129
- setp(d_buffer.get(), d_buffer.get() + buf_size - 1);
130
- }
131
-
132
- pythonbuf(pythonbuf&&) = default;
133
-
134
- /// Sync before destroy
135
- ~pythonbuf() override {
136
- _sync();
137
- }
138
- };
139
-
140
- PYBIND11_NAMESPACE_END(detail)
141
-
142
-
143
- /** \rst
144
- This a move-only guard that redirects output.
145
-
146
- .. code-block:: cpp
147
-
148
- #include <pybind11/iostream.h>
149
-
150
- ...
151
-
152
- {
153
- py::scoped_ostream_redirect output;
154
- std::cout << "Hello, World!"; // Python stdout
155
- } // <-- return std::cout to normal
156
-
157
- You can explicitly pass the c++ stream and the python object,
158
- for example to guard stderr instead.
159
-
160
- .. code-block:: cpp
161
-
162
- {
163
- py::scoped_ostream_redirect output{std::cerr, py::module::import("sys").attr("stderr")};
164
- std::cout << "Hello, World!";
165
- }
166
- \endrst */
167
- class scoped_ostream_redirect {
168
- protected:
169
- std::streambuf *old;
170
- std::ostream &costream;
171
- detail::pythonbuf buffer;
172
-
173
- public:
174
- explicit scoped_ostream_redirect(std::ostream &costream = std::cout,
175
- const object &pyostream
176
- = module_::import("sys").attr("stdout"))
177
- : costream(costream), buffer(pyostream) {
178
- old = costream.rdbuf(&buffer);
179
- }
180
-
181
- ~scoped_ostream_redirect() {
182
- costream.rdbuf(old);
183
- }
184
-
185
- scoped_ostream_redirect(const scoped_ostream_redirect &) = delete;
186
- scoped_ostream_redirect(scoped_ostream_redirect &&other) = default;
187
- scoped_ostream_redirect &operator=(const scoped_ostream_redirect &) = delete;
188
- scoped_ostream_redirect &operator=(scoped_ostream_redirect &&) = delete;
189
- };
190
-
191
-
192
- /** \rst
193
- Like `scoped_ostream_redirect`, but redirects cerr by default. This class
194
- is provided primary to make ``py::call_guard`` easier to make.
195
-
196
- .. code-block:: cpp
197
-
198
- m.def("noisy_func", &noisy_func,
199
- py::call_guard<scoped_ostream_redirect,
200
- scoped_estream_redirect>());
201
-
202
- \endrst */
203
- class scoped_estream_redirect : public scoped_ostream_redirect {
204
- public:
205
- explicit scoped_estream_redirect(std::ostream &costream = std::cerr,
206
- const object &pyostream
207
- = module_::import("sys").attr("stderr"))
208
- : scoped_ostream_redirect(costream, pyostream) {}
209
- };
210
-
211
-
212
- PYBIND11_NAMESPACE_BEGIN(detail)
213
-
214
- // Class to redirect output as a context manager. C++ backend.
215
- class OstreamRedirect {
216
- bool do_stdout_;
217
- bool do_stderr_;
218
- std::unique_ptr<scoped_ostream_redirect> redirect_stdout;
219
- std::unique_ptr<scoped_estream_redirect> redirect_stderr;
220
-
221
- public:
222
- explicit OstreamRedirect(bool do_stdout = true, bool do_stderr = true)
223
- : do_stdout_(do_stdout), do_stderr_(do_stderr) {}
224
-
225
- void enter() {
226
- if (do_stdout_)
227
- redirect_stdout.reset(new scoped_ostream_redirect());
228
- if (do_stderr_)
229
- redirect_stderr.reset(new scoped_estream_redirect());
230
- }
231
-
232
- void exit() {
233
- redirect_stdout.reset();
234
- redirect_stderr.reset();
235
- }
236
- };
237
-
238
- PYBIND11_NAMESPACE_END(detail)
239
-
240
- /** \rst
241
- This is a helper function to add a C++ redirect context manager to Python
242
- instead of using a C++ guard. To use it, add the following to your binding code:
243
-
244
- .. code-block:: cpp
245
-
246
- #include <pybind11/iostream.h>
247
-
248
- ...
249
-
250
- py::add_ostream_redirect(m, "ostream_redirect");
251
-
252
- You now have a Python context manager that redirects your output:
253
-
254
- .. code-block:: python
255
-
256
- with m.ostream_redirect():
257
- m.print_to_cout_function()
258
-
259
- This manager can optionally be told which streams to operate on:
260
-
261
- .. code-block:: python
262
-
263
- with m.ostream_redirect(stdout=true, stderr=true):
264
- m.noisy_function_with_error_printing()
265
-
266
- \endrst */
267
- inline class_<detail::OstreamRedirect>
268
- add_ostream_redirect(module_ m, const std::string &name = "ostream_redirect") {
269
- return class_<detail::OstreamRedirect>(std::move(m), name.c_str(), module_local())
270
- .def(init<bool, bool>(), arg("stdout") = true, arg("stderr") = true)
271
- .def("__enter__", &detail::OstreamRedirect::enter)
272
- .def("__exit__", [](detail::OstreamRedirect &self_, const args &) { self_.exit(); });
273
- }
274
-
275
- PYBIND11_NAMESPACE_END(PYBIND11_NAMESPACE)