| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284328532863287328832893290329132923293329432953296329732983299330033013302330333043305330633073308330933103311331233133314331533163317331833193320332133223323332433253326332733283329333033313332333333343335333633373338333933403341334233433344334533463347334833493350335133523353335433553356335733583359336033613362336333643365336633673368336933703371337233733374337533763377337833793380338133823383338433853386338733883389339033913392339333943395339633973398339934003401340234033404340534063407340834093410341134123413341434153416341734183419342034213422342334243425342634273428342934303431343234333434343534363437343834393440344134423443344434453446344734483449345034513452345334543455345634573458345934603461346234633464346534663467346834693470347134723473347434753476347734783479348034813482348334843485348634873488348934903491349234933494349534963497349834993500350135023503350435053506350735083509351035113512351335143515351635173518351935203521352235233524352535263527352835293530353135323533353435353536353735383539354035413542354335443545354635473548354935503551355235533554355535563557355835593560356135623563356435653566356735683569357035713572357335743575357635773578357935803581358235833584358535863587358835893590359135923593359435953596359735983599360036013602360336043605360636073608360936103611361236133614361536163617361836193620362136223623362436253626362736283629363036313632363336343635363636373638363936403641364236433644364536463647364836493650365136523653365436553656365736583659366036613662366336643665366636673668366936703671367236733674367536763677367836793680368136823683368436853686368736883689369036913692369336943695369636973698369937003701370237033704370537063707370837093710371137123713371437153716371737183719372037213722372337243725372637273728372937303731373237333734373537363737373837393740374137423743374437453746374737483749375037513752375337543755375637573758375937603761376237633764376537663767376837693770377137723773377437753776377737783779378037813782378337843785378637873788378937903791379237933794379537963797379837993800380138023803380438053806380738083809381038113812381338143815381638173818381938203821382238233824382538263827382838293830383138323833383438353836383738383839384038413842384338443845384638473848384938503851385238533854385538563857385838593860386138623863386438653866386738683869387038713872387338743875387638773878387938803881388238833884388538863887388838893890389138923893389438953896389738983899390039013902390339043905390639073908390939103911391239133914391539163917391839193920392139223923392439253926392739283929393039313932393339343935393639373938393939403941394239433944394539463947394839493950395139523953395439553956395739583959396039613962396339643965396639673968396939703971397239733974397539763977397839793980398139823983398439853986398739883989399039913992399339943995399639973998399940004001400240034004400540064007400840094010401140124013401440154016401740184019402040214022402340244025402640274028402940304031403240334034403540364037403840394040404140424043404440454046404740484049405040514052405340544055405640574058405940604061406240634064406540664067406840694070407140724073407440754076407740784079408040814082408340844085408640874088408940904091409240934094409540964097409840994100410141024103410441054106410741084109411041114112411341144115411641174118411941204121412241234124412541264127412841294130413141324133413441354136413741384139414041414142414341444145414641474148414941504151415241534154415541564157415841594160416141624163416441654166416741684169417041714172417341744175417641774178417941804181418241834184418541864187418841894190419141924193419441954196419741984199420042014202420342044205420642074208420942104211421242134214421542164217421842194220422142224223422442254226422742284229423042314232423342344235423642374238423942404241424242434244424542464247424842494250425142524253425442554256425742584259426042614262426342644265426642674268426942704271427242734274427542764277427842794280428142824283428442854286428742884289429042914292429342944295429642974298429943004301430243034304430543064307430843094310431143124313431443154316431743184319432043214322432343244325432643274328432943304331433243334334433543364337433843394340434143424343434443454346434743484349435043514352435343544355435643574358435943604361436243634364436543664367436843694370437143724373437443754376437743784379438043814382438343844385438643874388438943904391439243934394439543964397439843994400440144024403440444054406440744084409441044114412441344144415441644174418441944204421442244234424442544264427442844294430443144324433443444354436443744384439444044414442444344444445444644474448444944504451445244534454445544564457445844594460446144624463446444654466446744684469447044714472447344744475447644774478447944804481448244834484448544864487448844894490449144924493449444954496449744984499450045014502450345044505450645074508450945104511451245134514451545164517451845194520452145224523452445254526452745284529453045314532453345344535453645374538453945404541454245434544454545464547454845494550455145524553455445554556455745584559456045614562456345644565456645674568456945704571457245734574457545764577457845794580458145824583458445854586458745884589459045914592459345944595459645974598459946004601460246034604460546064607460846094610461146124613461446154616461746184619462046214622462346244625462646274628462946304631463246334634463546364637463846394640464146424643464446454646464746484649465046514652465346544655465646574658465946604661466246634664466546664667466846694670467146724673467446754676467746784679468046814682468346844685468646874688468946904691469246934694469546964697469846994700470147024703470447054706470747084709471047114712471347144715471647174718471947204721472247234724472547264727472847294730473147324733473447354736473747384739474047414742474347444745474647474748474947504751475247534754475547564757475847594760476147624763476447654766476747684769477047714772477347744775477647774778477947804781478247834784478547864787478847894790479147924793479447954796479747984799480048014802480348044805480648074808480948104811481248134814481548164817481848194820482148224823482448254826482748284829483048314832483348344835483648374838483948404841484248434844484548464847484848494850485148524853485448554856485748584859486048614862486348644865486648674868486948704871487248734874487548764877487848794880488148824883488448854886488748884889489048914892489348944895489648974898489949004901490249034904490549064907490849094910491149124913491449154916491749184919492049214922492349244925492649274928492949304931493249334934493549364937493849394940494149424943494449454946494749484949495049514952495349544955495649574958495949604961496249634964496549664967496849694970497149724973497449754976497749784979498049814982498349844985498649874988498949904991499249934994499549964997499849995000500150025003500450055006500750085009501050115012501350145015501650175018501950205021502250235024502550265027502850295030 |
- #!/usr/bin/env python
- # SPDX-License-Identifier: GPL-2.0
- # exported-sql-viewer.py: view data from sql database
- # Copyright (c) 2014-2018, Intel Corporation.
- # To use this script you will need to have exported data using either the
- # export-to-sqlite.py or the export-to-postgresql.py script. Refer to those
- # scripts for details.
- #
- # Following on from the example in the export scripts, a
- # call-graph can be displayed for the pt_example database like this:
- #
- # python tools/perf/scripts/python/exported-sql-viewer.py pt_example
- #
- # Note that for PostgreSQL, this script supports connecting to remote databases
- # by setting hostname, port, username, password, and dbname e.g.
- #
- # python tools/perf/scripts/python/exported-sql-viewer.py "hostname=myhost username=myuser password=mypassword dbname=pt_example"
- #
- # The result is a GUI window with a tree representing a context-sensitive
- # call-graph. Expanding a couple of levels of the tree and adjusting column
- # widths to suit will display something like:
- #
- # Call Graph: pt_example
- # Call Path Object Count Time(ns) Time(%) Branch Count Branch Count(%)
- # v- ls
- # v- 2638:2638
- # v- _start ld-2.19.so 1 10074071 100.0 211135 100.0
- # |- unknown unknown 1 13198 0.1 1 0.0
- # >- _dl_start ld-2.19.so 1 1400980 13.9 19637 9.3
- # >- _d_linit_internal ld-2.19.so 1 448152 4.4 11094 5.3
- # v-__libc_start_main@plt ls 1 8211741 81.5 180397 85.4
- # >- _dl_fixup ld-2.19.so 1 7607 0.1 108 0.1
- # >- __cxa_atexit libc-2.19.so 1 11737 0.1 10 0.0
- # >- __libc_csu_init ls 1 10354 0.1 10 0.0
- # |- _setjmp libc-2.19.so 1 0 0.0 4 0.0
- # v- main ls 1 8182043 99.6 180254 99.9
- #
- # Points to note:
- # The top level is a command name (comm)
- # The next level is a thread (pid:tid)
- # Subsequent levels are functions
- # 'Count' is the number of calls
- # 'Time' is the elapsed time until the function returns
- # Percentages are relative to the level above
- # 'Branch Count' is the total number of branches for that function and all
- # functions that it calls
- # There is also a "All branches" report, which displays branches and
- # possibly disassembly. However, presently, the only supported disassembler is
- # Intel XED, and additionally the object code must be present in perf build ID
- # cache. To use Intel XED, libxed.so must be present. To build and install
- # libxed.so:
- # git clone https://github.com/intelxed/mbuild.git mbuild
- # git clone https://github.com/intelxed/xed
- # cd xed
- # ./mfile.py --share
- # sudo ./mfile.py --prefix=/usr/local install
- # sudo ldconfig
- #
- # Example report:
- #
- # Time CPU Command PID TID Branch Type In Tx Branch
- # 8107675239590 2 ls 22011 22011 return from interrupt No ffffffff86a00a67 native_irq_return_iret ([kernel]) -> 7fab593ea260 _start (ld-2.19.so)
- # 7fab593ea260 48 89 e7 mov %rsp, %rdi
- # 8107675239899 2 ls 22011 22011 hardware interrupt No 7fab593ea260 _start (ld-2.19.so) -> ffffffff86a012e0 page_fault ([kernel])
- # 8107675241900 2 ls 22011 22011 return from interrupt No ffffffff86a00a67 native_irq_return_iret ([kernel]) -> 7fab593ea260 _start (ld-2.19.so)
- # 7fab593ea260 48 89 e7 mov %rsp, %rdi
- # 7fab593ea263 e8 c8 06 00 00 callq 0x7fab593ea930
- # 8107675241900 2 ls 22011 22011 call No 7fab593ea263 _start+0x3 (ld-2.19.so) -> 7fab593ea930 _dl_start (ld-2.19.so)
- # 7fab593ea930 55 pushq %rbp
- # 7fab593ea931 48 89 e5 mov %rsp, %rbp
- # 7fab593ea934 41 57 pushq %r15
- # 7fab593ea936 41 56 pushq %r14
- # 7fab593ea938 41 55 pushq %r13
- # 7fab593ea93a 41 54 pushq %r12
- # 7fab593ea93c 53 pushq %rbx
- # 7fab593ea93d 48 89 fb mov %rdi, %rbx
- # 7fab593ea940 48 83 ec 68 sub $0x68, %rsp
- # 7fab593ea944 0f 31 rdtsc
- # 7fab593ea946 48 c1 e2 20 shl $0x20, %rdx
- # 7fab593ea94a 89 c0 mov %eax, %eax
- # 7fab593ea94c 48 09 c2 or %rax, %rdx
- # 7fab593ea94f 48 8b 05 1a 15 22 00 movq 0x22151a(%rip), %rax
- # 8107675242232 2 ls 22011 22011 hardware interrupt No 7fab593ea94f _dl_start+0x1f (ld-2.19.so) -> ffffffff86a012e0 page_fault ([kernel])
- # 8107675242900 2 ls 22011 22011 return from interrupt No ffffffff86a00a67 native_irq_return_iret ([kernel]) -> 7fab593ea94f _dl_start+0x1f (ld-2.19.so)
- # 7fab593ea94f 48 8b 05 1a 15 22 00 movq 0x22151a(%rip), %rax
- # 7fab593ea956 48 89 15 3b 13 22 00 movq %rdx, 0x22133b(%rip)
- # 8107675243232 2 ls 22011 22011 hardware interrupt No 7fab593ea956 _dl_start+0x26 (ld-2.19.so) -> ffffffff86a012e0 page_fault ([kernel])
- from __future__ import print_function
- import sys
- # Only change warnings if the python -W option was not used
- if not sys.warnoptions:
- import warnings
- # PySide2 causes deprecation warnings, ignore them.
- warnings.filterwarnings("ignore", category=DeprecationWarning)
- import argparse
- import weakref
- import threading
- import string
- try:
- # Python2
- import cPickle as pickle
- # size of pickled integer big enough for record size
- glb_nsz = 8
- except ImportError:
- import pickle
- glb_nsz = 16
- import re
- import os
- import random
- import copy
- import math
- from libxed import LibXED
- pyside_version_1 = True
- if not "--pyside-version-1" in sys.argv:
- try:
- from PySide2.QtCore import *
- from PySide2.QtGui import *
- from PySide2.QtSql import *
- from PySide2.QtWidgets import *
- pyside_version_1 = False
- except:
- pass
- if pyside_version_1:
- from PySide.QtCore import *
- from PySide.QtGui import *
- from PySide.QtSql import *
- from decimal import Decimal, ROUND_HALF_UP
- from ctypes import CDLL, Structure, create_string_buffer, addressof, sizeof, \
- c_void_p, c_bool, c_byte, c_char, c_int, c_uint, c_longlong, c_ulonglong
- from multiprocessing import Process, Array, Value, Event
- # xrange is range in Python3
- try:
- xrange
- except NameError:
- xrange = range
- def printerr(*args, **keyword_args):
- print(*args, file=sys.stderr, **keyword_args)
- # Data formatting helpers
- def tohex(ip):
- if ip < 0:
- ip += 1 << 64
- return "%x" % ip
- def offstr(offset):
- if offset:
- return "+0x%x" % offset
- return ""
- def dsoname(name):
- if name == "[kernel.kallsyms]":
- return "[kernel]"
- return name
- def findnth(s, sub, n, offs=0):
- pos = s.find(sub)
- if pos < 0:
- return pos
- if n <= 1:
- return offs + pos
- return findnth(s[pos + 1:], sub, n - 1, offs + pos + 1)
- # Percent to one decimal place
- def PercentToOneDP(n, d):
- if not d:
- return "0.0"
- x = (n * Decimal(100)) / d
- return str(x.quantize(Decimal(".1"), rounding=ROUND_HALF_UP))
- # Helper for queries that must not fail
- def QueryExec(query, stmt):
- ret = query.exec_(stmt)
- if not ret:
- raise Exception("Query failed: " + query.lastError().text())
- # Background thread
- class Thread(QThread):
- done = Signal(object)
- def __init__(self, task, param=None, parent=None):
- super(Thread, self).__init__(parent)
- self.task = task
- self.param = param
- def run(self):
- while True:
- if self.param is None:
- done, result = self.task()
- else:
- done, result = self.task(self.param)
- self.done.emit(result)
- if done:
- break
- # Tree data model
- class TreeModel(QAbstractItemModel):
- def __init__(self, glb, params, parent=None):
- super(TreeModel, self).__init__(parent)
- self.glb = glb
- self.params = params
- self.root = self.GetRoot()
- self.last_row_read = 0
- def Item(self, parent):
- if parent.isValid():
- return parent.internalPointer()
- else:
- return self.root
- def rowCount(self, parent):
- result = self.Item(parent).childCount()
- if result < 0:
- result = 0
- self.dataChanged.emit(parent, parent)
- return result
- def hasChildren(self, parent):
- return self.Item(parent).hasChildren()
- def headerData(self, section, orientation, role):
- if role == Qt.TextAlignmentRole:
- return self.columnAlignment(section)
- if role != Qt.DisplayRole:
- return None
- if orientation != Qt.Horizontal:
- return None
- return self.columnHeader(section)
- def parent(self, child):
- child_item = child.internalPointer()
- if child_item is self.root:
- return QModelIndex()
- parent_item = child_item.getParentItem()
- return self.createIndex(parent_item.getRow(), 0, parent_item)
- def index(self, row, column, parent):
- child_item = self.Item(parent).getChildItem(row)
- return self.createIndex(row, column, child_item)
- def DisplayData(self, item, index):
- return item.getData(index.column())
- def FetchIfNeeded(self, row):
- if row > self.last_row_read:
- self.last_row_read = row
- if row + 10 >= self.root.child_count:
- self.fetcher.Fetch(glb_chunk_sz)
- def columnAlignment(self, column):
- return Qt.AlignLeft
- def columnFont(self, column):
- return None
- def data(self, index, role):
- if role == Qt.TextAlignmentRole:
- return self.columnAlignment(index.column())
- if role == Qt.FontRole:
- return self.columnFont(index.column())
- if role != Qt.DisplayRole:
- return None
- item = index.internalPointer()
- return self.DisplayData(item, index)
- # Table data model
- class TableModel(QAbstractTableModel):
- def __init__(self, parent=None):
- super(TableModel, self).__init__(parent)
- self.child_count = 0
- self.child_items = []
- self.last_row_read = 0
- def Item(self, parent):
- if parent.isValid():
- return parent.internalPointer()
- else:
- return self
- def rowCount(self, parent):
- return self.child_count
- def headerData(self, section, orientation, role):
- if role == Qt.TextAlignmentRole:
- return self.columnAlignment(section)
- if role != Qt.DisplayRole:
- return None
- if orientation != Qt.Horizontal:
- return None
- return self.columnHeader(section)
- def index(self, row, column, parent):
- return self.createIndex(row, column, self.child_items[row])
- def DisplayData(self, item, index):
- return item.getData(index.column())
- def FetchIfNeeded(self, row):
- if row > self.last_row_read:
- self.last_row_read = row
- if row + 10 >= self.child_count:
- self.fetcher.Fetch(glb_chunk_sz)
- def columnAlignment(self, column):
- return Qt.AlignLeft
- def columnFont(self, column):
- return None
- def data(self, index, role):
- if role == Qt.TextAlignmentRole:
- return self.columnAlignment(index.column())
- if role == Qt.FontRole:
- return self.columnFont(index.column())
- if role != Qt.DisplayRole:
- return None
- item = index.internalPointer()
- return self.DisplayData(item, index)
- # Model cache
- model_cache = weakref.WeakValueDictionary()
- model_cache_lock = threading.Lock()
- def LookupCreateModel(model_name, create_fn):
- model_cache_lock.acquire()
- try:
- model = model_cache[model_name]
- except:
- model = None
- if model is None:
- model = create_fn()
- model_cache[model_name] = model
- model_cache_lock.release()
- return model
- def LookupModel(model_name):
- model_cache_lock.acquire()
- try:
- model = model_cache[model_name]
- except:
- model = None
- model_cache_lock.release()
- return model
- # Find bar
- class FindBar():
- def __init__(self, parent, finder, is_reg_expr=False):
- self.finder = finder
- self.context = []
- self.last_value = None
- self.last_pattern = None
- label = QLabel("Find:")
- label.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
- self.textbox = QComboBox()
- self.textbox.setEditable(True)
- self.textbox.currentIndexChanged.connect(self.ValueChanged)
- self.progress = QProgressBar()
- self.progress.setRange(0, 0)
- self.progress.hide()
- if is_reg_expr:
- self.pattern = QCheckBox("Regular Expression")
- else:
- self.pattern = QCheckBox("Pattern")
- self.pattern.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
- self.next_button = QToolButton()
- self.next_button.setIcon(parent.style().standardIcon(QStyle.SP_ArrowDown))
- self.next_button.released.connect(lambda: self.NextPrev(1))
- self.prev_button = QToolButton()
- self.prev_button.setIcon(parent.style().standardIcon(QStyle.SP_ArrowUp))
- self.prev_button.released.connect(lambda: self.NextPrev(-1))
- self.close_button = QToolButton()
- self.close_button.setIcon(parent.style().standardIcon(QStyle.SP_DockWidgetCloseButton))
- self.close_button.released.connect(self.Deactivate)
- self.hbox = QHBoxLayout()
- self.hbox.setContentsMargins(0, 0, 0, 0)
- self.hbox.addWidget(label)
- self.hbox.addWidget(self.textbox)
- self.hbox.addWidget(self.progress)
- self.hbox.addWidget(self.pattern)
- self.hbox.addWidget(self.next_button)
- self.hbox.addWidget(self.prev_button)
- self.hbox.addWidget(self.close_button)
- self.bar = QWidget()
- self.bar.setLayout(self.hbox)
- self.bar.hide()
- def Widget(self):
- return self.bar
- def Activate(self):
- self.bar.show()
- self.textbox.lineEdit().selectAll()
- self.textbox.setFocus()
- def Deactivate(self):
- self.bar.hide()
- def Busy(self):
- self.textbox.setEnabled(False)
- self.pattern.hide()
- self.next_button.hide()
- self.prev_button.hide()
- self.progress.show()
- def Idle(self):
- self.textbox.setEnabled(True)
- self.progress.hide()
- self.pattern.show()
- self.next_button.show()
- self.prev_button.show()
- def Find(self, direction):
- value = self.textbox.currentText()
- pattern = self.pattern.isChecked()
- self.last_value = value
- self.last_pattern = pattern
- self.finder.Find(value, direction, pattern, self.context)
- def ValueChanged(self):
- value = self.textbox.currentText()
- pattern = self.pattern.isChecked()
- index = self.textbox.currentIndex()
- data = self.textbox.itemData(index)
- # Store the pattern in the combo box to keep it with the text value
- if data == None:
- self.textbox.setItemData(index, pattern)
- else:
- self.pattern.setChecked(data)
- self.Find(0)
- def NextPrev(self, direction):
- value = self.textbox.currentText()
- pattern = self.pattern.isChecked()
- if value != self.last_value:
- index = self.textbox.findText(value)
- # Allow for a button press before the value has been added to the combo box
- if index < 0:
- index = self.textbox.count()
- self.textbox.addItem(value, pattern)
- self.textbox.setCurrentIndex(index)
- return
- else:
- self.textbox.setItemData(index, pattern)
- elif pattern != self.last_pattern:
- # Keep the pattern recorded in the combo box up to date
- index = self.textbox.currentIndex()
- self.textbox.setItemData(index, pattern)
- self.Find(direction)
- def NotFound(self):
- QMessageBox.information(self.bar, "Find", "'" + self.textbox.currentText() + "' not found")
- # Context-sensitive call graph data model item base
- class CallGraphLevelItemBase(object):
- def __init__(self, glb, params, row, parent_item):
- self.glb = glb
- self.params = params
- self.row = row
- self.parent_item = parent_item
- self.query_done = False
- self.child_count = 0
- self.child_items = []
- if parent_item:
- self.level = parent_item.level + 1
- else:
- self.level = 0
- def getChildItem(self, row):
- return self.child_items[row]
- def getParentItem(self):
- return self.parent_item
- def getRow(self):
- return self.row
- def childCount(self):
- if not self.query_done:
- self.Select()
- if not self.child_count:
- return -1
- return self.child_count
- def hasChildren(self):
- if not self.query_done:
- return True
- return self.child_count > 0
- def getData(self, column):
- return self.data[column]
- # Context-sensitive call graph data model level 2+ item base
- class CallGraphLevelTwoPlusItemBase(CallGraphLevelItemBase):
- def __init__(self, glb, params, row, comm_id, thread_id, call_path_id, time, insn_cnt, cyc_cnt, branch_count, parent_item):
- super(CallGraphLevelTwoPlusItemBase, self).__init__(glb, params, row, parent_item)
- self.comm_id = comm_id
- self.thread_id = thread_id
- self.call_path_id = call_path_id
- self.insn_cnt = insn_cnt
- self.cyc_cnt = cyc_cnt
- self.branch_count = branch_count
- self.time = time
- def Select(self):
- self.query_done = True
- query = QSqlQuery(self.glb.db)
- if self.params.have_ipc:
- ipc_str = ", SUM(insn_count), SUM(cyc_count)"
- else:
- ipc_str = ""
- QueryExec(query, "SELECT call_path_id, name, short_name, COUNT(calls.id), SUM(return_time - call_time)" + ipc_str + ", SUM(branch_count)"
- " FROM calls"
- " INNER JOIN call_paths ON calls.call_path_id = call_paths.id"
- " INNER JOIN symbols ON call_paths.symbol_id = symbols.id"
- " INNER JOIN dsos ON symbols.dso_id = dsos.id"
- " WHERE parent_call_path_id = " + str(self.call_path_id) +
- " AND comm_id = " + str(self.comm_id) +
- " AND thread_id = " + str(self.thread_id) +
- " GROUP BY call_path_id, name, short_name"
- " ORDER BY call_path_id")
- while query.next():
- if self.params.have_ipc:
- insn_cnt = int(query.value(5))
- cyc_cnt = int(query.value(6))
- branch_count = int(query.value(7))
- else:
- insn_cnt = 0
- cyc_cnt = 0
- branch_count = int(query.value(5))
- child_item = CallGraphLevelThreeItem(self.glb, self.params, self.child_count, self.comm_id, self.thread_id, query.value(0), query.value(1), query.value(2), query.value(3), int(query.value(4)), insn_cnt, cyc_cnt, branch_count, self)
- self.child_items.append(child_item)
- self.child_count += 1
- # Context-sensitive call graph data model level three item
- class CallGraphLevelThreeItem(CallGraphLevelTwoPlusItemBase):
- def __init__(self, glb, params, row, comm_id, thread_id, call_path_id, name, dso, count, time, insn_cnt, cyc_cnt, branch_count, parent_item):
- super(CallGraphLevelThreeItem, self).__init__(glb, params, row, comm_id, thread_id, call_path_id, time, insn_cnt, cyc_cnt, branch_count, parent_item)
- dso = dsoname(dso)
- if self.params.have_ipc:
- insn_pcnt = PercentToOneDP(insn_cnt, parent_item.insn_cnt)
- cyc_pcnt = PercentToOneDP(cyc_cnt, parent_item.cyc_cnt)
- br_pcnt = PercentToOneDP(branch_count, parent_item.branch_count)
- ipc = CalcIPC(cyc_cnt, insn_cnt)
- self.data = [ name, dso, str(count), str(time), PercentToOneDP(time, parent_item.time), str(insn_cnt), insn_pcnt, str(cyc_cnt), cyc_pcnt, ipc, str(branch_count), br_pcnt ]
- else:
- self.data = [ name, dso, str(count), str(time), PercentToOneDP(time, parent_item.time), str(branch_count), PercentToOneDP(branch_count, parent_item.branch_count) ]
- self.dbid = call_path_id
- # Context-sensitive call graph data model level two item
- class CallGraphLevelTwoItem(CallGraphLevelTwoPlusItemBase):
- def __init__(self, glb, params, row, comm_id, thread_id, pid, tid, parent_item):
- super(CallGraphLevelTwoItem, self).__init__(glb, params, row, comm_id, thread_id, 1, 0, 0, 0, 0, parent_item)
- if self.params.have_ipc:
- self.data = [str(pid) + ":" + str(tid), "", "", "", "", "", "", "", "", "", "", ""]
- else:
- self.data = [str(pid) + ":" + str(tid), "", "", "", "", "", ""]
- self.dbid = thread_id
- def Select(self):
- super(CallGraphLevelTwoItem, self).Select()
- for child_item in self.child_items:
- self.time += child_item.time
- self.insn_cnt += child_item.insn_cnt
- self.cyc_cnt += child_item.cyc_cnt
- self.branch_count += child_item.branch_count
- for child_item in self.child_items:
- child_item.data[4] = PercentToOneDP(child_item.time, self.time)
- if self.params.have_ipc:
- child_item.data[6] = PercentToOneDP(child_item.insn_cnt, self.insn_cnt)
- child_item.data[8] = PercentToOneDP(child_item.cyc_cnt, self.cyc_cnt)
- child_item.data[11] = PercentToOneDP(child_item.branch_count, self.branch_count)
- else:
- child_item.data[6] = PercentToOneDP(child_item.branch_count, self.branch_count)
- # Context-sensitive call graph data model level one item
- class CallGraphLevelOneItem(CallGraphLevelItemBase):
- def __init__(self, glb, params, row, comm_id, comm, parent_item):
- super(CallGraphLevelOneItem, self).__init__(glb, params, row, parent_item)
- if self.params.have_ipc:
- self.data = [comm, "", "", "", "", "", "", "", "", "", "", ""]
- else:
- self.data = [comm, "", "", "", "", "", ""]
- self.dbid = comm_id
- def Select(self):
- self.query_done = True
- query = QSqlQuery(self.glb.db)
- QueryExec(query, "SELECT thread_id, pid, tid"
- " FROM comm_threads"
- " INNER JOIN threads ON thread_id = threads.id"
- " WHERE comm_id = " + str(self.dbid))
- while query.next():
- child_item = CallGraphLevelTwoItem(self.glb, self.params, self.child_count, self.dbid, query.value(0), query.value(1), query.value(2), self)
- self.child_items.append(child_item)
- self.child_count += 1
- # Context-sensitive call graph data model root item
- class CallGraphRootItem(CallGraphLevelItemBase):
- def __init__(self, glb, params):
- super(CallGraphRootItem, self).__init__(glb, params, 0, None)
- self.dbid = 0
- self.query_done = True
- if_has_calls = ""
- if IsSelectable(glb.db, "comms", columns = "has_calls"):
- if_has_calls = " WHERE has_calls = " + glb.dbref.TRUE
- query = QSqlQuery(glb.db)
- QueryExec(query, "SELECT id, comm FROM comms" + if_has_calls)
- while query.next():
- if not query.value(0):
- continue
- child_item = CallGraphLevelOneItem(glb, params, self.child_count, query.value(0), query.value(1), self)
- self.child_items.append(child_item)
- self.child_count += 1
- # Call graph model parameters
- class CallGraphModelParams():
- def __init__(self, glb, parent=None):
- self.have_ipc = IsSelectable(glb.db, "calls", columns = "insn_count, cyc_count")
- # Context-sensitive call graph data model base
- class CallGraphModelBase(TreeModel):
- def __init__(self, glb, parent=None):
- super(CallGraphModelBase, self).__init__(glb, CallGraphModelParams(glb), parent)
- def FindSelect(self, value, pattern, query):
- if pattern:
- # postgresql and sqlite pattern patching differences:
- # postgresql LIKE is case sensitive but sqlite LIKE is not
- # postgresql LIKE allows % and _ to be escaped with \ but sqlite LIKE does not
- # postgresql supports ILIKE which is case insensitive
- # sqlite supports GLOB (text only) which uses * and ? and is case sensitive
- if not self.glb.dbref.is_sqlite3:
- # Escape % and _
- s = value.replace("%", "\\%")
- s = s.replace("_", "\\_")
- # Translate * and ? into SQL LIKE pattern characters % and _
- if sys.version_info[0] == 3:
- trans = str.maketrans("*?", "%_")
- else:
- trans = string.maketrans("*?", "%_")
- match = " LIKE '" + str(s).translate(trans) + "'"
- else:
- match = " GLOB '" + str(value) + "'"
- else:
- match = " = '" + str(value) + "'"
- self.DoFindSelect(query, match)
- def Found(self, query, found):
- if found:
- return self.FindPath(query)
- return []
- def FindValue(self, value, pattern, query, last_value, last_pattern):
- if last_value == value and pattern == last_pattern:
- found = query.first()
- else:
- self.FindSelect(value, pattern, query)
- found = query.next()
- return self.Found(query, found)
- def FindNext(self, query):
- found = query.next()
- if not found:
- found = query.first()
- return self.Found(query, found)
- def FindPrev(self, query):
- found = query.previous()
- if not found:
- found = query.last()
- return self.Found(query, found)
- def FindThread(self, c):
- if c.direction == 0 or c.value != c.last_value or c.pattern != c.last_pattern:
- ids = self.FindValue(c.value, c.pattern, c.query, c.last_value, c.last_pattern)
- elif c.direction > 0:
- ids = self.FindNext(c.query)
- else:
- ids = self.FindPrev(c.query)
- return (True, ids)
- def Find(self, value, direction, pattern, context, callback):
- class Context():
- def __init__(self, *x):
- self.value, self.direction, self.pattern, self.query, self.last_value, self.last_pattern = x
- def Update(self, *x):
- self.value, self.direction, self.pattern, self.last_value, self.last_pattern = x + (self.value, self.pattern)
- if len(context):
- context[0].Update(value, direction, pattern)
- else:
- context.append(Context(value, direction, pattern, QSqlQuery(self.glb.db), None, None))
- # Use a thread so the UI is not blocked during the SELECT
- thread = Thread(self.FindThread, context[0])
- thread.done.connect(lambda ids, t=thread, c=callback: self.FindDone(t, c, ids), Qt.QueuedConnection)
- thread.start()
- def FindDone(self, thread, callback, ids):
- callback(ids)
- # Context-sensitive call graph data model
- class CallGraphModel(CallGraphModelBase):
- def __init__(self, glb, parent=None):
- super(CallGraphModel, self).__init__(glb, parent)
- def GetRoot(self):
- return CallGraphRootItem(self.glb, self.params)
- def columnCount(self, parent=None):
- if self.params.have_ipc:
- return 12
- else:
- return 7
- def columnHeader(self, column):
- if self.params.have_ipc:
- headers = ["Call Path", "Object", "Count ", "Time (ns) ", "Time (%) ", "Insn Cnt", "Insn Cnt (%)", "Cyc Cnt", "Cyc Cnt (%)", "IPC", "Branch Count ", "Branch Count (%) "]
- else:
- headers = ["Call Path", "Object", "Count ", "Time (ns) ", "Time (%) ", "Branch Count ", "Branch Count (%) "]
- return headers[column]
- def columnAlignment(self, column):
- if self.params.have_ipc:
- alignment = [ Qt.AlignLeft, Qt.AlignLeft, Qt.AlignRight, Qt.AlignRight, Qt.AlignRight, Qt.AlignRight, Qt.AlignRight, Qt.AlignRight, Qt.AlignRight, Qt.AlignRight, Qt.AlignRight, Qt.AlignRight ]
- else:
- alignment = [ Qt.AlignLeft, Qt.AlignLeft, Qt.AlignRight, Qt.AlignRight, Qt.AlignRight, Qt.AlignRight, Qt.AlignRight ]
- return alignment[column]
- def DoFindSelect(self, query, match):
- QueryExec(query, "SELECT call_path_id, comm_id, thread_id"
- " FROM calls"
- " INNER JOIN call_paths ON calls.call_path_id = call_paths.id"
- " INNER JOIN symbols ON call_paths.symbol_id = symbols.id"
- " WHERE calls.id <> 0"
- " AND symbols.name" + match +
- " GROUP BY comm_id, thread_id, call_path_id"
- " ORDER BY comm_id, thread_id, call_path_id")
- def FindPath(self, query):
- # Turn the query result into a list of ids that the tree view can walk
- # to open the tree at the right place.
- ids = []
- parent_id = query.value(0)
- while parent_id:
- ids.insert(0, parent_id)
- q2 = QSqlQuery(self.glb.db)
- QueryExec(q2, "SELECT parent_id"
- " FROM call_paths"
- " WHERE id = " + str(parent_id))
- if not q2.next():
- break
- parent_id = q2.value(0)
- # The call path root is not used
- if ids[0] == 1:
- del ids[0]
- ids.insert(0, query.value(2))
- ids.insert(0, query.value(1))
- return ids
- # Call tree data model level 2+ item base
- class CallTreeLevelTwoPlusItemBase(CallGraphLevelItemBase):
- def __init__(self, glb, params, row, comm_id, thread_id, calls_id, call_time, time, insn_cnt, cyc_cnt, branch_count, parent_item):
- super(CallTreeLevelTwoPlusItemBase, self).__init__(glb, params, row, parent_item)
- self.comm_id = comm_id
- self.thread_id = thread_id
- self.calls_id = calls_id
- self.call_time = call_time
- self.time = time
- self.insn_cnt = insn_cnt
- self.cyc_cnt = cyc_cnt
- self.branch_count = branch_count
- def Select(self):
- self.query_done = True
- if self.calls_id == 0:
- comm_thread = " AND comm_id = " + str(self.comm_id) + " AND thread_id = " + str(self.thread_id)
- else:
- comm_thread = ""
- if self.params.have_ipc:
- ipc_str = ", insn_count, cyc_count"
- else:
- ipc_str = ""
- query = QSqlQuery(self.glb.db)
- QueryExec(query, "SELECT calls.id, name, short_name, call_time, return_time - call_time" + ipc_str + ", branch_count"
- " FROM calls"
- " INNER JOIN call_paths ON calls.call_path_id = call_paths.id"
- " INNER JOIN symbols ON call_paths.symbol_id = symbols.id"
- " INNER JOIN dsos ON symbols.dso_id = dsos.id"
- " WHERE calls.parent_id = " + str(self.calls_id) + comm_thread +
- " ORDER BY call_time, calls.id")
- while query.next():
- if self.params.have_ipc:
- insn_cnt = int(query.value(5))
- cyc_cnt = int(query.value(6))
- branch_count = int(query.value(7))
- else:
- insn_cnt = 0
- cyc_cnt = 0
- branch_count = int(query.value(5))
- child_item = CallTreeLevelThreeItem(self.glb, self.params, self.child_count, self.comm_id, self.thread_id, query.value(0), query.value(1), query.value(2), query.value(3), int(query.value(4)), insn_cnt, cyc_cnt, branch_count, self)
- self.child_items.append(child_item)
- self.child_count += 1
- # Call tree data model level three item
- class CallTreeLevelThreeItem(CallTreeLevelTwoPlusItemBase):
- def __init__(self, glb, params, row, comm_id, thread_id, calls_id, name, dso, call_time, time, insn_cnt, cyc_cnt, branch_count, parent_item):
- super(CallTreeLevelThreeItem, self).__init__(glb, params, row, comm_id, thread_id, calls_id, call_time, time, insn_cnt, cyc_cnt, branch_count, parent_item)
- dso = dsoname(dso)
- if self.params.have_ipc:
- insn_pcnt = PercentToOneDP(insn_cnt, parent_item.insn_cnt)
- cyc_pcnt = PercentToOneDP(cyc_cnt, parent_item.cyc_cnt)
- br_pcnt = PercentToOneDP(branch_count, parent_item.branch_count)
- ipc = CalcIPC(cyc_cnt, insn_cnt)
- self.data = [ name, dso, str(call_time), str(time), PercentToOneDP(time, parent_item.time), str(insn_cnt), insn_pcnt, str(cyc_cnt), cyc_pcnt, ipc, str(branch_count), br_pcnt ]
- else:
- self.data = [ name, dso, str(call_time), str(time), PercentToOneDP(time, parent_item.time), str(branch_count), PercentToOneDP(branch_count, parent_item.branch_count) ]
- self.dbid = calls_id
- # Call tree data model level two item
- class CallTreeLevelTwoItem(CallTreeLevelTwoPlusItemBase):
- def __init__(self, glb, params, row, comm_id, thread_id, pid, tid, parent_item):
- super(CallTreeLevelTwoItem, self).__init__(glb, params, row, comm_id, thread_id, 0, 0, 0, 0, 0, 0, parent_item)
- if self.params.have_ipc:
- self.data = [str(pid) + ":" + str(tid), "", "", "", "", "", "", "", "", "", "", ""]
- else:
- self.data = [str(pid) + ":" + str(tid), "", "", "", "", "", ""]
- self.dbid = thread_id
- def Select(self):
- super(CallTreeLevelTwoItem, self).Select()
- for child_item in self.child_items:
- self.time += child_item.time
- self.insn_cnt += child_item.insn_cnt
- self.cyc_cnt += child_item.cyc_cnt
- self.branch_count += child_item.branch_count
- for child_item in self.child_items:
- child_item.data[4] = PercentToOneDP(child_item.time, self.time)
- if self.params.have_ipc:
- child_item.data[6] = PercentToOneDP(child_item.insn_cnt, self.insn_cnt)
- child_item.data[8] = PercentToOneDP(child_item.cyc_cnt, self.cyc_cnt)
- child_item.data[11] = PercentToOneDP(child_item.branch_count, self.branch_count)
- else:
- child_item.data[6] = PercentToOneDP(child_item.branch_count, self.branch_count)
- # Call tree data model level one item
- class CallTreeLevelOneItem(CallGraphLevelItemBase):
- def __init__(self, glb, params, row, comm_id, comm, parent_item):
- super(CallTreeLevelOneItem, self).__init__(glb, params, row, parent_item)
- if self.params.have_ipc:
- self.data = [comm, "", "", "", "", "", "", "", "", "", "", ""]
- else:
- self.data = [comm, "", "", "", "", "", ""]
- self.dbid = comm_id
- def Select(self):
- self.query_done = True
- query = QSqlQuery(self.glb.db)
- QueryExec(query, "SELECT thread_id, pid, tid"
- " FROM comm_threads"
- " INNER JOIN threads ON thread_id = threads.id"
- " WHERE comm_id = " + str(self.dbid))
- while query.next():
- child_item = CallTreeLevelTwoItem(self.glb, self.params, self.child_count, self.dbid, query.value(0), query.value(1), query.value(2), self)
- self.child_items.append(child_item)
- self.child_count += 1
- # Call tree data model root item
- class CallTreeRootItem(CallGraphLevelItemBase):
- def __init__(self, glb, params):
- super(CallTreeRootItem, self).__init__(glb, params, 0, None)
- self.dbid = 0
- self.query_done = True
- if_has_calls = ""
- if IsSelectable(glb.db, "comms", columns = "has_calls"):
- if_has_calls = " WHERE has_calls = " + glb.dbref.TRUE
- query = QSqlQuery(glb.db)
- QueryExec(query, "SELECT id, comm FROM comms" + if_has_calls)
- while query.next():
- if not query.value(0):
- continue
- child_item = CallTreeLevelOneItem(glb, params, self.child_count, query.value(0), query.value(1), self)
- self.child_items.append(child_item)
- self.child_count += 1
- # Call Tree data model
- class CallTreeModel(CallGraphModelBase):
- def __init__(self, glb, parent=None):
- super(CallTreeModel, self).__init__(glb, parent)
- def GetRoot(self):
- return CallTreeRootItem(self.glb, self.params)
- def columnCount(self, parent=None):
- if self.params.have_ipc:
- return 12
- else:
- return 7
- def columnHeader(self, column):
- if self.params.have_ipc:
- headers = ["Call Path", "Object", "Call Time", "Time (ns) ", "Time (%) ", "Insn Cnt", "Insn Cnt (%)", "Cyc Cnt", "Cyc Cnt (%)", "IPC", "Branch Count ", "Branch Count (%) "]
- else:
- headers = ["Call Path", "Object", "Call Time", "Time (ns) ", "Time (%) ", "Branch Count ", "Branch Count (%) "]
- return headers[column]
- def columnAlignment(self, column):
- if self.params.have_ipc:
- alignment = [ Qt.AlignLeft, Qt.AlignLeft, Qt.AlignRight, Qt.AlignRight, Qt.AlignRight, Qt.AlignRight, Qt.AlignRight, Qt.AlignRight, Qt.AlignRight, Qt.AlignRight, Qt.AlignRight, Qt.AlignRight ]
- else:
- alignment = [ Qt.AlignLeft, Qt.AlignLeft, Qt.AlignRight, Qt.AlignRight, Qt.AlignRight, Qt.AlignRight, Qt.AlignRight ]
- return alignment[column]
- def DoFindSelect(self, query, match):
- QueryExec(query, "SELECT calls.id, comm_id, thread_id"
- " FROM calls"
- " INNER JOIN call_paths ON calls.call_path_id = call_paths.id"
- " INNER JOIN symbols ON call_paths.symbol_id = symbols.id"
- " WHERE calls.id <> 0"
- " AND symbols.name" + match +
- " ORDER BY comm_id, thread_id, call_time, calls.id")
- def FindPath(self, query):
- # Turn the query result into a list of ids that the tree view can walk
- # to open the tree at the right place.
- ids = []
- parent_id = query.value(0)
- while parent_id:
- ids.insert(0, parent_id)
- q2 = QSqlQuery(self.glb.db)
- QueryExec(q2, "SELECT parent_id"
- " FROM calls"
- " WHERE id = " + str(parent_id))
- if not q2.next():
- break
- parent_id = q2.value(0)
- ids.insert(0, query.value(2))
- ids.insert(0, query.value(1))
- return ids
- # Vertical layout
- class HBoxLayout(QHBoxLayout):
- def __init__(self, *children):
- super(HBoxLayout, self).__init__()
- self.layout().setContentsMargins(0, 0, 0, 0)
- for child in children:
- if child.isWidgetType():
- self.layout().addWidget(child)
- else:
- self.layout().addLayout(child)
- # Horizontal layout
- class VBoxLayout(QVBoxLayout):
- def __init__(self, *children):
- super(VBoxLayout, self).__init__()
- self.layout().setContentsMargins(0, 0, 0, 0)
- for child in children:
- if child.isWidgetType():
- self.layout().addWidget(child)
- else:
- self.layout().addLayout(child)
- # Vertical layout widget
- class VBox():
- def __init__(self, *children):
- self.vbox = QWidget()
- self.vbox.setLayout(VBoxLayout(*children))
- def Widget(self):
- return self.vbox
- # Tree window base
- class TreeWindowBase(QMdiSubWindow):
- def __init__(self, parent=None):
- super(TreeWindowBase, self).__init__(parent)
- self.model = None
- self.find_bar = None
- self.view = QTreeView()
- self.view.setSelectionMode(QAbstractItemView.ContiguousSelection)
- self.view.CopyCellsToClipboard = CopyTreeCellsToClipboard
- self.context_menu = TreeContextMenu(self.view)
- def DisplayFound(self, ids):
- if not len(ids):
- return False
- parent = QModelIndex()
- for dbid in ids:
- found = False
- n = self.model.rowCount(parent)
- for row in xrange(n):
- child = self.model.index(row, 0, parent)
- if child.internalPointer().dbid == dbid:
- found = True
- self.view.setExpanded(parent, True)
- self.view.setCurrentIndex(child)
- parent = child
- break
- if not found:
- break
- return found
- def Find(self, value, direction, pattern, context):
- self.view.setFocus()
- self.find_bar.Busy()
- self.model.Find(value, direction, pattern, context, self.FindDone)
- def FindDone(self, ids):
- found = True
- if not self.DisplayFound(ids):
- found = False
- self.find_bar.Idle()
- if not found:
- self.find_bar.NotFound()
- # Context-sensitive call graph window
- class CallGraphWindow(TreeWindowBase):
- def __init__(self, glb, parent=None):
- super(CallGraphWindow, self).__init__(parent)
- self.model = LookupCreateModel("Context-Sensitive Call Graph", lambda x=glb: CallGraphModel(x))
- self.view.setModel(self.model)
- for c, w in ((0, 250), (1, 100), (2, 60), (3, 70), (4, 70), (5, 100)):
- self.view.setColumnWidth(c, w)
- self.find_bar = FindBar(self, self)
- self.vbox = VBox(self.view, self.find_bar.Widget())
- self.setWidget(self.vbox.Widget())
- AddSubWindow(glb.mainwindow.mdi_area, self, "Context-Sensitive Call Graph")
- # Call tree window
- class CallTreeWindow(TreeWindowBase):
- def __init__(self, glb, parent=None, thread_at_time=None):
- super(CallTreeWindow, self).__init__(parent)
- self.model = LookupCreateModel("Call Tree", lambda x=glb: CallTreeModel(x))
- self.view.setModel(self.model)
- for c, w in ((0, 230), (1, 100), (2, 100), (3, 70), (4, 70), (5, 100)):
- self.view.setColumnWidth(c, w)
- self.find_bar = FindBar(self, self)
- self.vbox = VBox(self.view, self.find_bar.Widget())
- self.setWidget(self.vbox.Widget())
- AddSubWindow(glb.mainwindow.mdi_area, self, "Call Tree")
- if thread_at_time:
- self.DisplayThreadAtTime(*thread_at_time)
- def DisplayThreadAtTime(self, comm_id, thread_id, time):
- parent = QModelIndex()
- for dbid in (comm_id, thread_id):
- found = False
- n = self.model.rowCount(parent)
- for row in xrange(n):
- child = self.model.index(row, 0, parent)
- if child.internalPointer().dbid == dbid:
- found = True
- self.view.setExpanded(parent, True)
- self.view.setCurrentIndex(child)
- parent = child
- break
- if not found:
- return
- found = False
- while True:
- n = self.model.rowCount(parent)
- if not n:
- return
- last_child = None
- for row in xrange(n):
- self.view.setExpanded(parent, True)
- child = self.model.index(row, 0, parent)
- child_call_time = child.internalPointer().call_time
- if child_call_time < time:
- last_child = child
- elif child_call_time == time:
- self.view.setCurrentIndex(child)
- return
- elif child_call_time > time:
- break
- if not last_child:
- if not found:
- child = self.model.index(0, 0, parent)
- self.view.setExpanded(parent, True)
- self.view.setCurrentIndex(child)
- return
- found = True
- self.view.setExpanded(parent, True)
- self.view.setCurrentIndex(last_child)
- parent = last_child
- # ExecComm() gets the comm_id of the command string that was set when the process exec'd i.e. the program name
- def ExecComm(db, thread_id, time):
- query = QSqlQuery(db)
- QueryExec(query, "SELECT comm_threads.comm_id, comms.c_time, comms.exec_flag"
- " FROM comm_threads"
- " INNER JOIN comms ON comms.id = comm_threads.comm_id"
- " WHERE comm_threads.thread_id = " + str(thread_id) +
- " ORDER BY comms.c_time, comms.id")
- first = None
- last = None
- while query.next():
- if first is None:
- first = query.value(0)
- if query.value(2) and Decimal(query.value(1)) <= Decimal(time):
- last = query.value(0)
- if not(last is None):
- return last
- return first
- # Container for (x, y) data
- class XY():
- def __init__(self, x=0, y=0):
- self.x = x
- self.y = y
- def __str__(self):
- return "XY({}, {})".format(str(self.x), str(self.y))
- # Container for sub-range data
- class Subrange():
- def __init__(self, lo=0, hi=0):
- self.lo = lo
- self.hi = hi
- def __str__(self):
- return "Subrange({}, {})".format(str(self.lo), str(self.hi))
- # Graph data region base class
- class GraphDataRegion(object):
- def __init__(self, key, title = "", ordinal = ""):
- self.key = key
- self.title = title
- self.ordinal = ordinal
- # Function to sort GraphDataRegion
- def GraphDataRegionOrdinal(data_region):
- return data_region.ordinal
- # Attributes for a graph region
- class GraphRegionAttribute():
- def __init__(self, colour):
- self.colour = colour
- # Switch graph data region represents a task
- class SwitchGraphDataRegion(GraphDataRegion):
- def __init__(self, key, exec_comm_id, pid, tid, comm, thread_id, comm_id):
- super(SwitchGraphDataRegion, self).__init__(key)
- self.title = str(pid) + " / " + str(tid) + " " + comm
- # Order graph legend within exec comm by pid / tid / time
- self.ordinal = str(pid).rjust(16) + str(exec_comm_id).rjust(8) + str(tid).rjust(16)
- self.exec_comm_id = exec_comm_id
- self.pid = pid
- self.tid = tid
- self.comm = comm
- self.thread_id = thread_id
- self.comm_id = comm_id
- # Graph data point
- class GraphDataPoint():
- def __init__(self, data, index, x, y, altx=None, alty=None, hregion=None, vregion=None):
- self.data = data
- self.index = index
- self.x = x
- self.y = y
- self.altx = altx
- self.alty = alty
- self.hregion = hregion
- self.vregion = vregion
- # Graph data (single graph) base class
- class GraphData(object):
- def __init__(self, collection, xbase=Decimal(0), ybase=Decimal(0)):
- self.collection = collection
- self.points = []
- self.xbase = xbase
- self.ybase = ybase
- self.title = ""
- def AddPoint(self, x, y, altx=None, alty=None, hregion=None, vregion=None):
- index = len(self.points)
- x = float(Decimal(x) - self.xbase)
- y = float(Decimal(y) - self.ybase)
- self.points.append(GraphDataPoint(self, index, x, y, altx, alty, hregion, vregion))
- def XToData(self, x):
- return Decimal(x) + self.xbase
- def YToData(self, y):
- return Decimal(y) + self.ybase
- # Switch graph data (for one CPU)
- class SwitchGraphData(GraphData):
- def __init__(self, db, collection, cpu, xbase):
- super(SwitchGraphData, self).__init__(collection, xbase)
- self.cpu = cpu
- self.title = "CPU " + str(cpu)
- self.SelectSwitches(db)
- def SelectComms(self, db, thread_id, last_comm_id, start_time, end_time):
- query = QSqlQuery(db)
- QueryExec(query, "SELECT id, c_time"
- " FROM comms"
- " WHERE c_thread_id = " + str(thread_id) +
- " AND exec_flag = " + self.collection.glb.dbref.TRUE +
- " AND c_time >= " + str(start_time) +
- " AND c_time <= " + str(end_time) +
- " ORDER BY c_time, id")
- while query.next():
- comm_id = query.value(0)
- if comm_id == last_comm_id:
- continue
- time = query.value(1)
- hregion = self.HRegion(db, thread_id, comm_id, time)
- self.AddPoint(time, 1000, None, None, hregion)
- def SelectSwitches(self, db):
- last_time = None
- last_comm_id = None
- last_thread_id = None
- query = QSqlQuery(db)
- QueryExec(query, "SELECT time, thread_out_id, thread_in_id, comm_out_id, comm_in_id, flags"
- " FROM context_switches"
- " WHERE machine_id = " + str(self.collection.machine_id) +
- " AND cpu = " + str(self.cpu) +
- " ORDER BY time, id")
- while query.next():
- flags = int(query.value(5))
- if flags & 1:
- # Schedule-out: detect and add exec's
- if last_thread_id == query.value(1) and last_comm_id is not None and last_comm_id != query.value(3):
- self.SelectComms(db, last_thread_id, last_comm_id, last_time, query.value(0))
- continue
- # Schedule-in: add data point
- if len(self.points) == 0:
- start_time = self.collection.glb.StartTime(self.collection.machine_id)
- hregion = self.HRegion(db, query.value(1), query.value(3), start_time)
- self.AddPoint(start_time, 1000, None, None, hregion)
- time = query.value(0)
- comm_id = query.value(4)
- thread_id = query.value(2)
- hregion = self.HRegion(db, thread_id, comm_id, time)
- self.AddPoint(time, 1000, None, None, hregion)
- last_time = time
- last_comm_id = comm_id
- last_thread_id = thread_id
- def NewHRegion(self, db, key, thread_id, comm_id, time):
- exec_comm_id = ExecComm(db, thread_id, time)
- query = QSqlQuery(db)
- QueryExec(query, "SELECT pid, tid FROM threads WHERE id = " + str(thread_id))
- if query.next():
- pid = query.value(0)
- tid = query.value(1)
- else:
- pid = -1
- tid = -1
- query = QSqlQuery(db)
- QueryExec(query, "SELECT comm FROM comms WHERE id = " + str(comm_id))
- if query.next():
- comm = query.value(0)
- else:
- comm = ""
- return SwitchGraphDataRegion(key, exec_comm_id, pid, tid, comm, thread_id, comm_id)
- def HRegion(self, db, thread_id, comm_id, time):
- key = str(thread_id) + ":" + str(comm_id)
- hregion = self.collection.LookupHRegion(key)
- if hregion is None:
- hregion = self.NewHRegion(db, key, thread_id, comm_id, time)
- self.collection.AddHRegion(key, hregion)
- return hregion
- # Graph data collection (multiple related graphs) base class
- class GraphDataCollection(object):
- def __init__(self, glb):
- self.glb = glb
- self.data = []
- self.hregions = {}
- self.xrangelo = None
- self.xrangehi = None
- self.yrangelo = None
- self.yrangehi = None
- self.dp = XY(0, 0)
- def AddGraphData(self, data):
- self.data.append(data)
- def LookupHRegion(self, key):
- if key in self.hregions:
- return self.hregions[key]
- return None
- def AddHRegion(self, key, hregion):
- self.hregions[key] = hregion
- # Switch graph data collection (SwitchGraphData for each CPU)
- class SwitchGraphDataCollection(GraphDataCollection):
- def __init__(self, glb, db, machine_id):
- super(SwitchGraphDataCollection, self).__init__(glb)
- self.machine_id = machine_id
- self.cpus = self.SelectCPUs(db)
- self.xrangelo = glb.StartTime(machine_id)
- self.xrangehi = glb.FinishTime(machine_id)
- self.yrangelo = Decimal(0)
- self.yrangehi = Decimal(1000)
- for cpu in self.cpus:
- self.AddGraphData(SwitchGraphData(db, self, cpu, self.xrangelo))
- def SelectCPUs(self, db):
- cpus = []
- query = QSqlQuery(db)
- QueryExec(query, "SELECT DISTINCT cpu"
- " FROM context_switches"
- " WHERE machine_id = " + str(self.machine_id))
- while query.next():
- cpus.append(int(query.value(0)))
- return sorted(cpus)
- # Switch graph data graphics item displays the graphed data
- class SwitchGraphDataGraphicsItem(QGraphicsItem):
- def __init__(self, data, graph_width, graph_height, attrs, event_handler, parent=None):
- super(SwitchGraphDataGraphicsItem, self).__init__(parent)
- self.data = data
- self.graph_width = graph_width
- self.graph_height = graph_height
- self.attrs = attrs
- self.event_handler = event_handler
- self.setAcceptHoverEvents(True)
- def boundingRect(self):
- return QRectF(0, 0, self.graph_width, self.graph_height)
- def PaintPoint(self, painter, last, x):
- if not(last is None or last.hregion.pid == 0 or x < self.attrs.subrange.x.lo):
- if last.x < self.attrs.subrange.x.lo:
- x0 = self.attrs.subrange.x.lo
- else:
- x0 = last.x
- if x > self.attrs.subrange.x.hi:
- x1 = self.attrs.subrange.x.hi
- else:
- x1 = x - 1
- x0 = self.attrs.XToPixel(x0)
- x1 = self.attrs.XToPixel(x1)
- y0 = self.attrs.YToPixel(last.y)
- colour = self.attrs.region_attributes[last.hregion.key].colour
- width = x1 - x0 + 1
- if width < 2:
- painter.setPen(colour)
- painter.drawLine(x0, self.graph_height - y0, x0, self.graph_height)
- else:
- painter.fillRect(x0, self.graph_height - y0, width, self.graph_height - 1, colour)
- def paint(self, painter, option, widget):
- last = None
- for point in self.data.points:
- self.PaintPoint(painter, last, point.x)
- if point.x > self.attrs.subrange.x.hi:
- break;
- last = point
- self.PaintPoint(painter, last, self.attrs.subrange.x.hi + 1)
- def BinarySearchPoint(self, target):
- lower_pos = 0
- higher_pos = len(self.data.points)
- while True:
- pos = int((lower_pos + higher_pos) / 2)
- val = self.data.points[pos].x
- if target >= val:
- lower_pos = pos
- else:
- higher_pos = pos
- if higher_pos <= lower_pos + 1:
- return lower_pos
- def XPixelToData(self, x):
- x = self.attrs.PixelToX(x)
- if x < self.data.points[0].x:
- x = 0
- pos = 0
- low = True
- else:
- pos = self.BinarySearchPoint(x)
- low = False
- return (low, pos, self.data.XToData(x))
- def EventToData(self, event):
- no_data = (None,) * 4
- if len(self.data.points) < 1:
- return no_data
- x = event.pos().x()
- if x < 0:
- return no_data
- low0, pos0, time_from = self.XPixelToData(x)
- low1, pos1, time_to = self.XPixelToData(x + 1)
- hregions = set()
- hregion_times = []
- if not low1:
- for i in xrange(pos0, pos1 + 1):
- hregion = self.data.points[i].hregion
- hregions.add(hregion)
- if i == pos0:
- time = time_from
- else:
- time = self.data.XToData(self.data.points[i].x)
- hregion_times.append((hregion, time))
- return (time_from, time_to, hregions, hregion_times)
- def hoverMoveEvent(self, event):
- time_from, time_to, hregions, hregion_times = self.EventToData(event)
- if time_from is not None:
- self.event_handler.PointEvent(self.data.cpu, time_from, time_to, hregions)
- def hoverLeaveEvent(self, event):
- self.event_handler.NoPointEvent()
- def mousePressEvent(self, event):
- if event.button() != Qt.RightButton:
- super(SwitchGraphDataGraphicsItem, self).mousePressEvent(event)
- return
- time_from, time_to, hregions, hregion_times = self.EventToData(event)
- if hregion_times:
- self.event_handler.RightClickEvent(self.data.cpu, hregion_times, event.screenPos())
- # X-axis graphics item
- class XAxisGraphicsItem(QGraphicsItem):
- def __init__(self, width, parent=None):
- super(XAxisGraphicsItem, self).__init__(parent)
- self.width = width
- self.max_mark_sz = 4
- self.height = self.max_mark_sz + 1
- def boundingRect(self):
- return QRectF(0, 0, self.width, self.height)
- def Step(self):
- attrs = self.parentItem().attrs
- subrange = attrs.subrange.x
- t = subrange.hi - subrange.lo
- s = (3.0 * t) / self.width
- n = 1.0
- while s > n:
- n = n * 10.0
- return n
- def PaintMarks(self, painter, at_y, lo, hi, step, i):
- attrs = self.parentItem().attrs
- x = lo
- while x <= hi:
- xp = attrs.XToPixel(x)
- if i % 10:
- if i % 5:
- sz = 1
- else:
- sz = 2
- else:
- sz = self.max_mark_sz
- i = 0
- painter.drawLine(xp, at_y, xp, at_y + sz)
- x += step
- i += 1
- def paint(self, painter, option, widget):
- # Using QPainter::drawLine(int x1, int y1, int x2, int y2) so x2 = width -1
- painter.drawLine(0, 0, self.width - 1, 0)
- n = self.Step()
- attrs = self.parentItem().attrs
- subrange = attrs.subrange.x
- if subrange.lo:
- x_offset = n - (subrange.lo % n)
- else:
- x_offset = 0.0
- x = subrange.lo + x_offset
- i = (x / n) % 10
- self.PaintMarks(painter, 0, x, subrange.hi, n, i)
- def ScaleDimensions(self):
- n = self.Step()
- attrs = self.parentItem().attrs
- lo = attrs.subrange.x.lo
- hi = (n * 10.0) + lo
- width = attrs.XToPixel(hi)
- if width > 500:
- width = 0
- return (n, lo, hi, width)
- def PaintScale(self, painter, at_x, at_y):
- n, lo, hi, width = self.ScaleDimensions()
- if not width:
- return
- painter.drawLine(at_x, at_y, at_x + width, at_y)
- self.PaintMarks(painter, at_y, lo, hi, n, 0)
- def ScaleWidth(self):
- n, lo, hi, width = self.ScaleDimensions()
- return width
- def ScaleHeight(self):
- return self.height
- def ScaleUnit(self):
- return self.Step() * 10
- # Scale graphics item base class
- class ScaleGraphicsItem(QGraphicsItem):
- def __init__(self, axis, parent=None):
- super(ScaleGraphicsItem, self).__init__(parent)
- self.axis = axis
- def boundingRect(self):
- scale_width = self.axis.ScaleWidth()
- if not scale_width:
- return QRectF()
- return QRectF(0, 0, self.axis.ScaleWidth() + 100, self.axis.ScaleHeight())
- def paint(self, painter, option, widget):
- scale_width = self.axis.ScaleWidth()
- if not scale_width:
- return
- self.axis.PaintScale(painter, 0, 5)
- x = scale_width + 4
- painter.drawText(QPointF(x, 10), self.Text())
- def Unit(self):
- return self.axis.ScaleUnit()
- def Text(self):
- return ""
- # Switch graph scale graphics item
- class SwitchScaleGraphicsItem(ScaleGraphicsItem):
- def __init__(self, axis, parent=None):
- super(SwitchScaleGraphicsItem, self).__init__(axis, parent)
- def Text(self):
- unit = self.Unit()
- if unit >= 1000000000:
- unit = int(unit / 1000000000)
- us = "s"
- elif unit >= 1000000:
- unit = int(unit / 1000000)
- us = "ms"
- elif unit >= 1000:
- unit = int(unit / 1000)
- us = "us"
- else:
- unit = int(unit)
- us = "ns"
- return " = " + str(unit) + " " + us
- # Switch graph graphics item contains graph title, scale, x/y-axis, and the graphed data
- class SwitchGraphGraphicsItem(QGraphicsItem):
- def __init__(self, collection, data, attrs, event_handler, first, parent=None):
- super(SwitchGraphGraphicsItem, self).__init__(parent)
- self.collection = collection
- self.data = data
- self.attrs = attrs
- self.event_handler = event_handler
- margin = 20
- title_width = 50
- self.title_graphics = QGraphicsSimpleTextItem(data.title, self)
- self.title_graphics.setPos(margin, margin)
- graph_width = attrs.XToPixel(attrs.subrange.x.hi) + 1
- graph_height = attrs.YToPixel(attrs.subrange.y.hi) + 1
- self.graph_origin_x = margin + title_width + margin
- self.graph_origin_y = graph_height + margin
- x_axis_size = 1
- y_axis_size = 1
- self.yline = QGraphicsLineItem(0, 0, 0, graph_height, self)
- self.x_axis = XAxisGraphicsItem(graph_width, self)
- self.x_axis.setPos(self.graph_origin_x, self.graph_origin_y + 1)
- if first:
- self.scale_item = SwitchScaleGraphicsItem(self.x_axis, self)
- self.scale_item.setPos(self.graph_origin_x, self.graph_origin_y + 10)
- self.yline.setPos(self.graph_origin_x - y_axis_size, self.graph_origin_y - graph_height)
- self.axis_point = QGraphicsLineItem(0, 0, 0, 0, self)
- self.axis_point.setPos(self.graph_origin_x - 1, self.graph_origin_y +1)
- self.width = self.graph_origin_x + graph_width + margin
- self.height = self.graph_origin_y + margin
- self.graph = SwitchGraphDataGraphicsItem(data, graph_width, graph_height, attrs, event_handler, self)
- self.graph.setPos(self.graph_origin_x, self.graph_origin_y - graph_height)
- if parent and 'EnableRubberBand' in dir(parent):
- parent.EnableRubberBand(self.graph_origin_x, self.graph_origin_x + graph_width - 1, self)
- def boundingRect(self):
- return QRectF(0, 0, self.width, self.height)
- def paint(self, painter, option, widget):
- pass
- def RBXToPixel(self, x):
- return self.attrs.PixelToX(x - self.graph_origin_x)
- def RBXRangeToPixel(self, x0, x1):
- return (self.RBXToPixel(x0), self.RBXToPixel(x1 + 1))
- def RBPixelToTime(self, x):
- if x < self.data.points[0].x:
- return self.data.XToData(0)
- return self.data.XToData(x)
- def RBEventTimes(self, x0, x1):
- x0, x1 = self.RBXRangeToPixel(x0, x1)
- time_from = self.RBPixelToTime(x0)
- time_to = self.RBPixelToTime(x1)
- return (time_from, time_to)
- def RBEvent(self, x0, x1):
- time_from, time_to = self.RBEventTimes(x0, x1)
- self.event_handler.RangeEvent(time_from, time_to)
- def RBMoveEvent(self, x0, x1):
- if x1 < x0:
- x0, x1 = x1, x0
- self.RBEvent(x0, x1)
- def RBReleaseEvent(self, x0, x1, selection_state):
- if x1 < x0:
- x0, x1 = x1, x0
- x0, x1 = self.RBXRangeToPixel(x0, x1)
- self.event_handler.SelectEvent(x0, x1, selection_state)
- # Graphics item to draw a vertical bracket (used to highlight "forward" sub-range)
- class VerticalBracketGraphicsItem(QGraphicsItem):
- def __init__(self, parent=None):
- super(VerticalBracketGraphicsItem, self).__init__(parent)
- self.width = 0
- self.height = 0
- self.hide()
- def SetSize(self, width, height):
- self.width = width + 1
- self.height = height + 1
- def boundingRect(self):
- return QRectF(0, 0, self.width, self.height)
- def paint(self, painter, option, widget):
- colour = QColor(255, 255, 0, 32)
- painter.fillRect(0, 0, self.width, self.height, colour)
- x1 = self.width - 1
- y1 = self.height - 1
- painter.drawLine(0, 0, x1, 0)
- painter.drawLine(0, 0, 0, 3)
- painter.drawLine(x1, 0, x1, 3)
- painter.drawLine(0, y1, x1, y1)
- painter.drawLine(0, y1, 0, y1 - 3)
- painter.drawLine(x1, y1, x1, y1 - 3)
- # Graphics item to contain graphs arranged vertically
- class VertcalGraphSetGraphicsItem(QGraphicsItem):
- def __init__(self, collection, attrs, event_handler, child_class, parent=None):
- super(VertcalGraphSetGraphicsItem, self).__init__(parent)
- self.collection = collection
- self.top = 10
- self.width = 0
- self.height = self.top
- self.rubber_band = None
- self.rb_enabled = False
- first = True
- for data in collection.data:
- child = child_class(collection, data, attrs, event_handler, first, self)
- child.setPos(0, self.height + 1)
- rect = child.boundingRect()
- if rect.right() > self.width:
- self.width = rect.right()
- self.height = self.height + rect.bottom() + 1
- first = False
- self.bracket = VerticalBracketGraphicsItem(self)
- def EnableRubberBand(self, xlo, xhi, rb_event_handler):
- if self.rb_enabled:
- return
- self.rb_enabled = True
- self.rb_in_view = False
- self.setAcceptedMouseButtons(Qt.LeftButton)
- self.rb_xlo = xlo
- self.rb_xhi = xhi
- self.rb_event_handler = rb_event_handler
- self.mousePressEvent = self.MousePressEvent
- self.mouseMoveEvent = self.MouseMoveEvent
- self.mouseReleaseEvent = self.MouseReleaseEvent
- def boundingRect(self):
- return QRectF(0, 0, self.width, self.height)
- def paint(self, painter, option, widget):
- pass
- def RubberBandParent(self):
- scene = self.scene()
- view = scene.views()[0]
- viewport = view.viewport()
- return viewport
- def RubberBandSetGeometry(self, rect):
- scene_rectf = self.mapRectToScene(QRectF(rect))
- scene = self.scene()
- view = scene.views()[0]
- poly = view.mapFromScene(scene_rectf)
- self.rubber_band.setGeometry(poly.boundingRect())
- def SetSelection(self, selection_state):
- if self.rubber_band:
- if selection_state:
- self.RubberBandSetGeometry(selection_state)
- self.rubber_band.show()
- else:
- self.rubber_band.hide()
- def SetBracket(self, rect):
- if rect:
- x, y, width, height = rect.x(), rect.y(), rect.width(), rect.height()
- self.bracket.setPos(x, y)
- self.bracket.SetSize(width, height)
- self.bracket.show()
- else:
- self.bracket.hide()
- def RubberBandX(self, event):
- x = event.pos().toPoint().x()
- if x < self.rb_xlo:
- x = self.rb_xlo
- elif x > self.rb_xhi:
- x = self.rb_xhi
- else:
- self.rb_in_view = True
- return x
- def RubberBandRect(self, x):
- if self.rb_origin.x() <= x:
- width = x - self.rb_origin.x()
- rect = QRect(self.rb_origin, QSize(width, self.height))
- else:
- width = self.rb_origin.x() - x
- top_left = QPoint(self.rb_origin.x() - width, self.rb_origin.y())
- rect = QRect(top_left, QSize(width, self.height))
- return rect
- def MousePressEvent(self, event):
- self.rb_in_view = False
- x = self.RubberBandX(event)
- self.rb_origin = QPoint(x, self.top)
- if self.rubber_band is None:
- self.rubber_band = QRubberBand(QRubberBand.Rectangle, self.RubberBandParent())
- self.RubberBandSetGeometry(QRect(self.rb_origin, QSize(0, self.height)))
- if self.rb_in_view:
- self.rubber_band.show()
- self.rb_event_handler.RBMoveEvent(x, x)
- else:
- self.rubber_band.hide()
- def MouseMoveEvent(self, event):
- x = self.RubberBandX(event)
- rect = self.RubberBandRect(x)
- self.RubberBandSetGeometry(rect)
- if self.rb_in_view:
- self.rubber_band.show()
- self.rb_event_handler.RBMoveEvent(self.rb_origin.x(), x)
- def MouseReleaseEvent(self, event):
- x = self.RubberBandX(event)
- if self.rb_in_view:
- selection_state = self.RubberBandRect(x)
- else:
- selection_state = None
- self.rb_event_handler.RBReleaseEvent(self.rb_origin.x(), x, selection_state)
- # Switch graph legend data model
- class SwitchGraphLegendModel(QAbstractTableModel):
- def __init__(self, collection, region_attributes, parent=None):
- super(SwitchGraphLegendModel, self).__init__(parent)
- self.region_attributes = region_attributes
- self.child_items = sorted(collection.hregions.values(), key=GraphDataRegionOrdinal)
- self.child_count = len(self.child_items)
- self.highlight_set = set()
- self.column_headers = ("pid", "tid", "comm")
- def rowCount(self, parent):
- return self.child_count
- def headerData(self, section, orientation, role):
- if role != Qt.DisplayRole:
- return None
- if orientation != Qt.Horizontal:
- return None
- return self.columnHeader(section)
- def index(self, row, column, parent):
- return self.createIndex(row, column, self.child_items[row])
- def columnCount(self, parent=None):
- return len(self.column_headers)
- def columnHeader(self, column):
- return self.column_headers[column]
- def data(self, index, role):
- if role == Qt.BackgroundRole:
- child = self.child_items[index.row()]
- if child in self.highlight_set:
- return self.region_attributes[child.key].colour
- return None
- if role == Qt.ForegroundRole:
- child = self.child_items[index.row()]
- if child in self.highlight_set:
- return QColor(255, 255, 255)
- return self.region_attributes[child.key].colour
- if role != Qt.DisplayRole:
- return None
- hregion = self.child_items[index.row()]
- col = index.column()
- if col == 0:
- return hregion.pid
- if col == 1:
- return hregion.tid
- if col == 2:
- return hregion.comm
- return None
- def SetHighlight(self, row, set_highlight):
- child = self.child_items[row]
- top_left = self.createIndex(row, 0, child)
- bottom_right = self.createIndex(row, len(self.column_headers) - 1, child)
- self.dataChanged.emit(top_left, bottom_right)
- def Highlight(self, highlight_set):
- for row in xrange(self.child_count):
- child = self.child_items[row]
- if child in self.highlight_set:
- if child not in highlight_set:
- self.SetHighlight(row, False)
- elif child in highlight_set:
- self.SetHighlight(row, True)
- self.highlight_set = highlight_set
- # Switch graph legend is a table
- class SwitchGraphLegend(QWidget):
- def __init__(self, collection, region_attributes, parent=None):
- super(SwitchGraphLegend, self).__init__(parent)
- self.data_model = SwitchGraphLegendModel(collection, region_attributes)
- self.model = QSortFilterProxyModel()
- self.model.setSourceModel(self.data_model)
- self.view = QTableView()
- self.view.setModel(self.model)
- self.view.setEditTriggers(QAbstractItemView.NoEditTriggers)
- self.view.verticalHeader().setVisible(False)
- self.view.sortByColumn(-1, Qt.AscendingOrder)
- self.view.setSortingEnabled(True)
- self.view.resizeColumnsToContents()
- self.view.resizeRowsToContents()
- self.vbox = VBoxLayout(self.view)
- self.setLayout(self.vbox)
- sz1 = self.view.columnWidth(0) + self.view.columnWidth(1) + self.view.columnWidth(2) + 2
- sz1 = sz1 + self.view.verticalScrollBar().sizeHint().width()
- self.saved_size = sz1
- def resizeEvent(self, event):
- self.saved_size = self.size().width()
- super(SwitchGraphLegend, self).resizeEvent(event)
- def Highlight(self, highlight_set):
- self.data_model.Highlight(highlight_set)
- self.update()
- def changeEvent(self, event):
- if event.type() == QEvent.FontChange:
- self.view.resizeRowsToContents()
- self.view.resizeColumnsToContents()
- # Need to resize rows again after column resize
- self.view.resizeRowsToContents()
- super(SwitchGraphLegend, self).changeEvent(event)
- # Random colour generation
- def RGBColourTooLight(r, g, b):
- if g > 230:
- return True
- if g <= 160:
- return False
- if r <= 180 and g <= 180:
- return False
- if r < 60:
- return False
- return True
- def GenerateColours(x):
- cs = [0]
- for i in xrange(1, x):
- cs.append(int((255.0 / i) + 0.5))
- colours = []
- for r in cs:
- for g in cs:
- for b in cs:
- # Exclude black and colours that look too light against a white background
- if (r, g, b) == (0, 0, 0) or RGBColourTooLight(r, g, b):
- continue
- colours.append(QColor(r, g, b))
- return colours
- def GenerateNColours(n):
- for x in xrange(2, n + 2):
- colours = GenerateColours(x)
- if len(colours) >= n:
- return colours
- return []
- def GenerateNRandomColours(n, seed):
- colours = GenerateNColours(n)
- random.seed(seed)
- random.shuffle(colours)
- return colours
- # Graph attributes, in particular the scale and subrange that change when zooming
- class GraphAttributes():
- def __init__(self, scale, subrange, region_attributes, dp):
- self.scale = scale
- self.subrange = subrange
- self.region_attributes = region_attributes
- # Rounding avoids errors due to finite floating point precision
- self.dp = dp # data decimal places
- self.Update()
- def XToPixel(self, x):
- return int(round((x - self.subrange.x.lo) * self.scale.x, self.pdp.x))
- def YToPixel(self, y):
- return int(round((y - self.subrange.y.lo) * self.scale.y, self.pdp.y))
- def PixelToXRounded(self, px):
- return round((round(px, 0) / self.scale.x), self.dp.x) + self.subrange.x.lo
- def PixelToYRounded(self, py):
- return round((round(py, 0) / self.scale.y), self.dp.y) + self.subrange.y.lo
- def PixelToX(self, px):
- x = self.PixelToXRounded(px)
- if self.pdp.x == 0:
- rt = self.XToPixel(x)
- if rt > px:
- return x - 1
- return x
- def PixelToY(self, py):
- y = self.PixelToYRounded(py)
- if self.pdp.y == 0:
- rt = self.YToPixel(y)
- if rt > py:
- return y - 1
- return y
- def ToPDP(self, dp, scale):
- # Calculate pixel decimal places:
- # (10 ** dp) is the minimum delta in the data
- # scale it to get the minimum delta in pixels
- # log10 gives the number of decimals places negatively
- # subtrace 1 to divide by 10
- # round to the lower negative number
- # change the sign to get the number of decimals positively
- x = math.log10((10 ** dp) * scale)
- if x < 0:
- x -= 1
- x = -int(math.floor(x) - 0.1)
- else:
- x = 0
- return x
- def Update(self):
- x = self.ToPDP(self.dp.x, self.scale.x)
- y = self.ToPDP(self.dp.y, self.scale.y)
- self.pdp = XY(x, y) # pixel decimal places
- # Switch graph splitter which divides the CPU graphs from the legend
- class SwitchGraphSplitter(QSplitter):
- def __init__(self, parent=None):
- super(SwitchGraphSplitter, self).__init__(parent)
- self.first_time = False
- def resizeEvent(self, ev):
- if self.first_time:
- self.first_time = False
- sz1 = self.widget(1).view.columnWidth(0) + self.widget(1).view.columnWidth(1) + self.widget(1).view.columnWidth(2) + 2
- sz1 = sz1 + self.widget(1).view.verticalScrollBar().sizeHint().width()
- sz0 = self.size().width() - self.handleWidth() - sz1
- self.setSizes([sz0, sz1])
- elif not(self.widget(1).saved_size is None):
- sz1 = self.widget(1).saved_size
- sz0 = self.size().width() - self.handleWidth() - sz1
- self.setSizes([sz0, sz1])
- super(SwitchGraphSplitter, self).resizeEvent(ev)
- # Graph widget base class
- class GraphWidget(QWidget):
- graph_title_changed = Signal(object)
- def __init__(self, parent=None):
- super(GraphWidget, self).__init__(parent)
- def GraphTitleChanged(self, title):
- self.graph_title_changed.emit(title)
- def Title(self):
- return ""
- # Display time in s, ms, us or ns
- def ToTimeStr(val):
- val = Decimal(val)
- if val >= 1000000000:
- return "{} s".format((val / 1000000000).quantize(Decimal("0.000000001")))
- if val >= 1000000:
- return "{} ms".format((val / 1000000).quantize(Decimal("0.000001")))
- if val >= 1000:
- return "{} us".format((val / 1000).quantize(Decimal("0.001")))
- return "{} ns".format(val.quantize(Decimal("1")))
- # Switch (i.e. context switch i.e. Time Chart by CPU) graph widget which contains the CPU graphs and the legend and control buttons
- class SwitchGraphWidget(GraphWidget):
- def __init__(self, glb, collection, parent=None):
- super(SwitchGraphWidget, self).__init__(parent)
- self.glb = glb
- self.collection = collection
- self.back_state = []
- self.forward_state = []
- self.selection_state = (None, None)
- self.fwd_rect = None
- self.start_time = self.glb.StartTime(collection.machine_id)
- i = 0
- hregions = collection.hregions.values()
- colours = GenerateNRandomColours(len(hregions), 1013)
- region_attributes = {}
- for hregion in hregions:
- if hregion.pid == 0 and hregion.tid == 0:
- region_attributes[hregion.key] = GraphRegionAttribute(QColor(0, 0, 0))
- else:
- region_attributes[hregion.key] = GraphRegionAttribute(colours[i])
- i = i + 1
- # Default to entire range
- xsubrange = Subrange(0.0, float(collection.xrangehi - collection.xrangelo) + 1.0)
- ysubrange = Subrange(0.0, float(collection.yrangehi - collection.yrangelo) + 1.0)
- subrange = XY(xsubrange, ysubrange)
- scale = self.GetScaleForRange(subrange)
- self.attrs = GraphAttributes(scale, subrange, region_attributes, collection.dp)
- self.item = VertcalGraphSetGraphicsItem(collection, self.attrs, self, SwitchGraphGraphicsItem)
- self.scene = QGraphicsScene()
- self.scene.addItem(self.item)
- self.view = QGraphicsView(self.scene)
- self.view.centerOn(0, 0)
- self.view.setAlignment(Qt.AlignLeft | Qt.AlignTop)
- self.legend = SwitchGraphLegend(collection, region_attributes)
- self.splitter = SwitchGraphSplitter()
- self.splitter.addWidget(self.view)
- self.splitter.addWidget(self.legend)
- self.point_label = QLabel("")
- self.point_label.setSizePolicy(QSizePolicy.Preferred, QSizePolicy.Fixed)
- self.back_button = QToolButton()
- self.back_button.setIcon(self.style().standardIcon(QStyle.SP_ArrowLeft))
- self.back_button.setDisabled(True)
- self.back_button.released.connect(lambda: self.Back())
- self.forward_button = QToolButton()
- self.forward_button.setIcon(self.style().standardIcon(QStyle.SP_ArrowRight))
- self.forward_button.setDisabled(True)
- self.forward_button.released.connect(lambda: self.Forward())
- self.zoom_button = QToolButton()
- self.zoom_button.setText("Zoom")
- self.zoom_button.setDisabled(True)
- self.zoom_button.released.connect(lambda: self.Zoom())
- self.hbox = HBoxLayout(self.back_button, self.forward_button, self.zoom_button, self.point_label)
- self.vbox = VBoxLayout(self.splitter, self.hbox)
- self.setLayout(self.vbox)
- def GetScaleForRangeX(self, xsubrange):
- # Default graph 1000 pixels wide
- dflt = 1000.0
- r = xsubrange.hi - xsubrange.lo
- return dflt / r
- def GetScaleForRangeY(self, ysubrange):
- # Default graph 50 pixels high
- dflt = 50.0
- r = ysubrange.hi - ysubrange.lo
- return dflt / r
- def GetScaleForRange(self, subrange):
- # Default graph 1000 pixels wide, 50 pixels high
- xscale = self.GetScaleForRangeX(subrange.x)
- yscale = self.GetScaleForRangeY(subrange.y)
- return XY(xscale, yscale)
- def PointEvent(self, cpu, time_from, time_to, hregions):
- text = "CPU: " + str(cpu)
- time_from = time_from.quantize(Decimal(1))
- rel_time_from = time_from - self.glb.StartTime(self.collection.machine_id)
- text = text + " Time: " + str(time_from) + " (+" + ToTimeStr(rel_time_from) + ")"
- self.point_label.setText(text)
- self.legend.Highlight(hregions)
- def RightClickEvent(self, cpu, hregion_times, pos):
- if not IsSelectable(self.glb.db, "calls", "WHERE parent_id >= 0"):
- return
- menu = QMenu(self.view)
- for hregion, time in hregion_times:
- thread_at_time = (hregion.exec_comm_id, hregion.thread_id, time)
- menu_text = "Show Call Tree for {} {}:{} at {}".format(hregion.comm, hregion.pid, hregion.tid, time)
- menu.addAction(CreateAction(menu_text, "Show Call Tree", lambda a=None, args=thread_at_time: self.RightClickSelect(args), self.view))
- menu.exec_(pos)
- def RightClickSelect(self, args):
- CallTreeWindow(self.glb, self.glb.mainwindow, thread_at_time=args)
- def NoPointEvent(self):
- self.point_label.setText("")
- self.legend.Highlight({})
- def RangeEvent(self, time_from, time_to):
- time_from = time_from.quantize(Decimal(1))
- time_to = time_to.quantize(Decimal(1))
- if time_to <= time_from:
- self.point_label.setText("")
- return
- rel_time_from = time_from - self.start_time
- rel_time_to = time_to - self.start_time
- text = " Time: " + str(time_from) + " (+" + ToTimeStr(rel_time_from) + ") to: " + str(time_to) + " (+" + ToTimeStr(rel_time_to) + ")"
- text = text + " duration: " + ToTimeStr(time_to - time_from)
- self.point_label.setText(text)
- def BackState(self):
- return (self.attrs.subrange, self.attrs.scale, self.selection_state, self.fwd_rect)
- def PushBackState(self):
- state = copy.deepcopy(self.BackState())
- self.back_state.append(state)
- self.back_button.setEnabled(True)
- def PopBackState(self):
- self.attrs.subrange, self.attrs.scale, self.selection_state, self.fwd_rect = self.back_state.pop()
- self.attrs.Update()
- if not self.back_state:
- self.back_button.setDisabled(True)
- def PushForwardState(self):
- state = copy.deepcopy(self.BackState())
- self.forward_state.append(state)
- self.forward_button.setEnabled(True)
- def PopForwardState(self):
- self.attrs.subrange, self.attrs.scale, self.selection_state, self.fwd_rect = self.forward_state.pop()
- self.attrs.Update()
- if not self.forward_state:
- self.forward_button.setDisabled(True)
- def Title(self):
- time_from = self.collection.xrangelo + Decimal(self.attrs.subrange.x.lo)
- time_to = self.collection.xrangelo + Decimal(self.attrs.subrange.x.hi)
- rel_time_from = time_from - self.start_time
- rel_time_to = time_to - self.start_time
- title = "+" + ToTimeStr(rel_time_from) + " to +" + ToTimeStr(rel_time_to)
- title = title + " (" + ToTimeStr(time_to - time_from) + ")"
- return title
- def Update(self):
- selected_subrange, selection_state = self.selection_state
- self.item.SetSelection(selection_state)
- self.item.SetBracket(self.fwd_rect)
- self.zoom_button.setDisabled(selected_subrange is None)
- self.GraphTitleChanged(self.Title())
- self.item.update(self.item.boundingRect())
- def Back(self):
- if not self.back_state:
- return
- self.PushForwardState()
- self.PopBackState()
- self.Update()
- def Forward(self):
- if not self.forward_state:
- return
- self.PushBackState()
- self.PopForwardState()
- self.Update()
- def SelectEvent(self, x0, x1, selection_state):
- if selection_state is None:
- selected_subrange = None
- else:
- if x1 - x0 < 1.0:
- x1 += 1.0
- selected_subrange = Subrange(x0, x1)
- self.selection_state = (selected_subrange, selection_state)
- self.zoom_button.setDisabled(selected_subrange is None)
- def Zoom(self):
- selected_subrange, selection_state = self.selection_state
- if selected_subrange is None:
- return
- self.fwd_rect = selection_state
- self.item.SetSelection(None)
- self.PushBackState()
- self.attrs.subrange.x = selected_subrange
- self.forward_state = []
- self.forward_button.setDisabled(True)
- self.selection_state = (None, None)
- self.fwd_rect = None
- self.attrs.scale.x = self.GetScaleForRangeX(self.attrs.subrange.x)
- self.attrs.Update()
- self.Update()
- # Slow initialization - perform non-GUI initialization in a separate thread and put up a modal message box while waiting
- class SlowInitClass():
- def __init__(self, glb, title, init_fn):
- self.init_fn = init_fn
- self.done = False
- self.result = None
- self.msg_box = QMessageBox(glb.mainwindow)
- self.msg_box.setText("Initializing " + title + ". Please wait.")
- self.msg_box.setWindowTitle("Initializing " + title)
- self.msg_box.setWindowIcon(glb.mainwindow.style().standardIcon(QStyle.SP_MessageBoxInformation))
- self.init_thread = Thread(self.ThreadFn, glb)
- self.init_thread.done.connect(lambda: self.Done(), Qt.QueuedConnection)
- self.init_thread.start()
- def Done(self):
- self.msg_box.done(0)
- def ThreadFn(self, glb):
- conn_name = "SlowInitClass" + str(os.getpid())
- db, dbname = glb.dbref.Open(conn_name)
- self.result = self.init_fn(db)
- self.done = True
- return (True, 0)
- def Result(self):
- while not self.done:
- self.msg_box.exec_()
- self.init_thread.wait()
- return self.result
- def SlowInit(glb, title, init_fn):
- init = SlowInitClass(glb, title, init_fn)
- return init.Result()
- # Time chart by CPU window
- class TimeChartByCPUWindow(QMdiSubWindow):
- def __init__(self, glb, parent=None):
- super(TimeChartByCPUWindow, self).__init__(parent)
- self.glb = glb
- self.machine_id = glb.HostMachineId()
- self.collection_name = "SwitchGraphDataCollection " + str(self.machine_id)
- collection = LookupModel(self.collection_name)
- if collection is None:
- collection = SlowInit(glb, "Time Chart", self.Init)
- self.widget = SwitchGraphWidget(glb, collection, self)
- self.view = self.widget
- self.base_title = "Time Chart by CPU"
- self.setWindowTitle(self.base_title + self.widget.Title())
- self.widget.graph_title_changed.connect(self.GraphTitleChanged)
- self.setWidget(self.widget)
- AddSubWindow(glb.mainwindow.mdi_area, self, self.windowTitle())
- def Init(self, db):
- return LookupCreateModel(self.collection_name, lambda : SwitchGraphDataCollection(self.glb, db, self.machine_id))
- def GraphTitleChanged(self, title):
- self.setWindowTitle(self.base_title + " : " + title)
- # Child data item finder
- class ChildDataItemFinder():
- def __init__(self, root):
- self.root = root
- self.value, self.direction, self.pattern, self.last_value, self.last_pattern = (None,) * 5
- self.rows = []
- self.pos = 0
- def FindSelect(self):
- self.rows = []
- if self.pattern:
- pattern = re.compile(self.value)
- for child in self.root.child_items:
- for column_data in child.data:
- if re.search(pattern, str(column_data)) is not None:
- self.rows.append(child.row)
- break
- else:
- for child in self.root.child_items:
- for column_data in child.data:
- if self.value in str(column_data):
- self.rows.append(child.row)
- break
- def FindValue(self):
- self.pos = 0
- if self.last_value != self.value or self.pattern != self.last_pattern:
- self.FindSelect()
- if not len(self.rows):
- return -1
- return self.rows[self.pos]
- def FindThread(self):
- if self.direction == 0 or self.value != self.last_value or self.pattern != self.last_pattern:
- row = self.FindValue()
- elif len(self.rows):
- if self.direction > 0:
- self.pos += 1
- if self.pos >= len(self.rows):
- self.pos = 0
- else:
- self.pos -= 1
- if self.pos < 0:
- self.pos = len(self.rows) - 1
- row = self.rows[self.pos]
- else:
- row = -1
- return (True, row)
- def Find(self, value, direction, pattern, context, callback):
- self.value, self.direction, self.pattern, self.last_value, self.last_pattern = (value, direction,pattern, self.value, self.pattern)
- # Use a thread so the UI is not blocked
- thread = Thread(self.FindThread)
- thread.done.connect(lambda row, t=thread, c=callback: self.FindDone(t, c, row), Qt.QueuedConnection)
- thread.start()
- def FindDone(self, thread, callback, row):
- callback(row)
- # Number of database records to fetch in one go
- glb_chunk_sz = 10000
- # Background process for SQL data fetcher
- class SQLFetcherProcess():
- def __init__(self, dbref, sql, buffer, head, tail, fetch_count, fetching_done, process_target, wait_event, fetched_event, prep):
- # Need a unique connection name
- conn_name = "SQLFetcher" + str(os.getpid())
- self.db, dbname = dbref.Open(conn_name)
- self.sql = sql
- self.buffer = buffer
- self.head = head
- self.tail = tail
- self.fetch_count = fetch_count
- self.fetching_done = fetching_done
- self.process_target = process_target
- self.wait_event = wait_event
- self.fetched_event = fetched_event
- self.prep = prep
- self.query = QSqlQuery(self.db)
- self.query_limit = 0 if "$$last_id$$" in sql else 2
- self.last_id = -1
- self.fetched = 0
- self.more = True
- self.local_head = self.head.value
- self.local_tail = self.tail.value
- def Select(self):
- if self.query_limit:
- if self.query_limit == 1:
- return
- self.query_limit -= 1
- stmt = self.sql.replace("$$last_id$$", str(self.last_id))
- QueryExec(self.query, stmt)
- def Next(self):
- if not self.query.next():
- self.Select()
- if not self.query.next():
- return None
- self.last_id = self.query.value(0)
- return self.prep(self.query)
- def WaitForTarget(self):
- while True:
- self.wait_event.clear()
- target = self.process_target.value
- if target > self.fetched or target < 0:
- break
- self.wait_event.wait()
- return target
- def HasSpace(self, sz):
- if self.local_tail <= self.local_head:
- space = len(self.buffer) - self.local_head
- if space > sz:
- return True
- if space >= glb_nsz:
- # Use 0 (or space < glb_nsz) to mean there is no more at the top of the buffer
- nd = pickle.dumps(0, pickle.HIGHEST_PROTOCOL)
- self.buffer[self.local_head : self.local_head + len(nd)] = nd
- self.local_head = 0
- if self.local_tail - self.local_head > sz:
- return True
- return False
- def WaitForSpace(self, sz):
- if self.HasSpace(sz):
- return
- while True:
- self.wait_event.clear()
- self.local_tail = self.tail.value
- if self.HasSpace(sz):
- return
- self.wait_event.wait()
- def AddToBuffer(self, obj):
- d = pickle.dumps(obj, pickle.HIGHEST_PROTOCOL)
- n = len(d)
- nd = pickle.dumps(n, pickle.HIGHEST_PROTOCOL)
- sz = n + glb_nsz
- self.WaitForSpace(sz)
- pos = self.local_head
- self.buffer[pos : pos + len(nd)] = nd
- self.buffer[pos + glb_nsz : pos + sz] = d
- self.local_head += sz
- def FetchBatch(self, batch_size):
- fetched = 0
- while batch_size > fetched:
- obj = self.Next()
- if obj is None:
- self.more = False
- break
- self.AddToBuffer(obj)
- fetched += 1
- if fetched:
- self.fetched += fetched
- with self.fetch_count.get_lock():
- self.fetch_count.value += fetched
- self.head.value = self.local_head
- self.fetched_event.set()
- def Run(self):
- while self.more:
- target = self.WaitForTarget()
- if target < 0:
- break
- batch_size = min(glb_chunk_sz, target - self.fetched)
- self.FetchBatch(batch_size)
- self.fetching_done.value = True
- self.fetched_event.set()
- def SQLFetcherFn(*x):
- process = SQLFetcherProcess(*x)
- process.Run()
- # SQL data fetcher
- class SQLFetcher(QObject):
- done = Signal(object)
- def __init__(self, glb, sql, prep, process_data, parent=None):
- super(SQLFetcher, self).__init__(parent)
- self.process_data = process_data
- self.more = True
- self.target = 0
- self.last_target = 0
- self.fetched = 0
- self.buffer_size = 16 * 1024 * 1024
- self.buffer = Array(c_char, self.buffer_size, lock=False)
- self.head = Value(c_longlong)
- self.tail = Value(c_longlong)
- self.local_tail = 0
- self.fetch_count = Value(c_longlong)
- self.fetching_done = Value(c_bool)
- self.last_count = 0
- self.process_target = Value(c_longlong)
- self.wait_event = Event()
- self.fetched_event = Event()
- glb.AddInstanceToShutdownOnExit(self)
- self.process = Process(target=SQLFetcherFn, args=(glb.dbref, sql, self.buffer, self.head, self.tail, self.fetch_count, self.fetching_done, self.process_target, self.wait_event, self.fetched_event, prep))
- self.process.start()
- self.thread = Thread(self.Thread)
- self.thread.done.connect(self.ProcessData, Qt.QueuedConnection)
- self.thread.start()
- def Shutdown(self):
- # Tell the thread and process to exit
- self.process_target.value = -1
- self.wait_event.set()
- self.more = False
- self.fetching_done.value = True
- self.fetched_event.set()
- def Thread(self):
- if not self.more:
- return True, 0
- while True:
- self.fetched_event.clear()
- fetch_count = self.fetch_count.value
- if fetch_count != self.last_count:
- break
- if self.fetching_done.value:
- self.more = False
- return True, 0
- self.fetched_event.wait()
- count = fetch_count - self.last_count
- self.last_count = fetch_count
- self.fetched += count
- return False, count
- def Fetch(self, nr):
- if not self.more:
- # -1 inidcates there are no more
- return -1
- result = self.fetched
- extra = result + nr - self.target
- if extra > 0:
- self.target += extra
- # process_target < 0 indicates shutting down
- if self.process_target.value >= 0:
- self.process_target.value = self.target
- self.wait_event.set()
- return result
- def RemoveFromBuffer(self):
- pos = self.local_tail
- if len(self.buffer) - pos < glb_nsz:
- pos = 0
- n = pickle.loads(self.buffer[pos : pos + glb_nsz])
- if n == 0:
- pos = 0
- n = pickle.loads(self.buffer[0 : glb_nsz])
- pos += glb_nsz
- obj = pickle.loads(self.buffer[pos : pos + n])
- self.local_tail = pos + n
- return obj
- def ProcessData(self, count):
- for i in xrange(count):
- obj = self.RemoveFromBuffer()
- self.process_data(obj)
- self.tail.value = self.local_tail
- self.wait_event.set()
- self.done.emit(count)
- # Fetch more records bar
- class FetchMoreRecordsBar():
- def __init__(self, model, parent):
- self.model = model
- self.label = QLabel("Number of records (x " + "{:,}".format(glb_chunk_sz) + ") to fetch:")
- self.label.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
- self.fetch_count = QSpinBox()
- self.fetch_count.setRange(1, 1000000)
- self.fetch_count.setValue(10)
- self.fetch_count.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
- self.fetch = QPushButton("Go!")
- self.fetch.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
- self.fetch.released.connect(self.FetchMoreRecords)
- self.progress = QProgressBar()
- self.progress.setRange(0, 100)
- self.progress.hide()
- self.done_label = QLabel("All records fetched")
- self.done_label.hide()
- self.spacer = QLabel("")
- self.close_button = QToolButton()
- self.close_button.setIcon(parent.style().standardIcon(QStyle.SP_DockWidgetCloseButton))
- self.close_button.released.connect(self.Deactivate)
- self.hbox = QHBoxLayout()
- self.hbox.setContentsMargins(0, 0, 0, 0)
- self.hbox.addWidget(self.label)
- self.hbox.addWidget(self.fetch_count)
- self.hbox.addWidget(self.fetch)
- self.hbox.addWidget(self.spacer)
- self.hbox.addWidget(self.progress)
- self.hbox.addWidget(self.done_label)
- self.hbox.addWidget(self.close_button)
- self.bar = QWidget()
- self.bar.setLayout(self.hbox)
- self.bar.show()
- self.in_progress = False
- self.model.progress.connect(self.Progress)
- self.done = False
- if not model.HasMoreRecords():
- self.Done()
- def Widget(self):
- return self.bar
- def Activate(self):
- self.bar.show()
- self.fetch.setFocus()
- def Deactivate(self):
- self.bar.hide()
- def Enable(self, enable):
- self.fetch.setEnabled(enable)
- self.fetch_count.setEnabled(enable)
- def Busy(self):
- self.Enable(False)
- self.fetch.hide()
- self.spacer.hide()
- self.progress.show()
- def Idle(self):
- self.in_progress = False
- self.Enable(True)
- self.progress.hide()
- self.fetch.show()
- self.spacer.show()
- def Target(self):
- return self.fetch_count.value() * glb_chunk_sz
- def Done(self):
- self.done = True
- self.Idle()
- self.label.hide()
- self.fetch_count.hide()
- self.fetch.hide()
- self.spacer.hide()
- self.done_label.show()
- def Progress(self, count):
- if self.in_progress:
- if count:
- percent = ((count - self.start) * 100) / self.Target()
- if percent >= 100:
- self.Idle()
- else:
- self.progress.setValue(percent)
- if not count:
- # Count value of zero means no more records
- self.Done()
- def FetchMoreRecords(self):
- if self.done:
- return
- self.progress.setValue(0)
- self.Busy()
- self.in_progress = True
- self.start = self.model.FetchMoreRecords(self.Target())
- # Brance data model level two item
- class BranchLevelTwoItem():
- def __init__(self, row, col, text, parent_item):
- self.row = row
- self.parent_item = parent_item
- self.data = [""] * (col + 1)
- self.data[col] = text
- self.level = 2
- def getParentItem(self):
- return self.parent_item
- def getRow(self):
- return self.row
- def childCount(self):
- return 0
- def hasChildren(self):
- return False
- def getData(self, column):
- return self.data[column]
- # Brance data model level one item
- class BranchLevelOneItem():
- def __init__(self, glb, row, data, parent_item):
- self.glb = glb
- self.row = row
- self.parent_item = parent_item
- self.child_count = 0
- self.child_items = []
- self.data = data[1:]
- self.dbid = data[0]
- self.level = 1
- self.query_done = False
- self.br_col = len(self.data) - 1
- def getChildItem(self, row):
- return self.child_items[row]
- def getParentItem(self):
- return self.parent_item
- def getRow(self):
- return self.row
- def Select(self):
- self.query_done = True
- if not self.glb.have_disassembler:
- return
- query = QSqlQuery(self.glb.db)
- QueryExec(query, "SELECT cpu, to_dso_id, to_symbol_id, to_sym_offset, short_name, long_name, build_id, sym_start, to_ip"
- " FROM samples"
- " INNER JOIN dsos ON samples.to_dso_id = dsos.id"
- " INNER JOIN symbols ON samples.to_symbol_id = symbols.id"
- " WHERE samples.id = " + str(self.dbid))
- if not query.next():
- return
- cpu = query.value(0)
- dso = query.value(1)
- sym = query.value(2)
- if dso == 0 or sym == 0:
- return
- off = query.value(3)
- short_name = query.value(4)
- long_name = query.value(5)
- build_id = query.value(6)
- sym_start = query.value(7)
- ip = query.value(8)
- QueryExec(query, "SELECT samples.dso_id, symbol_id, sym_offset, sym_start"
- " FROM samples"
- " INNER JOIN symbols ON samples.symbol_id = symbols.id"
- " WHERE samples.id > " + str(self.dbid) + " AND cpu = " + str(cpu) +
- " ORDER BY samples.id"
- " LIMIT 1")
- if not query.next():
- return
- if query.value(0) != dso:
- # Cannot disassemble from one dso to another
- return
- bsym = query.value(1)
- boff = query.value(2)
- bsym_start = query.value(3)
- if bsym == 0:
- return
- tot = bsym_start + boff + 1 - sym_start - off
- if tot <= 0 or tot > 16384:
- return
- inst = self.glb.disassembler.Instruction()
- f = self.glb.FileFromNamesAndBuildId(short_name, long_name, build_id)
- if not f:
- return
- mode = 0 if Is64Bit(f) else 1
- self.glb.disassembler.SetMode(inst, mode)
- buf_sz = tot + 16
- buf = create_string_buffer(tot + 16)
- f.seek(sym_start + off)
- buf.value = f.read(buf_sz)
- buf_ptr = addressof(buf)
- i = 0
- while tot > 0:
- cnt, text = self.glb.disassembler.DisassembleOne(inst, buf_ptr, buf_sz, ip)
- if cnt:
- byte_str = tohex(ip).rjust(16)
- for k in xrange(cnt):
- byte_str += " %02x" % ord(buf[i])
- i += 1
- while k < 15:
- byte_str += " "
- k += 1
- self.child_items.append(BranchLevelTwoItem(0, self.br_col, byte_str + " " + text, self))
- self.child_count += 1
- else:
- return
- buf_ptr += cnt
- tot -= cnt
- buf_sz -= cnt
- ip += cnt
- def childCount(self):
- if not self.query_done:
- self.Select()
- if not self.child_count:
- return -1
- return self.child_count
- def hasChildren(self):
- if not self.query_done:
- return True
- return self.child_count > 0
- def getData(self, column):
- return self.data[column]
- # Brance data model root item
- class BranchRootItem():
- def __init__(self):
- self.child_count = 0
- self.child_items = []
- self.level = 0
- def getChildItem(self, row):
- return self.child_items[row]
- def getParentItem(self):
- return None
- def getRow(self):
- return 0
- def childCount(self):
- return self.child_count
- def hasChildren(self):
- return self.child_count > 0
- def getData(self, column):
- return ""
- # Calculate instructions per cycle
- def CalcIPC(cyc_cnt, insn_cnt):
- if cyc_cnt and insn_cnt:
- ipc = Decimal(float(insn_cnt) / cyc_cnt)
- ipc = str(ipc.quantize(Decimal(".01"), rounding=ROUND_HALF_UP))
- else:
- ipc = "0"
- return ipc
- # Branch data preparation
- def BranchDataPrepBr(query, data):
- data.append(tohex(query.value(8)).rjust(16) + " " + query.value(9) + offstr(query.value(10)) +
- " (" + dsoname(query.value(11)) + ")" + " -> " +
- tohex(query.value(12)) + " " + query.value(13) + offstr(query.value(14)) +
- " (" + dsoname(query.value(15)) + ")")
- def BranchDataPrepIPC(query, data):
- insn_cnt = query.value(16)
- cyc_cnt = query.value(17)
- ipc = CalcIPC(cyc_cnt, insn_cnt)
- data.append(insn_cnt)
- data.append(cyc_cnt)
- data.append(ipc)
- def BranchDataPrep(query):
- data = []
- for i in xrange(0, 8):
- data.append(query.value(i))
- BranchDataPrepBr(query, data)
- return data
- def BranchDataPrepWA(query):
- data = []
- data.append(query.value(0))
- # Workaround pyside failing to handle large integers (i.e. time) in python3 by converting to a string
- data.append("{:>19}".format(query.value(1)))
- for i in xrange(2, 8):
- data.append(query.value(i))
- BranchDataPrepBr(query, data)
- return data
- def BranchDataWithIPCPrep(query):
- data = []
- for i in xrange(0, 8):
- data.append(query.value(i))
- BranchDataPrepIPC(query, data)
- BranchDataPrepBr(query, data)
- return data
- def BranchDataWithIPCPrepWA(query):
- data = []
- data.append(query.value(0))
- # Workaround pyside failing to handle large integers (i.e. time) in python3 by converting to a string
- data.append("{:>19}".format(query.value(1)))
- for i in xrange(2, 8):
- data.append(query.value(i))
- BranchDataPrepIPC(query, data)
- BranchDataPrepBr(query, data)
- return data
- # Branch data model
- class BranchModel(TreeModel):
- progress = Signal(object)
- def __init__(self, glb, event_id, where_clause, parent=None):
- super(BranchModel, self).__init__(glb, None, parent)
- self.event_id = event_id
- self.more = True
- self.populated = 0
- self.have_ipc = IsSelectable(glb.db, "samples", columns = "insn_count, cyc_count")
- if self.have_ipc:
- select_ipc = ", insn_count, cyc_count"
- prep_fn = BranchDataWithIPCPrep
- prep_wa_fn = BranchDataWithIPCPrepWA
- else:
- select_ipc = ""
- prep_fn = BranchDataPrep
- prep_wa_fn = BranchDataPrepWA
- sql = ("SELECT samples.id, time, cpu, comm, pid, tid, branch_types.name,"
- " CASE WHEN in_tx = '0' THEN 'No' ELSE 'Yes' END,"
- " ip, symbols.name, sym_offset, dsos.short_name,"
- " to_ip, to_symbols.name, to_sym_offset, to_dsos.short_name"
- + select_ipc +
- " FROM samples"
- " INNER JOIN comms ON comm_id = comms.id"
- " INNER JOIN threads ON thread_id = threads.id"
- " INNER JOIN branch_types ON branch_type = branch_types.id"
- " INNER JOIN symbols ON symbol_id = symbols.id"
- " INNER JOIN symbols to_symbols ON to_symbol_id = to_symbols.id"
- " INNER JOIN dsos ON samples.dso_id = dsos.id"
- " INNER JOIN dsos AS to_dsos ON samples.to_dso_id = to_dsos.id"
- " WHERE samples.id > $$last_id$$" + where_clause +
- " AND evsel_id = " + str(self.event_id) +
- " ORDER BY samples.id"
- " LIMIT " + str(glb_chunk_sz))
- if pyside_version_1 and sys.version_info[0] == 3:
- prep = prep_fn
- else:
- prep = prep_wa_fn
- self.fetcher = SQLFetcher(glb, sql, prep, self.AddSample)
- self.fetcher.done.connect(self.Update)
- self.fetcher.Fetch(glb_chunk_sz)
- def GetRoot(self):
- return BranchRootItem()
- def columnCount(self, parent=None):
- if self.have_ipc:
- return 11
- else:
- return 8
- def columnHeader(self, column):
- if self.have_ipc:
- return ("Time", "CPU", "Command", "PID", "TID", "Branch Type", "In Tx", "Insn Cnt", "Cyc Cnt", "IPC", "Branch")[column]
- else:
- return ("Time", "CPU", "Command", "PID", "TID", "Branch Type", "In Tx", "Branch")[column]
- def columnFont(self, column):
- if self.have_ipc:
- br_col = 10
- else:
- br_col = 7
- if column != br_col:
- return None
- return QFont("Monospace")
- def DisplayData(self, item, index):
- if item.level == 1:
- self.FetchIfNeeded(item.row)
- return item.getData(index.column())
- def AddSample(self, data):
- child = BranchLevelOneItem(self.glb, self.populated, data, self.root)
- self.root.child_items.append(child)
- self.populated += 1
- def Update(self, fetched):
- if not fetched:
- self.more = False
- self.progress.emit(0)
- child_count = self.root.child_count
- count = self.populated - child_count
- if count > 0:
- parent = QModelIndex()
- self.beginInsertRows(parent, child_count, child_count + count - 1)
- self.insertRows(child_count, count, parent)
- self.root.child_count += count
- self.endInsertRows()
- self.progress.emit(self.root.child_count)
- def FetchMoreRecords(self, count):
- current = self.root.child_count
- if self.more:
- self.fetcher.Fetch(count)
- else:
- self.progress.emit(0)
- return current
- def HasMoreRecords(self):
- return self.more
- # Report Variables
- class ReportVars():
- def __init__(self, name = "", where_clause = "", limit = ""):
- self.name = name
- self.where_clause = where_clause
- self.limit = limit
- def UniqueId(self):
- return str(self.where_clause + ";" + self.limit)
- # Branch window
- class BranchWindow(QMdiSubWindow):
- def __init__(self, glb, event_id, report_vars, parent=None):
- super(BranchWindow, self).__init__(parent)
- model_name = "Branch Events " + str(event_id) + " " + report_vars.UniqueId()
- self.model = LookupCreateModel(model_name, lambda: BranchModel(glb, event_id, report_vars.where_clause))
- self.view = QTreeView()
- self.view.setUniformRowHeights(True)
- self.view.setSelectionMode(QAbstractItemView.ContiguousSelection)
- self.view.CopyCellsToClipboard = CopyTreeCellsToClipboard
- self.view.setModel(self.model)
- self.ResizeColumnsToContents()
- self.context_menu = TreeContextMenu(self.view)
- self.find_bar = FindBar(self, self, True)
- self.finder = ChildDataItemFinder(self.model.root)
- self.fetch_bar = FetchMoreRecordsBar(self.model, self)
- self.vbox = VBox(self.view, self.find_bar.Widget(), self.fetch_bar.Widget())
- self.setWidget(self.vbox.Widget())
- AddSubWindow(glb.mainwindow.mdi_area, self, report_vars.name + " Branch Events")
- def ResizeColumnToContents(self, column, n):
- # Using the view's resizeColumnToContents() here is extrememly slow
- # so implement a crude alternative
- mm = "MM" if column else "MMMM"
- font = self.view.font()
- metrics = QFontMetrics(font)
- max = 0
- for row in xrange(n):
- val = self.model.root.child_items[row].data[column]
- len = metrics.width(str(val) + mm)
- max = len if len > max else max
- val = self.model.columnHeader(column)
- len = metrics.width(str(val) + mm)
- max = len if len > max else max
- self.view.setColumnWidth(column, max)
- def ResizeColumnsToContents(self):
- n = min(self.model.root.child_count, 100)
- if n < 1:
- # No data yet, so connect a signal to notify when there is
- self.model.rowsInserted.connect(self.UpdateColumnWidths)
- return
- columns = self.model.columnCount()
- for i in xrange(columns):
- self.ResizeColumnToContents(i, n)
- def UpdateColumnWidths(self, *x):
- # This only needs to be done once, so disconnect the signal now
- self.model.rowsInserted.disconnect(self.UpdateColumnWidths)
- self.ResizeColumnsToContents()
- def Find(self, value, direction, pattern, context):
- self.view.setFocus()
- self.find_bar.Busy()
- self.finder.Find(value, direction, pattern, context, self.FindDone)
- def FindDone(self, row):
- self.find_bar.Idle()
- if row >= 0:
- self.view.setCurrentIndex(self.model.index(row, 0, QModelIndex()))
- else:
- self.find_bar.NotFound()
- # Line edit data item
- class LineEditDataItem(object):
- def __init__(self, glb, label, placeholder_text, parent, id = "", default = ""):
- self.glb = glb
- self.label = label
- self.placeholder_text = placeholder_text
- self.parent = parent
- self.id = id
- self.value = default
- self.widget = QLineEdit(default)
- self.widget.editingFinished.connect(self.Validate)
- self.widget.textChanged.connect(self.Invalidate)
- self.red = False
- self.error = ""
- self.validated = True
- if placeholder_text:
- self.widget.setPlaceholderText(placeholder_text)
- def TurnTextRed(self):
- if not self.red:
- palette = QPalette()
- palette.setColor(QPalette.Text,Qt.red)
- self.widget.setPalette(palette)
- self.red = True
- def TurnTextNormal(self):
- if self.red:
- palette = QPalette()
- self.widget.setPalette(palette)
- self.red = False
- def InvalidValue(self, value):
- self.value = ""
- self.TurnTextRed()
- self.error = self.label + " invalid value '" + value + "'"
- self.parent.ShowMessage(self.error)
- def Invalidate(self):
- self.validated = False
- def DoValidate(self, input_string):
- self.value = input_string.strip()
- def Validate(self):
- self.validated = True
- self.error = ""
- self.TurnTextNormal()
- self.parent.ClearMessage()
- input_string = self.widget.text()
- if not len(input_string.strip()):
- self.value = ""
- return
- self.DoValidate(input_string)
- def IsValid(self):
- if not self.validated:
- self.Validate()
- if len(self.error):
- self.parent.ShowMessage(self.error)
- return False
- return True
- def IsNumber(self, value):
- try:
- x = int(value)
- except:
- x = 0
- return str(x) == value
- # Non-negative integer ranges dialog data item
- class NonNegativeIntegerRangesDataItem(LineEditDataItem):
- def __init__(self, glb, label, placeholder_text, column_name, parent):
- super(NonNegativeIntegerRangesDataItem, self).__init__(glb, label, placeholder_text, parent)
- self.column_name = column_name
- def DoValidate(self, input_string):
- singles = []
- ranges = []
- for value in [x.strip() for x in input_string.split(",")]:
- if "-" in value:
- vrange = value.split("-")
- if len(vrange) != 2 or not self.IsNumber(vrange[0]) or not self.IsNumber(vrange[1]):
- return self.InvalidValue(value)
- ranges.append(vrange)
- else:
- if not self.IsNumber(value):
- return self.InvalidValue(value)
- singles.append(value)
- ranges = [("(" + self.column_name + " >= " + r[0] + " AND " + self.column_name + " <= " + r[1] + ")") for r in ranges]
- if len(singles):
- ranges.append(self.column_name + " IN (" + ",".join(singles) + ")")
- self.value = " OR ".join(ranges)
- # Positive integer dialog data item
- class PositiveIntegerDataItem(LineEditDataItem):
- def __init__(self, glb, label, placeholder_text, parent, id = "", default = ""):
- super(PositiveIntegerDataItem, self).__init__(glb, label, placeholder_text, parent, id, default)
- def DoValidate(self, input_string):
- if not self.IsNumber(input_string.strip()):
- return self.InvalidValue(input_string)
- value = int(input_string.strip())
- if value <= 0:
- return self.InvalidValue(input_string)
- self.value = str(value)
- # Dialog data item converted and validated using a SQL table
- class SQLTableDataItem(LineEditDataItem):
- def __init__(self, glb, label, placeholder_text, table_name, match_column, column_name1, column_name2, parent):
- super(SQLTableDataItem, self).__init__(glb, label, placeholder_text, parent)
- self.table_name = table_name
- self.match_column = match_column
- self.column_name1 = column_name1
- self.column_name2 = column_name2
- def ValueToIds(self, value):
- ids = []
- query = QSqlQuery(self.glb.db)
- stmt = "SELECT id FROM " + self.table_name + " WHERE " + self.match_column + " = '" + value + "'"
- ret = query.exec_(stmt)
- if ret:
- while query.next():
- ids.append(str(query.value(0)))
- return ids
- def DoValidate(self, input_string):
- all_ids = []
- for value in [x.strip() for x in input_string.split(",")]:
- ids = self.ValueToIds(value)
- if len(ids):
- all_ids.extend(ids)
- else:
- return self.InvalidValue(value)
- self.value = self.column_name1 + " IN (" + ",".join(all_ids) + ")"
- if self.column_name2:
- self.value = "( " + self.value + " OR " + self.column_name2 + " IN (" + ",".join(all_ids) + ") )"
- # Sample time ranges dialog data item converted and validated using 'samples' SQL table
- class SampleTimeRangesDataItem(LineEditDataItem):
- def __init__(self, glb, label, placeholder_text, column_name, parent):
- self.column_name = column_name
- self.last_id = 0
- self.first_time = 0
- self.last_time = 2 ** 64
- query = QSqlQuery(glb.db)
- QueryExec(query, "SELECT id, time FROM samples ORDER BY id DESC LIMIT 1")
- if query.next():
- self.last_id = int(query.value(0))
- self.first_time = int(glb.HostStartTime())
- self.last_time = int(glb.HostFinishTime())
- if placeholder_text:
- placeholder_text += ", between " + str(self.first_time) + " and " + str(self.last_time)
- super(SampleTimeRangesDataItem, self).__init__(glb, label, placeholder_text, parent)
- def IdBetween(self, query, lower_id, higher_id, order):
- QueryExec(query, "SELECT id FROM samples WHERE id > " + str(lower_id) + " AND id < " + str(higher_id) + " ORDER BY id " + order + " LIMIT 1")
- if query.next():
- return True, int(query.value(0))
- else:
- return False, 0
- def BinarySearchTime(self, lower_id, higher_id, target_time, get_floor):
- query = QSqlQuery(self.glb.db)
- while True:
- next_id = int((lower_id + higher_id) / 2)
- QueryExec(query, "SELECT time FROM samples WHERE id = " + str(next_id))
- if not query.next():
- ok, dbid = self.IdBetween(query, lower_id, next_id, "DESC")
- if not ok:
- ok, dbid = self.IdBetween(query, next_id, higher_id, "")
- if not ok:
- return str(higher_id)
- next_id = dbid
- QueryExec(query, "SELECT time FROM samples WHERE id = " + str(next_id))
- next_time = int(query.value(0))
- if get_floor:
- if target_time > next_time:
- lower_id = next_id
- else:
- higher_id = next_id
- if higher_id <= lower_id + 1:
- return str(higher_id)
- else:
- if target_time >= next_time:
- lower_id = next_id
- else:
- higher_id = next_id
- if higher_id <= lower_id + 1:
- return str(lower_id)
- def ConvertRelativeTime(self, val):
- mult = 1
- suffix = val[-2:]
- if suffix == "ms":
- mult = 1000000
- elif suffix == "us":
- mult = 1000
- elif suffix == "ns":
- mult = 1
- else:
- return val
- val = val[:-2].strip()
- if not self.IsNumber(val):
- return val
- val = int(val) * mult
- if val >= 0:
- val += self.first_time
- else:
- val += self.last_time
- return str(val)
- def ConvertTimeRange(self, vrange):
- if vrange[0] == "":
- vrange[0] = str(self.first_time)
- if vrange[1] == "":
- vrange[1] = str(self.last_time)
- vrange[0] = self.ConvertRelativeTime(vrange[0])
- vrange[1] = self.ConvertRelativeTime(vrange[1])
- if not self.IsNumber(vrange[0]) or not self.IsNumber(vrange[1]):
- return False
- beg_range = max(int(vrange[0]), self.first_time)
- end_range = min(int(vrange[1]), self.last_time)
- if beg_range > self.last_time or end_range < self.first_time:
- return False
- vrange[0] = self.BinarySearchTime(0, self.last_id, beg_range, True)
- vrange[1] = self.BinarySearchTime(1, self.last_id + 1, end_range, False)
- return True
- def AddTimeRange(self, value, ranges):
- n = value.count("-")
- if n == 1:
- pass
- elif n == 2:
- if value.split("-")[1].strip() == "":
- n = 1
- elif n == 3:
- n = 2
- else:
- return False
- pos = findnth(value, "-", n)
- vrange = [value[:pos].strip() ,value[pos+1:].strip()]
- if self.ConvertTimeRange(vrange):
- ranges.append(vrange)
- return True
- return False
- def DoValidate(self, input_string):
- ranges = []
- for value in [x.strip() for x in input_string.split(",")]:
- if not self.AddTimeRange(value, ranges):
- return self.InvalidValue(value)
- ranges = [("(" + self.column_name + " >= " + r[0] + " AND " + self.column_name + " <= " + r[1] + ")") for r in ranges]
- self.value = " OR ".join(ranges)
- # Report Dialog Base
- class ReportDialogBase(QDialog):
- def __init__(self, glb, title, items, partial, parent=None):
- super(ReportDialogBase, self).__init__(parent)
- self.glb = glb
- self.report_vars = ReportVars()
- self.setWindowTitle(title)
- self.setMinimumWidth(600)
- self.data_items = [x(glb, self) for x in items]
- self.partial = partial
- self.grid = QGridLayout()
- for row in xrange(len(self.data_items)):
- self.grid.addWidget(QLabel(self.data_items[row].label), row, 0)
- self.grid.addWidget(self.data_items[row].widget, row, 1)
- self.status = QLabel()
- self.ok_button = QPushButton("Ok", self)
- self.ok_button.setDefault(True)
- self.ok_button.released.connect(self.Ok)
- self.ok_button.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
- self.cancel_button = QPushButton("Cancel", self)
- self.cancel_button.released.connect(self.reject)
- self.cancel_button.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
- self.hbox = QHBoxLayout()
- #self.hbox.addStretch()
- self.hbox.addWidget(self.status)
- self.hbox.addWidget(self.ok_button)
- self.hbox.addWidget(self.cancel_button)
- self.vbox = QVBoxLayout()
- self.vbox.addLayout(self.grid)
- self.vbox.addLayout(self.hbox)
- self.setLayout(self.vbox)
- def Ok(self):
- vars = self.report_vars
- for d in self.data_items:
- if d.id == "REPORTNAME":
- vars.name = d.value
- if not vars.name:
- self.ShowMessage("Report name is required")
- return
- for d in self.data_items:
- if not d.IsValid():
- return
- for d in self.data_items[1:]:
- if d.id == "LIMIT":
- vars.limit = d.value
- elif len(d.value):
- if len(vars.where_clause):
- vars.where_clause += " AND "
- vars.where_clause += d.value
- if len(vars.where_clause):
- if self.partial:
- vars.where_clause = " AND ( " + vars.where_clause + " ) "
- else:
- vars.where_clause = " WHERE " + vars.where_clause + " "
- self.accept()
- def ShowMessage(self, msg):
- self.status.setText("<font color=#FF0000>" + msg)
- def ClearMessage(self):
- self.status.setText("")
- # Selected branch report creation dialog
- class SelectedBranchDialog(ReportDialogBase):
- def __init__(self, glb, parent=None):
- title = "Selected Branches"
- items = (lambda g, p: LineEditDataItem(g, "Report name:", "Enter a name to appear in the window title bar", p, "REPORTNAME"),
- lambda g, p: SampleTimeRangesDataItem(g, "Time ranges:", "Enter time ranges", "samples.id", p),
- lambda g, p: NonNegativeIntegerRangesDataItem(g, "CPUs:", "Enter CPUs or ranges e.g. 0,5-6", "cpu", p),
- lambda g, p: SQLTableDataItem(g, "Commands:", "Only branches with these commands will be included", "comms", "comm", "comm_id", "", p),
- lambda g, p: SQLTableDataItem(g, "PIDs:", "Only branches with these process IDs will be included", "threads", "pid", "thread_id", "", p),
- lambda g, p: SQLTableDataItem(g, "TIDs:", "Only branches with these thread IDs will be included", "threads", "tid", "thread_id", "", p),
- lambda g, p: SQLTableDataItem(g, "DSOs:", "Only branches with these DSOs will be included", "dsos", "short_name", "samples.dso_id", "to_dso_id", p),
- lambda g, p: SQLTableDataItem(g, "Symbols:", "Only branches with these symbols will be included", "symbols", "name", "symbol_id", "to_symbol_id", p),
- lambda g, p: LineEditDataItem(g, "Raw SQL clause: ", "Enter a raw SQL WHERE clause", p))
- super(SelectedBranchDialog, self).__init__(glb, title, items, True, parent)
- # Event list
- def GetEventList(db):
- events = []
- query = QSqlQuery(db)
- QueryExec(query, "SELECT name FROM selected_events WHERE id > 0 ORDER BY id")
- while query.next():
- events.append(query.value(0))
- return events
- # Is a table selectable
- def IsSelectable(db, table, sql = "", columns = "*"):
- query = QSqlQuery(db)
- try:
- QueryExec(query, "SELECT " + columns + " FROM " + table + " " + sql + " LIMIT 1")
- except:
- return False
- return True
- # SQL table data model item
- class SQLTableItem():
- def __init__(self, row, data):
- self.row = row
- self.data = data
- def getData(self, column):
- return self.data[column]
- # SQL table data model
- class SQLTableModel(TableModel):
- progress = Signal(object)
- def __init__(self, glb, sql, column_headers, parent=None):
- super(SQLTableModel, self).__init__(parent)
- self.glb = glb
- self.more = True
- self.populated = 0
- self.column_headers = column_headers
- self.fetcher = SQLFetcher(glb, sql, lambda x, y=len(column_headers): self.SQLTableDataPrep(x, y), self.AddSample)
- self.fetcher.done.connect(self.Update)
- self.fetcher.Fetch(glb_chunk_sz)
- def DisplayData(self, item, index):
- self.FetchIfNeeded(item.row)
- return item.getData(index.column())
- def AddSample(self, data):
- child = SQLTableItem(self.populated, data)
- self.child_items.append(child)
- self.populated += 1
- def Update(self, fetched):
- if not fetched:
- self.more = False
- self.progress.emit(0)
- child_count = self.child_count
- count = self.populated - child_count
- if count > 0:
- parent = QModelIndex()
- self.beginInsertRows(parent, child_count, child_count + count - 1)
- self.insertRows(child_count, count, parent)
- self.child_count += count
- self.endInsertRows()
- self.progress.emit(self.child_count)
- def FetchMoreRecords(self, count):
- current = self.child_count
- if self.more:
- self.fetcher.Fetch(count)
- else:
- self.progress.emit(0)
- return current
- def HasMoreRecords(self):
- return self.more
- def columnCount(self, parent=None):
- return len(self.column_headers)
- def columnHeader(self, column):
- return self.column_headers[column]
- def SQLTableDataPrep(self, query, count):
- data = []
- for i in xrange(count):
- data.append(query.value(i))
- return data
- # SQL automatic table data model
- class SQLAutoTableModel(SQLTableModel):
- def __init__(self, glb, table_name, parent=None):
- sql = "SELECT * FROM " + table_name + " WHERE id > $$last_id$$ ORDER BY id LIMIT " + str(glb_chunk_sz)
- if table_name == "comm_threads_view":
- # For now, comm_threads_view has no id column
- sql = "SELECT * FROM " + table_name + " WHERE comm_id > $$last_id$$ ORDER BY comm_id LIMIT " + str(glb_chunk_sz)
- column_headers = []
- query = QSqlQuery(glb.db)
- if glb.dbref.is_sqlite3:
- QueryExec(query, "PRAGMA table_info(" + table_name + ")")
- while query.next():
- column_headers.append(query.value(1))
- if table_name == "sqlite_master":
- sql = "SELECT * FROM " + table_name
- else:
- if table_name[:19] == "information_schema.":
- sql = "SELECT * FROM " + table_name
- select_table_name = table_name[19:]
- schema = "information_schema"
- else:
- select_table_name = table_name
- schema = "public"
- QueryExec(query, "SELECT column_name FROM information_schema.columns WHERE table_schema = '" + schema + "' and table_name = '" + select_table_name + "'")
- while query.next():
- column_headers.append(query.value(0))
- if pyside_version_1 and sys.version_info[0] == 3:
- if table_name == "samples_view":
- self.SQLTableDataPrep = self.samples_view_DataPrep
- if table_name == "samples":
- self.SQLTableDataPrep = self.samples_DataPrep
- super(SQLAutoTableModel, self).__init__(glb, sql, column_headers, parent)
- def samples_view_DataPrep(self, query, count):
- data = []
- data.append(query.value(0))
- # Workaround pyside failing to handle large integers (i.e. time) in python3 by converting to a string
- data.append("{:>19}".format(query.value(1)))
- for i in xrange(2, count):
- data.append(query.value(i))
- return data
- def samples_DataPrep(self, query, count):
- data = []
- for i in xrange(9):
- data.append(query.value(i))
- # Workaround pyside failing to handle large integers (i.e. time) in python3 by converting to a string
- data.append("{:>19}".format(query.value(9)))
- for i in xrange(10, count):
- data.append(query.value(i))
- return data
- # Base class for custom ResizeColumnsToContents
- class ResizeColumnsToContentsBase(QObject):
- def __init__(self, parent=None):
- super(ResizeColumnsToContentsBase, self).__init__(parent)
- def ResizeColumnToContents(self, column, n):
- # Using the view's resizeColumnToContents() here is extrememly slow
- # so implement a crude alternative
- font = self.view.font()
- metrics = QFontMetrics(font)
- max = 0
- for row in xrange(n):
- val = self.data_model.child_items[row].data[column]
- len = metrics.width(str(val) + "MM")
- max = len if len > max else max
- val = self.data_model.columnHeader(column)
- len = metrics.width(str(val) + "MM")
- max = len if len > max else max
- self.view.setColumnWidth(column, max)
- def ResizeColumnsToContents(self):
- n = min(self.data_model.child_count, 100)
- if n < 1:
- # No data yet, so connect a signal to notify when there is
- self.data_model.rowsInserted.connect(self.UpdateColumnWidths)
- return
- columns = self.data_model.columnCount()
- for i in xrange(columns):
- self.ResizeColumnToContents(i, n)
- def UpdateColumnWidths(self, *x):
- # This only needs to be done once, so disconnect the signal now
- self.data_model.rowsInserted.disconnect(self.UpdateColumnWidths)
- self.ResizeColumnsToContents()
- # Convert value to CSV
- def ToCSValue(val):
- if '"' in val:
- val = val.replace('"', '""')
- if "," in val or '"' in val:
- val = '"' + val + '"'
- return val
- # Key to sort table model indexes by row / column, assuming fewer than 1000 columns
- glb_max_cols = 1000
- def RowColumnKey(a):
- return a.row() * glb_max_cols + a.column()
- # Copy selected table cells to clipboard
- def CopyTableCellsToClipboard(view, as_csv=False, with_hdr=False):
- indexes = sorted(view.selectedIndexes(), key=RowColumnKey)
- idx_cnt = len(indexes)
- if not idx_cnt:
- return
- if idx_cnt == 1:
- with_hdr=False
- min_row = indexes[0].row()
- max_row = indexes[0].row()
- min_col = indexes[0].column()
- max_col = indexes[0].column()
- for i in indexes:
- min_row = min(min_row, i.row())
- max_row = max(max_row, i.row())
- min_col = min(min_col, i.column())
- max_col = max(max_col, i.column())
- if max_col > glb_max_cols:
- raise RuntimeError("glb_max_cols is too low")
- max_width = [0] * (1 + max_col - min_col)
- for i in indexes:
- c = i.column() - min_col
- max_width[c] = max(max_width[c], len(str(i.data())))
- text = ""
- pad = ""
- sep = ""
- if with_hdr:
- model = indexes[0].model()
- for col in range(min_col, max_col + 1):
- val = model.headerData(col, Qt.Horizontal, Qt.DisplayRole)
- if as_csv:
- text += sep + ToCSValue(val)
- sep = ","
- else:
- c = col - min_col
- max_width[c] = max(max_width[c], len(val))
- width = max_width[c]
- align = model.headerData(col, Qt.Horizontal, Qt.TextAlignmentRole)
- if align & Qt.AlignRight:
- val = val.rjust(width)
- text += pad + sep + val
- pad = " " * (width - len(val))
- sep = " "
- text += "\n"
- pad = ""
- sep = ""
- last_row = min_row
- for i in indexes:
- if i.row() > last_row:
- last_row = i.row()
- text += "\n"
- pad = ""
- sep = ""
- if as_csv:
- text += sep + ToCSValue(str(i.data()))
- sep = ","
- else:
- width = max_width[i.column() - min_col]
- if i.data(Qt.TextAlignmentRole) & Qt.AlignRight:
- val = str(i.data()).rjust(width)
- else:
- val = str(i.data())
- text += pad + sep + val
- pad = " " * (width - len(val))
- sep = " "
- QApplication.clipboard().setText(text)
- def CopyTreeCellsToClipboard(view, as_csv=False, with_hdr=False):
- indexes = view.selectedIndexes()
- if not len(indexes):
- return
- selection = view.selectionModel()
- first = None
- for i in indexes:
- above = view.indexAbove(i)
- if not selection.isSelected(above):
- first = i
- break
- if first is None:
- raise RuntimeError("CopyTreeCellsToClipboard internal error")
- model = first.model()
- row_cnt = 0
- col_cnt = model.columnCount(first)
- max_width = [0] * col_cnt
- indent_sz = 2
- indent_str = " " * indent_sz
- expanded_mark_sz = 2
- if sys.version_info[0] == 3:
- expanded_mark = "\u25BC "
- not_expanded_mark = "\u25B6 "
- else:
- expanded_mark = unicode(chr(0xE2) + chr(0x96) + chr(0xBC) + " ", "utf-8")
- not_expanded_mark = unicode(chr(0xE2) + chr(0x96) + chr(0xB6) + " ", "utf-8")
- leaf_mark = " "
- if not as_csv:
- pos = first
- while True:
- row_cnt += 1
- row = pos.row()
- for c in range(col_cnt):
- i = pos.sibling(row, c)
- if c:
- n = len(str(i.data()))
- else:
- n = len(str(i.data()).strip())
- n += (i.internalPointer().level - 1) * indent_sz
- n += expanded_mark_sz
- max_width[c] = max(max_width[c], n)
- pos = view.indexBelow(pos)
- if not selection.isSelected(pos):
- break
- text = ""
- pad = ""
- sep = ""
- if with_hdr:
- for c in range(col_cnt):
- val = model.headerData(c, Qt.Horizontal, Qt.DisplayRole).strip()
- if as_csv:
- text += sep + ToCSValue(val)
- sep = ","
- else:
- max_width[c] = max(max_width[c], len(val))
- width = max_width[c]
- align = model.headerData(c, Qt.Horizontal, Qt.TextAlignmentRole)
- if align & Qt.AlignRight:
- val = val.rjust(width)
- text += pad + sep + val
- pad = " " * (width - len(val))
- sep = " "
- text += "\n"
- pad = ""
- sep = ""
- pos = first
- while True:
- row = pos.row()
- for c in range(col_cnt):
- i = pos.sibling(row, c)
- val = str(i.data())
- if not c:
- if model.hasChildren(i):
- if view.isExpanded(i):
- mark = expanded_mark
- else:
- mark = not_expanded_mark
- else:
- mark = leaf_mark
- val = indent_str * (i.internalPointer().level - 1) + mark + val.strip()
- if as_csv:
- text += sep + ToCSValue(val)
- sep = ","
- else:
- width = max_width[c]
- if c and i.data(Qt.TextAlignmentRole) & Qt.AlignRight:
- val = val.rjust(width)
- text += pad + sep + val
- pad = " " * (width - len(val))
- sep = " "
- pos = view.indexBelow(pos)
- if not selection.isSelected(pos):
- break
- text = text.rstrip() + "\n"
- pad = ""
- sep = ""
- QApplication.clipboard().setText(text)
- def CopyCellsToClipboard(view, as_csv=False, with_hdr=False):
- view.CopyCellsToClipboard(view, as_csv, with_hdr)
- def CopyCellsToClipboardHdr(view):
- CopyCellsToClipboard(view, False, True)
- def CopyCellsToClipboardCSV(view):
- CopyCellsToClipboard(view, True, True)
- # Context menu
- class ContextMenu(object):
- def __init__(self, view):
- self.view = view
- self.view.setContextMenuPolicy(Qt.CustomContextMenu)
- self.view.customContextMenuRequested.connect(self.ShowContextMenu)
- def ShowContextMenu(self, pos):
- menu = QMenu(self.view)
- self.AddActions(menu)
- menu.exec_(self.view.mapToGlobal(pos))
- def AddCopy(self, menu):
- menu.addAction(CreateAction("&Copy selection", "Copy to clipboard", lambda: CopyCellsToClipboardHdr(self.view), self.view))
- menu.addAction(CreateAction("Copy selection as CS&V", "Copy to clipboard as CSV", lambda: CopyCellsToClipboardCSV(self.view), self.view))
- def AddActions(self, menu):
- self.AddCopy(menu)
- class TreeContextMenu(ContextMenu):
- def __init__(self, view):
- super(TreeContextMenu, self).__init__(view)
- def AddActions(self, menu):
- i = self.view.currentIndex()
- text = str(i.data()).strip()
- if len(text):
- menu.addAction(CreateAction('Copy "' + text + '"', "Copy to clipboard", lambda: QApplication.clipboard().setText(text), self.view))
- self.AddCopy(menu)
- # Table window
- class TableWindow(QMdiSubWindow, ResizeColumnsToContentsBase):
- def __init__(self, glb, table_name, parent=None):
- super(TableWindow, self).__init__(parent)
- self.data_model = LookupCreateModel(table_name + " Table", lambda: SQLAutoTableModel(glb, table_name))
- self.model = QSortFilterProxyModel()
- self.model.setSourceModel(self.data_model)
- self.view = QTableView()
- self.view.setModel(self.model)
- self.view.setEditTriggers(QAbstractItemView.NoEditTriggers)
- self.view.verticalHeader().setVisible(False)
- self.view.sortByColumn(-1, Qt.AscendingOrder)
- self.view.setSortingEnabled(True)
- self.view.setSelectionMode(QAbstractItemView.ContiguousSelection)
- self.view.CopyCellsToClipboard = CopyTableCellsToClipboard
- self.ResizeColumnsToContents()
- self.context_menu = ContextMenu(self.view)
- self.find_bar = FindBar(self, self, True)
- self.finder = ChildDataItemFinder(self.data_model)
- self.fetch_bar = FetchMoreRecordsBar(self.data_model, self)
- self.vbox = VBox(self.view, self.find_bar.Widget(), self.fetch_bar.Widget())
- self.setWidget(self.vbox.Widget())
- AddSubWindow(glb.mainwindow.mdi_area, self, table_name + " Table")
- def Find(self, value, direction, pattern, context):
- self.view.setFocus()
- self.find_bar.Busy()
- self.finder.Find(value, direction, pattern, context, self.FindDone)
- def FindDone(self, row):
- self.find_bar.Idle()
- if row >= 0:
- self.view.setCurrentIndex(self.model.mapFromSource(self.data_model.index(row, 0, QModelIndex())))
- else:
- self.find_bar.NotFound()
- # Table list
- def GetTableList(glb):
- tables = []
- query = QSqlQuery(glb.db)
- if glb.dbref.is_sqlite3:
- QueryExec(query, "SELECT name FROM sqlite_master WHERE type IN ( 'table' , 'view' ) ORDER BY name")
- else:
- QueryExec(query, "SELECT table_name FROM information_schema.tables WHERE table_schema = 'public' AND table_type IN ( 'BASE TABLE' , 'VIEW' ) ORDER BY table_name")
- while query.next():
- tables.append(query.value(0))
- if glb.dbref.is_sqlite3:
- tables.append("sqlite_master")
- else:
- tables.append("information_schema.tables")
- tables.append("information_schema.views")
- tables.append("information_schema.columns")
- return tables
- # Top Calls data model
- class TopCallsModel(SQLTableModel):
- def __init__(self, glb, report_vars, parent=None):
- text = ""
- if not glb.dbref.is_sqlite3:
- text = "::text"
- limit = ""
- if len(report_vars.limit):
- limit = " LIMIT " + report_vars.limit
- sql = ("SELECT comm, pid, tid, name,"
- " CASE"
- " WHEN (short_name = '[kernel.kallsyms]') THEN '[kernel]'" + text +
- " ELSE short_name"
- " END AS dso,"
- " call_time, return_time, (return_time - call_time) AS elapsed_time, branch_count, "
- " CASE"
- " WHEN (calls.flags = 1) THEN 'no call'" + text +
- " WHEN (calls.flags = 2) THEN 'no return'" + text +
- " WHEN (calls.flags = 3) THEN 'no call/return'" + text +
- " ELSE ''" + text +
- " END AS flags"
- " FROM calls"
- " INNER JOIN call_paths ON calls.call_path_id = call_paths.id"
- " INNER JOIN symbols ON call_paths.symbol_id = symbols.id"
- " INNER JOIN dsos ON symbols.dso_id = dsos.id"
- " INNER JOIN comms ON calls.comm_id = comms.id"
- " INNER JOIN threads ON calls.thread_id = threads.id" +
- report_vars.where_clause +
- " ORDER BY elapsed_time DESC" +
- limit
- )
- column_headers = ("Command", "PID", "TID", "Symbol", "Object", "Call Time", "Return Time", "Elapsed Time (ns)", "Branch Count", "Flags")
- self.alignment = (Qt.AlignLeft, Qt.AlignLeft, Qt.AlignLeft, Qt.AlignLeft, Qt.AlignLeft, Qt.AlignLeft, Qt.AlignLeft, Qt.AlignRight, Qt.AlignRight, Qt.AlignLeft)
- super(TopCallsModel, self).__init__(glb, sql, column_headers, parent)
- def columnAlignment(self, column):
- return self.alignment[column]
- # Top Calls report creation dialog
- class TopCallsDialog(ReportDialogBase):
- def __init__(self, glb, parent=None):
- title = "Top Calls by Elapsed Time"
- items = (lambda g, p: LineEditDataItem(g, "Report name:", "Enter a name to appear in the window title bar", p, "REPORTNAME"),
- lambda g, p: SQLTableDataItem(g, "Commands:", "Only calls with these commands will be included", "comms", "comm", "comm_id", "", p),
- lambda g, p: SQLTableDataItem(g, "PIDs:", "Only calls with these process IDs will be included", "threads", "pid", "thread_id", "", p),
- lambda g, p: SQLTableDataItem(g, "TIDs:", "Only calls with these thread IDs will be included", "threads", "tid", "thread_id", "", p),
- lambda g, p: SQLTableDataItem(g, "DSOs:", "Only calls with these DSOs will be included", "dsos", "short_name", "dso_id", "", p),
- lambda g, p: SQLTableDataItem(g, "Symbols:", "Only calls with these symbols will be included", "symbols", "name", "symbol_id", "", p),
- lambda g, p: LineEditDataItem(g, "Raw SQL clause: ", "Enter a raw SQL WHERE clause", p),
- lambda g, p: PositiveIntegerDataItem(g, "Record limit:", "Limit selection to this number of records", p, "LIMIT", "100"))
- super(TopCallsDialog, self).__init__(glb, title, items, False, parent)
- # Top Calls window
- class TopCallsWindow(QMdiSubWindow, ResizeColumnsToContentsBase):
- def __init__(self, glb, report_vars, parent=None):
- super(TopCallsWindow, self).__init__(parent)
- self.data_model = LookupCreateModel("Top Calls " + report_vars.UniqueId(), lambda: TopCallsModel(glb, report_vars))
- self.model = self.data_model
- self.view = QTableView()
- self.view.setModel(self.model)
- self.view.setEditTriggers(QAbstractItemView.NoEditTriggers)
- self.view.verticalHeader().setVisible(False)
- self.view.setSelectionMode(QAbstractItemView.ContiguousSelection)
- self.view.CopyCellsToClipboard = CopyTableCellsToClipboard
- self.context_menu = ContextMenu(self.view)
- self.ResizeColumnsToContents()
- self.find_bar = FindBar(self, self, True)
- self.finder = ChildDataItemFinder(self.model)
- self.fetch_bar = FetchMoreRecordsBar(self.data_model, self)
- self.vbox = VBox(self.view, self.find_bar.Widget(), self.fetch_bar.Widget())
- self.setWidget(self.vbox.Widget())
- AddSubWindow(glb.mainwindow.mdi_area, self, report_vars.name)
- def Find(self, value, direction, pattern, context):
- self.view.setFocus()
- self.find_bar.Busy()
- self.finder.Find(value, direction, pattern, context, self.FindDone)
- def FindDone(self, row):
- self.find_bar.Idle()
- if row >= 0:
- self.view.setCurrentIndex(self.model.index(row, 0, QModelIndex()))
- else:
- self.find_bar.NotFound()
- # Action Definition
- def CreateAction(label, tip, callback, parent=None, shortcut=None):
- action = QAction(label, parent)
- if shortcut != None:
- action.setShortcuts(shortcut)
- action.setStatusTip(tip)
- action.triggered.connect(callback)
- return action
- # Typical application actions
- def CreateExitAction(app, parent=None):
- return CreateAction("&Quit", "Exit the application", app.closeAllWindows, parent, QKeySequence.Quit)
- # Typical MDI actions
- def CreateCloseActiveWindowAction(mdi_area):
- return CreateAction("Cl&ose", "Close the active window", mdi_area.closeActiveSubWindow, mdi_area)
- def CreateCloseAllWindowsAction(mdi_area):
- return CreateAction("Close &All", "Close all the windows", mdi_area.closeAllSubWindows, mdi_area)
- def CreateTileWindowsAction(mdi_area):
- return CreateAction("&Tile", "Tile the windows", mdi_area.tileSubWindows, mdi_area)
- def CreateCascadeWindowsAction(mdi_area):
- return CreateAction("&Cascade", "Cascade the windows", mdi_area.cascadeSubWindows, mdi_area)
- def CreateNextWindowAction(mdi_area):
- return CreateAction("Ne&xt", "Move the focus to the next window", mdi_area.activateNextSubWindow, mdi_area, QKeySequence.NextChild)
- def CreatePreviousWindowAction(mdi_area):
- return CreateAction("Pre&vious", "Move the focus to the previous window", mdi_area.activatePreviousSubWindow, mdi_area, QKeySequence.PreviousChild)
- # Typical MDI window menu
- class WindowMenu():
- def __init__(self, mdi_area, menu):
- self.mdi_area = mdi_area
- self.window_menu = menu.addMenu("&Windows")
- self.close_active_window = CreateCloseActiveWindowAction(mdi_area)
- self.close_all_windows = CreateCloseAllWindowsAction(mdi_area)
- self.tile_windows = CreateTileWindowsAction(mdi_area)
- self.cascade_windows = CreateCascadeWindowsAction(mdi_area)
- self.next_window = CreateNextWindowAction(mdi_area)
- self.previous_window = CreatePreviousWindowAction(mdi_area)
- self.window_menu.aboutToShow.connect(self.Update)
- def Update(self):
- self.window_menu.clear()
- sub_window_count = len(self.mdi_area.subWindowList())
- have_sub_windows = sub_window_count != 0
- self.close_active_window.setEnabled(have_sub_windows)
- self.close_all_windows.setEnabled(have_sub_windows)
- self.tile_windows.setEnabled(have_sub_windows)
- self.cascade_windows.setEnabled(have_sub_windows)
- self.next_window.setEnabled(have_sub_windows)
- self.previous_window.setEnabled(have_sub_windows)
- self.window_menu.addAction(self.close_active_window)
- self.window_menu.addAction(self.close_all_windows)
- self.window_menu.addSeparator()
- self.window_menu.addAction(self.tile_windows)
- self.window_menu.addAction(self.cascade_windows)
- self.window_menu.addSeparator()
- self.window_menu.addAction(self.next_window)
- self.window_menu.addAction(self.previous_window)
- if sub_window_count == 0:
- return
- self.window_menu.addSeparator()
- nr = 1
- for sub_window in self.mdi_area.subWindowList():
- label = str(nr) + " " + sub_window.name
- if nr < 10:
- label = "&" + label
- action = self.window_menu.addAction(label)
- action.setCheckable(True)
- action.setChecked(sub_window == self.mdi_area.activeSubWindow())
- action.triggered.connect(lambda a=None,x=nr: self.setActiveSubWindow(x))
- self.window_menu.addAction(action)
- nr += 1
- def setActiveSubWindow(self, nr):
- self.mdi_area.setActiveSubWindow(self.mdi_area.subWindowList()[nr - 1])
- # Help text
- glb_help_text = """
- <h1>Contents</h1>
- <style>
- p.c1 {
- text-indent: 40px;
- }
- p.c2 {
- text-indent: 80px;
- }
- }
- </style>
- <p class=c1><a href=#reports>1. Reports</a></p>
- <p class=c2><a href=#callgraph>1.1 Context-Sensitive Call Graph</a></p>
- <p class=c2><a href=#calltree>1.2 Call Tree</a></p>
- <p class=c2><a href=#allbranches>1.3 All branches</a></p>
- <p class=c2><a href=#selectedbranches>1.4 Selected branches</a></p>
- <p class=c2><a href=#topcallsbyelapsedtime>1.5 Top calls by elapsed time</a></p>
- <p class=c1><a href=#charts>2. Charts</a></p>
- <p class=c2><a href=#timechartbycpu>2.1 Time chart by CPU</a></p>
- <p class=c1><a href=#tables>3. Tables</a></p>
- <h1 id=reports>1. Reports</h1>
- <h2 id=callgraph>1.1 Context-Sensitive Call Graph</h2>
- The result is a GUI window with a tree representing a context-sensitive
- call-graph. Expanding a couple of levels of the tree and adjusting column
- widths to suit will display something like:
- <pre>
- Call Graph: pt_example
- Call Path Object Count Time(ns) Time(%) Branch Count Branch Count(%)
- v- ls
- v- 2638:2638
- v- _start ld-2.19.so 1 10074071 100.0 211135 100.0
- |- unknown unknown 1 13198 0.1 1 0.0
- >- _dl_start ld-2.19.so 1 1400980 13.9 19637 9.3
- >- _d_linit_internal ld-2.19.so 1 448152 4.4 11094 5.3
- v-__libc_start_main@plt ls 1 8211741 81.5 180397 85.4
- >- _dl_fixup ld-2.19.so 1 7607 0.1 108 0.1
- >- __cxa_atexit libc-2.19.so 1 11737 0.1 10 0.0
- >- __libc_csu_init ls 1 10354 0.1 10 0.0
- |- _setjmp libc-2.19.so 1 0 0.0 4 0.0
- v- main ls 1 8182043 99.6 180254 99.9
- </pre>
- <h3>Points to note:</h3>
- <ul>
- <li>The top level is a command name (comm)</li>
- <li>The next level is a thread (pid:tid)</li>
- <li>Subsequent levels are functions</li>
- <li>'Count' is the number of calls</li>
- <li>'Time' is the elapsed time until the function returns</li>
- <li>Percentages are relative to the level above</li>
- <li>'Branch Count' is the total number of branches for that function and all functions that it calls
- </ul>
- <h3>Find</h3>
- Ctrl-F displays a Find bar which finds function names by either an exact match or a pattern match.
- The pattern matching symbols are ? for any character and * for zero or more characters.
- <h2 id=calltree>1.2 Call Tree</h2>
- The Call Tree report is very similar to the Context-Sensitive Call Graph, but the data is not aggregated.
- Also the 'Count' column, which would be always 1, is replaced by the 'Call Time'.
- <h2 id=allbranches>1.3 All branches</h2>
- The All branches report displays all branches in chronological order.
- Not all data is fetched immediately. More records can be fetched using the Fetch bar provided.
- <h3>Disassembly</h3>
- Open a branch to display disassembly. This only works if:
- <ol>
- <li>The disassembler is available. Currently, only Intel XED is supported - see <a href=#xed>Intel XED Setup</a></li>
- <li>The object code is available. Currently, only the perf build ID cache is searched for object code.
- The default directory ~/.debug can be overridden by setting environment variable PERF_BUILDID_DIR.
- One exception is kcore where the DSO long name is used (refer dsos_view on the Tables menu),
- or alternatively, set environment variable PERF_KCORE to the kcore file name.</li>
- </ol>
- <h4 id=xed>Intel XED Setup</h4>
- To use Intel XED, libxed.so must be present. To build and install libxed.so:
- <pre>
- git clone https://github.com/intelxed/mbuild.git mbuild
- git clone https://github.com/intelxed/xed
- cd xed
- ./mfile.py --share
- sudo ./mfile.py --prefix=/usr/local install
- sudo ldconfig
- </pre>
- <h3>Instructions per Cycle (IPC)</h3>
- If available, IPC information is displayed in columns 'insn_cnt', 'cyc_cnt' and 'IPC'.
- <p><b>Intel PT note:</b> The information applies to the blocks of code ending with, and including, that branch.
- Due to the granularity of timing information, the number of cycles for some code blocks will not be known.
- In that case, 'insn_cnt', 'cyc_cnt' and 'IPC' are zero, but when 'IPC' is displayed it covers the period
- since the previous displayed 'IPC'.
- <h3>Find</h3>
- Ctrl-F displays a Find bar which finds substrings by either an exact match or a regular expression match.
- Refer to Python documentation for the regular expression syntax.
- All columns are searched, but only currently fetched rows are searched.
- <h2 id=selectedbranches>1.4 Selected branches</h2>
- This is the same as the <a href=#allbranches>All branches</a> report but with the data reduced
- by various selection criteria. A dialog box displays available criteria which are AND'ed together.
- <h3>1.4.1 Time ranges</h3>
- The time ranges hint text shows the total time range. Relative time ranges can also be entered in
- ms, us or ns. Also, negative values are relative to the end of trace. Examples:
- <pre>
- 81073085947329-81073085958238 From 81073085947329 to 81073085958238
- 100us-200us From 100us to 200us
- 10ms- From 10ms to the end
- -100ns The first 100ns
- -10ms- The last 10ms
- </pre>
- N.B. Due to the granularity of timestamps, there could be no branches in any given time range.
- <h2 id=topcallsbyelapsedtime>1.5 Top calls by elapsed time</h2>
- The Top calls by elapsed time report displays calls in descending order of time elapsed between when the function was called and when it returned.
- The data is reduced by various selection criteria. A dialog box displays available criteria which are AND'ed together.
- If not all data is fetched, a Fetch bar is provided. Ctrl-F displays a Find bar.
- <h1 id=charts>2. Charts</h1>
- <h2 id=timechartbycpu>2.1 Time chart by CPU</h2>
- This chart displays context switch information when that data is available. Refer to context_switches_view on the Tables menu.
- <h3>Features</h3>
- <ol>
- <li>Mouse over to highight the task and show the time</li>
- <li>Drag the mouse to select a region and zoom by pushing the Zoom button</li>
- <li>Go back and forward by pressing the arrow buttons</li>
- <li>If call information is available, right-click to show a call tree opened to that task and time.
- Note, the call tree may take some time to appear, and there may not be call information for the task or time selected.
- </li>
- </ol>
- <h3>Important</h3>
- The graph can be misleading in the following respects:
- <ol>
- <li>The graph shows the first task on each CPU as running from the beginning of the time range.
- Because tracing might start on different CPUs at different times, that is not necessarily the case.
- Refer to context_switches_view on the Tables menu to understand what data the graph is based upon.</li>
- <li>Similarly, the last task on each CPU can be showing running longer than it really was.
- Again, refer to context_switches_view on the Tables menu to understand what data the graph is based upon.</li>
- <li>When the mouse is over a task, the highlighted task might not be visible on the legend without scrolling if the legend does not fit fully in the window</li>
- </ol>
- <h1 id=tables>3. Tables</h1>
- The Tables menu shows all tables and views in the database. Most tables have an associated view
- which displays the information in a more friendly way. Not all data for large tables is fetched
- immediately. More records can be fetched using the Fetch bar provided. Columns can be sorted,
- but that can be slow for large tables.
- <p>There are also tables of database meta-information.
- For SQLite3 databases, the sqlite_master table is included.
- For PostgreSQL databases, information_schema.tables/views/columns are included.
- <h3>Find</h3>
- Ctrl-F displays a Find bar which finds substrings by either an exact match or a regular expression match.
- Refer to Python documentation for the regular expression syntax.
- All columns are searched, but only currently fetched rows are searched.
- <p>N.B. Results are found in id order, so if the table is re-ordered, find-next and find-previous
- will go to the next/previous result in id order, instead of display order.
- """
- # Help window
- class HelpWindow(QMdiSubWindow):
- def __init__(self, glb, parent=None):
- super(HelpWindow, self).__init__(parent)
- self.text = QTextBrowser()
- self.text.setHtml(glb_help_text)
- self.text.setReadOnly(True)
- self.text.setOpenExternalLinks(True)
- self.setWidget(self.text)
- AddSubWindow(glb.mainwindow.mdi_area, self, "Exported SQL Viewer Help")
- # Main window that only displays the help text
- class HelpOnlyWindow(QMainWindow):
- def __init__(self, parent=None):
- super(HelpOnlyWindow, self).__init__(parent)
- self.setMinimumSize(200, 100)
- self.resize(800, 600)
- self.setWindowTitle("Exported SQL Viewer Help")
- self.setWindowIcon(self.style().standardIcon(QStyle.SP_MessageBoxInformation))
- self.text = QTextBrowser()
- self.text.setHtml(glb_help_text)
- self.text.setReadOnly(True)
- self.text.setOpenExternalLinks(True)
- self.setCentralWidget(self.text)
- # PostqreSQL server version
- def PostqreSQLServerVersion(db):
- query = QSqlQuery(db)
- QueryExec(query, "SELECT VERSION()")
- if query.next():
- v_str = query.value(0)
- v_list = v_str.strip().split(" ")
- if v_list[0] == "PostgreSQL" and v_list[2] == "on":
- return v_list[1]
- return v_str
- return "Unknown"
- # SQLite version
- def SQLiteVersion(db):
- query = QSqlQuery(db)
- QueryExec(query, "SELECT sqlite_version()")
- if query.next():
- return query.value(0)
- return "Unknown"
- # About dialog
- class AboutDialog(QDialog):
- def __init__(self, glb, parent=None):
- super(AboutDialog, self).__init__(parent)
- self.setWindowTitle("About Exported SQL Viewer")
- self.setMinimumWidth(300)
- pyside_version = "1" if pyside_version_1 else "2"
- text = "<pre>"
- text += "Python version: " + sys.version.split(" ")[0] + "\n"
- text += "PySide version: " + pyside_version + "\n"
- text += "Qt version: " + qVersion() + "\n"
- if glb.dbref.is_sqlite3:
- text += "SQLite version: " + SQLiteVersion(glb.db) + "\n"
- else:
- text += "PostqreSQL version: " + PostqreSQLServerVersion(glb.db) + "\n"
- text += "</pre>"
- self.text = QTextBrowser()
- self.text.setHtml(text)
- self.text.setReadOnly(True)
- self.text.setOpenExternalLinks(True)
- self.vbox = QVBoxLayout()
- self.vbox.addWidget(self.text)
- self.setLayout(self.vbox)
- # Font resize
- def ResizeFont(widget, diff):
- font = widget.font()
- sz = font.pointSize()
- font.setPointSize(sz + diff)
- widget.setFont(font)
- def ShrinkFont(widget):
- ResizeFont(widget, -1)
- def EnlargeFont(widget):
- ResizeFont(widget, 1)
- # Unique name for sub-windows
- def NumberedWindowName(name, nr):
- if nr > 1:
- name += " <" + str(nr) + ">"
- return name
- def UniqueSubWindowName(mdi_area, name):
- nr = 1
- while True:
- unique_name = NumberedWindowName(name, nr)
- ok = True
- for sub_window in mdi_area.subWindowList():
- if sub_window.name == unique_name:
- ok = False
- break
- if ok:
- return unique_name
- nr += 1
- # Add a sub-window
- def AddSubWindow(mdi_area, sub_window, name):
- unique_name = UniqueSubWindowName(mdi_area, name)
- sub_window.setMinimumSize(200, 100)
- sub_window.resize(800, 600)
- sub_window.setWindowTitle(unique_name)
- sub_window.setAttribute(Qt.WA_DeleteOnClose)
- sub_window.setWindowIcon(sub_window.style().standardIcon(QStyle.SP_FileIcon))
- sub_window.name = unique_name
- mdi_area.addSubWindow(sub_window)
- sub_window.show()
- # Main window
- class MainWindow(QMainWindow):
- def __init__(self, glb, parent=None):
- super(MainWindow, self).__init__(parent)
- self.glb = glb
- self.setWindowTitle("Exported SQL Viewer: " + glb.dbname)
- self.setWindowIcon(self.style().standardIcon(QStyle.SP_ComputerIcon))
- self.setMinimumSize(200, 100)
- self.mdi_area = QMdiArea()
- self.mdi_area.setHorizontalScrollBarPolicy(Qt.ScrollBarAsNeeded)
- self.mdi_area.setVerticalScrollBarPolicy(Qt.ScrollBarAsNeeded)
- self.setCentralWidget(self.mdi_area)
- menu = self.menuBar()
- file_menu = menu.addMenu("&File")
- file_menu.addAction(CreateExitAction(glb.app, self))
- edit_menu = menu.addMenu("&Edit")
- edit_menu.addAction(CreateAction("&Copy", "Copy to clipboard", self.CopyToClipboard, self, QKeySequence.Copy))
- edit_menu.addAction(CreateAction("Copy as CS&V", "Copy to clipboard as CSV", self.CopyToClipboardCSV, self))
- edit_menu.addAction(CreateAction("&Find...", "Find items", self.Find, self, QKeySequence.Find))
- edit_menu.addAction(CreateAction("Fetch &more records...", "Fetch more records", self.FetchMoreRecords, self, [QKeySequence(Qt.Key_F8)]))
- edit_menu.addAction(CreateAction("&Shrink Font", "Make text smaller", self.ShrinkFont, self, [QKeySequence("Ctrl+-")]))
- edit_menu.addAction(CreateAction("&Enlarge Font", "Make text bigger", self.EnlargeFont, self, [QKeySequence("Ctrl++")]))
- reports_menu = menu.addMenu("&Reports")
- if IsSelectable(glb.db, "calls"):
- reports_menu.addAction(CreateAction("Context-Sensitive Call &Graph", "Create a new window containing a context-sensitive call graph", self.NewCallGraph, self))
- if IsSelectable(glb.db, "calls", "WHERE parent_id >= 0"):
- reports_menu.addAction(CreateAction("Call &Tree", "Create a new window containing a call tree", self.NewCallTree, self))
- self.EventMenu(GetEventList(glb.db), reports_menu)
- if IsSelectable(glb.db, "calls"):
- reports_menu.addAction(CreateAction("&Top calls by elapsed time", "Create a new window displaying top calls by elapsed time", self.NewTopCalls, self))
- if IsSelectable(glb.db, "context_switches"):
- charts_menu = menu.addMenu("&Charts")
- charts_menu.addAction(CreateAction("&Time chart by CPU", "Create a new window displaying time charts by CPU", self.TimeChartByCPU, self))
- self.TableMenu(GetTableList(glb), menu)
- self.window_menu = WindowMenu(self.mdi_area, menu)
- help_menu = menu.addMenu("&Help")
- help_menu.addAction(CreateAction("&Exported SQL Viewer Help", "Helpful information", self.Help, self, QKeySequence.HelpContents))
- help_menu.addAction(CreateAction("&About Exported SQL Viewer", "About this application", self.About, self))
- def Try(self, fn):
- win = self.mdi_area.activeSubWindow()
- if win:
- try:
- fn(win.view)
- except:
- pass
- def CopyToClipboard(self):
- self.Try(CopyCellsToClipboardHdr)
- def CopyToClipboardCSV(self):
- self.Try(CopyCellsToClipboardCSV)
- def Find(self):
- win = self.mdi_area.activeSubWindow()
- if win:
- try:
- win.find_bar.Activate()
- except:
- pass
- def FetchMoreRecords(self):
- win = self.mdi_area.activeSubWindow()
- if win:
- try:
- win.fetch_bar.Activate()
- except:
- pass
- def ShrinkFont(self):
- self.Try(ShrinkFont)
- def EnlargeFont(self):
- self.Try(EnlargeFont)
- def EventMenu(self, events, reports_menu):
- branches_events = 0
- for event in events:
- event = event.split(":")[0]
- if event == "branches":
- branches_events += 1
- dbid = 0
- for event in events:
- dbid += 1
- event = event.split(":")[0]
- if event == "branches":
- label = "All branches" if branches_events == 1 else "All branches " + "(id=" + dbid + ")"
- reports_menu.addAction(CreateAction(label, "Create a new window displaying branch events", lambda a=None,x=dbid: self.NewBranchView(x), self))
- label = "Selected branches" if branches_events == 1 else "Selected branches " + "(id=" + dbid + ")"
- reports_menu.addAction(CreateAction(label, "Create a new window displaying branch events", lambda a=None,x=dbid: self.NewSelectedBranchView(x), self))
- def TimeChartByCPU(self):
- TimeChartByCPUWindow(self.glb, self)
- def TableMenu(self, tables, menu):
- table_menu = menu.addMenu("&Tables")
- for table in tables:
- table_menu.addAction(CreateAction(table, "Create a new window containing a table view", lambda a=None,t=table: self.NewTableView(t), self))
- def NewCallGraph(self):
- CallGraphWindow(self.glb, self)
- def NewCallTree(self):
- CallTreeWindow(self.glb, self)
- def NewTopCalls(self):
- dialog = TopCallsDialog(self.glb, self)
- ret = dialog.exec_()
- if ret:
- TopCallsWindow(self.glb, dialog.report_vars, self)
- def NewBranchView(self, event_id):
- BranchWindow(self.glb, event_id, ReportVars(), self)
- def NewSelectedBranchView(self, event_id):
- dialog = SelectedBranchDialog(self.glb, self)
- ret = dialog.exec_()
- if ret:
- BranchWindow(self.glb, event_id, dialog.report_vars, self)
- def NewTableView(self, table_name):
- TableWindow(self.glb, table_name, self)
- def Help(self):
- HelpWindow(self.glb, self)
- def About(self):
- dialog = AboutDialog(self.glb, self)
- dialog.exec_()
- def TryOpen(file_name):
- try:
- return open(file_name, "rb")
- except:
- return None
- def Is64Bit(f):
- result = sizeof(c_void_p)
- # ELF support only
- pos = f.tell()
- f.seek(0)
- header = f.read(7)
- f.seek(pos)
- magic = header[0:4]
- if sys.version_info[0] == 2:
- eclass = ord(header[4])
- encoding = ord(header[5])
- version = ord(header[6])
- else:
- eclass = header[4]
- encoding = header[5]
- version = header[6]
- if magic == chr(127) + "ELF" and eclass > 0 and eclass < 3 and encoding > 0 and encoding < 3 and version == 1:
- result = True if eclass == 2 else False
- return result
- # Global data
- class Glb():
- def __init__(self, dbref, db, dbname):
- self.dbref = dbref
- self.db = db
- self.dbname = dbname
- self.home_dir = os.path.expanduser("~")
- self.buildid_dir = os.getenv("PERF_BUILDID_DIR")
- if self.buildid_dir:
- self.buildid_dir += "/.build-id/"
- else:
- self.buildid_dir = self.home_dir + "/.debug/.build-id/"
- self.app = None
- self.mainwindow = None
- self.instances_to_shutdown_on_exit = weakref.WeakSet()
- try:
- self.disassembler = LibXED()
- self.have_disassembler = True
- except:
- self.have_disassembler = False
- self.host_machine_id = 0
- self.host_start_time = 0
- self.host_finish_time = 0
- def FileFromBuildId(self, build_id):
- file_name = self.buildid_dir + build_id[0:2] + "/" + build_id[2:] + "/elf"
- return TryOpen(file_name)
- def FileFromNamesAndBuildId(self, short_name, long_name, build_id):
- # Assume current machine i.e. no support for virtualization
- if short_name[0:7] == "[kernel" and os.path.basename(long_name) == "kcore":
- file_name = os.getenv("PERF_KCORE")
- f = TryOpen(file_name) if file_name else None
- if f:
- return f
- # For now, no special handling if long_name is /proc/kcore
- f = TryOpen(long_name)
- if f:
- return f
- f = self.FileFromBuildId(build_id)
- if f:
- return f
- return None
- def AddInstanceToShutdownOnExit(self, instance):
- self.instances_to_shutdown_on_exit.add(instance)
- # Shutdown any background processes or threads
- def ShutdownInstances(self):
- for x in self.instances_to_shutdown_on_exit:
- try:
- x.Shutdown()
- except:
- pass
- def GetHostMachineId(self):
- query = QSqlQuery(self.db)
- QueryExec(query, "SELECT id FROM machines WHERE pid = -1")
- if query.next():
- self.host_machine_id = query.value(0)
- else:
- self.host_machine_id = 0
- return self.host_machine_id
- def HostMachineId(self):
- if self.host_machine_id:
- return self.host_machine_id
- return self.GetHostMachineId()
- def SelectValue(self, sql):
- query = QSqlQuery(self.db)
- try:
- QueryExec(query, sql)
- except:
- return None
- if query.next():
- return Decimal(query.value(0))
- return None
- def SwitchesMinTime(self, machine_id):
- return self.SelectValue("SELECT time"
- " FROM context_switches"
- " WHERE time != 0 AND machine_id = " + str(machine_id) +
- " ORDER BY id LIMIT 1")
- def SwitchesMaxTime(self, machine_id):
- return self.SelectValue("SELECT time"
- " FROM context_switches"
- " WHERE time != 0 AND machine_id = " + str(machine_id) +
- " ORDER BY id DESC LIMIT 1")
- def SamplesMinTime(self, machine_id):
- return self.SelectValue("SELECT time"
- " FROM samples"
- " WHERE time != 0 AND machine_id = " + str(machine_id) +
- " ORDER BY id LIMIT 1")
- def SamplesMaxTime(self, machine_id):
- return self.SelectValue("SELECT time"
- " FROM samples"
- " WHERE time != 0 AND machine_id = " + str(machine_id) +
- " ORDER BY id DESC LIMIT 1")
- def CallsMinTime(self, machine_id):
- return self.SelectValue("SELECT calls.call_time"
- " FROM calls"
- " INNER JOIN threads ON threads.thread_id = calls.thread_id"
- " WHERE calls.call_time != 0 AND threads.machine_id = " + str(machine_id) +
- " ORDER BY calls.id LIMIT 1")
- def CallsMaxTime(self, machine_id):
- return self.SelectValue("SELECT calls.return_time"
- " FROM calls"
- " INNER JOIN threads ON threads.thread_id = calls.thread_id"
- " WHERE calls.return_time != 0 AND threads.machine_id = " + str(machine_id) +
- " ORDER BY calls.return_time DESC LIMIT 1")
- def GetStartTime(self, machine_id):
- t0 = self.SwitchesMinTime(machine_id)
- t1 = self.SamplesMinTime(machine_id)
- t2 = self.CallsMinTime(machine_id)
- if t0 is None or (not(t1 is None) and t1 < t0):
- t0 = t1
- if t0 is None or (not(t2 is None) and t2 < t0):
- t0 = t2
- return t0
- def GetFinishTime(self, machine_id):
- t0 = self.SwitchesMaxTime(machine_id)
- t1 = self.SamplesMaxTime(machine_id)
- t2 = self.CallsMaxTime(machine_id)
- if t0 is None or (not(t1 is None) and t1 > t0):
- t0 = t1
- if t0 is None or (not(t2 is None) and t2 > t0):
- t0 = t2
- return t0
- def HostStartTime(self):
- if self.host_start_time:
- return self.host_start_time
- self.host_start_time = self.GetStartTime(self.HostMachineId())
- return self.host_start_time
- def HostFinishTime(self):
- if self.host_finish_time:
- return self.host_finish_time
- self.host_finish_time = self.GetFinishTime(self.HostMachineId())
- return self.host_finish_time
- def StartTime(self, machine_id):
- if machine_id == self.HostMachineId():
- return self.HostStartTime()
- return self.GetStartTime(machine_id)
- def FinishTime(self, machine_id):
- if machine_id == self.HostMachineId():
- return self.HostFinishTime()
- return self.GetFinishTime(machine_id)
- # Database reference
- class DBRef():
- def __init__(self, is_sqlite3, dbname):
- self.is_sqlite3 = is_sqlite3
- self.dbname = dbname
- self.TRUE = "TRUE"
- self.FALSE = "FALSE"
- # SQLite prior to version 3.23 does not support TRUE and FALSE
- if self.is_sqlite3:
- self.TRUE = "1"
- self.FALSE = "0"
- def Open(self, connection_name):
- dbname = self.dbname
- if self.is_sqlite3:
- db = QSqlDatabase.addDatabase("QSQLITE", connection_name)
- else:
- db = QSqlDatabase.addDatabase("QPSQL", connection_name)
- opts = dbname.split()
- for opt in opts:
- if "=" in opt:
- opt = opt.split("=")
- if opt[0] == "hostname":
- db.setHostName(opt[1])
- elif opt[0] == "port":
- db.setPort(int(opt[1]))
- elif opt[0] == "username":
- db.setUserName(opt[1])
- elif opt[0] == "password":
- db.setPassword(opt[1])
- elif opt[0] == "dbname":
- dbname = opt[1]
- else:
- dbname = opt
- db.setDatabaseName(dbname)
- if not db.open():
- raise Exception("Failed to open database " + dbname + " error: " + db.lastError().text())
- return db, dbname
- # Main
- def Main():
- usage_str = "exported-sql-viewer.py [--pyside-version-1] <database name>\n" \
- " or: exported-sql-viewer.py --help-only"
- ap = argparse.ArgumentParser(usage = usage_str, add_help = False)
- ap.add_argument("--pyside-version-1", action='store_true')
- ap.add_argument("dbname", nargs="?")
- ap.add_argument("--help-only", action='store_true')
- args = ap.parse_args()
- if args.help_only:
- app = QApplication(sys.argv)
- mainwindow = HelpOnlyWindow()
- mainwindow.show()
- err = app.exec_()
- sys.exit(err)
- dbname = args.dbname
- if dbname is None:
- ap.print_usage()
- print("Too few arguments")
- sys.exit(1)
- is_sqlite3 = False
- try:
- f = open(dbname, "rb")
- if f.read(15) == b'SQLite format 3':
- is_sqlite3 = True
- f.close()
- except:
- pass
- dbref = DBRef(is_sqlite3, dbname)
- db, dbname = dbref.Open("main")
- glb = Glb(dbref, db, dbname)
- app = QApplication(sys.argv)
- glb.app = app
- mainwindow = MainWindow(glb)
- glb.mainwindow = mainwindow
- mainwindow.show()
- err = app.exec_()
- glb.ShutdownInstances()
- db.close()
- sys.exit(err)
- if __name__ == "__main__":
- Main()
|