TITLE, "A-Line for Hadron Production Target (May 2009)"

  OPTION, -ECHO, DOUBLE, VERIFY
  !ASSIGN, PRINT="AlineE.print"
  !ASSIGN, ECHO="AlineE.echo"

! ==============================================================================
! global parameters
! ------------------------------------------------------------------------------

  Cb   := 1.e10/CLIGHT
  E0   := 13.64 !LCLS design
  Brho := Cb*E0

! LCLS BEAM definition (Yuantao Ding data)

  EMITXN := 1.2E-06 !normalized horizontal emittance (m)
  EMITYN := 0.7E-06 !normalized vertical emittance (m)
  BLENG  := 9.7E-06 !bunch length (m)
  ESPRD  := 5.8E-04 !energy spread (1)

  EMITX := EMITXN/(E0/EMASS)
  EMITY := EMITYN/(E0/EMASS)

  BEAM, PARTICLE=ELECTRON, ENERGY=E0, EX=EMITX, EY=EMITY, &
        SIGT=BLENG, SIGE=ESPRD

! LCLS design Twiss at DBMARK29 (M. Woodley data)

  !TBETX := 31.226097276424 !twiss beta x (m)
  !TBETY := 62.913645471612 !twiss beta y (m)
  !TALFX :=  0.550967214803 !twiss alpha x
  !TALFY := -1.317155083937 !twiss alpha y

! LCLS design Twiss at DBMARK29 (Yuantao Ding data)

  TBETX := 32.0290 !twiss beta x (m)
  TBETY := 67.4900 !twiss beta y (m)
  TALFX :=  0.7820 !twiss alpha x
  TALFY := -1.4850 !twiss alpha y

  TWSS0 : BETA0, BETX=TBETX, ALFX=TALFX, BETY=TBETY, ALFY=TALFY, &
                 ENERGY=E0

! LCLS design Twiss at S100 (M. Woodley data)

  !BXS100 := 35.295438648732
  !BYS100 := 56.287363531453
  !AXS100 := -1.228588949606
  !AYS100 :=  1.87187487513

! LCLS design Twiss at S100 (Yuantao Ding data)

  BXS100 := 35.232048931311
  BYS100 := 60.653598326647
  AXS100 := -1.010953264738
  AYS100 :=  1.949919261099

  GXS100 := (1+AXS100*AXS100)/BXS100
  GYS100 := (1+AYS100*AYS100)/BYS100
  SIG11  :=  EMITX*BXS100
  SIG21  := -EMITX*AXS100
  SIG22  :=  EMITX*GXS100
  SIG33  :=  EMITY*BYS100
  SIG43  := -EMITY*AYS100
  SIG44  :=  EMITY*GYS100
  C21    :=  SIG21/SQRT(SIG11*SIG22)
  C43    :=  SIG43/SQRT(SIG33*SIG44)

  SIGS100 : SIGMA0, SIGX=SQRT(SIG11), SIGPX=SQRT(SIG22), R21=C21, &
                    SIGY=SQRT(SIG33), SIGPY=SQRT(SIG44), R43=C43, &
                    SIGT=BLENG,       SIGPT=ESPRD

  TWS100 : BETA0, BETX=BXS100, ALFX=AXS100, BETY=BYS100, ALFY=AYS100, &
                  ENERGY=E0

! Initial XYZ coordinates at DBMARK29 (per T. Fieguth)

  XM29  :=  0
  YM29  :=  0.0099335
  ZM29  := -2.085976
  THM29 :=  0
  PHM29 := -0.004762
  PSM29 :=  0

! Based on the above, the initial XYZ coordinates at S100 are

  XS100  :=  0
  YS100  :=  5.543152234E-09
  ZS100  :=  3.483050032E-07
  THS100 :=  0
  PHS100 := -0.004762
  PSS100 :=  0

