fox/G4-DESCSS
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add: simulation result, Particle Definition;

Browse Source 
change: model format;
add: template class for PrimaryGeneratorAction;
add: random algorithm in PrimaryGeneratorAction;
This commit is contained in:
liuyihui 2022-05-16 18:08:47 +08:00
parent 3b2b20209d
commit f09c1d4407
16 changed files with 358 additions and 96 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
CMakeLists.txt
View File

@ -20,7 +20,10 @@ add_executable(DESCSS main.cpp ${sources} ${headers})
target_link_libraries(DESCSS ${Geant4_LIBRARIES}) target_link_libraries(DESCSS ${Geant4_LIBRARIES})
set(DESCSS_SCRIPTS set(DESCSS_SCRIPTS
debug.mac
electron.mac
vis.mac vis.mac
assets/model.txt
) )
foreach(_script ${DESCSS_SCRIPTS}) foreach(_script ${DESCSS_SCRIPTS})

7
G4.code-workspace
View File

@ -105,7 +105,12 @@
"wcsv_ntuple": "cpp", "wcsv_ntuple": "cpp",
"bufobj": "cpp", "bufobj": "cpp",
"aidas": "cpp", "aidas": "cpp",
"element": "cpp" "element": "cpp",
"colorf": "cpp",
"cid": "cpp",
"mem": "cpp",
"sout": "cpp",
"typedefs": "cpp"
} }
} }
} }

81
README.md
View File

