import numpy as np
import struct 
import os


ser_pre = np.fromfile('ser.old', dtype='<i4')

######################
TDin1=220
######################
TDdi=len(ser_pre)/(TDin1*4)

print TDdi

if len(ser_pre) % (TDin1*4) != 0:
   print 'ser file size does NOT match 8 times TD1 block size!'
else:
   ser_temp = np.copy(ser_pre)
   index = np.arange(len(ser_pre)/4,len(ser_pre))
   ser_post1 = np.delete(ser_temp,index)
   ser_post2 = np.copy(ser_post1)
   ser_post3 = np.copy(ser_post1)
   ser_post4 = np.copy(ser_post1)


for i in range(1,TDin1/2+1):
   ######################
   #   FID extraction   #
   ######################
   c = (i-1)*TDdi*8
   index = np.arange(c,c+TDdi,2)
   s1_r = np.take(ser_pre,index)
   index = np.arange(c+1,c+TDdi+1,2)
   s1_i = np.take(ser_pre,index)
   index = np.arange(c+TDdi,c+2*TDdi,2)
   s2_r = np.take(ser_pre,index)
   index = np.arange(c+TDdi+1,c+2*TDdi+1,2)
   s2_i = np.take(ser_pre,index)
   index = np.arange(c+2*TDdi,c+3*TDdi,2)
   s3_r = np.take(ser_pre,index)
   index = np.arange(c+2*TDdi+1,c+3*TDdi+1,2)
   s3_i = np.take(ser_pre,index)
   index = np.arange(c+3*TDdi,c+4*TDdi,2)
   s4_r = np.take(ser_pre,index)
   index = np.arange(c+3*TDdi+1,c+4*TDdi+1,2)
   s4_i = np.take(ser_pre,index)
   index = np.arange(c+4*TDdi,c+5*TDdi,2)
   s5_r = np.take(ser_pre,index)
   index = np.arange(c+4*TDdi+1,c+5*TDdi+1,2)
   s5_i = np.take(ser_pre,index)
   index = np.arange(c+5*TDdi,c+6*TDdi,2)
   s6_r = np.take(ser_pre,index)
   index = np.arange(c+5*TDdi+1,c+6*TDdi+1,2)
   s6_i = np.take(ser_pre,index)
   index = np.arange(c+6*TDdi,c+7*TDdi,2)
   s7_r = np.take(ser_pre,index)
   index = np.arange(c+6*TDdi+1,c+7*TDdi+1,2)
   s7_i = np.take(ser_pre,index)
   index = np.arange(c+7*TDdi,c+8*TDdi,2)
   s8_r = np.take(ser_pre,index)
   index = np.arange(c+7*TDdi+1,c+8*TDdi+1,2)
   s8_i = np.take(ser_pre,index)
   ######################
   # 1st & 2nd spectra  #
   ######################
   Sa = np.empty(TDdi)
   Sb = np.empty(TDdi)
   Sc = np.empty(TDdi)
   Sd = np.empty(TDdi)
   ######################
   index = np.arange(0,TDdi,2)
   np.put(Sa,index,s1_r)
   index = np.arange(1,TDdi+1,2)
   np.put(Sa,index,s1_i)
   index = np.arange(0,TDdi,2)
   np.put(Sb,index,s2_r)
   index = np.arange(1,TDdi+1,2)
   np.put(Sb,index,s2_i)
   index = np.arange(0,TDdi,2)
   np.put(Sc,index,np.negative(s3_i))
   index = np.arange(1,TDdi+1,2)
   np.put(Sc,index,s3_r)
   index = np.arange(0,TDdi,2)
   np.put(Sd,index,np.negative(s4_i))
   index = np.arange(1,TDdi+1,2)
   np.put(Sd,index,s4_r)
   ######################
   temp1 = np.add(Sa,Sb)
   temp2 = np.add(temp1,Sc)
   ss1_r = np.subtract(temp2,Sd)
   temp1 = np.add(Sa,Sb)
   temp2 = np.subtract(temp1,Sc)
   ss2_r = np.add(temp2,Sd)
   ######################
   index = np.arange(0,TDdi,2)
   np.put(Sa,index,np.negative(s1_i))
   index = np.arange(1,TDdi+1,2)
   np.put(Sa,index,s1_r)
   index = np.arange(0,TDdi,2)
   np.put(Sb,index,np.negative(s2_i))
   index = np.arange(1,TDdi+1,2)
   np.put(Sb,index,s2_r)
   index = np.arange(0,TDdi,2)
   np.put(Sc,index,s3_r)
   index = np.arange(1,TDdi+1,2)
   np.put(Sc,index,s3_i)
   index = np.arange(0,TDdi,2)
   np.put(Sd,index,s4_r)
   index = np.arange(1,TDdi+1,2)
   np.put(Sd,index,s4_i)
   ######################
   temp1 = np.subtract(Sa,Sb)
   temp2 = np.subtract(temp1,Sc)
   ss1_i = np.subtract(temp2,Sd)
   temp1 = np.subtract(Sb,Sa)
   temp2 = np.subtract(temp1,Sc)
   ss2_i = np.subtract(temp2,Sd)
   ######################
   #    1st spectrum    #
   ######################
   ss1_ri = np.concatenate((ss1_r,ss1_i))
   index = np.arange((i-1)*TDdi*2,i*TDdi*2)
   if len(ss1_ri) != len(ser_post1)/(TDin1/2):
      print 'ss1_ri size does NOT match TD1 block size!'
   else:
      np.put(ser_post1,index,ss1_ri)
   ######################
   #    2nd spectrum    #
   ######################
   #ss2_ri = np.concatenate((ss2_r,ss2_i))
   ss2_ri = np.concatenate((ss2_r,np.negative(ss2_i)))
   #ss2_ri = np.concatenate((ss2_r,np.negative(ss2_i)))
   index = np.arange((i-1)*TDdi*2,i*TDdi*2)
   if len(ss2_ri) != len(ser_post2)/(TDin1/2):
      print 'ss2_ri size does NOT match TD1 block size!'
   else:
      np.put(ser_post2,index,ss2_ri)
   ######################
   # 3rd & 4th spectra  #
   ######################
   Sa = np.empty(TDdi)
   Sb = np.empty(TDdi)
   Sc = np.empty(TDdi)
   Sd = np.empty(TDdi)
   ######################
   index = np.arange(0,TDdi,2)
   np.put(Sa,index,s5_r)
   index = np.arange(1,TDdi+1,2)
   np.put(Sa,index,s5_i)
   index = np.arange(0,TDdi,2)
   np.put(Sb,index,s6_r)
   index = np.arange(1,TDdi+1,2)
   np.put(Sb,index,s6_i)
   index = np.arange(0,TDdi,2)
   np.put(Sc,index,np.negative(s7_i))
   index = np.arange(1,TDdi+1,2)
   np.put(Sc,index,s7_r)
   index = np.arange(0,TDdi,2)
   np.put(Sd,index,np.negative(s8_i))
   index = np.arange(1,TDdi+1,2)
   np.put(Sd,index,s8_r)
   ######################
   temp1 = np.add(Sa,Sb)
   temp2 = np.add(temp1,Sc)
   ss3_r = np.subtract(temp2,Sd)
   temp1 = np.add(Sa,Sb)
   temp2 = np.subtract(temp1,Sc)
   ss4_r = np.add(temp2,Sd)
   ######################
   index = np.arange(0,TDdi,2)
   np.put(Sa,index,np.negative(s5_i))
   index = np.arange(1,TDdi+1,2)
   np.put(Sa,index,s5_r)
   index = np.arange(0,TDdi,2)
   np.put(Sb,index,np.negative(s6_i))
   index = np.arange(1,TDdi+1,2)
   np.put(Sb,index,s6_r)
   index = np.arange(0,TDdi,2)
   np.put(Sc,index,s7_r)
   index = np.arange(1,TDdi+1,2)
   np.put(Sc,index,s7_i)
   index = np.arange(0,TDdi,2)
   np.put(Sd,index,s8_r)
   index = np.arange(1,TDdi+1,2)
   np.put(Sd,index,s8_i)
   ######################
   temp1 = np.subtract(Sa,Sb)
   temp2 = np.subtract(temp1,Sc)
   ss3_i = np.subtract(temp2,Sd)
   temp1 = np.subtract(Sb,Sa)
   temp2 = np.subtract(temp1,Sc)
   ss4_i = np.subtract(temp2,Sd)
   ######################
   #    3rd spectrum    #
   ######################
   ss3_ri = np.concatenate((ss3_r,ss3_i))
   index = np.arange((i-1)*TDdi*2,i*TDdi*2)
   if len(ss3_ri) != len(ser_post3)/(TDin1/2):
      print 'ss1_ri size does NOT match TD1 block size!'
   else:
      np.put(ser_post3,index,ss3_ri)
   ######################
   #    4th spectrum    #
   ######################
   #ss4_ri = np.concatenate((ss4_r,ss4_i))
   ss4_ri = np.concatenate((ss4_r,np.negative(ss4_i)))
   #ss4_ri = np.concatenate((ss4_r,np.negative(ss4_i)))
   index = np.arange((i-1)*TDdi*2,i*TDdi*2)
   if len(ss4_ri) != len(ser_post4)/(TDin1/2):
      print 'ss2_ri size does NOT match TD1 block size!'
   else:
      np.put(ser_post4,index,ss4_ri)
############################
print len(ser_pre)
print len(ser_post1)
print len(ser_post2)
print len(ser_post3)
print len(ser_post4)

ser_post1.tofile('ser4')
ser_post2.tofile('ser1')
ser_post3.tofile('ser2')
ser_post4.tofile('ser3')