nucleardatapy 0.2.0__py3-none-any.whl → 1.0.0__py3-none-any.whl

nucleardatapy/fig/eos_setupAMLeq_fig.py

@@ -3,7 +3,7 @@ import matplotlib.pyplot as plt
 
 import nucleardatapy as nuda
 
-def eos_setupAMLeq_xe_fig( pname, models_micro, models_pheno ):
+def eos_setupAMLeq_xe_fig( pname, micro_mbs, pheno_models, band ):
     """
     Plot nuclear chart (N versus Z).\
     The plot is 1x2 with:\
@@ -22,57 +22,101 @@ def eos_setupAMLeq_xe_fig( pname, models_micro, models_pheno ):
     #
     print(f'Plot name: {pname}')
     #
-    # xe at beta-equilibrium
+    # xe at micro-equilibrium
     #
     asy = 0.5
     #
     fig, axs = plt.subplots(1,2)
     fig.tight_layout() # Or equivalently,  "plt.tight_layout()"
-    fig.subplots_adjust(left=0.12, bottom=0.12, right=None, top=0.98, wspace=0.05, hspace=0.3 )
+    fig.subplots_adjust(left=0.12, bottom=0.12, right=0.95, top=0.90, wspace=0.05, hspace=0.3 )
     #
-    axs[0].set_xlabel(r'$n_\text{nuc}$ (fm$^{-3}$)')
-    axs[0].set_ylabel(r'electron fraction $x_e$')
-    axs[0].set_xlim([0, 0.28])
+    axs[0].set_xlabel(r'$n_\text{nuc}$ (fm$^{-3}$)',fontsize='14')
+    axs[0].set_ylabel(r'electron fraction $x_e$',fontsize='14')
+    axs[0].set_xlim([0, 0.33])
     axs[0].set_ylim([0.1, 0.3])
     #
-    axs[1].set_xlabel(r'$n_\text{nuc}$ (fm$^{-3}$)')
-    #axs[1].set_ylabel(r'electron fraction $x_e$')
-    axs[1].set_xlim([0, 0.28])
+    axs[1].set_xlabel(r'$n_\text{nuc}$ (fm$^{-3}$)',fontsize='14')
+    #axs[1].set_ylabel(r'electron fraction $x_e$',fontsize='14')
+    axs[1].set_xlim([0, 0.33])
     axs[1].set_ylim([0.1, 0.3])
     axs[1].tick_params('y', labelleft=False)
     #
-    for model in models_micro:
+    mb_check = []
+    #
+    for kmb,mb in enumerate(micro_mbs):
+        #
+        print('mb:',mb,kmb)
+        #
+        models, models_lower = nuda.matter.micro_esym_models_mb( mb )
+        #
+        print('models:',models)
         #
-        beta = nuda.eos.setupAMLeq( model = model, kind = 'micro', asy = asy )
-        if nuda.env.verb_output: beta.print_outputs( )
+        if mb == 'VAR':
+            models.remove('1998-VAR-AM-APR-fit')
+            models_lower.remove('1998-var-am-apr-fit')
         #
-        if beta.esym is not None: 
-            print('model:',model)
-            axs[0].plot( beta.den, beta.x_el, linestyle=beta.linestyle, label=beta.label, markevery=beta.every )
-            #axs[0].plot( beta.den, beta.x_el, marker='o', linestyle=beta.linestyle, label=beta.label, markevery=beta.every )
+        for model in models:
+            #
+            micro = nuda.eos.setupAMLeq( model = model, kind = 'micro', asy = asy )
+            if nuda.env.verb_output: micro.print_outputs( )
+            #
+            check = nuda.matter.setupCheck( eos = micro, band = band )
+            #
+            if check.isInside:
+                lstyle = 'solid'
+            else:
+                lstyle = 'dashed'
+                #continue
+            #
+            if micro.x_el is not None: 
+                print('model:',model)
+                if mb in mb_check:
+                    axs[0].plot( micro.den, micro.x_el, marker='o', linestyle=lstyle, markevery=micro.every, color=nuda.param.col[kmb] )
+                else:
+                    mb_check.append(mb)
+                    axs[0].plot( micro.den, micro.x_el, marker='o', linestyle=lstyle, label=mb, markevery=micro.every, color=nuda.param.col[kmb] )
+            # end of model
+        # end of mb
     axs[0].text(0.08,0.22,'microscopic models',fontsize='10')
-    #axs[0].legend(loc='upper left',fontsize='8', ncol=3)
     #
-    for model in models_pheno:
+    model_check = []
+    #
+    for kmodel,model in enumerate(pheno_models):
         #
         params, params_lower = nuda.matter.pheno_esym_params( model = model )
         #
         for param in params:
             #
-            beta = nuda.eos.setupAMLeq( model = model, param = param, kind = 'pheno', asy = asy )
-            if beta.esym is not None: 
-                print('model:',model,' param:',param)
-                #beta.label=None
-                axs[1].plot( beta.den, beta.x_el, linestyle=beta.linestyle, label=beta.label, markevery=beta.every )
+            pheno = nuda.eos.setupAMLeq( model = model, param = param, kind = 'pheno', asy = asy )
             if nuda.env.verb_output: pheno.print_outputs( )
+            #
+            check = nuda.matter.setupCheck( eos = pheno, band = band )
+            #
+            if check.isInside:
+                lstyle = 'solid'
+            else:
+                lstyle = 'dashed'
+                #continue
+            #
+            if pheno.x_el is not None:
+                print('model:',model,' param:',param)
+                if model in model_check:
+                    axs[1].plot( pheno.den, pheno.x_el, linestyle=lstyle, markevery=pheno.every, color=nuda.param.col[kmodel] )
+                else:
+                    model_check.append(model)
+                    axs[1].plot( pheno.den, pheno.x_el, linestyle=lstyle, label=model, markevery=pheno.every, color=nuda.param.col[kmodel] )
+            # end of param
+        # end of model
     #
     axs[1].text(0.08,0.22,'phenomenological models',fontsize='10')
     #
+    fig.legend(loc='upper left',bbox_to_anchor=(0.15,1.0),columnspacing=2,fontsize='8',ncol=5,frameon=False)
+    #
     if pname is not None:
     	plt.savefig(pname, dpi=200)
     	plt.close()
 
-def eos_setupAMLeq_xmu_fig( pname, models_micro, models_pheno ):
+def eos_setupAMLeq_xmu_fig( pname, micro_mbs, pheno_models, band ):
     """
     Plot nuclear chart (N versus Z).\
     The plot is 1x2 with:\