@ -154,34 +154,34 @@ $$
|:--------:|:--------------------------------------------------------------------:|:-------:|:-----------------------:|:-----------------------:|:-----------------------:|:-------:|:-----------:| |:--------:|:--------------------------------------------------------------------:|:-------:|:-----------------------:|:-----------------------:|:-----------------------:|:-------:|:-----------:|
| $e^{-}$ | $j=j_0e^{-\frac{E}{E_0}}$ | 0.98107 | 0.1 | 10 | 9.23242E+11 | 923242 | 999999.8096 | | $e^{-}$ | $j=j_0e^{-\frac{E}{E_0}}$ | 0.98107 | 0.1 | 10 | 9.23242E+11 | 923242 | 999999.8096 |
| `p` | $j=j_0(E-\beta)^{-\alpha}$ | 0.99883 | 50 | 1000 | 42853994.15 | 100000 | 428.5399415 | | `p` | $j=j_0(E-\beta)^{-\alpha}$ | 0.99883 | 50 | 1000 | 42853994.15 | 100000 | 428.5399415 |
| `H` | $\frac{C_1\beta^{\alpha-1}}{pc}\left(\frac{pc}{pc+C_2}\right)^{C_3}$ | 0.98915 | 200 | 100000 | 12854310.92 | 100000 | 128.5431092 | | `H` | $\frac{C_1\beta^{\alpha-1}}{pc}\left(\frac{pc}{pc+C_2}\right)^{C_3}$ | 0.98915 | 220 | 100000 | 12854310.92 | 100000 | 128.5431092 |
| `He` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.98936 | 200 | 100000 | 1718828.12 | 100000 | 17.1882812 | | `He` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.98936 | 220 | 100000 | 1720188.128 | 100000 | 17.20188128 |
| `Li` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99755 | 200 | 100000 | 12982.36284 | 100000 | 0.129823628 | | `Li` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99755 | 220 | 100000 | 12988.97006 | 100000 | 0.129889701 |
| `Be` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99708 | 200 | 100000 | 6687.159271 | 100000 | 0.066871593 | | `Be` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99708 | 220 | 100000 | 6691.176848 | 100000 | 0.066911768 |
| `B` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99569 | 200 | 100000 | 17093.65985 | 100000 | 0.170936599 | | `B` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99569 | 220 | 100000 | 17105.83698 | 100000 | 0.17105837 |
| `C` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.98936 | 200 | 100000 | 52245.07239 | 100000 | 0.522450724 | | `C` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.98936 | 220 | 100000 | 52286.55444 | 100000 | 0.522865544 |
| `N` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.98892 | 200 | 100000 | 13564.66838 | 100000 | 0.135646684 | | `N` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.98892 | 220 | 100000 | 13576.5542 | 100000 | 0.135765542 |
| `O` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.98936 | 200 | 100000 | 48807.93662 | 100000 | 0.488079366 | | `O` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.98936 | 220 | 100000 | 48846.63508 | 100000 | 0.488466351 |
| `F` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99323 | 200 | 100000 | 1122.681148 | 100000 | 0.011226811 | | `F` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99323 | 220 | 100000 | 1123.492483 | 100000 | 0.011234925 |
| `Ne` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99014 | 200 | 100000 | 8065.236798 | 100000 | 0.080652368 | | `Ne` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99014 | 220 | 100000 | 8071.585382 | 100000 | 0.080715854 |
| `Na` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99263 | 200 | 100000 | 1865.503552 | 100000 | 0.018655036 | | `Na` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99263 | 220 | 100000 | 1866.8702 | 100000 | 0.018668702 |
| `Mg` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.98972 | 200 | 100000 | 10928.24228 | 100000 | 0.109282423 | | `Mg` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.98972 | 220 | 100000 | 10936.4597 | 100000 | 0.109364597 |
| `Al` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99217 | 200 | 100000 | 1938.943418 | 100000 | 0.019389434 | | `Al` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99217 | 220 | 100000 | 1940.381408 | 100000 | 0.019403814 |
| `Si` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.98873 | 200 | 100000 | 8262.870777 | 100000 | 0.082628708 | | `Si` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.98873 | 220 | 100000 | 8269.257173 | 100000 | 0.082692572 |
| `P` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99183 | 200 | 100000 | 413.5749377 | 100000 | 0.004135749 | | `P` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99183 | 220 | 100000 | 413.8897994 | 100000 | 0.004138898 |
| `S` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.98873 | 200 | 100000 | 1659.707 | 100000 | 0.01659707 | | `S` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.98873 | 220 | 100000 | 1660.990128 | 100000 | 0.016609901 |
| `Cl` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99087 | 200 | 100000 | 371.7198798 | 100000 | 0.003717199 | | `Cl` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99087 | 220 | 100000 | 372.0271181 | 100000 | 0.003720271 |
| `Ar` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99432 | 200 | 100000 | 751.2576633 | 100000 | 0.007512577 | | `Ar` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99432 | 220 | 100000 | 751.7977189 | 100000 | 0.007517977 |
| `K` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.98963 | 200 | 100000 | 466.3178791 | 100000 | 0.004663179 | | `K` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.98963 | 220 | 100000 | 466.7118907 | 100000 | 0.004667119 |
| `Ca` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.98935 | 200 | 100000 | 1161.134206 | 100000 | 0.011611342 | | `Ca` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.98935 | 220 | 100000 | 1161.900661 | 100000 | 0.011619007 |
| `Sc` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99251 | 200 | 100000 | 231.3722059 | 100000 | 0.002313722 | | `Sc` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99251 | 220 | 100000 | 231.5541271 | 100000 | 0.002315541 |
| `Ti` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.9934 | 200 | 100000 | 828.6605823 | 100000 | 0.008286606 | | `Ti` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.9934 | 220 | 100000 | 829.2810844 | 100000 | 0.008292811 |
| `V` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99424 | 200 | 100000 | 403.8859746 | 100000 | 0.00403886 | | `V` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99424 | 220 | 100000 | 404.1724078 | 100000 | 0.004041724 |
| `Cr` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99313 | 200 | 100000 | 783.4909673 | 100000 | 0.00783491 | | `Cr` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99313 | 220 | 100000 | 784.0902938 | 100000 | 0.007840903 |
| `Mn` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99385 | 200 | 100000 | 570.5548534 | 100000 | 0.005705549 | | `Mn` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99385 | 220 | 100000 | 570.9732265 | 100000 | 0.005709732 |
| `Fe` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99351 | 200 | 100000 | 6024.476681 | 100000 | 0.060244767 | | `Fe` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99351 | 220 | 100000 | 6028.059963 | 100000 | 0.0602806 |
| `Co` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99426 | 200 | 100000 | 20.89166947 | 100000 | 0.000208917 | | `Co` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99426 | 220 | 100000 | 20.90356212 | 100000 | 0.000209036 |
| `Ni` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99231 | 200 | 100000 | 292.8131737 | 100000 | 0.002928132 | | `Ni` | $C_1e^{-C_2E}(1-e^{-C_3E+C_4})$ | 0.99231 | 220 | 100000 | 292.9961042 | 100000 | 0.002929961 |
</div> </div>
@ -212,15 +212,7 @@ $$
2. `Acceptance-Rejection Method`（`ARM`） 2. `Acceptance-Rejection Method`（`ARM`）
对于一个在$[a, b]$的分布，设其概率密度函数(PDF)为$F(x)$，首先生成一个均匀分布随机数$X \sim U(x_{min}, x_{max})$，随后独立生成另一个均匀分布随机数$Y \sim U(y_{min}, y_{max})$，如果$Y \le F(X)$，则保留$X$，保留下来的$X$满足$F(x)$的分布。 对于一个在$[a, b]$的分布，设其概率密度函数(PDF)为$F(x)$，首先生成一个均匀分布随机数$X \sim U(x_{min}, x_{max})$，随后独立生成另一个均匀分布随机数$Y \sim U(y_{min}, y_{max})$，如果$Y \le F(X)$，则保留$X$，保留下来的$X$满足$F(x)$的分布。
`ITM`也被称为反演法，关键是需要获取累积分布函数的逆函数，效率较高；`ARM`本质上是一种模拟算法，效率较低，但是适应性更广，特别是对于一些复杂分布函数。`GCR`质子和俘获辐射粒子的模型有三类，其中`GCR`质子的函数较为简单，其余两类较为复杂，因此分别采用`ITM`和`ARM`。 `ITM`也被称为反演法，关键是需要获取累积分布函数的逆函数，效率较高；`ARM`本质上是一种模拟算法，效率很低，但是适应性更广，特别是对于一些复杂分布函数。
<div align=center>
| $F(E)$ | $\int F(E)\mathrm{d}E$ | $C(x)=\int_a^xF(u)\mathrm{d}u$ | $C^{-1}(y)$ |
|:--------------------------:|:------------------------------------------:|:----------------------------------------------------------------------------:|:---------------------------------------------------------------------------------------------------:|
| $j=j_0(E-\beta)^{-\alpha}$ | $\frac{j_0(E-\beta)^{1-\alpha}}{1-\alpha}$ | $\frac{j_0}{1-\alpha}\left((x-\beta)^{1-\alpha}-(a-\beta)^{1-\alpha}\right)$ | $\sqrt[1-\alpha]{\frac{(y+C)(1-\alpha)}{j_0}}+\beta,\ C=\frac{j_0(a-\beta)^{1-\alpha}}{1-\alpha}$ |
</div>
1. 位置生成 1. 位置生成
而发射位置要求在球面上均匀分布。由于球面的面积元$\mathrm{d}\Omega=\sin\theta\mathrm{d}\phi\mathrm{d}\theta$是$\theta$的函数，如果位置向量$\vec{r}(\rho, \phi, \theta)$的$\phi$和$\theta$是在$\phi\in[0, 2\pi)$和$\theta\in[0, \pi]$上均匀分布，选取的点将会在两极较为密集（即），因此可以采用`ITM`方法生成$\theta$：$C^{-1}(z)=\cos^{-1}(z)$。 而发射位置要求在球面上均匀分布。由于球面的面积元$\mathrm{d}\Omega=\sin\theta\mathrm{d}\phi\mathrm{d}\theta$是$\theta$的函数，如果位置向量$\vec{r}(\rho, \phi, \theta)$的$\phi$和$\theta$是在$\phi\in[0, 2\pi)$和$\theta\in[0, \pi]$上均匀分布，选取的点将会在两极较为密集（即），因此可以采用`ITM`方法生成$\theta$：$C^{-1}(z)=\cos^{-1}(z)$。
@ -243,6 +235,21 @@ P_z = -\cos\theta
$$ $$
为了保证发射方向是朝向球面内，因此需要限制$\theta$的最大角度为$90^{\circ}$。 为了保证发射方向是朝向球面内，因此需要限制$\theta$的最大角度为$90^{\circ}$。
3. 能量
* `GCR`质子的函数较为简单，采用`ITM`；（下表中$j_0$已经归一化）
<div align=center>
| $F(E)$ | $\int F(E)\mathrm{d}E$ | $C(x)=\int_a^xF(u)\mathrm{d}u$ | $C^{-1}(y)$ |
|:--------------------------:|:------------------------------------------:|:----------------------------------------------------------------------------:|:---------------------------------------------------------------------------------------------------:|
| $j=j_0(E-\beta)^{-\alpha}$ | $\frac{j_0(E-\beta)^{1-\alpha}}{1-\alpha}$ | $\frac{j_0}{1-\alpha}\left((x-\beta)^{1-\alpha}-(a-\beta)^{1-\alpha}\right)$ | $\sqrt[1-\alpha]{\frac{(y+C)(1-\alpha)}{j_0}}+\beta,\ C=\frac{j_0(a-\beta)^{1-\alpha}}{1-\alpha}$ |
</div>
<div align=center><img src="docs/simu-tp.png" style="max-width: 50%;"></div>
* 俘获质子的函数较为复杂，使用`ARM`生成`100000`个随机数需要`12`分钟，效率满足要求；
<div align=center><img src="docs/simu-gcrp.png" style="max-width: 50%;"></div>
* 其余俘获辐射粒子同样使用`ARM`，生成`100000`个随机数需要`4.5`分钟。
<div align=center><img src="docs/simu-gcri.png" style="max-width: 50%;"></div>
## 空间站建模 ## 空间站建模
1. 尺寸与分区[^6] 1. 尺寸与分区[^6]
<div align=center><img src="docs/size.webp" style="max-width: 50%;"></div> <div align=center><img src="docs/size.webp" style="max-width: 50%;"></div>

