csiborgtools/notebooks/test_cosma.ipynb

1306 lines
116 KiB
Plaintext
Raw Normal View History

{
"cells": [
{
"cell_type": "code",
"execution_count": 1,
"id": "5a38ed25",
"metadata": {
"ExecuteTime": {
"end_time": "2023-03-24T14:16:01.928614Z",
"start_time": "2023-03-24T14:15:34.242247Z"
},
"scrolled": true
},
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"../csiborgtools/field/__init__.py:20: UserWarning: MAS_library not found, `DensityField` will not be available\n",
" warn(\"MAS_library not found, `DensityField` will not be available\", UserWarning) # noqa\n"
]
}
],
"source": [
"import sys\n",
"from os.path import join\n",
"import numpy as np\n",
"import matplotlib.pyplot as plt\n",
"from numba import jit\n",
"from tqdm import tqdm, trange\n",
"import joblib\n",
"sys.path.append(\"../\")\n",
"import csiborgtools\n",
"\n",
"%matplotlib widget\n",
"%load_ext autoreload\n",
"%autoreload 2"
]
},
{
"cell_type": "code",
"execution_count": 2,
"id": "6f3ae40e",
"metadata": {},
"outputs": [],
"source": [
"srcdir = \"/cosma8/data/dp016/dc-stis1/csiborg_new\"\n",
"dumpdir = \"/cosma8/data/dp016/dc-stis1/csiborg_dump\"\n",
"paths = csiborgtools.read.CSiBORGPaths(srcdir=srcdir, dumpdir=dumpdir)\n",
"\n",
"\n",
"reader = csiborgtools.read.ParticleReader(paths)"
]
},
{
"cell_type": "code",
"execution_count": 3,
"id": "33abfe57",
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Reading in output `00001` with ncpu = `324`.\n",
"Opened 324 particle files.\n"
]
},
{
"name": "stderr",
"output_type": "stream",
"text": [
"100%|██████████| 324/324 [03:57<00:00, 1.36it/s]\n"
]
}
],
"source": [
"part = reader.read_particle(1, 7444, [\"x\", \"y\", \"z\", \"M\", \"ID\"], verbose=True)"
]
},
{
"cell_type": "code",
"execution_count": 4,
"id": "32ab40ee",
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"ls: cannot access /cosma8/data/dp016/dc-stis1/csiborg_new/ramsess_out_7444/outputs_00001/: No such file or directory\n"
]
}
],
"source": [
"/cosma8/data/dp016/dc-stis1/csiborg_new/ramses_out_7444_new/output_00001/info_00001.txt"
]
},
{
"cell_type": "code",
"execution_count": 9,
"id": "fc116dff",
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"cat: /cosma8/data/dp016/dc-stis1/csiborg_new/ramses_out_7444/output_00001/info_00001.txt: No such file or directory\n"
]
}
],
"source": [
"!cat /cosma8/data/dp016/dc-stis1/csiborg_new/ramses_out_7444/output_00001/info_00001.txt"
]
},
{
"cell_type": "code",
"execution_count": 8,
"id": "e6b05ceb",
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"ncpu = 324\n",
"ndim = 3\n",
"levelmin = 8\n",
"levelmax = 19\n",
"ngridmax = 2800000\n",
"nstep_coarse= 0\n",
"\n",
"boxlen = 0.100000000000000E+01\n",
"time = -0.260788354371079E+02\n",
"aexp = 0.142857000000000E-01\n",
"H0 = 0.705000000000000E+02\n",
"omega_m = 0.307000011205673E+00\n",
"omega_l = 0.693000018596649E+00\n",
"omega_k = -0.298023223876953E-07\n",
"omega_b = 0.000000000000000E+00\n",
"unit_l = 0.423740962969913E+26\n",
"unit_d = 0.983942902810118E-24\n",
"unit_t = 0.893232429772148E+14\n",
"\n",
"ordering type=hilbert \n",
" DOMAIN ind_min ind_max\n",
" 1 0.000000000000000E+00 0.890921902481080E+17\n",
" 2 0.890921902481080E+17 0.903741664479150E+17\n",
" 3 0.903741664479150E+17 0.905655754235576E+17\n",
" 4 0.905655754235576E+17 0.907569842649825E+17\n",
" 5 0.907569842649825E+17 0.910375727335997E+17\n",
" 6 0.910375727335997E+17 0.916149055929713E+17\n",
" 7 0.916149055929713E+17 0.918096761858294E+17\n",
" 8 0.918096761858294E+17 0.920010850272543E+17\n",
" 9 0.920010850272543E+17 0.942688305781146E+17\n",
" 10 0.942688305781146E+17 0.945672973849723E+17\n",
" 11 0.945672973849723E+17 0.947587062263972E+17\n",
" 12 0.947587062263972E+17 0.949501751973642E+17\n",
" 13 0.949501751973642E+17 0.955129510703923E+17\n",
" 14 0.955129510703923E+17 0.958726141818962E+17\n",
" 15 0.958726141818962E+17 0.963097707172332E+17\n",
" 16 0.963097707172332E+17 0.965012398224179E+17\n",
" 17 0.965012398224179E+17 0.966926486638428E+17\n",
" 18 0.966926486638428E+17 0.968840576394854E+17\n",
" 19 0.968840576394854E+17 0.970754664809103E+17\n",
" 20 0.970754664809103E+17 0.972668753223352E+17\n",
" 21 0.972668753223352E+17 0.974582842979779E+17\n",
" 22 0.974582842979779E+17 0.976496931394028E+17\n",
" 23 0.976496931394028E+17 0.978411019808276E+17\n",
" 24 0.978411019808276E+17 0.980325109564703E+17\n",
" 25 0.980325109564703E+17 0.982239197978952E+17\n",
" 26 0.982239197978952E+17 0.984153286393201E+17\n",
" 27 0.984153286393201E+17 0.986067376149627E+17\n",
" 28 0.986067376149627E+17 0.987981464563876E+17\n",
" 29 0.987981464563876E+17 0.989895554320302E+17\n",
" 30 0.989895554320302E+17 0.991810244029972E+17\n",
" 31 0.991810244029972E+17 0.993724332444221E+17\n",
" 32 0.993724332444221E+17 0.995638422200648E+17\n",
" 33 0.995638422200648E+17 0.997552510614897E+17\n",
" 34 0.997552510614897E+17 0.999467200324567E+17\n",
" 35 0.999467200324567E+17 0.100143568852615E+18\n",
" 36 0.100143568852615E+18 0.100840961751384E+18\n",
" 37 0.100840961751384E+18 0.101193427034571E+18\n",
" 38 0.101193427034571E+18 0.101812721724424E+18\n",
" 39 0.101812721724424E+18 0.107245430957408E+18\n",
" 40 0.107245430957408E+18 0.115866535335035E+18\n",
" 41 0.115866535335035E+18 0.204958656863666E+18\n",
" 42 0.204958656863666E+18 0.252936659161580E+18\n",
" 43 0.252936659161580E+18 0.257043514674643E+18\n",
" 44 0.257043514674643E+18 0.258590894486192E+18\n",
" 45 0.258590894486192E+18 0.259300492205621E+18\n",
" 46 0.259300492205621E+18 0.259620404321059E+18\n",
" 47 0.259620404321059E+18 0.260240103140491E+18\n",
" 48 0.260240103140491E+18 0.260431693444284E+18\n",
" 49 0.260431693444284E+18 0.260623102285709E+18\n",
" 50 0.260623102285709E+18 0.260814511127134E+18\n",
" 51 0.260814511127134E+18 0.261005920102777E+18\n",
" 52 0.261005920102777E+18 0.261197381557551E+18\n",
" 53 0.261197381557551E+18 0.261388790398976E+18\n",
" 54 0.261388790398976E+18 0.261580199374619E+18\n",
" 55 0.261580199374619E+18 0.261771608216044E+18\n",
" 56 0.261771608216044E+18 0.261963017057468E+18\n",
" 57 0.261963017057468E+18 0.262154426033111E+18\n",
" 58 0.262154426033111E+18 0.262345834874536E+18\n",
" 59 0.262345834874536E+18 0.262537243850179E+18\n",
" 60 0.262537243850179E+18 0.262728652691603E+18\n",
" 61 0.262728652691603E+18 0.262920061533028E+18\n",
" 62 0.262920061533028E+18 0.263111470508671E+18\n",
" 63 0.263111470508671E+18 0.263302879350096E+18\n",
" 64 0.263302879350096E+18 0.263494288191521E+18\n",
" 65 0.263494288191521E+18 0.263685697167163E+18\n",
" 66 0.263685697167163E+18 0.263877106008588E+18\n",
" 67 0.263877106008588E+18 0.264074332383216E+18\n",
" 68 0.264074332383216E+18 0.264505062976389E+18\n",
" 69 0.264505062976389E+18 0.265212269338558E+18\n",
" 70 0.265212269338558E+18 0.265425398098035E+18\n",
" 71 0.265425398098035E+18 0.265616859552809E+18\n",
" 72 0.265616859552809E+18 0.265859579026342E+18\n",
" 73 0.265859579026342E+18 0.268130852221747E+18\n",
" 74 0.268130852221747E+18 0.268372849603903E+18\n",
" 75 0.268372849603903E+18 0.268567500742984E+18\n",
" 76 0.268567500742984E+18 0.268759689523626E+18\n",
" 77 0.268759689523626E+18 0.269423404241650E+18\n",
" 78 0.269423404241650E+18 0.269614813217292E+18\n",
" 79 0.269614813217292E+18 0.269806222058717E+18\n",
" 80 0.269806222058717E+18 0.269997630900142E+18\n",
" 81 0.269997630900142E+18 0.306265813826929E+18\n",
" 82 0.306265813826929E+18 0.306556648074248E+18\n",
" 83 0.306556648074248E+18 0.306748056915673E+18\n",
" 84 0.306748056915673E+18 0.306939465757098E+18\n",
" 85 0.306939465757098E+18 0.307221961728066E+18\n",
" 86 0.307221961728066E+18 0.307794217399222E+18\n",
" 87 0.307794217399222E+18 0.307988715395875E+18\n",
" 88 0.307988715395875E+18 0.308180124237300E+18\n",
" 89 0.308180124237300E+18 0.310444717550600E+18\n",
" 90 0.310444717550600E+18 0.310740185664979E+18\n",
" 91 0.310740185664979E+18 0.310931594506404E+18\n",
" 92 0.310931594506404E+18 0.311123041063010E+18\n",
" 93 0.311123041063010E+18 0.311680816251929E+18\n",
" 94 0.311680816251929E+18 0.312038281548136E+18\n",
" 95 0.312038281548136E+18 0.312471011985457E+18\n",
" 96 0.312471011985457E+18 0.312662450891653E+18\n",
" 97 0.312662450891653E+18 0.312853859867296E+18\n",
" 98 0.312853859867296E+18 0.313045268708721E+18\n",
" 99 0.313045268708721E+18 0.313236677684363E+18\n",
" 100 0.313236677684363E+18 0.313428086525788E+18\n",
" 101 0.313428086525788E+18 0.313619495367213E+18\n",
" 102 0.313619495367213E+18 0.313810904342856E+18\n",
" 103 0.313810904342856E+18 0.314002313184281E+18\n",
" 104 0.314002313184281E+18 0.314193722025705E+18\n",
" 105 0.314193722025705E+18 0.314385131001348E+18\n",
" 106 0.314385131001348E+18 0.314576539842773E+18\n",
" 107 0.314576539842773E+18 0.314767948684198E+18\n",
" 108 0.314767948684198E+18 0.314959357659841E+18\n",
" 109 0.314959357659841E+18 0.315150766501265E+18\n",
" 110 0.315150766501265E+18 0.315342175476908E+18\n",
" 111 0.315342175476908E+18 0.315533614383104E+18\n",
" 112 0.315533614383104E+18 0.315725023224529E+18\n",
" 113 0.315725023224529E+18 0.315916432200172E+18\n",
" 114 0.315916432200172E+18 0.316107878622560E+18\n",
" 115 0.316107878622560E+18 0.316302087245791E+18\n",
" 116 0.316302087245791E+18 0.316995494012256E+18\n",
" 117 0.316995494012256E+18 0.317341733203476E+18\n",
" 118 0.317341733203476E+18 0.317950474621813E+18\n",
" 119 0.317950474621813E+18 0.322765279123735E+18\n",
" 120 0.322765279123735E+18 0.323602548101480E+18\n",
" 121 0.323602548101480E+18 0.404285202853003E+18\n",
" 122 0.404285202853003E+18 0.493377324381635E+18\n",
" 123 0.493377324381635E+18 0.505798757690900E+18\n",
" 124 0.505798757690900E+18 0.506944051119915E+18\n",
" 125 0.506944051119915E+18 0.507135512574689E+18\n",
" 126 0.507135512574689E+18 0.507326921550332E+18\n",
" 127 0.507326921550332E+18 0.507518330391757E+18\n",
" 128 0.507518330391757E+18 0.507896890285621E+18\n",
" 129 0.507896890285621E+18 0.508206872973541E+18\n",
" 130 0.508206872973541E+18 0.508401475257369E+18\n",
" 131 0.508401475257369E+18 0.508600811534680E+18\n",
" 132 0.508600811534680E+18 0.509602348381766E+18\n",
" 133 0.509602348381766E+18 0.514502795336876E+18\n",
" 134 0.514502795336876E+18 0.515690615787225E+18\n",
" 135 0.515690615787225E+18 0.516460272584491E+18\n",
" 136 0.516460272584491E+18 0.516797047278404E+18\n",
" 137 0.516797047278404E+18 0.516988456254046E+18\n",
" 138 0.516988456254046E+18 0.517179865095471E+18\n",
" 139 0.517179865095471E+18 0.517371326550245E+18\n",
" 140 0.517371326550245E+18 0.517587226905805E+18\n",
" 141 0.517587226905805E+18 0.517778688494797E+18\n",
" 142 0.517778688494797E+18 0.517970097336222E+18\n",
" 143 0.517970097336222E+18 0.518161506177647E+18\n",
" 144 0.518161506177647E+18 0.518352915153289E+18\n",
" 145 0.518352915153289E+18 0.518544323994714E+18\n",
" 146 0.518544323994714E+18 0.518735732970357E+18\n",
" 147 0.518735732970357E+18 0.518927141811782E+18\n",
" 148 0.518927141811782E+18 0.519118550653207E+18\n",
" 149 0.519118550653207E+18 0.519309959628849E+18\n",
" 150 0.519309959628849E+18 0.519501368470274E+18\n",
" 151 0.519501368470274E+18 0.519692777311699E+18\n",
" 152 0.519692777311699E+18 0.519884186287342E+18\n",
" 153 0.519884186287342E+18 0.520075595128766E+18\n",
" 154 0.520075595128766E+18 0.520267003970191E+18\n",
" 155 0.520267003970191E+18 0.520458412945834E+18\n",
" 156 0.520458412945834E+18 0.520649874400608E+18\n",
" 157 0.520649874400608E+18 0.520844424205304E+18\n",
" 158 0.520844424205304E+18 0.521057601685815E+18\n",
" 159 0.521057601685815E+18 0.521249010661458E+18\n",
" 160 0.521249010661458E+18 0.521948424208122E+18\n",
" 161 0.521948424208122E+18 0.522338053038014E+18\n",
" 162 0.522338053038014E+18 0.598485619475218E+18\n",
" 163 0.598485619475218E+18 0.630677923535258E+18\n",
" 164 0.630677923535258E+18 0.631528434165613E+18\n",
" 165 0.631528434165613E+18 0.631765173702492E+18\n",
" 166 0.631765173702492E+18 0.631956582543917E+18\n",
" 167 0.631956582543917E+18 0.632172306134729E+18\n",
" 168 0.632172306134729E+18 0.632363767589503E+18\n",
" 169 0.632363767589503E+18 0.632555176430928E+18\n",
" 170 0.632555176430928E+18 0.632746585406570E+18\n",
" 171 0.632746585406570E+18 0.632937994247995E+18\n",
" 172 0.632937994247995E+18 0.633129403089420E+18\n",
" 173 0.633129403089420E+18 0.633320812065063E+18\n",
" 174 0.633320812065063E+18 0.633512220906488E+18\n",
" 175 0.633512220906488E+18 0.633703629747913E+18\n",
" 176 0.633703629747913E+18 0.633895038723555E+18\n",
" 177 0.633895038723555E+18 0.634086447564980E+18\n",
" 178 0.634086447564980E+18 0.634277856540623E+18\n",
" 179 0.634277856540623E+18 0.634469265382048E+18\n",
" 180 0.634469265382048E+18 0.634660674223473E+18\n",
" 181 0.634660674223473E+18 0.634852083199115E+18\n",
" 182 0.634852083199115E+18 0.635043492040540E+18\n",
" 183 0.635043492040540E+18 0.635234938462929E+18\n",
" 184 0.635234938462929E+18 0.635447299497001E+18\n",
" 185 0.635447299497001E+18 0.635641591737876E+18\n",
" 186 0.635641591737876E+18 0.635833030778290E+18\n",
" 187 0.635833030778290E+18 0.636024439619715E+18\n",
" 188 0.636024439619715E+18 0.636215848461140E+18\n",
" 189 0.636215848461140E+18 0.636750937767018E+18\n",
" 190 0.636750937767018E+18 0.638195828250378E+18\n",
" 191 0.638195828250378E+18 0.643163132008595E+18\n",
" 192 0.643163132008595E+18 0.643481123736781E+18\n",
" 193 0.643481123736781E+18 0.644410277365809E+18\n",
" 194 0.644410277365809E+18 0.644604712548565E+18\n",
" 195 0.644604712548565E+18 0.644796121389990E+18\n",
" 196 0.644796121389990E+18 0.645181291506958E+18\n",
" 197 0.645181291506958E+18 0.645480259617030E+18\n",
" 198 0.645480259617030E+18 0.645671668458455E+18\n",
" 199 0.645671668458455E+18 0.645863077434098E+18\n",
" 200 0.645863077434098E+18 0.646078413001458E+18\n",
" 201 0.646078413001458E+18 0.647196133008867E+18\n",
" 202 0.647196133008867E+18 0.692703801651495E+18\n",
" 203 0.692703801651495E+18 0.781795991899603E+18\n",
" 204 0.781795991899603E+18 0.829396574483251E+18\n",
" 205 0.829396574483251E+18 0.833478057611231E+18\n",
" 206 0.833478057611231E+18 0.835041939358220E+18\n",
" 207 0.835041939358220E+18 0.835748986135511E+18\n",
" 208 0.835748986135511E+18 0.836066892501221E+18\n",
" 209 0.836066892501221E+18 0.836684653485097E+18\n",
" 210 0.836684653485097E+18 0.836876197752209E+18\n",
" 211 0.836876197752209E+18 0.837067606593634E+18\n",
" 212 0.837067606593634E+18 0.837259015435059E+18\n",
" 213 0.837259015435059E+18 0.837450424410702E+18\n",
" 214 0.837450424410702E+18 0.837641863316898E+18\n",
" 215 0.837641863316898E+18 0.837833272158323E+18\n",
" 216 0.837833272158323E+18 0.838024681133965E+18\n",
" 217 0.838024681133965E+18 0.838216089975390E+18\n",
" 218 0.838216089975390E+18 0.838407498951033E+18\n",
" 219 0.838407498951033E+18 0.838598907792458E+18\n",
" 220 0.838598907792458E+18 0.838790316633883E+18\n",
" 221 0.838790316633883E+18 0.838981725609525E+18\n",
" 222 0.838981725609525E+18 0.839173134450950E+18\n",
" 223 0.839173134450950E+18 0.839364543292375E+18\n",
" 224 0.839364543292375E+18 0.839555952268018E+18\n",
" 225 0.839555952268018E+18 0.839747361109443E+18\n",
" 226 0.839747361109443E+18 0.839938769950867E+18\n",
" 227 0.839938769950867E+18 0.840130178926510E+18\n",
" 228 0.840130178926510E+18 0.840321587767935E+18\n",
" 229 0.840321587767935E+18 0.840515962821149E+18\n",
" 230 0.840515962821149E+18 0.840945666917138E+18\n",
" 231 0.840945666917138E+18 0.841646435928637E+18\n",
" 232 0.841646435928637E+18 0.841858285861601E+18\n",
" 233 0.841858285861601E+18 0.842049724767797E+18\n",
" 234 0.842049724767797E+18 0.842281563210121E+18\n",
" 235 0.842281563210121E+18 0.842722369393918E+18\n",
" 236 0.842722369393918E+18 0.844794685611311E+18\n",
" 237 0.844794685611311E+18 0.844986094452736E+18\n",
" 238 0.844986094452736E+18 0.845180402799739E+18\n",
" 239 0.845180402799739E+18 0.845839342990524E+18\n",
" 240 0.845839342990524E+18 0.846030781896720E+18\n",
" 241 0.846030781896720E+18 0.846222190738145E+18\n",
" 242 0.846222190738145E+18 0.846413599713788E+18\n",
" 243 0.846413599713788E+18 0.864502974527832E+18\n",
" 244 0.864502974527832E+18 0.882968182065725E+18\n",
" 245 0.882968182065725E+18 0.883159590907150E+18\n",
" 246 0.883159590907150E+18 0.883350999882793E+18\n",
" 247 0.883350999882793E+18 0.883542438788989E+18\n",
" 248 0.883542438788989E+18 0.884202639686894E+18\n",
" 249 0.884202639686894E+18 0.884397114195444E+18\n",
" 250 0.884397114195444E+18 0.884588523036869E+18\n",
" 251 0.884588523036869E+18 0.886662527579062E+18\n",
" 252 0.886662527579062E+18 0.887140268334121E+18\n",
" 253 0.887140268334121E+18 0.887338684146254E+18\n",
" 254 0.887338684146254E+18 0.887530145735246E+18\n",
" 255 0.887530145735246E+18 0.887743060014793E+18\n",
" 256 0.887743060014793E+18 0.888448298610852E+18\n",
" 257 0.888448298610852E+18 0.888883030502932E+18\n",
" 258 0.888883030502932E+18 0.889075884332417E+18\n",
" 259 0.889075884332417E+18 0.889267293308060E+18\n",
" 260 0.889267293308060E+18 0.889458702149485E+18\n",
" 261 0.889458702149485E+18 0.889650110990909E+18\n",
" 262 0.889650110990909E+18 0.889841519966552E+18\n",
" 263 0.889841519966552E+18 0.890032928807977E+18\n",
" 264 0.890032928807977E+18 0.890224337783620E+18\n",
" 265 0.890224337783620E+18 0.890415746625044E+18\n",
" 266 0.890415746625044E+18 0.890607155466469E+18\n",
" 267 0.890607155466469E+18 0.890798564442112E+18\n",
" 268 0.890798564442112E+18 0.890989973283537E+18\n",
" 269 0.890989973283537E+18 0.891181382124962E+18\n",
" 270 0.891181382124962E+18 0.891372791100604E+18\n",
" 271 0.891372791100604E+18 0.891564199942029E+18\n",
" 272 0.891564199942029E+18 0.891755608783454E+18\n",
" 273 0.891755608783454E+18 0.891947055340061E+18\n",
" 274 0.891947055340061E+18 0.892138464181486E+18\n",
" 275 0.892138464181486E+18 0.892329873022910E+18\n",
" 276 0.892329873022910E+18 0.892521312063324E+18\n",
" 277 0.892521312063324E+18 0.892712833647641E+18\n",
" 278 0.892712833647641E+18 0.893333954101248E+18\n",
" 279 0.893333954101248E+18 0.893653827830415E+18\n",
" 280 0.893653827830415E+18 0.894359090988319E+18\n",
" 281 0.894359090988319E+18 0.899130547593282E+18\n",
" 282 0.899130547593282E+18 0.900009270521627E+18\n",
" 283 0.900009270521627E+18 0.958721087984632E+18\n",
" 284 0.958721087984632E+18 0.104545954810770E+19\n",
" 285 0.104545954810770E+19 0.104631566863643E+19\n",
" 286 0.104631566863643E+19 0.105113412305525E+19\n",
" 287 0.105113412305525E+19 0.105174324551772E+19\n",
" 288 0.105174324551772E+19 0.105209241884269E+19\n",
" 289 0.105209241884269E+19 0.105278706349926E+19\n",
" 290 0.105278706349926E+19 0.105298166296451E+19\n",
" 291 0.105298166296451E+19 0.105317312455351E+19\n",
" 292 0.105317312455351E+19 0.105336453339493E+19\n",
" 293 0.105336453339493E+19 0.105355594223636E+19\n",
" 294 0.105355594223636E+19 0.105374735121200E+19\n",
" 295 0.105374735121200E+19 0.105393881266677E+19\n",
" 296 0.105393881266677E+19 0.105413022150820E+19\n",
" 297 0.105413022150820E+19 0.105432163048384E+19\n",
" 298 0.105432163048384E+19 0.105451303932527E+19\n",
" 299 0.105451303932527E+19 0.105470444830091E+19\n",
" 300 0.105470444830091E+19 0.105489585714233E+19\n",
" 301 0.105489585714233E+19 0.105508726598376E+19\n",
" 302 0.105508726598376E+19 0.105527867495940E+19\n",
" 303 0.105527867495940E+19 0.105547008380083E+19\n",
" 304 0.105547008380083E+19 0.105566149264225E+19\n",
" 305 0.105566149264225E+19 0.105585290161789E+19\n",
" 306 0.105585290161789E+19 0.105604431045932E+19\n",
" 307 0.105604431045932E+19 0.105623571930074E+19\n",
" 308 0.105623571930074E+19 0.105642712827639E+19\n",
" 309 0.105642712827639E+19 0.105661858973116E+19\n",
" 310 0.105661858973116E+19 0.105705459105451E+19\n",
" 311 0.105705459105451E+19 0.105742519707160E+19\n",
" 312 0.105742519707160E+19 0.105797542036583E+19\n",
" 313 0.105797542036583E+19 0.105816688182061E+19\n",
" 314 0.105816688182061E+19 0.105835829066203E+19\n",
" 315 0.105835829066203E+19 0.105865633938067E+19\n",
" 316 0.105865633938067E+19 0.106092287187170E+19\n",
" 317 0.106092287187170E+19 0.106111428071313E+19\n",
" 318 0.106111428071313E+19 0.106130893198642E+19\n",
" 319 0.106130893198642E+19 0.106188462242346E+19\n",
" 320 0.106188462242346E+19 0.106216483535087E+19\n",
" 321 0.106216483535087E+19 0.106235624432651E+19\n",
" 322 0.106235624432651E+19 0.106254765316794E+19\n",
" 323 0.106254765316794E+19 0.106382938307822E+19\n",
" 324 0.106382938307822E+19 0.115292150460685E+19\n"
]
}
],
"source": [
"!cat /cosma8/data/dp016/dc-stis1/csiborg_new/ramses_out_7444_new/output_00001/info_00001.txt"
]
},
{
"cell_type": "code",
"execution_count": 10,
"id": "dc8ade85",
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
"'/cosma8/data/dp016/dc-stis1/csiborg_dump'"
]
},
"execution_count": 10,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"paths.dumpdir\n"
]
},
{
"cell_type": "code",
"execution_count": 8,
"id": "1891c6ff",
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
"101"
]
},
"execution_count": 8,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"len(paths.get_ics(tonew=True))"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "7101a155",
"metadata": {},
"outputs": [],
"source": [
" fname = \"ramses_out_{}_{}.npy\".format(\n",
" str(self.nsim).zfill(5), str(self.nsnap).zfill(5))\n",
" data = numpy.load(join(self.paths.dumpdir, fname))\n"
]
},
{
"cell_type": "code",
"execution_count": 2,
"id": "190d39e6",
"metadata": {
"ExecuteTime": {
"end_time": "2023-03-24T14:16:12.485845Z",
"start_time": "2023-03-24T14:16:01.930739Z"
}
},
"outputs": [],
"source": [
"cat0 = csiborgtools.read.ClumpsCatalogue(7468)\n",
"catx = csiborgtools.read.ClumpsCatalogue(7588)"
]
},
{
"cell_type": "code",
"execution_count": 3,
"id": "09c93ab0",
"metadata": {
"ExecuteTime": {
"end_time": "2023-03-24T14:16:31.435607Z",
"start_time": "2023-03-24T14:16:12.487458Z"
},
"scrolled": true
},
"outputs": [],
"source": [
"reader = csiborgtools.read.PairOverlap(cat0, catx, max_dist=150 / 0.705)"
]
},
{
"cell_type": "code",
"execution_count": 31,
"id": "650cbe8a",
"metadata": {
"ExecuteTime": {
"end_time": "2023-03-24T14:27:04.213308Z",
"start_time": "2023-03-24T14:27:00.679174Z"
},
"scrolled": false
},
"outputs": [
{
"data": {
"application/javascript": "/* Put everything inside the global mpl namespace */\n/* global mpl */\nwindow.mpl = {};\n\nmpl.get_websocket_type = function () {\n if (typeof WebSocket !== 'undefined') {\n return WebSocket;\n } else if (typeof MozWebSocket !== 'undefined') {\n return MozWebSocket;\n } else {\n alert(\n 'Your browser does not have WebSocket support. ' +\n 'Please try Chrome, Safari or Firefox ≥ 6. ' +\n 'Firefox 4 and 5 are also supported but you ' +\n 'have to enable WebSockets in about:config.'\n );\n }\n};\n\nmpl.figure = function (figure_id, websocket, ondownload, parent_element) {\n this.id = figure_id;\n\n this.ws = websocket;\n\n this.supports_binary = this.ws.binaryType !== undefined;\n\n if (!this.supports_binary) {\n var warnings = document.getElementById('mpl-warnings');\n if (warnings) {\n warnings.style.display = 'block';\n warnings.textContent =\n 'This browser does not support binary websocket messages. ' +\n 'Performance may be slow.';\n }\n }\n\n this.imageObj = new Image();\n\n this.context = undefined;\n this.message = undefined;\n this.canvas = undefined;\n this.rubberband_canvas = undefined;\n this.rubberband_context = undefined;\n this.format_dropdown = undefined;\n\n this.image_mode = 'full';\n\n this.root = document.createElement('div');\n this.root.setAttribute('style', 'display: inline-block');\n this._root_extra_style(this.root);\n\n parent_element.appendChild(this.root);\n\n this._init_header(this);\n this._init_canvas(this);\n this._init_toolbar(this);\n\n var fig = this;\n\n this.waiting = false;\n\n this.ws.onopen = function () {\n fig.send_message('supports_binary', { value: fig.supports_binary });\n fig.send_message('send_image_mode', {});\n if (fig.ratio !== 1) {\n fig.send_message('set_device_pixel_ratio', {\n device_pixel_ratio: fig.ratio,\n });\n }\n fig.send_message('refresh', {});\n };\n\n this.imageObj.onload = function () {\n if (fig.image_mode === 'full') {\n // Full images could contain transparency (where diff images\n // almost always do), so we need to clear the canvas so that\n // there is no ghosting.\n fig.context.clearRect(0, 0, fig.canvas.width, fig.canvas.height);\n }\n fig.context.drawImage(fig.imageObj, 0, 0);\n };\n\n this.imageObj.onunload = function () {\n fig.ws.close();\n };\n\n this.ws.onmessage = this._make_on_message_function(this);\n\n this.ondownload = ondownload;\n};\n\nmpl.figure.prototype._init_header = function () {\n var titlebar = document.createElement('div');\n titlebar.classList =\n 'ui-dialog-titlebar ui-widget-header ui-corner-all ui-helper-clearfix';\n var titletext = document.createElement('div');\n titletext.classList = 'ui-dialog-title';\n titletext.setAttribute(\n 'style',\n 'width: 100%; text-align: center; padding: 3px;'\n );\n titlebar.appendChild(titletext);\n this.root.appendChild(titlebar);\n this.header = titletext;\n};\n\nmpl.figure.prototype._canvas_extra_style = function (_canvas_div) {};\n\nmpl.figure.prototype._root_extra_style = function (_canvas_div) {};\n\nmpl.figure.prototype._init_canvas = function () {\n var fig = this;\n\n var canvas_div = (this.canvas_div = document.createElement('div'));\n canvas_div.setAttribute(\n 'style',\n 'border: 1px solid #ddd;' +\n 'box-sizing: content-box;' +\n 'clear: both;' +\n 'min-height: 1px;' +\n 'min-width: 1px;' +\n 'outline: 0;' +\n 'overflow: hidden;' +\n 'position: relative;' +\n 'resize: both;'\n );\n\n function on_keyboard_event_closure(name) {\n return function (event) {\n return fig.key
"text/plain": [
"<IPython.core.display.Javascript object>"
]
},
"metadata": {},
"output_type": "display_data"
},
{
"data": {
"text/html": [
"<img src=\"data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAoAAAAHgCAYAAAA10dzkAAAAAXNSR0IArs4c6QAAAERlWElmTU0AKgAAAAgAAYdpAAQAAAABAAAAGgAAAAAAA6ABAAMAAAABAAEAAKACAAQAAAABAAACgKADAAQAAAABAAAB4AAAAAAfNMscAABAAElEQVR4AezdB5wU5d3A8f/d0cGjSEekiFJEiiAISsCIIBKVxALECBIleW1RsUFUQEVBUcSCoCjBhqBR0ViwoKgIioDYRUCQIlXpCgd3+87/0Vl39/budnf2dnZmf8/7GXdn5pmZZ76zb+7P0yYrYCUhIYAAAggggAACCGSMQHbG3Ck3igACCCCAAAIIIGAECAD5ISCAAAIIIIAAAhkmQACYYQ+c20UAAQQQQAABBAgA+Q0ggAACCCCAAAIZJkAAmGEPnNtFAAEEEEAAAQQIAPkNIIAAAggggAACGSZAAJhhD5zbRQABBBBAAAEECAD5DSCAAAIIIIAAAhkmQACYYQ+c20UAAQQQQAABBAgA+Q0ggAACCCCAAAIZJkAAmGEPnNtFAAEEEEAAAQQIAPkNIIAAAggggAACGSZAAJhhD5zbRQABBBBAAAEECAD5DSCAAAIIIIAAAhkmQACYYQ+c20UAAQQQQAABBAgA+Q0ggAACCCCAAAIZJkAAmGEPnNtNrcD06dMlKytL1qxZk9oLczUEEEAAAQSKESAALAaHXQh4UeD11183QacGnjNnzix0C3l5edK0aVOTp127doX2O91QUFAgtWrVkjvvvNOcyu3yOL0fPX7Pnj0yatQoOfXUU6VGjRrGToP7yDRv3rygvfqHLh9++GFk9kLr8Rwfa5k+/vhjueyyy+Too4+WypUry+GHHy7nnnuufPvtt4WuH7nhtttuM/fQunXryF2sI4CAxwXKeLz8FB8BBCIEPv30U7OlQoUK8tVXX0XsFXnooYdkw4YNZnvbtm0L7Xe6YdGiRbJt2zbp27evOZXb5XF6P3q83s8tt9xigic100CtuPSvf/1LjjvuuLAszZo1C1svbiWW42Mt0x133CEffPCBnHPOOdKmTRvZtGmTPPDAA3LssceKBqVFBXfr16+X22+/3QSNxZWVfQgg4E0BAkBvPjdKjUCRAp999plUq1ZNOnbsKF9//XVYPq01GjNmjJx22mkye/ZsKY0A8NVXX5VGjRqZGie9uNvlCQNIcKVevXqyceNGqVu3rixevLhQcBd52m7dusnZZ58duTnm9ViOj7VMw4YNkxkzZki5cuWC1+/fv78cc8wxMm7cOHnyySeD20O/XHPNNXL88cdLfn6+CYBD9/EdAQS8L0ATsPefIXfgMYFPPvlE+vTpI7m5uVKlShU5+eSTTU1M5G1oLZMGcVqTd8QRR5iau9GjR5smuci8oeta46Z/3HWJrAG8++675eDBg6YpU48pjQDwlVdeCdb+6TXcLo+WwWkqX768Cf7iOc/u3buNdTzHhOYt6fhYy9S1a9ew4E+vceSRR5oAPfIfCPb133vvPfnvf/8rEydOtDfxiQACPhMgAPTZA+V20lvgyy+/FK3d0aDouuuuk5tuuklWr14tPXr0kI8++ihYeA0Stb/Zjz/+KDfffLNceOGFpglSa+2KS9q/b/ny5aapT5v2VqxYEQxCtm7dKhoAjhgxQr7//ntzGm0STGbS5kUtu9YwanK7PFqGAwcOmBosbTItadH+i8lIQ4YMMQG+Bu8nnXSSqTWM57xOjy/pWoFAQDZv3iw1a9YslFVr/C6//HK56KKLzD8iCmVgAwII+EKAJmBfPEZuwisCN954owlI5s+fbwZiaLkHDRokzZs3NwHhu+++a25FBxzk5OSYvlv169c327TjfsuWLc33ov6jNX4a8GhgpwGgfl+5cqW0aNHCNP1WrVrVDAg466yzRJsQdbBGMpM2/2rQ88c//tGc1u3yaCG0/5sGYbEkDcYbN24cS9aoebSZVW01ANbgSu//rrvuMkH/ggULpH379lGPszc6Pd4+T0mfTz31lOkHqv0aI9OUKVPMPxDeeuutyF2sI4CAjwQIAH30MLmV9BbQmpU33nhD+vXrFwz+tMQaiP31r3+VqVOnyq5du0yne/3j++c//1ns4E/z6SACbTr+3//+p6tRk/a306RNuzrqU0ehajOfBmX6h33y5Mnmu+YrjeZfDQA12KpYsaIph9vlsS3efPNNU56S/qN9/JwkbW7VxU5nnHGG6QuoAbnWvM6ZM8feFfXT6fFRTxqx8ZtvvpFLL71UunTpIoMHDw7bqzXOI0eONDXTyf7HQdiFWEEAAdcFCABdfwQUIFMEtAn2559/NrV9kfesNXva/Lhu3Tozzcgvv/xiAr7IfCWNJNWm5ezsbFP7p1N+NGnSxNRCPf/886Yfof7B3759u+gIz/POOy94eq310sEAJ554YnBbcV+i5dfaRg20xo4dGzw00fIET5CEL9WrV5eePXsm4UyJnUKf2Zlnnin6DPQfAVqzG09yenzotbSJXkdna02w9vGLLIvWUOs0N9oETEIAAX8LEAD6+/lydxkmoDVuOsefBn+atBlYA49ly5YF/+BrUKYp2f3/tFlbazDt/n96jWSVRwNWbRZ///335ZBDDpEBAwaIjm6NDGD0mpFJ+yH+9NNPkZujrmutVyznjHpwMRsbNmxo+kPu3bvX9A0sJmvUXU6P15Pu3LnT1CDv2LHDOIbWLut+7S/68MMPm4EfP/zwg24yad++faYrgU5mrgOXNEAkIYCA9wUYBOL9Z8gdeERAg4tKlSqZQRqRRdZmOa250z/0tWvXNs202ncvMkXbFppHA67QwE4DwKVLl5ppS7RJWVNos6yua2f/tWvXSq9evcyoZO0fZg9W0elkOnToYPrRaV5N0fLrdh3926pVK9HaQTslUh77WPtTy6YjpbV2UgfKaM2VTmKsU5nEkrTvnTazx7JoDWxppO+++848Ux31nUhyerwGcaeffrpxe/nll81ziiyHzg2ptdA6B6HWHNuLmqu3rkfrMxh5HtYRQMAbAtQAeuM5UUofCGjNkgZZL774onk1nB0o6WhMnadNAxytYdGkTZY64ldrYuyaGg3+XnvttSIltHlvy5YtYQGgzkVXtmxZ0wRpH6hBmU4hogNPND3yyCOifQ7tJmCtMdNBI1deeaW8/fbbpgZRg4dVq1aJNqdG5rfPq/3//vSnP9mrZsLhRMoTPMFvX66//npT+6f9JDVpGbS/pPavi5xy5rdDwj60r2Oq+gBqM39k3zmtcX3ppZdM7ZsG+XbS7gAa3OpgEXs0bjzH2+cp6VObnTVYXrhwofntad+/aEn/sfDCCy8U2qXNwjolzb333mu6ERTKwAYEEPCkAAGgJx8bhfaqgE7CrMGIBnuXXHKJlClTxszvt3///uCr0/TedL4/HTBywgknyMUXX2z6junbG/SPtDbnRkvRmnZ11GnkyFPNpwNE9NrRktb42DVBul+DB50PTgcwDBw4MNohZiobHWyig0zslKzyaOBiT1b8xBNPmDnsdIJi7c+olvYbR+zrRn4mqw+g+mvzqd08qoNxtGlak/aZ0351aqUDYHQwh9bk6ihgbVbVml+ddDk06RtTdMCMNm3r89YUz/GaP5YyXX311SYA1SBem8JtSz1e09/+9jfzqUGoDlCKTPZcgNH2ReZlHQEEPCRgzQdFQgCBUhL4z3/+E7D+5yBgTS8SvILVJBvo3bt3wGoODFiBQcAKAgJWM2Vwv/1l7ty5ASt4C1hTgwSsiaADVs1bwPpjHrBG9NpZwj6td++aa1k1hWHbQ1es2qCAFaAELrjggtDNAevNHQGrf53ZZr0/OGAFqGH7rcAkYE1nEtwWml83WoFIwAqAAtZAkGAeJ+UJnsT6oteyAlKzqXv37oHx48eb71atWsAKokOzlup3LYc+y2iL/XytWrJAp06dAlY/uYAVYAesZueAFWAFrP51hcr2zjvvmHNZAWBwXzzH60GxlEnNopXZ3ha8eBFf9HjrHwxF7GUzAgh4VSBLC279DwEJAQQ8IKC1MNo/Tz
],
"text/plain": [
"<IPython.core.display.HTML object>"
]
},
"metadata": {},
"output_type": "display_data"
}
],
"source": [
"ks = np.argsort(reader.cat0(\"totpartmass\"))[::-1]\n",
"k = ks[1]\n",
"\n",
"\n",
"plt.figure()\n",
"plt.scatter(reader.dist(False, \"r200\")[k], reader.mass_ratio()[k], c=reader.overlap(False)[k])\n",
"plt.colorbar(label=\"Overlap\")\n",
"\n",
"plt.title(r\"$\\log M_{{\\rm tot}} / M_\\odot = {:.4f}$\".format(np.log10(reader.cat0(\"totpartmass\")[k])))\n",
"plt.xlabel(r\"$\\Delta r_i / R_{200c}$\")\n",
"plt.ylabel(r\"$|\\log \\dfrac{M_i}{M_{\\rm tot}}|$\")\n",
"plt.tight_layout()\n",
"plt.show()"
]
},
{
"cell_type": "code",
"execution_count": 32,
"id": "51dd52f0",
"metadata": {
"ExecuteTime": {
"end_time": "2023-03-24T14:49:25.961273Z",
"start_time": "2023-03-24T14:27:05.607189Z"
},
"scrolled": true
},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Starting: 2023-03-24 14:27:05.644524.\n",
"Loaded `clump0`: 2023-03-24 14:41:48.868024.\n",
"Loaded `clumpx`: 2023-03-24 14:49:25.871648.\n"
]
}
],
"source": [
"print(\"Starting: {}.\".format(datetime.now()))\n",
"clumps0 = np.load(\"/mnt/extraspace/rstiskalek/csiborg/initmatch/clump_7468_particles.npy\", allow_pickle=True)\n",
"print(\"Loaded `clump0`: {}.\".format(datetime.now()))\n",
"clumpsx = np.load(\"/mnt/extraspace/rstiskalek/csiborg/initmatch/clump_7588_particles.npy\", allow_pickle=True)\n",
"print(\"Loaded `clumpx`: {}.\".format(datetime.now()))\n",
"\n",
"overlapper = csiborgtools.match.ParticleOverlap()\n",
"\n",
"hid2clumps0 = {hid: n for n, hid in enumerate(clumps0[\"ID\"])}\n",
"hid2clumpsx = {hid: n for n, hid in enumerate(clumpsx[\"ID\"])}"
]
},
{
"cell_type": "code",
"execution_count": 33,
"id": "bcef2505",
"metadata": {
"ExecuteTime": {
"end_time": "2023-03-24T15:09:47.965132Z",
"start_time": "2023-03-24T15:08:50.141995Z"
}
},
"outputs": [],
"source": [
"# Convert positions to cell IDs\n",
"overlapper.clumps_pos2cell(clumps0)\n",
"overlapper.clumps_pos2cell(clumpsx)\n",
"\n",
"mins0, maxs0 = csiborgtools.match.get_halolims(clumps0, overlapper.inv_clength, overlapper.nshift)\n",
"minsx, maxsx = csiborgtools.match.get_halolims(clumpsx, overlapper.inv_clength, overlapper.nshift)"
]
},
{
"cell_type": "code",
"execution_count": 34,
"id": "e2c24b54",
"metadata": {
"ExecuteTime": {
"end_time": "2023-03-24T15:10:35.352871Z",
"start_time": "2023-03-24T15:09:47.966986Z"
}
},
"outputs": [],
"source": [
"delta_bckg = overlapper.make_bckg_delta(clumps0)\n",
"delta_bckg = overlapper.make_bckg_delta(clumpsx, delta=delta_bckg)"
]
},
{
"cell_type": "code",
"execution_count": 369,
"id": "fb4e8c0a",
"metadata": {
"ExecuteTime": {
"end_time": "2023-03-24T17:04:22.358413Z",
"start_time": "2023-03-24T17:01:54.600744Z"
}
},
"outputs": [],
"source": [
"smooth_kwargs = {\"sigma\": 1, \"truncate\": 4, \"mode\": \"constant\", \"cval\": 0.0}\n",
"\n",
"delta_bckg_smooth = gaussian_filter(delta_bckg, **smooth_kwargs)"
]
},
{
"cell_type": "code",
"execution_count": 363,
"id": "bb707fb2",
"metadata": {
"ExecuteTime": {
"end_time": "2023-03-24T17:00:16.973318Z",
"start_time": "2023-03-24T17:00:16.011637Z"
}
},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Ratio is 0.9820815\n",
"Original overlap is 0.6785714\n",
"Smoothed overlap is 0.6664124\n"
]
},
{
"data": {
"text/plain": [
"0.32628544480462635"
]
},
"execution_count": 363,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"# k = 24734 # skull!\n",
"\n",
"ks = np.argsort(reader.cat0(\"totpartmass\"))[::-1]\n",
"# k = ks[1]\n",
"k = 331\n",
"n = 0\n",
"\n",
"print(\"Ratio is \", summed_ratio[k])\n",
"\n",
"print(\"Original overlap is \", overlap_raw[k][n])\n",
"print(\"Smoothed overlap is \", overlap_smoothed[k][n])\n",
"\n",
"index_cl0 = hid2clumps0[reader.cat0(\"index\", k)]\n",
"cl0 = clumps0[index_cl0][0]\n",
"mins_cl0, maxs_cl0 = mins0[index_cl0], maxs0[index_cl0]\n",
"\n",
"index_clx = hid2clumpsx[reader.catx(\"index\", reader[\"match_indxs\"][k][n])]\n",
"clx = clumpsx[index_clx][0]\n",
"mins_clx, maxs_clx = minsx[index_clx], maxsx[index_clx]\n",
"\n",
"\n",
"\n",
"delta1, delta2, cellmins, nonzero = overlapper.make_deltas(\n",
" cl0, clx, mins_cl0, maxs_cl0, mins_clx, maxs_clx, smooth_kwargs=smooth_kwargs)\n",
"\n",
"csiborgtools.match.calculate_overlap(delta1, delta2, cellmins, delta_bckg_smooth)"
]
},
{
"cell_type": "code",
"execution_count": 364,
"id": "5eeed44f",
"metadata": {
"ExecuteTime": {
"end_time": "2023-03-24T17:00:19.825171Z",
"start_time": "2023-03-24T17:00:19.787750Z"
}
},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"NGP/smoothed overlap 0.6785714 0.6664124\n",
"0.32628544480462635\n",
"Sum is 0.32628544480462635\n",
"Originally NGP/smoothed was 0.6785714 0.6664124\n"
]
}
],
"source": [
"xs = []\n",
"for n in range(reader[\"match_indxs\"][k].size):\n",
"\n",
" index_clx = hid2clumpsx[reader.catx(\"index\", reader[\"match_indxs\"][k][n])]\n",
" clx = clumpsx[index_clx][0]\n",
" mins_clx, maxs_clx = minsx[index_clx], maxsx[index_clx]\n",
" \n",
" print(\"NGP/smoothed overlap \", overlap_raw[k][n], overlap_smoothed[k][n])\n",
" delta1, delta2, cellmins, nonzero1 = overlapper.make_deltas(\n",
" cl0, clx, mins_cl0, maxs_cl0, mins_clx, maxs_clx, smooth_kwargs=smooth_kwargs)\n",
" \n",
" x = csiborgtools.match.calculate_overlap(delta1, delta2, cellmins, delta_bckg_smooth)\n",
" print(x)\n",
" xs.append(x)\n",
" \n",
"print(\"Sum is \", sum(xs))\n",
"print(\"Originally NGP/smoothed was \", summed_raw[k], summed_smoothed[k])"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "1db1bc57",
"metadata": {},
"outputs": [],
"source": []
},
{
"cell_type": "code",
"execution_count": null,
"id": "5883ecc7",
"metadata": {
"ExecuteTime": {
"end_time": "2023-01-31T17:51:03.510067Z",
"start_time": "2023-01-31T17:51:03.469080Z"
}
},
"outputs": [],
"source": []
},
{
"cell_type": "code",
"execution_count": null,
"id": "a58b300c",
"metadata": {},
"outputs": [],
"source": []
},
{
"cell_type": "code",
"execution_count": null,
"id": "56b90375",
"metadata": {
"ExecuteTime": {
"end_time": "2023-01-28T18:54:51.064154Z",
"start_time": "2023-01-28T18:54:47.314086Z"
}
},
"outputs": [],
"source": [
"dlogm = [None] * len(indxs)\n",
"mass = [None] * len(indxs)\n",
"for k in trange(len(indxs)):\n",
" dlogm[k] = np.abs(np.log10(cat[0][\"totpartmass\"][k]) - np.log10(cat[1][\"totpartmass\"][indxs[k]]))\n",
" mass[k] = np.ones(indxs[k].size) * cat[0][\"totpartmass\"][k]\n",
"dlogm = np.asanyarray(dlogm)\n",
"mass = np.asanyarray(mass)"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "e44414b7",
"metadata": {
"ExecuteTime": {
"end_time": "2023-01-28T18:56:19.841434Z",
"start_time": "2023-01-28T18:56:19.041227Z"
}
},
"outputs": [],
"source": [
"plt.figure()\n",
"plt.scatter(np.concatenate(dlogm), np.concatenate(overlap), s=1, rasterized=True)\n",
"t = np.linspace(0, 2)\n",
"plt.plot(t, 10**(-t), c=\"red\", label=r\"$10^{-|\\log M_1 / M_2|}$\")\n",
"plt.xlabel(r\"$|\\log M_1 / M_2|$\")\n",
"plt.ylabel(r\"$\\mathcal{O}$\")\n",
"plt.legend()\n",
"plt.tight_layout()\n",
"# plt.savefig(\"../plots/mass_comparison.png\", dpi=450)\n",
"plt.show()"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "03cec1b7",
"metadata": {},
"outputs": [],
"source": []
},
{
"cell_type": "code",
"execution_count": null,
"id": "1cde4797",
"metadata": {
"ExecuteTime": {
"end_time": "2023-01-28T15:19:36.573717Z",
"start_time": "2023-01-28T15:19:34.985074Z"
}
},
"outputs": [],
"source": [
"for k in trange(len(indxs)):\n",
" if np.any((dlogm[k] > 1.75) & (overlap[k] > 0.15)):\n",
" print(k)"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "58b2cb87",
"metadata": {
"ExecuteTime": {
"end_time": "2023-01-28T15:27:57.709539Z",
"start_time": "2023-01-28T15:27:57.131915Z"
}
},
"outputs": [],
"source": [
"k = 97788\n",
"print(dlogm[k])\n",
"print(overlap[k])\n",
"n = np.argmax(overlap[k])\n",
"\n",
"index_cl0 = [cl[1] for cl in clumps0].index(cat[0][k][\"index\"])\n",
"cl0 = clumps0[index_cl0][0]\n",
"mins_cl0, maxs_cl0 = mins0[index_cl0], maxs0[index_cl0]\n",
"\n",
"index_clx = [cl[1] for cl in clumpsx].index(cat[1][\"index\"][indxs[k]][n])\n",
"clx = clumpsx[index_clx][0]\n",
"mins_clx, maxs_clx = minsx[index_clx], maxsx[index_clx]"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "f5193d37",
"metadata": {
"ExecuteTime": {
"end_time": "2023-01-28T15:28:02.049121Z",
"start_time": "2023-01-28T15:28:02.016020Z"
}
},
"outputs": [],
"source": [
"delta1, delta2, cellmins = overlapper.make_deltas(cl0, clx, mins_cl0, maxs_cl0, mins_clx, maxs_clx)"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "6e0176db",
"metadata": {
"ExecuteTime": {
"end_time": "2023-01-28T15:28:03.044781Z",
"start_time": "2023-01-28T15:28:03.010050Z"
}
},
"outputs": [],
"source": [
"overlapper.overlap(delta1, delta2, cellmins, delta)"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "a3993216",
"metadata": {
"ExecuteTime": {
"end_time": "2023-01-28T15:28:03.662552Z",
"start_time": "2023-01-28T15:28:03.630680Z"
}
},
"outputs": [],
"source": [
"delta1.sum() / delta2.sum()"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "120b0b61",
"metadata": {},
"outputs": [],
"source": []
},
{
"cell_type": "code",
"execution_count": null,
"id": "5170b359",
"metadata": {
"ExecuteTime": {
"end_time": "2023-01-28T15:28:05.961500Z",
"start_time": "2023-01-28T15:28:05.857277Z"
}
},
"outputs": [],
"source": [
"plt.figure()\n",
"plt.imshow(np.sum(delta1, axis=2))\n",
"plt.show()\n",
"\n",
"plt.figure()\n",
"plt.imshow(np.sum(delta2, axis=2))\n",
"plt.show()"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "adcca1e1",
"metadata": {},
"outputs": [],
"source": []
},
{
"cell_type": "code",
"execution_count": null,
"id": "d74da689",
"metadata": {},
"outputs": [],
"source": []
},
{
"cell_type": "code",
"execution_count": null,
"id": "0989f96e",
"metadata": {
"ExecuteTime": {
"start_time": "2023-01-26T09:49:07.667Z"
}
},
"outputs": [],
"source": [
"ncounter = len(indxs[k])\n",
"true_overlap = np.full(ncounter, np.nan)\n",
"spherical_overlap = np.full(ncounter, np.nan)\n",
"\n",
"for n in trange(len(indxs[k])):\n",
" clx = clumpsx[[cl[1] for cl in clumpsx].index(cat[1][\"index\"][indxs[k]][n])][0]\n",
" \n",
" R1 = (3 * cl0.size / (4 * np.pi))**(1./3) * 1 / 2048\n",
" R2 = (3 * clx.size / (4 * np.pi))**(1./3) * 1 / 2048\n",
" d = np.linalg.norm([np.mean(cl0[p]) - np.mean(clx[p]) for p in ('x', 'y', 'z')])\n",
" \n",
" spherical_overlap[n] = csiborgtools.match.spherical_overlap(R1, R2, d)\n",
" true_overlap[n] = overlapper(cl0, clx, delta)\n",
" \n",
"# print(true_overlap, spherical_overlap)\n",
" "
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "6007e537",
"metadata": {
"ExecuteTime": {
"start_time": "2023-01-26T09:49:07.668Z"
}
},
"outputs": [],
"source": [
"plt.figure()\n",
"plt.scatter(true_overlap, spherical_overlap)\n",
"\n",
"t = np.linspace(0, 1, 100)\n",
"plt.plot(t, t, c=\"k\", ls=\"--\")\n",
"\n",
"plt.xlabel(\"True overlap\")\n",
"plt.ylabel(\"Spherical overlap\")\n",
"# plt.xscale(\"log\")\n",
"# plt.yscale(\"log\")\n",
"plt.show()"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "fd1a9591",
"metadata": {},
"outputs": [],
"source": []
},
{
"cell_type": "code",
"execution_count": null,
"id": "2a4062f2",
"metadata": {
"ExecuteTime": {
"start_time": "2023-01-26T09:49:07.670Z"
}
},
"outputs": [],
"source": [
"R1 = (3 * cl0.size / (4 * np.pi))**(1./3) * 1 / 2048\n",
"R2 = (3 * clx.size / (4 * np.pi))**(1./3) * 1 / 2048\n",
"d = np.linalg.norm([np.mean(cl0[p]) - np.mean(clx[p]) for p in ('x', 'y', 'z')])"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "d2b0dcd5",
"metadata": {
"ExecuteTime": {
"end_time": "2023-01-23T20:52:54.565480Z",
"start_time": "2023-01-23T20:52:54.534775Z"
}
},
"outputs": [],
"source": []
},
{
"cell_type": "code",
"execution_count": null,
"id": "64634315",
"metadata": {},
"outputs": [],
"source": []
},
{
"cell_type": "code",
"execution_count": null,
"id": "9cfcc924",
"metadata": {
"ExecuteTime": {
"end_time": "2023-01-23T19:00:54.795153Z",
"start_time": "2023-01-23T19:00:54.447475Z"
}
},
"outputs": [],
"source": []
},
{
"cell_type": "code",
"execution_count": null,
"id": "a747a632",
"metadata": {
"ExecuteTime": {
"start_time": "2023-01-26T09:49:07.686Z"
}
},
"outputs": [],
"source": [
"box = cat[0].box\n",
"maverage = box.box2solarmass(clumps0[2][0][\"M\"][0])\n",
"cell = box.box2mpc(1/2048)"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "27bb5c36",
"metadata": {
"ExecuteTime": {
"start_time": "2023-01-26T09:49:07.686Z"
}
},
"outputs": [],
"source": [
"n_sim = 0\n",
"import numpy\n",
"\n",
"R = (3 * cat.cats[n_sim][\"npart\"] / (4 * numpy.pi))**(1./3) * 1 / 2048\n",
"R = cat.cats[n_sim].box.box2mpc(R)"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "03a7825f",
"metadata": {
"ExecuteTime": {
"start_time": "2023-01-26T09:49:07.687Z"
}
},
"outputs": [],
"source": [
"# dlogm = [None] * len(indxs)\n",
"# for k in trange(len(indxs)):\n",
"# dlogm[k] = np.abs(np.log10(cat[0][\"totpartmass\"][k]) - np.log10(cat[1][\"totpartmass\"][indxs[k]]))\n",
"# dlogm = np.asanyarray(dlogm)\n",
"\n",
"normdist = [None] * len(indxs)\n",
"masses = [None] * len(indxs)\n",
"for k in trange(len(indxs)):\n",
" normdist[k] = dist0[k] / ((3 * cat[0][\"totpartmass\"][k] / (4 * np.pi * maverage))**(1/3) * cell)\n",
" masses[k] = np.log10(np.ones(indxs[k].size) * cat[0][\"totpartmass\"][k])\n",
" \n",
"normdist = np.asanyarray(normdist)\n",
"masses = np.asanyarray(masses)"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "e0330ca5",
"metadata": {
"ExecuteTime": {
"start_time": "2023-01-26T09:49:07.688Z"
},
"scrolled": false
},
"outputs": [],
"source": [
"plt.figure()\n",
"\n",
"# plt.scatter(np.concatenate(normdist), np.concatenate(overlap), c=np.concatenate(masses), s=4)\n",
"\n",
"plt.scatter(np.concatenate(normdist), np.concatenate(masses), c=np.concatenate(overlap), s=4)\n",
"\n",
"\n",
"plt.colorbar()\n",
"# plt.xlabel(r\"$z = 0$ normalised separation by $\\hat{R}$\")\n",
"# plt.xlabel(r\"Absolute difference in total mass [dex]\")\n",
"# plt.xscale(\"log\")\n",
"# plt.ylabel(r\"$\\mathcal{O}$\")\n",
"plt.xscale(\"log\")\n",
"plt.tight_layout()\n",
"# plt.savefig(\"../plots/another_view.png\", dpi=450)\n",
"plt.show()"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "de23a8a1",
"metadata": {},
"outputs": [],
"source": []
},
{
"cell_type": "code",
"execution_count": null,
"id": "a6b5e4f8",
"metadata": {},
"outputs": [],
"source": []
},
{
"cell_type": "code",
"execution_count": null,
"id": "43bc17db",
"metadata": {},
"outputs": [],
"source": []
},
{
"cell_type": "code",
"execution_count": null,
"id": "b4df25af",
"metadata": {},
"outputs": [],
"source": []
},
{
"cell_type": "code",
"execution_count": null,
"id": "0d25a16d",
"metadata": {
"ExecuteTime": {
"start_time": "2023-01-26T09:49:07.690Z"
}
},
"outputs": [],
"source": [
"cl0 = clumps0[[cl[1] for cl in clumps0].index(cat[0][k][\"index\"])][0]\n",
"\n",
"\n",
"\n",
"\n",
"\n",
"\n",
"clx = clumpsx[[cl[1] for cl in clumpsx].index(cat[1][\"index\"][indxs[k]][n])][0]"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "be26cbcc",
"metadata": {},
"outputs": [],
"source": []
}
],
"metadata": {
"kernelspec": {
"display_name": "Python 3",
"language": "python",
"name": "python3"
},
"language_info": {
"codemirror_mode": {
"name": "ipython",
"version": 3
},
"file_extension": ".py",
"mimetype": "text/x-python",
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.6.8"
}
},
"nbformat": 4,
"nbformat_minor": 5
}