rasmus/nafuma
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add function to pick out scans based on timestamps

Browse source
This commit is contained in:
rasmusvt 2022-08-16 12:08:52 +02:00
parent a77eb23a38
commit f31849b08b
1 changed files with 24 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

27
nafuma/xanes/plot.py
View file

@ -4,6 +4,7 @@ from matplotlib.ticker import (MultipleLocator, FormatStrFormatter,AutoMinorLoca
import pandas as pd import pandas as pd
import numpy as np import numpy as np
import math import math
import datetime
#import ipywidgets as widgets #import ipywidgets as widgets
#from IPython.display import display #from IPython.display import display
@ -13,7 +14,7 @@ import nafuma.plotting as btp
import nafuma.auxillary as aux import nafuma.auxillary as aux
def plot_xanes(data, options=None): def plot_xanes(data, options={}):
# Update options # Update options
@ -30,8 +31,6 @@ def plot_xanes(data, options=None):
options = aux.update_options(options=options, required_options=required_options, default_options=default_options) options = aux.update_options(options=options, required_options=required_options, default_options=default_options)
print(options['xunit'])
if not 'xanes_data' in data.keys(): if not 'xanes_data' in data.keys():
data['xanes_data'] = xas.io.load_data(data=data, options=options) data['xanes_data'] = xas.io.load_data(data=data, options=options)
@ -59,6 +58,28 @@ def plot_xanes(data, options=None):
return fig, ax return fig, ax
def pick_out_scans(metadata: dict, timestamp: list):
# If either start or end are None, set to way back when or way into the future
if not timestamp[0]:
timestamp[0] = datetime.datetime.strptime('1970 01 01 00:00:00', '%Y %m %d %H:%M:%S')
else:
timestamp[0] = datetime.datetime.strptime(timestamp[0], "%d.%b %y %H.%M.%S")
if not timestamp[1]:
timestamp[1] = datetime.datetime.strptime('3000 01 01 00:00:00', '%Y %m %d %H:%M:%S')
else:
timestamp[1] = datetime.datetime.strptime(timestamp[1], "%d.%b %y %H.%M.%S")
scans = []
for i, time in enumerate(metadata['time']):
if time >= timestamp[0] and time <= timestamp[1]:
scans.append(i)
return scans