66
assets/model.txt
View File

@ -1,37 +1,29 @@
# GCR 559.76377, -1.53424, 2.40873, 1.5156509
125.89223, 44.00673, -0.000460574, 112356 125.89223, 44.00673, -0.000460574, 112356, 361.781666
0.01815, 0.000297726, 0.00184, 0.49271 0.01815, 0.000297726, 0.00184, 0.49271, 48.4143048
0.000184794, 0.000389118, 0.0021, 0.49406 0.000184794, 0.000389118, 0.0021, 0.49406, 0.3655716
0.0000953693, 0.000386546, 0.002, 0.48222 0.0000953693, 0.000386546, 0.002, 0.48222, 0.1883217
0.000245835, 0.000383147, 0.00182, 0.45495 0.000245835, 0.000383147, 0.00182, 0.45495, 0.4814399
0.000551454, 0.000297607, 0.00184, 0.49310 0.000551454, 0.000297607, 0.00184, 0.49310, 1.4715932
0.000177397, 0.000347376, 0.00161, 0.43464 0.000177397, 0.000347376, 0.00161, 0.43464, 0.382109
0.000515269, 0.000297659, 0.00184, 0.49294 0.000515269, 0.000297659, 0.00184, 0.49294, 1.3747775
0.0000121025, 0.000307953, 0.00208, 0.53260 0.0000121025, 0.000307953, 0.00208, 0.53260, 0.0316204
0.0000855093, 0.000299579, 0.00188, 0.50022 0.0000855093, 0.000299579, 0.00188, 0.50022, 0.2271729
0.0000200236, 0.000306074, 0.00204, 0.52582 0.0000200236, 0.000306074, 0.00204, 0.52582, 0.0525426
0.000107752, 0.000283677, 0.00197, 0.52149 0.000107752, 0.000283677, 0.00197, 0.52149, 0.3078042
0.0000207575, 0.000304806, 0.00201, 0.52089 0.0000207575, 0.000304806, 0.00201, 0.52089, 0.0546116
0.0000809795, 0.000281074, 0.00191, 0.51232 0.0000809795, 0.000281074, 0.00191, 0.51232, 0.2327364
0.00000442291, 0.000303868, 0.00198, 0.51717 0.00000442291, 0.000303868, 0.00198, 0.51717, 0.0116488
0.0000162654, 0.000281067, 0.00191, 0.51235 0.0000162654, 0.000281067, 0.00191, 0.51235, 0.0467482
0.00000463438, 0.00033962, 0.00178, 0.46921 0.00000463438, 0.00033962, 0.00178, 0.46921, 0.0104706
0.00000956898, 0.000351681, 0.00199, 0.49986 0.00000956898, 0.000351681, 0.00199, 0.49986, 0.0211592
0.00000578307, 0.000336363, 0.00172, 0.45907 0.00000578307, 0.000336363, 0.00172, 0.45907, 0.0131355
0.0000109853, 0.000266092, 0.00154, 0.42213 0.0000109853, 0.000266092, 0.00154, 0.42213, 0.0327014
0.00000291072, 0.000344777, 0.00187, 0.48297 0.00000291072, 0.000344777, 0.00187, 0.48297, 0.006517
0.0000104814, 0.000348052, 0.00193, 0.49105 0.0000104814, 0.000348052, 0.00193, 0.49105, 0.0233399
0.0000051388, 0.000351427, 0.00199, 0.49888 0.0000051388, 0.000351427, 0.00199, 0.49888, 0.0113753
0.0000098941, 0.000346998, 0.00191, 0.48874 0.0000098941, 0.000346998, 0.00191, 0.48874, 0.022068
0.00000723814, 0.000349723, 0.00196, 0.49543 0.00000723814, 0.000349723, 0.00196, 0.49543, 0.0160699
0.0000592702, 0.000279954, 0.00175, 0.45382 0.0000592702, 0.000279954, 0.00175, 0.45382, 0.1696584
0.00000020753, 0.000283367, 0.0018, 0.46005 0.00000020753, 0.000283367, 0.0018, 0.46005, 0.0005883
0.00000284117, 0.000275123, 0.00168, 0.44440 0.00000284117, 0.000275123, 0.00168, 0.44440, 0.0082463
# Trapped Electron
# j0 E0
1333680 0.0824
# Trapped Proton
# j0 alpha beta
559.76377 2.40873 −1.53424

