import foton
import ezca

ezca = ezca.Ezca()

design_zp = f'zpk([10.6],[0.38],1,"n")'
fd_zp = foton.FilterDesign(design_zp,16384)

for iOptic in ['SR3','PR3']:
    print(' ')
    print('    Starting with '+iOptic)
    print(' ')
    
    # foton_file_obj = foton.FilterFile('/opt/rtcds/userapps/release/sus/h1/filterfiles/H1SUS'+iOptic+'.txt');    
    foton_file_obj = foton.FilterFile(F'/opt/rtcds/userapps/release/sus/h1/filterfiles/H1SUS{iOptic}.txt')
    #for iStage in ['M1','M2','M3']:
    for iStage in ['M1','M2','M3']:
        if iStage == 'M1':
            #osemList = ['H1','H2','H3','V1','V2','V3']
            #osemList = ['LF','RT','SD']
            
            osemList = ['T1','T2','T3','LF','RT','SD']
        else:
            osemList = ['UL','LL','UR','LR']
        for iOSEM in osemList:
            foton_file_obj[f'{iOptic}_{iStage}_OSEMINF_{iOSEM}'][0].set_filterdesign(fd_zp)
            foton_file_obj[f'{iOptic}_{iStage}_OSEMINF_{iOSEM}'][0].name = '10.6:0.38'

            foton_file_obj[iOptic+'_'+iStage+'_WD_OSEMAC_BANDLIM_'+iOSEM][5].set_filterdesign(fd_zp)
            foton_file_obj[iOptic+'_'+iStage+'_WD_OSEMAC_BANDLIM_'+iOSEM][5].name = '10.6:0.38'
            
    foton_file_obj.write()
    print(' ')
    print('    Done with '+iOptic)
    print(' ')