def main():
"""
NAME
IODP_csv_magic.py
DESCRIPTION
converts ODP LIMS sample format files to magic_measurements format files
SYNTAX
IODP_csv_magic.py [command line options]
OPTIONS
-h: prints the help message and quits.
-f FILE: specify input .csv file, default is all in directory
-F FILE: specify output measurements file, default is magic_measurements.txt
-Fsp FILE: specify output er_specimens.txt file, default is er_specimens.txt
-Fsa FILE: specify output er_samples.txt file, default is er_samples.txt
-Fsi FILE: specify output er_sites.txt file, default is er_sites.txt
-A : don't average replicate measurements
INPUTS
IODP .csv file format exported from LIMS database
"""
#
#
version_num=pmag.get_version()
meas_file='magic_measurements.txt'
spec_file='er_specimens.txt'
samp_file='er_samples.txt'
site_file='er_sites.txt'
csv_file=''
ErSpecs,ErSamps,ErSites,ErLocs,ErCits=[],[],[],[],[]
MagRecs=[]
citation="This study"
dir_path,demag='.','NRM'
args=sys.argv
noave=0
depth_method='a'
if '-WD' in args:
ind=args.index("-WD")
dir_path=args[ind+1]
print "dir_path", dir_path
if "-h" in args:
print main.__doc__
sys.exit()
if "-A" in args: noave=1
if '-f' in args:
ind=args.index("-f")
# csv_file=args[ind+1] # original
csv_file=dir_path + '/' + args[ind+1] # LJ
if '-F' in args:
ind=args.index("-F")
meas_file=args[ind+1]
if '-Fsp' in args:
ind=args.index("-Fsp")
spec_file=dir_path+'/'+args[ind+1]
Specs,file_type=pmag.magic_read(spec_file)
else:
spec_file=dir_path+'/'+spec_file
if '-Fsi' in args:
ind=args.index("-Fsi")
site_file=args[ind+1]
if '-Fsa' in args:
ind=args.index("-Fsa")
samp_file=dir_path+'/'+args[ind+1]
ErSamps,file_type=pmag.magic_read(samp_file)
else:
samp_file=dir_path+'/'+samp_file
site_file=dir_path+'/'+site_file
meas_file=dir_path+'/'+meas_file
if csv_file=="":
filelist=os.listdir(dir_path) # read in list of files to import
else:
filelist=[csv_file]
specimens,samples,sites=[],[],[]
MagRecs,SpecRecs,SampRecs,SiteRecs=[],[],[],[]
for samp in ErSamps:
if samp['er_sample_name'] not in samples:
samples.append(samp['er_sample_name'])
SampRecs.append(samp)
print "filelist", filelist # LJ
for file in filelist: # parse each file
if file[-3:].lower()=='csv':
print 'processing: ',file
input=open(file,'rU').readlines()
keys=input[0].replace('\n','').split(',') # splits on underscores
if "Interval Top (cm) on SHLF" in keys:interval_key="Interval Top (cm) on SHLF"
if " Interval Bot (cm) on SECT" in keys:interval_key=" Interval Bot (cm) on SECT"
if "Top Depth (m)" in keys:depth_key="Top Depth (m)"
if "CSF-A Top (m)" in keys:depth_key="CSF-A Top (m)"
if "CSF-B Top (m)" in keys:
comp_depth_key="CSF-B Top (m)" # use this model if available
else:
comp_depth_key=""
if "Demag level (mT)" in keys:demag_key="Demag level (mT)"
if "Demag Level (mT)" in keys: demag_key="Demag Level (mT)"
if "Inclination (Tray- and Bkgrd-Corrected) (deg)" in keys:inc_key="Inclination (Tray- and Bkgrd-Corrected) (deg)"
if "Inclination background + tray corrected (deg)" in keys:inc_key="Inclination background + tray corrected (deg)"
if "Inclination background + tray corrected (\xc2\xb0)" in keys:inc_key="Inclination background + tray corrected (\xc2\xb0)"
if "Declination (Tray- and Bkgrd-Corrected) (deg)" in keys:dec_key="Declination (Tray- and Bkgrd-Corrected) (deg)"
if "Declination background + tray corrected (deg)" in keys:dec_key="Declination background + tray corrected (deg)"
if "Declination background + tray corrected (\xc2\xb0)" in keys:dec_key="Declination background + tray corrected (\xc2\xb0)"
if "Intensity (Tray- and Bkgrd-Corrected) (A/m)" in keys:int_key="Intensity (Tray- and Bkgrd-Corrected) (A/m)"
if "Intensity background + tray corrected (A/m)" in keys:int_key="Intensity background + tray corrected (A/m)"
if "Core Type" in keys:
type="Core Type"
else: type="Type"
for line in input[1:]:
InRec={}
for k in range(len(keys)):InRec[keys[k]]=line.split(',')[k]
try:
run_number=""
inst="ODP-SRM"
volume='15.59' # set default volume to this
MagRec,SpecRec,SampRec,SiteRec={},{},{},{}
expedition=InRec['Exp']
location=InRec['Site']+InRec['Hole']
# Maintain backward compatibility for the ever-changing LIMS format (Argh!)
while len(InRec['Core'])<3:
InRec['Core']='0'+InRec['Core']
if "Last Tray Measurment" in InRec.keys() and "Discrete" in InRec['Last Tray Measurement'] or 'dscr' in csv_file : # assume discrete sample
specimen=expedition+'-'+location+'-'+InRec['Core']+InRec[type]+"-"+InRec['Section']+'-'+InRec['Section Half']+'-'+InRec[interval_key]
else: # mark as continuous measurements
specimen=expedition+'-'+location+'-'+InRec['Core']+InRec[type]+"_"+InRec['Section']+InRec['Section Half']+'-'+InRec[interval_key]
SpecRec['er_expedition_name']=expedition
SpecRec['er_location_name']=location
SpecRec['er_site_name']=specimen
SpecRec['er_citation_names']=citation
for key in SpecRec.keys():SampRec[key]=SpecRec[key]
for key in SpecRec.keys():SiteRec[key]=SpecRec[key]
SampRec['sample_azimuth']='0'
SampRec['sample_dip']='0'
SampRec['sample_core_depth']=InRec[depth_key]
if comp_depth_key!='':
SampRec['sample_composite_depth']=InRec[comp_depth_key]
if "Discrete" in InRec['Last Tray Measurement']:
SampRec['magic_method_codes']='FS-C-DRILL-IODP:SP-SS-C:SO-V'
else:
SampRec['magic_method_codes']='FS-C-DRILL-IODP:SO-V'
SpecRec['er_specimen_name']=specimen
SpecRec['er_sample_name']=specimen
SampRec['er_sample_name']=specimen
SampRec['er_specimen_names']=specimen
SiteRec['er_specimen_names']=specimen
for key in SpecRec.keys():MagRec[key]=SpecRec[key]
# set up measurement record - default is NRM
MagRec['er_analyst_mail_names']=InRec['Test Entered By']
MagRec['magic_software_packages']=version_num
MagRec["treatment_temp"]='%8.3e' % (273) # room temp in kelvin
MagRec["measurement_temp"]='%8.3e' % (273) # room temp in kelvin
MagRec["treatment_ac_field"]=0
MagRec["treatment_dc_field"]='0'
MagRec["treatment_dc_field_phi"]='0'
MagRec["treatment_dc_field_theta"]='0'
MagRec["measurement_flag"]='g' # assume all data are "good"
MagRec["measurement_standard"]='u' # assume all data are "good"
SpecRec['er_specimen_alternatives']=InRec['Text Id']
if 'Sample Area (cm?)' in InRec.keys() and InRec['Sample Area (cm?)']!= "": volume=InRec['Sample Area (cm?)']
if InRec['Run Number']!= "": run_number=InRec['Run Number']
datestamp=InRec['Test Changed On'].split() # date time is second line of file
mmddyy=datestamp[0].split('/') # break into month day year
if len(mmddyy[0])==1: mmddyy[0]='0'+mmddyy[0] # make 2 characters
if len(mmddyy[1])==1: mmddyy[1]='0'+mmddyy[1] # make 2 characters
if len(datestamp[1])==1: datestamp[1]='0'+datestamp[1] # make 2 characters
date='20'+mmddyy[2]+':'+mmddyy[0]+":"+mmddyy[1] +':' +datestamp[1]+":00.00"
MagRec["measurement_date"]=date
MagRec["magic_method_codes"]='LT-NO'
if InRec[demag_key]!="0":
MagRec['magic_method_codes'] = 'LT-AF-Z'
inst=inst+':ODP-SRM-AF' # measured on shipboard in-line 2G AF
treatment_value=float(InRec[demag_key].strip('"'))*1e-3 # convert mT => T
MagRec["treatment_ac_field"]=treatment_value # AF demag in treat mT => T
if InRec['Treatment Type']!="":
if 'Alternating Frequency' in InRec['Treatment Type']:
MagRec['magic_method_codes'] = 'LT-AF-Z'
inst=inst+':ODP-DTECH' # measured on shipboard Dtech D2000
treatment_value=float(InRec['Treatment Value'])*1e-3 # convert mT => T
MagRec["treatment_ac_field"]=treatment_value # AF demag in treat mT => T
elif 'Thermal' in InRec['Treatment Type']:
MagRec['magic_method_codes'] = 'LT-T-Z'
inst=inst+':ODP-TDS' # measured on shipboard Schonstedt thermal demagnetizer
treatment_value=float(InRec['Treatment Value'])+273 # convert C => K
MagRec["treatment_temp"]='%8.3e'%(treatment_value) #
MagRec["measurement_standard"]='u' # assume all data are "good"
vol=float(volume)*1e-6 # convert from cc to m^3
if run_number!="":
MagRec['external_database_ids']=run_number
MagRec['external_database_names']='LIMS'
else:
MagRec['external_database_ids']=""
MagRec['external_database_names']=''
MagRec['measurement_inc']=InRec[inc_key].strip('"')
MagRec['measurement_dec']=InRec[dec_key].strip('"')
intens= InRec[int_key].strip('"')
MagRec['measurement_magn_moment']='%8.3e'%(float(intens)*vol) # convert intensity from A/m to Am^2 using vol
MagRec['magic_instrument_codes']=inst
MagRec['measurement_number']='1'
MagRec['measurement_csd']=''
MagRec['measurement_positions']=''
MagRecs.append(MagRec)
if specimen not in specimens:
specimens.append(specimen)
SpecRecs.append(SpecRec)
if MagRec['er_sample_name'] not in samples:
samples.append(MagRec['er_sample_name'])
SampRecs.append(SampRec)
if MagRec['er_site_name'] not in sites:
sites.append(MagRec['er_site_name'])
SiteRecs.append(SiteRec)
except:
pass
if len(SpecRecs)>0:
pmag.magic_write(spec_file,SpecRecs,'er_specimens')
print 'specimens stored in ',spec_file
if len(SampRecs)>0:
SampOut,keys=pmag.fillkeys(SampRecs)
pmag.magic_write(samp_file,SampOut,'er_samples')
print 'samples stored in ',samp_file
if len(SiteRecs)>0:
pmag.magic_write(site_file,SiteRecs,'er_sites')
print 'sites stored in ',site_file
MagSort=pmag.sortbykeys(MagRecs,["er_specimen_name","treatment_ac_field"])
MagOuts=[]
for MagRec in MagSort:
MagRec["treatment_ac_field"]='%8.3e'%(MagRec['treatment_ac_field']) # convert to string
MagOuts.append(MagRec)
Fixed=pmag.measurements_methods(MagOuts,noave)
pmag.magic_write(meas_file,Fixed,'magic_measurements')
print 'data stored in ',meas_file