! ------------------------------------------------------------------------------
! Gaussian 1-sigma size including 1st and 2nd order dispersion
! ------------------------------------------------------------------------------

 sigmax: string, "sqrt(beam[ex]*betx+dx*dx*beam[sige]*beam[sige]+
         0.5*ddx*ddx*beam[sige]*beam[sige]*beam[sige]*beam[sige])"
 sigmay: string, "sqrt(beam[ey]*bety+dy*dy*beam[sige]*beam[sige]+
         0.5*ddy*ddy*beam[sige]*beam[sige]*beam[sige]*beam[sige])"

! ==============================================================================
! load XSIF file for existing A-line
! ------------------------------------------------------------------------------

  CALL, FILENAME="AlineE_original.xsif"

! ==============================================================================

  SETPLOT, XSIZE=25, YSIZE=15
  SETPLOT, LWIDTH=5, LSCALE=1.5, SSCALE=1.5, RSCALE=1.5

! set ROLLS to zero

  SET, ROLLON, 0.0

! ==============================================================================
! Existing Aline e- optics with original twiss
! ------------------------------------------------------------------------------

COMMENT
    TITLE, "Original A-Line with original Twiss"
    LBETX := 45.859927352919 !twiss beta x (m)
    LALFX := 2.143329866801  !twiss alpha x
    LBETY := 70.863524917589 !twiss beta y (m)
    LALFY := -0.992810572364 !twiss alpha y
    TWSSL : BETA0, BETX=LBETX, ALFX=LALFX, BETY=LBETY, ALFY=LALFY
    USE, BSYALINE
    !PRINT, FULL
    TWISS, COUPLE, BETA0=TWSSL, SAVE
    PLOT, TABLE=TWISS, HAXIS=S, VAXIS=BETX,BETY, &
      STYLE=1, COLOUR=100, SPLINE=.T., RANGE=#S/Q28[2] !, FILE="Aline"
    PLOT, TABLE=TWISS, HAXIS=S, VAXIS=rBETX,rBETY, &
      STYLE=1, COLOUR=100, SPLINE=.T., RANGE=#S/WALL !, FILE="Aline"
    PLOT, TABLE=TWISS, HAXIS=S, VAXIS=DX, &
      STYLE=1, COLOUR=100, SPLINE=.T., RANGE=#S/WALL !, FILE="Aline"
    !TWISS, CHROM, BETA0=TWSSL, SAVE
    !PLOT, TABLE=TWISS, HAXIS=S, VAXIS=DDX,DDY &
    !  STYLE=1, COLOUR=100, SPLINE=.T., RANGE=#S/WALL !, FILE="Aline"
    !MATCH, BETA0=TWSSL
    !ENDMATCH
    STOP
ENDCOMMENT

!!COMMENT !Geometry with existing XYZ (Rolls must be ON)
    SET, ROLLON, 1.0
    USE, BSYALINE
    PRINT, FULL
    SURVEY, Y0=YM29, Z0=ZM29, PHI0=PHM29, TAPE="survey_aline.txt"
!!ENDCOMMENT

COMMENT !Geometry with XYZ=0 at S100
    USE, S100ALINE
    PRINT, FULL
    SURVEY !, TAPE
    STOP
ENDCOMMENT

COMMENT !Geometry with existing XYZ (Rolls must be ON)
    SET, ROLLON, 1.0
    USE, S100ALINE
    PRINT, FULL
    SURVEY, Y0=YS100, Z0=ZS100, PHI0=PHS100, TAPE
    STOP
ENDCOMMENT

! coordinates @ROLL2 in original A-Line SURVEY (deck ROLLs @ zero)
! (M. Woodley)

  !C1 := 6.557772894E-01 !X
  !C2 := 0               !Y
  !C3 := 1.318828501E+02 !Z
  !C4 := 8.738490149E-03 !THETA
  !C5 := 0               !PHI
  !C6 := 0               !PSI