@@ -91,56 +135,103 @@ def eos_setupAMLeq_xmu_fig( pname, models_micro, models_pheno ):
     #
     print(f'Plot name: {pname}')
     #
-    # xmu at beta-equilibrium
+    # xmu at micro-equilibrium
     #
     asy = 0.5
     #
     fig, axs = plt.subplots(1,2)
     fig.tight_layout() # Or equivalently,  "plt.tight_layout()"
-    fig.subplots_adjust(left=0.12, bottom=0.12, right=None, top=0.98, wspace=0.05, hspace=0.3 )
+    fig.subplots_adjust(left=0.12, bottom=0.12, right=0.95, top=0.90, wspace=0.05, hspace=0.3 )
     #
-    axs[0].set_xlabel(r'$n_\text{nuc}$ (fm$^{-3}$)')
-    axs[0].set_ylabel(r'muon fraction $x_\mu$')
-    axs[0].set_xlim([0, 0.28])
+    axs[0].set_xlabel(r'$n_\text{nuc}$ (fm$^{-3}$)',fontsize='14')
+    axs[0].set_ylabel(r'muon fraction $x_\mu$',fontsize='14')
+    axs[0].set_xlim([0, 0.33])
     axs[0].set_ylim([0, 0.15])
     #
-    axs[1].set_xlabel(r'$n_\text{nuc}$ (fm$^{-3}$)')
-    #axs[1].set_ylabel(r'muon fraction $x_\mu$')
-    axs[1].set_xlim([0, 0.28])
+    axs[1].set_xlabel(r'$n_\text{nuc}$ (fm$^{-3}$)',fontsize='14')
+    #axs[1].set_ylabel(r'muon fraction $x_\mu$',fontsize='14')
+    axs[1].set_xlim([0, 0.33])
     axs[1].set_ylim([0, 0.15])
     axs[1].tick_params('y', labelleft=False)
     #
-    for model in models_micro:
+    mb_check = []
+    #
+    for kmb,mb in enumerate(micro_mbs):
+        #
+        print('mb:',mb,kmb)
+        #
+        models, models_lower = nuda.matter.micro_esym_models_mb( mb )
         #
-        beta = nuda.eos.setupAMLeq( model = model, kind = 'micro', asy = asy )
-        if nuda.env.verb_output: beta.print_outputs( )
+        print('models:',models)
         #
