#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Sun Aug  1 17:24:59 2021
Rotina genérica de leitura dos dados de TMS em arquivo excel

Inputs (strings):
    
    i-> intensidade de estimulação (110 ou 120)
    s-> identificação do sujeito (04, 05, 07 ou 09)
    m-> músculo Alvo ('FCP', 'ADM' ou 'FRC')

    dirname -> pasta com as tabelas xlsx (default ./)

Os dados lidos estarão disponíveis no DataFrame df
# df['x'] -> coordenada x
# df['y'] -> coordenada y
# df['z'] -> coordenada z
# df['Ampl'] -> Amplitude do EMG
# df['Lat'] -> Latência do EMG

@author: werley
"""

import pandas as pd
import numpy as np
 
##INPUTS

# Intensidade da estimulação da bobina  (110 ou 120)
i= '120'

# Identificação do sujeito (04, 05, 07 ou 09)
s = '04'

# Músculo Alvo ('FCP', 'ADM' ou 'FRC')
m = 'FRC'

# Localização das planilhas de dados
dirname = './'

##

# String fixa do arquivo de leitura
filename = s +'_'+ i +'_MRI_'+ m + '_processed.xlsx';

if m=='FCP': blankcol_shift=0;
elif m=='ADM': blankcol_shift=2;
else: blankcol_shift=4;
   
cols = [3,4,5,8+blankcol_shift,9+blankcol_shift]

df = pd.read_excel(dirname + filename,skiprows=range(1,6),usecols = cols, engine='openpyxl');

df.columns = ['x', 'y', 'z', 'Ampl', 'Lat']

if 'df' in locals():
    print ('DataFrame ''df'' criado.' )