! ==============================================================================
! Existing Aline e- optics with LCLS initial twiss
! ------------------------------------------------------------------------------

COMMENT !Match to alpha=0 at EFOC (s=351m)
    SXE := 0.1e-3
    SYE := 0.1e-3
    DXE := 0
    DYE := 0
    BXE := (SXE*SXE-DXE*DXE*ESPRD*ESPRD)/EMITX
    BYE := (SYE*SYE-DYE*DYE*ESPRD*ESPRD)/EMITY
    VALUE, BXE,BYE
    USE, S100ALINE
    MATCH, BETA0=TWS100
      VARY, KQ30, STEP=1.E-4 !, LOWER=-K1max, UPPER=+K1max
      VARY, KQ38, STEP=1.E-4 !, LOWER=-K1max, UPPER=+K1max
      CONSTR, EFOC, alfx=0, alfy=0
      WEIGHT, betx=1e-4, bety=1e-4
      CONSTR, EFOC, betx<BXE, bety<BYE
      LMDIF, TOL=1.E-20, CALLS=20000
    ENDMATCH
    VALUE, KQ30,KQ38
    SAVE
ENDCOMMENT

COMMENT !Match to alpha=0 at EFOC2 (s=385m)
    SXE := 0.1e-3
    SYE := 0.1e-3
    DXE := 0
    DYE := 0
    BXE := (SXE*SXE-DXE*DXE*ESPRD*ESPRD)/EMITX
    BYE := (SYE*SYE-DYE*DYE*ESPRD*ESPRD)/EMITY
    VALUE, BXE,BYE
    USE, S100ALINE
    MATCH, BETA0=TWS100
      VARY, KQ30, STEP=1.E-4 !, LOWER=-K1max, UPPER=+K1max
      VARY, KQ38, STEP=1.E-4 !, LOWER=-K1max, UPPER=+K1max
      CONSTR, EFOC2, alfx=0, alfy=0
      WEIGHT, betx=1e-4, bety=1e-4
      CONSTR, EFOC2, betx<BXE, bety<BYE
      LMDIF, TOL=1.E-20, CALLS=20000
    ENDMATCH
    VALUE, KQ30,KQ38
    SAVE
ENDCOMMENT

COMMENT !Match point-to-point from PR10 to EFOC2 (s=385m)
    SXE := 0.1e-3
    SYE := 0.1e-3
    DXE := 0
    DYE := 0
    BXE := (SXE*SXE-DXE*DXE*ESPRD*ESPRD)/EMITX
    BYE := (SYE*SYE-DYE*DYE*ESPRD*ESPRD)/EMITY
    VALUE, BXE,BYE
    USE, S100ALINE
    MATCH, BETA0=TWS100
      VARY, KQ30, STEP=1.E-4 !, LOWER=-K1max, UPPER=+K1max
      VARY, KQ38, STEP=1.E-4 !, LOWER=-K1max, UPPER=+K1max
      rmatrix, PR10/EFOC2, rm(1,2)=0, rm(3,4)=0
      WEIGHT, betx=1e-4, bety=1e-4
      CONSTR, EFOC2, betx<BXE, bety<BYE
      LMDIF, TOL=1.E-20, CALLS=20000
    ENDMATCH
    VALUE, KQ30,KQ38
    SAVE
ENDCOMMENT

!!COMMENT !Plot twiss starting from MK29
    TITLE, S="Existing A-line starting from MK29"
    SET, ROLLON, 1.0
    USE, BSYALINE, range=#S/ESAE
    PRINT, FULL
    TWISS, COUPLE, BETA0=TWSS0, SAVE !, TAPE="twiss.tape", RTAPE="rmat.tape"
    PLOT, TABLE=TWISS, HAXIS=S, VAXIS1=rBETX,rBETY, VAXIS2=DX,DY, &
      STYLE=1, COLOUR=100, SPLINE=.T., RANGE=#S/ESAE !, FILE="Aline"
    !PLOT, TABLE=TWISS, HAXIS=S, VAXIS=rBETX,rBETY, &
    !  STYLE=1, COLOUR=100, SPLINE=.T., RANGE=#S/ESAE !, FILE="Aline"
