import os
import re
import gdb
import time
import signal
import common_functions as cf
def exit_handler(event):
   global nosalir
   nosalir=False
  
   print(str("event type: exit"))
   try:
        print("exit code: {}".format(str(event.exit_code)))
   except Exception as err:
        err_str = "ERROR: {}".format(str(err))
        print(err_str)


"""
Handler that will put a breakpoint on the kernel after
signal
"""
def handler(signum, frame):
      global pid,pillo,t0
      print ("handler:+++++++++++========================++++++++++++")
      #print ("+++++++++++========================++++++++++++") 
      #gdb.execute("nexti")
      settimeslice()
      pillo=time.time()-t0
      os.kill(os.getpid(), signal.SIGINT)
      #print ("+++++++++++========================++++++++++++") 
def setalarm():
    signal.signal(signal.SIGALRM,handler)
    #print ("No disponible")
def settimeslice():  
     #signal.alarm(1)   
     signal.setitimer(signal.ITIMER_REAL,1.0,1.0)
     #print ("No disponible")
def pausealarm():
     signal.setitimer(signal.ITIMER_REAL,0) 
     #print ("No disponible") 
     

def selectrd():
  linea= cf.execute_command(gdb=gdb, to_execute="x/1i $pc")
  print (linea)#+"-"+str(len(linea))+"-"+linea[0])
  print ("============")
  lista=re.findall(r"R(\d+)", linea[0])
  #Si no hay nexti 
  listareg=[" R{} ".format(x) for x in lista]
  strlistareg="info registers "
  for x in listareg:
    strlistareg+=x;
  print (strlistareg)  
  valores= cf.execute_command(gdb=gdb, to_execute=strlistareg)  
  print("===========VALORES==========")
  regs={}
  for x in valores:
    print(x)
    m = re.match(r".*R(\d+).*0x([0-9a-fA-F]+).*", x)
    if m:
      regs[m.group(1)]=m.group(2)
      print (str(m.group(0))+"  "+str(m.group(1)))
  #Loop----
  gdb.execute("nexti")
  linea= cf.execute_command(gdb=gdb, to_execute="x/1i $pc")
  valores= cf.execute_command(gdb=gdb, to_execute=strlistareg)
  regdst={}
  for x in valores:
    print(x)
    m = re.match(r".*R(\d+).*0x([0-9a-fA-F]+).*", x)
    if m:
      if (regs[m.group(1)]!=m.group(2)):
        regdst.add(m.group(1))
  
  
  
  print (type(lista))
  print(lista)
def set_event(event):
      global trun,ocurrencias,t,pillo,t0,bp
      #pausealarm()
   
      if (isinstance(event, gdb.BreakpointEvent)):
        print ("Bp")
        t0=time.clock()
        bp.enabled=False
        ocurrencias=ocurrencias+1
        #signal.setitimer(signal.ITIMER_REAL,3)
      else:  
        print ("set_event")
        #signal.setitimer(signal.ITIMER_REAL,0)
        trun=(time.clock()-t0)
        #signal.setitimer(signal.ITIMER_REAL,2)
      #signal.setitimer(signal.ITIMER_REAL,0.1)
      t=time.time()-t0
      print ("\n Pongo: "+str(t)+"  Pillo;"+str(pillo)+"\n")
      #settimeslice()   
      #gdb.execute("c")

def main():      
    global ocurrencias,t,nosalir,trun,pid,t0,pillo,bp
    
    was_hit = False
    #pid=os.getpid()
    ocurrencias=0
    # Initialize GDB to run the appset pagination off
    gdb.execute('catch signal SIGALRM') 
    gdb.execute("set confirm off")
    gdb.execute("set pagination off")
    gdb.execute("set target-async off")
    gdb.execute("set non-stop off")
    setalarm()
    settimeslice()
    # Connecting to a exit handler event
    gdb.events.exited.connect(exit_handler)

    # Connecting to a stop signal event
    gdb.events.stop.connect(set_event)
    #gdb.events.cont.connect(set_event_cont)
    #gdb.execute("file ~/rodinia_3.1/cuda/lud/cuda/lud_cuda")
    #gdb.execute("set arg -s 10000")
    #gdb.execute("break lud_cuda")
  

    t0=time.time()
    print (t0)

    pillo=t0
    gdb.execute("file codes/mmElem/matrixMul")
    gdb.execute("set arg -wA=16384 -hA=16384 -hB=16384 -wB=16384")
    #gdb.execute("set cuda break_on_launch application")
   # bp=gdb.Breakpoint('matrixMulCUDA')
    #gdb.execute('handle SIGALRM stop')
    #gdb.execute('handle SIGALRM ignore')
    #gdb.execute('handle SIGALRM')
    gdb.execute('catch signal SIGALRM')  
    gdb.execute('r')
    #selectrd()
    print ("1")
    gdb.execute("c")
    print ("1")
    gdb.execute("c")
    gdb.execute("c")
    gdb.execute("c")
    pausealarm()
    gdb.execute("c")
    gdb.execute("c")
  #nosalir=True
   # while nosalir:
    #  gdb.execute("finish")
     # gdb.execute("c")"
    print (" Ocurrencias "+str(ocurrencias)+" Tiempo acumulado de ejecucciones "+ str(trun))
    f=open("tmpxxx_return_profiler.conf","w")
    f.write("Ocurrencias ="+str(ocurrencias)+"\n Tiempo "+str(trun)+"\n")
    f.close()

main()