#!/usr/bin/env /usr/bin/python3
import os
import cgi, cgitb
import subprocess


def FindName(A, Z):
  result = subprocess.run(["../Cleopatra/IsotopeShort", str(A), str(Z)], stdout=subprocess.PIPE).stdout.decode('utf-8')
  ppp1 = result.find('Name:')
  ppp2 = result.find('|')
  return result[ppp1+5:ppp2]
  
def FindAZ(AZ):
  result = subprocess.run(["../Cleopatra/IsotopeShort", AZ], stdout=subprocess.PIPE).stdout.decode('utf-8')
  pA = result.find('A:')
  pZ = result.find('Z:')
  return [int(result[pA+2:pA+5]), int(result[pZ+2:pZ+5])]


form = cgi.FieldStorage()

beam_AZ = form.getvalue('beam_AZ')
#beam_A = form.getvalue('beam_A')
#beam_Z = form.getvalue('beam_Z')
beam_Jpi = form.getvalue('beam_Jpi')
beam_Ex = form.getvalue('beam_Ex')

target_AZ = form.getvalue('target_AZ')
#target_A = form.getvalue('target_A')
#target_Z = form.getvalue('target_Z')

lRecoil_AZ = form.getvalue('lRecoil_AZ')
#lRecoil_A = form.getvalue('lRecoil_A')
#lRecoil_Z = form.getvalue('lRecoil_Z')

KEA = form.getvalue('KEA')
BField = form.getvalue('BField')
posArray = form.getvalue('posArray')
posRecoil = form.getvalue('posRecoil')

isDWBA = form.getvalue('DWBA');
if isDWBA == "On" :
  isDWBA = True
else:
  isDWBA = False

SSType = form.getvalue('SSType');

ExList = form.getvalue('Ex');
JpiList = form.getvalue('Jpi');
OrbList = form.getvalue('Orb');

op1 = form.getvalue('op1')
op2 = form.getvalue('op2')

plot = form.getvalue('plot')
gate = form.getvalue('gate')

#====== ch directory
os.chdir(r"files")
#================== write reactionConfig.txt

beam = FindAZ(beam_AZ)
target = FindAZ(target_AZ)
lRecoil = FindAZ(lRecoil_AZ)

fn1 = "sim_reactionConfig.txt"

f = open(fn1, "w")
f.write("%s //beam_A\n" % beam[0])
f.write("%s //beam_Z\n" % beam[1])
f.write("%s //target_A\n" % target[0])
f.write("%s //target_Z\n" % target[1])
f.write("%s //recoil_light_A\n" % lRecoil[0])
f.write("%s //recoil_light_Z\n" % lRecoil[1])
f.write("%.3f     //beam_energy_in_MeV/u\n" % float(KEA))
f.write("0.000  //beam_energy_sigma_in_MeV/u\n")
f.write("0.000  //beam_angle_in_mrad\n")
f.write("0.000  //beam_emittance_in_mrad\n")
f.write("0.000  //beam_x-offset_in_mm\n")
f.write("0.000  //beam_y-offset_in_mm\n")
f.write("100000  //number_of_event_generated\n")
f.write("false  //isTargetScattering\n")
f.write("0.913  //Target_density_in_g/cm3\n")
f.write("2.2e-4  //Target_thickness_in_cm\n")
f.write("../SRIM/20F_in_CD2.txt  //stopping_power_for_beam\n")
f.write("../SRIM/3H_in_CD2.txt   //stopping_power_for_light_recoil\n")
f.write("../SRIM19F_in_CD2.txt  //stopping_power_for_heavy_recoil\n")
f.write("false     //isDacay\n")
f.write("32        //decayNucleus_A\n")
f.write("14        //decayNucleus_Z\n")
f.write("false     //isReDo\n")
f.write("%s        //excitation_energy_of_A[MeV]\n" % beam_Ex)
f.write("#===== end of file")
f.close()

#================== write detectorGeo.txt
bore="462.5"
prepDistant="11.5"
detWidth="10.0"
detLength="50.0"
nSide="4"
pos=['58.6', '117.9', '176.8', '235.8', '294.0']

if SSType == "SOLARIS" :
  nSide="6"
  pos=['58.6', '117.9', '176.8', '235.8', '294.0', '353.2', '412.1', '471.0', '530.0']