-        if beta.esym is not None: 
-            print('model:',model)
-            axs[0].plot( beta.den, beta.x_mu, marker='o', linestyle=beta.linestyle, label=beta.label, markevery=beta.every )
+        if mb == 'VAR':
+            models.remove('1998-VAR-AM-APR-fit')
+            models_lower.remove('1998-var-am-apr-fit')
+        #
+        for model in models:
+            #
+            micro = nuda.eos.setupAMLeq( model = model, kind = 'micro', asy = asy )
+            if nuda.env.verb_output: micro.print_outputs( )
+            #
+            check = nuda.matter.setupCheck( eos = micro, band = band )
+            #
+            if check.isInside:
+                lstyle = 'solid'
+            else:
+                lstyle = 'dashed'
+                #continue
+            #
+            if micro.x_mu is not None: 
+                print('model:',model)
+                if mb in mb_check:
+                    axs[0].plot( micro.den, micro.x_mu, marker='o', linestyle=lstyle, markevery=micro.every, color=nuda.param.col[kmb] )
+                else:
+                    mb_check.append(mb)
+                    axs[0].plot( micro.den, micro.x_mu, marker='o', linestyle=lstyle, label=mb, markevery=micro.every, color=nuda.param.col[kmb] )
+            # end of model
+        # end of mb
+    #
     axs[0].text(0.08,0.12,'microscopic models',fontsize='10')
-    #axs[0].legend(loc='upper left',fontsize='8', ncol=3)
     #
-    for model in models_pheno:
+    model_check = []
+    #
+    for kmodel,model in enumerate(pheno_models):
         #
         params, params_lower = nuda.matter.pheno_esym_params( model = model )
         #
         for param in params:
             #
-            beta = nuda.eos.setupAMLeq( model = model, param = param, kind = 'pheno', asy = asy )
-            if beta.esym is not None: 
-                print('model:',model,' param:',param)
-                #beta.label=None
-                axs[1].plot( beta.den, beta.x_mu, linestyle=beta.linestyle, label=beta.label, markevery=beta.every )
+            pheno = nuda.eos.setupAMLeq( model = model, param = param, kind = 'pheno', asy = asy )
             if nuda.env.verb_output: pheno.print_outputs( )
+            #
+            check = nuda.matter.setupCheck( eos = pheno, band = band )
+            #
+            if check.isInside:
+                lstyle = 'solid'
+            else:
+                lstyle = 'dashed'
+                #continue
+            #
+            if pheno.x_mu is not None: 
+                print('model:',model,' param:',param)
+                #micro.label=None
+                if model in model_check:
+                    axs[1].plot( pheno.den, pheno.x_mu, linestyle=lstyle, markevery=pheno.every, color=nuda.param.col[kmodel] )
+                else:
+                    model_check.append(model)
+                    axs[1].plot( pheno.den, pheno.x_mu, linestyle=lstyle, label=model, markevery=pheno.every, color=nuda.param.col[kmodel] )
+            # end of param
+        # end of model
     #
     axs[1].text(0.08,0.12,'phenomenological models',fontsize='10')
     #
+    fig.legend(loc='upper left',bbox_to_anchor=(0.15,1.0),columnspacing=2,fontsize='8',ncol=5,frameon=False)
+    #
     if pname is not None:
     	plt.savefig(pname, dpi=200)
     	plt.close()
 