1
debug.txt
View File

@ -1 +0,0 @@
22210

BIN
docs/simu-gcri.png Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 23 KiB

BIN
docs/simu-gcrp.png Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 27 KiB

BIN
docs/simu-tp.png Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 22 KiB

5
include/ActionInitialization.h
View File

@ -3,12 +3,15 @@
#include "G4VUserActionInitialization.hh" #include "G4VUserActionInitialization.hh"
extern G4String particleType;
class ActionInitialization : public G4VUserActionInitialization { class ActionInitialization : public G4VUserActionInitialization {
public: public:
ActionInitialization(); ActionInitialization();
~ActionInitialization() override; ~ActionInitialization() override;
void Build() const override; void BuildForMaster() const override;
void Build() const;
}; };
#endif #endif

69
include/Particle.h Normal file
View File

@ -0,0 +1,69 @@
#ifndef DESCSS_Particle_h
#define DESCSS_Particle_h
#include "G4IonTable.hh"
#include "G4ParticleDefinition.hh"
#include "G4ParticleTable.hh"
#include "globals.hh"
class Particle {
public:
Particle(G4String name, G4int Z, G4int A) {
if (Z == 1)
ion = G4ParticleTable::GetParticleTable()->FindParticle("proton");
else if (Z == 2)
ion = G4ParticleTable::GetParticleTable()->FindParticle("alpha");
else
ion = G4IonTable::GetIonTable()->GetIon(Z, A, 0);
this->name = name;
this->Z = Z, this->A = A;
};
void setC1(G4double v) { C1 = v; };
void setC2(G4double v) { C2 = v; };
void setC3(G4double v) { C3 = v; };
void setC4(G4double v) { C4 = v; };
void setSum(G4double v) { sum = v; };
virtual G4double pdf(G4double E);
protected:
G4ParticleDefinition* ion;
G4int Z, A;
G4String name;
G4double sum;
G4double C1, C2, C3, C4;
};
class TProton : public Particle {
public:
TProton();
G4double pdf(G4double E) {
G4double C = C1 * pow(50 - C2, 1 - C3) / sum / (1 - C3);
return pow(sum * (C + y) * (1 - C3) / C1, 1 / (1 - C3)) + C2;
}
};
class GCRProton : public Particle {
public:
GCRProton();
G4double pdf(G4double E) {
G4double pc = getPc(E);
G4double beta = getBeta(E);
return C1 * pow(beta, C2 - 1) * pow(pc / (pc + C3), C4) / pc / sum;
}
private:
G4double M = 931.5;
G4double getPc(G4double E) { return sqrt(2 * M * E + E * E); }
G4double getBeta(G4double E) {
G4double pc = getPc(E);
return pc / sqrt(M * M + pc * pc);
}
};
class GCRIon : public Particle {
public:
GCRIon();
G4double pdf(G4double E) { return C1 * exp(-C2 * E) * (1 - exp(-C3 * E + C4)) / sum; }
};
#endif