fn2 = "sim_detectorGeo.txt"
f = open(fn2, "w")
f.write("%s    //Bfield_[T]\n" % BField)
f.write("0.00  //Bfield_direction_to_z-axis_[deg]_should_not_use\n")
f.write("%s    //bore_[mm]\n" % bore)
f.write("%s    //distance_from_axis_[mm]\n" % prepDistant)
f.write("%s     //width_of_detector_[mm]\n" % detWidth)
f.write("%s       //length_of_detector_[mm]\n" % detLength)
f.write("%s      //recoil_position_+_for_downstream_[mm]\n" % posRecoil)
f.write("10.0     //inner_radius_of_recoil_detector_[mm]\n")
f.write("40.2     //outter_radius_of_recoil_detector_[mm]\n")
f.write("false    //is_coincident_with_recoil\n")
f.write("0        //Recoil_1_position_[mm]_when_0_disable_tree_branch\n")
f.write("0        //Recoil_2_position_[mm]\n")
f.write("0.00   //Elum_1_position_[mm]_(just_another_recoil_detector_but_for_light_recoil)\n")
f.write("0.00   //Elum_2_position_[mm]_when_Elum=0_disable_tree_branch\n")
f.write("0       //support_length_[mm]\n")
f.write("%s     //first_position_-_for_upstream_[mm]\n" % posArray)
f.write("0.03     //energy_resolution_of_PSD_array_[MeV]\n")
f.write("1.00     //position_resolution_of_PSD_array_[mm]\n")
f.write("Out           //detector_facing_Out_or_In\n")
f.write("%s        //number_of_detector_as_same_side\n" % nSide)
f.write("0.00     //1st_detector_near_position_in_reference_to_det6_[mm]\n")
for a in pos:
  f.write("%s\n" % a)
f.write("#============= end of file")
f.close()

#================== if DWBA, write DWBA.txt, else write Ex.txt
fn4 = "sim_DWBA.txt"
'''
beamName=""
hRecoil_A = int(beam_A) + int(target_A) - int(lRecoil_A)
hRecoil_Z = int(beam_Z) + int(target_Z) - int(lRecoil_Z)
hRecoilName=""

targetName=""
lRecoilName=""

if target_A == "1" and target_Z == "0":
   targetName="n"
if target_A == "1" and target_Z == "1":
   targetName="p"
if target_A == "2" and target_Z == "1":
   targetName="d"
if target_A == "3" and target_Z == "1":
   targetName="t"
if target_A == "3" and target_Z == "2":
   targetName="3He"
if target_A == "4" and target_Z == "2":
   targetName="a"


if lRecoil_A == "1" and lRecoil_Z == "0":
   lRecoilName="n"
if lRecoil_A == "1" and lRecoil_Z == "1":
   lRecoilName="p"
if lRecoil_A == "2" and lRecoil_Z == "1":
   lRecoilName="d"
if lRecoil_A == "3" and lRecoil_Z == "1":
   lRecoilName="t"
if lRecoil_A == "3" and lRecoil_Z == "2":
   lRecoilName="3He"
if lRecoil_A == "4" and lRecoil_Z == "2":
   lRecoilName="a"
'''

hRecoil = [ beam[0] + target[0] - lRecoil[0], beam[1] + target[1] - lRecoil[1] ]
hRecoil_AZ = FindName(hRecoil[0], hRecoil[1]);

if isDWBA :
  #===== find the symbol of Beam
  reactionName = beam_AZ + "(" + target_AZ +","+lRecoil_AZ+")" + hRecoil_AZ;

  f = open(fn4, "w")

  if isinstance(ExList, list):
     nState = len(ExList)
     for i in range(0, nState):
       f.write("%s  %s   %s   %s    %.3f   %sMeV/u   %s%s\n" % (reactionName, beam_Jpi, OrbList[i], JpiList[i], float(ExList[i]), KEA, op1, op2))
  else :
     f.write("%s  %s   %s   %s    %.3f   %sMeV/u   %s%s\n" % (reactionName, beam_Jpi, OrbList, JpiList, float(ExList), KEA, op1, op2))

  f.close()


fnEx="sim_Ex.txt"
if isDWBA == False:
  f = open(fnEx, "w")
  f.write("//Ex   rel_xsec   SF   sigma_in_MeV\n")
  if isinstance(ExList, list):
    for i in range(0, len(ExList)):
      f.write("%.3f  1.0   1.0  0.000\n" % float(ExList[i]))
  else:
    f.write("%.3f 1.0  1.0  0.000\n" % float(ExList))
  f.write("#==============_End_of_file")
  f.close()

#================== write PlotConfig.txt (to be done)
fn5=""

#==================== Run Simulation
dwba_1=""
outFile=""
dwba_3=""
dwba_4=""
isPtolemy=False
dwbaPNG=""
dwbaRootFile=""
dwbaExFile=""
pngName=""
result=""
haha=""

