GOLEM Example: Shot Data Analysis¶

# Python 3
import configparser as cp
import numpy as np
import matplotlib.pyplot as plt
from urllib.request import urlopen, HTTPError
%matplotlib inline

def get_shot_config(shot):
    """
    Get the GOLEM shot configuration.
    
    Parameters
    ----------
    shot :  int
            GOLEM shot number
    
    Returns
    -------
    dict
            GOLEM shot configuration dictionnary
    
    """
    url = f'http://golem.fjfi.cvut.cz/shots/{shot}/data_configuration.cfg'
    try:
        with urlopen(url) as response:
            config_str = response.read().decode('utf-8')
    
        config = cp.RawConfigParser()
        data_types = dict()
        config.read_string(config_str)
        for data_type in config.sections():
            data_types[data_type] = dict(config.items(data_type))
        return data_types
    except HTTPError:
        print('Problem with the network? Can''t open the config file')
        return None

def get_shot_data_dict(shot, signame):
    """
    Returns the data dictionnary of a signal for given shot.
    
    Parameters
    ----------
    shot :  int
            GOLEM shot number
    signame : str
            GOLEM signal name

    Returns
    -------
    data : Numpy npz format Data
            GOLEM signal data

    """
    baseURL = "http://golem.fjfi.cvut.cz/utils/data/"
    url = baseURL + str(shot) +'/' + signame + '.npz'
    print(f'Openning {url} ...')
    # The source file gets downloaded and saved into a temporary directory 
    ds = np.DataSource()
    return np.load(ds.open(url, mode='br'))# as npz: # Python 3 needs to open the file in binary mode

def dict_to_y(data_dict):
    """
    Returns time data arrays (y, t) from a GOLEM signal data.
    
    Parameters
    ----------
    data_dict : dict
            GOLEM signal data.
    
    Returns
    -------
    t : Numpy array
        time data
    y : Numpy array
        data array

    """
    y = data_dict['data']
    t = np.linspace(data_dict['t_start'], data_dict['t_end'], len(y))
    return t, y

shot = 22667

cfg = get_shot_config(shot)