11
include/PrimaryGeneratorAction.h
View File

@ -1,24 +1,27 @@
#ifndef DESCSS_PrimaryGeneratorAction_h #ifndef DESCSS_PrimaryGeneratorAction_h
#define DESCSS_PrimaryGeneratorAction_h #define DESCSS_PrimaryGeneratorAction_h
#include "G4ParticleGun.hh"
#include "G4GeneralParticleSource.hh" #include "G4GeneralParticleSource.hh"
#include "G4ParticleGun.hh"
#include "G4VUserPrimaryGeneratorAction.hh" #include "G4VUserPrimaryGeneratorAction.hh"
#include "globals.hh" #include "globals.hh"
class G4GeneralParticleSource;
class G4ParticleGun; class G4ParticleGun;
class G4Event; class G4Event;
class G4Box;
template <typename T>
class PrimaryGeneratorAction : public G4VUserPrimaryGeneratorAction { class PrimaryGeneratorAction : public G4VUserPrimaryGeneratorAction {
public: public:
PrimaryGeneratorAction(); PrimaryGeneratorAction();
~PrimaryGeneratorAction(); ~PrimaryGeneratorAction();
virtual void GeneratePrimaries(G4Event*); virtual void GeneratePrimaries(G4Event*);
const G4GeneralParticleSource* GetParticleGun() const { return fParticleGun; } const T* GetParticleGun() const { return fParticleGun; }
private: private:
G4GeneralParticleSource* fParticleGun; G4String particleType;
T* fParticleGun = nullptr;
}; };
#endif #endif