if isDWBA :
  dwba_1 = subprocess.run(["../Cleopatra/InFileCreator", fn4, "0", "180", "0.1"] , stdout=subprocess.PIPE).stdout.decode('utf-8')
  cmd = "../Cleopatra/ptolemy <%s.in> %s.out" % (fn4, fn4)
  outFile = fn4+ ".out"
  os.system(cmd)
  if os.path.exists(outFile) :
    dwba_3 = subprocess.run(["../Cleopatra/ExtractXSec", outFile, "1"], stdout=subprocess.PIPE).stdout.decode('utf-8')
    dwbaRootFile = fn4 + ".root"
    dwbaExFile = fn4 + ".Ex.txt"
    dwba_4 = subprocess.run(["../Cleopatra/PlotTGraphTObjArray", dwbaRootFile, "1"], stdout=subprocess.PIPE).stdout.decode('utf-8')
    dwbaPNG = dwba_4.splitlines().pop()


if isDWBA :
  result = subprocess.run(["../Cleopatra/Transfer", fn1, fn2, dwbaExFile, dwbaRootFile, "transfer.root", "reaction.dat", "0"], stdout=subprocess.PIPE).stdout.decode('utf-8')    
else:
  result = subprocess.run(["../Cleopatra/Transfer", fn1, fn2, fnEx, "", "transfer.root", "reaction.dat", "0"], stdout=subprocess.PIPE).stdout.decode('utf-8')


if fn5 =="":
   haha = subprocess.run(["../Cleopatra/PlotSimulation", "transfer.root"],  stdout=subprocess.PIPE).stdout.decode('utf-8')
else:
   haha = subprocess.run(["../Cleopatra/PlotSimulation", "transfer.root", fn5],  stdout=subprocess.PIPE).stdout.decode('utf-8')
pngName=haha.splitlines().pop()



print ("Content-Type: text/html\r\n\r\n")
print ("<html>")
print ("<style> img {max-width: 100%} </style>")
print ("<style> body { font-family: courier, courier new, serif;} </style>")
#print ("<style> a { color: #F7CF3C;} </style>")
print ("<style> a { color: #1364A7;} </style>")
print ("<body>")

'''
print("<br> %s" % beam_A)
print("<br> %s" % beam_Z)
print("<br> %s" % beam_Jpi)
print("<br> %s" % beam_Ex)
print("<br> %s" % target_A)
print("<br> %s" % target_Z)
print("<br> %s" % lRecoil_A)
print("<br> %s" % lRecoil_Z)
print("<br> %s" % KEA)
print("<br> %s" % BField)
print("<br> %s" % posArray)
print("<br> %s" % posRecoil)
print("<br> %s" % SSType)
print("<br> do DWBA : |%s|" % isDWBA)
print("<br> %s" % ExList)
print("<br> %s" % JpiList)
print("<br> %s" % OrbList)
print("<br> %s" % op1)
print("<br> %s" % op2)
print("<br> %s" % plot)
print("<br> %s" % gate)
'''


if dwbaPNG != "" :
  print ("<img src=\"files/%s\">" % dwbaPNG)
  print ("<br> Downlaod the <a href=\"files/%s\" download=\"%s\">%s</a>" % (fn4, fn4, fn4))
  print ("<br> Downlaod the <a href=\"files/%s\" download=\"%s\">%s</a>" % (fn4+".in", fn4+".in", fn4+".in"))
  print ("<br> Downlaod the <a href=\"files/%s\" download=\"%s\">%s</a>" % (fn4+".out", fn4+".out", fn4+".out"))
  print ("<br> Downlaod the <a href=\"files/%s\" download=\"%s\">%s</a>" % (fn4+".Xsec.txt", fn4+".Xsec.txt", fn4+".Xec.txt"))
  print ("<br> Downlaod the <a href=\"files/%s\" download=\"%s\">%s</a>" % (fn4+".root", fn4+".root", fn4+".root"))
  print ("<br>")

if pngName != "" :
  print ("<img src=\"files/%s\">" % pngName)
  print ("<br> Downlaod the <a href=\"files/%s\" download=\"%s\">reactionConfig file</a>" % (fn1, fn1))
  print ("<br> Downlaod the <a href=\"files/%s\" download=\"%s\">detectorGeo file</a>" % (fn2, fn2))
  print ("<br> Downlaod the <a href=\"files/%s\" download=\"%s\">Ex file</a>" % (fnEx, fnEx))
  print ("<br> Downlaod the simulation <a href=\"files/transfer.root\" download=\"transfer.root\">root file</a>")

print ("<br><br><br>#################################### programs text output")
if isDWBA:
  print ("<br><span style=\"white-space: pre-line\"> %s </span>" % dwba_1)
  print ("<br><span style=\"white-space: pre-line\"> %s </span>" % dwba_3)
  print ("<br><span style=\"white-space: pre-line\"> %s </span>" % dwba_4)
print ("<br><span style=\"white-space: pre-line\"> %s </span>" % result)
print ("<br><span style=\"white-space: pre-line\"> %s </span>" % haha)
print ("</body>")
print ("</html>")