histoGe/query.py at master · ffavela/histoGe · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
#import sys
import os.path
#from os.path import basename
#import re
import pandas as pd #para imprimir en forma de tabla
#from matplotlib import pyplot as plt
#import numpy as np
#from scipy.optimize import curve_fit
#from scipy import asarray as ar,exp
#from math import sqrt, pi
#import time
#import signal
#import keyboard

# mainPath=sys.path[0] # sources dir
#from myLibs.parsers import *
#from myLibs.gilmoreStats import *
#from myLibs.fitting import *
#from myLibs.autoPeakFunk import *
#from myLibs.QueryDB import *
#from myLibs.plotting import *
from myLibs.QueryDB import OpenDatabase, EnergyRange, CloseDatabase


def QueryFun(ListOpt):
    #print(String)
    List = ListOpt.copy()
    List.pop(0)
    pathfile = os.path.realpath(__file__)
    pathfile = pathfile.strip('query.py')
    conexion = OpenDatabase(pathfile)
    if '--all' in List:
        allflag = True
        List.remove('--all')
    else:
        allflag = False
    if len(List) == 2:
        iEner = List[0]
        fEner = List[1]
    else:
        print('')
        return 30

    try:
        iEner=float(iEner)
        fEner=float(fEner)
    except ValueError:
        print('ERROR: Argument or arguments cannot be converted to float.')
        return 31
    if iEner > fEner:
        iEnerAux = iEner
        iEner = fEner
        fEner = iEnerAux
        del iEnerAux

    DBInfo = EnergyRange(conexion,iEner,fEner)
    print('\nThe energy range consulted is %.2f keV to %.2f keV.\n' % (iEner,fEner))
    if len(DBInfo) != 0:
        Eg , Ig , Decay, Half , Parent = [],[],[],[],[]
        for Ele in DBInfo:
            Eg.append(str(Ele[1])+' ('+str(Ele[2])+')')
            Ig.append(str(Ele[3])+' ('+str(Ele[4])+')')
            Decay.append(Ele[5])
            Half.append(str(Ele[6]) +' ' +str(Ele[7]) + ' ('+str(Ele[8])+')')
            Parent.append(Ele[10])
        pd.set_option('display.max_rows', 30)#imprime todas las filas

        if allflag:
            pd.set_option('display.max_rows', None)#imprime todas las filas
        else:
            pd.set_option('display.max_rows', len(Ele))

        df = pd.DataFrame(list(zip(Eg,Ig,Decay,Half,Parent)),columns=['Eg [keV]','Ig (%)','Decay mode','Half Life','Parent'])#crea  la tabla
        print(df) #imprime la tabla

    print("\n%d results were found" %(len(DBInfo)))
    CloseDatabase(conexion)
    return 0