Merge pull request #11 from sealden89/laserTests

Adding tests for laser interaction rates.

Author: stewartboogert

08_processes/laserwire_compton_cumulative.tpl

@@ -0,0 +1,27 @@
+
+lb1: laserflux, wavelength=532.0*nm,
+m2=1.2,
+pulseDuration=71.0*ps,
+pulseEnergy={{ LASER_ENERGY }}*mJ,
+w0=2.3*um;
+
+laser1: laserwire, l=0.01*m, wireLength=0.1*m, laserOffsetTheta=-pi/2, laserOffsetPhi=0,laserOffsetY=0.0*m, laserOffsetZ=0*m, laserOffsetX=0, laserBeam="lb1";
+
+lat : line=(laser1);
+use, lat;
+
+beam, particle="e-",
+      energy={{ BEAM_ENERGY }}*GeV,
+      X0=0.0*m,
+      Xp0=0.0,
+      Y0=0.0*m,
+      Yp0=0.0, 
+      distrType="gauss",
+      sigmaX=119*um,
+      sigmaY=1.07*um,
+      sigmaE=0,
+      sigmaT=0;
+option, 
+physicsList="laser_cumulative_compton_scattering";
+
+sample, range=laser1;

08_processes/laserwire_compton_multiStep.tpl

@@ -0,0 +1,27 @@
+
+lb1: laserflux, wavelength=532.0*nm,
+m2=1.2,
+pulseDuration=71.0*ps,
+pulseEnergy={{ LASER_ENERGY }}*mJ,
+w0=2.3*um;
+
+laser1: laserwire, l=0.01*m, wireLength=0.1*m, laserOffsetTheta=-pi/2, laserOffsetPhi=0,laserOffsetY=0.0*m, laserOffsetZ=0*m, laserOffsetX=0, laserBeam="lb1";
+
+lat : line=(laser1);
+use, lat;
+
+beam, particle="e-",
+      energy={{ BEAM_ENERGY }}*GeV,
+      X0=0.0*m,
+      Xp0=0.0,
+      Y0=0.0*m,
+      Yp0=0.0, 
+      distrType="gauss",
+      sigmaX=119*um,
+      sigmaY=1.07*um,
+      sigmaE=0,
+      sigmaT=0;
+option, 
+physicsList="laser_compton_scattering", scaleFactorLaser=1000;
+
+sample, range=laser1;

08_processes/laserwire_photodetachment_cumulative.tpl

@@ -0,0 +1,26 @@
+lb1: laserflux, wavelength=1064.0*nm,
+m2=1.2,
+pulseDuration=106.0*ns,
+pulseEnergy={{ LASER_ENERGY }}*mJ,
+w0=50*mm;
+
+laser1: laserwire, l=0.01*m, wireLength=0.1*m, laserOffsetTheta=-pi/2, laserOffsetPhi=0,laserOffsetY=0.0*m, laserOffsetZ=0*m, laserOffsetX=0, laserBeam="lb1";
+
+lat : line=(laser1);
+use, lat;
+
+beam, particle="ion 1 1 -1",
+      energy={{ BEAM_ENERGY }}*MeV,
+      X0=0.0*m,
+      Xp0=0.0,
+      Y0=0.0*m,
+      Yp0=0.0, 
+      distrType="gauss",
+      sigmaX=2*mm,
+      sigmaY=4*mm,
+      sigmaE=0,
+      sigmaT=0;
+option, 
+physicsList="laser_cumulative_photo_detachment";
+
+sample, range=laser1;

08_processes/laserwire_photodetachment_multiStep.tpl

@@ -0,0 +1,26 @@
+lb1: laserflux, wavelength=1064.0*nm,
+m2=1.2,
+pulseDuration=106.0*ns,
+pulseEnergy={{ LASER_ENERGY }}*mJ,
+w0=50*mm;
+
+laser1: laserwire, l=0.01*m, wireLength=0.1*m, laserOffsetTheta=-pi/2, laserOffsetPhi=0,laserOffsetY=0.0*m, laserOffsetZ=0*m, laserOffsetX=0, laserBeam="lb1";
+
+lat : line=(laser1);
+use, lat;
+
+beam, particle="ion 1 1 -1",
+      energy={{ BEAM_ENERGY }}*MeV,
+      X0=0.0*m,
+      Xp0=0.0,
+      Y0=0.0*m,
+      Yp0=0.0, 
+      distrType="gauss",
+      sigmaX=2*mm,
+      sigmaY=4*mm,
+      sigmaE=0,
+      sigmaT=0;
+option, 
+physicsList="laser_photo_detachment", scaleFactorLaser=1000;
+
+sample, range=laser1;

08_processes/test_laserwire_compton_cumulative.py

