Setup new COCOs combination for em_coupling matrices

These have now been defined for all components. Maybe someone smarter than me can explain the difference between sigma_Ip and sigma_B0 since it would simplify things if we could eliminate one instead of multiplying them multiple places...
gafusion · Mar 20, 2024 · 1db3bbf · 1db3bbf
1 parent 4ab29a3
commit 1db3bbf
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/omas/omas_cocos.py b/omas/omas_cocos.py
@@ -30,6 +30,7 @@
 #        transforms['F'] = transforms['TOR']
 #        transforms['POL'] = sigma_B0_eff * sigma_rhotp_eff
 #        transforms['BP'] = transforms['POL']
+#        transforms['PSI/TOR'] = sigma_Ip_eff * sigma_B0_eff * sigma_Bp_eff * (2 * numpy.pi) ** exp_Bp_eff
 #        transforms[None] = 1
 #
 
@@ -594,9 +595,18 @@
 _cocos_signals['edge_transport.model.:.ggd.:.total_ion_energy.d_pol.:.values']='?'                                             #[ADD?]# 0.750000 # _pol
 
 # EM_COUPLING
-_cocos_signals['em_coupling.b_field_pol_probes_active']='?'                                                #[ADD?]# 2.000000 # b  b_field
-_cocos_signals['em_coupling.b_field_pol_probes_passive']='?'                                               #[ADD?]# 2.000000 # b  b_field
-_cocos_signals['em_coupling.b_field_pol_probes_plasma']='?'                                                #[ADD?]# 2.000000 # b  b_field
+_cocos_signals['em_coupling.b_field_pol_probes_active']='TOR'                                                     # 3.000000 # b  b_field  [T/A]
+_cocos_signals['em_coupling.b_field_pol_probes_passive']='TOR'                                                    # 3.000000 # b  b_field  [T/A]
+_cocos_signals['em_coupling.b_field_pol_probes_plasma']='TOR'                                                     # 3.000000 # b  b_field  [T/A]
+_cocos_signals['em_coupling.mutual_active_active']=None                                                           # 0.000000 # 
+_cocos_signals['em_coupling.mutual_loops_active']='PSI/TOR'                                                       # 0.000000 # 
+_cocos_signals['em_coupling.mutual_loops_passive']='PSI/TOR'                                                      # 0.000000 # 
+_cocos_signals['em_coupling.mutual_loops_plasma']='PSI/TOR'                                                       # 0.000000 # 
+_cocos_signals['em_coupling.mutual_passive_active']=None                                                          # 0.000000 # 
+_cocos_signals['em_coupling.mutual_passive_passive']=None                                                         # 0.000000 # 
+_cocos_signals['em_coupling.mutual_plasma_active']='PSI/TOR'                                                      # 0.000000 # 
+_cocos_signals['em_coupling.mutual_plasma_passive']='PSI/TOR'                                                     # 0.000000 # 
+_cocos_signals['em_coupling.mutual_plasma_plasma']='PSI/TOR'                                                      # 0.000000 # 
 
 # EQUILIBRIUM
 _cocos_signals['equilibrium.time_slice.:.global_quantities.magnetic_axis.b_field_tor']='TOR'                                    # 4.000000 # b  b_field  _tor  [T]

diff --git a/omas/omas_physics.py b/omas/omas_physics.py
@@ -2661,6 +2661,7 @@ def cocos_transform(cocosin_index, cocosout_index):
     transforms['F'] = transforms['TOR']
     transforms['POL'] = sigma_B0_eff * sigma_rhotp_eff
     transforms['BP'] = transforms['POL']
+    transforms['PSI/TOR'] = sigma_Ip_eff * sigma_B0_eff * sigma_Bp_eff * (2 * numpy.pi) ** exp_Bp_eff
     transforms[None] = 1
 
     printd(transforms, topic='cocos')
@@ -2959,6 +2960,7 @@ def generate_cocos_signals(structures=[], threshold=0, write=True, verbose=False
 #        transforms['F'] = transforms['TOR']
 #        transforms['POL'] = sigma_B0_eff * sigma_rhotp_eff
 #        transforms['BP'] = transforms['POL']
+#        transforms['PSI/TOR'] = sigma_Ip_eff * sigma_B0_eff * sigma_Bp_eff * (2 * numpy.pi) ** exp_Bp_eff
 #        transforms[None] = 1
 #