Analysis of DRP signals, analysis of particle transport

The following script shows the basic data aquision from Double Rake Probe

To get more detailed information about Double Rake Probe visit its wikipage.

DRP consists of two rakes of Langmuir probes that can be operated in two basic modes.
The first one is floating potential mode ($V_\mathrm{float}$), where each pin is insulated
and we can measure an approximation of plasma potential.

The other mode is ion saturation current ($I_\mathrm{sat}$), where -100 V voltage is
applied to each pin. This voltage repels electrons so one can measure the plasma density.

To measure transport using DRP one has to measure 4 floating potential ($V_\mathrm{fl}$)
signals and 2 ion saturation current ($I_\mathrm{sat}$) signals, however, quality $I_\mathrm{sat}$
measurements may turn out to be challenging to perform, so in the following
script, only $V_\mathrm{float}$ measurements are considered.

Needed Python modules

import matplotlib.pyplot as plt
import numpy as np
import xarray as xr
try:
    import xrscipy.signal as dsp
except:
    print('xrscipy.signal module not found')
# import xarray_dsp.wavelets as wave

# for better readibility of spectrograms
from matplotlib.colors import LogNorm


# to keep the attributes
xr.set_options(keep_attrs=True)

# the data is downloaded in the directory/folder
# from where this script was started 
ds = np.DataSource('/tmp')

#inline plots
%matplotlib inline

Download (and save the data)

One might need to modify ulr_base variable in download_data function and the channel names in channel_list.

The correct url_base depends on the DAS used and the channel_list
represents the names of particular signals.
It is recommended to save the data to your computer if runnig the scipts locally,
as it saves time and makes your analysis independent on internet connection.

Eventually the data is converted into xarray.DataArray structure for convenience.

# test shot_no, the followin configuration should give the right data
# shot_no = 36443
# test channel_list
# channel_list = ['DRP-R1', 'DRP-R2','DRP-R3','DRP-R4','DRP-R5','DRP-R6',
#               'DRP-L1', 'DRP-L2' ,'DRP-L3' ,'DRP-L4' ,'DRP-L5']
# test url_base
# "http://golem.fjfi.cvut.cz/shots/{}/Diagnostics/DoubleRakeProbe/".format(shot_no)

# a little helper function
def strip_off_data(base_url: str,data_name: str):
    """ This function is used to get data like parameters
    or comments from golem database"""
    url = base_url + data_name
    file = ds.open(url)
    parameters = np.loadtxt(file)
    return parameters

def check_plasma_existence(shot_no: int, no_plasma_override: bool=False):
    """Checks, whether there was plasma, if there was not plasma
    raises RuntimeError. It is possible to override no_plasma and
    proceed in the code execution."""
#     base_url = "http://golem.fjfi.cvut.cz/shots/{}/Diagnostics/BasicDiagnostics/Results/"
    base_url = f"http://golem.fjfi.cvut.cz/shots/{shot_no}/Diagnostics/PlasmaDetection/Results/"    
    # check whether there was plasma
    
    b_plasma = strip_off_data(base_url,"b_plasma")
    if not b_plasma == 1.0:
        if not no_plasma_override:
            raise RuntimeError("There was no plasma.")
    
            


def download_data(shot_no: int, channel_list: list, no_plasma_override: bool=False):
    """Downloads the data and returns a tuple
    of time numpy.ndarray and data numpy.ndarray"""
    url_base = f"http://golem.fjfi.cvut.cz/shots/{shot_no}/Diagnostics/DoubleRakeProbe/"
    
    check_plasma_existence(shot_no, no_plasma_override)
    
    data_all = np.ones(shape=(40960,len(channel_list)))
    i = 0
    
    

    for channel in channel_list:
        # open the file
        file = ds.open(url_base+"{}.csv".format(channel))
        # transform to numpy format
        np_file = np.loadtxt(file, delimiter=',')
        # add to the complete array
        data_all[:,i] = np_file[:,1]
        # saves the time from the diag. output
        if i == 0:
            time = np_file[:,0] * 10e2 # to have time in ms
        i += 1
        
        # the 10e2 coefficient is given by a voltage divider
    return time, data_all * 10e2