!!ENDCOMMENT

COMMENT !LCLS twiss at S100
    TITLE, "A-Line with LCLS initial twiss"
    USE, MK29S100
    SAVEBETA, PLACE=#E, label=TWS100
    TWISS, COUPLE, BETA0=TWSS0
    VALUE, TWS100[betx],TWS100[bety],TWS100[alfx],TWS100[alfy]
ENDCOMMENT

COMMENT !Plot twiss
    TITLE, "Primary e- in existing A-line with focus at 351 m."
    SET, ROLLON, 1.0
    USE, S100ALINE
    PRINT, FULL
    TWISS, COUPLE, BETA0=TWS100, SAVE !, TAPE
    !PLOT, TABLE=TWISS, HAXIS=S, VAXIS=rBETX,rBETY, &
    !  STYLE=1, COLOUR=100, SPLINE=.T., RANGE=#S/ESAE !, FILE="Aline"
    !PLOT, TABLE=TWISS, HAXIS=S, VAXIS=DX, &
    !  STYLE=1, COLOUR=100, SPLINE=.T., RANGE=#S/ESAE !, FILE="Aline"
    PLOT, TABLE=TWISS, HAXIS=S, VAXIS1=rBETX,rBETY, VAXIS2=DX,DY, &
      STYLE=1, COLOUR=100, SPLINE=.T., RANGE=#S/ESAE !, FILE="Aline"
    !TWISS, CHROM, BETA0=TWS100, SAVE
    !PLOT, TABLE=TWISS, HAXIS=S, VAXIS=DDX,DDY &
    !  STYLE=1, COLOUR=100, SPLINE=.T., RANGE=#S/ESAE !, FILE="Aline"
    !PLOT, TABLE=TWISS, HAXIS=S, VAXIS=SIGMAX,SIGMAY &
    !  STYLE=1, COLOUR=100, SPLINE=.F., RANGE=#S/ESAE !, FILE="Aline"
    !SELECT, OPTICS, full
    !ENVELOPE, SIGMA0=SIGS100, SAVE=ENVELOPE
    !PLOT, TABLE=ENVELOPE, HAXIS=S, VAXIS=SIGX,SIGY, &
    !  STYLE=1, COLOUR=100, SPLINE=.F., RANGE=#S/ESAE !, FILE="Aline"
    !MATCH, BETA0=TWS100
    !ENDMATCH
ENDCOMMENT

COMMENT !Twiss and beam size at EFOC and EFOC2
    USE, S100ALINE
    SAVEBETA, PLACE=EFOC, label=TWSSEF
    SAVEBETA, PLACE=EFOC2, label=TWSSEF2
    TWISS, COUPLE, BETA0=TWS100
    VALUE, TWSSEF[betx],TWSSEF[bety],TWSSEF[alfx],TWSSEF[alfy],TWSSEF[dx]
    VALUE, sqrt(EMITX*TWSSEF[betx]+ESPRD*ESPRD*TWSSEF[dx]*TWSSEF[dx])*1e6
    VALUE, sqrt(EMITY*TWSSEF[bety]+ESPRD*ESPRD*TWSSEF[dy]*TWSSEF[dy])*1e6
    VALUE, TWSSEF2[betx],TWSSEF2[bety],TWSSEF2[alfx],TWSSEF2[alfy],TWSSEF2[dx]
    VALUE, sqrt(EMITX*TWSSEF2[betx]+ESPRD*ESPRD*TWSSEF2[dx]*TWSSEF2[dx])*1e6
    VALUE, sqrt(EMITY*TWSSEF2[bety]+ESPRD*ESPRD*TWSSEF2[dy]*TWSSEF2[dy])*1e6
