12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751 |
- #!/usr/bin/env python3
- # by svsd_val
- # jabber : svsd_val@jabber.ru
- # mail to: svsdval@gmail.com
- import sys
- import os
- import re
- filepath=''
- if ( len(sys.argv) < 2 ):
- if sys.platform.startswith('win'):
- from tkinter import Tk
- from tkinter import filedialog as fd
- root=Tk()
- root.withdraw()
- filepath= fd.askopenfilename(filetypes=(("Video Files", ".mpg .mkv .avi .webm .mp4"), ("All Files", "*.*")))
- root.destroy()
- print("get file [" + filepath +"]")
- else:
- print("halt, no args")
- sys.exit( 0 )
- else:
- filepath = sys.argv[1]
- if not os.path.exists( filepath ):
- has_pytube = False
- try:
- from pytube import YouTube
- has_pytube = True
- except:
- pass
- if has_pytube:
- print("Downloading video by url: %s ..." % filepath)
- yt = YouTube( filepath )
- videos = [ { 'itag' : i.itag, 'res' : int(re.sub('[^0-9]','', i.resolution)), 'progressive' : int(i.is_progressive) } for i in yt.streams.filter(file_extension='mp4') if i.mime_type.find("video") != -1 ]
- print(videos)
- videos = sorted( videos , key = lambda d : ( -d['progressive'], - d['res']) )
- print('sorted by progressive (has video & audio in same file) and video resolution')
- for i in videos:
- print('processing: %s' % i)
- filepath = "%s_%s_%s.mp4" % ( re.sub(r'[\W_]','_', yt.title), i['itag'], i['res'])
- yt.streams.get_by_itag(i['itag']).download( "./" , filepath, skip_existing=True)
- break
- else:
- print("file not exists [" + filepath +"], and no pytube has installed..., exit.")
- sys.exit( 0 )
- import math
- import ntpath
- import time
- from os.path import expanduser
- import cv2
- import pygame
- from midiutil.MidiFile import MIDIFile
- from OpenGL.GL import *
- from OpenGL.GLU import *
- from pygame.locals import *
- print(f'open file [{filepath}]')
- vidcap = cv2.VideoCapture( filepath )
- outputmid= ntpath.basename( filepath ) + '_output.mid'
- settingsfile= filepath + '.ini'
- import datetime
- import video2midi.settings as settings
- from video2midi.gl import *
- from video2midi.midi import *
- from video2midi.prefs import prefs
- width=640
- height=480
- mpos = [0,0]
- keygrab=0
- keygrabid=-1
- lastkeygrabid=-1
- frame= 0
- printed_for_frame=0
- convertCvtColor=1
- # For OpenCV 2.X ..
- CAP_PROP_FRAME_COUNT =0
- CAP_PROP_POS_FRAMES =0
- CAP_PROP_POS_MSEC =0
- CAP_PROP_FRAME_WIDTH =0
- CAP_PROP_FRAME_HEIGHT=0
- CAP_PROP_FPS =0
- COLOR_BGR2RGB =0
- print("OpenCV version:" + cv2.__version__ )
- if cv2.__version__.startswith('2.'):
- CAP_PROP_FRAME_COUNT = cv2.cv.CV_CAP_PROP_FRAME_COUNT
- CAP_PROP_POS_FRAMES = cv2.cv.CV_CAP_PROP_POS_FRAMES
- CAP_PROP_POS_MSEC = cv2.cv.CV_CAP_PROP_POS_MSEC
- CAP_PROP_FRAME_WIDTH = cv2.cv.CV_CAP_PROP_FRAME_WIDTH
- CAP_PROP_FRAME_HEIGHT = cv2.cv.CV_CAP_PROP_FRAME_HEIGHT
- CAP_PROP_FPS = cv2.cv.CV_CAP_PROP_FPS
- else:
- # 3, 4 , etc ...
- CAP_PROP_FRAME_COUNT = cv2.CAP_PROP_FRAME_COUNT
- CAP_PROP_POS_FRAMES = cv2.CAP_PROP_POS_FRAMES
- CAP_PROP_POS_MSEC = cv2.CAP_PROP_POS_MSEC
- CAP_PROP_FRAME_WIDTH = cv2.CAP_PROP_FRAME_WIDTH
- CAP_PROP_FRAME_HEIGHT = cv2.CAP_PROP_FRAME_HEIGHT
- CAP_PROP_FPS = cv2.CAP_PROP_FPS
- COLOR_BGR2RGB = cv2.COLOR_BGR2RGB
- vidcap.set(CAP_PROP_POS_FRAMES, frame)
- vidcap.set(cv2.CAP_PROP_BUFFERSIZE, 2)
- success,image = vidcap.read()
- debug_keys = 0
- length = int(vidcap.get(CAP_PROP_FRAME_COUNT))
- video_width = int(vidcap.get(CAP_PROP_FRAME_WIDTH))
- video_height = int(vidcap.get(CAP_PROP_FRAME_HEIGHT))
- fps = float(vidcap.get(CAP_PROP_FPS))
- width = video_width
- height = video_height
- def fit_to_the_screen() -> None:
- global width, height
- infoObject = pygame.display.Info()
- if (width > infoObject.current_w) or ( height > infoObject.current_h):
- print("try fit window to the screen")
- print("current window size: %sx%s" %(width,height))
- print("current screen size: %sx%s" %(infoObject.current_w, infoObject.current_h))
- ratio = ( width / infoObject.current_w)
- width = int(width / ratio * 0.9 )
- height = int(height / ratio *0.9)
- print("new window size: %sx%s" %(width,height))
- pygame.init()
- fit_to_the_screen()
- endframe = length
- showoutputpath = 0
- def resize_window() -> None:
- global screen, width, height
- if prefs.resize:
- width = prefs.resize_width
- height = prefs.resize_height
- else:
- width = video_width
- height = video_height
- fit_to_the_screen()
- screen = pygame.display.set_mode((width,height), DOUBLEBUF|OPENGL|pygame.RESIZABLE)
- doinit()
- # set start frame
- def getFrame(framenum:int = -1) -> None:
- global image
- global success
- global width
- global height
- global convertCvtColor
- global fps
- if ( fps == 0 ):
- return
- goto_frame_by_msec=False
- if ( framenum != -1 ):
- #vidcap.set(CAP_PROP_POS_FRAMES, int(framenum) )
- # problems with mpeg formats ...
- if goto_frame_by_msec:
- oldframenum = int(round(vidcap.get(1)))
- frametime = framenum * 1000.0 / fps
- print("go to frame time :" + str(frametime))
- success = vidcap.set(CAP_PROP_POS_MSEC, frametime)
- if not success:
- print("Cannot set frame position from video file at " + str(framenum))
- success = vidcap.set(CAP_PROP_POS_FRAMES, int(oldframenum) )
- else:
- success = vidcap.set(CAP_PROP_POS_FRAMES, framenum )
- curframe = vidcap.get(CAP_PROP_POS_FRAMES)
- if (curframe != framenum ):
- print("OpenCV bug, Requesting frame " + str(framenum) + " but get position on " +str(curframe))
- success,image = vidcap.read()
- # if ( resize == 1 ):
- # image = cv2.resize(image, (resize_width , resize_height))
- # print "resize to "+str(resize_width) + "x"+ str(resize_height)
- getFrame()
- print("video " + str(width) + "x" + str(height) +" fps: " + str(fps))
- # add some notes
- channel = 0
- volume = 100
- basenote = prefs.octave * 12
- notes=[]
- notes_db=[]
- notes_de=[]
- notes_channel=[]
- notes_tmp=[]
- notes_pressed_color=[]
- colorWindow_colorBtns_channel_labels=[]
- colorWindow_colorBtns_channel_btns=[]
- separate_note_id=-1
- screen=0
- colorBtns = []
- #quantized notes to the grid.
- use_snap_notes_to_grid = False
- notes_grid_size=32
- midi_file_format = 0
- line_height = 20
- running = 1
- #cfg
- home = expanduser("~")
- inifile = os.path.join( home, '.v2m.ini')
- if os.path.exists( 'v2m.ini' ):
- inifile="v2m.ini"
- print("local config file exists.")
- def update_size() -> None:
- global width, height
- if ( prefs.resize == 1 ):
- width = prefs.resize_width
- height = prefs.resize_height
- else:
- fit_to_the_screen()
- def loadsettings(cfgfile: str) -> None:
- global colorBtns, colorWindow_colorBtns_channel_labels
- settings.loadsettings(cfgfile)
- settings.compatibleColors(colorBtns)
- if len(colorWindow_colorBtns_channel_labels) > 0:
- for i in range(len(colorBtns)):
- colorWindow_colorBtns_channel_labels[i].text = "Ch:" + str(prefs.keyp_colors_channel[i]+1)
- update_size
- if 'glwindows' in globals():
- glBindTexture(GL_TEXTURE_2D, Gl.bgImgGL)
- loadImage(prefs.startframe)
- settingsWindow_slider1.setvalue(prefs.keyp_delta)
- settingsWindow_slider2.setvalue(prefs.minimal_duration * 100)
- settingsWindow_slider3.setvalue(prefs.tempo)
- settingsWindow_slider7.setvalue(prefs.keys_pos_cnt)
- sparks_switch.switch_status = prefs.use_sparks
- sparks_slider_delta.value = 0
- sparks_slider_delta.id =-1
- settingsWindow_rollcheck_button.switch_status = prefs.rollcheck
- settingsWindow_rollcheck_priority_button.switch_status = prefs.rollcheck_priority
- use_percolor_delta.switch_status = prefs.use_percolor_delta
- notes_overlap_btn.switch_status = prefs.notes_overlap
- ignore_notes_with_minimal_duration_btn.switch_status = prefs.ignore_minimal_duration
- update_size
- for i in range(144):
- notes.append(0)
- notes_db.append(0)
- notes_de.append(0)
- notes_channel.append(0)
- notes_tmp.append(0)
- notes_pressed_color.append([0,0,0])
- prefs.keyp_colors_alternate.append([0,0,0])
- prefs.keyp_colors_alternate_sensitivity.append(0)
- def v_rotate(v, ang):
- radAng = ang * math.pi/180
- return [ (v[1] * math.cos(radAng)) - (v[0] * math.sin(radAng)), (v[1] * math.sin(radAng)) + (v[0] * math.cos(radAng)) ]
- def updatekeys( append=0 ):
- xx=0
- if append == 1:
- print(f'clear keys, set to {prefs.keys_pos_cnt}')
- prefs.keys_pos = []
-
- for idx in range (prefs.keys_pos_cnt):
- i = idx // 12
- j = idx % 12
- # for i in range(12):
- # for j in range(12):
- if (append == 1) or (i*12+j > len(prefs.keys_pos)-1):
- prefs.keys_pos.append( [0,0] )
- prefs.keys_pos[i*12+j][0] = int(round( xx ))
- prefs.keys_pos[i*12+j][1] = 0
- if (j == 1) or ( j ==3 ) or ( j == 6 ) or ( j == 8) or ( j == 10 ):
- prefs.keys_pos[i*12+j][1] = prefs.yoffset_blackkeys
- xx += -prefs.whitekey_width
- # keys_pos[i*12+j][0] = int(round( xx + whitekey_width *0.5 ))
- # tune by wuzhuoqing
- if (j == 1) or ( j == 6 ):
- prefs.keys_pos[i*12+j][0] = int(round( xx + prefs.whitekey_width * prefs.blackkey_relative_position ))
- if (j == 8 ):
- prefs.keys_pos[i*12+j][0] = int(round( xx + prefs.whitekey_width * 0.5 ))
- if ( j ==3 ) or ( j == 10 ):
- prefs.keys_pos[i*12+j][0] = int(round( xx + prefs.whitekey_width * (1.0 - prefs.blackkey_relative_position) ))
- xx += prefs.whitekey_width
- for i in range(len(prefs.keys_pos)):
- prefs.keys_pos[i] = v_rotate( prefs.keys_pos[i] , prefs.keys_angle )
- prefs.keys_pos[i][0] = - prefs.keys_pos[i][0]
- updatekeys( 1 )
- loadsettings(inifile)
- tStart = t0 = time.time()-1
- frames = 0
- def snap_to_grid( input_value , input_grid_size ):
- quantized = int( (input_value - int(input_value)) * input_grid_size ) / input_grid_size
- result = (quantized + int(input_value))
- #print ("value before:", input_value , " after :", result)
- return result
- def framerate():
- global t0, frames
- t = time.time()
- frames += 1
- if t - t0 >= 1.0:
- seconds = t - t0
- if ( seconds != 0) :
- fps = frames / seconds
- print("%.0f frames in %3.1f seconds = %6.3f FPS" % (frames,seconds,fps))
- t0 = t
- frames = 0
- def loadImage(idframe=130):
- global image
- global convertCvtColor
- if running != 0:
- getFrame(idframe)
- #image2=cv2.resize(image, (int(video_width/4) , int(video_height/4)))
- print("load image from video " + str(width) + "x" + str(height) + " frame: "+ str(idframe))
- glPixelStorei(GL_UNPACK_ALIGNMENT,1)
- glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_NEAREST)
- glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST)
- glTexEnvf(GL_TEXTURE_ENV, GL_TEXTURE_ENV_MODE, GL_DECAL)
- error_on_load=False
- try:
- if ( convertCvtColor == 1 ):
- #print ("Loading RGB texture")
- glTexImage2D(GL_TEXTURE_2D, 0, 3, video_width, video_height, 0, GL_RGB, GL_UNSIGNED_BYTE, cv2.cvtColor(image,COLOR_BGR2RGB) )
- else:
- #print ("Loading BGR texture")
- glTexImage2D(GL_TEXTURE_2D, 0, 3, video_width, video_height, 0, GL_BGR, GL_UNSIGNED_BYTE, image )
- return
- except Exception as E:
- error_on_load=True
- print("Can't load image from video to OpenGL: %s" % E);
- if error_on_load:
- rvideo_width, rvideo_height = 512, 512
- print("Trying resize video image to %sx%s" % (rvideo_width, rvideo_height));
- try:
- rimage = cv2.resize(image , (rvideo_width, rvideo_height))
- if ( convertCvtColor == 1 ):
- glTexImage2D(GL_TEXTURE_2D, 0, 3, rvideo_width, rvideo_height, 0, GL_RGB, GL_UNSIGNED_BYTE, cv2.cvtColor(rimage,COLOR_BGR2RGB) )
- else:
- glTexImage2D(GL_TEXTURE_2D, 0, 3, rvideo_width, rvideo_height, 0, GL_BGR, GL_UNSIGNED_BYTE, rimage )
- except Exception as E:
- print("Can't load image from video to OpenGL: %s" % E);
- def update_channels(sender):
- print( 'update_channels...' +str(sender.index))
- i=abs(sender.index) -1
- if (sender.index > 0):
- prefs.keyp_colors_channel[i]= prefs.keyp_colors_channel[i] + 1
- else:
- prefs.keyp_colors_channel[i]= prefs.keyp_colors_channel[i] - 1
- if (prefs.keyp_colors_channel[i] > 15):
- prefs.keyp_colors_channel[i] = 15
- if (prefs.keyp_colors_channel[i] < 0):
- prefs.keyp_colors_channel[i] = 0
- colorWindow_colorBtns_channel_labels[i].text = "Ch:" + str(prefs.keyp_colors_channel[i]+1)
- def disable_color(sender):
- print( 'disabled color...' +str(sender.index))
- if sender.index < len(prefs.keyp_colors):
- prefs.keyp_colors[ sender.index ] = [0,0,0]
- # prefs.keyp_colors_channel[i]= prefs.keyp_colors_channel[i] + 1
- def readkeycolor(i):
- pixx=int(prefs.xoffset_whitekeys + prefs.keys_pos[i][0])
- pixy=int(prefs.yoffset_whitekeys + prefs.keys_pos[i][1])
- if ( pixx >= width ) or ( pixy >= height ) or ( pixx < 0 ) or ( pixy < 0 ): return
- if ( prefs.resize == 1 ):
- pixxo=pixx
- pixyo=pixy
- pixx= int(round( pixx * ( video_width / float(prefs.resize_width) )))
- pixy= int(round( pixy * ( video_height / float(prefs.resize_height) )))
- if ( pixx > video_width -1 ): pixx = video_width-1
- if ( pixy > video_height-1 ): pixy= video_height-1
- # print "original x:"+str(pixxo) + "x" +str(pixyo) + " mapped :" +str(pixx) +"x"+str(pixy)
- keybgr=image[pixy,pixx]
- key=[ keybgr[2], keybgr[1],keybgr[0] ]
- prefs.keyp_colors_alternate[i] = key
- def readcolors(sender):
- for i in range( len(prefs.keys_pos) ):
- readkeycolor(i)
- def update_alternate_sensitivity(sender,value):
- global lastkeygrabid
- if ( lastkeygrabid != -1 ):
- prefs.keyp_colors_alternate_sensitivity[ lastkeygrabid ] = value
- def update_sparks_delta(sender,value):
- if (sender.id == -1):
- return
- if (sender.id < len(prefs.keyp_colors)) :
- prefs.keyp_colors_sparks_sensitivity[sender.id] = sender.value
- #print("keyp_colors_sparks_sensitivity["+str(sender.id)+"] = "+ str(sender.value) )
- def update_blackkey_relative_position(sender,value):
- prefs.blackkey_relative_position = value * 0.001
- updatekeys()
- def update_sync_notes_start_pos_time_delta(sender,value):
- prefs.sync_notes_start_pos_time_delta = value *0.001
- def change_use_alternate_keys(sender):
- global extra_label1
- prefs.use_alternate_keys = not prefs.use_alternate_keys
- update_alternate_label()
- def update_alternate_label():
- extra_label1.text = "Use alternate:"+str(prefs.use_alternate_keys)
- def change_use_sparks(sender):
- prefs.use_sparks = sender.switch_status
- # sender.text = "use sparks:"+str(use_sparks)
- def change_rollcheck(sender):
- prefs.rollcheck = sender.switch_status
- def change_rollcheck_priority(sender):
- prefs.rollcheck_priority = sender.switch_status
- def updatecolor(sender):
- if (lastkeygrabid != -1):
- readkeycolor(lastkeygrabid)
- def update_sparks_y_pos (sender):
- if (sender.text == 'y+'):
- prefs.keyp_spark_y_pos = prefs.keyp_spark_y_pos -1
- else:
- prefs.keyp_spark_y_pos = prefs.keyp_spark_y_pos +1
- def update_line_height(sender,value):
- global line_height
- line_height = value
- def snap_notes_to_the_grid(sender):
- global use_snap_notes_to_grid
- use_snap_notes_to_grid = sender.switch_status
- def raise_octave(*args):
- global basenote
- prefs.octave += 1
- if (prefs.octave > 7): prefs.octave = 7
- basenote = prefs.octave * 12
- def lower_octave(*args):
- global basenote
- prefs.octave -= 1
- if (prefs.octave < 0): prefs.octave = 0
- basenote = prefs.octave * 12
- def onPallete_click(sender, index):
- selected_color_delta.color = sender.color
- if index < len(prefs.percolor_delta):
- selected_color_delta.setvalue( prefs.percolor_delta[ index ] )
- sparks_slider_delta.id = Gl.keyp_colormap_id
- sparks_slider_delta.color = prefs.keyp_colors[Gl.keyp_colormap_id]
- sparks_slider_delta.setvalue( prefs.keyp_colors_sparks_sensitivity[Gl.keyp_colormap_id] )
- def change_use_percolor_delta(sender):
- prefs.use_percolor_delta = sender.switch_status
- def update_percolor_delta(sender,value):
- if (Gl.keyp_colormap_id == -1):
- return
- if (Gl.keyp_colormap_id < len(prefs.percolor_delta)):
- prefs.percolor_delta[ Gl.keyp_colormap_id ] = sender.value
- #print("changed percolor delta for color with id ["+str(sender.id)+"] = "+ str(sender.value) )
- def showOrhideallwindows(sender):
- if sender is None:
- ShowHideButton.switch_status = not ShowHideButton.switch_status
- print('switch hidden for all windows')
- for i in glwindows:
- #print("i.type =%s" % (str(type(i))) )
- if isinstance(i, GLWindow ):
- i.fullhidden = ShowHideButton.switch_status
- def start_recreate_midi(sender):
- global running
- if prefs.autoclose == 1:
- running = 0
- else:
- reconstruct()
- def set_start_frame_to_current_frame(sender):
- if sender.index == 0:
- prefs.startframe = int(round(vidcap.get(1)))
- else:
- prefs.startframe = 0
- print("set start frame = "+ str(prefs.startframe))
- def sef_end_frame_to_current_frame(sender):
- global endframe
- if sender.index == 0:
- endframe = int(round(vidcap.get(1)))
- else:
- endframe = length
- print("set end frame = "+ str(endframe), sender.index)
- def switch_notes_overlap(sender):
- if sender is None:
- prefs.notes_overlap = not prefs.notes_overlap
- notes_overlap_btn.switch_status = prefs.notes_overlap
- else:
- prefs.notes_overlap = notes_overlap_btn.switch_status
- def switch_sync_notes_start_pos(sender):
- prefs.sync_notes_start_pos = sender.switch_status
- def change_save_to_disk_per_channel(sender):
- prefs.save_to_disk_per_channel = sender.switch_status
- def switch_ignore_notes_with_minimal_duration(sender):
- if sender is None:
- prefs.ignore_minimal_duration = not prefs.ignore_minimal_duration
- ignore_notes_with_minimal_duration_btn.switch_status = prefs.ignore_minimal_duration
- else:
- prefs.ignore_minimal_duration = ignore_notes_with_minimal_duration_btn.switch_status
- def switch_resize_windows(sender):
- prefs.resize = not prefs.resize
- resize_window()
- def scroll_by_steps( steps ):
- global frame
- frame+=steps
- if (frame > length *0.99):
- frame=math.trunc(length *0.99)
- if (frame < 1):
- frame=1
- glBindTexture(GL_TEXTURE_2D, Gl.bgImgGL)
- loadImage(frame)
- def scroll_forward_by_frame(sender):
- scroll_by_steps(1)
- def scroll_fast_forward(sender):
- scroll_by_steps(100)
- def scroll_prev_by_frame(sender):
- scroll_by_steps(-1)
- def scroll_fast_prev(sender):
- scroll_by_steps(-100)
- def scroll_to_start(sender):
- global frame
- frame=0
- glBindTexture(GL_TEXTURE_2D, Gl.bgImgGL)
- loadImage(frame)
- def scroll_to_end(sender):
- global frame
- frame=length-100
- glBindTexture(GL_TEXTURE_2D, Gl.bgImgGL)
- loadImage(frame)
- def btndown_save_settings(sender):
- settings.savesettings(settingsfile)
- def btndown_load_settings(sender):
- old_resize = prefs.resize
- loadsettings( settingsfile )
- update_alternate_label()
- if (prefs.resize != old_resize):
- resize_window()
- def change_autoclose(sender):
- prefs.autoclose = sender.switch_status
- def rotate_cw(sender):
- prefs.keys_angle -= 5
- updatekeys()
- def rotate_ccw(sender):
- prefs.keys_angle += 5
- updatekeys()
- def update_keys_pos_cnt(sender,value):
- prefs.keys_pos_cnt=int(value)
-
- def change_cnt(sender):
- print('change count')
- updatekeys(1)
- def is_black_key(key_id : int) -> bool:
- j = key_id % 12
- return (j == 1) or ( j == 3 ) or ( j == 6 ) or ( j == 8) or ( j == 10 )
- def vertical_align_keys( separate_black_keys = 1, align = 1 ):
- print(f"lastkeygrabid {lastkeygrabid}")
- if lastkeygrabid < 0 or lastkeygrabid > len(prefs.keys_pos):
- return
- y = prefs.keys_pos[lastkeygrabid][align]
- selected_black_key = is_black_key(lastkeygrabid)
-
- for idx in range (len(prefs.keys_pos)):
- if separate_black_keys == 1:
- if selected_black_key:
- if is_black_key(idx):
- prefs.keys_pos[idx][align] = y
- else:
- if not is_black_key(idx):
- prefs.keys_pos[idx][align] = y
- else:
- prefs.keys_pos[idx][align] = y
-
- def valign(sender):
- vertical_align_keys(align=1)
- def halign(sender):
- vertical_align_keys(align=0)
- wh = ( (len(prefs.keyp_colors) // 2)+2 ) * 24 - 24
- colorWindow = GLWindow(24, 50, 274, wh, "color map")
- settingsWindow = GLWindow(24+275, 80, 550, 380, "Settings")
- helpWindow = GLWindow(24+270, 50, 750, 535, "help")
- extraWindow = GLWindow(24+270+550+6, 80, 510, 250, "extra/experimental")
- sparksWindow = GLWindow(24+270+550+6, 300, 510, 185, "sparks & color settings")
- glwindows = []
- ShowHideButton = GLButton(0,0 ,13,13, 1, [128,128,128], "" , showOrhideallwindows ,switch=1, switch_status=0 )
- ShowHideButton.active = 2
- glwindows.append(ShowHideButton)
- glwindows.append(colorWindow)
- glwindows.append(settingsWindow)
- glwindows.append(helpWindow)
- glwindows.append(extraWindow)
- glwindows.append(sparksWindow)
- helpWindow.hidden=1
- helpWindow_label1 = GLLabel(0,0, """h - on window title, show/hide the window
- q - begin to recreate midi
- s - set start frame, (mods : shift, set processing start frame to the beginning)
- e - set end frame, (mods : shift, set processing end frame to the ending)
- p - if key is set, force separate to 2 channels (on single color video)
- o - enable or disable overlap notes
- i - enable or disable ignore/lengthening of notes with minimal duration
- r - enable or disable resize function
- Mouse wheel - keys adjustment
- Left mouse button - dragging the selected key / select color from the color map
- CTRL + Left mouse button - update selected color in the color map
- CTRL + 0 - disable selected color in the color map
- Right mouse button - dragging all keys, if the key is selected, the transfer is carried out relative to it.
- Arrows - keys adjustment (mods : shift) ( Atl+Arrows UP/Down - sparks position adjustment )
- +(PLUS) / - (MINUS) - rotate keys by 5*
- PageUp/PageDown - scrolling video (mods : shift)
- Home/End - go to the beginning or end of the video
- [ / ] - change base octave
- F2 / F3 - save / load settings, F4 - move all windows to the mouse point
- Escape - quit, TAB - Show/Hide all windows
- Space - abort re-creation and save midi file to disk
- 4,6,8,2 on numpad - move the selected key by 1 pixel on each axis
- 1,3 - vertical / horizontal alignment""")
- settingsWindow.appendChild( GLButton(260, 20 ,140,20,0 , [128,128,128], "start recreate midi" , start_recreate_midi , hint = "q - hot key") )
- settingsWindow.appendChild( GLButton(260, 40 ,140,20,0 , [128,128,128], "set start frame" , set_start_frame_to_current_frame, hint = "s - hot key, (mods : shift + s, set processing start frame to the beginning)" ) )
- settingsWindow.appendChild( GLButton(260+141, 40 ,140,20,0, [128,128,128], "set end frame" , sef_end_frame_to_current_frame , hint = "e - hot key, (mods : shift + e, set processing end frame to the ending)" ) )
- notes_overlap_btn = GLButton(260, 80 ,140,20,0, [128,128,128], "notes overlap" , switch_notes_overlap , hint = "o - hot key", switch=1, switch_status=0)
- ignore_notes_with_minimal_duration_btn = GLButton(260,100 ,272,20,0, [128,128,128], "ignore notes with minimal duration", switch_ignore_notes_with_minimal_duration, hint = "i - hot key", switch=1, switch_status=0)
- settingsWindow.appendChild( notes_overlap_btn )
- settingsWindow.appendChild( ignore_notes_with_minimal_duration_btn )
- settingsWindow.appendChild( GLButton(260+141, 80 ,140,20,0, [128,128,128], "sync notes" , switch_sync_notes_start_pos , hint = "sync notes start pos", switch=1, switch_status=0) )
- settingsWindow.appendChild( GLButton(260,120 ,140,20,0, [128,128,128], "resize window" , switch_resize_windows , hint = "r - hot key") )
- exit_switch = GLButton(260+141, 120 ,140,20,1, [128,128,128], "auto-close" ,change_autoclose,switch=1, switch_status= prefs.autoclose, hint = "exit after the completion of the midi reconstruction" )
- settingsWindow.appendChild( exit_switch )
- settingsWindow.appendChild( GLButton(260 , 140 ,140,20,0, [128,128,128], "save settings" , btndown_save_settings , hint = "F2 - hot key, save current settings" ) )
- settingsWindow.appendChild( GLButton(260+141, 140 ,140,20,0, [128,128,128], "load settings" , btndown_load_settings , hint = "F3 - hot key, load saved settings" ) )
- navbtns_info = [
- {'name' : "[<", 'hint' : 'Home - hot key, go to first frame',
- 'func' : scroll_to_start },
- {'name' : "<<", 'hint' : 'PageDown - hot key, fast scroll backward',
- 'func' : scroll_fast_prev },
- {'name' : " <", 'hint' : 'Shift+PageDown - shortcut, scroll backward by frame',
- 'func' : scroll_prev_by_frame },
- {'name' : " >", 'hint' : 'Shift+PageUp - shortcut, scroll forward by frame',
- 'func' : scroll_forward_by_frame },
- {'name' : ">>", 'hint' : 'PageUp - hot key,fast scroll forward',
- 'func' : scroll_fast_forward },
- {'name' : " >]", 'hint' : 'End - hot key, go to last frame',
- 'func' : scroll_to_end },
- {'name' : "R+", 'hint' : 'rotate the keys clockwise, hot key +',
- 'func' : rotate_cw },
- {'name' : "R-", 'hint' : 'rotate the keys counterclockwise, hot key -',
- 'func' : rotate_ccw }
- ]
- #btnfuncs = [ None, None, None, None, None, None ]
- for i in range(len( navbtns_info )):
- settingsWindow.appendChild( GLButton(260 + i * 32,230 ,32,20,0, [128,128,128], navbtns_info[i]['name'] , navbtns_info[i]['func'], hint = navbtns_info[i]['hint']) )
- settingsWindow.appendChild( GLButton(260 , 295 ,140,20,0, [128,128,128], "update count", change_cnt , hint = "Change keys count" ) )
- settingsWindow.appendChild( GLButton(260+141, 295 ,70,20,0, [128,128,128], "v. align", valign , hint = "vertical alignment of keys to the selected key" ) )
- settingsWindow.appendChild( GLButton(260+141+70, 295 ,70,20,0, [128,128,128], "h. align", halign , hint = "horizontal alignment of keys to the selected key" ) )
- helpWindow.appendChild(helpWindow_label1)
- settingsWindow_label1 = GLLabel(1,0, "base octave: " + str(prefs.octave))
- # + "\nnotes overlap: " + str(prefs.notes_overlap) + "\nignore minimal duration: " + str(prefs.ignore_minimal_duration))
- settingsWindow.appendChild(settingsWindow_label1)
- settingsWindow.appendChild( GLButton(130,0 ,20,20,1, [128,128,128], "+", raise_octave, hint = "] - hot key, move up base octave (+12 tones)" ) )
- settingsWindow.appendChild( GLButton(150,0 ,20,20,1, [128,128,128], " -", lower_octave, hint = "[ - hot key, move down base octave (-12 tones)" ) )
- settingsWindow_slider1 = GLSlider(1,40, 240,18, 0,130,prefs.keyp_delta,label="Sensitivity")
- settingsWindow_slider1.round=1
- settingsWindow.appendChild(settingsWindow_slider1)
- settingsWindow_slider2 = GLSlider(1,90, 240,18, 0,200,prefs.minimal_duration*100,label="Minimal note duration (sec)")
- settingsWindow_slider2.round=0
- settingsWindow.appendChild(settingsWindow_slider2)
- settingsWindow_slider3 = GLSlider(1,133, 240,18, 30,240,prefs.tempo,label="Output tempo for midi")
- settingsWindow_slider3.round=0
- settingsWindow.appendChild(settingsWindow_slider3)
- settingsWindow_slider4 = GLSlider(1,175, 240,18, 0,2,midi_file_format,label="Output midi format")
- settingsWindow_slider4.round=0
- settingsWindow.appendChild(settingsWindow_slider4)
- settingsWindow_slider5 = GLSlider(1,215, 240,18, 0,1000,prefs.blackkey_relative_position * 1000, update_blackkey_relative_position, label="black key relative pos")
- settingsWindow_slider5.round=0
- settingsWindow.appendChild(settingsWindow_slider5)
- settingsWindow_slider6 = GLSlider(1,255, 240,18, 0,1000,prefs.sync_notes_start_pos_time_delta, update_sync_notes_start_pos_time_delta, label="sync notes time delta (ms)")
- settingsWindow_slider6.round=0
- settingsWindow.appendChild(settingsWindow_slider6)
- settingsWindow_slider7 = GLSlider(1,295, 240,18, 12,144,prefs.keys_pos_cnt,update_keys_pos_cnt, label="Keys count")
- settingsWindow_slider7.round=0
- settingsWindow.appendChild(settingsWindow_slider7)
- settingsWindow_rollcheck_button = GLButton(260,160 ,140,22,1, [128,128,128], "roll check" ,change_rollcheck,switch=1, switch_status=prefs.rollcheck )
- settingsWindow.appendChild(settingsWindow_rollcheck_button)
- settingsWindow.appendChild( GLButton(260+141, 160 ,140,20,1, [128,128,128], "per channel save" ,change_save_to_disk_per_channel,switch=1, switch_status= prefs.save_to_disk_per_channel, hint = "split the output midi per channels" ) )
- settingsWindow_rollcheck_priority_button = GLButton(260,180 ,222,22,1, [128,128,128], "rollcheck white keys priority" ,change_rollcheck_priority,switch=1, switch_status=prefs.rollcheck_priority )
- settingsWindow.appendChild(settingsWindow_rollcheck_priority_button)
- # for i in range( len( keyp_colors ) ):
- #keyp_colormap_colors_pos.append ([ (i % 2) * 32, ( i // 2 ) * 20 ])
- print ('creating new colors '+str(len( prefs.keyp_colors )))
- sparks_slider_delta = GLSlider(6,25, 150,18, -50,150,50,update_sparks_delta, label="Sparks delta")
- for i in range( len( prefs.keyp_colors ) ):
- cx,cy = (i % 2) * 130, ( i // 2 ) * 20
- offsetx,offsety=4,4
- colorBtns.append( GLColorButton(offsetx+cx,offsety+cy ,20,20,i, prefs.keyp_colors[i], onPallete_click ) )
- colorWindow.appendChild(colorBtns[i])
- colorWindow_label1 = GLLabel(offsetx+25+cx,offsety+cy , "Ch:" + str(prefs.keyp_colors_channel[i]+1) )
- colorWindow_colorBtns_channel_labels.append( colorWindow_label1 )
- colorWindow.appendChild(colorWindow_label1)
- colorWindow_colorBtns_channel_btns.append( GLButton(offsetx+cx+70,offsety+cy ,20,20,(i+1), [128,128,128], "+" ,update_channels) )
- colorWindow_colorBtns_channel_btns.append( GLButton(offsetx+cx+70+20,offsety+cy ,20,20,-(i+1), [128,128,128], "-" ,update_channels) )
- colorWindow_colorBtns_channel_btns.append( GLButton(offsetx+cx+70+40,offsety+cy ,20,20,i, [128,128,128], "x" ,disable_color, hint="ctrl+0 - shortcut, disable selected color") )
- for i in colorWindow_colorBtns_channel_btns:
- colorWindow.appendChild( i )
- extraWindow.appendChild( GLButton(5, 20 ,128,25,1, [128,128,128], "read colors" ,readcolors) )
- extraWindow.appendChild( GLButton(135,20 ,128,25,1, [128,128,128], "update color" ,updatecolor) )
- extraWindow.appendChild( GLButton(265,20 ,138,25,1, [128,128,128], "enable/disable" ,change_use_alternate_keys) )
- extraWindow.appendChild( GLButton(265,45 ,155,22,1, [96 ,96 ,128], "snap notes to grid" ,snap_notes_to_the_grid,switch=1, switch_status=use_snap_notes_to_grid) )
- extra_label1 = GLLabel(6,0, "Use alternate:"+str(prefs.use_alternate_keys) )
- extraWindow.appendChild( extra_label1 )
- #extra_label2 = GLLabel(0,67, "Selected key sensitivity:"+str(0) )
- extra_slider1 = GLSlider(6,65, 240,18, -100,100,0,update_alternate_sensitivity, label="Selected key sensitivity")
- #extra_slider1.showvalue=True
- #showvaluesinlabel=0
- extraWindow.appendChild(extra_slider1)
- extra_label3 = GLLabel( 6,90, """to select the key press ctrl + left mouse button on the key rect.
- to deselect the key press ctrl + left mouse button on empty space.""" )
- extraWindow.appendChild( extra_label3 )
- extraWindow_slider2 = GLSlider(5,155, 240,18, 0,2000, line_height, update_line_height, label="length of vertical key lines")
- extraWindow_slider2.round=0
- extraWindow.appendChild(extraWindow_slider2)
- sparks_slider_height = GLSlider(160,25, 150,18, 1,60,1,None, label="Sparks height")
- sparks_slider_height.round=0
- sparks_switch = GLButton(313,24 ,100,22,1, [128,128,128], "use sparks" ,change_use_sparks,switch=1, switch_status=prefs.use_sparks )
- sparksWindow.appendChild( sparks_slider_delta )
- sparksWindow.appendChild( sparks_slider_height )
- sparksWindow.appendChild( sparks_switch )
- #
- sparksWindow.appendChild( GLButton(413 ,24 ,32,22,1, [96,96,128], "y+" ,update_sparks_y_pos, hint="move sparks higher") )
- sparksWindow.appendChild( GLButton(413+33,24 ,32,22,1, [96,96,128], "y-" ,update_sparks_y_pos, hint="move sparks lower") )
- sparksWindow.appendChild( GLLabel( 6,50, "alt + up / down - move sparks label up or down " ))
- selected_color_delta = GLSlider(6,100, 200,18, 0,130,50,update_percolor_delta, label="percolor sensitivity")
- selected_color_delta.round=1
- use_percolor_delta = GLButton(313,100 ,190,22,1, [128,128,128], "use percolor sensitivity" ,change_use_percolor_delta,switch=1, switch_status=prefs.use_sparks )
- sparksWindow.appendChild( selected_color_delta )
- sparksWindow.appendChild( use_percolor_delta )
- #colorSettingsWindow.appendChild( GLButton(413 ,24 ,64,22,1, [96,96,128], "Pallette" , None ) )
- #
- #extra_slider2.showvalue=True
- #extra.appendChild(extra_label2)
- #loadsettings( settingsfile )
- #frame=801
- def getkeyp_pixel_pos( x:int, y:int ) -> list[int]:
- pixx=int(prefs.xoffset_whitekeys + x)
- pixy=int(prefs.yoffset_whitekeys + y)
- if ( pixx >= width ) or ( pixy >= height ) or ( pixx < 0 ) or ( pixy < 0 ):
- return [-1,-1]
- #if ( prefs.resize == 1 ):
- if 1==1: #disabled
- pixx= int(round( pixx * ( video_width / float(width) )))
- pixy= int(round( pixy * ( video_height / float(height) )))
- if ( pixx > video_width -1 ): pixx = video_width-1
- if ( pixy > video_height-1 ): pixy= video_height-1
- return [pixx,pixy]
- def iswhitekey( key_num: int ) -> int:
- j = key_num % 12
- if (j == 1) or ( j ==3 ) or ( j == 6 ) or ( j == 8) or ( j == 10 ):
- return 1
- return 0
- def drawframe( lastimage = None):
- global pyfont
- global helptext
- global mousex, mousey
- global keyp_colormap_colors_pos
- global keyp_colormap_pos
- global frame, image
- global printed_for_frame
- global notes_tmp
- global notes_pressed_color
- #global old_spark_color
- #global cur_spark_color
- print_for_frame_debug = False
- if printed_for_frame != frame:
- print_for_frame_debug = True
- printed_for_frame = frame
- scale=1.0
- mousex, mousey = pygame.mouse.get_pos()
- glClear(GL_COLOR_BUFFER_BIT|GL_DEPTH_BUFFER_BIT)
- glViewport (0, 0, width, height)
- glMatrixMode (GL_PROJECTION)
- glLoadIdentity ()
- glOrtho(0, width, height, 0, -1, 100)
- glMatrixMode(GL_MODELVIEW)
- glLoadIdentity()
- glDisable(GL_DEPTH_TEST)
- glScale(scale,scale,1)
- glColor4f(1.0, 1.0, 1.0, 1.0)
- glBindTexture(GL_TEXTURE_2D, Gl.bgImgGL)
- glEnable(GL_TEXTURE_2D)
- DrawQuad(0,0,width,height)
- glEnable(GL_BLEND)
- glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA)
- glColor4f(1.0, 0.5, 1.0, 0.5)
- glPushMatrix()
- glTranslatef(prefs.xoffset_whitekeys,prefs.yoffset_whitekeys,0)
- glDisable(GL_TEXTURE_2D)
- for i in range( len( prefs.keys_pos) ):
- pixpos = getkeyp_pixel_pos(prefs.keys_pos[i][0],prefs.keys_pos[i][1])
- if (pixpos[0] == -1) and (pixpos[1] == -1):
- continue
- if lastimage is not None:
- keybgr=lastimage[ pixpos[1], pixpos[0] ]
- else:
- keybgr=image[ pixpos[1], pixpos[0] ]
- key= [ keybgr[2], keybgr[1],keybgr[0] ]
- keybgr=[0,0,0]
- sparkkey=[0,0,0]
- if prefs.use_sparks:
- sh = int(sparks_slider_height.value)
- if sh == 0:
- sh = 1
- for spark_y_add_pos in range (sh):
- sparkpixpos = getkeyp_pixel_pos(prefs.keys_pos[i][0],prefs.keyp_spark_y_pos - spark_y_add_pos )
- if not ((sparkpixpos[0] == -1) and (sparkpixpos[1] == -1)):
- keybgr = image[ sparkpixpos[1], sparkpixpos[0] ]
- sparkkey = [ sparkkey[0] + keybgr[2],
- sparkkey[1] + keybgr[1],
- sparkkey[2] + keybgr[0] ]
- sparkkey = [ sparkkey[0] / sh, sparkkey[1] / sh,sparkkey[2] / sh]
- #cur_spark_color[i] = sparkkey
- else:
- sparkkey = [0,0,0]
- note=i
- if ( note > 144 ):
- print("skip note > 144")
- continue
- keypressed=0
- pressedcolor=[0,0,0]
- if prefs.use_alternate_keys:
- delta = prefs.keyp_delta + prefs.keyp_colors_alternate_sensitivity[i]
- if ( abs( int(key[0]) - prefs.keyp_colors_alternate[i][0] ) > delta ) and ( abs( int(key[1]) - prefs.keyp_colors_alternate[i][1] ) > delta ) and ( abs( int(key[2]) - prefs.keyp_colors_alternate[i][2] ) > delta ):
- keypressed=1
- pressedcolor=prefs.keyp_colors_alternate[i]
- else:
- for key_id in range( len(prefs.keyp_colors) ):
- keyc = prefs.keyp_colors[key_id]
- spark_delta = prefs.keyp_colors_sparks_sensitivity[key_id]
- delta = prefs.keyp_delta
- if prefs.use_percolor_delta:
- if key_id < len( prefs.percolor_delta ):
- delta = prefs.percolor_delta[ key_id ]
- if (keyc[0] != 0 ) or (keyc[1] != 0 ) or (keyc[2] != 0 ) :
- if ( abs( int(key[0]) - keyc[0] ) < delta ) and ( abs( int(key[1]) - keyc[1] ) < delta ) and ( abs( int(key[2]) - keyc[2] ) < delta ):
- keypressed=1
- pressedcolor = keyc
- notes_pressed_color[i] = keyc
- if prefs.use_sparks:
- #unpressed_by_spark_delta = ( abs( int(sparkkey[0]) - keyc[0] ) < spark_delta ) and ( abs( int(sparkkey[1]) - keyc[1] ) < spark_delta ) and ( abs( int(sparkkey[2]) - keyc[2] ) < spark_delta )
- has_spark_delta = ((sparkkey[0] - keyc[0] ) > spark_delta ) or ((sparkkey[1] - keyc[1] ) > spark_delta ) or ((sparkkey[2] - keyc[2] ) > spark_delta )
- #unpressed_by_spark_fade = ( cur_spark_color[i][0] < old_spark_color[i][0]) and ( cur_spark_color[i][1] < old_spark_color[i][1]) and ( cur_spark_color[i][2] < old_spark_color[i][2])
- #unpressed_by_spark_fade_delta = ( abs( cur_spark_color[i][0] - old_spark_color[i][0]) > 20 )
- if print_for_frame_debug:
- print("note %d key_id %d spark_delta %d sparkkey vs keyc %d %d, %d %d, %d %d" % (note, key_id, spark_delta, sparkkey[0], keyc[0], sparkkey[1], keyc[1], sparkkey[2], keyc[2]))
- if ( not has_spark_delta ):
- keypressed=2
- notes_tmp[i] = keypressed
- if prefs.rollcheck:
- for i in range(1, len( prefs.keys_pos) -1 ):
- if prefs.rollcheck_priority == 0:
- if not iswhitekey(i):
- # Priority on Black keys
- if notes_tmp[i+1] >0: notes_tmp[i] = 0
- if notes_tmp[i-1] >0: notes_tmp[i] = 0
- else:
- if iswhitekey(i):
- # Priority on White keys
- if notes_tmp[i+1] >0: notes_tmp[i] = 0
- if notes_tmp[i-1] >0: notes_tmp[i] = 0
- for i in range( len( prefs.keys_pos) ):
- keypressed = notes_tmp[i]
- pressedcolor = notes_pressed_color[i]
- glPushMatrix()
- glTranslatef(prefs.keys_pos[i][0],prefs.keys_pos[i][1],0)
- glColor4f(1,1,1,0.5)
- if iswhitekey(i):
- glColor4f(0.57,0.57,0.57,0.55)
- DrawQuad(-0.5,-line_height,0.5, line_height )
- if ( keypressed != 0 ):
- #glColor4f(1.0, 0.5, 1.0, 0.9)
- glColor4f(pressedcolor[0]/255.0,pressedcolor[1]/255.0,pressedcolor[2]/255.0,0.9)
- DrawQuad(-6,-7,6,7)
- glColor4f(0,0,0,1)
- if ( keypressed == 1):
- DrawRect(-7,-9,7,9,3)
- else:
- DrawRect(-5,-7,5,7,3)
- else:
- glColor4f(0,0,0,1)
- DrawRect(-7,-7,7,7,1)
- glColor4f(0.5, 1, 1.0, 0.7)
- DrawQuad(-5,-5,5,5)
- if ( lastkeygrabid == i ):
- glColor4f(0.0, 0.5, 1.0, 0.7)
- DrawQuad(-4,-4,4,4)
- if ( separate_note_id == i ):
- glColor4f(0,1,0,1)
- DrawRect(-7,-12,7,12,2)
- if prefs.octave * 12 == i:
- glColor4f(1,0,0,1)
- DrawRect(-9,9,9,12,3)
- DrawQuad(-1,-1,1,1)
- glPopMatrix()
- glColor4f(0.0, 1.0, 1.0, 0.7)
- # Sparks
- if prefs.use_sparks:
- glPushMatrix()
- glTranslatef(prefs.keys_pos[i][0], prefs.keyp_spark_y_pos ,0)
- glColor4f(0.5, 1, 1.0, 0.7)
- DrawQuad(-1,-1,1,1)
- DrawQuad(-0.5,-sparks_slider_height.value ,0.5,0)
- glPopMatrix()
- glPopMatrix()
- glDisable(GL_BLEND)
- glDisable(GL_TEXTURE_2D)
- for i in range(len(glwindows)):
- glwindows[i].draw()
- # drawing hints over all windows
- for i in range(len(glwindows)):
- glwindows[i].drawhint()
- prefs.keyp_delta = int(settingsWindow_slider1.value)
- prefs.minimal_duration = settingsWindow_slider2.value *0.01
- prefs.tempo = int(settingsWindow_slider3.value)
- settingsWindow_label1.text = "base octave: " + str(prefs.octave)
- # + "\nnotes overlap: " + str(prefs.notes_overlap) + "\nignore minimal duration: " + str(prefs.ignore_minimal_duration)
- #settingsWindow_label2.text = "Sensitivity:"+str(keyp_delta)+"\n\nMinimal note duration (sec):"+format(minimal_duration,'.2f' ) + "\n\nOutput tempo for midi:" + str(tempo)
- for i in range(len(prefs.keyp_colors)):
- colorBtns[i].color = prefs.keyp_colors[i]
- glPushMatrix()
- glTranslatef(mousex,mousey,0)
- glColor4f(0.2, 0.5, 1, 0.9)
- DrawQuad(-1,-1,1,1)
- glPopMatrix()
- if showoutputpath > time.time():
- drawHint( width *0.5, height -20, prefs.save_to_disk_message, True)
- def processmidi():
- global frame
- global width
- global height
- global length
- global fps
- global notes
- global notes_db
- global notes_de
- global notes_channel
- global success,image
- global separate_note_id
- global outputmid
- global basenote
- print("video " + str(width) + "x" + str(height))
- basenote = prefs.octave * 12
- mf = midinotes( int(midi_file_format))
- track = 0 # the only track
- time = 0 # start at the beginning
- mf.setup_track(time, prefs.miditrackname, prefs.tempo)
- first_note_time=0
- channel_has_note = [ 0 for x in range(16) ]
- for i in range(len(prefs.keyp_colors_channel)):
- mf.addProgramChange(track, prefs.keyp_colors_channel[i], prefs.keyp_colors_channel_prog[i])
- print("starting from frame:" + str(prefs.startframe))
- getFrame( prefs.startframe )
- notecnt=0
- lastimage = image.copy()
- while success:
- if (frame % 10 == 0):
- glBindTexture(GL_TEXTURE_2D, Gl.bgImgGL)
- if (frame % 200 == 0):
- loadImage(frame)
- lastimage = image.copy()
- #glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR)
- #glTexImage2D(GL_TEXTURE_2D, 0, 3, video_width, video_height, 0, GL_BGR, GL_UNSIGNED_BYTE, image )
- #glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_NEAREST)
- glEnable(GL_TEXTURE_2D)
- drawframe( lastimage )
- glColor4f(1.0, 0.5, 1.0, 0.5)
- glDisable(GL_TEXTURE_2D)
- p= frame / float( length )
- DrawQuad(0,height *0.5 -10, p * width ,height *0.5 +10)
- # glPopMatrix();:
- pygame.display.flip()
- # if (frame % 100 == 0):
- print("processing frame: " + str(frame) + " / " + str(length) + " % " + str( math.trunc(p * 100)))
- # if ( resize == 1 ):
- # image=cv2.resize(image, (width , height))
- # processing white keys
- for i in range( len(prefs.keys_pos) ):
- pixpos = getkeyp_pixel_pos(prefs.keys_pos[i][0],prefs.keys_pos[i][1])
- if (pixpos[0] == -1) and (pixpos[1] == -1):
- continue
- keybgr=image[ pixpos[1], pixpos[0] ]
- key= [ keybgr[2], keybgr[1],keybgr[0] ]
- keybgr=[0,0,0]
- sparkkey=[0,0,0]
- if prefs.use_sparks:
- sh = int(sparks_slider_height.value)
- if sh == 0:
- sh = 1
- for spark_y_add_pos in range (sh):
- sparkpixpos = getkeyp_pixel_pos(prefs.keys_pos[i][0],prefs.keyp_spark_y_pos - spark_y_add_pos )
- if not ((sparkpixpos[0] == -1) and (sparkpixpos[1] == -1)):
- keybgr = image[ sparkpixpos[1], sparkpixpos[0] ]
- sparkkey = [ sparkkey[0] + keybgr[2],
- sparkkey[1] + keybgr[1],
- sparkkey[2] + keybgr[0] ]
- sparkkey = [ sparkkey[0] / sh, sparkkey[1] / sh,sparkkey[2] / sh]
- else:
- sparkkey = [0,0,0]
- note=i
- if ( note > 144 ):
- print("skip note > 144")
- continue
- keypressed=0
- note_channel=0
- # deltaclr = abs( int(key[0]) - keyp_colors[0][0] ) + abs( int(key[1]) - keyp_colors[0][1] ) + abs( int(key[2]) - keyp_colors[0][2] )
- deltaclr = prefs.keyp_delta*prefs.keyp_delta*prefs.keyp_delta
- deltaid = 0
- if prefs.use_alternate_keys:
- delta = prefs.keyp_delta + prefs.keyp_colors_alternate_sensitivity[i]
- if ( abs( int(key[0]) - prefs.keyp_colors_alternate[i][0] ) > delta ) and ( abs( int(key[1]) - prefs.keyp_colors_alternate[i][1] ) > delta ) and ( abs( int(key[2]) - prefs.keyp_colors_alternate[i][2] ) > delta ):
- keypressed = 1
- pressedcolor = prefs.keyp_colors_alternate[i]
- else:
- for j in range(len(prefs.keyp_colors)):
- delta = prefs.keyp_delta
- if prefs.use_percolor_delta:
- if j < len( prefs.percolor_delta ):
- delta = prefs.percolor_delta[ j ]
- deltaclr = delta*delta*delta
- if (prefs.keyp_colors[j][0] != 0 ) or ( prefs.keyp_colors[j][1] != 0 ) or ( prefs.keyp_colors[j][2] != 0 ):
- if ( abs( int(key[0]) - prefs.keyp_colors[j][0] ) < delta ) and ( abs( int(key[1]) - prefs.keyp_colors[j][1] ) < delta ) and ( abs( int(key[2]) - prefs.keyp_colors[j][2] ) < delta ):
- delta = abs( int(key[0]) - prefs.keyp_colors[j][0] ) + abs( int(key[1]) - prefs.keyp_colors[j][1] ) + abs( int(key[2]) - prefs.keyp_colors[j][2] )
- if ( delta < deltaclr ):
- deltaclr = delta
- deltaid = j
- keypressed=1
- if prefs.use_sparks:
- has_spark_delta = ((sparkkey[0] - prefs.keyp_colors[j][0] ) > prefs.keyp_colors_sparks_sensitivity[j] ) or ((sparkkey[1] - prefs.keyp_colors[j][1] ) > prefs.keyp_colors_sparks_sensitivity[j] ) or ((sparkkey[2] - prefs.keyp_colors[j][2] ) > prefs.keyp_colors_sparks_sensitivity[j] )
- #if ( abs( int(sparkkey[0]) - keyp_colors[j][0] ) < keyp_colors_sparks_sensitivity[j] ) and ( abs( int(sparkkey[1]) - keyp_colors[j][1] ) < keyp_colors_sparks_sensitivity[j] ) and ( abs( int(sparkkey[2]) - keyp_colors[j][2] ) < keyp_colors_sparks_sensitivity[j] ):
- if ( not has_spark_delta ):
- keypressed=0
- if ( keypressed != 0 ):
- note_channel=prefs.keyp_colors_channel[ deltaid ]
- if ( prefs.debug == 1 ):
- if (keypressed == 1 ):
- cv2.rectangle(image, (pixx-5,pixy-5), (pixx+5,pixy+5), (128,128,255), -1 )
- cv2.putText(image, str(note_channel), (pixx-5,pixy-10), 0, 0.3, (64,128,255))
- # cv2.rectangle(image, (pixx-5,pixy-5), (pixx+5,pixy+5), (255,0,255))
- cv2.rectangle(image, (pixx-1,pixy-1), (pixx+1,pixy+1), (255,0,255))
- # cv2.putText(image, str(note), (pixx-5,pixy+20), 0, 0.5, (255,0,255))
- # reg pressed key; when keypressed==2 and previous keypressed state is 0 or 2 we should also goes here
- if keypressed==1 or (keypressed==2 and notes[note] != 1):
- # if key is not pressed
- if ( notes[note] == 0 ):
- if ( debug_keys == 1 ):
- print("note pressed on :" + str( note ))
- notes_db[ note ] = frame
- if (first_note_time == 0):
- first_note_time = frame / fps
- notes_channel[ note ] = note_channel
- if ( separate_note_id != -1 ):
- if ( separate_note_id < note ):
- notes_channel[ note ] = 0
- else:
- notes_channel[ note ] = 1
- # always update to last press state
- notes[ note ] = keypressed
- notes_tmp[ note] = keypressed
- # save fall notes and then we can check for a near keys with priority...
- if prefs.rollcheck:
- for i in range(1, len( prefs.keys_pos)-1 ):
- if notes[ i ] != 0:
- if prefs.rollcheck_priority == 0:
- if not iswhitekey(i):
- # Priority on Black keys
- if notes[i+1] >0 and notes_tmp[i] >0: notes[i] = 0
- if notes[i-1] >0 and notes_tmp[i] >0: notes[i] = 0
- else:
- if iswhitekey(i):
- # Priority on White keys
- if notes[i+1] >0 and notes_tmp[i] >0: notes[i] = 0
- if notes[i-1] >0 and notes_tmp[i] >0: notes[i] = 0
- #
- for i in range( len( prefs.keys_pos) ):
- note=i
- keypressed = notes[ note ]
- if notes_tmp[ i ] != 0:
- if ( notes[note] != 0 ) and ( notes_channel[ note ] != note_channel ) and ( prefs.notes_overlap == 1 ):
- # case if one key over other
- time = notes_db[note] / fps
- duration = ( frame - notes_db[note] ) / fps
- if (use_snap_notes_to_grid == 1):
- #print ("1 time:", time , "first_note_time:",first_note_time)
- time = snap_to_grid( time - first_note_time , notes_grid_size ) + 1
- duration = snap_to_grid( duration , notes_grid_size )
- #print ("1 time after:", time , "after before:",duration)
- ignore = 0
- if ( duration < prefs.minimal_duration ):
- if ( debug_keys == 1 ):
- print(" duration:" + str(duration) + " < minimal_duration:" + str(prefs.minimal_duration))
- duration = prefs.minimal_duration
- if ( prefs.ignore_minimal_duration == 1 ):
- ignore=1
- if ( debug_keys == 1 ):
- print("keys (one over other), note released :" + str(note) + " de = " + str(notes_de[note]) + "- db =" + str(notes_db[note]))
- print("midi add white keys, note : " +str(note) + " time:" +str(time) + " duration:" + str(duration))
- if ( not ignore ):
- mf.addNote(track, notes_channel[note] , basenote + note, time * prefs.tempo / 60.0 , duration * prefs.tempo / 60.0 , volume )
- channel_has_note[ note_channel ] = 1
- notecnt+=1
- notes_db[ note ] = frame
- notes_channel[ note ] = note_channel
- else:
- # if key been presed and released: two cases goes here keypressed==0 or (keypressed==2 and previous state is keypressed==1)
- if ( notes[note] != 0):
- notes[ note ] = 0
- notes_de[ note ] = frame
- time = notes_db[note] / fps
- duration = ( notes_de[note] - notes_db[note] ) / fps
- if (use_snap_notes_to_grid):
- if (first_note_time == 0):
- first_note_time = time
- #print ("2 time:", time , "first_note_time:",first_note_time)
- time = snap_to_grid( time - first_note_time , notes_grid_size ) + 1
- duration = snap_to_grid( duration , notes_grid_size )
- ignore=0
- if ( duration < prefs.minimal_duration ):
- if ( debug_keys == 1 ):
- print(" duration:" + str(duration) + " < minimal_duration:" + str(prefs.minimal_duration))
- duration = prefs.minimal_duration
- if ( prefs.ignore_minimal_duration == 1 ):
- ignore=1
- if ( debug_keys == 1 ):
- print("keys, note released :" + str(note ) + " de = " + str(notes_de[note]) + "- db =" + str(notes_db[note]))
- print("midi add white keys, note : " +str(note) + " time:" +str(time) + " duration:" + str(duration))
- if ( not ignore ):
- mf.addNote(track, notes_channel[note] , basenote+ note, time * prefs.tempo / 60.0 , duration * prefs.tempo / 60.0 , volume )
- channel_has_note[ note_channel ] = 1
- notecnt+=1
- # coming here when use sparks is true and previous state is keypressed==1. We consider the key is released and then pressed again
- if (keypressed==2):
- notes[ note ] = keypressed
- notes_db[ note ] = frame
- notes_channel[ note ] = note_channel
- xapp=0
- if ( prefs.debug == 1 ):
- cv2.imwrite("/tmp/frame%d.jpg" % frame, image) # save frame as JPEG file
- # success,image = vidcap.read()
- getFrame()
- frame += 1
- framerate()
- if ( frame > endframe ):
- success = False
- for event in pygame.event.get():
- if event.type == pygame.QUIT:
- success = False
- pygame.quit()
- quit()
- elif event.type == pygame.KEYDOWN:
- if event.key == pygame.K_SPACE:
- success = False
- if event.key == pygame.K_ESCAPE:
- running = 0
- pygame.quit()
- quit()
- print("saved notes: " + str(notecnt))
- #search free id for name ...
- fileid=0
- while os.path.exists( outputmid ):
- outputmid = ntpath.basename( filepath ) + "_"+str(fileid)+ "_output.mid"
- fileid+=1
- if ( fileid > 999 ): break
- if prefs.sync_notes_start_pos:
- mf.sync_start_pos(prefs.sync_notes_start_pos_time_delta, False)
- if prefs.save_to_disk_per_channel:
- status, prefs.save_to_disk_message = mf.save_to_disk_per_channel(outputmid)
- else:
- status, prefs.save_to_disk_message = mf.save_to_disk(outputmid)
- return status
- def doinit():
- doinitGl()
- loadImage()
- GenFontTexture()
- def reconstruct():
- global frame
- global showoutputpath
- helpWindow.hidden=1
- frame=prefs.startframe
- t1 = datetime.datetime.now()
- processmidi()
- t2 = datetime.datetime.now()
- print(""" processing time: {} / {} = {}; """.format( t1,t2, t2-t1 ))
- frame=prefs.startframe
- getFrame(frame)
- showoutputpath = time.time() + 5
- def main():
- global pyfont
- global mousex, mousey
- global keyp_colormap_colors_pos
- global keyp_colormap_pos
- global success,image
- global endframe
- global basenote
- global glwindows
- global separate_note_id
- global frame
- global width,height
- global screen
- global lastkeygrabid
- global running
- #global old_spark_color, cur_spark_color
- keygrab=0
- keygrabid=-1
- keygrabaddx=0
- lastkeygrabid=-1
- #pyfont = pygame.font.SysFont('Sans', 20)
- #pygame.RESIZABLE
- screen = pygame.display.set_mode( (width,height) , DOUBLEBUF|OPENGL|pygame.RESIZABLE)
- pygame.display.set_caption(filepath)
- doinit()
- clock = pygame.time.Clock()
- # set start frame
- vidcap.set(CAP_PROP_POS_FRAMES, frame)
- while running==1:
- mouseOnWindows = False
- # mousex, mousey = pygame.mouse.get_pos()
- drawframe()
- mods = pygame.key.get_mods()
- for event in pygame.event.get():
- if event.type == pygame.QUIT:
- running = 0
- pygame.quit()
- quit()
- elif event.type == pygame.VIDEORESIZE:
- prefs.resize = 1
- prefs.resize_width = event.w
- prefs.resize_height = event.h
- width = prefs.resize_width
- height = prefs.resize_height
- screen = pygame.display.set_mode( (width,height) , DOUBLEBUF|OPENGL|pygame.RESIZABLE)
- elif event.type == pygame.KEYUP:
- for wnd in glwindows:
- wnd.update_key_up(event.key)
- elif event.type == pygame.KEYDOWN:
- for wnd in glwindows:
- wnd.update_key_down(event.key)
- # print event.key
- if event.key == pygame.K_q:
- if prefs.autoclose == 1:
- running = 0
- else:
- reconstruct()
- if event.key == pygame.K_o:
- #prefs.notes_overlap = not prefs.notes_overlap
- switch_notes_overlap(None)
- if event.key == pygame.K_i:
- #prefs.ignore_minimal_duration = not prefs.ignore_minimal_duration
- switch_ignore_notes_with_minimal_duration(None)
- if event.key == pygame.K_s:
- if mods & pygame.KMOD_SHIFT:
- prefs.startframe = 0
- else:
- prefs.startframe = int(round(vidcap.get(1)))
- print("set start frame = "+ str(prefs.startframe))
- if event.key == pygame.K_e:
- if mods & pygame.KMOD_SHIFT:
- endframe = length
- else:
- endframe = int(round(vidcap.get(1)))
- print("set end frame = "+ str(endframe))
- if event.key == pygame.K_ESCAPE:
- running = 0
- pygame.quit()
- quit()
- if event.key == pygame.K_F2:
- btndown_save_settings(None)
- if event.key == pygame.K_F3:
- btndown_load_settings(None)
- if event.key == pygame.K_F4:
- for i in range(len(glwindows)):
- # if isinstance(glwindows[i],GLButton):
- # continue
- glwindows[i].x = mousex;
- glwindows[i].y = mousey;
- if event.key == pygame.K_r:
- switch_resize_windows(None)
- if event.key == pygame.K_RIGHTBRACKET:
- raise_octave()
- if event.key == pygame.K_LEFTBRACKET:
- lower_octave()
- if event.key == pygame.K_PLUS or event.key == pygame.K_KP_PLUS or event.key == pygame.K_EQUALS:
- prefs.keys_angle -= 5
- updatekeys()
- if event.key == pygame.K_MINUS or event.key == pygame.K_KP_MINUS:
- prefs.keys_angle += 5
- updatekeys()
- if event.key == pygame.K_UP:
- if mods & pygame.KMOD_ALT:
- prefs.keyp_spark_y_pos -= 1
- else:
- if mods & pygame.KMOD_SHIFT:
- prefs.yoffset_blackkeys -= 1
- else:
- prefs.yoffset_blackkeys -= 2
- updatekeys( )
- if event.key == pygame.K_DOWN:
- if mods & pygame.KMOD_ALT:
- prefs.keyp_spark_y_pos += 1
- else:
- if mods & pygame.KMOD_SHIFT:
- prefs.yoffset_blackkeys += 1
- else:
- prefs.yoffset_blackkeys += 2
- updatekeys( )
- if event.key == pygame.K_TAB:
- showOrhideallwindows(None)
- if event.key == pygame.K_LEFT:
- if mods & pygame.KMOD_SHIFT:
- prefs.whitekey_width-=0.1
- else:
- prefs.whitekey_width-=1.0
- updatekeys( )
- if event.key == pygame.K_RIGHT:
- if mods & pygame.KMOD_SHIFT:
- prefs.whitekey_width+=0.1
- else:
- prefs.whitekey_width+=1.0
- updatekeys( )
- if event.key == pygame.K_HOME:
- scroll_to_start(None)
- if event.key == pygame.K_END:
- scroll_to_end(None)
- if event.key == pygame.K_0:
- if mods & pygame.KMOD_CTRL and Gl.keyp_colormap_id != -1:
- prefs.keyp_colors[Gl.keyp_colormap_id][0] = 0
- prefs.keyp_colors[Gl.keyp_colormap_id][1] = 0
- prefs.keyp_colors[Gl.keyp_colormap_id][2] = 0
- if event.key == pygame.K_PAGEUP:
- if mods & pygame.KMOD_SHIFT:
- scroll_forward_by_frame(None)
- else:
- scroll_fast_forward(None)
- if event.key == pygame.K_PAGEDOWN:
- if mods & pygame.KMOD_SHIFT:
- scroll_prev_by_frame(None)
- else:
- scroll_fast_prev(None)
- if event.key == pygame.K_p:
- size=5
- separate_note_id=-1
- for i in range( len( prefs.keys_pos) ):
- if (abs( mousex - (prefs.keys_pos[i][0] + prefs.xoffset_whitekeys) )< size) and (abs( mousey - (prefs.keys_pos[i][1] + prefs.yoffset_whitekeys) )< size):
- separate_note_id=i
- if event.key == pygame.K_KP4:
- if lastkeygrabid >0 and lastkeygrabid < len(prefs.keys_pos):
- prefs.keys_pos[lastkeygrabid][0] -= 1
- if event.key == pygame.K_KP6:
- if lastkeygrabid >0 and lastkeygrabid < len(prefs.keys_pos):
- prefs.keys_pos[lastkeygrabid][0] += 1
- if event.key == pygame.K_KP8:
- if lastkeygrabid >0 and lastkeygrabid < len(prefs.keys_pos):
- prefs.keys_pos[lastkeygrabid][1] -= 1
- if event.key == pygame.K_KP2:
- if lastkeygrabid >0 and lastkeygrabid < len(prefs.keys_pos):
- prefs.keys_pos[lastkeygrabid][1] += 1
- if event.key == pygame.K_KP1:
- vertical_align_keys(1, 1)
- if event.key == pygame.K_KP3:
- vertical_align_keys(1, 0)
- elif event.type == pygame.MOUSEBUTTONUP:
- for i in range( len(glwindows)-1, -1 , -1):
- #print("process mouse up on windiws id: ", i)
- if glwindows[i].update_mouse_up(mousex,mousey,event.button) == 1:
- mouseOnWindows=True
- resort=True
- break
- if ( event.button == 1 ):
- keygrab = 0
- keygrabid = -1
- if ( event.button == 3 ):
- keygrab = 0
- elif event.type == pygame.MOUSEBUTTONDOWN:
- resort=False
- for i in range( len(glwindows)-1, -1 , -1):
- #print("process mouse down on windiws id: ", i)
- if glwindows[i].update_mouse_down(mousex,mousey,event.button) == 1:
- mouseOnWindows=True
- resort=True
- break
- if resort:
- glwindows.sort(key=lambda x: x.active, reverse=False)
- # print event.button
- if ( event.button == 4 ):
- prefs.whitekey_width+=0.05
- # print "whitekey_width="+str(whitekey_width)
- updatekeys( )
- # scale+=0.1
- if ( event.button == 5 ):
- prefs.whitekey_width-=0.05
- # print "whitekey_width="+str(whitekey_width)
- updatekeys( )
- if ( event.button == 1 ):
- if mods & pygame.KMOD_CTRL and Gl.keyp_colormap_id != -1:
- pixx = int(mousex)
- pixy = int(mousey)
- if not (( pixx >= width ) or ( pixy >= height ) or ( pixx < 0 ) or ( pixy < 0 )):
- if ( prefs.resize == 1 ):
- pixx= int(round( pixx * ( video_width / float(prefs.resize_width) )))
- pixy= int(round( pixy * ( video_height / float(prefs.resize_height) )))
- if ( pixx > video_width -1 ): pixx = video_width-1
- if ( pixy > video_height-1 ): pixy = video_height-1
- print("original mouse x:"+str(mousex) + "x" +str(mousey) + " mapped :" +str(pixx) +"x"+str(pixy))
- keybgr=image[pixy,pixx]
- prefs.keyp_colors[Gl.keyp_colormap_id][0] = keybgr[2]
- prefs.keyp_colors[Gl.keyp_colormap_id][1] = keybgr[1]
- prefs.keyp_colors[Gl.keyp_colormap_id][2] = keybgr[0]
- else:
- # if not (mods & pygame.KMOD_CTRL):
- if not colorWindow.active and not mouseOnWindows:
- Gl.keyp_colormap_id = -1
- #keyp_colormap_id = -1
- size=5
- if (mods & pygame.KMOD_CTRL):
- lastkeygrabid=-1
- for i in range( len( prefs.keys_pos) ):
- if (abs( mousex - (prefs.keys_pos[i][0] + prefs.xoffset_whitekeys) )< size) and (abs( mousey - (prefs.keys_pos[i][1] + prefs.yoffset_whitekeys) )< size):
- keygrab=1
- if not ( mods & pygame.KMOD_CTRL ):
- keygrabid=i
- lastkeygrabid=i
- extra_slider1.setvalue( prefs.keyp_colors_alternate_sensitivity[i] )
- print("ok click found on : "+str(keygrabid))
- break
- # if ( event.button == 2 ):
- # lastkeygrabid=-1
- if ( event.button == 3 ):
- keygrab = 2
- size=5
- print("x offset " + str(prefs.xoffset_whitekeys) + " y offset: " +str(prefs.yoffset_whitekeys))
- keygrabaddx=0
- for i in range( len( prefs.keys_pos) ):
- if (abs( mousex - (prefs.keys_pos[i][0] + prefs.xoffset_whitekeys) )< size) and (abs( mousey - (prefs.keys_pos[i][1] + prefs.yoffset_whitekeys) )< size):
- keygrab=2
- keygrabaddx=prefs.keys_pos[i][0]
- print("ok click found on : "+str(keygrabid))
- break
- if ( keygrab == 1) and ( keygrabid >-1 ):
- # print "moving keyid = " + str(keygrabid)
- prefs.keys_pos[ keygrabid ][0] = mousex - prefs.xoffset_whitekeys
- prefs.keys_pos[ keygrabid ][1] = mousey - prefs.yoffset_whitekeys
- if ( keygrab == 2):
- # print "moving offsets : "+ str(mousex) + " x " + str(mousey)
- prefs.xoffset_whitekeys = mousex - keygrabaddx
- prefs.yoffset_whitekeys = mousey
- for wnd in glwindows:
- wnd.update_mouse_move(mousex,mousey)
- pygame.display.flip()
- #framerate()
- #limit fps to 60 and get the frame time in milliseconds
- ms = clock.tick(60)
- main()
- if prefs.autoclose == 1:
- reconstruct()
- print ('done...')
|