Main_PIXE.py

# -*- coding: utf-8 -*-
"""
Created on Wed Jul  3 10:32:14 2019

@author: HP
"""

import os
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import Func_PIXE as fx
import xlsxwriter as xw
import time

time_start=time.time()
folder_name='6.6.19'
input_path='F://桌面文件2019.2.11//Nuclear Physics Study//DataFiles//'+folder_name
output_file='F://桌面文件2019.2.11//Nuclear Physics Study//PyAnalysis//'+folder_name+'_PIXE.xlsx'
filetype='.mca'

def get_filename(path,filetype):
    name=[]
    files=os.listdir(path)
    for i in files:
        if filetype in i: 
            name.append(i)
    return name

filename=get_filename(input_path,filetype)

HeadLine=21
workbook=xw.Workbook(output_file)
worksheet1=workbook.add_worksheet('ElementAnalysis')#表一输出分析过的数据
worksheet2=workbook.add_worksheet('Raw Results')#表二是原始数据集，以供参考
style_plain=workbook.add_format({'align':'center'})#设置普通居中格式
style_bold=workbook.add_format({'align':'center','bold':True})#设置加粗字体格式
NewFormat=np.array([])
row=1
col=0
for item in filename[0:1]:
    counter=filename.index(item)+1
    temp=pd.read_csv(input_path+'//'+item,sep=' ',header=None,skiprows=HeadLine-1,nrows=1,dtype='str')#读出来字符串，先检查HeadLine最后一行是不是<<DATA>>
    if not temp[0][0]=='<<DATA>>':#如果不是<<DATA>>，也就意味着出错了
        print(counter,'HeadLine error, '+item+' skipped')
        continue
    temp=pd.read_csv(input_path+'//'+item,sep=' ',header=None,skiprows=HeadLine,nrows=4096)#读取文件，以空格分隔，没有标题，调过前面的行，读取4096行
    temp=temp.values#dataframe转成array
    Spectrum0=fx.Channel2Energy(temp)#补上calibration的能量值
    Spectrum0_copy=Spectrum0.copy()
    worksheet2.write(0,col,item.replace('.mca','_PIXE'))#输出样品名
    worksheet1.write(row,0,item.replace('.mca','_PIXE'))#输出样品名
    temp=fx.PeakAnalysis_and_Export(Spectrum0_copy,1,col,worksheet1,worksheet2,style_plain,style_bold)#输出峰信息
    if len(NewFormat)==0:#把表一中的数据设成大数组，用于高亮显示感兴趣的点
        NewFormat=temp
    else:
        NewFormat=np.vstack((NewFormat,temp))
    row+=1
    col+=5
    time_end=time.time()
    time_used=time_end-time_start
    time_left=time_used/counter*(len(filename)-counter)#动态修正剩余时间
    time_used=round(time_used)
    time_left=round(time_left)
    str_place='{0:<4}{1:<25}{2:>3}{3:^1}{4:<}{5:}'
    print(str_place.format(counter,item,time_used,'/',time_left,'s',sep=''))
fx.ColorTable(NewFormat,workbook,worksheet1)#高亮显示突出的数据
worksheet1.freeze_panes(1, 1)#冻结窗口
#workbook.close()