-def eos_setupAMLeq_xexmu_fig( pname, models_micro, models_pheno ):
+def eos_setupAMLeq_xexmu_fig( pname, micro_mbs, pheno_models, band ):
     """
     Plot nuclear chart (N versus Z).\
     The plot is 1x2 with:\
@@ -161,44 +252,84 @@ def eos_setupAMLeq_xexmu_fig( pname, models_micro, models_pheno ):
     #
     fig, axs = plt.subplots(1,1)
     #fig.tight_layout() # Or equivalently,  "plt.tight_layout()"
-    fig.subplots_adjust(left=0.12, bottom=0.12, right=None, top=0.95, wspace=0.3, hspace=0.3 )
+    fig.subplots_adjust(left=0.12, bottom=0.12, right=0.95, top=0.90, wspace=0.3, hspace=0.3 )
     #
-    axs.set_xlabel(r'$n_\text{nuc}$ (fm$^{-3}$)',fontsize='10')
-    axs.set_ylabel(r'$x_e$, $x_\mu$',fontsize='10')
-    axs.set_xlim([0, 0.28])
+    axs.set_xlabel(r'$n_\text{nuc}$ (fm$^{-3}$)',fontsize='14')
+    axs.set_ylabel(r'$x_e$, $x_\mu$',fontsize='14')
+    axs.set_xlim([0, 0.33])
     axs.set_ylim([0, 0.5])
     #
     asys = [ 0.1, 0.3, 0.5, 0.7, 0.9 ]
     #asys = [ 0.5 ]
     #
-    for inda,asy in enumerate(asys):
+    for iasy,asy in enumerate(asys):
+        #
+        mb_check = []
         #
-        for model in models_micro:
+        for kmb,mb in enumerate(micro_mbs):
             #
-            continue
-            am = nuda.eos.setupAMLeq( model = model, kind = 'micro', asy = asy )
-            if nuda.env.verb_output: am.print_outputs( )
+            print('mb:',mb,kmb)
             #
-            if am.esym is not None: 
-                print('model:',model)
-                axs.plot( am.den, am.x_mu, marker='o', linestyle=am.linestyle, label=am.label, markevery=am.every )
-        #axs[0].text(0.02,12,'microscopic models',fontsize='10')
-        #axs[0].text(0.02,10,'for $\delta=$'+str(asy),fontsize='10')
+            models, models_lower = nuda.matter.micro_esym_models_mb( mb )
+            #
+            print('models:',models)
+            #
+            if mb == 'VAR':
+                models.remove('1998-VAR-AM-APR-fit')
+                models_lower.remove('1998-var-am-apr-fit')
+            #
+            for model in models:
+                #
+                continue
+                micro = nuda.eos.setupAMLeq( model = model, kind = 'micro', asy = asy )
+                if nuda.env.verb_output: micro.print_outputs( )
+                #
+                check = nuda.matter.setupCheck( eos = micro, band = band )
+                #
+                if check.isInside:
+                    lstyle = 'solid'
+                else:
+                    lstyle = 'dashed'
+                    #continue
+                #
+                if micro.esym is not None: 
+                    print('model:',model)
+                    axs.plot( micro.den, micro.x_mu, marker='o', linestyle=lstyle, label=micro.label, markevery=micro.every )
+                # end of model
+            # end of mb
+        #
+        model_check = []
         #
-        for model in models_pheno:
+        for kmodel,model in enumerate(pheno_models):
             #
             params, params_lower = nuda.matter.pheno_esym_params( model = model )
             #
             for param in params:
                 #
-                am = nuda.eos.setupAMLeq( model = model, param = param, kind = 'pheno', asy = asy )
-                if am.esym is not None: 
-                    print('model:',model,' param:',param)
-                    #beta.label=None
-                    axs.plot( am.den, am.x_el, linestyle='solid', label=r'$\delta=$'+str(asy), color=nuda.param.col[inda] )
-                    axs.plot( am.den, am.x_mu, linestyle='dashed', color=nuda.param.col[inda] )
+                pheno = nuda.eos.setupAMLeq( model = model, param = param, kind = 'pheno', asy = asy )
                 if nuda.env.verb_output: pheno.print_outputs( )
+                #
+                check = nuda.matter.setupCheck( eos = pheno, band = band )
+                #
+                if check.isInside:
+                    lstyle = 'solid'
+                else:
+                    lstyle = 'dashed'
+                    #continue
+                #
+                if pheno.esym is not None: 
+                    print('model:',model,' param:',param)
+                    if model in model_check:
+                        axs.plot( pheno.den, pheno.x_el, linestyle='solid', color=nuda.param.col[iasy] )
+                        axs.plot( pheno.den, pheno.x_mu, linestyle='dashed', color=nuda.param.col[iasy] )
+                    else:
+                        model_check.append(model)
+                        axs.plot( pheno.den, pheno.x_el, linestyle='solid', label=r'$\delta=$'+str(asy), color=nuda.param.col[iasy] )
+                        axs.plot( pheno.den, pheno.x_mu, linestyle='dashed', color=nuda.param.col[iasy] )
+                    #
                 break
+                # end of param
+            # end of model
         #
         axs.legend(loc='upper right',fontsize='10',ncol=3)
         axs.text(0.05,0.35,r'$x_e$',fontsize='14')
@@ -206,4 +337,5 @@ def eos_setupAMLeq_xexmu_fig( pname, models_micro, models_pheno ):
     #
     if pname is not None:
     	plt.savefig(pname, dpi=200)
-    	plt.close()
+    	plt.close()
+    #