ENDCOMMENT

COMMENT
  USE, S100EFOC
  SAVELINE, NAME="S100EFOC", FILENAME="ES100toEFOC.xsif"
  TWISS, COUPLE, BETA0=TWS100
ENDCOMMENT

! ------------------------------------------------------------------------------
! For secondary e- beam produced at PR10

COMMENT
  USE, S100ALINE, range=PR10/#E
  PRINT, FULL
  SURVEY, X0=2.057349314E+01, THETA0=-4.188790205E-01, TAPE
  SELECT, OPTICS, FULL
  OPTICS, BETA0=TWS100, COLUMNS=S,KEYWORD,NAME,L,K0L,K1L
  STOP
ENDCOMMENT

COMMENT
  USE, S100ALINE, range=#S/PR10
  PRINT, FULL
  TWISS, COUPLE, BETA0=TWS100, SAVE, TAPE
  PLOT, HAXIS=S, VAXIS=rBETX,rBETY, STYLE=1, COLOUR=100, SPLINE=.T.
  PLOT, HAXIS=S, VAXIS=DX, STYLE=1, COLOUR=100, SPLINE=.T.
  TWISS, CHROM, BETA0=TWS100, SAVE
  PLOT, HAXIS=S, VAXIS=SIGMAX,SIGMAY, STYLE=1, COLOUR=100, SPLINE=.F.
  !SELECT, OPTICS, range=#S/PR10
  !ENVELOPE, SIGMA0=SIGS100, SAVE=ENVELOPE
  !PLOT, TABLE=ENVELOPE, HAXIS=S, VAXIS=SIGX,SIGY, &
  !  STYLE=1, COLOUR=100, SPLINE=.F.
ENDCOMMENT

COMMENT
  KQ30 := -0.026277669054 !for r12=r34=0 from PR10 to EFOC2
  KQ38 :=  0.019139729274
  SIGXH  := 0.306E-3      !horiz. rms size (m)
  SIGYH  := 0.186E-3      !vert. rms size (m)
  SIGPXH := 0.14E-3       !horiz. rms divergence (rad)
  SIGPYH := 0.14E-3       !vert. rms divergence (rad)
  EMITX  := SIGXH*SIGPXH  !horiz. emittance (m)
  EMITY  := SIGYH*SIGPYH  !vert. emittance (m)
  ESPRD  := 0.01          !rms energy spread (1)
  TBETX  := SIGXH/SIGPXH  !twiss beta x (m)
  TALFX  := 0.0           !twiss alpha x
  TBETY  := SIGYH/SIGPYH  !twiss beta y (m)
  TALFY  := 0.0           !twiss alpha y
  BEAM, ENERGY=E0, EX=EMITX, EY=EMITY, SIGT=BLENG, SIGE=ESPRD
  TITLE, "Secondary e- in existing A-line with focus at 385 m."
  USE, S100ALINE, range=PR10/ESAE
  PRINT, FULL
  SAVEBETA, PLACE=EFOC2, label=TWSSE
  TWISS, COUPLE, BETA0=TWSS0, SAVE
  !PLOT, HAXIS=S, VAXIS=rBETX,rBETY, STYLE=1, COLOUR=100, SPLINE=.T.
  !PLOT, HAXIS=S, VAXIS=DX, STYLE=1, COLOUR=100, SPLINE=.T.
  !TWISS, COUPLE, BETA0=TWSS0, SAVE, deltap=-0.01
  !PLOT, HAXIS=S, VAXIS=rBETX,rBETY, STYLE=1, COLOUR=100, SPLINE=.T.
  !TWISS, COUPLE, BETA0=TWSS0, SAVE, deltap=+0.01
  !PLOT, HAXIS=S, VAXIS=rBETX,rBETY, STYLE=1, COLOUR=100, SPLINE=.T.
  TWISS, CHROM, BETA0=TWSS0, SAVE
  PLOT, HAXIS=S, VAXIS=SIGMAX,SIGMAY, STYLE=1, COLOUR=100, SPLINE=.F.
  !MATCH, BETA0=TWSS0
  !RMATRIX, PR10/EFOC2, RM(6,6)=1
  !ENDMATCH
  VALUE, EMITX,EMITY
  VALUE, TWSSE[betx],TWSSE[bety],TWSSE[alfx],TWSSE[alfy],TWSSE[dx]
  VALUE, sqrt(EMITX*TWSSE[betx]+ESPRD*ESPRD*TWSSE[dx]*TWSSE[dx])*1e6
  VALUE, sqrt(EMITY*TWSSE[bety]+ESPRD*ESPRD*TWSSE[dy]*TWSSE[dy])*1e6
