program multipass_test

use bmad

implicit none

type (lat_struct), target :: lat
type (ele_struct), pointer :: ele

! Init
call bmad_parser ('multipass_and_superimpose.bmad', lat)

open (1, file = 'output.now')

! Forking with a branch element

call bmad_parser ('branch_fork.bmad', lat)
call write_bmad_lattice_file ('lat1.bmad', lat)
call bmad_parser ('lat1.bmad', lat)

ele => lat%branch(1)%ele(2)
write (1, '(3a)')       '"BF-01"  STR  "', trim(ele%name), '"'
write (1, '(a, 2i3)')   '"BF-02"  ABS   0 ', nint(ele%value(ix_to_branch$)), nint(ele%value(ix_to_element$))
write (1, '(a, f10.4)') '"BF-03"  REL  1e-12 ', ele%value(p0c$)

! Multipass and superimpose

call bmad_parser ('multipass_and_superimpose.bmad', lat)

write (1, '(3a)') '"MS-01"  STR  "', trim(lat%ele(01)%name), '"'
write (1, '(3a)') '"MS-02"  STR  "', trim(lat%ele(02)%name), '"'
write (1, '(3a)') '"MS-03"  STR  "', trim(lat%ele(03)%name), '"'
write (1, '(3a)') '"MS-04"  STR  "', trim(lat%ele(04)%name), '"'
write (1, '(3a)') '"MS-05"  STR  "', trim(lat%ele(05)%name), '"'
write (1, '(3a)') '"MS-06"  STR  "', trim(lat%ele(06)%name), '"'
write (1, '(3a)') '"MS-07"  STR  "', trim(lat%ele(07)%name), '"'
write (1, '(3a)') '"MS-08"  STR  "', trim(lat%ele(08)%name), '"'
write (1, '(3a)') '"MS-09"  STR  "', trim(lat%ele(09)%name), '"'
write (1, '(3a)') '"MS-10"  STR  "', trim(lat%ele(10)%name), '"'
write (1, '(3a)') '"MS-11"  STR  "', trim(lat%ele(11)%name), '"'
write (1, '(3a)') '"MS-12"  STR  "', trim(lat%ele(12)%name), '"'
write (1, '(3a)') '"MS-13"  STR  "', trim(lat%ele(13)%name), '"'

! fiducial and flexible patch

call bmad_parser ('patch.bmad', lat)
call write_bmad_lattice_file ('lat2.bmad', lat)
call bmad_parser ('lat2.bmad', lat)

write (1, '(a, f12.6)')  '"P-0S" ABS 0', lat%branch(1)%ele(0)%s
write (1, '(a, es14.6)') '"P-0T" ABS 0', lat%branch(1)%ele(0)%ref_time
ele => lat%branch(1)%ele(6)
write (1, '(a, 3f12.6)') '"P-6MI" ABS 0', ele%value(z_offset$), ele%value(x_pitch$), ele%value(y_pitch$)
ele => lat%branch(1)%ele(7)
write (1, '(a, 3f12.6)') '"P-7FR" ABS 0', ele%floor%r
write (1, '(a, 3f12.6)') '"P-7FA" ABS 0', ele%floor%theta, ele%floor%phi, ele%floor%psi


! And close

close (1)

end program