6
main.cpp
View File

@ -9,9 +9,15 @@
#include "G4VisExecutive.hh" #include "G4VisExecutive.hh"
#include "QBBC.hh" #include "QBBC.hh"
extern G4String particleType;
int main(int argc, char** argv) { int main(int argc, char** argv) {
G4UIExecutive* ui = nullptr; G4UIExecutive* ui = nullptr;
if (argc == 1) ui = new G4UIExecutive(argc, argv); if (argc == 1) ui = new G4UIExecutive(argc, argv);
if (argc >= 3)
particleType = argv[2];
else
particleType = "TE";
G4MTRunManager* runManager = new G4MTRunManager; G4MTRunManager* runManager = new G4MTRunManager;
G4VisExecutive* visManager = new G4VisExecutive; G4VisExecutive* visManager = new G4VisExecutive;

23
src/ActionInitialization.cpp
View File

@ -1,8 +1,27 @@
#include "ActionInitialization.h" #include "ActionInitialization.h"
#include "PrimaryGeneratorAction.h" #include "PrimaryGeneratorAction.h"
#include <time.h>
ActionInitialization::ActionInitialization() {} #include "Randomize.hh"
G4String particleType;
class G4GeneralParticleSource;
class G4ParticleGun;
ActionInitialization::ActionInitialization() {
G4long seed = time(NULL);
CLHEP::HepRandom::setTheEngine(new CLHEP::RanecuEngine());
CLHEP::HepRandom::setTheSeed(seed);
}
ActionInitialization::~ActionInitialization() {} ActionInitialization::~ActionInitialization() {}
void ActionInitialization::Build() const { SetUserAction(new PrimaryGeneratorAction); } void ActionInitialization::BuildForMaster() const {}
void ActionInitialization::Build() const {
if (particleType == "TE")
SetUserAction(new PrimaryGeneratorAction<G4GeneralParticleSource>);
else
SetUserAction(new PrimaryGeneratorAction<G4ParticleGun>);
}

4
src/DetectorConstruction.cpp
View File

@ -17,8 +17,6 @@ DetectorConstruction::DetectorConstruction() {}
DetectorConstruction::~DetectorConstruction() {} DetectorConstruction::~DetectorConstruction() {}
std::ofstream dataFile("F:/Project/Geant4/DESCSS/debug.txt");
static void ConstructSectionSphere(G4LogicalVolume* fMotherLogical, G4double zBias) { static void ConstructSectionSphere(G4LogicalVolume* fMotherLogical, G4double zBias) {
G4double pRadius = 2.8 * m / 2; G4double pRadius = 2.8 * m / 2;
G4double pRadius2 = 1.2 * m; G4double pRadius2 = 1.2 * m;
@ -255,7 +253,5 @@ G4VPhysicalVolume* DetectorConstruction::Construct() {
// 大柱段 // 大柱段
ConstructSectionBig(logicWorld, 0.815 * m); ConstructSectionBig(logicWorld, 0.815 * m);
dataFile << logicWorld->GetMass() / kg << std::endl;
return physicsWorld; return physicsWorld;
} }

83
src/PrimaryGeneratorAction.cpp
View File

@ -1,9 +1,86 @@
#include "PrimaryGeneratorAction.h" #include "PrimaryGeneratorAction.h"
PrimaryGeneratorAction::PrimaryGeneratorAction() { #include "G4PhysicalConstants.hh"
#include "G4SystemOfUnits.hh"
#include "Randomize.hh"
PrimaryGeneratorAction<G4GeneralParticleSource>::PrimaryGeneratorAction() {
fParticleGun = new G4GeneralParticleSource(); fParticleGun = new G4GeneralParticleSource();
} }
PrimaryGeneratorAction::~PrimaryGeneratorAction() { delete fParticleGun; } PrimaryGeneratorAction<G4ParticleGun>::PrimaryGeneratorAction() {
fParticleGun = new G4ParticleGun();
}
void PrimaryGeneratorAction::GeneratePrimaries(G4Event* e) { fParticleGun->GeneratePrimaryVertex(e); } template <typename T>
PrimaryGeneratorAction<T>::~PrimaryGeneratorAction() {
delete fParticleGun;
}
std::string line;
G4double M = 931.5;
G4double C1, C2, C3, C4, sum;
std::ifstream modelFile("F:/Project/Geant4/DESCSS/assets/model.txt");
G4double randomPhi() {
G4double v = G4UniformRand();
return 2 * pi * v;
}
G4double randomTheta() {
G4double u = G4UniformRand();
return std::acos(2 * u - 1);
}
G4ThreeVector randomPos(G4double rho) {
G4double phi = randomPhi();
G4double theta = randomTheta();
G4double x = rho * std::sin(theta) * std::cos(phi);
G4double y = rho * std::sin(theta) * std::sin(phi);
G4double z = rho * std::cos(theta);
return G4ThreeVector(x, y, z);
}
G4ThreeVector randomDir() {
G4double phi = randomPhi();
G4double theta = randomTheta();
G4double x = -std::sin(theta) * std::cos(phi);
G4double y = -std::sin(theta) * std::sin(phi);
G4double z = -std::cos(theta);
return G4ThreeVector(x, y, z);
}
G4double ITM(G4double Emin, G4double Emax, G4double (*f)(G4double)) {
G4double x = G4UniformRand();
return (*f)(x);
}
G4double ARM(G4double Emin, G4double Emax, G4double (*f)(G4double)) {
G4double x, y;
while (true) {
x = G4UniformRand() * (Emax - Emin) + Emin;
y = G4UniformRand();
if (y <= (*f)(x)) break;
}
return x;
}
void PrimaryGeneratorAction<G4GeneralParticleSource>::GeneratePrimaries(G4Event* e) {
fParticleGun->GeneratePrimaryVertex(e);
}
void PrimaryGeneratorAction<G4ParticleGun>::GeneratePrimaries(G4Event* e) {
G4double E = 0;
G4double R = 15. * m;
G4ThreeVector pos = G4ThreeVector(); // randomPos(R);
G4ThreeVector dir = G4ThreeVector(); // randomDir();
fParticleGun->SetParticleEnergy(E);
fParticleGun->SetParticlePosition(pos);
fParticleGun->SetParticleMomentumDirection(dir);
fParticleGun->GeneratePrimaryVertex(e);
}

83
utils/check.py Normal file
View File

@ -0,0 +1,83 @@
import math
import numpy as np
from matplotlib import pyplot as plt
def trap_proton():
def pdf(E, C1=559.76377, C2=-1.53424, C3=2.40873, sum=1.51565):
return C1 * pow(E - C2, -C3) / sum
pdf = np.vectorize(pdf)
def pdf_int(E, C1=559.76377, C2=-1.53424, C3=2.40873, sum=1.5156509):
return C1 * math.pow(E - C2, 1 - C3) / (1 - C3) / sum
def pdf_sum(E):
return pdf_int(E) - pdf_int(50)
def pdf_inv(y, C1=559.76377, C2=-1.53424, C3=2.40873, sum=1.5156509):
return pow(sum * (pdf_int(50) + y) * (1 - C3) / C1, 1 / (1 - C3)) + C2
with open('debug.txt', 'r') as f:
data = f.readlines()
data = np.array([float(s) for s in data[0].split(', ')[:-1]])
x = np.linspace(50, 1000, 1000)
plt.plot(x, pdf(x), label="f(x)")
plt.hist(data, bins=np.linspace(50, 1000, 500), density=True, histtype='step', label="Simulation")
plt.legend()
plt.ylabel('Frequency')
plt.xlabel('Energy/(MeV)')
plt.savefig('docs/simu-tp.png', bbox_inches='tight')
def gcr_proton():
M = 931.5
def get_pc(E):
return math.sqrt(2 * M * E + E * E)
def get_beta(E):
pc = get_pc(E)
return pc / math.sqrt(M * M + pc * pc)
def pdf(E, C1=125.89223, alpha=44.00673, C2=-0.000460574, C3=112356, sum=361.781665965026):
pc = get_pc(E)
beta = get_beta(E)
return C1 * math.pow(beta, alpha - 1) * math.pow(pc / (pc + C2), C3) / pc / sum
pdf = np.vectorize(pdf)
with open('debug.txt', 'r') as f:
data = f.readlines()
data = np.array([float(s) for s in data[0].split(', ')[:-1]])
x = np.linspace(220, 1e5, 10000)
plt.plot(x, pdf(x), label="f(x)")
plt.hist(data, bins=np.linspace(220, 1e5, 100), density=True, histtype='step', label="Simulation")
plt.legend()
plt.ylabel('Frequency')
plt.xlabel('Energy/(MeV)')
plt.savefig('docs/simu-gcrp.png', bbox_inches='tight')
def gcr_ion():
def pdf(E, C1=0.01815, C2=0.000297726, C3=0.00184, C4=0.49271, sum=48.4143048):
return C1 * math.exp(-C2 * E) * (1 - math.exp(-C3 * E + C4)) / sum
pdf = np.vectorize(pdf)
with open('debug.txt', 'r') as f:
data = f.readlines()
data = np.array([float(s) for s in data[0].split(', ')[:-1]])
x = np.linspace(220, 1e5, 10000)
plt.plot(x, pdf(x), label="f(x)")
plt.hist(data, bins=np.linspace(220, 1e5, 300), density=True, histtype='step', label="Simulation")
plt.legend()
plt.ylabel('Frequency')
plt.xlabel('Energy/(MeV)')
plt.savefig('docs/simu-gcri.png', bbox_inches='tight')
trap_proton()
gcr_proton()
gcr_ion()