ENDCOMMENT

COMMENT
  USE, S100ALINE, range=PR10/#E
  !SAVELINE, NAME="ALINE", FILENAME="Aline.xsif"
  !TWISS, COUPLE, BETA0=TWS100
ENDCOMMENT

COMMENT ! to test Takashi's particles
  USE, S100ALINE, range=PR10/#E
  print, full
  TWISS, COUPLE, BETA0=TWS100, save, &
    px=3.1100E-4, py=1.9800E-4, deltap=-4.4723E-3
  PLOT, HAXIS=S, VAXIS=X,Y, STYLE=1, COLOUR=100, SPLINE=.T., &
    vmin=-0.15, vmax=0.05
ENDCOMMENT

! ------------------------------------------------------------------------------

  STOP

! ==============================================================================
! load XSIF file for electrons to hadron target
! ------------------------------------------------------------------------------

  CALL, FILENAME="AlineE_target.xsif"

! ==============================================================================
! Aline e- optics to hadron target with LCLS initial twiss
! ------------------------------------------------------------------------------

COMMENT !Match to beta waist at target
    SXT := 0.1e-3
    SYT := 0.1e-3
    DXT := 0
    DYT := 0
    BXT := (SXT*SXT-DXT*DXT*ESPRD*ESPRD)/EMITX
    BYT := (SYT*SYT-DYT*DYT*ESPRD*ESPRD)/EMITY
    VALUE, BXT,BYT
    USE, S100TGT
    MATCH, BETA0=TWS100
      VARY, KQ27, STEP=1.E-4 !, LOWER=-K1max, UPPER=+K1max
      VARY, KQ28, STEP=1.E-4 !, LOWER=-K1max, UPPER=+K1max
      !WEIGHT, alfx=1e-6, alfy=1e-6
      CONSTR, MTGT, alfx=0, alfy=0
      WEIGHT, betx=1e-4, bety=1e-4
      CONSTR, MTGT, betx<BXT, bety<BYT
      LMDIF, TOL=1.E-20, CALLS=20000
    ENDMATCH
    VALUE, KQ27,KQ28
    SAVE
ENDCOMMENT

COMMENT !Geometry with XYZ=0 at S100
    USE, S100TGT
    PRINT, FULL
    SURVEY, TAPE
    STOP
ENDCOMMENT

COMMENT !Geometry with existing XYZ (Rolls must be ON)
    SET, ROLLON, 1.0
    USE, S100TGT
    PRINT, FULL
    SURVEY, Y0=YS100, Z0=ZS100, PHI0=PHS100, TAPE
    STOP
ENDCOMMENT

