#!/usr/bin/env python

import glob, os, io

src = '/lustre/athorpe/ch4/ORNL_Nature_CH4/Selects_Renamed_2/'
dst = '/lustre/athorpe/ch4/ORNL_Nature_CH4/Selects_Renamed_2/'


os.chdir(src)
for file in glob.glob("*.hdr"):
      #Remove last few lines of header 
      readFile = open(file)
      lines = readFile.readlines()
      print('len(lines):',len(lines))
      print('lines:',lines)
      print('line0:',lines[3])
      print('type lines:',type(lines))    
      # Find indicies for missing results
      indices = [i for i, s in enumerate(lines) if 'map info' in s]
      print('indices:',indices)
      mi_int=indices[0]
      print('mi_int:',mi_int)
      mi=lines[mi_int]
      print('map info:',mi)

      indices2 = [m for m, n in enumerate(lines) if 'data ignore value' in n]
      print('indices:',indices2)
      di_int=indices2[0]
      print('di_int:',di_int)
      di=lines[di_int]
      print('data ignore:',di)


      #test=lines.index('description*')
      #print('test:',test) 
      #if 'ENVI' in lines:
      #  print("Yes")
      #indx1=lines.index("map") 
      #print('indx1:',indx1) 
      #Select only the relevant lines
      
      lines_sel=lines[0:10] #These stay the same
      lines_sel2=lines[mi_int]
      lines_sel3=lines[di_int]      

      #lines_sel=lines[0:12]
      #print('lines_sel:',lines_sel)
      #lines_sel_str=str(lines_sel)
      #print('lines_sel_str:',lines_sel_str)
      ##lines_sel2=lines[14]
      ##lines_sel3=lines[18]
      #lines_sel2_str=str(lines_sel2)
      #from io import StringIO
      #concat=StringIO()
      #concat.write(lines_sel_str)
      #concat.write(lines_sel2_str)
      #print(.join(lines_sel,lines_sel2)
      #print(concat.getvalue())
      #print(lines_sel)
      #print(lines_sel2)
      readFile.close()
      w = open(file,'w')
      print('w:',w)
      w.writelines([item for item in lines_sel[:]])
      w.writelines([item for item in lines_sel2[:]])
      w.writelines([item for item in lines_sel3[:]])
      w.close()