Skip to content

Commit

Permalink
new file added
Browse files Browse the repository at this point in the history
  • Loading branch information
@snonis
snonis committed Mar 5, 2024
1 parent 72c9a6e commit c139c74
Show file tree
Hide file tree
Showing 4 changed files with 163 additions and 119 deletions.
4 changes: 3 additions & 1 deletion grand/sim/efield2voltage.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -51,6 +51,7 @@ class Efield2Voltage:

def __init__(self, f_input, f_output="", seed=None, padding_factor=1.0, du_type='GP300'):
# du_type='GP300' (using hfss simulations), 'GP300_nec' (using nec simulations), 'GP300_mat' (using matlab simulations), 'Horizon'
self.du_type = du_type
self.f_input = f_input
self.f_output = f_output
self.seed = seed # used to generate same set of random numbers. (gal noise)
Expand Down Expand Up @@ -147,7 +148,8 @@ def get_event(self, event_idx=None, event_number=None, run_number=None):
self.fft_size,
self.freqs_mhz,
self.nb_du,
seed=self.seed
seed=self.seed,
du_type=self.du_type
)
# compute total transfer function of RF chain. Can be computed only once in __init__ if length of time traces does not change between events.
if self.params["add_rf_chain"]:
Expand Down
278 changes: 160 additions & 118 deletions grand/sim/noise/Test_for_new_galaxy_py.ipynb
View file
Original file line number Diff line number Diff line change
Expand Up @@ -127,7 +127,7 @@
},
{
"cell_type": "code",
"execution_count": 28,
"execution_count": 14,
"id": "e9ed7793-f841-4d30-b0f6-ed17287fb642",
"metadata": {},
"outputs": [],
Expand All @@ -138,6 +138,7 @@
"\n",
"import h5py\n",
"import numpy as np\n",
"from matplotlib import pyplot as plt\n",
"from grand import grand_add_path_data\n",
"\n",
"def interpol_at_new_x(a_x, a_y, new_x):\n",
Expand Down Expand Up @@ -299,7 +300,7 @@
},
{
"cell_type": "code",
"execution_count": 29,
"execution_count": 6,
"id": "97d7bf38-3984-4dd1-8d5e-344e82e44745",
"metadata": {},
"outputs": [],
Expand All @@ -308,159 +309,200 @@
"nb_ant = 300\n",
"size_out = 20\n",
"freqs_mhz = np.arange(30, 251, 1)\n",
"lest=18"
"lst=18"
]
},
{
"cell_type": "code",
"execution_count": 47,
"execution_count": 7,
"id": "e33111c4-5d00-4e4b-9c5c-f40485b7e673",
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
"array([[[-2.98126189e+00+1.95813677e+00j,\n",
" 3.34572569e+00+1.52550714e+00j,\n",
" 3.90075698e+00+1.70692047e+01j, ...,\n",
" 1.62385095e+00+1.32892821e+01j,\n",
" -3.95829279e+01+2.49934012e+01j,\n",
" -6.45968966e-01+9.76450234e-01j],\n",
" [ 5.55975952e+01-1.32173454e+01j,\n",
" -9.65557324e+00+1.52011275e+01j,\n",
" -1.52305429e+01+1.04342555e+01j, ...,\n",
" -2.79036579e-02+8.57266986e+00j,\n",
" 3.66742709e+00+1.11378748e+01j,\n",
" -1.14187999e+01+1.18710568e+01j],\n",
" [ 1.73499185e+01+1.26067507e+01j,\n",
" 3.97780489e+01+4.59216261e+01j,\n",
" 3.45848326e+01+8.23762318e+00j, ...,\n",
" 1.99766509e+00-6.70664021e+00j,\n",
" 8.83469145e+00-7.79161673e-02j,\n",
" 2.49444577e+01-1.08397325e+01j]],\n",
"array([[[-3.12530053e+00+2.05274347e+00j,\n",
" 3.55623994e+00+1.62149259e+00j,\n",
" 4.19390880e+00+1.83519989e+01j, ...,\n",
" 1.46539732e+00+1.19925282e+01j,\n",
" -3.55640550e+01+2.24558096e+01j,\n",
" -5.77326941e-01+8.72690571e-01j],\n",
" [ 6.71111716e+01-1.59544946e+01j,\n",
" -1.19110551e+01+1.87520163e+01j,\n",
" -1.91588388e+01+1.31254822e+01j, ...,\n",
" -3.48623194e-02+1.07105368e+01j,\n",
" 4.55955491e+00+1.38472424e+01j,\n",
" -1.40874984e+01+1.46454527e+01j],\n",
" [ 2.71679927e+01+1.97407330e+01j,\n",
" 5.70902500e+01+6.59076347e+01j,\n",
" 4.59040972e+01+1.09337136e+01j, ...,\n",
" 4.34029895e+00-1.45714232e+01j,\n",
" 1.95397942e+01-1.72328132e-01j,\n",
" 5.49045445e+01-2.38590303e+01j]],\n",
"\n",
" [[-4.25112547e+00-4.76449391e+01j,\n",
" 1.23592422e+01+1.69003828e+01j,\n",
" 2.91478958e+01+9.39616605e+00j, ...,\n",
" -6.48019945e+00-5.28586127e+00j,\n",
" 2.04149889e+00-2.95650825e+00j,\n",
" 2.76024199e+00+1.76537189e+00j],\n",
" [ 9.51549912e+00-1.64077254e+01j,\n",
" -8.11200447e+00-8.45343178e+00j,\n",
" -4.92422101e+00-4.70626151e+00j, ...,\n",
" -3.35140704e+00+1.44300580e+01j,\n",
" 2.81923370e+00-1.94347596e+00j,\n",
" 4.21489514e+00+3.18112529e+00j],\n",
" [ 7.05050204e+00-3.81387014e+01j,\n",
" -1.66900917e+00+9.50846736e+00j,\n",
" -9.05053622e+00+1.54775010e+01j, ...,\n",
" -1.92548874e+00+3.14550127e+00j,\n",
" -5.54301582e-01+4.26402218e+00j,\n",
" 1.41774940e+01-1.68475649e+01j]],\n",
" [[-4.45651713e+00-4.99468879e+01j,\n",
" 1.31368901e+01+1.79637609e+01j,\n",
" 3.13384344e+01+1.01023119e+01j, ...,\n",
" -5.84786856e+00-4.77007262e+00j,\n",
" 1.83422457e+00-2.65633261e+00j,\n",
" 2.46693285e+00+1.57777974e+00j],\n",
" [ 1.14860417e+01-1.98055630e+01j,\n",
" -1.00069183e+01-1.04281009e+01j,\n",
" -6.19428715e+00-5.92011106e+00j, ...,\n",
" -4.18718660e+00+1.80286504e+01j,\n",
" 3.50503242e+00-2.41624036e+00j,\n",
" 5.19996226e+00+3.92458908e+00j],\n",
" [ 1.10402817e+01-5.97208545e+01j,\n",
" -2.39539529e+00+1.36467422e+01j,\n",
" -1.20126848e+01+2.05431299e+01j, ...,\n",
" -4.18348241e+00+6.83418654e+00j,\n",
" -1.22595553e+00+9.43078955e+00j,\n",
" 3.12056835e+01-3.70827013e+01j]],\n",
"\n",
" [[ 2.46449288e+00+8.74427143e+00j,\n",
" 8.94120590e-02+1.09957101e-01j,\n",
" 2.05214927e+00+4.61180598e+01j, ...,\n",
" -1.15732565e+01-1.76137749e-01j,\n",
" 1.36484656e+01-4.71293746e+00j,\n",
" 1.50112291e+01-7.44271711e+00j],\n",
" [-2.83985311e-01-1.07330862e+01j,\n",
" 1.97984327e+01-1.38239782e+01j,\n",
" -1.28038452e+01-3.61483851e+00j, ...,\n",
" 4.37628205e+00-5.34144687e+00j,\n",
" 7.29616161e+00+1.06997942e+01j,\n",
" -1.27214329e+01+1.72831359e+00j],\n",
" [-9.76627955e+00+4.78766635e+00j,\n",
" -3.43935958e+01+1.77598104e+01j,\n",
" 2.13951385e+01+3.49512175e+00j, ...,\n",
" 1.15969745e+01+1.91578653e+01j,\n",
" -5.31613097e+00-2.22819318e+00j,\n",
" 1.08157113e+01+1.81434378e+01j]],\n",
" [[ 2.58356401e+00+9.16674788e+00j,\n",
" 9.50378975e-02+1.16875640e-01j,\n",
" 2.20637352e+00+4.95839495e+01j, ...,\n",
" -1.04439506e+01-1.58950417e-01j,\n",
" 1.22627306e+01-4.23443074e+00j,\n",
" 1.34161042e+01-6.65183828e+00j],\n",
" [-3.42795168e-01-1.29557760e+01j,\n",
" 2.44232234e+01-1.70531735e+01j,\n",
" -1.61062417e+01-4.54718578e+00j, ...,\n",
" 5.46764669e+00-6.67350594e+00j,\n",
" 9.07100501e+00+1.33025955e+01j,\n",
" -1.56945710e+01+2.13223939e+00j],\n",
" [-1.52928794e+01+7.49693920e+00j,\n",
" -4.93623755e+01+2.54892346e+01j,\n",
" 2.83975501e+01+4.63903961e+00j, ...,\n",
" 2.51965840e+01+4.16240255e+01j,\n",
" -1.17577513e+01-4.92812186e+00j,\n",
" 2.38061581e+01+3.99350108e+01j]],\n",
"\n",
" ...,\n",
"\n",
" [[-3.75645404e+01+9.08761748e+00j,\n",
" -6.02381593e+00+3.58775211e+00j,\n",
" -3.92645630e+00+9.61112998e+00j, ...,\n",
" 2.15991515e+00-2.16198424e+00j,\n",
" -7.02978643e+00+3.93540637e+00j,\n",
" 2.42113208e+01+7.69228643e+00j],\n",
" [-3.56406126e+01+2.96816882e+00j,\n",
" 9.48585973e+00+1.49213436e+01j,\n",
" 3.12251646e-01-1.12877623e+00j, ...,\n",
" 1.22450017e+00+2.40252295e-01j,\n",
" -9.67705290e+00+7.88618707e+00j,\n",
" -1.74436422e+01-2.15937970e+00j],\n",
" [-1.54335685e+01+1.04075212e+00j,\n",
" -8.14508211e-01-3.62003959e-01j,\n",
" -1.90773699e+01-6.46175519e+00j, ...,\n",
" -7.26387838e+00-8.15758656e+00j,\n",
" -4.27787477e+00-2.18729332e+00j,\n",
" 8.19552634e-01-4.48506986e-01j]],\n",
" [[-3.93794582e+01+9.52668258e+00j,\n",
" -6.40283656e+00+3.81349474e+00j,\n",
" -4.22153949e+00+1.03334309e+01j, ...,\n",
" 1.94915295e+00-1.95102015e+00j,\n",
" -6.31604897e+00+3.53584274e+00j,\n",
" 2.16385748e+01+6.87488783e+00j],\n",
" [-4.30213440e+01+3.58283999e+00j,\n",
" 1.17016976e+01+1.84068767e+01j,\n",
" 3.92788293e-01-1.41991273e+00j, ...,\n",
" 1.52986810e+00+3.00166819e-01j,\n",
" -1.20310651e+01+9.80455838e+00j,\n",
" -2.15204122e+01-2.66405036e+00j],\n",
" [-2.41672073e+01+1.62969907e+00j,\n",
" -1.16899845e+00-5.19555310e-01j,\n",
" -2.53211994e+01-8.57662205e+00j, ...,\n",
" -1.57821268e+01-1.77238740e+01j,\n",
" -9.46142751e+00-4.83766316e+00j,\n",
" 1.80389426e+00-9.87196117e-01j]],\n",
"\n",
" [[-2.40480168e+01-8.72329271e+00j,\n",
" 4.83559416e+01-4.45632593e-03j,\n",
" -3.10297829e+00+8.68956800e-01j, ...,\n",
" -6.92104610e+00-1.22317870e+01j,\n",
" -4.93756120e+00-2.01867604e+01j,\n",
" 2.61960931e+00+4.96169586e+00j],\n",
" [ 2.10652133e+01-1.07146668e+01j,\n",
" -5.19046581e+00+1.64251317e+01j,\n",
" -7.85483755e+00+2.26473176e+01j, ...,\n",
" -9.23096901e+00-1.82112189e+00j,\n",
" 6.92135288e+00+5.39631114e+00j,\n",
" 1.86477750e+01+6.37274570e+00j],\n",
" [-1.83916957e+01+7.79279917e+00j,\n",
" -2.24541975e+01-1.03552013e+01j,\n",
" 2.56768287e+01-2.41578324e+01j, ...,\n",
" 4.89726339e+00-5.33603044e+00j,\n",
" -5.53107057e+00+2.22815171e+00j,\n",
" 5.21676831e-01-1.23413512e+01j]],\n",
" [[-2.52098884e+01-9.14475559e+00j,\n",
" 5.13985146e+01-4.73671954e-03j,\n",
" -3.33617501e+00+9.34261117e-01j, ...,\n",
" -6.24569787e+00-1.10382224e+01j,\n",
" -4.43624833e+00-1.81371893e+01j,\n",
" 2.34124409e+00+4.43445559e+00j],\n",
" [ 2.54275592e+01-1.29335422e+01j,\n",
" -6.40292633e+00+2.02619403e+01j,\n",
" -9.88077491e+00+2.84885647e+01j, ...,\n",
" -1.15330037e+01-2.27527635e+00j,\n",
" 8.60502138e+00+6.70900235e+00j,\n",
" 2.30059641e+01+7.86212611e+00j],\n",
" [-2.87992969e+01+1.22026343e+01j,\n",
" -3.22267126e+01-1.48619917e+01j,\n",
" 3.40805940e+01-3.20644456e+01j, ...,\n",
" 1.06402155e+01-1.15935186e+01j,\n",
" -1.22331358e+01+4.92803016e+00j,\n",
" 1.14824820e+00-2.71642012e+01j]],\n",
"\n",
" [[ 4.57011139e+00-1.17604596e+01j,\n",
" -2.74530379e+01+1.08363255e+01j,\n",
" -2.76764268e+00+1.02845000e+01j, ...,\n",
" 1.70265206e+00-7.85779749e+00j,\n",
" 6.34665535e+00+4.05867604e+00j,\n",
" 2.66361846e-01-1.39901670e+01j],\n",
" [-3.66697266e+01+3.99145277e+01j,\n",
" -3.98007759e+01-3.13569314e+01j,\n",
" -6.18573693e+00+3.65642969e+01j, ...,\n",
" 1.63201493e+01+1.23345122e+01j,\n",
" -1.75284278e+00-5.33320460e+00j,\n",
" -5.02708500e+00+4.24148981e+00j],\n",
" [ 2.92394157e+01-3.20169100e-01j,\n",
" 4.46545789e+00-1.67560944e+00j,\n",
" 2.12066200e+01+2.25211434e+01j, ...,\n",
" 1.61550018e+00+3.47771179e+00j,\n",
" -1.06643087e+01+1.44591655e+01j,\n",
" -1.18397093e+01+2.14809282e+00j]]])"
" [[ 4.79091474e+00-1.23286621e+01j,\n",
" -2.91803928e+01+1.15181509e+01j,\n",
" -2.97563807e+00+1.10574064e+01j, ...,\n",
" 1.53650910e+00-7.09104206e+00j,\n",
" 5.70227650e+00+3.64659679e+00j,\n",
" 2.38057674e-01-1.25035423e+01j],\n",
" [-4.42635748e+01+4.81803342e+01j,\n",
" -4.90979896e+01-3.86817155e+01j,\n",
" -7.78117609e+00+4.59950426e+01j, ...,\n",
" 2.03900958e+01+1.54105138e+01j,\n",
" -2.17923430e+00-6.63054469e+00j,\n",
" -6.20196978e+00+5.23277239e+00j],\n",
" [ 4.57855885e+01-5.01348278e-01j,\n",
" 6.40891432e+00-2.40486813e+00j,\n",
" 2.81473313e+01+2.98920849e+01j, ...,\n",
" 3.50997461e+00+7.55597569e+00j,\n",
" -2.35863808e+01+3.19795116e+01j,\n",
" -2.60600512e+01+4.72810670e+00j]]])"
]
},
"execution_count": 47,
"execution_count": 7,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"galactic_noise(lst, size_out, freqs_mhz, nb_ant, seed=0, du_type='GP300')"
"galactic_noise(lst, size_out, freqs_mhz, nb_ant, seed=0, du_type='GP300_nec')"
]
},
{
"cell_type": "code",
"execution_count": null,
"execution_count": 10,
"id": "506d460e-d0af-461f-a28e-76459b5d5197",
"metadata": {},
"outputs": [],
"source": []
"source": [
"GN_nec=galactic_noise(lst, size_out, freqs_mhz, nb_ant, seed=0, du_type='GP300_nec')"
]
},
{
"cell_type": "code",
"execution_count": null,
"execution_count": 11,
"id": "ce51e00f-dcf7-439c-bf03-fae70b4cf545",
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
"(300, 3, 221)"
]
},
"execution_count": 11,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"GN_nec.shape"
]
},
{
"cell_type": "code",
"execution_count": 21,
"id": "9d70dbf9-7990-4d19-ab80-6fde746c653a",
"metadata": {},
"outputs": [],
"source": []
"source": [
"#plt.plot(freqs_mhz,abs(GN_nec[0,2,:]))\n",
"#plt.plot(freqs_mhz,abs(GN_nec[1,2,:]))\n",
"#plt.plot(freqs_mhz,abs(GN_nec[2,2,:]))\n",
"#plt.plot(freqs_mhz,abs(GN_nec[3,2,:]))\n",
"#plt.plot(freqs_mhz,abs(GN_nec[4,2,:]))\n",
"#plt.plot(freqs_mhz,abs(GN_nec[5,2,:]))\n",
"#plt.plot(freqs_mhz,abs(GN_nec[6,2,:]))"
]
},
{
"cell_type": "code",
"execution_count": 22,
"id": "51afc1f5-a158-4bf4-91bb-9d8a7eef5956",
"metadata": {},
"outputs": [],
"source": [
"np.save(\"galaxy_oc_nec.npy\",GN_nec)"
]
},
{
"cell_type": "code",
Expand Down
Binary file modified scripts/test_voltage1.root
View file
Binary file not shown.
Binary file modified scripts/test_voltage2.root
View file
Binary file not shown.

0 comments on commit c139c74

Please sign in to comment.