!!COMMENT !Plot twiss
    TITLE, "Electrons from S100 to hadron target"
    USE, S100TGT
    !PRINT, FULL
    !SURVEY, TAPE="surveyEtoTGT"
    TWISS, BETA0=TWS100, SAVE, COUPLE
    PLOT, TABLE=TWISS, HAXIS=S, VAXIS=rBETX,rBETY, &
      STYLE=1, COLOUR=100, SPLINE=.T., RANGE=#S/#E !, FILE="Aline"
    PLOT, TABLE=TWISS, HAXIS=S, VAXIS=DX,DY, &
      STYLE=1, COLOUR=100, SPLINE=.T., RANGE=#S/#E !, FILE="Aline"
    TWISS, CHROM, BETA0=TWS100, SAVE
    PLOT, TABLE=TWISS, HAXIS=S, VAXIS=DDX,DDY &
      STYLE=1, COLOUR=100, SPLINE=.T., RANGE=#S/#E !, FILE="Aline"
    PLOT, TABLE=TWISS, HAXIS=S, VAXIS=SIGMAX,SIGMAY &
      STYLE=1, COLOUR=100, SPLINE=.F., RANGE=#S/#E !, FILE="Aline"
    SELECT, OPTICS, full
    ENVELOPE, SIGMA0=SIGS100, SAVE=ENVELOPE
    PLOT, TABLE=ENVELOPE, HAXIS=S, VAXIS=SIGX,SIGY, &
      STYLE=1, COLOUR=100, SPLINE=.F., RANGE=#S/#E !, FILE="Aline"
    !MATCH, BETA0=TWS100
    !rmatrix, BEGB/ENDB, rm(6,6)=1
    !ENDMATCH
!!ENDCOMMENT

!!COMMENT !Twiss and beam size at target
    USE, S100TGT
    SAVEBETA, PLACE=MTGT[1], label=TWTGT
    TWISS, COUPLE, BETA0=TWS100
    VALUE, TWTGT[betx],TWTGT[bety],TWTGT[alfx],TWTGT[alfy],TWTGT[dx]
    VALUE, sqrt(EMITX*TWTGT[betx]+ESPRD*ESPRD*TWTGT[dx]*TWTGT[dx])*1e6
    VALUE, sqrt(EMITY*TWTGT[bety]+ESPRD*ESPRD*TWTGT[dy]*TWTGT[dy])*1e6
!!ENDCOMMENT

COMMENT
  USE, S100TGT
  SAVELINE, NAME="S100TGT", FILENAME="S100TGT.xsif"
  STOP
ENDCOMMENT

! ------------------------------------------------------------------------------

  STOP

! ------------------------------------------------------------------------------





! ==============================================================================
! load XSIF file for electrons to ESA
! ------------------------------------------------------------------------------
! Aline e- optics to ESA with LCLS initial twiss

  !CALL, FILENAME="AlineE_ESAv0.xsif"   !with existing Q30,Q38

  CALL, FILENAME="AlineH_ESAv1.xsif"
  CALL, FILENAME="AlineE_ESAv1.xsif"

! ==============================================================================

COMMENT !Match e- to beta waist at s~351m
    SXE := 0.1e-3
    SYE := 0.1e-3
    DXE := 0
    DYE := 0
    BXE := (SXE*SXE-DXE*DXE*ESPRD*ESPRD)/EMITX
    BYE := (SYE*SYE-DYE*DYE*ESPRD*ESPRD)/EMITY
    VALUE, BXE,BYE
    USE, S100ESA
    MATCH, BETA0=TWS100
      VARY, KQ30, STEP=1.E-4 !, LOWER=-K1max, UPPER=+K1max
      VARY, KQ38, STEP=1.E-4 !, LOWER=-K1max, UPPER=+K1max
      !VARY, KQH1, STEP=1.E-4 !, LOWER=-K1max, UPPER=+K1max
      !VARY, KQH2, STEP=1.E-4 !, LOWER=-K1max, UPPER=+K1max
      CONSTR, EFOC, alfx=0, alfy=0
      WEIGHT, betx=1e-3, bety=1e-3
      CONSTR, EFOC, betx<BXE, bety<BYE
      !WEIGHT, betx=1e-6, bety=1e-6
      !CONSTR, QH1[1], betx<100, bety<100
      !CONSTR, QH2[1], betx<100, bety<100
      LMDIF, TOL=1.E-20, CALLS=20000
    ENDMATCH
    VALUE, KQ30,KQ38,KQH1,KQH2,KQH3,KQH4
    SAVE