# list of channel names in the database
channel_list = ['DRP-R1', 'DRP-R2','DRP-R3','DRP-R4','DRP-R5','DRP-R6',
               'DRP-L1', 'DRP-L2' ,'DRP-L3' ,'DRP-L4' ,'DRP-L5','DRP-L6']
# shot is set to 0
# i.e. the last shot
# attention! when running the script locally
# leaving the 0 will not update the downloaded data to the current shot
# because it first looks wheather the data has already been downloaded

shot_no = 43235


time, data = download_data(shot_no, channel_list ,False)

# creation of xarray.DataArray
# it is with offset
data_all_off = xr.DataArray(data,
                            coords={'time':time,
                                    'signal':channel_list},
                            dims=['time','signal']
#                             ,attrs = {'shot_no'}
                            )

Download the time of the plasma start and the end

def get_plasma_start_end(shot_no: int, no_plasma_override: bool=False):
    """Downloads the plasma start and end if there was plasma."""
    # the following should not change in principle, but
    # if your are having problems with downloading the data,
    # check the url as adviced above
    base_url = f"http://golem.fjfi.cvut.cz/shots/{shot_no}/Diagnostics/PlasmaDetection/Results/"
    
    # checks whether there was plasma
    # and if yes, get the plasma start and end
    # this check is done here again because it is
    # meaningless to try to download plasma_start and plasma_end
    # if there was no plasma,
    check_plasma_existence(shot_no, no_plasma_override)
    
    t_start = strip_off_data(base_url,"t_plasma_start")
    t_end = strip_off_data(base_url,"t_plasma_end")
    return t_start, t_end

# add the plasma start and end as a attribute of 
# all_data xarray.DataArray
t_start, t_end = get_plasma_start_end(shot_no,False)


data_all_off.attrs = {"plasma_start":t_start, "plasma_end":t_end}

Subtraction of offset

# time window of period before the plasma start
# where we calculate the offset we subtract
time_off = slice(data_all_off.attrs['plasma_start']-2.5,data_all_off.attrs['plasma_start']-1)

# all the downloaded data without offset
data_all = data_all_off - data_all_off.sel(time=time_off).mean(dim='time')

Visual check of the signals

time_window = slice(data_all.attrs['plasma_start']-0.5,data_all.attrs['plasma_end']+1)
# the following list just to sort the pin coordinates such that
# it is shown as it is physically in the vessel
signal_order = ['DRP-R1', 'DRP-L1','DRP-R2','DRP-L2','DRP-R3','DRP-L3',
               'DRP-R4', 'DRP-L4' ,'DRP-R5' ,'DRP-L5' ,'DRP-R6', 'DRP-L6']

# here is the part, where the plasma is is selected
sel_plasma = data_all.sel(time=time_window,signal=signal_order)

# plot and plot setup
all_plot = sel_plasma.plot.line(x="time", col='signal', col_wrap=2,
                                figsize=(12,12),
#                                 subplot_kws={'gridspec_kw:{'wspace':0.1}}
                               )

fontsize = 13
# axes labels
all_plot.set_xlabels('time [ms]',fontsize=fontsize)
all_plot.set_ylabels('$V_\mathrm{float}$ [V]',fontsize=fontsize)


# make the figure nicer
plt.tight_layout()

Spectrograms of the data

we keep using the signal order and time window from the cell above → so we continue to work with the sel_plasma variable

spectrogram = dsp.spectrogram(sel_plasma, nperseg=512,dim='time')

spct_plot = spectrogram.plot(x='time',col='signal',col_wrap=2,
                             norm=LogNorm(),
                             figsize=(12,12),
                             cbar_kwargs={'label':r'_[-]'}
                            )

fontsize = 13
# axes labels
spct_plot.set_xlabels('time [ms]',fontsize=fontsize)
spct_plot.set_ylabels('$f$ [kHz]',fontsize=fontsize)

plt.ylim(0,75)
# save the figure with the name expected by DoubleRakeProbe.sh
plt.savefig('icon-fig.png')