@@ -0,0 +1,39 @@
+import pytest
+import pybdsim
+import os
+
+def test() :
+
+    os.chdir(os.path.dirname(__file__))
+    
+    base_name     = "laserwire_compton_cumulative"
+    template_name = base_name+".tpl"
+    gmad_name     = base_name+".gmad"
+    root_name     = base_name+".root"
+
+    data = {
+        'LASER_ENERGY': '150',
+        'BEAM_ENERGY' : '1.3'
+    }
+
+    pybdsim.Run.RenderGmadJinjaTemplate(template_name,gmad_name,data)
+    pybdsim.Run.Bdsim(gmad_name,base_name,10000,1)
+
+    d = pybdsim.Data.Load(root_name)
+    samplerData = pybdsim.Data.SamplerData(d,'laser1') 
+    weights=samplerData.data.get("weight")
+    partid=samplerData.data.get("partID")
+    wpho=[]
+
+    for i in range(len(partid)):
+        if partid[i]==22:
+            wpho.append(weights[i])
+
+
+    Npho=sum(wpho)
+    ref_Npho=0.626774271968543
+    
+
+    print('test ',sum(wpho))
+    print('ref ', ref_Npho)
+    assert (Npho==pytest.approx(ref_Npho,abs=1e-3))

08_processes/test_laserwire_compton_multiStep.py

@@ -0,0 +1,35 @@
+import pytest
+import pybdsim
+import os
+
+def test() :
+
+    os.chdir(os.path.dirname(__file__))
+    
+    base_name     = "laserwire_compton_multiStep"
+    template_name = base_name+".tpl"
+    gmad_name     = base_name+".gmad"
+    root_name     = base_name+".root"
+
+    l  = 2.0 
+    data = {
+        'LASER_ENERGY': '150',
+        'BEAM_ENERGY' : '1.3'
+    }
+
+    pybdsim.Run.RenderGmadJinjaTemplate(template_name,gmad_name,data)
+    pybdsim.Run.Bdsim(gmad_name,base_name,10000,1)
+
+    d = pybdsim.Data.Load(root_name)
+    samplerData = pybdsim.Data.SamplerData(d,'laser1') 
+    partid=samplerData.data.get('partID')
+    Npho=0
+    for i in range(len(partid)):
+        if partid[i]==22:
+            Npho+=1
+
+    ref_Npho=637
+    
+    print('test ',Npho)
+    print('ref ', ref_Npho)
+    assert (Npho==ref_Npho)

08_processes/test_laserwire_photodetachment_cumulative.py

@@ -0,0 +1,36 @@
+import pytest
+import pybdsim
+import os
+
+def test() :
+
+    os.chdir(os.path.dirname(__file__))
+    
+    base_name     = "laserwire_photodetachment_cumulative"
+    template_name = base_name+".tpl"
+    gmad_name     = base_name+".gmad"
+    root_name     = base_name+".root"
+
+    data = {
+        'LASER_ENERGY': '67.4',
+        'BEAM_ENERGY' : '0.942022+0.003'
+    }
+
+    pybdsim.Run.RenderGmadJinjaTemplate(template_name,gmad_name,data)
+    pybdsim.Run.Bdsim(gmad_name,base_name,10000,1)
+
+    d = pybdsim.Data.Load(root_name)
+    samplerData = pybdsim.Data.SamplerData(d,'laser1') 
+    weights=samplerData.data.get("weight")
+    partid=samplerData.data.get("partID")
+
+    we=[]
+    for i in range(len(weights)):
+        if partid[i]==11:
+            we.append(weights[i])
+    Ne=sum(we)
+    ref_Ne=5.843840659450507
+    
+    print('test ', Ne)
+    print('ref ', ref_Ne)
+    assert (Ne==ref_Ne)

08_processes/test_laserwire_photodetachment_multiStep.py

@@ -0,0 +1,34 @@
+import pytest
+import pybdsim
+import os
+
+def test() :
+
+    os.chdir(os.path.dirname(__file__))
+    
+    base_name     = "laserwire_photodetachment_multiStep"
+    template_name = base_name+".tpl"
+    gmad_name     = base_name+".gmad"
+    root_name     = base_name+".root"
+
+    data = {
+        'LASER_ENERGY': '67.4',
+        'BEAM_ENERGY' : '0.942022+0.003'
+    }
+
+    pybdsim.Run.RenderGmadJinjaTemplate(template_name,gmad_name,data)
+    pybdsim.Run.Bdsim(gmad_name,base_name,10000,1)
+
+    d = pybdsim.Data.Load(root_name)
+    samplerData = pybdsim.Data.SamplerData(d,'laser1') 
+    partid=samplerData.data.get('partID')
+    Ne=0
+    for i in range(len(partid)):
+        if partid[i]==11:
+            Ne+=1
+
+    ref_Ne=6852
+    
+    print('test ',Ne)
+    print('ref ', ref_Ne)
+    assert (Ne==ref_Ne)