ENDCOMMENT

COMMENT !Verify/adjust geometry in the new e- chicane
    USE, (Q27B3,NEWB3,D123G,NEWB3)
    PRINT, FULL
    SURVEY !, TAPE
    STOP
ENDCOMMENT

COMMENT !Verify lengths
    USE, (Q27B3,NEWB3,D123G,NEWB3)
    SELECT, OPTICS, FULL
    OPTICS, BETX=1, BETY=1, COLUMNS=S,KEYWORD,NAME,L
    STOP
ENDCOMMENT

COMMENT !Geometry with XYZ=0 at S100
    USE, S100ESA
    PRINT, FULL
    SURVEY !, TAPE
    STOP
ENDCOMMENT

COMMENT !Geometry with existing XYZ (Rolls must be ON)
    SET, ROLLON, 1.0
    USE, S100ESA
    PRINT, FULL
    SURVEY, Y0=YS100, Z0=ZS100, PHI0=PHS100, TAPE
    STOP
ENDCOMMENT

!!COMMENT !Plot twiss
    TITLE, "Electrons from S100 to ESA"
    USE, S100ESA
    !PRINT, FULL
    TWISS, COUPLE, BETA0=TWS100, SAVE
    PLOT, TABLE=TWISS, HAXIS=S, VAXIS=rBETX,rBETY, &
      STYLE=1, COLOUR=100, SPLINE=.T.!, RANGE=#S/HDET !, FILE="Aline"
    PLOT, TABLE=TWISS, HAXIS=S, VAXIS=DX, &
      STYLE=1, COLOUR=100, SPLINE=.T.!, RANGE=#S/HDET !, FILE="Aline"
    !TWISS, CHROM, BETA0=TWS100, SAVE
    !PLOT, TABLE=TWISS, HAXIS=S, VAXIS=DDX,DDY &
    !  STYLE=1, COLOUR=100, SPLINE=.T., RANGE=#S/HDET !, FILE="Aline"
    !PLOT, TABLE=TWISS, HAXIS=S, VAXIS=SIGMAX,SIGMAY &
    !  STYLE=1, COLOUR=100, SPLINE=.F.!, RANGE=#S/HDET !, FILE="Aline"
    !SELECT, OPTICS, full
    !ENVELOPE, SIGMA0=SIGS100, SAVE=ENVELOPE
    !PLOT, TABLE=ENVELOPE, HAXIS=S, VAXIS=SIGX,SIGY, &
    !  STYLE=1, COLOUR=100, SPLINE=.F., RANGE=#S/HDET !, FILE="Aline"
    !MATCH, BETA0=TWS100
    !rmatrix, Q28U[2]/Q30[1], rm(6,6)=1
    !ENDMATCH
!!ENDCOMMENT

!!COMMENT !Twiss and beam size at s~351m
    USE, S100ESA
    SAVEBETA, PLACE=EFOC, label=TWSSE
    TWISS, COUPLE, BETA0=TWS100
    VALUE, TWSSE[betx],TWSSE[bety],TWSSE[alfx],TWSSE[alfy],TWSSE[dx]
    VALUE, sqrt(EMITX*TWSSE[betx]+ESPRD*ESPRD*TWSSE[dx]*TWSSE[dx])*1e6
    VALUE, sqrt(EMITY*TWSSE[bety]+ESPRD*ESPRD*TWSSE[dy]*TWSSE[dy])*1e6
!!ENDCOMMENT

!!COMMENT
  USE, S100ESA
  SAVELINE, NAME="S100ESA", FILENAME="EtoEFOCv1.xsif"
  STOP
!!ENDCOMMENT

! ==============================================================================

  STOP