DxbcConverter.cpp 291 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771377237733774377537763777377837793780378137823783378437853786378737883789379037913792379337943795379637973798379938003801380238033804380538063807380838093810381138123813381438153816381738183819382038213822382338243825382638273828382938303831383238333834383538363837383838393840384138423843384438453846384738483849385038513852385338543855385638573858385938603861386238633864386538663867386838693870387138723873387438753876387738783879388038813882388338843885388638873888388938903891389238933894389538963897389838993900390139023903390439053906390739083909391039113912391339143915391639173918391939203921392239233924392539263927392839293930393139323933393439353936393739383939394039413942394339443945394639473948394939503951395239533954395539563957395839593960396139623963396439653966396739683969397039713972397339743975397639773978397939803981398239833984398539863987398839893990399139923993399439953996399739983999400040014002400340044005400640074008400940104011401240134014401540164017401840194020402140224023402440254026402740284029403040314032403340344035403640374038403940404041404240434044404540464047404840494050405140524053405440554056405740584059406040614062406340644065406640674068406940704071407240734074407540764077407840794080408140824083408440854086408740884089409040914092409340944095409640974098409941004101410241034104410541064107410841094110411141124113411441154116411741184119412041214122412341244125412641274128412941304131413241334134413541364137413841394140414141424143414441454146414741484149415041514152415341544155415641574158415941604161416241634164416541664167416841694170417141724173417441754176417741784179418041814182418341844185418641874188418941904191419241934194419541964197419841994200420142024203420442054206420742084209421042114212421342144215421642174218421942204221422242234224422542264227422842294230423142324233423442354236423742384239424042414242424342444245424642474248424942504251425242534254425542564257425842594260426142624263426442654266426742684269427042714272427342744275427642774278427942804281428242834284428542864287428842894290429142924293429442954296429742984299430043014302430343044305430643074308430943104311431243134314431543164317431843194320432143224323432443254326432743284329433043314332433343344335433643374338433943404341434243434344434543464347434843494350435143524353435443554356435743584359436043614362436343644365436643674368436943704371437243734374437543764377437843794380438143824383438443854386438743884389439043914392439343944395439643974398439944004401440244034404440544064407440844094410441144124413441444154416441744184419442044214422442344244425442644274428442944304431443244334434443544364437443844394440444144424443444444454446444744484449445044514452445344544455445644574458445944604461446244634464446544664467446844694470447144724473447444754476447744784479448044814482448344844485448644874488448944904491449244934494449544964497449844994500450145024503450445054506450745084509451045114512451345144515451645174518451945204521452245234524452545264527452845294530453145324533453445354536453745384539454045414542454345444545454645474548454945504551455245534554455545564557455845594560456145624563456445654566456745684569457045714572457345744575457645774578457945804581458245834584458545864587458845894590459145924593459445954596459745984599460046014602460346044605460646074608460946104611461246134614461546164617461846194620462146224623462446254626462746284629463046314632463346344635463646374638463946404641464246434644464546464647464846494650465146524653465446554656465746584659466046614662466346644665466646674668466946704671467246734674467546764677467846794680468146824683468446854686468746884689469046914692469346944695469646974698469947004701470247034704470547064707470847094710471147124713471447154716471747184719472047214722472347244725472647274728472947304731473247334734473547364737473847394740474147424743474447454746474747484749475047514752475347544755475647574758475947604761476247634764476547664767476847694770477147724773477447754776477747784779478047814782478347844785478647874788478947904791479247934794479547964797479847994800480148024803480448054806480748084809481048114812481348144815481648174818481948204821482248234824482548264827482848294830483148324833483448354836483748384839484048414842484348444845484648474848484948504851485248534854485548564857485848594860486148624863486448654866486748684869487048714872487348744875487648774878487948804881488248834884488548864887488848894890489148924893489448954896489748984899490049014902490349044905490649074908490949104911491249134914491549164917491849194920492149224923492449254926492749284929493049314932493349344935493649374938493949404941494249434944494549464947494849494950495149524953495449554956495749584959496049614962496349644965496649674968496949704971497249734974497549764977497849794980498149824983498449854986498749884989499049914992499349944995499649974998499950005001500250035004500550065007500850095010501150125013501450155016501750185019502050215022502350245025502650275028502950305031503250335034503550365037503850395040504150425043504450455046504750485049505050515052505350545055505650575058505950605061506250635064506550665067506850695070507150725073507450755076507750785079508050815082508350845085508650875088508950905091509250935094509550965097509850995100510151025103510451055106510751085109511051115112511351145115511651175118511951205121512251235124512551265127512851295130513151325133513451355136513751385139514051415142514351445145514651475148514951505151515251535154515551565157515851595160516151625163516451655166516751685169517051715172517351745175517651775178517951805181518251835184518551865187518851895190519151925193519451955196519751985199520052015202520352045205520652075208520952105211521252135214521552165217521852195220522152225223522452255226522752285229523052315232523352345235523652375238523952405241524252435244524552465247524852495250525152525253525452555256525752585259526052615262526352645265526652675268526952705271527252735274527552765277527852795280528152825283528452855286528752885289529052915292529352945295529652975298529953005301530253035304530553065307530853095310531153125313531453155316531753185319532053215322532353245325532653275328532953305331533253335334533553365337533853395340534153425343534453455346534753485349535053515352535353545355535653575358535953605361536253635364536553665367536853695370537153725373537453755376537753785379538053815382538353845385538653875388538953905391539253935394539553965397539853995400540154025403540454055406540754085409541054115412541354145415541654175418541954205421542254235424542554265427542854295430543154325433543454355436543754385439544054415442544354445445544654475448544954505451545254535454545554565457545854595460546154625463546454655466546754685469547054715472547354745475547654775478547954805481548254835484548554865487548854895490549154925493549454955496549754985499550055015502550355045505550655075508550955105511551255135514551555165517551855195520552155225523552455255526552755285529553055315532553355345535553655375538553955405541554255435544554555465547554855495550555155525553555455555556555755585559556055615562556355645565556655675568556955705571557255735574557555765577557855795580558155825583558455855586558755885589559055915592559355945595559655975598559956005601560256035604560556065607560856095610561156125613561456155616561756185619562056215622562356245625562656275628562956305631563256335634563556365637563856395640564156425643564456455646564756485649565056515652565356545655565656575658565956605661566256635664566556665667566856695670567156725673567456755676567756785679568056815682568356845685568656875688568956905691569256935694569556965697569856995700570157025703570457055706570757085709571057115712571357145715571657175718571957205721572257235724572557265727572857295730573157325733573457355736573757385739574057415742574357445745574657475748574957505751575257535754575557565757575857595760576157625763576457655766576757685769577057715772577357745775577657775778577957805781578257835784578557865787578857895790579157925793579457955796579757985799580058015802580358045805580658075808580958105811581258135814581558165817581858195820582158225823582458255826582758285829583058315832583358345835583658375838583958405841584258435844584558465847584858495850585158525853585458555856585758585859586058615862586358645865586658675868586958705871587258735874587558765877587858795880588158825883588458855886588758885889589058915892589358945895589658975898589959005901590259035904590559065907590859095910591159125913591459155916591759185919592059215922592359245925592659275928592959305931593259335934593559365937593859395940594159425943594459455946594759485949595059515952595359545955595659575958595959605961596259635964596559665967596859695970597159725973597459755976597759785979598059815982598359845985598659875988598959905991599259935994599559965997599859996000600160026003600460056006600760086009601060116012601360146015601660176018601960206021602260236024602560266027602860296030603160326033603460356036603760386039604060416042604360446045604660476048604960506051605260536054605560566057605860596060606160626063606460656066606760686069607060716072607360746075607660776078607960806081608260836084608560866087608860896090609160926093609460956096609760986099610061016102610361046105610661076108610961106111611261136114611561166117611861196120612161226123612461256126612761286129613061316132613361346135613661376138613961406141614261436144614561466147614861496150615161526153615461556156615761586159616061616162616361646165616661676168616961706171617261736174617561766177617861796180618161826183618461856186618761886189619061916192619361946195619661976198619962006201620262036204620562066207620862096210621162126213621462156216621762186219622062216222622362246225622662276228622962306231623262336234623562366237623862396240624162426243624462456246624762486249625062516252625362546255625662576258625962606261626262636264626562666267626862696270627162726273627462756276627762786279628062816282628362846285628662876288628962906291629262936294629562966297629862996300630163026303630463056306630763086309631063116312631363146315631663176318631963206321632263236324632563266327632863296330633163326333633463356336633763386339634063416342634363446345634663476348634963506351635263536354635563566357635863596360636163626363636463656366636763686369637063716372637363746375637663776378637963806381638263836384638563866387638863896390639163926393639463956396639763986399640064016402640364046405640664076408640964106411641264136414641564166417641864196420642164226423642464256426642764286429643064316432643364346435643664376438643964406441644264436444644564466447644864496450645164526453645464556456645764586459646064616462646364646465646664676468646964706471647264736474647564766477647864796480648164826483648464856486648764886489649064916492649364946495649664976498649965006501650265036504650565066507650865096510651165126513651465156516651765186519652065216522652365246525652665276528652965306531653265336534653565366537653865396540654165426543654465456546654765486549655065516552655365546555655665576558655965606561656265636564656565666567656865696570657165726573657465756576657765786579658065816582658365846585658665876588658965906591659265936594659565966597659865996600660166026603660466056606660766086609661066116612661366146615661666176618661966206621662266236624662566266627662866296630663166326633663466356636663766386639664066416642664366446645664666476648664966506651665266536654665566566657665866596660666166626663666466656666666766686669667066716672667366746675667666776678667966806681668266836684668566866687668866896690669166926693669466956696669766986699670067016702670367046705670667076708670967106711671267136714671567166717671867196720672167226723672467256726672767286729673067316732673367346735673667376738673967406741674267436744674567466747674867496750675167526753675467556756675767586759676067616762676367646765676667676768676967706771677267736774677567766777677867796780678167826783678467856786678767886789679067916792679367946795679667976798679968006801680268036804680568066807680868096810681168126813681468156816681768186819682068216822682368246825682668276828682968306831683268336834683568366837683868396840684168426843684468456846684768486849685068516852685368546855685668576858685968606861686268636864686568666867686868696870687168726873687468756876687768786879688068816882688368846885688668876888688968906891689268936894689568966897689868996900690169026903690469056906690769086909691069116912691369146915691669176918691969206921692269236924692569266927692869296930693169326933693469356936693769386939694069416942694369446945694669476948694969506951695269536954695569566957695869596960696169626963696469656966696769686969697069716972697369746975697669776978697969806981698269836984698569866987698869896990699169926993699469956996699769986999700070017002700370047005700670077008700970107011701270137014701570167017701870197020702170227023702470257026702770287029703070317032703370347035703670377038703970407041704270437044704570467047704870497050705170527053705470557056705770587059706070617062706370647065706670677068706970707071707270737074707570767077707870797080708170827083708470857086708770887089709070917092709370947095709670977098709971007101710271037104710571067107710871097110711171127113711471157116711771187119712071217122712371247125712671277128712971307131713271337134713571367137713871397140714171427143714471457146714771487149715071517152715371547155715671577158715971607161716271637164716571667167716871697170717171727173717471757176717771787179718071817182718371847185718671877188718971907191719271937194719571967197719871997200720172027203720472057206720772087209721072117212721372147215721672177218721972207221722272237224722572267227722872297230723172327233723472357236723772387239724072417242724372447245724672477248724972507251725272537254725572567257725872597260726172627263726472657266726772687269727072717272727372747275727672777278727972807281728272837284728572867287728872897290729172927293729472957296729772987299730073017302730373047305730673077308730973107311731273137314731573167317731873197320732173227323732473257326732773287329733073317332733373347335733673377338733973407341734273437344734573467347734873497350735173527353735473557356735773587359736073617362736373647365
  1. ///////////////////////////////////////////////////////////////////////////////
  2. // //
  3. // DxbcConverter.cpp //
  4. // Copyright (C) Microsoft Corporation. All rights reserved. //
  5. // This file is distributed under the University of Illinois Open Source //
  6. // License. See LICENSE.TXT for details. //
  7. // //
  8. // Implements the DirectX DXBC to DXIL converter. //
  9. // //
  10. ///////////////////////////////////////////////////////////////////////////////
  11. #include "llvm/Support/Debug.h"
  12. #include "DxbcConverterImpl.h"
  13. #include "DxilConvPasses/DxilCleanup.h"
  14. #include "dxc/DxilContainer/DxilContainer.h"
  15. #include "dxc/DxilContainer/DxilContainerAssembler.h"
  16. #include "dxc/DxilContainer/DxilContainerReader.h"
  17. #define DXBCCONV_DBG 0
  18. namespace hlsl {
  19. __override HRESULT STDMETHODCALLTYPE DxbcConverter::Convert(_In_reads_bytes_(DxbcSize) LPCVOID pDxbc,
  20. _In_ UINT32 DxbcSize,
  21. _In_opt_z_ LPCWSTR pExtraOptions,
  22. _Outptr_result_bytebuffer_maybenull_(*pDxilSize) LPVOID *ppDxil,
  23. _Out_ UINT32 *pDxilSize,
  24. _Outptr_result_maybenull_z_ LPWSTR *ppDiag) {
  25. DxcThreadMalloc TM(m_pMalloc);
  26. LARGE_INTEGER start, end;
  27. QueryPerformanceCounter(&start);
  28. DxcRuntimeEtw_DxcTranslate_Start();
  29. HRESULT hr = S_OK;
  30. try {
  31. sys::fs::MSFileSystem *pFSPtr;
  32. IFT(CreateMSFileSystemForDisk(&pFSPtr));
  33. unique_ptr<sys::fs::MSFileSystem> pFS(pFSPtr);
  34. sys::fs::AutoPerThreadSystem pTS(pFS.get());
  35. IFTLLVM(pTS.error_code());
  36. struct StdErrFlusher {
  37. ~StdErrFlusher() { dbgs().flush(); }
  38. } S;
  39. ConvertImpl(pDxbc, DxbcSize, pExtraOptions, ppDxil, pDxilSize, ppDiag);
  40. DxcRuntimeEtw_DxcTranslate_TranslateStats(DxbcSize, DxbcSize, (const BYTE *)pDxbc, *pDxilSize);
  41. hr = S_OK;
  42. }
  43. CATCH_CPP_ASSIGN_HRESULT();
  44. DxcRuntimeEtw_DxcTranslate_Stop(hr);
  45. QueryPerformanceCounter(&end);
  46. LogConvertResult(false, &start, &end, pDxbc, DxbcSize, pExtraOptions, *ppDxil, *pDxilSize, hr);
  47. return hr;
  48. }
  49. __override HRESULT STDMETHODCALLTYPE DxbcConverter::ConvertInDriver(_In_reads_bytes_(8) const UINT32 *pBytecode,
  50. _In_opt_z_ LPCVOID pInputSignature,
  51. _In_ UINT32 NumInputSignatureElements,
  52. _In_opt_z_ LPCVOID pOutputSignature,
  53. _In_ UINT32 NumOutputSignatureElements,
  54. _In_opt_z_ LPCVOID pPatchConstantSignature,
  55. _In_ UINT32 NumPatchConstantSignatureElements,
  56. _In_opt_z_ LPCWSTR pExtraOptions,
  57. _Out_ IDxcBlob **ppDxilModule,
  58. _Outptr_result_maybenull_z_ LPWSTR *ppDiag) {
  59. DxcThreadMalloc TM(m_pMalloc);
  60. LARGE_INTEGER start, end;
  61. QueryPerformanceCounter(&start);
  62. DxcRuntimeEtw_DxcTranslate_Start();
  63. HRESULT hr = S_OK;
  64. UINT32 bcSize = pBytecode[1] * sizeof(UINT32);
  65. const BYTE *pDxilBytes = nullptr;
  66. UINT32 DxilByteCount = 0;
  67. try {
  68. sys::fs::MSFileSystem *pFSPtr;
  69. IFT(CreateMSFileSystemForDisk(&pFSPtr));
  70. unique_ptr<sys::fs::MSFileSystem> pFS(pFSPtr);
  71. sys::fs::AutoPerThreadSystem pTS(pFS.get());
  72. IFTLLVM(pTS.error_code());
  73. struct StdErrFlusher {
  74. ~StdErrFlusher() { dbgs().flush(); }
  75. } S;
  76. ConvertInDriverImpl(pBytecode,
  77. (const D3D12DDIARG_SIGNATURE_ENTRY_0012 *)pInputSignature,
  78. NumInputSignatureElements,
  79. (const D3D12DDIARG_SIGNATURE_ENTRY_0012 *)pOutputSignature,
  80. NumOutputSignatureElements,
  81. (const D3D12DDIARG_SIGNATURE_ENTRY_0012 *)pPatchConstantSignature,
  82. NumPatchConstantSignatureElements,
  83. pExtraOptions,
  84. ppDxilModule,
  85. ppDiag);
  86. pDxilBytes = (const BYTE *)(*ppDxilModule)->GetBufferPointer();
  87. DxilByteCount = (*ppDxilModule)->GetBufferSize();
  88. DxcRuntimeEtw_DxcTranslate_TranslateStats(bcSize, bcSize, (const BYTE *)pBytecode, DxilByteCount);
  89. hr = S_OK;
  90. }
  91. CATCH_CPP_ASSIGN_HRESULT();
  92. DxcRuntimeEtw_DxcTranslate_Stop(hr);
  93. QueryPerformanceCounter(&end);
  94. LogConvertResult(true, &start, &end, pBytecode, bcSize, pExtraOptions, pDxilBytes, DxilByteCount, hr);
  95. return hr;
  96. }
  97. DxbcConverter::DxbcConverter()
  98. : m_dwRef(0)
  99. , m_pPR(nullptr)
  100. , m_pOP(nullptr)
  101. , m_pSM(nullptr)
  102. , m_DxbcMajor(0)
  103. , m_DxbcMinor(0)
  104. , m_pUnusedF32(nullptr)
  105. , m_pUnusedI32(nullptr)
  106. , m_NumTempRegs(0)
  107. , m_pIcbGV(nullptr)
  108. , m_bDisableHashCheck(false)
  109. , m_bRunDxilCleanup(true)
  110. , m_bLegacyCBufferLoad(true)
  111. , m_TGSMCount(0)
  112. , m_DepthRegType(D3D10_SB_OPERAND_TYPE_NULL)
  113. , m_bHasStencilRef(false)
  114. , m_bHasCoverageOut(false)
  115. , m_bControlPointPhase(false)
  116. , m_bPatchConstantPhase(false)
  117. , m_pInterfaceDataBuffer(nullptr)
  118. , m_pClassInstanceCBuffers(nullptr)
  119. , m_pClassInstanceSamplers(nullptr)
  120. , m_pClassInstanceComparisonSamplers(nullptr)
  121. , m_NumIfaces(0)
  122. , m_FcallCount(0) {
  123. DXASSERT(OP::CheckOpCodeTable(), "incorrect entry in OpCode property table");
  124. }
  125. DxbcConverter::~DxbcConverter() {
  126. }
  127. static void AddDxilPipelineStateValidationToDXBC( DxilModule *pModule,
  128. DxilPipelineStateValidation &PSV);
  129. static void EmitIdentMetadata(llvm::Module *pModule, LPCSTR pValue) {
  130. llvm::NamedMDNode *IdentMetadata =
  131. pModule->getOrInsertNamedMetadata("llvm.ident");
  132. llvm::LLVMContext &Ctx = pModule->getContext();
  133. llvm::Metadata *IdentNode[] = {llvm::MDString::get(Ctx, pValue)};
  134. IdentMetadata->addOperand(llvm::MDNode::get(Ctx, IdentNode));
  135. }
  136. void WritePart(AbstractMemoryStream *pStream, const void *pData, size_t size) {
  137. ULONG cbWritten = 0;
  138. pStream->Write(pData, size, &cbWritten);
  139. }
  140. void WritePart(AbstractMemoryStream *pStream, const SmallVectorImpl<char> &Data) {
  141. WritePart(pStream, Data.data(), Data.size());
  142. }
  143. void DxbcConverter::ConvertImpl(_In_reads_bytes_(DxbcSize) LPCVOID pDxbc,
  144. _In_ UINT32 DxbcSize,
  145. _In_opt_z_ LPCWSTR pExtraOptions,
  146. _Outptr_result_bytebuffer_maybenull_(*pDxilSize) LPVOID *ppDxil,
  147. _Out_ UINT32 *pDxilSize,
  148. _Outptr_result_maybenull_z_ LPWSTR *ppDiag) {
  149. IFTARG(pDxbc);
  150. IFTARG(ppDxil);
  151. IFTARG(pDxilSize);
  152. *ppDxil = nullptr;
  153. *pDxilSize = 0;
  154. if (ppDiag)
  155. *ppDiag = nullptr;
  156. // Parse pExtraOptions.
  157. ParseExtraOptions(pExtraOptions);
  158. // Create the module.
  159. m_pModule = std::make_unique<llvm::Module>("main", m_Ctx);
  160. // Setup DxilModule.
  161. m_pPR = &(m_pModule->GetOrCreateDxilModule(/*skipInit*/true));
  162. m_pOP = m_pPR->GetOP();
  163. // Open DXBC container.
  164. DxilContainerReader dxbcReader;
  165. IFT(dxbcReader.Load(pDxbc, DxbcSize));
  166. const void *pMaxPtr = (const char *)pDxbc + DxbcSize;
  167. IFTBOOL(pDxbc < pMaxPtr, DXC_E_INCORRECT_DXBC);
  168. // Obtain the code blob.
  169. UINT uCodeBlob;
  170. IFT(dxbcReader.FindFirstPartKind(DXBC_GenericShaderEx, &uCodeBlob));
  171. if (uCodeBlob == DXIL_CONTAINER_BLOB_NOT_FOUND) {
  172. IFT(dxbcReader.FindFirstPartKind(DXBC_GenericShader, &uCodeBlob));
  173. }
  174. IFTBOOL(uCodeBlob != DXIL_CONTAINER_BLOB_NOT_FOUND, DXC_E_INCORRECT_DXBC);
  175. const CShaderToken *pByteCode;
  176. IFTBOOL(dxbcReader.GetPartContent(uCodeBlob, (const void **)&pByteCode) == S_OK, DXC_E_INCORRECT_DXBC);
  177. // Parse DXBC container.
  178. D3D10ShaderBinary::CShaderCodeParser Parser;
  179. // 1. Collect information about the shader.
  180. Parser.SetShader(pByteCode);
  181. AnalyzeShader(Parser);
  182. // 2. Parse input signature(s).
  183. ExtractInputSignatureFromDXBC(dxbcReader, pMaxPtr);
  184. ConvertSignature(*m_pInputSignature, m_pPR->GetInputSignature());
  185. if (m_pSM->IsDS()) {
  186. ExtractPatchConstantSignatureFromDXBC(dxbcReader, pMaxPtr);
  187. ConvertSignature(*m_pPatchConstantSignature, m_pPR->GetPatchConstOrPrimSignature());
  188. }
  189. // 3. Parse output signature(s).
  190. ExtractOutputSignatureFromDXBC(dxbcReader, pMaxPtr);
  191. ConvertSignature(*m_pOutputSignature, m_pPR->GetOutputSignature());
  192. if (m_pSM->IsHS()) {
  193. ExtractPatchConstantSignatureFromDXBC(dxbcReader, pMaxPtr);
  194. ConvertSignature(*m_pPatchConstantSignature, m_pPR->GetPatchConstOrPrimSignature());
  195. }
  196. // 3.5. Callback before conversion
  197. PreConvertHook(pByteCode);
  198. // 4. Transform DXBC to DXIL.
  199. Parser.SetShader(pByteCode);
  200. ConvertInstructions(Parser);
  201. // 5. Emit medatada.
  202. m_pPR->EmitDxilMetadata();
  203. EmitIdentMetadata(m_pModule.get(), "dxbc2dxil 1.2");
  204. // 6. Cleanup/Optimize DXIL.
  205. Optimize();
  206. // 7. Callback after conversion
  207. PostConvertHook(pByteCode);
  208. // Serialize DXIL.
  209. SmallVector<char, 4*1024> DxilBuffer;
  210. SerializeDxil(DxilBuffer);
  211. // Wrap LLVM module in a DXBC container.
  212. size_t DXILSize = DxilBuffer.size_in_bytes();
  213. DxilContainerWriter *pContainerWriter = hlsl::NewDxilContainerWriter();
  214. pContainerWriter->AddPart(DXBC_DXIL, DXILSize, [=](AbstractMemoryStream *pStream) {
  215. WritePart(pStream, DxilBuffer);
  216. });
  217. SmallVector<char, 512> PSVBuffer; // 512 bytes is enough for 30 resources + header
  218. {
  219. UINT uCBuffers = m_pPR->GetCBuffers().size();
  220. UINT uSamplers = m_pPR->GetSamplers().size();
  221. UINT uSRVs = m_pPR->GetSRVs().size();
  222. UINT uUAVs = m_pPR->GetUAVs().size();
  223. UINT uTotalResources = uCBuffers + uSamplers + uSRVs + uUAVs;
  224. uint32_t PSVBufferSize = 0;
  225. DxilPipelineStateValidation PSV;
  226. PSV.InitNew(uTotalResources, nullptr, &PSVBufferSize);
  227. PSVBuffer.resize(PSVBufferSize);
  228. PSV.InitNew(uTotalResources, PSVBuffer.data(), &PSVBufferSize);
  229. AddDxilPipelineStateValidationToDXBC(m_pPR, PSV);
  230. pContainerWriter->AddPart(DXBC_PipelineStateValidation, PSVBufferSize, [=](AbstractMemoryStream *pStream) {
  231. WritePart(pStream, PSVBuffer);
  232. });
  233. }
  234. UINT64 featureBody = 0;
  235. { // Append original IO signatures to DXIL blob
  236. DXBCFourCC IOSigFourCCArray[] = {
  237. DXBC_InputSignature11_1,
  238. DXBC_InputSignature,
  239. DXBC_OutputSignature11_1,
  240. DXBC_OutputSignature5,
  241. DXBC_OutputSignature,
  242. DXBC_PatchConstantSignature11_1,
  243. DXBC_PatchConstantSignature
  244. };
  245. UINT NumSigs = sizeof(IOSigFourCCArray) / sizeof(IOSigFourCCArray[0]);
  246. UINT uBlob = DXIL_CONTAINER_BLOB_NOT_FOUND;
  247. UINT uElemSize = 0;
  248. const void* pBlobData = nullptr;
  249. for(UINT i = 0; i < NumSigs; i++) {
  250. IFT(dxbcReader.FindFirstPartKind(IOSigFourCCArray[i], &uBlob));
  251. if(uBlob != DXIL_CONTAINER_BLOB_NOT_FOUND) {
  252. IFT(dxbcReader.GetPartContent(uBlob, &pBlobData, &uElemSize));
  253. pContainerWriter->AddPart(IOSigFourCCArray[i], uElemSize, [=](AbstractMemoryStream *pStream) {
  254. WritePart(pStream, pBlobData, uElemSize);
  255. });
  256. }
  257. }
  258. // Add DXBC_RootSignature and DXBC_ShaderFeatureInfo if present
  259. IFT(dxbcReader.FindFirstPartKind(DXBC_RootSignature, &uBlob));
  260. if(uBlob != DXIL_CONTAINER_BLOB_NOT_FOUND) {
  261. IFT(dxbcReader.GetPartContent(uBlob, &pBlobData, &uElemSize));
  262. pContainerWriter->AddPart(DXBC_RootSignature, uElemSize, [=](AbstractMemoryStream *pStream) {
  263. WritePart(pStream, pBlobData, uElemSize);
  264. });
  265. }
  266. IFT(dxbcReader.FindFirstPartKind(DXBC_ShaderFeatureInfo, &uBlob));
  267. if(uBlob != DXIL_CONTAINER_BLOB_NOT_FOUND) {
  268. IFT(dxbcReader.GetPartContent(uBlob, &pBlobData, &uElemSize));
  269. pContainerWriter->AddPart(DXBC_ShaderFeatureInfo, uElemSize, [=](AbstractMemoryStream *pStream) {
  270. WritePart(pStream, pBlobData, uElemSize);
  271. });
  272. }
  273. else
  274. {
  275. // Add one anyway
  276. uElemSize = sizeof(UINT64);
  277. pContainerWriter->AddPart(DXBC_ShaderFeatureInfo, uElemSize, [=](AbstractMemoryStream *pStream) {
  278. WritePart(pStream, (void*)&featureBody, sizeof(featureBody));
  279. });
  280. }
  281. }
  282. // Serialize the container
  283. UINT32 OutputSize = pContainerWriter->size();
  284. CComHeapPtr<void> pOutput;
  285. IFTBOOL(pOutput.AllocateBytes(OutputSize), E_OUTOFMEMORY);
  286. CComPtr<AbstractMemoryStream> pOutputStream;
  287. IFT(CreateFixedSizeMemoryStream((LPBYTE)pOutput.m_pData, OutputSize, &pOutputStream));
  288. pContainerWriter->write(pOutputStream);
  289. pOutputStream.Detach();
  290. *ppDxil = pOutput.Detach();
  291. *pDxilSize = OutputSize;
  292. m_pBuilder.reset();
  293. m_pModule.reset();
  294. // Diagnostics.
  295. if (ppDiag)
  296. *ppDiag = nullptr;
  297. }
  298. void DxbcConverter::ConvertInDriverImpl(_In_reads_bytes_(8) const UINT32 *pByteCode,
  299. _In_opt_z_ const D3D12DDIARG_SIGNATURE_ENTRY_0012 *pInputSignature,
  300. _In_ UINT32 NumInputSignatureElements,
  301. _In_opt_z_ const D3D12DDIARG_SIGNATURE_ENTRY_0012 *pOutputSignature,
  302. _In_ UINT32 NumOutputSignatureElements,
  303. _In_opt_z_ const D3D12DDIARG_SIGNATURE_ENTRY_0012 *pPatchConstantSignature,
  304. _In_ UINT32 NumPatchConstantSignatureElements,
  305. _In_opt_z_ LPCWSTR pExtraOptions,
  306. _Out_ IDxcBlob **ppDxcBlob,
  307. _Outptr_result_maybenull_z_ LPWSTR *ppDiag) {
  308. IFTARG(pByteCode);
  309. IFTARG(ppDxcBlob);
  310. UINT SizeInUINTs = pByteCode[1];
  311. IFTBOOL(SizeInUINTs >= 2, DXC_E_ERROR_PARSING_DXBC_BYTECODE);
  312. *ppDxcBlob = nullptr;
  313. if (ppDiag)
  314. *ppDiag = nullptr;
  315. // Parse pExtraOptions.
  316. ParseExtraOptions(pExtraOptions);
  317. // Create the module.
  318. m_pModule = std::make_unique<llvm::Module>("main", m_Ctx);
  319. // Setup DxilModule.
  320. m_pPR = &(m_pModule->GetOrCreateDxilModule(/*skipInit*/true));
  321. m_pOP = m_pPR->GetOP();
  322. // Parse DXBC bytecode.
  323. D3D10ShaderBinary::CShaderCodeParser Parser;
  324. // 1. Collect information about the shader.
  325. Parser.SetShader(pByteCode);
  326. AnalyzeShader(Parser);
  327. // 2. Parse input signature(s).
  328. ExtractSignatureFromDDI(pInputSignature, NumInputSignatureElements, *m_pInputSignature);
  329. ConvertSignature(*m_pInputSignature, m_pPR->GetInputSignature());
  330. if (m_pSM->IsDS()) {
  331. ExtractSignatureFromDDI(pPatchConstantSignature, NumPatchConstantSignatureElements, *m_pPatchConstantSignature);
  332. ConvertSignature(*m_pPatchConstantSignature, m_pPR->GetPatchConstOrPrimSignature());
  333. }
  334. // 3. Parse output signature(s).
  335. ExtractSignatureFromDDI(pOutputSignature, NumOutputSignatureElements, *m_pOutputSignature);
  336. ConvertSignature(*m_pOutputSignature, m_pPR->GetOutputSignature());
  337. if (m_pSM->IsHS()) {
  338. ExtractSignatureFromDDI(pPatchConstantSignature, NumPatchConstantSignatureElements, *m_pPatchConstantSignature);
  339. ConvertSignature(*m_pPatchConstantSignature, m_pPR->GetPatchConstOrPrimSignature());
  340. }
  341. // 3.5. Callback before conversion
  342. PreConvertHook(pByteCode);
  343. // 4. Transform DXBC to DXIL.
  344. Parser.SetShader(pByteCode);
  345. ConvertInstructions(Parser);
  346. // 5. Emit medatada.
  347. m_pPR->EmitDxilMetadata();
  348. // 6. Cleanup/Optimize DXIL.
  349. Optimize();
  350. // 7. Callback after conversion
  351. PostConvertHook(pByteCode);
  352. // Serialize DXIL.
  353. SmallVector<char, 8*1024> DxilBuffer;
  354. raw_svector_ostream DxilStream(DxilBuffer);
  355. WriteBitcodeToFile(m_pModule.get(), DxilStream);
  356. DxilStream.flush();
  357. IFT(DxcCreateBlobOnHeapCopy(DxilBuffer.data(), DxilBuffer.size_in_bytes(), ppDxcBlob));
  358. m_pBuilder.reset();
  359. m_pModule.reset();
  360. // Diagnostics.
  361. if (ppDiag)
  362. *ppDiag = nullptr;
  363. }
  364. void DxbcConverter::ParseExtraOptions(const wchar_t *pExtraOptions) {
  365. if (pExtraOptions == nullptr) return;
  366. // This is temporary implementation for now.
  367. wstring Str(pExtraOptions);
  368. if (Str.find(L"-disableHashCheck") != wstring::npos)
  369. m_bDisableHashCheck = true;
  370. // Opt out from DXIL cleanup pass.
  371. if (Str.find(L"-no-dxil-cleanup") != wstring::npos)
  372. m_bRunDxilCleanup = false;
  373. }
  374. void DxbcConverter::SetShaderGlobalFlags(unsigned GlobalFlags) {
  375. // GlobalFlags takes the set of flags defined for D3D10_SB_OPCODE_DCL_GLOBAL_FLAGS:
  376. m_pPR->m_ShaderFlags.SetDisableOptimizations (DXBC::IsFlagDisableOptimizations (GlobalFlags)); // ~D3D11_1_SB_GLOBAL_FLAG_SKIP_OPTIMIZATION
  377. m_pPR->m_ShaderFlags.SetDisableMathRefactoring (DXBC::IsFlagDisableMathRefactoring (GlobalFlags)); // ~D3D10_SB_GLOBAL_FLAG_REFACTORING_ALLOWED
  378. m_pPR->m_ShaderFlags.SetEnableDoublePrecision (DXBC::IsFlagEnableDoublePrecision (GlobalFlags)); // D3D11_SB_GLOBAL_FLAG_ENABLE_DOUBLE_PRECISION_FLOAT_OPS
  379. m_pPR->m_ShaderFlags.SetForceEarlyDepthStencil (DXBC::IsFlagForceEarlyDepthStencil (GlobalFlags)); // D3D11_SB_GLOBAL_FLAG_FORCE_EARLY_DEPTH_STENCIL
  380. m_pPR->m_ShaderFlags.SetLowPrecisionPresent (DXBC::IsFlagEnableMinPrecision (GlobalFlags)); // D3D11_1_SB_GLOBAL_FLAG_ENABLE_MINIMUM_PRECISION
  381. m_pPR->m_ShaderFlags.SetEnableDoubleExtensions (DXBC::IsFlagEnableDoubleExtensions (GlobalFlags)); // D3D11_1_SB_GLOBAL_FLAG_ENABLE_DOUBLE_EXTENSIONS
  382. m_pPR->m_ShaderFlags.SetEnableMSAD (DXBC::IsFlagEnableMSAD (GlobalFlags)); // D3D11_1_SB_GLOBAL_FLAG_ENABLE_SHADER_EXTENSIONS
  383. if (IsSM51Plus()) {
  384. m_pPR->m_ShaderFlags.SetAllResourcesBound (DXBC::IsFlagAllResourcesBound (GlobalFlags)); // D3D12_SB_GLOBAL_FLAG_ALL_RESOURCES_BOUND
  385. }
  386. m_pPR->m_ShaderFlags.SetEnableRawAndStructuredBuffers (DXBC::IsFlagEnableRawAndStructuredBuffers(GlobalFlags)); // D3D12_SB_GLOBAL_FLAG_ALL_RESOURCES_BOUND
  387. }
  388. void DxbcConverter::ExtractInputSignatureFromDXBC(DxilContainerReader &dxbcReader, const void *pMaxPtr) {
  389. // Obtain the input signature blob.
  390. UINT uBlob;
  391. IFT(dxbcReader.FindFirstPartKind(DXBC_InputSignature11_1, &uBlob));
  392. UINT uElemSize = sizeof(D3D11_INTERNALSHADER_PARAMETER_11_1);
  393. if (uBlob == DXIL_CONTAINER_BLOB_NOT_FOUND) {
  394. IFT(dxbcReader.FindFirstPartKind(DXBC_InputSignature, &uBlob));
  395. uElemSize = sizeof(D3D10_INTERNALSHADER_PARAMETER);
  396. }
  397. IFTBOOL(uBlob != DXIL_CONTAINER_BLOB_NOT_FOUND, DXC_E_INCORRECT_DXBC);
  398. // Parse signature elements.
  399. const D3D10_INTERNALSHADER_SIGNATURE *pSig;
  400. IFT(dxbcReader.GetPartContent(uBlob, (const void**)&pSig))
  401. ExtractSignatureFromDXBC(pSig, uElemSize, pMaxPtr, *m_pInputSignature);
  402. }
  403. void DxbcConverter::ExtractOutputSignatureFromDXBC(DxilContainerReader &dxbcReader, const void *pMaxPtr) {
  404. // Obtain the output signature blob.
  405. UINT uBlob;
  406. IFT(dxbcReader.FindFirstPartKind(DXBC_OutputSignature11_1, &uBlob));
  407. UINT uElemSize = sizeof(D3D11_INTERNALSHADER_PARAMETER_11_1);
  408. if (uBlob == DXIL_CONTAINER_BLOB_NOT_FOUND) {
  409. IFT(dxbcReader.FindFirstPartKind(DXBC_OutputSignature5, &uBlob));
  410. uElemSize = sizeof(D3D11_INTERNALSHADER_PARAMETER_FOR_GS);
  411. }
  412. if (uBlob == DXIL_CONTAINER_BLOB_NOT_FOUND) {
  413. IFT(dxbcReader.FindFirstPartKind(DXBC_OutputSignature, &uBlob));
  414. uElemSize = sizeof(D3D10_INTERNALSHADER_PARAMETER);
  415. }
  416. IFTBOOL(uBlob != DXIL_CONTAINER_BLOB_NOT_FOUND, DXC_E_INCORRECT_DXBC);
  417. // Parse signature elements.
  418. const D3D10_INTERNALSHADER_SIGNATURE *pSig;
  419. IFT(dxbcReader.GetPartContent(uBlob, (const void**)&pSig));
  420. ExtractSignatureFromDXBC(pSig, uElemSize, pMaxPtr, *m_pOutputSignature);
  421. }
  422. void DxbcConverter::ExtractPatchConstantSignatureFromDXBC(DxilContainerReader &dxbcReader, const void *pMaxPtr) {
  423. // Obtain the patch-constant signature blob.
  424. UINT uBlob;
  425. IFT(dxbcReader.FindFirstPartKind(DXBC_PatchConstantSignature11_1, &uBlob));
  426. UINT uElemSize = sizeof(D3D11_INTERNALSHADER_PARAMETER_11_1);
  427. if (uBlob == DXIL_CONTAINER_BLOB_NOT_FOUND) {
  428. IFT(dxbcReader.FindFirstPartKind(DXBC_PatchConstantSignature, &uBlob));
  429. uElemSize = sizeof(D3D10_INTERNALSHADER_PARAMETER);
  430. }
  431. IFTBOOL(uBlob != DXIL_CONTAINER_BLOB_NOT_FOUND, DXC_E_INCORRECT_DXBC);
  432. // Parse signature elements.
  433. const D3D10_INTERNALSHADER_SIGNATURE *pSig;
  434. IFT(dxbcReader.GetPartContent(uBlob, (const void**)&pSig));
  435. ExtractSignatureFromDXBC(pSig, uElemSize, pMaxPtr, *m_pPatchConstantSignature);
  436. }
  437. void DxbcConverter::ExtractSignatureFromDXBC(const D3D10_INTERNALSHADER_SIGNATURE *pSig,
  438. UINT uElemSize, const void *pMaxPtr,
  439. SignatureHelper &SigHelper) {
  440. // Verify signature offsets are within the blob.
  441. const char *pCheck = (const char *)pSig;
  442. const char *pCheck2 = pCheck + sizeof(D3D10_INTERNALSHADER_SIGNATURE);
  443. IFTBOOL(pCheck != nullptr && pCheck < pMaxPtr && pCheck2 <= pMaxPtr, DXC_E_INCORRECT_DXBC);
  444. pCheck = (const char *)pSig + pSig->ParameterInfo;
  445. pCheck2 = pCheck + pSig->Parameters * uElemSize;
  446. IFTBOOL(pCheck <= pMaxPtr && pCheck2 <= pMaxPtr && pCheck <= pCheck2, DXC_E_INCORRECT_DXBC);
  447. unsigned uParamCount = pSig->Parameters;
  448. const char *pSigBase = (const char *)pSig;
  449. const char *pParamBase = pSigBase + pSig->ParameterInfo;
  450. // This is to test in-driver conversion.
  451. #define TestDDISignature 0
  452. #if TestDDISignature
  453. vector<D3D12DDIARG_SIGNATURE_ENTRY_0012> TestDDI;
  454. TestDDI.resize(uParamCount);
  455. memset(TestDDI.data(), 0, TestDDI.size()*sizeof(D3D12DDIARG_SIGNATURE_ENTRY_0012));
  456. unsigned EdgeTess = 0, InsideEdgeTess = 0;
  457. #endif
  458. for (unsigned iElement = 0; iElement < uParamCount; iElement++) {
  459. D3D11_INTERNALSHADER_PARAMETER_11_1 P = {0};
  460. // Properly copy parameters for the serialized form into P.
  461. switch (uElemSize) {
  462. case sizeof(D3D11_INTERNALSHADER_PARAMETER_11_1):
  463. memcpy(&P, pParamBase + iElement*uElemSize, uElemSize);
  464. break;
  465. case sizeof(D3D11_INTERNALSHADER_PARAMETER_FOR_GS):
  466. memcpy(&P, pParamBase + iElement*uElemSize, uElemSize);
  467. break;
  468. case sizeof(D3D10_INTERNALSHADER_PARAMETER):
  469. static_assert(sizeof(D3D11_INTERNALSHADER_PARAMETER_FOR_GS) ==
  470. sizeof(D3D10_INTERNALSHADER_PARAMETER) + FIELD_OFFSET(D3D11_INTERNALSHADER_PARAMETER_FOR_GS, SemanticName),
  471. "Incorrect assumptions about field offset");
  472. memcpy(&P.SemanticName, pParamBase + iElement*uElemSize, uElemSize);
  473. break;
  474. default:
  475. IFT(DXC_E_INCORRECT_DXBC);
  476. }
  477. // Extract data from the blob.
  478. SignatureHelper::ElementRecord E;
  479. // Existing tests use testasm to create shaders with incorrect semantic names.
  480. // The converter is compensating for this.
  481. if (P.SystemValue == D3D_NAME_UNDEFINED) {
  482. // Retrive name from the signature blob.
  483. CheckDxbcString(pSigBase + P.SemanticName, pMaxPtr);
  484. E.SemanticName = string(pSigBase + P.SemanticName);
  485. } else {
  486. // Recover canonical SV_ name.
  487. E.SemanticName = string(DXBC::GetSemanticNameFromD3DName(P.SystemValue));
  488. }
  489. unsigned SemanticIndex = DXBC::GetSemanticIndexFromD3DName(P.SystemValue);
  490. E.SemanticIndex = (SemanticIndex == UINT_MAX) ? P.SemanticIndex : SemanticIndex;
  491. E.StartRow = P.Register;
  492. E.StartCol = CMask(P.Mask).GetFirstActiveComp();
  493. E.Rows = 1;
  494. E.Cols = CMask(P.Mask).GetNumActiveRangeComps();
  495. E.Stream = P.Stream;
  496. E.ComponentType = DXBC::GetCompTypeWithMinPrec(P.ComponentType, (D3D11_SB_OPERAND_MIN_PRECISION)P.MinPrecision);
  497. #if TestDDISignature
  498. D3D12DDIARG_SIGNATURE_ENTRY_0012 &D = TestDDI[iElement];
  499. D.Register = P.Register;
  500. D.Mask = P.Mask;
  501. D.Stream = P.Stream;
  502. D.RegisterComponentType = (D3D10_SB_REGISTER_COMPONENT_TYPE)P.ComponentType;
  503. D.MinPrecision = (D3D11_SB_OPERAND_MIN_PRECISION)P.MinPrecision;
  504. switch (P.SystemValue) {
  505. case D3D_NAME_UNDEFINED: D.SystemValue = D3D10_SB_NAME_UNDEFINED; break;
  506. case D3D_NAME_POSITION: D.SystemValue = D3D10_SB_NAME_POSITION; break;
  507. case D3D_NAME_CLIP_DISTANCE: D.SystemValue = D3D10_SB_NAME_CLIP_DISTANCE; break;
  508. case D3D_NAME_CULL_DISTANCE: D.SystemValue = D3D10_SB_NAME_CULL_DISTANCE; break;
  509. case D3D_NAME_RENDER_TARGET_ARRAY_INDEX: D.SystemValue = D3D10_SB_NAME_RENDER_TARGET_ARRAY_INDEX; break;
  510. case D3D_NAME_VIEWPORT_ARRAY_INDEX: D.SystemValue = D3D10_SB_NAME_VIEWPORT_ARRAY_INDEX; break;
  511. case D3D_NAME_VERTEX_ID: D.SystemValue = D3D10_SB_NAME_VERTEX_ID; break;
  512. case D3D_NAME_PRIMITIVE_ID: D.SystemValue = D3D10_SB_NAME_PRIMITIVE_ID; break;
  513. case D3D_NAME_INSTANCE_ID: D.SystemValue = D3D10_SB_NAME_INSTANCE_ID; break;
  514. case D3D_NAME_IS_FRONT_FACE: D.SystemValue = D3D10_SB_NAME_IS_FRONT_FACE; break;
  515. case D3D_NAME_SAMPLE_INDEX: D.SystemValue = D3D10_SB_NAME_SAMPLE_INDEX; break;
  516. case D3D_NAME_FINAL_QUAD_EDGE_TESSFACTOR:
  517. switch (EdgeTess) {
  518. case 0: D.SystemValue = D3D11_SB_NAME_FINAL_QUAD_U_EQ_0_EDGE_TESSFACTOR; break;
  519. case 1: D.SystemValue = D3D11_SB_NAME_FINAL_QUAD_V_EQ_0_EDGE_TESSFACTOR; break;
  520. case 2: D.SystemValue = D3D11_SB_NAME_FINAL_QUAD_U_EQ_1_EDGE_TESSFACTOR; break;
  521. case 3: D.SystemValue = D3D11_SB_NAME_FINAL_QUAD_V_EQ_1_EDGE_TESSFACTOR; break;
  522. default:
  523. DXASSERT_NOMSG(false);
  524. }
  525. EdgeTess++;
  526. break;
  527. case D3D_NAME_FINAL_QUAD_INSIDE_TESSFACTOR:
  528. switch (InsideEdgeTess) {
  529. case 0: D.SystemValue = D3D11_SB_NAME_FINAL_QUAD_U_INSIDE_TESSFACTOR; break;
  530. case 1: D.SystemValue = D3D11_SB_NAME_FINAL_QUAD_V_INSIDE_TESSFACTOR; break;
  531. default:
  532. DXASSERT_NOMSG(false);
  533. }
  534. InsideEdgeTess++;
  535. break;
  536. case D3D_NAME_FINAL_TRI_EDGE_TESSFACTOR:
  537. switch (EdgeTess) {
  538. case 0: D.SystemValue = D3D11_SB_NAME_FINAL_TRI_U_EQ_0_EDGE_TESSFACTOR; break;
  539. case 1: D.SystemValue = D3D11_SB_NAME_FINAL_TRI_V_EQ_0_EDGE_TESSFACTOR; break;
  540. case 2: D.SystemValue = D3D11_SB_NAME_FINAL_TRI_W_EQ_0_EDGE_TESSFACTOR; break;
  541. default:
  542. DXASSERT_NOMSG(false);
  543. }
  544. EdgeTess++;
  545. break;
  546. case D3D_NAME_FINAL_TRI_INSIDE_TESSFACTOR: D.SystemValue = D3D11_SB_NAME_FINAL_TRI_INSIDE_TESSFACTOR; break;
  547. case D3D_NAME_FINAL_LINE_DETAIL_TESSFACTOR: D.SystemValue = D3D11_SB_NAME_FINAL_LINE_DETAIL_TESSFACTOR; break;
  548. case D3D_NAME_FINAL_LINE_DENSITY_TESSFACTOR: D.SystemValue = D3D11_SB_NAME_FINAL_LINE_DENSITY_TESSFACTOR; break;
  549. case D3D_NAME_TARGET:
  550. case D3D_NAME_DEPTH:
  551. case D3D_NAME_COVERAGE:
  552. case D3D_NAME_DEPTH_GREATER_EQUAL:
  553. case D3D_NAME_DEPTH_LESS_EQUAL:
  554. case D3D_NAME_STENCIL_REF:
  555. case D3D_NAME_INNER_COVERAGE: D.SystemValue = D3D10_SB_NAME_UNDEFINED; break;
  556. default:
  557. DXASSERT_NOMSG(false);
  558. }
  559. #else
  560. SigHelper.m_ElementRecords.emplace_back(E);
  561. #endif
  562. }
  563. #if TestDDISignature
  564. ExtractSignatureFromDDI(TestDDI.data(), (unsigned)TestDDI.size(), SigHelper);
  565. #endif
  566. }
  567. void DxbcConverter::ExtractSignatureFromDDI(const D3D12DDIARG_SIGNATURE_ENTRY_0012 *pElements,
  568. unsigned NumElements,
  569. SignatureHelper &SigHelper) {
  570. string NamePrefix;
  571. if (SigHelper.IsInput())
  572. NamePrefix = "_in";
  573. else if (SigHelper.IsOutput())
  574. NamePrefix = "_out";
  575. else
  576. NamePrefix = "_pc";
  577. unsigned iArbitrarySemantic = 0;
  578. for (unsigned iElement = 0; iElement < NumElements; iElement++) {
  579. const D3D12DDIARG_SIGNATURE_ENTRY_0012 &P = pElements[iElement];
  580. // Extract data from DDI signature element record.
  581. SignatureHelper::ElementRecord E;
  582. E.StartRow = P.Register;
  583. E.StartCol = CMask(P.Mask).GetFirstActiveComp();
  584. E.Rows = 1;
  585. E.Cols = CMask(P.Mask).GetNumActiveRangeComps();
  586. E.Stream = P.Stream;
  587. if (P.SystemValue == D3D10_SB_NAME_UNDEFINED) {
  588. E.ComponentType = DXBC::GetCompTypeWithMinPrec((D3D_REGISTER_COMPONENT_TYPE)P.RegisterComponentType, (D3D11_SB_OPERAND_MIN_PRECISION)P.MinPrecision);
  589. // For PS output, try to disambiguate semantic based on register index.
  590. if (m_pSM->IsPS() && SigHelper.IsOutput()) {
  591. if (P.Register != -1) {
  592. // This must be SV_Target.
  593. E.SemanticName = "SV_Target";
  594. E.SemanticIndex = P.Register;
  595. } else {
  596. E.SemanticIndex = P.Register;
  597. switch (P.RegisterComponentType) {
  598. case D3D10_SB_REGISTER_COMPONENT_UINT32:
  599. case D3D10_SB_REGISTER_COMPONENT_SINT32: {
  600. // This must be SV_StencilRef.
  601. if (m_bHasStencilRef) {
  602. E.SemanticName = "SV_StencilRef";
  603. } else if (m_bHasCoverageOut) {
  604. E.SemanticName = "SV_Coverage";
  605. } else {
  606. IFTBOOL(false, DXC_E_INCORRECT_DDI_SIGNATURE);
  607. }
  608. break;
  609. }
  610. case D3D10_SB_REGISTER_COMPONENT_FLOAT32: {
  611. // This must be SV_Depth*.
  612. switch (m_DepthRegType) {
  613. case D3D10_SB_OPERAND_TYPE_OUTPUT_DEPTH:
  614. E.SemanticName = "SV_Depth";
  615. break;
  616. case D3D11_SB_OPERAND_TYPE_OUTPUT_DEPTH_GREATER_EQUAL:
  617. E.SemanticName = "SV_DepthGreaterEqual";
  618. break;
  619. case D3D11_SB_OPERAND_TYPE_OUTPUT_DEPTH_LESS_EQUAL:
  620. E.SemanticName = "SV_DepthLessEqual";
  621. break;
  622. case D3D10_SB_OPERAND_TYPE_NULL:
  623. default:
  624. IFT(DXC_E_INCORRECT_DDI_SIGNATURE);
  625. }
  626. break;
  627. }
  628. default:
  629. IFT(DXC_E_INCORRECT_DDI_SIGNATURE);
  630. }
  631. }
  632. } else {
  633. // Arbitrary semantic.
  634. E.SemanticName = NamePrefix + std::to_string(iArbitrarySemantic++);
  635. E.SemanticIndex = iElement;
  636. }
  637. } else {
  638. E.SemanticName = string(DXBC::GetD3D10SBName(P.SystemValue));
  639. E.SemanticIndex = DXBC::GetD3D10SBSemanticIndex(P.SystemValue);
  640. if (P.RegisterComponentType != D3D_REGISTER_COMPONENT_UNKNOWN) {
  641. E.ComponentType = DXBC::GetCompTypeWithMinPrec((D3D_REGISTER_COMPONENT_TYPE)P.RegisterComponentType, (D3D11_SB_OPERAND_MIN_PRECISION)P.MinPrecision);
  642. } else {
  643. E.ComponentType = DXBC::GetD3DRegCompType(P.SystemValue);
  644. }
  645. }
  646. // This would happen is component type is not supplied by the runtime.
  647. IFTBOOL(!E.ComponentType.IsInvalid(), DXC_E_INCORRECT_DDI_SIGNATURE);
  648. SigHelper.m_ElementRecords.emplace_back(E);
  649. }
  650. }
  651. void DxbcConverter::ConvertSignature(SignatureHelper &SigHelper, DxilSignature &DxilSig) {
  652. // Sort SigHelper.m_UsedElements for upcoming binary search.
  653. std::sort(SigHelper.m_UsedElements.begin(), SigHelper.m_UsedElements.end(), SignatureHelper::UsedElement::LTByStreamAndStartRowAndStartCol());
  654. if (!SigHelper.m_Ranges.empty()) {
  655. // Adjust range columns to tightly include components of signature elements.
  656. for (size_t iRange = 0; iRange < SigHelper.m_Ranges.size(); iRange++) {
  657. SignatureHelper::Range &R = SigHelper.m_Ranges[iRange];
  658. unsigned RangeStartCol = UINT32_MAX;
  659. unsigned RangeEndCol = UINT32_MAX;
  660. for (size_t iElement = 0; iElement < SigHelper.m_ElementRecords.size(); iElement++) {
  661. const SignatureHelper::ElementRecord &SigElem = SigHelper.m_ElementRecords[iElement];
  662. unsigned StartRow = SigElem.StartRow;
  663. unsigned StartCol = SigElem.StartCol;
  664. unsigned Rows = SigElem.Rows; DXASSERT_LOCALVAR_NOMSG(Rows, Rows == 1);
  665. unsigned Cols = SigElem.Cols;
  666. unsigned Stream = SigElem.Stream;
  667. if (R.OutputStream != Stream)
  668. continue;
  669. if (R.StartRow <= StartRow && StartRow < R.StartRow+R.Rows) {
  670. if (!(StartCol+Cols-1 < R.GetStartCol() || R.GetEndCol() < StartCol)) {
  671. // Signature element overlaps with the declared range.
  672. if (RangeStartCol != UINT32_MAX) {
  673. RangeStartCol = std::min(RangeStartCol, StartCol);
  674. RangeEndCol = std::max(RangeEndCol, StartCol+Cols-1);
  675. } else {
  676. RangeStartCol = StartCol;
  677. RangeEndCol = StartCol+Cols-1;
  678. }
  679. }
  680. }
  681. }
  682. R.StartCol = RangeStartCol;
  683. R.Cols = RangeEndCol - RangeStartCol + 1;
  684. }
  685. // Coalesce declaration ranges if they overlap.
  686. std::sort(SigHelper.m_Ranges.begin(), SigHelper.m_Ranges.end(), SignatureHelper::Range::LTRangeByStreamAndStartRowAndStartCol());
  687. unsigned iLastEntryIndex = 0;
  688. for (size_t i = 1; i < SigHelper.m_Ranges.size(); i++) {
  689. // Current range into which we try to coalesce.
  690. SignatureHelper::Range &R1 = SigHelper.m_Ranges[iLastEntryIndex];
  691. // A range that is a candidate for coalescing.
  692. const SignatureHelper::Range &R2 = SigHelper.m_Ranges[i];
  693. // Do R1 and R2 overlap?
  694. DXASSERT_NOMSG(R1.GetStartRow() <= R2.GetStartRow());
  695. bool bOverlaps = (R1.GetStartRow() <= R2.GetStartRow() && R2.GetStartRow() <= R1.GetEndRow()) &&
  696. !(R1.GetEndCol() < R2.GetStartCol() || R2.GetEndCol() < R1.GetStartCol());
  697. if (bOverlaps) {
  698. // Coalesce ranges.
  699. R1.Rows = std::max(R1.Rows, R2.GetEndRow() - R1.GetStartRow() + 1);
  700. unsigned StartCol = std::min(R1.GetStartCol(), R2.GetStartCol());
  701. unsigned EndCol = std::max(R1.GetEndCol(), R2.GetEndCol());
  702. R1.StartCol = StartCol;
  703. R1.Cols = EndCol - R1.StartCol + 1;
  704. } else {
  705. iLastEntryIndex++;
  706. SigHelper.m_Ranges[iLastEntryIndex] = SigHelper.m_Ranges[i];
  707. }
  708. }
  709. SigHelper.m_Ranges.resize(iLastEntryIndex + 1);
  710. }
  711. // map range elements from SigHelper.m_ElementRecords to dxil signature element index
  712. std::map<unsigned, unsigned> RangeElementToDxilElement;
  713. for (size_t iElement = 0; iElement < SigHelper.m_ElementRecords.size(); iElement++) {
  714. const SignatureHelper::ElementRecord &SigElem = SigHelper.m_ElementRecords[iElement];
  715. const string &SemanticName = SigElem.SemanticName;
  716. unsigned SemanticIndex = SigElem.SemanticIndex;
  717. unsigned StartRow = SigElem.StartRow;
  718. unsigned StartCol = SigElem.StartCol;
  719. unsigned Rows = SigElem.Rows; DXASSERT_NOMSG(Rows == 1);
  720. unsigned Cols = SigElem.Cols;
  721. unsigned Stream = SigElem.Stream;
  722. CompType ComponentType = SigElem.ComponentType;
  723. // Determine interpolation mode by matching the corresponding decl record.
  724. D3D_INTERPOLATION_MODE D3DInterpMode = D3D_INTERPOLATION_UNDEFINED;
  725. if (m_pSM->IsPS() && SigHelper.IsInput()) {
  726. bool bFirstUse = false;
  727. if (!SigHelper.m_UsedElements.empty()) {
  728. for (unsigned i = 0; i < Cols; i++) {
  729. unsigned c = StartCol + i;
  730. // Find used-element lower bound.
  731. SignatureHelper::UsedElement E1;
  732. E1.Row = StartRow;
  733. E1.StartCol = StartCol;
  734. E1.OutputStream = Stream;
  735. auto it = std::lower_bound(SigHelper.m_UsedElements.begin(), SigHelper.m_UsedElements.end(), E1, SignatureHelper::UsedElement::LTByStreamAndStartRowAndStartCol());
  736. if (it != SigHelper.m_UsedElements.end()) {
  737. SignatureHelper::UsedElement &E2 = *it;
  738. if (E2.Row == E1.Row && (E2.StartCol <= c && c < E2.StartCol+E2.Cols)) {
  739. if (!bFirstUse) {
  740. bFirstUse = true;
  741. D3DInterpMode = E2.InterpolationMode;
  742. } else {
  743. DXASSERT_DXBC(D3DInterpMode == E2.InterpolationMode);
  744. }
  745. }
  746. }
  747. }
  748. }
  749. }
  750. // Create a new signature element.
  751. InterpolationMode::Kind IMK = DXBC::GetInterpolationModeKind(D3DInterpMode);
  752. unique_ptr<DxilSignatureElement> pE(SigHelper.m_Signature.CreateElement());
  753. pE->Initialize(SemanticName, ComponentType, InterpolationMode(IMK), Rows, Cols, StartRow, StartCol);
  754. pE->SetOutputStream(Stream);
  755. DxilSignatureElement &E = *pE;
  756. // Check range containment.
  757. bool bInRange = false;
  758. if (!SigHelper.m_Ranges.empty()) {
  759. // Search which range contains the element.
  760. for (size_t iRange = 0; iRange < SigHelper.m_Ranges.size(); iRange++) {
  761. SignatureHelper::Range &R = SigHelper.m_Ranges[iRange];
  762. if (R.OutputStream != Stream)
  763. continue;
  764. if (R.StartRow <= StartRow && StartRow < R.StartRow+R.Rows) {
  765. if (!(StartCol+Cols-1 < R.GetStartCol() || R.GetEndCol() < StartCol)) {
  766. // Found containment.
  767. bInRange = true;
  768. auto itKeyDxilEl = RangeElementToDxilElement.find(iElement);
  769. if (itKeyDxilEl == RangeElementToDxilElement.end()) {
  770. // First element in range
  771. unsigned iDxilElementIndex = (unsigned)SigHelper.m_Signature.GetElements().size();
  772. E.AppendSemanticIndex(SemanticIndex);
  773. // Search for all matching elements by semantic in range to expand
  774. // the range of this element:
  775. for (size_t iOtherEl = iElement + 1;
  776. iOtherEl < SigHelper.m_ElementRecords.size() && StartRow + Rows < R.StartRow + R.Rows;
  777. iOtherEl++) {
  778. // Skip elements that are part of another captured range already
  779. if (RangeElementToDxilElement.find(iOtherEl) != RangeElementToDxilElement.end())
  780. continue;
  781. const SignatureHelper::ElementRecord &OtherEl = SigHelper.m_ElementRecords[iOtherEl];
  782. // There should be no gaps for indexed element, so we're done if we find one.
  783. if (OtherEl.StartRow > StartRow + Rows)
  784. break;
  785. if (SemanticName.compare(OtherEl.SemanticName) == 0) {
  786. // OtherEl should always have one row
  787. DXASSERT_DXBC(OtherEl.Rows == 1);
  788. // should always be adding one row at a time in order, and single
  789. // indexed element should not have different start column.
  790. if (OtherEl.StartRow == StartRow + Rows &&
  791. StartCol == OtherEl.StartCol) {
  792. RangeElementToDxilElement[iOtherEl] = iDxilElementIndex;
  793. Cols = std::max(Cols, OtherEl.Cols);
  794. Rows++;
  795. E.AppendSemanticIndex(OtherEl.SemanticIndex);
  796. }
  797. }
  798. }
  799. // Adjust element dimensions to encompas matching elements.
  800. E.SetStartCol(StartCol);
  801. E.SetCols(Cols);
  802. E.SetRows(Rows);
  803. SigHelper.m_Signature.AppendElement(std::move(pE));
  804. } else {
  805. #ifdef DBG
  806. // Verify match with range representative element.
  807. DxilSignatureElement &RE = SigHelper.m_Signature.GetElement(itKeyDxilEl->second);
  808. DXASSERT_DXBC(RE.GetCompType() == E.GetCompType());
  809. DXASSERT_DXBC(*RE.GetInterpolationMode() == *E.GetInterpolationMode());
  810. #endif
  811. }
  812. break;
  813. } else {
  814. // Check that there is no overlap.
  815. DXASSERT_DXBC(StartCol+Cols <= R.StartCol || StartCol >= R.StartCol+R.Cols);
  816. }
  817. }
  818. }
  819. }
  820. if (!bInRange) {
  821. DXASSERT(E.GetSemanticIndexVec().empty(), "otherwise a bug");
  822. E.AppendSemanticIndex(SemanticIndex);
  823. SigHelper.m_Signature.AppendElement(std::move(pE));
  824. }
  825. }
  826. // Add SGVs that are not present in the signature blob.
  827. if (SigHelper.m_bHasInputCoverage || SigHelper.m_bHasInnerInputCoverage) {
  828. DXASSERT_DXBC(m_pSM->IsPS() && SigHelper.IsInput());
  829. string SemName;
  830. if (SigHelper.m_bHasInputCoverage) {
  831. DXASSERT_DXBC(!SigHelper.m_bHasInnerInputCoverage);
  832. SemName = string("SV_Coverage");
  833. } else {
  834. DXASSERT_DXBC(!SigHelper.m_bHasInputCoverage && SigHelper.m_bHasInnerInputCoverage);
  835. SemName = string("SV_InnerCoverage");
  836. }
  837. unique_ptr<DxilSignatureElement> E(SigHelper.m_Signature.CreateElement());
  838. E->Initialize(SemName, CompType::Kind::U32, InterpolationMode(), 1, 1, Semantic::kUndefinedRow, 0);
  839. E->AppendSemanticIndex(0);
  840. SigHelper.m_Signature.AppendElement(std::move(E));
  841. }
  842. // Set up DXBC <reg,comp> to Element mapping or DXBC OperandRegType to Element mapping,
  843. // depending on the semantic type.
  844. for (size_t iElem = 0; iElem < SigHelper.m_Signature.GetElements().size(); iElem++) {
  845. DxilSignatureElement &E = SigHelper.m_Signature.GetElement(iElem);
  846. bool bUpdateRegMap = E.IsAllocated();
  847. switch (E.GetKind()) {
  848. case Semantic::Kind::Coverage:
  849. case Semantic::Kind::InnerCoverage:
  850. case Semantic::Kind::Depth:
  851. case Semantic::Kind::DepthGreaterEqual:
  852. case Semantic::Kind::DepthLessEqual:
  853. case Semantic::Kind::StencilRef: {
  854. bUpdateRegMap = false;
  855. D3D10_SB_OPERAND_TYPE OperandRegType = DXBC::GetOperandRegType(E.GetKind(), /*IsOutput*/SigHelper.IsOutput());
  856. DXASSERT_DXBC(SigHelper.m_DxbcSgvToSignatureElement.find(OperandRegType) == SigHelper.m_DxbcSgvToSignatureElement.end());
  857. SigHelper.m_DxbcSgvToSignatureElement[OperandRegType] = (unsigned)iElem;
  858. break;
  859. }
  860. }
  861. if (bUpdateRegMap) {
  862. DXASSERT_NOMSG(E.IsAllocated());
  863. unsigned Stream = E.GetOutputStream();
  864. for (unsigned iRow = 0; iRow < E.GetRows(); iRow++) {
  865. unsigned r = E.GetStartRow() + iRow;
  866. for (unsigned iCol = 0; iCol < E.GetCols(); iCol++) {
  867. unsigned c = E.GetStartCol() + iCol;
  868. SignatureHelper::RegAndCompAndStream Key(r, c, Stream);
  869. DXASSERT(SigHelper.m_DxbcRegisterToSignatureElement.find(Key) == SigHelper.m_DxbcRegisterToSignatureElement.end(), "otherwise elements are wrong");
  870. SigHelper.m_DxbcRegisterToSignatureElement[Key] = (unsigned)iElem;
  871. }
  872. }
  873. }
  874. }
  875. // Clone signature elements into DxilModule.
  876. for (size_t i = 0; i < SigHelper.m_Signature.GetElements().size(); i++) {
  877. DxilSignatureElement &E = SigHelper.m_Signature.GetElement(i);
  878. DXIL::SemanticInterpretationKind I = E.GetInterpretation();
  879. switch (I) {
  880. case DXIL::SemanticInterpretationKind::NA:
  881. case DXIL::SemanticInterpretationKind::NotInSig:
  882. case DXIL::SemanticInterpretationKind::Invalid:
  883. continue;
  884. }
  885. unique_ptr<DxilSignatureElement> pClone(new DxilSignatureElement(E));
  886. switch (I) {
  887. case DXIL::SemanticInterpretationKind::NotPacked:
  888. case DXIL::SemanticInterpretationKind::Shadow:
  889. // Make sure element is unallocated in this case (DXBC allocates some of these)
  890. pClone->SetStartRow(Semantic::kUndefinedRow);
  891. pClone->SetStartCol(Semantic::kUndefinedCol);
  892. break;
  893. }
  894. DxilSig.AppendElement(std::move(pClone));
  895. }
  896. }
  897. static void AddDxilPipelineStateValidationToDXBC(
  898. DxilModule *pModule,
  899. DxilPipelineStateValidation &PSV)
  900. {
  901. UINT uCBuffers = pModule->GetCBuffers().size();
  902. UINT uSamplers = pModule->GetSamplers().size();
  903. UINT uSRVs = pModule->GetSRVs().size();
  904. UINT uUAVs = pModule->GetUAVs().size();
  905. UINT uTotalResources = uCBuffers + uSamplers + uSRVs + uUAVs;
  906. // Set DxilRuntimInfo
  907. PSVRuntimeInfo0 *pInfo = PSV.GetPSVRuntimeInfo0();
  908. const ShaderModel *pSM = pModule->GetShaderModel();
  909. pInfo->MinimumExpectedWaveLaneCount = 0;
  910. pInfo->MaximumExpectedWaveLaneCount = -1;
  911. switch (pSM->GetKind()) {
  912. case ShaderModel::Kind::Vertex: {
  913. pInfo->VS.OutputPositionPresent = 0;
  914. DxilSignature &S = pModule->GetOutputSignature();
  915. for (auto &&E : S.GetElements()) {
  916. if (E->GetKind() == Semantic::Kind::Position) {
  917. // Ideally, we might check never writes mask here,
  918. // but this is not yet part of the signature element in Dxil
  919. pInfo->VS.OutputPositionPresent = 1;
  920. break;
  921. }
  922. }
  923. break;
  924. }
  925. case ShaderModel::Kind::Hull: {
  926. pInfo->HS.InputControlPointCount = (UINT)pModule->GetInputControlPointCount();
  927. pInfo->HS.OutputControlPointCount = (UINT)pModule->GetOutputControlPointCount();
  928. pInfo->HS.TessellatorDomain = (UINT)pModule->GetTessellatorDomain();
  929. pInfo->HS.TessellatorOutputPrimitive = (UINT)pModule->GetTessellatorOutputPrimitive();
  930. break;
  931. }
  932. case ShaderModel::Kind::Domain: {
  933. pInfo->DS.InputControlPointCount = (UINT)pModule->GetInputControlPointCount();
  934. pInfo->DS.OutputPositionPresent = 0;
  935. DxilSignature &S = pModule->GetOutputSignature();
  936. for (auto &&E : S.GetElements()) {
  937. if (E->GetKind() == Semantic::Kind::Position) {
  938. // Ideally, we might check never writes mask here,
  939. // but this is not yet part of the signature element in Dxil
  940. pInfo->DS.OutputPositionPresent = 1;
  941. break;
  942. }
  943. }
  944. pInfo->DS.TessellatorDomain = (UINT)pModule->GetTessellatorDomain();
  945. break;
  946. }
  947. case ShaderModel::Kind::Geometry: {
  948. pInfo->GS.InputPrimitive = (UINT)pModule->GetInputPrimitive();
  949. // NOTE: For OutputTopology, pick one from a used stream, or if none
  950. // are used, use stream 0, and set OutputStreamMask to 1.
  951. pInfo->GS.OutputTopology = (UINT)pModule->GetStreamPrimitiveTopology();
  952. pInfo->GS.OutputStreamMask = pModule->GetActiveStreamMask();
  953. pInfo->GS.OutputPositionPresent = 0;
  954. DxilSignature &S = pModule->GetOutputSignature();
  955. for (auto &&E : S.GetElements()) {
  956. if (E->GetKind() == Semantic::Kind::Position) {
  957. // Ideally, we might check never writes mask here,
  958. // but this is not yet part of the signature element in Dxil
  959. pInfo->GS.OutputPositionPresent = 1;
  960. break;
  961. }
  962. }
  963. break;
  964. }
  965. case ShaderModel::Kind::Pixel: {
  966. pInfo->PS.DepthOutput = 0;
  967. pInfo->PS.SampleFrequency = 0;
  968. {
  969. DxilSignature &S = pModule->GetInputSignature();
  970. for (auto &&E : S.GetElements()) {
  971. if (E->GetInterpolationMode()->IsAnySample() ||
  972. E->GetKind() == Semantic::Kind::SampleIndex) {
  973. pInfo->PS.SampleFrequency = 1;
  974. break;
  975. }
  976. }
  977. }
  978. {
  979. DxilSignature &S = pModule->GetOutputSignature();
  980. for (auto &&E : S.GetElements()) {
  981. if (E->IsAnyDepth()) {
  982. pInfo->PS.DepthOutput = 1;
  983. break;
  984. }
  985. }
  986. }
  987. break;
  988. }
  989. }
  990. // Set resource binding information
  991. UINT uResIndex = 0;
  992. for (auto &&R : pModule->GetCBuffers()) {
  993. DXASSERT_LOCALVAR_NOMSG(uTotalResources, uResIndex < uTotalResources);
  994. PSVResourceBindInfo0 *pBindInfo = PSV.GetPSVResourceBindInfo0(uResIndex);
  995. DXASSERT_NOMSG(pBindInfo);
  996. pBindInfo->ResType = (UINT)PSVResourceType::CBV;
  997. pBindInfo->Space = R->GetSpaceID();
  998. pBindInfo->LowerBound = R->GetLowerBound();
  999. pBindInfo->UpperBound = R->GetUpperBound();
  1000. uResIndex++;
  1001. }
  1002. for (auto &&R : pModule->GetSamplers()) {
  1003. DXASSERT_NOMSG(uResIndex < uTotalResources);
  1004. PSVResourceBindInfo0 *pBindInfo = PSV.GetPSVResourceBindInfo0(uResIndex);
  1005. DXASSERT_NOMSG(pBindInfo);
  1006. pBindInfo->ResType = (UINT)PSVResourceType::Sampler;
  1007. pBindInfo->Space = R->GetSpaceID();
  1008. pBindInfo->LowerBound = R->GetLowerBound();
  1009. pBindInfo->UpperBound = R->GetUpperBound();
  1010. uResIndex++;
  1011. }
  1012. for (auto &&R : pModule->GetSRVs()) {
  1013. DXASSERT_NOMSG(uResIndex < uTotalResources);
  1014. PSVResourceBindInfo0 *pBindInfo = PSV.GetPSVResourceBindInfo0(uResIndex);
  1015. DXASSERT_NOMSG(pBindInfo);
  1016. if (R->IsStructuredBuffer()) {
  1017. pBindInfo->ResType = (UINT)PSVResourceType::SRVStructured;
  1018. } else if (R->IsRawBuffer()) {
  1019. pBindInfo->ResType = (UINT)PSVResourceType::SRVRaw;
  1020. } else {
  1021. pBindInfo->ResType = (UINT)PSVResourceType::SRVTyped;
  1022. }
  1023. pBindInfo->Space = R->GetSpaceID();
  1024. pBindInfo->LowerBound = R->GetLowerBound();
  1025. pBindInfo->UpperBound = R->GetUpperBound();
  1026. uResIndex++;
  1027. }
  1028. for (auto &&R : pModule->GetUAVs()) {
  1029. DXASSERT_NOMSG(uResIndex < uTotalResources);
  1030. PSVResourceBindInfo0 *pBindInfo = PSV.GetPSVResourceBindInfo0(uResIndex);
  1031. DXASSERT_NOMSG(pBindInfo);
  1032. if (R->IsStructuredBuffer()) {
  1033. if (R->HasCounter())
  1034. pBindInfo->ResType = (UINT)PSVResourceType::UAVStructuredWithCounter;
  1035. else
  1036. pBindInfo->ResType = (UINT)PSVResourceType::UAVStructured;
  1037. } else if (R->IsRawBuffer()) {
  1038. pBindInfo->ResType = (UINT)PSVResourceType::UAVRaw;
  1039. } else {
  1040. pBindInfo->ResType = (UINT)PSVResourceType::UAVTyped;
  1041. }
  1042. pBindInfo->Space = R->GetSpaceID();
  1043. pBindInfo->LowerBound = R->GetLowerBound();
  1044. pBindInfo->UpperBound = R->GetUpperBound();
  1045. uResIndex++;
  1046. }
  1047. DXASSERT_NOMSG(uResIndex == uTotalResources);
  1048. }
  1049. void DxbcConverter::AnalyzeShader(D3D10ShaderBinary::CShaderCodeParser &Parser) {
  1050. // Parse shader model.
  1051. D3D10_SB_TOKENIZED_PROGRAM_TYPE ShaderType = Parser.ShaderType();
  1052. m_DxbcMajor = Parser.ShaderMajorVersion();
  1053. m_DxbcMinor = Parser.ShaderMinorVersion();
  1054. ShaderModel::Kind ShaderKind = DXBC::GetShaderModelKind(ShaderType);
  1055. // The converter always promotes the shader version to 6.0.
  1056. m_pSM = ShaderModel::Get(ShaderKind, 6, 0);
  1057. m_pPR->SetShaderModel(m_pSM);
  1058. // By default refactoring is disallowed, unless we encounter
  1059. // dcl_globalflags allowRefactoring
  1060. m_pPR->m_ShaderFlags.SetDisableMathRefactoring(true);
  1061. // By default, all resources are assumed bound for SM5.0 shaders,
  1062. // unless we encounter interface declarations
  1063. m_pPR->m_ShaderFlags.SetAllResourcesBound(true);
  1064. // Setup signature helpers.
  1065. m_pInputSignature.reset(new SignatureHelper(m_pSM->GetKind(), DXIL::SignatureKind::Input));
  1066. m_pOutputSignature.reset(new SignatureHelper(m_pSM->GetKind(), DXIL::SignatureKind::Output));
  1067. m_pPatchConstantSignature.reset(new SignatureHelper(m_pSM->GetKind(), DXIL::SignatureKind::PatchConstOrPrim));
  1068. // Collect:
  1069. // 1. Declarations
  1070. // 2. Labels
  1071. // Declare:
  1072. // 1. Global symbols for resources/samplers.
  1073. // 2. Their types.
  1074. BYTE CurrentOutputStream = 0;
  1075. unsigned MaxOutputRegister = 0;
  1076. m_bControlPointPhase = false;
  1077. bool bPatchConstantPhase = false;
  1078. D3D10ShaderBinary::CInstruction Inst;
  1079. while(!Parser.EndOfShader()) {
  1080. Parser.ParseInstruction(&Inst);
  1081. switch (Inst.OpCode()) {
  1082. case D3D10_SB_OPCODE_DCL_CONSTANT_BUFFER: {
  1083. // Record this cbuffer declaration in DxilModule.
  1084. unsigned ID = m_pPR->AddCBuffer(unique_ptr<DxilCBuffer>(new DxilCBuffer));
  1085. DxilCBuffer &R = m_pPR->GetCBuffer(ID); // R == record
  1086. R.SetID(ID);
  1087. // Root signature bindings.
  1088. unsigned RangeID = Inst.m_Operands[0].m_Index[0].m_RegIndex;
  1089. unsigned CBufferSize = Inst.m_ConstantBufferDecl.Size * DXBC::kWidth * 4;
  1090. unsigned LB, RangeSize;
  1091. switch (Inst.m_Operands[0].m_IndexDimension) {
  1092. case D3D10_SB_OPERAND_INDEX_2D: // SM 5.0-
  1093. LB = RangeID;
  1094. RangeSize = 1;
  1095. break;
  1096. case D3D10_SB_OPERAND_INDEX_3D: // SM 5.1
  1097. LB = Inst.m_Operands[0].m_Index[1].m_RegIndex;
  1098. RangeSize = Inst.m_Operands[0].m_Index[2].m_RegIndex != UINT_MAX ? Inst.m_Operands[0].m_Index[2].m_RegIndex - LB + 1 : UINT_MAX;
  1099. break;
  1100. default:
  1101. DXASSERT_DXBC(false);
  1102. IFTARG(NULL);
  1103. }
  1104. R.SetLowerBound(LB);
  1105. R.SetRangeSize(RangeSize);
  1106. R.SetSpaceID(Inst.m_ConstantBufferDecl.Space);
  1107. // Declare global variable.
  1108. R.SetGlobalName(SynthesizeResGVName("CB", R.GetID()));
  1109. StructType *pResType = GetStructResElemType(CBufferSize);
  1110. R.SetGlobalSymbol(DeclareUndefPtr(pResType, DXIL::kCBufferAddrSpace));
  1111. // CBuffer-specific state.
  1112. R.SetSize(CBufferSize);
  1113. //R.SetImmediateIndexed(Inst.m_ConstantBufferDecl.AccessPattern == D3D10_SB_CONSTANT_BUFFER_IMMEDIATE_INDEXED);
  1114. // Record shader register/rangeID mapping for upcoming instruction conversion.
  1115. DXASSERT(m_CBufferRangeMap.find(RangeID) == m_CBufferRangeMap.end(), "otherwise overlapping declarations");
  1116. m_CBufferRangeMap[RangeID] = R.GetID();
  1117. break;
  1118. }
  1119. case D3D10_SB_OPCODE_DCL_SAMPLER: {
  1120. // Record this sampler declaration in DxilModule.
  1121. unsigned ID = m_pPR->AddSampler(unique_ptr<DxilSampler>(new DxilSampler));
  1122. DxilSampler &R = m_pPR->GetSampler(ID); // R == record
  1123. R.SetID(ID);
  1124. // Root signature bindings.
  1125. unsigned RangeID = Inst.m_Operands[0].m_Index[0].m_RegIndex;
  1126. unsigned LB, RangeSize;
  1127. switch (Inst.m_Operands[0].m_IndexDimension) {
  1128. case D3D10_SB_OPERAND_INDEX_1D: // SM 5.0-
  1129. LB = RangeID;
  1130. RangeSize = 1;
  1131. break;
  1132. case D3D10_SB_OPERAND_INDEX_3D: // SM 5.1
  1133. LB = Inst.m_Operands[0].m_Index[1].m_RegIndex;
  1134. RangeSize = Inst.m_Operands[0].m_Index[2].m_RegIndex != UINT_MAX ? Inst.m_Operands[0].m_Index[2].m_RegIndex - LB + 1 : UINT_MAX;
  1135. break;
  1136. default:
  1137. DXASSERT_DXBC(false);
  1138. IFTARG(NULL);
  1139. }
  1140. R.SetLowerBound(LB);
  1141. R.SetRangeSize(RangeSize);
  1142. R.SetSpaceID(Inst.m_SamplerDecl.Space);
  1143. // Declare global variable.
  1144. R.SetGlobalName(SynthesizeResGVName("S", R.GetID()));
  1145. string ResTypeName("dx.types.Sampler");
  1146. StructType *pResType = m_pModule->getTypeByName(ResTypeName);
  1147. if (pResType == nullptr) {
  1148. pResType = StructType::create(m_Ctx, ResTypeName);
  1149. }
  1150. R.SetGlobalSymbol(DeclareUndefPtr(pResType, DXIL::kDeviceMemoryAddrSpace));
  1151. // Sampler-specific state.
  1152. R.SetSamplerKind(DXBC::GetSamplerKind(Inst.m_SamplerDecl.SamplerMode));
  1153. // Record shader register/rangeID mapping for upcoming instruction conversion.
  1154. DXASSERT(m_SamplerRangeMap.find(RangeID) == m_SamplerRangeMap.end(), "otherwise overlapping declarations");
  1155. m_SamplerRangeMap[RangeID] = R.GetID();
  1156. break;
  1157. }
  1158. case D3D10_SB_OPCODE_DCL_RESOURCE:
  1159. case D3D11_SB_OPCODE_DCL_RESOURCE_RAW:
  1160. case D3D11_SB_OPCODE_DCL_RESOURCE_STRUCTURED: {
  1161. // Record this SRV declaration in DxilModule.
  1162. unsigned ID = m_pPR->AddSRV(unique_ptr<DxilResource>(new DxilResource));
  1163. DxilResource &R = m_pPR->GetSRV(ID); // R == record
  1164. R.SetID(ID);
  1165. R.SetRW(false);
  1166. // Root signature bindings.
  1167. unsigned RangeID = Inst.m_Operands[0].m_Index[0].m_RegIndex;
  1168. unsigned LB, RangeSize;
  1169. if (IsSM51Plus()) {
  1170. LB = Inst.m_Operands[0].m_Index[1].m_RegIndex;
  1171. RangeSize = Inst.m_Operands[0].m_Index[2].m_RegIndex != UINT_MAX ? Inst.m_Operands[0].m_Index[2].m_RegIndex - LB + 1 : UINT_MAX;
  1172. } else {
  1173. LB = RangeID;
  1174. RangeSize = 1;
  1175. }
  1176. R.SetLowerBound(LB);
  1177. R.SetRangeSize(RangeSize);
  1178. // Resource-specific state.
  1179. StructType *pResType = nullptr;
  1180. switch (Inst.OpCode()) {
  1181. case D3D10_SB_OPCODE_DCL_RESOURCE: {
  1182. R.SetSpaceID(Inst.m_ResourceDecl.Space);
  1183. R.SetKind(DXBC::GetResourceKind(Inst.m_ResourceDecl.Dimension));
  1184. const unsigned kTypedBufferElementSizeInBytes = 4;
  1185. R.SetElementStride(kTypedBufferElementSizeInBytes);
  1186. R.SetSampleCount(Inst.m_ResourceDecl.SampleCount);
  1187. CompType DeclCT = DXBC::GetDeclResCompType(Inst.m_ResourceDecl.ReturnType[0]);
  1188. if (DeclCT.IsInvalid()) DeclCT = CompType::getU32();
  1189. R.SetCompType(DeclCT);
  1190. pResType = GetTypedResElemType(DeclCT);
  1191. break;
  1192. }
  1193. case D3D11_SB_OPCODE_DCL_RESOURCE_RAW: {
  1194. R.SetSpaceID(Inst.m_RawSRVDecl.Space);
  1195. R.SetKind(DxilResource::Kind::RawBuffer);
  1196. const unsigned kRawBufferElementSizeInBytes = 1;
  1197. R.SetElementStride(kRawBufferElementSizeInBytes);
  1198. pResType = GetTypedResElemType(CompType::getU32());
  1199. break;
  1200. }
  1201. case D3D11_SB_OPCODE_DCL_RESOURCE_STRUCTURED: {
  1202. R.SetSpaceID(Inst.m_StructuredSRVDecl.Space);
  1203. R.SetKind(DxilResource::Kind::StructuredBuffer);
  1204. unsigned Stride = Inst.m_StructuredSRVDecl.ByteStride;
  1205. R.SetElementStride(Stride);
  1206. pResType = GetStructResElemType(Stride);
  1207. break;
  1208. }
  1209. default: ;
  1210. }
  1211. // Declare global variable.
  1212. R.SetGlobalName(SynthesizeResGVName("T", R.GetID()));
  1213. R.SetGlobalSymbol(DeclareUndefPtr(pResType, DXIL::kDeviceMemoryAddrSpace));
  1214. // Record shader register/rangeID mapping for upcoming instruction conversion.
  1215. DXASSERT(m_SRVRangeMap.find(RangeID) == m_SRVRangeMap.end(), "otherwise overlapping declarations");
  1216. m_SRVRangeMap[RangeID] = R.GetID();
  1217. break;
  1218. }
  1219. case D3D11_SB_OPCODE_DCL_UNORDERED_ACCESS_VIEW_TYPED:
  1220. case D3D11_SB_OPCODE_DCL_UNORDERED_ACCESS_VIEW_RAW:
  1221. case D3D11_SB_OPCODE_DCL_UNORDERED_ACCESS_VIEW_STRUCTURED: {
  1222. // Record this UAV declaration in DxilModule.
  1223. unsigned ID = m_pPR->AddUAV(unique_ptr<DxilResource>(new DxilResource));
  1224. DxilResource &R = m_pPR->GetUAV(ID); // R == record
  1225. R.SetID(ID);
  1226. R.SetRW(true);
  1227. // Root signature bindings.
  1228. unsigned RangeID = Inst.m_Operands[0].m_Index[0].m_RegIndex;
  1229. unsigned LB, RangeSize;
  1230. if (IsSM51Plus()) {
  1231. LB = Inst.m_Operands[0].m_Index[1].m_RegIndex;
  1232. RangeSize = Inst.m_Operands[0].m_Index[2].m_RegIndex != UINT_MAX ? Inst.m_Operands[0].m_Index[2].m_RegIndex - LB + 1 : UINT_MAX;
  1233. } else {
  1234. LB = RangeID;
  1235. RangeSize = 1;
  1236. }
  1237. R.SetLowerBound(LB);
  1238. R.SetRangeSize(RangeSize);
  1239. // Resource-specific state.
  1240. string GVTypeName;
  1241. raw_string_ostream GVTypeNameStream(GVTypeName);
  1242. StructType *pResType = nullptr;
  1243. unsigned Flags = 0;
  1244. switch (Inst.OpCode()) {
  1245. case D3D11_SB_OPCODE_DCL_UNORDERED_ACCESS_VIEW_TYPED: {
  1246. R.SetSpaceID(Inst.m_TypedUAVDecl.Space);
  1247. Flags = Inst.m_TypedUAVDecl.Flags;
  1248. R.SetKind(DXBC::GetResourceKind(Inst.m_TypedUAVDecl.Dimension));
  1249. const unsigned kTypedBufferElementSizeInBytes = 4;
  1250. R.SetElementStride(kTypedBufferElementSizeInBytes);
  1251. CompType DeclCT = DXBC::GetDeclResCompType(Inst.m_TypedUAVDecl.ReturnType[0]);
  1252. if (DeclCT.IsInvalid()) DeclCT = CompType::getU32();
  1253. R.SetCompType(DeclCT);
  1254. pResType = GetTypedResElemType(DeclCT);
  1255. break;
  1256. }
  1257. case D3D11_SB_OPCODE_DCL_UNORDERED_ACCESS_VIEW_RAW: {
  1258. R.SetSpaceID(Inst.m_RawUAVDecl.Space);
  1259. R.SetKind(DxilResource::Kind::RawBuffer);
  1260. Flags = Inst.m_RawUAVDecl.Flags;
  1261. const unsigned kRawBufferElementSizeInBytes = 1;
  1262. R.SetElementStride(kRawBufferElementSizeInBytes);
  1263. pResType = GetTypedResElemType(CompType::getU32());
  1264. break;
  1265. }
  1266. case D3D11_SB_OPCODE_DCL_UNORDERED_ACCESS_VIEW_STRUCTURED: {
  1267. R.SetSpaceID(Inst.m_StructuredUAVDecl.Space);
  1268. R.SetKind(DxilResource::Kind::StructuredBuffer);
  1269. Flags = Inst.m_StructuredUAVDecl.Flags;
  1270. unsigned Stride = Inst.m_StructuredUAVDecl.ByteStride;
  1271. R.SetElementStride(Stride);
  1272. pResType = GetStructResElemType(Stride);
  1273. break;
  1274. }
  1275. default: ;
  1276. }
  1277. R.SetGloballyCoherent((Flags & D3D11_SB_GLOBALLY_COHERENT_ACCESS) != 0);
  1278. R.SetHasCounter((Flags & D3D11_SB_UAV_HAS_ORDER_PRESERVING_COUNTER) != 0);
  1279. R.SetROV((Flags & D3D11_SB_RASTERIZER_ORDERED_ACCESS) != 0);
  1280. // Declare global variable.
  1281. R.SetGlobalName(SynthesizeResGVName("U", R.GetID()));
  1282. R.SetGlobalSymbol(DeclareUndefPtr(pResType, DXIL::kDeviceMemoryAddrSpace));
  1283. // Record shader register/rangeID mapping for upcoming instruction conversion.
  1284. DXASSERT(m_UAVRangeMap.find(RangeID) == m_UAVRangeMap.end(), "otherwise overlapping declarations");
  1285. m_UAVRangeMap[RangeID] = R.GetID();
  1286. break;
  1287. }
  1288. case D3D10_SB_OPCODE_DCL_INDEX_RANGE: {
  1289. unsigned RowRegIdx = (Inst.m_Operands[0].m_IndexDimension == D3D10_SB_OPERAND_INDEX_1D) ? 0 : 1;
  1290. SignatureHelper::Range R;
  1291. R.StartRow = Inst.m_Operands[0].m_Index[RowRegIdx].m_RegIndex;
  1292. R.StartCol = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask).GetFirstActiveComp();
  1293. R.Rows = Inst.m_IndexRangeDecl.RegCount;
  1294. R.Cols = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask).GetNumActiveRangeComps();
  1295. R.OutputStream = CurrentOutputStream;
  1296. switch (Inst.m_Operands[0].m_Type) {
  1297. case D3D10_SB_OPERAND_TYPE_INPUT:
  1298. m_pInputSignature->m_Ranges.emplace_back(R);
  1299. break;
  1300. case D3D10_SB_OPERAND_TYPE_OUTPUT:
  1301. if (!m_pSM->IsHS() || m_bControlPointPhase) {
  1302. m_pOutputSignature->m_Ranges.emplace_back(R);
  1303. } else {
  1304. DXASSERT_NOMSG(m_pSM->IsHS() && bPatchConstantPhase);
  1305. m_pPatchConstantSignature->m_Ranges.emplace_back(R);
  1306. }
  1307. break;
  1308. case D3D11_SB_OPERAND_TYPE_INPUT_PATCH_CONSTANT:
  1309. DXASSERT_DXBC(m_pSM->IsHS() || m_pSM->IsDS());
  1310. m_pPatchConstantSignature->m_Ranges.emplace_back(R);
  1311. break;
  1312. case D3D11_SB_OPERAND_TYPE_INPUT_CONTROL_POINT:
  1313. DXASSERT_DXBC(m_pSM->IsHS() || m_pSM->IsDS());
  1314. m_pInputSignature->m_Ranges.emplace_back(R);
  1315. break;
  1316. default:
  1317. DXASSERT_DXBC(false);
  1318. }
  1319. break;
  1320. }
  1321. case D3D10_SB_OPCODE_DCL_GS_INPUT_PRIMITIVE:
  1322. m_pPR->SetInputPrimitive(DXBC::GetInputPrimitive(Inst.m_InputPrimitiveDecl.Primitive));
  1323. break;
  1324. case D3D10_SB_OPCODE_DCL_GS_OUTPUT_PRIMITIVE_TOPOLOGY:
  1325. m_pPR->SetStreamPrimitiveTopology(DXBC::GetPrimitiveTopology(Inst.m_OutputTopologyDecl.Topology));
  1326. break;
  1327. case D3D10_SB_OPCODE_DCL_MAX_OUTPUT_VERTEX_COUNT:
  1328. m_pPR->SetMaxVertexCount(Inst.m_GSMaxOutputVertexCountDecl.MaxOutputVertexCount);
  1329. break;
  1330. case D3D10_SB_OPCODE_DCL_INPUT: {
  1331. D3D10_SB_OPERAND_TYPE RegType = Inst.m_Operands[0].m_Type;
  1332. switch (RegType) {
  1333. case D3D11_SB_OPERAND_TYPE_INPUT_COVERAGE_MASK:
  1334. m_pInputSignature->m_bHasInputCoverage = true;
  1335. break;
  1336. case D3D11_SB_OPERAND_TYPE_INNER_COVERAGE:
  1337. m_pInputSignature->m_bHasInnerInputCoverage = true;
  1338. break;
  1339. case D3D11_SB_OPERAND_TYPE_INPUT_THREAD_ID:
  1340. case D3D11_SB_OPERAND_TYPE_INPUT_THREAD_GROUP_ID:
  1341. case D3D11_SB_OPERAND_TYPE_INPUT_THREAD_ID_IN_GROUP:
  1342. case D3D11_SB_OPERAND_TYPE_INPUT_THREAD_ID_IN_GROUP_FLATTENED:
  1343. case D3D11_SB_OPERAND_TYPE_INPUT_DOMAIN_POINT:
  1344. case D3D11_SB_OPERAND_TYPE_OUTPUT_CONTROL_POINT_ID:
  1345. case D3D10_SB_OPERAND_TYPE_INPUT_PRIMITIVEID:
  1346. case D3D11_SB_OPERAND_TYPE_INPUT_FORK_INSTANCE_ID:
  1347. case D3D11_SB_OPERAND_TYPE_INPUT_JOIN_INSTANCE_ID:
  1348. case D3D11_SB_OPERAND_TYPE_CYCLE_COUNTER:
  1349. case D3D11_SB_OPERAND_TYPE_INPUT_GS_INSTANCE_ID:
  1350. break;
  1351. default: {
  1352. unsigned NumUnits, Row;
  1353. switch (Inst.m_Operands[0].m_IndexDimension) {
  1354. case D3D10_SB_OPERAND_INDEX_1D:
  1355. NumUnits = 0;
  1356. Row = Inst.m_Operands[0].m_Index[0].m_RegIndex;
  1357. break;
  1358. case D3D10_SB_OPERAND_INDEX_2D:
  1359. NumUnits = Inst.m_Operands[0].m_Index[0].m_RegIndex;
  1360. Row = Inst.m_Operands[0].m_Index[1].m_RegIndex;
  1361. break;
  1362. default:
  1363. DXASSERT(false, "there should no other index dimensions");
  1364. }
  1365. SignatureHelper::UsedElement E;
  1366. E.NumUnits = NumUnits;
  1367. E.Row = Row;
  1368. E.StartCol = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask).GetFirstActiveComp();
  1369. E.Cols = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask).GetNumActiveRangeComps();
  1370. E.InterpolationMode = D3D_INTERPOLATION_UNDEFINED;
  1371. E.MinPrecision = Inst.m_Operands[0].m_MinPrecision;
  1372. if (RegType == D3D10_SB_OPERAND_TYPE_INPUT) {
  1373. m_pInputSignature->m_UsedElements.emplace_back(E);
  1374. } else {
  1375. if (m_pSM->IsDS()) {
  1376. switch (RegType) {
  1377. case D3D11_SB_OPERAND_TYPE_INPUT_CONTROL_POINT:
  1378. m_pInputSignature->m_UsedElements.emplace_back(E);
  1379. break;
  1380. case D3D11_SB_OPERAND_TYPE_INPUT_PATCH_CONSTANT:
  1381. m_pPatchConstantSignature->m_UsedElements.emplace_back(E);
  1382. break;
  1383. default:
  1384. DXASSERT(false, "check unsupported case");
  1385. break;
  1386. }
  1387. }
  1388. if (m_pSM->IsHS()) {
  1389. switch (RegType) {
  1390. case D3D11_SB_OPERAND_TYPE_INPUT_CONTROL_POINT:
  1391. m_pInputSignature->m_UsedElements.emplace_back(E);
  1392. break;
  1393. case D3D11_SB_OPERAND_TYPE_INPUT_PATCH_CONSTANT:
  1394. break;
  1395. case D3D11_SB_OPERAND_TYPE_OUTPUT_CONTROL_POINT:
  1396. break;
  1397. default:
  1398. DXASSERT(false, "check unsupported case");
  1399. break;
  1400. }
  1401. }
  1402. }
  1403. break;
  1404. }
  1405. }
  1406. break;
  1407. }
  1408. case D3D10_SB_OPCODE_DCL_INPUT_SGV: {
  1409. SignatureHelper::UsedElement E;
  1410. E.Row = Inst.m_Operands[0].m_Index[0].m_RegIndex;
  1411. E.StartCol = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask).GetFirstActiveComp();
  1412. E.Cols = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask).GetNumActiveRangeComps();
  1413. E.InterpolationMode = D3D_INTERPOLATION_UNDEFINED;
  1414. E.MinPrecision = Inst.m_Operands[0].m_MinPrecision;
  1415. m_pInputSignature->m_UsedElements.emplace_back(E);
  1416. break;
  1417. }
  1418. case D3D10_SB_OPCODE_DCL_INPUT_SIV: {
  1419. unsigned NumUnits = 0;
  1420. unsigned Row = Inst.m_Operands[0].m_Index[0].m_RegIndex;
  1421. if (m_pSM->IsGS()) {
  1422. NumUnits = Inst.m_Operands[0].m_Index[0].m_RegIndex;
  1423. Row = Inst.m_Operands[0].m_Index[1].m_RegIndex;
  1424. }
  1425. SignatureHelper::UsedElement E;
  1426. E.NumUnits = NumUnits;
  1427. E.Row = Row;
  1428. E.StartCol = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask).GetFirstActiveComp();
  1429. E.Cols = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask).GetNumActiveRangeComps();
  1430. E.InterpolationMode = D3D_INTERPOLATION_UNDEFINED;
  1431. E.MinPrecision = Inst.m_Operands[0].m_MinPrecision;
  1432. switch (Inst.m_Operands[0].m_Type) {
  1433. case D3D10_SB_OPERAND_TYPE_INPUT:
  1434. m_pInputSignature->m_UsedElements.emplace_back(E);
  1435. break;
  1436. case D3D11_SB_OPERAND_TYPE_INPUT_PATCH_CONSTANT:
  1437. m_pPatchConstantSignature->m_UsedElements.emplace_back(E);
  1438. break;
  1439. default:
  1440. DXASSERT(false, "missing case");
  1441. break;
  1442. }
  1443. break;
  1444. }
  1445. case D3D10_SB_OPCODE_DCL_INPUT_PS: {
  1446. SignatureHelper::UsedElement E;
  1447. E.Row = Inst.m_Operands[0].m_Index[0].m_RegIndex;
  1448. E.StartCol = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask).GetFirstActiveComp();
  1449. E.Cols = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask).GetNumActiveRangeComps();
  1450. E.InterpolationMode = (D3D_INTERPOLATION_MODE)Inst.m_InputPSDecl.InterpolationMode;
  1451. E.MinPrecision = Inst.m_Operands[0].m_MinPrecision;
  1452. m_pInputSignature->m_UsedElements.emplace_back(E);
  1453. break;
  1454. }
  1455. case D3D10_SB_OPCODE_DCL_INPUT_PS_SGV: {
  1456. SignatureHelper::UsedElement E;
  1457. E.Row = Inst.m_Operands[0].m_Index[0].m_RegIndex;
  1458. E.StartCol = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask).GetFirstActiveComp();
  1459. E.Cols = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask).GetNumActiveRangeComps();
  1460. E.InterpolationMode = (D3D_INTERPOLATION_MODE)Inst.m_InputPSDeclSGV.InterpolationMode;
  1461. E.MinPrecision = Inst.m_Operands[0].m_MinPrecision;
  1462. m_pInputSignature->m_UsedElements.emplace_back(E);
  1463. break;
  1464. }
  1465. case D3D10_SB_OPCODE_DCL_INPUT_PS_SIV: {
  1466. SignatureHelper::UsedElement E;
  1467. E.Row = Inst.m_Operands[0].m_Index[0].m_RegIndex;
  1468. E.StartCol = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask).GetFirstActiveComp();
  1469. E.Cols = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask).GetNumActiveRangeComps();
  1470. E.InterpolationMode = (D3D_INTERPOLATION_MODE)Inst.m_InputPSDeclSIV.InterpolationMode;
  1471. E.MinPrecision = Inst.m_Operands[0].m_MinPrecision;
  1472. m_pInputSignature->m_UsedElements.emplace_back(E);
  1473. break;
  1474. }
  1475. case D3D10_SB_OPCODE_DCL_OUTPUT: {
  1476. D3D10_SB_OPERAND_TYPE RegType = Inst.m_Operands[0].m_Type;
  1477. switch (RegType) {
  1478. case D3D10_SB_OPERAND_TYPE_OUTPUT_DEPTH:
  1479. case D3D11_SB_OPERAND_TYPE_OUTPUT_DEPTH_GREATER_EQUAL:
  1480. case D3D11_SB_OPERAND_TYPE_OUTPUT_DEPTH_LESS_EQUAL:
  1481. m_DepthRegType = RegType;
  1482. __fallthrough;
  1483. case D3D11_SB_OPERAND_TYPE_OUTPUT_STENCIL_REF:
  1484. case D3D10_SB_OPERAND_TYPE_OUTPUT_COVERAGE_MASK: {
  1485. m_bHasStencilRef = RegType == D3D11_SB_OPERAND_TYPE_OUTPUT_STENCIL_REF;
  1486. m_bHasCoverageOut = RegType == D3D10_SB_OPERAND_TYPE_OUTPUT_COVERAGE_MASK;
  1487. SignatureHelper::UsedElement E;
  1488. E.Row = Semantic::kUndefinedRow;
  1489. E.StartCol = 0;
  1490. E.Cols = 1;
  1491. E.InterpolationMode = D3D_INTERPOLATION_UNDEFINED;
  1492. E.MinPrecision = Inst.m_Operands[0].m_MinPrecision;
  1493. m_pOutputSignature->m_UsedElements.emplace_back(E);
  1494. break;
  1495. }
  1496. default: {
  1497. SignatureHelper::UsedElement E;
  1498. E.Row = Inst.m_Operands[0].m_Index[0].m_RegIndex;
  1499. E.StartCol = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask).GetFirstActiveComp();
  1500. E.Cols = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask).GetNumActiveRangeComps();
  1501. E.InterpolationMode = D3D_INTERPOLATION_UNDEFINED;
  1502. E.MinPrecision = Inst.m_Operands[0].m_MinPrecision;
  1503. E.OutputStream = CurrentOutputStream;
  1504. if (!m_pSM->IsHS() || m_bControlPointPhase) {
  1505. m_pOutputSignature->m_UsedElements.emplace_back(E);
  1506. } else {
  1507. DXASSERT_NOMSG(m_pSM->IsHS() && bPatchConstantPhase);
  1508. m_pPatchConstantSignature->m_UsedElements.emplace_back(E);
  1509. }
  1510. MaxOutputRegister = std::max(MaxOutputRegister, E.Row);
  1511. break;
  1512. }
  1513. }
  1514. break;
  1515. }
  1516. case D3D10_SB_OPCODE_DCL_OUTPUT_SGV:
  1517. case D3D10_SB_OPCODE_DCL_OUTPUT_SIV: {
  1518. SignatureHelper::UsedElement E;
  1519. E.Row = Inst.m_Operands[0].m_Index[0].m_RegIndex;
  1520. E.StartCol = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask).GetFirstActiveComp();
  1521. E.Cols = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask).GetNumActiveRangeComps();
  1522. E.InterpolationMode = D3D_INTERPOLATION_UNDEFINED;
  1523. E.MinPrecision = Inst.m_Operands[0].m_MinPrecision;
  1524. E.OutputStream = CurrentOutputStream;
  1525. if (!m_pSM->IsHS() || m_bControlPointPhase) {
  1526. m_pOutputSignature->m_UsedElements.emplace_back(E);
  1527. } else {
  1528. DXASSERT_NOMSG(m_pSM->IsHS() && bPatchConstantPhase);
  1529. m_pPatchConstantSignature->m_UsedElements.emplace_back(E);
  1530. }
  1531. MaxOutputRegister = std::max(MaxOutputRegister, E.Row);
  1532. break;
  1533. }
  1534. case D3D10_SB_OPCODE_DCL_TEMPS:
  1535. m_NumTempRegs = std::max(m_NumTempRegs, Inst.m_TempsDecl.NumTemps);
  1536. break;
  1537. case D3D10_SB_OPCODE_DCL_INDEXABLE_TEMP: {
  1538. // Record x-register.
  1539. unsigned Reg = Inst.m_IndexableTempDecl.IndexableTempNumber;
  1540. unsigned NumRegs = Inst.m_IndexableTempDecl.NumRegisters;
  1541. CMask Mask = CMask::FromDXBC(Inst.m_IndexableTempDecl.Mask);
  1542. IndexableReg IR = { nullptr, nullptr, NumRegs, Mask.GetNumActiveRangeComps(), true };
  1543. if (!bPatchConstantPhase) {
  1544. // This is the main shader.
  1545. DXASSERT_DXBC(m_IndexableRegs.find(Reg) == m_IndexableRegs.end());
  1546. m_IndexableRegs[Reg] = IR;
  1547. } else {
  1548. // This is patch constant function.
  1549. // Can have dcl per phase
  1550. auto itIR = m_PatchConstantIndexableRegs.find(Reg);
  1551. if (itIR != m_PatchConstantIndexableRegs.end()) {
  1552. auto &theIR = itIR->second;
  1553. theIR.NumComps = std::max(theIR.NumComps, IR.NumComps);
  1554. theIR.NumComps = std::max(theIR.NumRegs, IR.NumRegs);
  1555. } else {
  1556. m_PatchConstantIndexableRegs[Reg] = IR;
  1557. }
  1558. }
  1559. break;
  1560. }
  1561. case D3D10_SB_OPCODE_DCL_GLOBAL_FLAGS:
  1562. SetShaderGlobalFlags(Inst.m_GlobalFlagsDecl.Flags);
  1563. break;
  1564. case D3D11_SB_OPCODE_DCL_STREAM: {
  1565. BYTE Stream = (BYTE)Inst.m_Operands[0].m_Index[0].m_RegIndex;
  1566. IFTBOOL(Stream < DXIL::kNumOutputStreams, DXC_E_INCORRECT_DXBC);
  1567. CurrentOutputStream = Stream;
  1568. m_pPR->SetStreamActive(Stream, true);
  1569. break;
  1570. }
  1571. case D3D11_SB_OPCODE_HS_DECLS:
  1572. break;
  1573. case D3D11_SB_OPCODE_DCL_INPUT_CONTROL_POINT_COUNT:
  1574. m_pPR->SetInputControlPointCount(Inst.m_InputControlPointCountDecl.InputControlPointCount);
  1575. break;
  1576. case D3D11_SB_OPCODE_DCL_OUTPUT_CONTROL_POINT_COUNT:
  1577. m_pPR->SetOutputControlPointCount(Inst.m_OutputControlPointCountDecl.OutputControlPointCount);
  1578. break;
  1579. case D3D11_SB_OPCODE_DCL_TESS_DOMAIN:
  1580. m_pPR->SetTessellatorDomain(DXBC::GetTessellatorDomain(Inst.m_TessellatorDomainDecl.TessellatorDomain));
  1581. break;
  1582. case D3D11_SB_OPCODE_DCL_TESS_PARTITIONING:
  1583. m_pPR->SetTessellatorPartitioning(DXBC::GetTessellatorPartitioning(Inst.m_TessellatorPartitioningDecl.TessellatorPartitioning));
  1584. break;
  1585. case D3D11_SB_OPCODE_DCL_TESS_OUTPUT_PRIMITIVE:
  1586. m_pPR->SetTessellatorOutputPrimitive(DXBC::GetTessellatorOutputPrimitive(Inst.m_TessellatorOutputPrimitiveDecl.TessellatorOutputPrimitive));
  1587. break;
  1588. case D3D11_SB_OPCODE_DCL_HS_MAX_TESSFACTOR:
  1589. m_pPR->SetMaxTessellationFactor(Inst.m_HSMaxTessFactorDecl.MaxTessFactor);
  1590. break;
  1591. case D3D11_SB_OPCODE_HS_CONTROL_POINT_PHASE:
  1592. DXASSERT_NOMSG(!m_bControlPointPhase && !bPatchConstantPhase);
  1593. m_bControlPointPhase = true;
  1594. break;
  1595. case D3D11_SB_OPCODE_HS_FORK_PHASE:
  1596. case D3D11_SB_OPCODE_HS_JOIN_PHASE:
  1597. m_bControlPointPhase = false;
  1598. bPatchConstantPhase = true;
  1599. m_PatchConstantPhaseInstanceCounts.push_back(1);
  1600. break;
  1601. case D3D11_SB_OPCODE_DCL_HS_FORK_PHASE_INSTANCE_COUNT:
  1602. m_PatchConstantPhaseInstanceCounts.back() = Inst.m_HSForkPhaseInstanceCountDecl.InstanceCount;
  1603. break;
  1604. case D3D11_SB_OPCODE_DCL_HS_JOIN_PHASE_INSTANCE_COUNT:
  1605. m_PatchConstantPhaseInstanceCounts.back() = Inst.m_HSJoinPhaseInstanceCountDecl.InstanceCount;
  1606. break;
  1607. case D3D11_SB_OPCODE_DCL_THREAD_GROUP:
  1608. m_pPR->SetNumThreads(Inst.m_ThreadGroupDecl.x,
  1609. Inst.m_ThreadGroupDecl.y,
  1610. Inst.m_ThreadGroupDecl.z);
  1611. break;
  1612. case D3D11_SB_OPCODE_DCL_THREAD_GROUP_SHARED_MEMORY_RAW:
  1613. case D3D11_SB_OPCODE_DCL_THREAD_GROUP_SHARED_MEMORY_STRUCTURED: {
  1614. TGSMEntry E;
  1615. E.Id = m_TGSMCount++;
  1616. if (Inst.OpCode() == D3D11_SB_OPCODE_DCL_THREAD_GROUP_SHARED_MEMORY_RAW) {
  1617. E.Stride = 1;
  1618. E.Count = Inst.m_RawTGSMDecl.ByteCount;
  1619. } else {
  1620. E.Stride = Inst.m_StructuredTGSMDecl.StructByteStride;
  1621. E.Count = Inst.m_StructuredTGSMDecl.StructCount;
  1622. }
  1623. // Declare global variable.
  1624. unsigned SizeInBytes = E.Stride*E.Count;
  1625. Type *pArrayType = ArrayType::get(Type::getInt8Ty(m_Ctx), SizeInBytes);
  1626. E.pVar = new GlobalVariable(*m_pModule, pArrayType,
  1627. false, GlobalValue::InternalLinkage,
  1628. UndefValue::get(pArrayType),
  1629. Twine("TGSM") + Twine(E.Id), nullptr,
  1630. GlobalVariable::NotThreadLocal, DXIL::kTGSMAddrSpace);
  1631. E.pVar->setAlignment(kRegCompAlignment);
  1632. // Mark GV as being used for LLVM.
  1633. m_pPR->GetLLVMUsed().push_back(E.pVar);
  1634. m_TGSMMap[Inst.m_Operands[0].m_Index[0].m_RegIndex] = E;
  1635. break;
  1636. }
  1637. case D3D11_SB_OPCODE_DCL_GS_INSTANCE_COUNT:
  1638. m_pPR->SetGSInstanceCount(Inst.m_GSInstanceCountDecl.InstanceCount);
  1639. break;
  1640. case D3D10_SB_OPCODE_CUSTOMDATA:
  1641. break;
  1642. case D3D11_SB_OPCODE_DCL_FUNCTION_BODY: {
  1643. DXASSERT_DXBC(Inst.m_NumOperands == 0);
  1644. unsigned FBIdx = Inst.m_FunctionBodyDecl.FunctionBodyNumber;
  1645. m_InterfaceFunctionBodies[FBIdx].pFunc = nullptr;
  1646. break;
  1647. }
  1648. case D3D11_SB_OPCODE_DCL_FUNCTION_TABLE: {
  1649. DXASSERT_DXBC(Inst.m_NumOperands == 0);
  1650. auto& FnTable = m_FunctionTables[Inst.m_FunctionTableDecl.FunctionTableNumber];
  1651. FnTable.assign(Inst.m_FunctionTableDecl.pFunctionIdentifiers, Inst.m_FunctionTableDecl.pFunctionIdentifiers + Inst.m_FunctionTableDecl.TableLength);
  1652. break;
  1653. }
  1654. case D3D11_SB_OPCODE_DCL_INTERFACE: {
  1655. DXASSERT_DXBC(Inst.m_NumOperands == 0);
  1656. auto& Iface = m_Interfaces[Inst.m_InterfaceDecl.InterfaceNumber];
  1657. Iface.Tables.assign(Inst.m_InterfaceDecl.pFunctionTableIdentifiers, Inst.m_InterfaceDecl.pFunctionTableIdentifiers + Inst.m_InterfaceDecl.TableLength);
  1658. #ifdef DBG
  1659. for (unsigned TableIdx : Iface.Tables) {
  1660. DXASSERT_DXBC(m_FunctionTables[TableIdx].size() == Inst.m_InterfaceDecl.ExpectedTableSize);
  1661. }
  1662. #endif
  1663. Iface.bDynamicallyIndexed = Inst.m_InterfaceDecl.bDynamicallyIndexed;
  1664. Iface.NumArrayEntries = Inst.m_InterfaceDecl.ArrayLength;
  1665. m_NumIfaces = std::max(m_NumIfaces, Inst.m_InterfaceDecl.InterfaceNumber + Iface.NumArrayEntries);
  1666. InsertInterfacesResourceDecls();
  1667. break;
  1668. }
  1669. case D3D10_SB_OPCODE_LABEL: {
  1670. m_bControlPointPhase = false;
  1671. bPatchConstantPhase = false;
  1672. DXASSERT_DXBC(Inst.m_NumOperands == 1);
  1673. DXASSERT_DXBC(Inst.m_Operands[0].m_Type == D3D10_SB_OPERAND_TYPE_LABEL ||
  1674. Inst.m_Operands[0].m_Type == D3D11_SB_OPERAND_TYPE_FUNCTION_BODY);
  1675. FunctionType *pFuncType = FunctionType::get(Type::getVoidTy(m_Ctx), false);
  1676. unsigned LabelIdx = Inst.m_Operands[0].m_Index[0].m_RegIndex;
  1677. LabelEntry Label;
  1678. const bool IsFb = Inst.m_Operands[0].m_Type == D3D11_SB_OPERAND_TYPE_FUNCTION_BODY;
  1679. auto& LabelMap = IsFb ? m_InterfaceFunctionBodies : m_Labels;
  1680. DXASSERT_DXBC((LabelMap.find(LabelIdx) == LabelMap.end()) == !IsFb); // Function bodies should be pre-declared, labels aren't
  1681. Label.pFunc = Function::Create(pFuncType, GlobalValue::LinkageTypes::InternalLinkage,
  1682. StringRef(IsFb ? "dx.fb." : "dx.label.") + Twine(LabelIdx), m_pModule.get());
  1683. Label.pFunc->setCallingConv(CallingConv::C);
  1684. LabelMap[LabelIdx] = Label;
  1685. break;
  1686. }
  1687. default:
  1688. break;
  1689. }
  1690. }
  1691. }
  1692. void DxbcConverter::ConvertInstructions(D3D10ShaderBinary::CShaderCodeParser &Parser) {
  1693. if (m_pPR->GetShaderModel()->IsGS()) {
  1694. // Set GS active stream mask.
  1695. if (m_pPR->GetActiveStreamMask() == 0
  1696. && !m_pPR->GetOutputSignature().GetElements().empty())
  1697. m_pPR->SetStreamActive(0, true);
  1698. // Make sure GS instance count is at least 1
  1699. if (m_pPR->GetGSInstanceCount() == 0)
  1700. m_pPR->SetGSInstanceCount(1);
  1701. }
  1702. // Add entry function declaration.
  1703. m_pPR->SetEntryFunctionName("main");
  1704. FunctionType *pEntryFuncType = FunctionType::get(Type::getVoidTy(m_Ctx), false);
  1705. Function *pFunction = Function::Create(pEntryFuncType, GlobalValue::LinkageTypes::ExternalLinkage,
  1706. m_pPR->GetEntryFunctionName(), m_pModule.get());
  1707. pFunction->setCallingConv(CallingConv::C);
  1708. m_pPR->SetEntryFunction(pFunction);
  1709. // Create main entry function.
  1710. BasicBlock *pBB = BasicBlock::Create(m_Ctx, "entry", pFunction);
  1711. m_pBuilder = std::make_unique< IRBuilder<> >(pBB);
  1712. FastMathFlags FMF;
  1713. if (!m_pPR->m_ShaderFlags.GetDisableMathRefactoring()) {
  1714. FMF.setUnsafeAlgebra();
  1715. }
  1716. m_pBuilder->SetFastMathFlags(FMF);
  1717. // Empty instruction stream.
  1718. if (Parser.EndOfShader()) {
  1719. m_pBuilder->CreateRetVoid();
  1720. return;
  1721. }
  1722. m_pUnusedF32 = UndefValue::get(Type::getFloatTy(m_Ctx));
  1723. m_pUnusedI32 = UndefValue::get(Type::getInt32Ty(m_Ctx));
  1724. // Create entry function scope.
  1725. DXASSERT_NOMSG(m_ScopeStack.IsEmpty());
  1726. (void)m_ScopeStack.Push(Scope::Function, nullptr);
  1727. m_ScopeStack.Top().SetEntry(true);
  1728. DeclareIndexableRegisters();
  1729. // Parse DXBC instructions and emit DXIL equivalents.
  1730. Value *pHullLoopInductionVar = nullptr;
  1731. m_bControlPointPhase = false;
  1732. bool bMustCloseHullLoop = false;
  1733. m_bPatchConstantPhase = false;
  1734. bool bInsertResourceHandles = true;
  1735. unsigned ForkJoinPhaseIndex = 0;
  1736. D3D10ShaderBinary::CInstruction Inst;
  1737. bool bPasshThroughCP = false;
  1738. bool bDoneParsing = false;
  1739. for (;;) {
  1740. AdvanceDxbcInstructionStream(Parser, Inst, bDoneParsing);
  1741. // Terminate HS phase (HullLoop), if necessary.
  1742. if (m_bPatchConstantPhase) {
  1743. bool bTerminateHullLoop = false;
  1744. if (bDoneParsing || bMustCloseHullLoop) {
  1745. bTerminateHullLoop = true;
  1746. } else {
  1747. switch (Inst.OpCode()) {
  1748. case D3D11_SB_OPCODE_HS_FORK_PHASE:
  1749. case D3D11_SB_OPCODE_HS_JOIN_PHASE:
  1750. case D3D10_SB_OPCODE_LABEL:
  1751. bTerminateHullLoop = true;
  1752. break;
  1753. }
  1754. }
  1755. if (bTerminateHullLoop) {
  1756. IFTBOOL(m_ScopeStack.Top().Kind == Scope::HullLoop, E_FAIL);
  1757. // Hull shader control point phase fork/join.
  1758. Scope &HullScope = m_ScopeStack.Top();
  1759. // Increment HullLoop instance ID.
  1760. Value *pOldInstID = m_pBuilder->CreateLoad(HullScope.pInductionVar);
  1761. Value *pNewInstID = m_pBuilder->CreateAdd(pOldInstID, m_pOP->GetU32Const(1));
  1762. (void)m_pBuilder->CreateStore(pNewInstID, HullScope.pInductionVar);
  1763. // Insert backedge cbranch to HullLoop and AfterHullLoop BBs.
  1764. Value *pCond = m_pBuilder->CreateICmpULT(pNewInstID, m_pOP->GetU32Const(HullScope.HullLoopTripCount));
  1765. m_pBuilder->CreateCondBr(pCond, HullScope.pHullLoopBB, HullScope.pPostScopeBB);
  1766. m_pPR->GetPatchConstantFunction()->getBasicBlockList().push_back(HullScope.pPostScopeBB);
  1767. m_pBuilder->SetInsertPoint(HullScope.pPostScopeBB);
  1768. m_ScopeStack.Pop();
  1769. // Skip dead instructions to the next phase, label or EOS.
  1770. for( ; !bDoneParsing ; ) {
  1771. if (Inst.OpCode() == D3D11_SB_OPCODE_HS_FORK_PHASE ||
  1772. Inst.OpCode() == D3D11_SB_OPCODE_HS_JOIN_PHASE ||
  1773. Inst.OpCode() == D3D10_SB_OPCODE_LABEL)
  1774. break;
  1775. AdvanceDxbcInstructionStream(Parser, Inst, bDoneParsing);
  1776. }
  1777. }
  1778. bMustCloseHullLoop = false;
  1779. }
  1780. // Terminate function, if necessary.
  1781. {
  1782. bool bTerminateFunc = false;
  1783. if (bDoneParsing) {
  1784. bTerminateFunc = true;
  1785. } else {
  1786. switch (Inst.OpCode()) {
  1787. case D3D11_SB_OPCODE_HS_FORK_PHASE:
  1788. case D3D11_SB_OPCODE_HS_JOIN_PHASE:
  1789. if (!m_bPatchConstantPhase)
  1790. bTerminateFunc = true;
  1791. break;
  1792. case D3D10_SB_OPCODE_LABEL:
  1793. bTerminateFunc = true;
  1794. break;
  1795. }
  1796. }
  1797. if (bTerminateFunc) {
  1798. Scope &Scope = m_ScopeStack.FindParentFunction();
  1799. IFTBOOL(Scope.Kind == Scope::Function, DXC_E_INCORRECT_DXBC);
  1800. m_pBuilder->CreateRetVoid();
  1801. m_ScopeStack.Pop();
  1802. IFT(m_ScopeStack.IsEmpty());
  1803. m_bPatchConstantPhase = false;
  1804. }
  1805. }
  1806. if (bDoneParsing)
  1807. break;
  1808. m_PreciseMask = CMask(Inst.GetPreciseMask());
  1809. // Fix up output register masks.
  1810. // DXBC instruction conversion relies on the output mask(s) determining
  1811. // what components need to be written.
  1812. // Some output operand types have write mask that is 0 -- fix this.
  1813. for (unsigned i = 0; i < std::min(Inst.m_NumOperands, (UINT)2); i++) {
  1814. D3D10ShaderBinary::COperandBase &O = Inst.m_Operands[i];
  1815. switch (O.m_Type) {
  1816. case D3D10_SB_OPERAND_TYPE_OUTPUT_DEPTH:
  1817. case D3D11_SB_OPERAND_TYPE_OUTPUT_DEPTH_GREATER_EQUAL:
  1818. case D3D11_SB_OPERAND_TYPE_OUTPUT_DEPTH_LESS_EQUAL:
  1819. case D3D11_SB_OPERAND_TYPE_OUTPUT_STENCIL_REF:
  1820. case D3D10_SB_OPERAND_TYPE_OUTPUT_COVERAGE_MASK:
  1821. DXASSERT_DXBC(O.m_WriteMask == 0);
  1822. O.SetMask(D3D10_SB_OPERAND_4_COMPONENT_MASK_X);
  1823. break;
  1824. }
  1825. }
  1826. if (bInsertResourceHandles) {
  1827. InsertSM50ResourceHandles();
  1828. bInsertResourceHandles = false;
  1829. }
  1830. switch (Inst.OpCode()) {
  1831. //
  1832. // Declarations.
  1833. //
  1834. case D3D10_SB_OPCODE_DCL_RESOURCE:
  1835. case D3D10_SB_OPCODE_DCL_CONSTANT_BUFFER:
  1836. case D3D10_SB_OPCODE_DCL_SAMPLER:
  1837. case D3D10_SB_OPCODE_DCL_INDEX_RANGE:
  1838. case D3D10_SB_OPCODE_DCL_GS_OUTPUT_PRIMITIVE_TOPOLOGY:
  1839. case D3D10_SB_OPCODE_DCL_GS_INPUT_PRIMITIVE:
  1840. case D3D10_SB_OPCODE_DCL_MAX_OUTPUT_VERTEX_COUNT:
  1841. case D3D10_SB_OPCODE_DCL_INPUT:
  1842. case D3D10_SB_OPCODE_DCL_INPUT_SGV:
  1843. case D3D10_SB_OPCODE_DCL_INPUT_SIV:
  1844. case D3D10_SB_OPCODE_DCL_INPUT_PS:
  1845. case D3D10_SB_OPCODE_DCL_INPUT_PS_SGV:
  1846. case D3D10_SB_OPCODE_DCL_INPUT_PS_SIV:
  1847. case D3D10_SB_OPCODE_DCL_OUTPUT:
  1848. case D3D10_SB_OPCODE_DCL_OUTPUT_SGV:
  1849. case D3D10_SB_OPCODE_DCL_OUTPUT_SIV:
  1850. case D3D10_SB_OPCODE_DCL_TEMPS:
  1851. case D3D10_SB_OPCODE_DCL_INDEXABLE_TEMP:
  1852. break;
  1853. case D3D10_SB_OPCODE_DCL_GLOBAL_FLAGS:
  1854. break;
  1855. case D3D11_SB_OPCODE_DCL_STREAM:
  1856. case D3D11_SB_OPCODE_DCL_FUNCTION_BODY:
  1857. case D3D11_SB_OPCODE_DCL_FUNCTION_TABLE:
  1858. case D3D11_SB_OPCODE_DCL_INTERFACE:
  1859. case D3D11_SB_OPCODE_HS_DECLS:
  1860. case D3D11_SB_OPCODE_DCL_INPUT_CONTROL_POINT_COUNT:
  1861. case D3D11_SB_OPCODE_DCL_OUTPUT_CONTROL_POINT_COUNT:
  1862. case D3D11_SB_OPCODE_DCL_TESS_DOMAIN:
  1863. case D3D11_SB_OPCODE_DCL_TESS_PARTITIONING:
  1864. case D3D11_SB_OPCODE_DCL_TESS_OUTPUT_PRIMITIVE:
  1865. case D3D11_SB_OPCODE_DCL_HS_MAX_TESSFACTOR:
  1866. case D3D11_SB_OPCODE_DCL_THREAD_GROUP:
  1867. case D3D11_SB_OPCODE_DCL_UNORDERED_ACCESS_VIEW_TYPED:
  1868. case D3D11_SB_OPCODE_DCL_UNORDERED_ACCESS_VIEW_RAW:
  1869. case D3D11_SB_OPCODE_DCL_UNORDERED_ACCESS_VIEW_STRUCTURED:
  1870. case D3D11_SB_OPCODE_DCL_THREAD_GROUP_SHARED_MEMORY_RAW:
  1871. case D3D11_SB_OPCODE_DCL_THREAD_GROUP_SHARED_MEMORY_STRUCTURED:
  1872. case D3D11_SB_OPCODE_DCL_RESOURCE_RAW:
  1873. case D3D11_SB_OPCODE_DCL_RESOURCE_STRUCTURED:
  1874. case D3D11_SB_OPCODE_DCL_GS_INSTANCE_COUNT:
  1875. break;
  1876. //
  1877. // Immediate constant buffer.
  1878. //
  1879. case D3D10_SB_OPCODE_CUSTOMDATA:
  1880. if (Inst.m_CustomData.Type == D3D10_SB_CUSTOMDATA_DCL_IMMEDIATE_CONSTANT_BUFFER) {
  1881. unsigned Size = Inst.m_CustomData.DataSizeInBytes >> 2;
  1882. DXASSERT_DXBC(m_pIcbGV == nullptr && Inst.m_CustomData.DataSizeInBytes == Size*4);
  1883. llvm::Constant *pIcbData = ConstantDataArray::get(m_Ctx, ArrayRef<float>((float*)Inst.m_CustomData.pData, Size));
  1884. m_pIcbGV = new GlobalVariable(*m_pModule, pIcbData->getType(), true, GlobalValue::InternalLinkage,
  1885. pIcbData, "dx.icb", nullptr,
  1886. GlobalVariable::NotThreadLocal, DXIL::kImmediateCBufferAddrSpace);
  1887. }
  1888. break;
  1889. //
  1890. // Mov, movc, swapc, dmov, dmovc.
  1891. //
  1892. case D3D10_SB_OPCODE_MOV: {
  1893. CMask WriteMask = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask);
  1894. CompType DstType = InferOperandType(Inst, 0, WriteMask);
  1895. CompType SrcType = InferOperandType(Inst, 1, WriteMask);
  1896. // For mov, movc, and swapc, use integer operation type unless
  1897. // operand modifiers imply floating point.
  1898. CompType OperationType = CompType::getI32();
  1899. if (!DstType.IsInvalid())
  1900. OperationType = DstType.GetBaseCompType();
  1901. else if (!SrcType.IsInvalid())
  1902. OperationType = SrcType;
  1903. if (Inst.m_Operands[1].Modifier() != D3D10_SB_OPERAND_MODIFIER_NONE || Inst.m_bSaturate) {
  1904. OperationType = CompType::getF32();
  1905. }
  1906. OperandValue In;
  1907. LoadOperand(In, Inst, 1, WriteMask, OperationType);
  1908. StoreOperand(In, Inst, 0, WriteMask, OperationType);
  1909. break;
  1910. }
  1911. case D3D10_SB_OPCODE_MOVC: {
  1912. CMask WriteMask = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask);
  1913. CompType DstType = InferOperandType(Inst, 0, WriteMask);
  1914. CompType Src2Type = InferOperandType(Inst, 2, WriteMask);
  1915. CompType Src3Type = InferOperandType(Inst, 3, WriteMask);
  1916. CompType OperationType = CompType::getI32();
  1917. if (Src2Type == Src3Type && !Src2Type.IsInvalid())
  1918. OperationType = Src2Type;
  1919. else if (!DstType.IsInvalid())
  1920. OperationType = DstType.GetBaseCompType();
  1921. else if (!Src2Type.IsInvalid())
  1922. OperationType = Src2Type;
  1923. else if (!Src3Type.IsInvalid())
  1924. OperationType = Src3Type;
  1925. if (Inst.m_Operands[2].Modifier() != D3D10_SB_OPERAND_MODIFIER_NONE ||
  1926. Inst.m_Operands[3].Modifier() != D3D10_SB_OPERAND_MODIFIER_NONE ||
  1927. Inst.m_bSaturate) {
  1928. OperationType = CompType::getF32();
  1929. }
  1930. OperandValue In1, In2, In3, Out;
  1931. LoadOperand(In1, Inst, 1, WriteMask, CompType::getI1());
  1932. LoadOperand(In2, Inst, 2, WriteMask, OperationType);
  1933. LoadOperand(In3, Inst, 3, WriteMask, OperationType);
  1934. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  1935. if (!WriteMask.IsSet(c)) continue;
  1936. Out[c] = m_pBuilder->CreateSelect(In1[c], In2[c], In3[c]);
  1937. }
  1938. StoreOperand(Out, Inst, 0, WriteMask, OperationType);
  1939. break;
  1940. }
  1941. case D3D11_SB_OPCODE_SWAPC: {
  1942. CMask WriteMask = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask | Inst.m_Operands[1].m_WriteMask);
  1943. CMask Dst1Mask = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask);
  1944. CMask Dst2Mask = CMask::FromDXBC(Inst.m_Operands[1].m_WriteMask);
  1945. CompType Dst1Type = InferOperandType(Inst, 0, WriteMask);
  1946. CompType Dst2Type = InferOperandType(Inst, 1, WriteMask);
  1947. CompType Src2Type = InferOperandType(Inst, 3, WriteMask);
  1948. CompType Src3Type = InferOperandType(Inst, 4, WriteMask);
  1949. CompType OperationType = CompType::getI32();
  1950. if (Src2Type == Src3Type && !Src2Type.IsInvalid())
  1951. OperationType = Src2Type;
  1952. else if (!Dst1Type.IsInvalid())
  1953. OperationType = Dst1Type.GetBaseCompType();
  1954. else if (!Dst2Type.IsInvalid())
  1955. OperationType = Dst2Type.GetBaseCompType();
  1956. else if (!Src2Type.IsInvalid())
  1957. OperationType = Src2Type;
  1958. else if (!Src3Type.IsInvalid())
  1959. OperationType = Src3Type;
  1960. if (Inst.m_Operands[3].Modifier() != D3D10_SB_OPERAND_MODIFIER_NONE ||
  1961. Inst.m_Operands[4].Modifier() != D3D10_SB_OPERAND_MODIFIER_NONE ||
  1962. Inst.m_bSaturate) {
  1963. OperationType = CompType::getF32();
  1964. }
  1965. OperandValue In1, In2, In3, Out1, Out2;
  1966. LoadOperand(In1, Inst, 2, WriteMask, CompType::getI1());
  1967. LoadOperand(In2, Inst, 3, WriteMask, OperationType);
  1968. LoadOperand(In3, Inst, 4, WriteMask, OperationType);
  1969. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  1970. if (!Dst1Mask.IsSet(c)) continue;
  1971. Out1[c] = m_pBuilder->CreateSelect(In1[c], In3[c], In2[c]);
  1972. }
  1973. StoreOperand(Out1, Inst, 0, Dst1Mask, OperationType);
  1974. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  1975. if (!Dst2Mask.IsSet(c)) continue;
  1976. Out2[c] = m_pBuilder->CreateSelect(In1[c], In2[c], In3[c]);
  1977. }
  1978. StoreOperand(Out2, Inst, 1, Dst2Mask, OperationType);
  1979. break;
  1980. }
  1981. //
  1982. // Floating point unary.
  1983. //
  1984. case D3D10_SB_OPCODE_EXP: ConvertUnary(OP::OpCode::Exp, CompType::getF32(), Inst); break;
  1985. case D3D10_SB_OPCODE_FRC: ConvertUnary(OP::OpCode::Frc, CompType::getF32(), Inst); break;
  1986. case D3D10_SB_OPCODE_LOG: ConvertUnary(OP::OpCode::Log, CompType::getF32(), Inst); break;
  1987. case D3D10_SB_OPCODE_SQRT: ConvertUnary(OP::OpCode::Sqrt, CompType::getF32(), Inst); break;
  1988. case D3D10_SB_OPCODE_RSQ: ConvertUnary(OP::OpCode::Rsqrt, CompType::getF32(), Inst); break;
  1989. case D3D10_SB_OPCODE_ROUND_NE: ConvertUnary(OP::OpCode::Round_ne, CompType::getF32(), Inst); break;
  1990. case D3D10_SB_OPCODE_ROUND_NI: ConvertUnary(OP::OpCode::Round_ni, CompType::getF32(), Inst); break;
  1991. case D3D10_SB_OPCODE_ROUND_PI: ConvertUnary(OP::OpCode::Round_pi, CompType::getF32(), Inst); break;
  1992. case D3D10_SB_OPCODE_ROUND_Z: ConvertUnary(OP::OpCode::Round_z, CompType::getF32(), Inst); break;
  1993. case D3D11_SB_OPCODE_RCP: {
  1994. CMask WriteMask = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask);
  1995. CompType OperationType = DXBC::GetCompTypeWithMinPrec(CompType::getF32(), Inst.m_Operands[0].m_MinPrecision);
  1996. OperandValue In, Out;
  1997. LoadOperand(In, Inst, 1, WriteMask, OperationType);
  1998. Value *One = m_pOP->GetFloatConst(1.0f);
  1999. if (OperationType.Is16Bit())
  2000. One = ConstantFP::get(m_pBuilder->getHalfTy(), 1.0);
  2001. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  2002. if (!WriteMask.IsSet(c)) continue;
  2003. Out[c] = m_pBuilder->CreateBinOp(Instruction::BinaryOps::FDiv, One, In[c]);
  2004. }
  2005. StoreOperand(Out, Inst, 0, WriteMask, OperationType);
  2006. break;
  2007. }
  2008. case D3D10_SB_OPCODE_SINCOS:
  2009. {
  2010. CMask WriteMaskSin;
  2011. CMask WriteMaskCos;
  2012. CompType OperationType;
  2013. if (Inst.m_Operands[0].m_Type != D3D10_SB_OPERAND_TYPE_NULL) {
  2014. WriteMaskSin = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask);
  2015. OperationType = DXBC::GetCompTypeWithMinPrec(CompType::getF32(), Inst.m_Operands[0].m_MinPrecision);
  2016. }
  2017. if (Inst.m_Operands[1].m_Type != D3D10_SB_OPERAND_TYPE_NULL) {
  2018. WriteMaskCos = CMask::FromDXBC(Inst.m_Operands[1].m_WriteMask);
  2019. CompType OperationTypeCos = DXBC::GetCompTypeWithMinPrec(CompType::getF32(), Inst.m_Operands[1].m_MinPrecision);
  2020. DXASSERT_DXBC(OperationType.GetKind() == CompType::Kind::Invalid || OperationType == OperationTypeCos);
  2021. OperationType = OperationTypeCos;
  2022. }
  2023. CMask WriteMaskAll = WriteMaskSin | WriteMaskCos;
  2024. Type *pOperationType = OperationType.GetLLVMType(m_Ctx);
  2025. OperandValue In;
  2026. LoadOperand(In, Inst, 2, WriteMaskAll, OperationType);
  2027. if (Inst.m_Operands[0].m_Type != D3D10_SB_OPERAND_TYPE_NULL) {
  2028. OperandValue Out;
  2029. Function *pFunc = m_pOP->GetOpFunc(OP::OpCode::Sin, pOperationType);
  2030. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  2031. if (!WriteMaskSin.IsSet(c)) continue;
  2032. Out[c] = m_pBuilder->CreateCall(pFunc, { m_pOP->GetU32Const((unsigned)OP::OpCode::Sin), In[c] });
  2033. }
  2034. StoreOperand(Out, Inst, 0, WriteMaskSin, OperationType);
  2035. }
  2036. if (Inst.m_Operands[1].m_Type != D3D10_SB_OPERAND_TYPE_NULL) {
  2037. OperandValue Out;
  2038. Function *pFunc = m_pOP->GetOpFunc(OP::OpCode::Cos, pOperationType);
  2039. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  2040. if (!WriteMaskCos.IsSet(c)) continue;
  2041. Out[c] = m_pBuilder->CreateCall(pFunc, { m_pOP->GetU32Const((unsigned)OP::OpCode::Cos), In[c] });
  2042. }
  2043. StoreOperand(Out, Inst, 1, WriteMaskCos, OperationType);
  2044. }
  2045. break;
  2046. }
  2047. //
  2048. // Integer unary.
  2049. //
  2050. case D3D11_SB_OPCODE_BFREV: ConvertUnary(OP::OpCode::Bfrev, CompType::getU32(), Inst); break;
  2051. case D3D11_SB_OPCODE_COUNTBITS: ConvertUnary(OP::OpCode::Countbits, CompType::getU32(), Inst); break;
  2052. case D3D11_SB_OPCODE_FIRSTBIT_HI: ConvertUnary(OP::OpCode::FirstbitHi, CompType::getU32(), Inst); break;
  2053. case D3D11_SB_OPCODE_FIRSTBIT_LO: ConvertUnary(OP::OpCode::FirstbitLo, CompType::getU32(), Inst); break;
  2054. case D3D11_SB_OPCODE_FIRSTBIT_SHI: ConvertUnary(OP::OpCode::FirstbitSHi, CompType::getI32(), Inst); break;
  2055. case D3D10_SB_OPCODE_INEG: {
  2056. CMask WriteMask = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask);
  2057. CompType OperationType = DXBC::GetCompTypeWithMinPrec(CompType::getI32(), Inst.m_Operands[0].m_MinPrecision);
  2058. OperandValue In, Out;
  2059. LoadOperand(In, Inst, 1, WriteMask, OperationType);
  2060. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  2061. if (!WriteMask.IsSet(c)) continue;
  2062. Out[c] = m_pBuilder->CreateNeg(In[c]);
  2063. }
  2064. StoreOperand(Out, Inst, 0, WriteMask, OperationType);
  2065. break;
  2066. }
  2067. case D3D10_SB_OPCODE_NOT: {
  2068. CMask WriteMask = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask);
  2069. CompType OperationType = DXBC::GetCompTypeWithMinPrec(CompType::getI32(), Inst.m_Operands[0].m_MinPrecision);
  2070. OperandValue In, Out;
  2071. LoadOperand(In, Inst, 1, WriteMask, OperationType);
  2072. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  2073. if (!WriteMask.IsSet(c)) continue;
  2074. Out[c] = m_pBuilder->CreateNot(In[c]);
  2075. }
  2076. StoreOperand(Out, Inst, 0, WriteMask, OperationType);
  2077. break;
  2078. }
  2079. //
  2080. // Floating point binary.
  2081. //
  2082. case D3D10_SB_OPCODE_ADD: ConvertBinary(Instruction::FAdd, CompType::getF32(), Inst); break;
  2083. case D3D10_SB_OPCODE_MUL: ConvertBinary(Instruction::FMul, CompType::getF32(), Inst); break;
  2084. case D3D10_SB_OPCODE_DIV: ConvertBinary(Instruction::FDiv, CompType::getF32(), Inst); break;
  2085. case D3D10_SB_OPCODE_MAX: ConvertBinary(OP::OpCode::FMax, CompType::getF32(), Inst); break;
  2086. case D3D10_SB_OPCODE_MIN: ConvertBinary(OP::OpCode::FMin, CompType::getF32(), Inst); break;
  2087. //
  2088. // Integer binary.
  2089. //
  2090. case D3D10_SB_OPCODE_IADD: ConvertBinary(Instruction::Add, CompType::getI32(), Inst); break;
  2091. case D3D10_SB_OPCODE_IMAX: ConvertBinary(OP::OpCode::IMax, CompType::getI32(), Inst); break;
  2092. case D3D10_SB_OPCODE_IMIN: ConvertBinary(OP::OpCode::IMin, CompType::getI32(), Inst); break;
  2093. case D3D10_SB_OPCODE_UMAX: ConvertBinary(OP::OpCode::UMax, CompType::getU32(), Inst); break;
  2094. case D3D10_SB_OPCODE_UMIN: ConvertBinary(OP::OpCode::UMin, CompType::getU32(), Inst); break;
  2095. case D3D10_SB_OPCODE_AND: ConvertBinary(Instruction::And, CompType::getI32(), Inst); break;
  2096. case D3D10_SB_OPCODE_OR: ConvertBinary(Instruction::Or, CompType::getI32(), Inst); break;
  2097. case D3D10_SB_OPCODE_XOR: ConvertBinary(Instruction::Xor, CompType::getI32(), Inst); break;
  2098. case D3D10_SB_OPCODE_ISHL: ConvertBinary(Instruction::Shl, CompType::getI32(), Inst); break;
  2099. case D3D10_SB_OPCODE_ISHR: ConvertBinary(Instruction::AShr, CompType::getI32(), Inst); break;
  2100. case D3D10_SB_OPCODE_USHR: ConvertBinary(Instruction::LShr, CompType::getI32(), Inst); break;
  2101. //
  2102. // Integer binary with two outputs.
  2103. //
  2104. case D3D10_SB_OPCODE_IMUL: ConvertBinaryWithTwoOuts(OP::OpCode::IMul, Inst); break;
  2105. case D3D10_SB_OPCODE_UMUL: ConvertBinaryWithTwoOuts(OP::OpCode::UMul, Inst); break;
  2106. case D3D10_SB_OPCODE_UDIV: ConvertBinaryWithTwoOuts(OP::OpCode::UDiv, Inst); break;
  2107. //
  2108. // Integer binary with carry.
  2109. //
  2110. case D3D11_SB_OPCODE_UADDC: ConvertBinaryWithCarry(OP::OpCode::UAddc, Inst); break;
  2111. case D3D11_SB_OPCODE_USUBB: ConvertBinaryWithCarry(OP::OpCode::USubb, Inst); break;
  2112. //
  2113. // Floating point tertiary.
  2114. //
  2115. case D3D10_SB_OPCODE_MAD: ConvertTertiary(OP::OpCode::FMad, CompType::getF32(), Inst); break;
  2116. //
  2117. // Integer tertiary.
  2118. //
  2119. case D3D10_SB_OPCODE_IMAD: ConvertTertiary(OP::OpCode::IMad, CompType::getI32(), Inst); break;
  2120. case D3D10_SB_OPCODE_UMAD: ConvertTertiary(OP::OpCode::UMad, CompType::getI32(), Inst); break;
  2121. case D3D11_1_SB_OPCODE_MSAD: ConvertTertiary(OP::OpCode::Msad, CompType::getI32(), Inst); break;
  2122. case D3D11_SB_OPCODE_IBFE: ConvertTertiary(OP::OpCode::Ibfe, CompType::getI32(), Inst); break;
  2123. case D3D11_SB_OPCODE_UBFE: ConvertTertiary(OP::OpCode::Ubfe, CompType::getI32(), Inst); break;
  2124. //
  2125. // Quaternary int.
  2126. //
  2127. case D3D11_SB_OPCODE_BFI: ConvertQuaternary(OP::OpCode::Bfi, CompType::getI32(), Inst); break;
  2128. //
  2129. // Logical comparison.
  2130. //
  2131. case D3D10_SB_OPCODE_EQ: ConvertComparison(CmpInst::FCMP_OEQ, CompType::getF32(), Inst); break;
  2132. case D3D10_SB_OPCODE_NE: ConvertComparison(CmpInst::FCMP_UNE, CompType::getF32(), Inst); break;
  2133. case D3D10_SB_OPCODE_LT: ConvertComparison(CmpInst::FCMP_OLT, CompType::getF32(), Inst); break;
  2134. case D3D10_SB_OPCODE_GE: ConvertComparison(CmpInst::FCMP_OGE, CompType::getF32(), Inst); break;
  2135. case D3D10_SB_OPCODE_IEQ: ConvertComparison(CmpInst::ICMP_EQ, CompType::getI32(), Inst); break;
  2136. case D3D10_SB_OPCODE_INE: ConvertComparison(CmpInst::ICMP_NE, CompType::getI32(), Inst); break;
  2137. case D3D10_SB_OPCODE_ILT: ConvertComparison(CmpInst::ICMP_SLT, CompType::getI32(), Inst); break;
  2138. case D3D10_SB_OPCODE_IGE: ConvertComparison(CmpInst::ICMP_SGE, CompType::getI32(), Inst); break;
  2139. case D3D10_SB_OPCODE_ULT: ConvertComparison(CmpInst::ICMP_ULT, CompType::getI32(), Inst); break;
  2140. case D3D10_SB_OPCODE_UGE: ConvertComparison(CmpInst::ICMP_UGE, CompType::getI32(), Inst); break;
  2141. //
  2142. // Dot product.
  2143. //
  2144. case D3D10_SB_OPCODE_DP2: ConvertDotProduct(OP::OpCode::Dot2, 2, CMask::MakeMask(1,1,0,0), Inst); break;
  2145. case D3D10_SB_OPCODE_DP3: ConvertDotProduct(OP::OpCode::Dot3, 3, CMask::MakeMask(1,1,1,0), Inst); break;
  2146. case D3D10_SB_OPCODE_DP4: ConvertDotProduct(OP::OpCode::Dot4, 4, CMask::MakeMask(1,1,1,1), Inst); break;
  2147. //
  2148. // Type conversions.
  2149. //
  2150. case D3D10_SB_OPCODE_ITOF: ConvertCast(CompType::getI32(), CompType::getF32(), Inst); break;
  2151. case D3D10_SB_OPCODE_UTOF: ConvertCast(CompType::getU32(), CompType::getF32(), Inst); break;
  2152. case D3D10_SB_OPCODE_FTOI: ConvertCast(CompType::getF32(), CompType::getI32(), Inst); break;
  2153. case D3D10_SB_OPCODE_FTOU: ConvertCast(CompType::getF32(), CompType::getU32(), Inst); break;
  2154. case D3D11_SB_OPCODE_F32TOF16: {
  2155. const unsigned DstIdx = 0;
  2156. const unsigned SrcIdx = 1;
  2157. CMask WriteMask = CMask::FromDXBC(Inst.m_Operands[DstIdx].m_WriteMask);
  2158. if (!WriteMask.IsZero()) {
  2159. OperandValue In, Out;
  2160. LoadOperand(In, Inst, SrcIdx, WriteMask, CompType::getF32());
  2161. OP::OpCode OpCode = OP::OpCode::LegacyF32ToF16;
  2162. CompType DstType = CompType::getU32();
  2163. Function *F = m_pOP->GetOpFunc(OpCode, Type::getVoidTy(m_Ctx));
  2164. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  2165. if (!WriteMask.IsSet(c)) continue;
  2166. Value *Args[2];
  2167. Args[0] = m_pOP->GetU32Const((unsigned)OpCode);
  2168. Args[1] = In[c];
  2169. Out[c] = m_pBuilder->CreateCall(F, Args);
  2170. }
  2171. StoreOperand(Out, Inst, DstIdx, WriteMask, DstType);
  2172. }
  2173. break;
  2174. }
  2175. case D3D11_SB_OPCODE_F16TOF32: {
  2176. const unsigned DstIdx = 0;
  2177. const unsigned SrcIdx = 1;
  2178. CMask WriteMask = CMask::FromDXBC(Inst.m_Operands[DstIdx].m_WriteMask);
  2179. if (!WriteMask.IsZero()) {
  2180. OperandValue In, Out;
  2181. D3D10_SB_OPERAND_MODIFIER SrcModifier = Inst.m_Operands[SrcIdx].m_Modifier;
  2182. Inst.m_Operands[SrcIdx].m_Modifier = D3D10_SB_OPERAND_MODIFIER_NONE;
  2183. LoadOperand(In, Inst, SrcIdx, WriteMask, CompType::getU32());
  2184. OP::OpCode OpCode = OP::OpCode::LegacyF16ToF32;
  2185. CompType DstType = CompType::getF32();
  2186. Function *F = m_pOP->GetOpFunc(OpCode, Type::getVoidTy(m_Ctx));
  2187. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  2188. if (!WriteMask.IsSet(c)) continue;
  2189. Value *Args[2];
  2190. Args[0] = m_pOP->GetU32Const((unsigned)OpCode);
  2191. Args[1] = In[c];
  2192. Value *pResult = m_pBuilder->CreateCall(F, Args);
  2193. // Special-case: propagate source operand modifiers to result.
  2194. if (SrcModifier & D3D10_SB_OPERAND_MODIFIER_ABS) {
  2195. Function *Fabs = m_pOP->GetOpFunc(OP::OpCode::FAbs, pResult->getType());
  2196. Value *Args[2];
  2197. Args[0] = m_pOP->GetU32Const((unsigned)OP::OpCode::FAbs);
  2198. Args[1] = pResult;
  2199. pResult = m_pBuilder->CreateCall(Fabs, Args);
  2200. }
  2201. if (SrcModifier & D3D10_SB_OPERAND_MODIFIER_NEG) {
  2202. pResult = MarkPrecise(m_pBuilder->CreateFNeg(MarkPrecise(pResult, c)), c);
  2203. }
  2204. Out[c] = pResult;
  2205. }
  2206. StoreOperand(Out, Inst, DstIdx, WriteMask, CompType::getF32());
  2207. }
  2208. break;
  2209. }
  2210. //
  2211. // Double-precision operations.
  2212. //
  2213. case D3D11_SB_OPCODE_DADD: ConvertBinary(Instruction::FAdd, CompType::getF64(), Inst); break;
  2214. case D3D11_SB_OPCODE_DMAX: ConvertBinary(OP::OpCode::FMax, CompType::getF64(), Inst); break;
  2215. case D3D11_SB_OPCODE_DMIN: ConvertBinary(OP::OpCode::FMin, CompType::getF64(), Inst); break;
  2216. case D3D11_SB_OPCODE_DMUL: ConvertBinary(Instruction::FMul, CompType::getF64(), Inst); break;
  2217. case D3D11_1_SB_OPCODE_DDIV: ConvertBinary(Instruction::FDiv, CompType::getF64(), Inst); break;
  2218. case D3D11_1_SB_OPCODE_DFMA: ConvertTertiary(OP::OpCode::Fma, CompType::getF64(), Inst); break;
  2219. case D3D11_SB_OPCODE_DEQ: ConvertComparison(CmpInst::FCMP_OEQ, CompType::getF64(), Inst); break;
  2220. case D3D11_SB_OPCODE_DGE: ConvertComparison(CmpInst::FCMP_OGE, CompType::getF64(), Inst); break;
  2221. case D3D11_SB_OPCODE_DLT: ConvertComparison(CmpInst::FCMP_OLT, CompType::getF64(), Inst); break;
  2222. case D3D11_SB_OPCODE_DNE: ConvertComparison(CmpInst::FCMP_UNE, CompType::getF64(), Inst); break;
  2223. case D3D11_SB_OPCODE_DMOV: {
  2224. CMask WriteMask = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask);
  2225. OperandValue In;
  2226. LoadOperand(In, Inst, 1, WriteMask, CompType::getF64());
  2227. StoreOperand(In, Inst, 0, WriteMask, CompType::getF64());
  2228. break;
  2229. }
  2230. case D3D11_SB_OPCODE_DMOVC: {
  2231. CMask WriteMask = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask);
  2232. CompType OperationType = CompType::getF64();
  2233. OperandValue In1, In2, In3, Out;
  2234. LoadOperand(In1, Inst, 1, CMask(1, 1, 0, 0), CompType::getI1());
  2235. LoadOperand(In2, Inst, 2, WriteMask, OperationType);
  2236. LoadOperand(In3, Inst, 3, WriteMask, OperationType);
  2237. for (BYTE c = 0; c < DXBC::kWidth; c += 2) {
  2238. if (!WriteMask.IsSet(c)) continue;
  2239. Out[c] = m_pBuilder->CreateSelect(In1[c>>1], In2[c], In3[c]);
  2240. }
  2241. StoreOperand(Out, Inst, 0, WriteMask, OperationType);
  2242. break;
  2243. }
  2244. case D3D11_1_SB_OPCODE_DRCP: {
  2245. CMask WriteMask = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask);
  2246. CompType OperationType = CompType::getF64();
  2247. OperandValue In, Out;
  2248. LoadOperand(In, Inst, 1, WriteMask, OperationType);
  2249. for (BYTE c = 0; c < DXBC::kWidth; c += 2) {
  2250. if (!WriteMask.IsSet(c)) continue;
  2251. Out[c] = m_pBuilder->CreateBinOp(Instruction::BinaryOps::FDiv, m_pOP->GetDoubleConst(1.0), In[c]);
  2252. }
  2253. StoreOperand(Out, Inst, 0, WriteMask, OperationType);
  2254. break;
  2255. }
  2256. case D3D11_SB_OPCODE_DTOF: ConvertFromDouble(CompType::getF32(), Inst); break;
  2257. case D3D11_1_SB_OPCODE_DTOI: ConvertFromDouble(CompType::getI32(), Inst); break;
  2258. case D3D11_1_SB_OPCODE_DTOU: ConvertFromDouble(CompType::getU32(), Inst); break;
  2259. case D3D11_SB_OPCODE_FTOD: ConvertToDouble (CompType::getF32(), Inst); break;
  2260. case D3D11_1_SB_OPCODE_ITOD: ConvertToDouble (CompType::getI32(), Inst); break;
  2261. case D3D11_1_SB_OPCODE_UTOD: ConvertToDouble (CompType::getU32(), Inst); break;
  2262. //
  2263. // Resource operations.
  2264. //
  2265. case D3D10_SB_OPCODE_SAMPLE:
  2266. case D3DWDDM1_3_SB_OPCODE_SAMPLE_CLAMP_FEEDBACK: {
  2267. OP::OpCode OpCode = OP::OpCode::Sample;
  2268. bool bHasFeedback = DXBC::HasFeedback(Inst.OpCode());
  2269. const unsigned uOpOutput = 0;
  2270. const unsigned uOpClamp = 4 + (bHasFeedback ? 1 : 0);
  2271. Value *Args[11];
  2272. LoadCommonSampleInputs(Inst, &Args[0]);
  2273. // Other arguments.
  2274. Args[0] = m_pOP->GetU32Const((unsigned)OpCode);
  2275. // Clamp.
  2276. Args[10] = m_pOP->GetFloatConst(0.f);
  2277. if (bHasFeedback) {
  2278. if (Inst.m_Operands[uOpClamp].m_Type != D3D10_SB_OPERAND_TYPE_IMMEDIATE32 ||
  2279. Inst.m_Operands[uOpClamp].m_Valuef[0] != 0.f) {
  2280. OperandValue InClamp;
  2281. LoadOperand(InClamp, Inst, uOpClamp, CMask::MakeXMask(), CompType::getF32());
  2282. Args[10] = InClamp[0];
  2283. }
  2284. }
  2285. // Function call.
  2286. CompType DstType = DXBC::GetCompTypeWithMinPrec(CompType::getF32(), Inst.m_Operands[uOpOutput].m_MinPrecision);
  2287. Type *pDstType = DstType.GetLLVMType(m_Ctx);
  2288. Function *F = m_pOP->GetOpFunc(OpCode, pDstType);
  2289. Value *pOpRet = m_pBuilder->CreateCall(F, Args);
  2290. StoreResRetOutputAndStatus(Inst, pOpRet, DstType);
  2291. break;
  2292. }
  2293. case D3D10_SB_OPCODE_SAMPLE_B:
  2294. case D3DWDDM1_3_SB_OPCODE_SAMPLE_B_CLAMP_FEEDBACK: {
  2295. OP::OpCode OpCode = OP::OpCode::SampleBias;
  2296. bool bHasFeedback = DXBC::HasFeedback(Inst.OpCode());
  2297. const unsigned uOpOutput = 0;
  2298. const unsigned uOpBias = 4 + (bHasFeedback ? 1 : 0);
  2299. const unsigned uOpClamp = uOpBias + 1;
  2300. Value *Args[12];
  2301. LoadCommonSampleInputs(Inst, &Args[0]);
  2302. // Other arguments.
  2303. Args[0] = m_pOP->GetU32Const((unsigned)OpCode);
  2304. OperandValue InBias;
  2305. LoadOperand(InBias, Inst, uOpBias, CMask::MakeXMask(), CompType::getF32());
  2306. Args[10] = InBias[0];
  2307. // Clamp.
  2308. Args[11] = m_pOP->GetFloatConst(0.f);
  2309. if (bHasFeedback) {
  2310. if (Inst.m_Operands[uOpClamp].m_Type != D3D10_SB_OPERAND_TYPE_IMMEDIATE32 ||
  2311. Inst.m_Operands[uOpClamp].m_Valuef[0] != 0.f) {
  2312. OperandValue InClamp;
  2313. LoadOperand(InClamp, Inst, uOpClamp, CMask::MakeXMask(), CompType::getF32());
  2314. Args[11] = InClamp[0];
  2315. }
  2316. }
  2317. // Function call.
  2318. CompType DstType = DXBC::GetCompTypeWithMinPrec(CompType::getF32(), Inst.m_Operands[uOpOutput].m_MinPrecision);
  2319. Type *pDstType = DstType.GetLLVMType(m_Ctx);
  2320. Function *F = m_pOP->GetOpFunc(OpCode, pDstType);
  2321. Value *pOpRet = m_pBuilder->CreateCall(F, Args);
  2322. StoreResRetOutputAndStatus(Inst, pOpRet, DstType);
  2323. break;
  2324. }
  2325. case D3D10_SB_OPCODE_SAMPLE_L:
  2326. case D3DWDDM1_3_SB_OPCODE_SAMPLE_L_FEEDBACK: {
  2327. OP::OpCode OpCode = OP::OpCode::SampleLevel;
  2328. bool bHasFeedback = DXBC::HasFeedback(Inst.OpCode());
  2329. const unsigned uOpOutput = 0;
  2330. const unsigned uOpLevel = 4 + (bHasFeedback ? 1 : 0);
  2331. Value *Args[11];
  2332. LoadCommonSampleInputs(Inst, &Args[0]);
  2333. // Other arguments.
  2334. Args[0] = m_pOP->GetU32Const((unsigned)OpCode);
  2335. OperandValue InLevel;
  2336. LoadOperand(InLevel, Inst, uOpLevel, CMask::MakeXMask(), CompType::getF32());
  2337. Args[10] = InLevel[0];
  2338. // Function call.
  2339. CompType DstType = DXBC::GetCompTypeWithMinPrec(CompType::getF32(), Inst.m_Operands[uOpOutput].m_MinPrecision);
  2340. Type *pDstType = DstType.GetLLVMType(m_Ctx);
  2341. Function *F = m_pOP->GetOpFunc(OpCode, pDstType);
  2342. Value *pOpRet = m_pBuilder->CreateCall(F, Args);
  2343. StoreResRetOutputAndStatus(Inst, pOpRet, DstType);
  2344. break;
  2345. }
  2346. case D3D10_SB_OPCODE_SAMPLE_D:
  2347. case D3DWDDM1_3_SB_OPCODE_SAMPLE_D_CLAMP_FEEDBACK: {
  2348. OP::OpCode OpCode = OP::OpCode::SampleGrad;
  2349. bool bHasFeedback = DXBC::HasFeedback(Inst.OpCode());
  2350. const unsigned uOpOutput = 0;
  2351. const unsigned uOpSRV = DXBC::GetResourceSlot(Inst.OpCode());
  2352. const unsigned uOpDx = 4 + (bHasFeedback ? 1 : 0);
  2353. const unsigned uOpDy = uOpDx + 1;
  2354. const unsigned uOpClamp = uOpDy + 1;
  2355. const DxilResource &R = GetSRVFromOperand(Inst, uOpSRV);
  2356. Value *Args[17];
  2357. LoadCommonSampleInputs(Inst, &Args[0]);
  2358. // Other arguments.
  2359. Args[0] = m_pOP->GetU32Const((unsigned)OpCode);
  2360. OperandValue InDx, InDy;
  2361. CMask DxDyMask = CMask::MakeFirstNCompMask(DXBC::GetNumResCoords(R.GetKind()));
  2362. LoadOperand(InDx, Inst, uOpDx, DxDyMask, CompType::getF32());
  2363. Args[10] = DxDyMask.IsSet(0) ? InDx[0] : m_pUnusedF32;
  2364. Args[11] = DxDyMask.IsSet(1) ? InDx[1] : m_pUnusedF32;
  2365. Args[12] = DxDyMask.IsSet(2) ? InDx[2] : m_pUnusedF32;
  2366. LoadOperand(InDy, Inst, uOpDy, DxDyMask, CompType::getF32());
  2367. Args[13] = DxDyMask.IsSet(0) ? InDy[0] : m_pUnusedF32;
  2368. Args[14] = DxDyMask.IsSet(1) ? InDy[1] : m_pUnusedF32;
  2369. Args[15] = DxDyMask.IsSet(2) ? InDy[2] : m_pUnusedF32;
  2370. // Clamp.
  2371. Args[16] = m_pOP->GetFloatConst(0.f);
  2372. if (bHasFeedback) {
  2373. if (Inst.m_Operands[uOpClamp].m_Type != D3D10_SB_OPERAND_TYPE_IMMEDIATE32 ||
  2374. Inst.m_Operands[uOpClamp].m_Valuef[0] != 0.f) {
  2375. OperandValue InClamp;
  2376. LoadOperand(InClamp, Inst, uOpClamp, CMask::MakeXMask(), CompType::getF32());
  2377. Args[16] = InClamp[0];
  2378. }
  2379. }
  2380. // Function call.
  2381. CompType DstType = DXBC::GetCompTypeWithMinPrec(CompType::getF32(), Inst.m_Operands[uOpOutput].m_MinPrecision);
  2382. Type *pDstType = DstType.GetLLVMType(m_Ctx);
  2383. Function *F = m_pOP->GetOpFunc(OpCode, pDstType);
  2384. Value *pOpRet = m_pBuilder->CreateCall(F, Args);
  2385. StoreResRetOutputAndStatus(Inst, pOpRet, DstType);
  2386. break;
  2387. }
  2388. case D3D10_SB_OPCODE_SAMPLE_C:
  2389. case D3DWDDM1_3_SB_OPCODE_SAMPLE_C_CLAMP_FEEDBACK: {
  2390. OP::OpCode OpCode = OP::OpCode::SampleCmp;
  2391. bool bHasFeedback = DXBC::HasFeedback(Inst.OpCode());
  2392. const unsigned uOpOutput = 0;
  2393. const unsigned uOpCmp = 4 + (bHasFeedback ? 1 : 0);
  2394. const unsigned uOpClamp = uOpCmp + 1;
  2395. Value *Args[12];
  2396. LoadCommonSampleInputs(Inst, &Args[0]);
  2397. // Other arguments.
  2398. Args[0] = m_pOP->GetU32Const((unsigned)OpCode);
  2399. OperandValue InCmp;
  2400. LoadOperand(InCmp, Inst, uOpCmp, CMask::MakeXMask(), CompType::getF32());
  2401. Args[10] = InCmp[0];
  2402. // Clamp.
  2403. Args[11] = m_pOP->GetFloatConst(0.f);
  2404. if (bHasFeedback) {
  2405. if (Inst.m_Operands[uOpClamp].m_Type != D3D10_SB_OPERAND_TYPE_IMMEDIATE32 ||
  2406. Inst.m_Operands[uOpClamp].m_Valuef[0] != 0.f) {
  2407. OperandValue InClamp;
  2408. LoadOperand(InClamp, Inst, uOpClamp, CMask::MakeXMask(), CompType::getF32());
  2409. Args[11] = InClamp[0];
  2410. }
  2411. }
  2412. // Function call.
  2413. CompType DstType = DXBC::GetCompTypeWithMinPrec(CompType::getF32(), Inst.m_Operands[uOpOutput].m_MinPrecision);
  2414. Type *pDstType = DstType.GetLLVMType(m_Ctx);
  2415. Function *F = m_pOP->GetOpFunc(OpCode, pDstType);
  2416. Value *pOpRet = m_pBuilder->CreateCall(F, Args);
  2417. StoreResRetOutputAndStatus(Inst, pOpRet, DstType);
  2418. break;
  2419. }
  2420. case D3D10_SB_OPCODE_SAMPLE_C_LZ:
  2421. case D3DWDDM1_3_SB_OPCODE_SAMPLE_C_LZ_FEEDBACK: {
  2422. OP::OpCode OpCode = OP::OpCode::SampleCmpLevelZero;
  2423. bool bHasFeedback = DXBC::HasFeedback(Inst.OpCode());
  2424. const unsigned uOpOutput = 0;
  2425. const unsigned uOpCmp = 4 + (bHasFeedback ? 1 : 0);
  2426. Value *Args[11];
  2427. LoadCommonSampleInputs(Inst, &Args[0]);
  2428. // Other arguments.
  2429. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  2430. OperandValue InCmp;
  2431. LoadOperand(InCmp, Inst, uOpCmp, CMask::MakeXMask(), CompType::getF32());
  2432. Args[10] = InCmp[0];
  2433. // Function call.
  2434. CompType DstType = DXBC::GetCompTypeWithMinPrec(CompType::getF32(), Inst.m_Operands[uOpOutput].m_MinPrecision);
  2435. Type *pDstType = DstType.GetLLVMType(m_Ctx);
  2436. Function *F = m_pOP->GetOpFunc(OpCode, pDstType);
  2437. Value *pOpRet = m_pBuilder->CreateCall(F, Args);
  2438. StoreResRetOutputAndStatus(Inst, pOpRet, DstType);
  2439. break;
  2440. }
  2441. case D3D10_SB_OPCODE_LD:
  2442. case D3D10_SB_OPCODE_LD_MS:
  2443. case D3DWDDM1_3_SB_OPCODE_LD_FEEDBACK:
  2444. case D3DWDDM1_3_SB_OPCODE_LD_MS_FEEDBACK: {
  2445. bool bIsTexture2DMS = Inst.OpCode() == D3D10_SB_OPCODE_LD_MS ||
  2446. Inst.OpCode() == D3DWDDM1_3_SB_OPCODE_LD_MS_FEEDBACK;
  2447. bool bHasFeedback = DXBC::HasFeedback(Inst.OpCode());
  2448. const unsigned uOpOutput = 0;
  2449. const unsigned uOpStatus = 1;
  2450. const unsigned uOpCoord = uOpStatus + (bHasFeedback ? 1 : 0);
  2451. const unsigned uOpRes = uOpCoord + 1;
  2452. const unsigned uOpSampleCount = uOpRes + 1;
  2453. DXASSERT_DXBC(Inst.m_Operands[uOpRes].m_Type == D3D10_SB_OPERAND_TYPE_RESOURCE);
  2454. // Resource.
  2455. OperandValue InSRV;
  2456. const DxilResource &R = LoadSRVOperand(InSRV, Inst, uOpRes, CMask::MakeXMask(), CompType::getInvalid());
  2457. // Return type.
  2458. CompType DstType = DXBC::GetCompTypeWithMinPrec(R.GetCompType().GetBaseCompType(), Inst.m_Operands[uOpOutput].m_MinPrecision);
  2459. Type *pDstType = DstType.GetLLVMType(m_Ctx);
  2460. // Create Load call.
  2461. Value *pOpRet;
  2462. if (R.GetKind() != DxilResource::Kind::TypedBuffer) {
  2463. OP::OpCode OpCode = OP::OpCode::TextureLoad;
  2464. // Coordinates.
  2465. OperandValue InCoord;
  2466. CMask CoordMask = CMask::MakeFirstNCompMask(DXBC::GetNumResCoords(R.GetKind()));
  2467. // MIP level.
  2468. if (!bIsTexture2DMS) {
  2469. CoordMask.Set(3);
  2470. }
  2471. LoadOperand(InCoord, Inst, uOpCoord, CoordMask, CompType::getI32());
  2472. Value *Args[9];
  2473. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  2474. Args[1] = InSRV[0]; // Texture SRV handle
  2475. if (!bIsTexture2DMS) {
  2476. Args[2] = InCoord[3]; // MIP level
  2477. } else {
  2478. BYTE Comp = Inst.m_Operands[uOpSampleCount].m_ComponentName;
  2479. OperandValue InSampleCount;
  2480. LoadOperand(InSampleCount, Inst, uOpSampleCount, CMask::MakeCompMask(Comp), CompType::getI32());
  2481. Args[2] = InSampleCount[Comp]; // Sample count
  2482. }
  2483. // Coordinates.
  2484. Args[3] = CoordMask.IsSet(0) ? InCoord[0] : m_pUnusedI32; // Coordinate 0
  2485. Args[4] = CoordMask.IsSet(1) ? InCoord[1] : m_pUnusedI32; // Coordinate 1
  2486. Args[5] = CoordMask.IsSet(2) ? InCoord[2] : m_pUnusedI32; // Coordinate 2
  2487. // Offsets.
  2488. CMask OffsetMask = CMask::MakeFirstNCompMask(DXBC::GetNumResOffsets(R.GetKind()));
  2489. Args[6] = OffsetMask.IsSet(0) ? m_pOP->GetU32Const(Inst.m_TexelOffset[0]) : m_pUnusedI32; // Offset 0
  2490. Args[7] = OffsetMask.IsSet(1) ? m_pOP->GetU32Const(Inst.m_TexelOffset[1]) : m_pUnusedI32; // Offset 1
  2491. Args[8] = OffsetMask.IsSet(2) ? m_pOP->GetU32Const(Inst.m_TexelOffset[2]) : m_pUnusedI32; // Offset 2
  2492. Function *F = m_pOP->GetOpFunc(OpCode, pDstType);
  2493. pOpRet = m_pBuilder->CreateCall(F, Args);
  2494. } else {
  2495. // R.GetKind() == DxilResource::TypedBuffer
  2496. OP::OpCode OpCode = OP::OpCode::BufferLoad;
  2497. Value *Args[4];
  2498. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  2499. Args[1] = InSRV[0]; // Buffer SRV handle
  2500. Args[2] = GetCoordValue(Inst, uOpCoord); // Coord 0: in elements
  2501. Args[3] = m_pUnusedI32; // Coord 1: unused
  2502. Function *F = m_pOP->GetOpFunc(OpCode, pDstType);
  2503. pOpRet = m_pBuilder->CreateCall(F, Args);
  2504. }
  2505. StoreResRetOutputAndStatus(Inst, pOpRet, DstType);
  2506. break;
  2507. }
  2508. case D3D11_SB_OPCODE_LD_UAV_TYPED:
  2509. case D3DWDDM1_3_SB_OPCODE_LD_UAV_TYPED_FEEDBACK: {
  2510. bool bHasStatus = DXBC::HasFeedback(Inst.OpCode());
  2511. const unsigned uOpOutput = 0;
  2512. const unsigned uOpStatus = 1;
  2513. const unsigned uOpCoord = uOpStatus + (bHasStatus ? 1 : 0);
  2514. const unsigned uOpUAV = uOpCoord + 1;
  2515. DXASSERT_DXBC(Inst.m_Operands[uOpUAV].m_Type == D3D11_SB_OPERAND_TYPE_UNORDERED_ACCESS_VIEW);
  2516. const DxilResource &R = m_pPR->GetUAV(m_UAVRangeMap[Inst.m_Operands[uOpUAV].m_Index[0].m_RegIndex]);
  2517. // Resource.
  2518. OperandValue InUAV;
  2519. LoadOperand(InUAV, Inst, uOpUAV, CMask::MakeXMask(), CompType::getInvalid());
  2520. // Return type.
  2521. CompType DstType = DXBC::GetCompTypeWithMinPrec(R.GetCompType().GetBaseCompType(), Inst.m_Operands[uOpOutput].m_MinPrecision);
  2522. Type *pDstType = DstType.GetLLVMType(m_Ctx);
  2523. // Create Load call.
  2524. Value *pOpRet;
  2525. if (R.GetKind() != DxilResource::Kind::TypedBuffer) {
  2526. OP::OpCode OpCode = OP::OpCode::TextureLoad;
  2527. // Coordinates.
  2528. OperandValue InCoord;
  2529. CMask CoordMask = CMask::MakeFirstNCompMask(DXBC::GetNumResCoords(R.GetKind()));
  2530. LoadOperand(InCoord, Inst, uOpCoord, CoordMask, CompType::getI32());
  2531. Value *Args[9];
  2532. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  2533. Args[1] = InUAV[0]; // RWTexture UAV handle
  2534. Args[2] = m_pUnusedI32; // MIP level.
  2535. // Coordinates.
  2536. Args[3] = CoordMask.IsSet(0) ? InCoord[0] : m_pUnusedI32; // Coordinate 0
  2537. Args[4] = CoordMask.IsSet(1) ? InCoord[1] : m_pUnusedI32; // Coordinate 1
  2538. Args[5] = CoordMask.IsSet(2) ? InCoord[2] : m_pUnusedI32; // Coordinate 2
  2539. // Offsets.
  2540. Args[6] = m_pUnusedI32; // Offset 0
  2541. Args[7] = m_pUnusedI32; // Offset 1
  2542. Args[8] = m_pUnusedI32; // Offset 2
  2543. Function *F = m_pOP->GetOpFunc(OpCode, pDstType);
  2544. pOpRet = m_pBuilder->CreateCall(F, Args);
  2545. } else {
  2546. // R.GetKind() == DxilResource::TypedBuffer
  2547. OP::OpCode OpCode = OP::OpCode::BufferLoad;
  2548. Value *Args[4];
  2549. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  2550. Args[1] = InUAV[0]; // RWBuffer UAV handle
  2551. Args[2] = GetCoordValue(Inst, uOpCoord); // Coord 0: in elements
  2552. Args[3] = m_pUnusedI32; // Coord 1: undef
  2553. Function *F = m_pOP->GetOpFunc(OpCode, pDstType);
  2554. pOpRet = m_pBuilder->CreateCall(F, Args);
  2555. }
  2556. StoreResRetOutputAndStatus(Inst, pOpRet, DstType);
  2557. break;
  2558. }
  2559. case D3D11_SB_OPCODE_STORE_UAV_TYPED: {
  2560. const unsigned uOpUAV = 0;
  2561. const unsigned uOpCoord = uOpUAV + 1;
  2562. const unsigned uOpValue = uOpCoord + 1;
  2563. DXASSERT_DXBC(Inst.m_Operands[uOpUAV].m_Type == D3D11_SB_OPERAND_TYPE_UNORDERED_ACCESS_VIEW);
  2564. const DxilResource &R = m_pPR->GetUAV(m_UAVRangeMap[Inst.m_Operands[uOpUAV].m_Index[0].m_RegIndex]);
  2565. OperandValue InUAV, InCoord, InValue;
  2566. // Resource.
  2567. LoadOperand(InUAV, Inst, uOpUAV, CMask::MakeXMask(), CompType::getInvalid());
  2568. // Coordinates.
  2569. CMask CoordMask = CMask::MakeFirstNCompMask(DXBC::GetNumResCoords(R.GetKind()));
  2570. LoadOperand(InCoord, Inst, uOpCoord, CoordMask, CompType::getI32());
  2571. // Value type.
  2572. CompType ValueType = DXBC::GetCompTypeWithMinPrec(R.GetCompType().GetBaseCompType(), Inst.m_Operands[uOpUAV].m_MinPrecision);
  2573. Type *pValueType = ValueType.GetLLVMType(m_Ctx);
  2574. // Value.
  2575. CMask ValueMask = CMask::FromDXBC(Inst.m_Operands[uOpUAV].m_WriteMask);
  2576. LoadOperand(InValue, Inst, uOpValue, ValueMask, ValueType);
  2577. // Create Store call.
  2578. if (R.GetKind() != DxilResource::Kind::TypedBuffer) {
  2579. OP::OpCode OpCode = OP::OpCode::TextureStore;
  2580. Value *Args[10];
  2581. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  2582. Args[1] = InUAV[0]; // RWTexture UAV handle
  2583. // Coordinates.
  2584. Args[2] = CoordMask.IsSet(0) ? InCoord[0] : m_pUnusedI32; // Coordinate 0
  2585. Args[3] = CoordMask.IsSet(1) ? InCoord[1] : m_pUnusedI32; // Coordinate 1
  2586. Args[4] = CoordMask.IsSet(2) ? InCoord[2] : m_pUnusedI32; // Coordinate 2
  2587. // Value.
  2588. Args[5] = ValueMask.IsSet(0) ? InValue[0] : m_pUnusedI32; // Value 0
  2589. Args[6] = ValueMask.IsSet(1) ? InValue[1] : m_pUnusedI32; // Value 1
  2590. Args[7] = ValueMask.IsSet(2) ? InValue[2] : m_pUnusedI32; // Value 2
  2591. Args[8] = ValueMask.IsSet(3) ? InValue[3] : m_pUnusedI32; // Value 3
  2592. Args[9] = m_pOP->GetU8Const(ValueMask.ToByte()); // Value mask
  2593. Function *F = m_pOP->GetOpFunc(OpCode, pValueType);
  2594. MarkPrecise(m_pBuilder->CreateCall(F, Args));
  2595. } else {
  2596. // R.GetKind() == DxilResource::TypedBuffer
  2597. OP::OpCode OpCode = OP::OpCode::BufferStore;
  2598. Value *Args[9];
  2599. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  2600. Args[1] = InUAV[0]; // RWBuffer UAV handle
  2601. Args[2] = InCoord[0]; // Coord 0: in elements
  2602. Args[3] = m_pUnusedI32; // Coord 1: unused
  2603. Args[4] = ValueMask.IsSet(0) ? InValue[0] : m_pUnusedI32; // Value 0
  2604. Args[5] = ValueMask.IsSet(1) ? InValue[1] : m_pUnusedI32; // Value 1
  2605. Args[6] = ValueMask.IsSet(2) ? InValue[2] : m_pUnusedI32; // Value 2
  2606. Args[7] = ValueMask.IsSet(3) ? InValue[3] : m_pUnusedI32; // Value 3
  2607. Args[8] = m_pOP->GetU8Const(ValueMask.ToByte()); // Value mask
  2608. Function *F = m_pOP->GetOpFunc(OpCode, pValueType);
  2609. MarkPrecise(m_pBuilder->CreateCall(F, Args));
  2610. }
  2611. break;
  2612. }
  2613. case D3D11_SB_OPCODE_LD_RAW:
  2614. case D3DWDDM1_3_SB_OPCODE_LD_RAW_FEEDBACK: {
  2615. bool bHasFeedback = DXBC::HasFeedback(Inst.OpCode());
  2616. const unsigned uOpOutput = 0;
  2617. const unsigned uOpStatus = 1;
  2618. const unsigned uOpByteOffset = uOpStatus + (bHasFeedback ? 1 : 0);
  2619. const unsigned uOpRes = uOpByteOffset + 1;
  2620. // Byte offset.
  2621. Value *pByteOffset = GetCoordValue(Inst, uOpByteOffset);
  2622. if (Inst.m_Operands[uOpRes].m_Type != D3D11_SB_OPERAND_TYPE_THREAD_GROUP_SHARED_MEMORY) {
  2623. OP::OpCode OpCode = OP::OpCode::BufferLoad;
  2624. OperandValue InRes, InByteOffset;
  2625. // Resource.
  2626. LoadOperand(InRes, Inst, uOpRes, CMask::MakeXMask(), CompType::getInvalid());
  2627. // Create Load call.
  2628. Value *Args[4];
  2629. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  2630. Args[1] = InRes[0]; // [RW]ByteAddressBuffer UAV/SRV handle
  2631. Args[2] = pByteOffset; // Coord 0: in bytes
  2632. Args[3] = m_pUnusedI32; // Coord 1: unused
  2633. CompType DstType = CompType::getI32();
  2634. Type *pDstType = DstType.GetLLVMType(m_Ctx);
  2635. Function *F = m_pOP->GetOpFunc(OpCode, pDstType);
  2636. Value *pOpRet = m_pBuilder->CreateCall(F, Args);
  2637. StoreResRetOutputAndStatus(Inst, pOpRet, DstType);
  2638. } else {
  2639. const unsigned uOpTGSM = uOpRes;
  2640. CompType SrcType = CompType::getI32();
  2641. ConvertLoadTGSM(Inst, uOpTGSM, uOpOutput, SrcType, pByteOffset);
  2642. }
  2643. break;
  2644. }
  2645. case D3D11_SB_OPCODE_STORE_RAW: {
  2646. const unsigned uOpRes = 0;
  2647. const unsigned uOpByteOffset = uOpRes + 1;
  2648. const unsigned uOpValue = uOpByteOffset + 1;
  2649. // Byte offset.
  2650. Value *pByteOffset = GetCoordValue(Inst, uOpByteOffset);
  2651. if (Inst.m_Operands[uOpRes].m_Type == D3D11_SB_OPERAND_TYPE_UNORDERED_ACCESS_VIEW) {
  2652. const unsigned uOpUAV = uOpRes;
  2653. OP::OpCode OpCode = OP::OpCode::BufferStore;
  2654. DXASSERT_DXBC(Inst.m_Operands[uOpUAV].m_Type == D3D11_SB_OPERAND_TYPE_UNORDERED_ACCESS_VIEW);
  2655. OperandValue InUAV, InByteOffset, InValue;
  2656. // Resource.
  2657. LoadOperand(InUAV, Inst, uOpUAV, CMask::MakeXMask(), CompType::getInvalid());
  2658. // Value type.
  2659. CompType ValueType = CompType::getI32();
  2660. Type *pValueType = ValueType.GetLLVMType(m_Ctx);
  2661. // Value.
  2662. CMask ValueMask = CMask::FromDXBC(Inst.m_Operands[uOpUAV].m_WriteMask);
  2663. LoadOperand(InValue, Inst, uOpValue, ValueMask, ValueType);
  2664. // Create Store call.
  2665. Value *Args[9];
  2666. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  2667. Args[1] = InUAV[0]; // RWByteAddressBuffer UAV handle
  2668. Args[2] = pByteOffset; // Coord 0: in bytes
  2669. Args[3] = m_pUnusedI32; // Coord 1: undef
  2670. Args[4] = ValueMask.IsSet(0) ? InValue[0] : m_pUnusedI32; // Value 0
  2671. Args[5] = ValueMask.IsSet(1) ? InValue[1] : m_pUnusedI32; // Value 1
  2672. Args[6] = ValueMask.IsSet(2) ? InValue[2] : m_pUnusedI32; // Value 2
  2673. Args[7] = ValueMask.IsSet(3) ? InValue[3] : m_pUnusedI32; // Value 3
  2674. Args[8] = m_pOP->GetU8Const(ValueMask.ToByte()); // Value mask
  2675. Function *F = m_pOP->GetOpFunc(OpCode, pValueType);
  2676. MarkPrecise(m_pBuilder->CreateCall(F, Args));
  2677. } else {
  2678. const unsigned uOpTGSM = uOpRes;
  2679. CompType ValueType = CompType::getI32();
  2680. ConvertStoreTGSM(Inst, uOpTGSM, uOpValue, ValueType, pByteOffset);
  2681. }
  2682. break;
  2683. }
  2684. case D3D11_SB_OPCODE_LD_STRUCTURED:
  2685. case D3DWDDM1_3_SB_OPCODE_LD_STRUCTURED_FEEDBACK: {
  2686. bool bHasFeedback = DXBC::HasFeedback(Inst.OpCode());
  2687. const unsigned uOpOutput = 0;
  2688. const unsigned uOpStatus = 1;
  2689. const unsigned uOpElementOffset = uOpStatus + (bHasFeedback ? 1 : 0);
  2690. const unsigned uOpStructByteOffset = uOpElementOffset + 1;
  2691. const unsigned uOpRes = uOpStructByteOffset + 1;
  2692. if (Inst.m_Operands[uOpRes].m_Type != D3D11_SB_OPERAND_TYPE_THREAD_GROUP_SHARED_MEMORY) {
  2693. OP::OpCode OpCode = OP::OpCode::BufferLoad;
  2694. OperandValue InRes, InElementOffset, InStructByteOffset;
  2695. // Resource.
  2696. LoadOperand(InRes, Inst, uOpRes, CMask::MakeXMask(), CompType::getInvalid());
  2697. // Create Load call.
  2698. Value *Args[4];
  2699. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  2700. Args[1] = InRes[0]; // [RW]ByteAddressBuffer UAV/SRV handle
  2701. Args[2] = GetCoordValue(Inst, uOpElementOffset); // Coord 1: element index
  2702. Args[3] = GetCoordValue(Inst, uOpStructByteOffset); // Coord 2: byte offset within the element
  2703. CompType DstType = CompType::getI32();
  2704. Type *pDstType = DstType.GetLLVMType(m_Ctx);
  2705. Function *F = m_pOP->GetOpFunc(OpCode, pDstType);
  2706. Value *pOpRet = m_pBuilder->CreateCall(F, Args);
  2707. StoreResRetOutputAndStatus(Inst, pOpRet, DstType);
  2708. } else {
  2709. const unsigned uOpTGSM = uOpRes;
  2710. DXASSERT_DXBC(Inst.m_Operands[uOpTGSM].m_Type == D3D11_SB_OPERAND_TYPE_THREAD_GROUP_SHARED_MEMORY);
  2711. const TGSMEntry &R = m_TGSMMap[Inst.m_Operands[uOpTGSM].m_Index[0].m_RegIndex];
  2712. CompType SrcType = CompType::getF32();
  2713. // Byte offset.
  2714. Value *pByteOffset = GetByteOffset(Inst, uOpElementOffset, uOpStructByteOffset, R.Stride);
  2715. ConvertLoadTGSM(Inst, uOpTGSM, uOpOutput, SrcType, pByteOffset);
  2716. }
  2717. break;
  2718. }
  2719. case D3D11_SB_OPCODE_STORE_STRUCTURED: {
  2720. const unsigned uOpRes = 0;
  2721. const unsigned uOpElementOffset = uOpRes + 1;
  2722. const unsigned uOpStructByteOffset = uOpElementOffset + 1;
  2723. const unsigned uOpValue = uOpStructByteOffset + 1;
  2724. if (Inst.m_Operands[0].m_Type == D3D11_SB_OPERAND_TYPE_UNORDERED_ACCESS_VIEW) {
  2725. OP::OpCode OpCode = OP::OpCode::BufferStore;
  2726. const unsigned uOpUAV = uOpRes;
  2727. DXASSERT_DXBC(Inst.m_Operands[uOpUAV].m_Type == D3D11_SB_OPERAND_TYPE_UNORDERED_ACCESS_VIEW);
  2728. OperandValue InUAV, InElementOffset, InStructByteOffset, InValue;
  2729. // Resource.
  2730. LoadOperand(InUAV, Inst, uOpUAV, CMask::MakeXMask(), CompType::getInvalid());
  2731. // Value type.
  2732. CompType ValueType = CompType::getI32();
  2733. Type *pValueType = ValueType.GetLLVMType(m_Ctx);
  2734. // Value.
  2735. CMask ValueMask = CMask::FromDXBC(Inst.m_Operands[uOpUAV].m_WriteMask);
  2736. LoadOperand(InValue, Inst, uOpValue, ValueMask, ValueType);
  2737. // Create Store call.
  2738. Value *Args[9];
  2739. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  2740. Args[1] = InUAV[0]; // RWByteAddressBuffer UAV handle
  2741. Args[2] = GetCoordValue(Inst, uOpElementOffset); // Coord 1: element index
  2742. Args[3] = GetCoordValue(Inst, uOpStructByteOffset); // Coord 2: byte offset within the element
  2743. Args[4] = ValueMask.IsSet(0) ? InValue[0] : m_pUnusedI32; // Value 0
  2744. Args[5] = ValueMask.IsSet(1) ? InValue[1] : m_pUnusedI32; // Value 1
  2745. Args[6] = ValueMask.IsSet(2) ? InValue[2] : m_pUnusedI32; // Value 2
  2746. Args[7] = ValueMask.IsSet(3) ? InValue[3] : m_pUnusedI32; // Value 3
  2747. Args[8] = m_pOP->GetU8Const(ValueMask.ToByte()); // Value mask
  2748. Function *F = m_pOP->GetOpFunc(OpCode, pValueType);
  2749. MarkPrecise(m_pBuilder->CreateCall(F, Args));
  2750. } else {
  2751. const unsigned uOpTGSM = uOpRes;
  2752. const TGSMEntry &R = m_TGSMMap[Inst.m_Operands[uOpTGSM].m_Index[0].m_RegIndex];
  2753. CompType ValueType = CompType::getF32();
  2754. // Byte offset.
  2755. Value *pByteOffset = GetByteOffset(Inst, uOpElementOffset, uOpStructByteOffset, R.Stride);
  2756. ConvertStoreTGSM(Inst, uOpTGSM, uOpValue, ValueType, pByteOffset);
  2757. }
  2758. break;
  2759. }
  2760. //
  2761. // Atomic operations.
  2762. //
  2763. case D3D11_SB_OPCODE_ATOMIC_AND:
  2764. case D3D11_SB_OPCODE_ATOMIC_OR:
  2765. case D3D11_SB_OPCODE_ATOMIC_XOR:
  2766. case D3D11_SB_OPCODE_ATOMIC_IADD:
  2767. case D3D11_SB_OPCODE_ATOMIC_IMAX:
  2768. case D3D11_SB_OPCODE_ATOMIC_IMIN:
  2769. case D3D11_SB_OPCODE_ATOMIC_UMAX:
  2770. case D3D11_SB_OPCODE_ATOMIC_UMIN:
  2771. case D3D11_SB_OPCODE_IMM_ATOMIC_IADD:
  2772. case D3D11_SB_OPCODE_IMM_ATOMIC_AND:
  2773. case D3D11_SB_OPCODE_IMM_ATOMIC_OR:
  2774. case D3D11_SB_OPCODE_IMM_ATOMIC_XOR:
  2775. case D3D11_SB_OPCODE_IMM_ATOMIC_EXCH:
  2776. case D3D11_SB_OPCODE_IMM_ATOMIC_IMAX:
  2777. case D3D11_SB_OPCODE_IMM_ATOMIC_IMIN:
  2778. case D3D11_SB_OPCODE_IMM_ATOMIC_UMAX:
  2779. case D3D11_SB_OPCODE_IMM_ATOMIC_UMIN:
  2780. case D3D11_SB_OPCODE_ATOMIC_CMP_STORE:
  2781. case D3D11_SB_OPCODE_IMM_ATOMIC_CMP_EXCH: {
  2782. bool bHasReturn = DXBC::AtomicBinOpHasReturn(Inst.OpCode());
  2783. bool bHasCompare = DXBC::IsCompareExchAtomicBinOp(Inst.OpCode());
  2784. const unsigned uOpRes = bHasReturn ? 1 : 0;
  2785. const unsigned uOpCoord = uOpRes + 1;
  2786. const unsigned uOpCompareValue = uOpCoord + (bHasCompare ? 1 : 0);
  2787. const unsigned uOpValue = uOpCompareValue + 1;
  2788. if (Inst.m_Operands[uOpRes].m_Type == D3D11_SB_OPERAND_TYPE_UNORDERED_ACCESS_VIEW) {
  2789. const unsigned uOpUAV = uOpRes;
  2790. const DxilResource &R = m_pPR->GetUAV(m_UAVRangeMap[Inst.m_Operands[uOpUAV].m_Index[0].m_RegIndex]);
  2791. OperandValue InUAV, InCoord, InCompareValue, InValue;
  2792. // Resource.
  2793. LoadOperand(InUAV, Inst, uOpUAV, CMask::MakeXMask(), CompType::getInvalid());
  2794. // Coordinates.
  2795. CMask CoordMask = CMask::MakeFirstNCompMask(DxilResource::GetNumCoords(R.GetKind()));
  2796. LoadOperand(InCoord, Inst, uOpCoord, CoordMask, CompType::getI32());
  2797. Value *pOffset[3];
  2798. pOffset[0] = InCoord[0];
  2799. pOffset[1] = CoordMask.IsSet(1) ? InCoord[1] : m_pUnusedI32;
  2800. pOffset[2] = CoordMask.IsSet(2) ? InCoord[2] : m_pUnusedI32;
  2801. // Value type.
  2802. CompType ValueType = CompType::getI32();
  2803. Type *pValueType = ValueType.GetLLVMType(m_Ctx);
  2804. // Compare value.
  2805. if (bHasCompare) {
  2806. LoadOperand(InCompareValue, Inst, uOpCompareValue, CMask::MakeXMask(), ValueType);
  2807. }
  2808. // Value.
  2809. LoadOperand(InValue, Inst, uOpValue, CMask::MakeXMask(), ValueType);
  2810. // Create atomic call.
  2811. Value *pOpRet;
  2812. if (!bHasCompare) {
  2813. OP::OpCode OpCode = OP::OpCode::AtomicBinOp;
  2814. Value *Args[7];
  2815. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  2816. Args[1] = InUAV[0]; // Typed (uint/int) UAV handle
  2817. Args[2] = m_pOP->GetU32Const((unsigned)DXBC::GetAtomicBinOp(Inst.OpCode())); // Atomic operation kind.
  2818. Args[3] = pOffset[0]; // Offset 0, in elements
  2819. Args[4] = pOffset[1]; // Offset 1
  2820. Args[5] = pOffset[2]; // Offset 2
  2821. Args[6] = InValue[0]; // New value
  2822. Function *F = m_pOP->GetOpFunc(OpCode, pValueType);
  2823. pOpRet = m_pBuilder->CreateCall(F, Args);
  2824. } else {
  2825. OP::OpCode OpCode = OP::OpCode::AtomicCompareExchange;
  2826. Value *Args[7];
  2827. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  2828. Args[1] = InUAV[0]; // Typed (uint/int) UAV handle
  2829. Args[2] = pOffset[0]; // Offset 0, in elements
  2830. Args[3] = pOffset[1]; // Offset 1
  2831. Args[4] = pOffset[2]; // Offset 2
  2832. Args[5] = InCompareValue[0]; // Compare value
  2833. Args[6] = InValue[0]; // New value
  2834. Function *F = m_pOP->GetOpFunc(OpCode, pValueType);
  2835. pOpRet = m_pBuilder->CreateCall(F, Args);
  2836. }
  2837. StoreBroadcastOutput(Inst, pOpRet, ValueType);
  2838. } else {
  2839. const unsigned uOpTGSM = uOpRes;
  2840. DXASSERT_DXBC(Inst.m_Operands[uOpTGSM].m_Type == D3D11_SB_OPERAND_TYPE_THREAD_GROUP_SHARED_MEMORY);
  2841. const TGSMEntry &R = m_TGSMMap[Inst.m_Operands[uOpTGSM].m_Index[0].m_RegIndex];
  2842. OperandValue InElementOffset, InCompareValue, InValue;
  2843. // Byte offset.
  2844. CMask ElementOffsetMask = CMask::MakeFirstNCompMask(R.Stride == 1 ? 1 : 2);
  2845. LoadOperand(InElementOffset, Inst, uOpCoord, ElementOffsetMask, CompType::getI32());
  2846. Value *pByteOffset = InElementOffset[0];
  2847. if (R.Stride > 1) { // Structured TGSM.
  2848. Value *pOffset2 = InElementOffset[1];
  2849. Value *pStride = m_pOP->GetU32Const(R.Stride);
  2850. pByteOffset = m_pBuilder->CreateAdd(m_pBuilder->CreateMul(pByteOffset, pStride), pOffset2);
  2851. }
  2852. // Value type.
  2853. CompType ValueType = CompType::getI32();
  2854. // Compare value.
  2855. if (bHasCompare) {
  2856. LoadOperand(InCompareValue, Inst, uOpCompareValue, CMask::MakeXMask(), ValueType);
  2857. }
  2858. CompType DstType = CompType::getI32();
  2859. Type *pDstType = Type::getInt32PtrTy(m_Ctx, DXIL::kTGSMAddrSpace);
  2860. // Value.
  2861. LoadOperand(InValue, Inst, uOpValue, CMask::MakeXMask(), ValueType);
  2862. // Create GEP.
  2863. Value *pGEPIndices[2] = { m_pOP->GetU32Const(0), pByteOffset };
  2864. Value *pPtrI8 = m_pBuilder->CreateGEP(R.pVar, pGEPIndices);
  2865. Value *pPtr = m_pBuilder->CreatePointerCast(pPtrI8, pDstType);
  2866. // Generate atomic instruction.
  2867. Value *pRetVal;
  2868. if (!bHasCompare) {
  2869. pRetVal = m_pBuilder->CreateAtomicRMW(DXBC::GetLlvmAtomicBinOp(Inst.OpCode()), pPtr, InValue[0], AtomicOrdering::Monotonic);
  2870. } else {
  2871. pRetVal = m_pBuilder->CreateAtomicCmpXchg(pPtr, InCompareValue[0], InValue[0], AtomicOrdering::Monotonic, AtomicOrdering::Monotonic);
  2872. Type *RetTypeFields[2] = { Type::getInt32Ty(m_Ctx), Type::getInt1Ty(m_Ctx) };
  2873. pRetVal = m_pBuilder->CreateExtractValue(pRetVal, 0);
  2874. }
  2875. StoreBroadcastOutput(Inst, pRetVal, ValueType);
  2876. }
  2877. break;
  2878. }
  2879. case D3D10_1_SB_OPCODE_GATHER4:
  2880. case D3DWDDM1_3_SB_OPCODE_GATHER4_FEEDBACK: {
  2881. OP::OpCode OpCode = OP::OpCode::TextureGather;
  2882. bool bHasFeedback = DXBC::HasFeedback(Inst.OpCode());
  2883. const unsigned uOpOutput = 0;
  2884. const unsigned uOpCoord = uOpOutput + 1 + (bHasFeedback ? 1 : 0);
  2885. const unsigned uOpSRV = DXBC::GetResourceSlot(Inst.OpCode());
  2886. const unsigned uOpSampler = uOpSRV + 1;
  2887. const DxilResource &R = GetSRVFromOperand(Inst, uOpSRV);
  2888. Value *Args[10];
  2889. LoadCommonSampleInputs(Inst, &Args[0]);
  2890. // Other arguments.
  2891. Args[0] = m_pOP->GetU32Const((unsigned)OpCode);
  2892. // Offset.
  2893. bool bUseOffset = (R.GetKind() == DxilResource::Kind::Texture2D) ||
  2894. (R.GetKind() == DxilResource::Kind::Texture2DArray);
  2895. if (!bUseOffset) {
  2896. Args[7] = m_pUnusedI32;
  2897. Args[8] = m_pUnusedI32;
  2898. }
  2899. // Channel.
  2900. unsigned uChannel = Inst.m_Operands[uOpSampler].m_ComponentName;
  2901. Args[9] = m_pOP->GetU32Const(uChannel);
  2902. // Function call.
  2903. CompType DstType = DXBC::GetCompTypeWithMinPrec(R.GetCompType().GetBaseCompType(), Inst.m_Operands[uOpOutput].m_MinPrecision);
  2904. Type *pDstType = DstType.GetLLVMType(m_Ctx);
  2905. Function *F = m_pOP->GetOpFunc(OpCode, pDstType);
  2906. Value *pOpRet = m_pBuilder->CreateCall(F, Args);
  2907. StoreResRetOutputAndStatus(Inst, pOpRet, DstType);
  2908. break;
  2909. }
  2910. case D3D11_SB_OPCODE_GATHER4_C:
  2911. case D3DWDDM1_3_SB_OPCODE_GATHER4_C_FEEDBACK: {
  2912. OP::OpCode OpCode = OP::OpCode::TextureGatherCmp;
  2913. bool bHasFeedback = DXBC::HasFeedback(Inst.OpCode());
  2914. const unsigned uOpOutput = 0;
  2915. const unsigned uOpCoord = uOpOutput + 1 + (bHasFeedback ? 1 : 0);
  2916. const unsigned uOpSRV = DXBC::GetResourceSlot(Inst.OpCode());
  2917. const unsigned uOpSampler = uOpSRV + 1;
  2918. const unsigned uOpCmp = uOpSampler + 1;
  2919. const DxilResource &R = GetSRVFromOperand(Inst, uOpSRV);
  2920. Value *Args[11];
  2921. LoadCommonSampleInputs(Inst, &Args[0]);
  2922. // Other arguments.
  2923. Args[0] = m_pOP->GetU32Const((unsigned)OpCode);
  2924. // Offset.
  2925. bool bUseOffset = (R.GetKind() == DxilResource::Kind::Texture2D) ||
  2926. (R.GetKind() == DxilResource::Kind::Texture2DArray);
  2927. if (!bUseOffset) {
  2928. Args[7] = m_pUnusedI32;
  2929. Args[8] = m_pUnusedI32;
  2930. }
  2931. // Channel.
  2932. unsigned uChannel = Inst.m_Operands[uOpSampler].m_ComponentName;
  2933. Args[9] = m_pOP->GetU32Const(uChannel);
  2934. // Comparison value.
  2935. OperandValue InCmp;
  2936. LoadOperand(InCmp, Inst, uOpCmp, CMask::MakeXMask(), CompType::getF32());
  2937. Args[10] = InCmp[0];
  2938. // Function call.
  2939. CompType DstType = DXBC::GetCompTypeWithMinPrec(R.GetCompType().GetBaseCompType(), Inst.m_Operands[uOpOutput].m_MinPrecision);
  2940. Type *pDstType = DstType.GetLLVMType(m_Ctx);
  2941. Function *F = m_pOP->GetOpFunc(OpCode, pDstType);
  2942. Value *pOpRet = m_pBuilder->CreateCall(F, Args);
  2943. StoreResRetOutputAndStatus(Inst, pOpRet, DstType);
  2944. break;
  2945. }
  2946. case D3D11_SB_OPCODE_GATHER4_PO:
  2947. case D3DWDDM1_3_SB_OPCODE_GATHER4_PO_FEEDBACK: {
  2948. OP::OpCode OpCode = OP::OpCode::TextureGather;
  2949. bool bHasFeedback = DXBC::HasFeedback(Inst.OpCode());
  2950. const unsigned uOpOutput = 0;
  2951. const unsigned uOpCoord = uOpOutput + 1 + (bHasFeedback ? 1 : 0);
  2952. const unsigned uOpOffset = uOpCoord + 1;
  2953. const unsigned uOpSRV = DXBC::GetResourceSlot(Inst.OpCode());
  2954. const unsigned uOpSampler = uOpSRV + 1;
  2955. const DxilResource &R = GetSRVFromOperand(Inst, uOpSRV);
  2956. Value *Args[10];
  2957. LoadCommonSampleInputs(Inst, &Args[0], false);
  2958. // Other arguments.
  2959. Args[0] = m_pOP->GetU32Const((unsigned)OpCode);
  2960. // Programmable offset.
  2961. OperandValue InOffset;
  2962. LoadOperand(InOffset, Inst, uOpOffset, CMask::MakeFirstNCompMask(2), CompType::getI32());
  2963. Args[7] = InOffset[0];
  2964. Args[8] = InOffset[1];
  2965. // Channel.
  2966. unsigned uChannel = Inst.m_Operands[uOpSampler].m_ComponentName;
  2967. Args[9] = m_pOP->GetU32Const(uChannel);
  2968. // Function call.
  2969. CompType DstType = DXBC::GetCompTypeWithMinPrec(R.GetCompType().GetBaseCompType(), Inst.m_Operands[uOpOutput].m_MinPrecision);
  2970. Type *pDstType = DstType.GetLLVMType(m_Ctx);
  2971. Function *F = m_pOP->GetOpFunc(OpCode, pDstType);
  2972. Value *pOpRet = m_pBuilder->CreateCall(F, Args);
  2973. StoreResRetOutputAndStatus(Inst, pOpRet, DstType);
  2974. break;
  2975. }
  2976. case D3D11_SB_OPCODE_GATHER4_PO_C:
  2977. case D3DWDDM1_3_SB_OPCODE_GATHER4_PO_C_FEEDBACK: {
  2978. OP::OpCode OpCode = OP::OpCode::TextureGatherCmp;
  2979. bool bHasFeedback = DXBC::HasFeedback(Inst.OpCode());
  2980. const unsigned uOpOutput = 0;
  2981. const unsigned uOpCoord = uOpOutput + 1 + (bHasFeedback ? 1 : 0);
  2982. const unsigned uOpOffset = uOpCoord + 1;
  2983. const unsigned uOpSRV = DXBC::GetResourceSlot(Inst.OpCode());
  2984. const unsigned uOpSampler = uOpSRV + 1;
  2985. const unsigned uOpCmp = uOpSampler + 1;
  2986. const DxilResource &R = GetSRVFromOperand(Inst, uOpSRV);
  2987. Value *Args[11];
  2988. LoadCommonSampleInputs(Inst, &Args[0], false);
  2989. // Other arguments.
  2990. Args[0] = m_pOP->GetU32Const((unsigned)OpCode);
  2991. // Programmable offset.
  2992. OperandValue InOffset;
  2993. LoadOperand(InOffset, Inst, uOpOffset, CMask::MakeFirstNCompMask(2), CompType::getI32());
  2994. Args[7] = InOffset[0];
  2995. Args[8] = InOffset[1];
  2996. // Channel.
  2997. unsigned uChannel = Inst.m_Operands[uOpSampler].m_ComponentName;
  2998. Args[9] = m_pOP->GetU32Const(uChannel);
  2999. // Comparison value.
  3000. OperandValue InCmp;
  3001. LoadOperand(InCmp, Inst, uOpCmp, CMask::MakeXMask(), CompType::getF32());
  3002. Args[10] = InCmp[0];
  3003. // Function call.
  3004. CompType DstType = DXBC::GetCompTypeWithMinPrec(R.GetCompType().GetBaseCompType(), Inst.m_Operands[uOpOutput].m_MinPrecision);
  3005. Type *pDstType = DstType.GetLLVMType(m_Ctx);
  3006. Function *F = m_pOP->GetOpFunc(OpCode, pDstType);
  3007. Value *pOpRet = m_pBuilder->CreateCall(F, Args);
  3008. StoreResRetOutputAndStatus(Inst, pOpRet, DstType);
  3009. break;
  3010. }
  3011. case D3D10_1_SB_OPCODE_SAMPLE_POS: {
  3012. const unsigned uOpOutput = 0;
  3013. const unsigned uOpResOrRast = uOpOutput + 1;
  3014. const unsigned uOpSample = uOpResOrRast + 1;
  3015. // Sample.
  3016. OperandValue InSample;
  3017. LoadOperand(InSample, Inst, uOpSample, CMask::MakeXMask(), CompType::getI32());
  3018. Value *pOpRet;
  3019. if (Inst.m_Operands[uOpResOrRast].m_Type == D3D10_SB_OPERAND_TYPE_RESOURCE) {
  3020. // Resource.
  3021. OP::OpCode OpCode = OP::OpCode::Texture2DMSGetSamplePosition;
  3022. OperandValue InRes;
  3023. LoadOperand(InRes, Inst, uOpResOrRast, CMask::MakeXMask(), CompType::getInvalid());
  3024. // Create SamplePosition call.
  3025. Value *Args[3];
  3026. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  3027. Args[1] = InRes[0]; // Resource handle
  3028. Args[2] = InSample[0]; // Sample index
  3029. Function *F = m_pOP->GetOpFunc(OpCode, Type::getVoidTy(m_Ctx));
  3030. pOpRet = m_pBuilder->CreateCall(F, Args);
  3031. } else {
  3032. // Render target.
  3033. OP::OpCode OpCode = OP::OpCode::RenderTargetGetSamplePosition;
  3034. // Create SamplePosition call.
  3035. Value *Args[2];
  3036. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  3037. Args[1] = InSample[0]; // Sample index
  3038. Function *F = m_pOP->GetOpFunc(OpCode, Type::getVoidTy(m_Ctx));
  3039. pOpRet = m_pBuilder->CreateCall(F, Args);
  3040. }
  3041. StoreSamplePosOutput(Inst, pOpRet);
  3042. break;
  3043. }
  3044. case D3DWDDM1_3_SB_OPCODE_CHECK_ACCESS_FULLY_MAPPED: {
  3045. OP::OpCode OpCode = OP::OpCode::CheckAccessFullyMapped;
  3046. OperandValue InStatus;
  3047. LoadOperand(InStatus, Inst, 1, CMask::MakeXMask(), CompType::getI32());
  3048. // Create CheckAccessFullyMapped call.
  3049. Value *Args[2];
  3050. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  3051. Args[1] = InStatus[Inst.m_Operands[0].m_ComponentName]; // Status
  3052. Function *F = m_pOP->GetOpFunc(OpCode, Type::getInt32Ty(m_Ctx));
  3053. Value *pRetValue = m_pBuilder->CreateCall(F, Args);
  3054. pRetValue = CastDxbcValue(pRetValue, CompType::getI1(), CompType::getI32());
  3055. StoreBroadcastOutput(Inst, pRetValue, CompType::getI32());
  3056. break;
  3057. }
  3058. case D3D10_SB_OPCODE_RESINFO: {
  3059. OP::OpCode OpCode = OP::OpCode::GetDimensions;
  3060. const unsigned uOpOutput = 0;
  3061. const unsigned uOpMipLevel = uOpOutput + 1;
  3062. const unsigned uOpRes = uOpMipLevel + 1;
  3063. // MipLevel.
  3064. OperandValue InMipLevel;
  3065. LoadOperand(InMipLevel, Inst, uOpMipLevel, CMask::MakeXMask(), CompType::getI32());
  3066. // Resource.
  3067. OperandValue InRes;
  3068. LoadOperand(InRes, Inst, uOpRes, CMask::MakeXMask(), CompType::getInvalid());
  3069. // Create GetDimensions call.
  3070. Value *Args[3];
  3071. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  3072. Args[1] = InRes[0]; // Resource handle
  3073. Args[2] = InMipLevel[0]; // MipLevel
  3074. Function *F = m_pOP->GetOpFunc(OpCode, Type::getVoidTy(m_Ctx));
  3075. Value *pOpRet = m_pBuilder->CreateCall(F, Args);
  3076. StoreGetDimensionsOutput(Inst, pOpRet);
  3077. break;
  3078. }
  3079. case D3D11_SB_OPCODE_BUFINFO: {
  3080. OP::OpCode OpCode = OP::OpCode::GetDimensions;
  3081. const unsigned uOpOutput = 0;
  3082. const unsigned uOpRes = uOpOutput + 1;
  3083. // Resource.
  3084. OperandValue InRes;
  3085. LoadOperand(InRes, Inst, uOpRes, CMask::MakeXMask(), CompType::getInvalid());
  3086. // Create GetDimensions call.
  3087. Value *Args[3];
  3088. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  3089. Args[1] = InRes[0]; // Resource handle
  3090. Args[2] = m_pUnusedI32; // MipLevel (undefined)
  3091. Function *F = m_pOP->GetOpFunc(OpCode, Type::getVoidTy(m_Ctx));
  3092. Value *pOpRet = m_pBuilder->CreateCall(F, Args);
  3093. Value *pOpWidth = m_pBuilder->CreateExtractValue(pOpRet, 0);
  3094. // Store output.
  3095. StoreBroadcastOutput(Inst, pOpWidth, CompType::getI32());
  3096. break;
  3097. }
  3098. case D3D10_1_SB_OPCODE_SAMPLE_INFO: {
  3099. const unsigned uOpOutput = 0;
  3100. const unsigned uOpResOrRast = uOpOutput + 1;
  3101. bool bDxbcRetFloat = true;
  3102. if (Inst.m_InstructionReturnType == D3D10_SB_INSTRUCTION_RETURN_UINT) {
  3103. bDxbcRetFloat = false;
  3104. }
  3105. // Return type.
  3106. CompType DstType;
  3107. if (bDxbcRetFloat) {
  3108. DstType = DXBC::GetCompTypeWithMinPrec(CompType::getF32(), Inst.m_Operands[uOpOutput].m_MinPrecision);
  3109. } else {
  3110. DstType = DXBC::GetCompTypeWithMinPrec(CompType::getI32(), Inst.m_Operands[uOpOutput].m_MinPrecision);
  3111. }
  3112. Value *pRetValue;
  3113. if (Inst.m_Operands[uOpResOrRast].m_Type == D3D10_SB_OPERAND_TYPE_RESOURCE) {
  3114. // Resource.
  3115. OP::OpCode OpCode = OP::OpCode::GetDimensions;
  3116. OperandValue InRes;
  3117. LoadOperand(InRes, Inst, uOpResOrRast, CMask::MakeXMask(), CompType::getInvalid());
  3118. // Create GetDimensions call.
  3119. Value *Args[3];
  3120. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  3121. Args[1] = InRes[0]; // Resource handle
  3122. Args[2] = m_pOP->GetU32Const(0); // MipLevel
  3123. Function *F = m_pOP->GetOpFunc(OpCode, Type::getVoidTy(m_Ctx));
  3124. Value *pOpRet = m_pBuilder->CreateCall(F, Args);
  3125. pRetValue = m_pBuilder->CreateExtractValue(pOpRet, 3);
  3126. } else {
  3127. OP::OpCode OpCode = OP::OpCode::RenderTargetGetSampleCount;
  3128. // Create SampleCount call.
  3129. Value *Args[1];
  3130. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  3131. Function *F = m_pOP->GetOpFunc(OpCode, Type::getVoidTy(m_Ctx));
  3132. pRetValue = m_pBuilder->CreateCall(F, Args);
  3133. }
  3134. Value *pZeroValue;
  3135. if (bDxbcRetFloat) {
  3136. pRetValue = m_pBuilder->CreateCast(Instruction::CastOps::UIToFP, pRetValue, Type::getFloatTy(m_Ctx));
  3137. pZeroValue = m_pOP->GetFloatConst(0.f);
  3138. } else {
  3139. pZeroValue = m_pOP->GetU32Const(0);
  3140. }
  3141. // Store output.
  3142. CMask OutputMask = CMask::FromDXBC(Inst.m_Operands[uOpOutput].m_WriteMask);
  3143. if (!OutputMask.IsZero()) {
  3144. OperandValue Out;
  3145. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  3146. if (!OutputMask.IsSet(c)) continue;
  3147. BYTE Comp = Inst.m_Operands[uOpResOrRast].m_Swizzle[c];
  3148. if (Comp == 0) {
  3149. Out[c] = pRetValue;
  3150. } else {
  3151. Out[c] = pZeroValue;
  3152. }
  3153. }
  3154. StoreOperand(Out, Inst, uOpOutput, OutputMask, DstType);
  3155. }
  3156. break;
  3157. }
  3158. case D3D11_SB_OPCODE_IMM_ATOMIC_ALLOC:
  3159. case D3D11_SB_OPCODE_IMM_ATOMIC_CONSUME: {
  3160. OP::OpCode OpCode = OP::OpCode::BufferUpdateCounter;
  3161. const unsigned uOpOutput = 0;
  3162. const unsigned uOpUAV = uOpOutput + 1;
  3163. bool bInc = Inst.OpCode() == D3D11_SB_OPCODE_IMM_ATOMIC_ALLOC;
  3164. // Resource.
  3165. OperandValue InRes;
  3166. LoadOperand(InRes, Inst, uOpUAV, CMask::MakeXMask(), CompType::getInvalid());
  3167. // SetHasCounter.
  3168. SetHasCounter(Inst, uOpUAV);
  3169. // Create BufferUpdateCounter call.
  3170. Value *Args[3];
  3171. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  3172. Args[1] = InRes[0]; // Resource handle
  3173. Args[2] = m_pOP->GetI8Const(bInc ? 1 : -1); // Inc or Dec
  3174. CompType DstType = DXBC::GetCompTypeWithMinPrec(CompType::getI32(), Inst.m_Operands[uOpOutput].m_MinPrecision);
  3175. Function *F = m_pOP->GetOpFunc(OpCode, Type::getVoidTy(m_Ctx));
  3176. Value *pOpRet = m_pBuilder->CreateCall(F, Args);
  3177. StoreBroadcastOutput(Inst, pOpRet, DstType);
  3178. break;
  3179. }
  3180. case D3D11_SB_OPCODE_SYNC: {
  3181. OP::OpCode OpCode = OP::OpCode::Barrier;
  3182. DXIL::BarrierMode BMode = DXBC::GetBarrierMode(Inst.m_SyncFlags.bThreadsInGroup,
  3183. Inst.m_SyncFlags.bUnorderedAccessViewMemoryGlobal,
  3184. Inst.m_SyncFlags.bUnorderedAccessViewMemoryGroup,
  3185. Inst.m_SyncFlags.bThreadGroupSharedMemory);
  3186. // Create BufferUpdateCounter call.
  3187. Value *Args[2];
  3188. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  3189. Args[1] = m_pOP->GetU32Const((unsigned)BMode); // Barrier mode
  3190. Function *F = m_pOP->GetOpFunc(OpCode, Type::getVoidTy(m_Ctx));
  3191. MarkPrecise(m_pBuilder->CreateCall(F, Args));
  3192. break;
  3193. }
  3194. //
  3195. // Control-flow operations.
  3196. //
  3197. case D3D10_SB_OPCODE_IF: {
  3198. DXASSERT_DXBC(Inst.m_NumOperands == 1);
  3199. // Create If-scope.
  3200. Scope &Scope = m_ScopeStack.Push(Scope::If, m_pBuilder->GetInsertBlock());
  3201. // Prepare condition.
  3202. Scope.pCond = LoadZNZCondition(Inst, 0);
  3203. // Create then-branch BB and set it as active.
  3204. Scope.pThenBB = BasicBlock::Create(m_Ctx, Twine("if") + Twine(Scope.NameIndex) + Twine(".then"), pFunction);
  3205. m_pBuilder->SetInsertPoint(Scope.pThenBB);
  3206. // Create endif BB.
  3207. Scope.pPostScopeBB = BasicBlock::Create(m_Ctx, Twine("if") + Twine(Scope.NameIndex) + Twine(".end"));
  3208. break;
  3209. }
  3210. case D3D10_SB_OPCODE_ELSE: {
  3211. // Get If-scope.
  3212. Scope &Scope = m_ScopeStack.Top();
  3213. IFTBOOL(Scope.Kind == Scope::If, E_FAIL);
  3214. // Terminate then-branch.
  3215. CreateBranchIfNeeded(m_pBuilder->GetInsertBlock(), Scope.pPostScopeBB);
  3216. // Create else-branch BB and set it as active.
  3217. Scope.pElseBB = BasicBlock::Create(m_Ctx, Twine("if") + Twine(Scope.NameIndex) + Twine(".else"), pFunction);
  3218. m_pBuilder->SetInsertPoint(Scope.pElseBB);
  3219. break;
  3220. }
  3221. case D3D10_SB_OPCODE_ENDIF: {
  3222. // Get If-scope.
  3223. Scope &Scope = m_ScopeStack.Top();
  3224. IFTBOOL(Scope.Kind == Scope::If, E_FAIL);
  3225. // Terminate else-branch.
  3226. CreateBranchIfNeeded(m_pBuilder->GetInsertBlock(), Scope.pPostScopeBB);
  3227. // Insert IF cbranch.
  3228. m_pBuilder->SetInsertPoint(Scope.pPreScopeBB);
  3229. if (Scope.pElseBB != nullptr) {
  3230. m_pBuilder->CreateCondBr(Scope.pCond, Scope.pThenBB, Scope.pElseBB);
  3231. } else {
  3232. m_pBuilder->CreateCondBr(Scope.pCond, Scope.pThenBB, Scope.pPostScopeBB);
  3233. }
  3234. // Set endif BB as active.
  3235. pFunction->getBasicBlockList().push_back(Scope.pPostScopeBB);
  3236. m_pBuilder->SetInsertPoint(Scope.pPostScopeBB);
  3237. // Finish If-scope.
  3238. m_ScopeStack.Pop();
  3239. break;
  3240. }
  3241. case D3D10_SB_OPCODE_LOOP: {
  3242. DXASSERT_DXBC(Inst.m_NumOperands == 0);
  3243. // Create Loop-scope.
  3244. Scope &Scope = m_ScopeStack.Push(Scope::Loop, m_pBuilder->GetInsertBlock());
  3245. // Create Loop and EndLoop BBs.
  3246. Scope.pLoopBB = BasicBlock::Create(m_Ctx, Twine("loop") + Twine(Scope.NameIndex), pFunction);
  3247. Scope.pPostScopeBB = BasicBlock::Create(m_Ctx, Twine("loop") + Twine(Scope.NameIndex) + Twine(".end"));
  3248. // Insert branch to Loop BB.
  3249. m_pBuilder->CreateBr(Scope.pLoopBB);
  3250. // Set Loop BB as active.
  3251. m_pBuilder->SetInsertPoint(Scope.pLoopBB);
  3252. break;
  3253. }
  3254. case D3D10_SB_OPCODE_ENDLOOP: {
  3255. // Get Loop-scope.
  3256. Scope &Scope = m_ScopeStack.Top();
  3257. IFTBOOL(Scope.Kind == Scope::Loop, E_FAIL);
  3258. // Insert back-edge.
  3259. CreateBranchIfNeeded(m_pBuilder->GetInsertBlock(), Scope.pLoopBB);
  3260. // Set EndLoop BB as active.
  3261. pFunction->getBasicBlockList().push_back(Scope.pPostScopeBB);
  3262. m_pBuilder->SetInsertPoint(Scope.pPostScopeBB);
  3263. // Finish Loop-scope.
  3264. m_ScopeStack.Pop();
  3265. break;
  3266. }
  3267. case D3D10_SB_OPCODE_SWITCH: {
  3268. DXASSERT_DXBC(Inst.m_NumOperands == 1);
  3269. // Create Switch-scope.
  3270. Scope &Scope = m_ScopeStack.Push(Scope::Switch, m_pBuilder->GetInsertBlock());
  3271. // Prepare selector.
  3272. BYTE Comp = (BYTE)Inst.m_Operands[0].m_ComponentName;
  3273. CMask ReadMask = CMask::MakeCompMask(Comp);
  3274. OperandValue In1;
  3275. LoadOperand(In1, Inst, 0, ReadMask, CompType::getI32());
  3276. Scope.pSelector = In1[Comp];
  3277. // Create 1st casegroup BB and set it as active.
  3278. BasicBlock *pBB = BasicBlock::Create(m_Ctx, Twine("switch") + Twine(Scope.NameIndex) +
  3279. Twine(".casegroup") + Twine(Scope.CaseGroupIndex++), pFunction);
  3280. m_pBuilder->SetInsertPoint(pBB);
  3281. // Create endswitch BB.
  3282. Scope.pPostScopeBB = BasicBlock::Create(m_Ctx, Twine("switch") + Twine(Scope.NameIndex) + Twine(".end"));
  3283. break;
  3284. }
  3285. case D3D10_SB_OPCODE_CASE: {
  3286. DXASSERT_DXBC(Inst.m_NumOperands == 1);
  3287. // Get Switch-scope.
  3288. Scope &Scope = m_ScopeStack.Top();
  3289. IFTBOOL(Scope.Kind == Scope::Switch, E_FAIL);
  3290. // Retrieve selector value.
  3291. const D3D10ShaderBinary::COperandBase &O = Inst.m_Operands[0];
  3292. DXASSERT_DXBC(O.m_Type == D3D10_SB_OPERAND_TYPE_IMMEDIATE32 && O.m_NumComponents == D3D10_SB_OPERAND_1_COMPONENT);
  3293. int CaseValue = O.m_Value[0];
  3294. // Remember case clause.
  3295. pair<unsigned, BasicBlock*> Case(CaseValue, m_pBuilder->GetInsertBlock());
  3296. Scope.SwitchCases.emplace_back(Case);
  3297. break;
  3298. }
  3299. case D3D10_SB_OPCODE_DEFAULT: {
  3300. DXASSERT_DXBC(Inst.m_NumOperands == 0);
  3301. // Get Switch-scope.
  3302. Scope &Scope = m_ScopeStack.Top();
  3303. IFTBOOL(Scope.Kind == Scope::Switch, E_FAIL);
  3304. // Remember default clause.
  3305. Scope.pDefaultBB = m_pBuilder->GetInsertBlock();
  3306. break;
  3307. }
  3308. case D3D10_SB_OPCODE_ENDSWITCH: {
  3309. // Get Switch-scope.
  3310. Scope &Scope = m_ScopeStack.Top();
  3311. IFTBOOL(Scope.Kind == Scope::Switch, E_FAIL);
  3312. // Terminate case/default BB.
  3313. CreateBranchIfNeeded(m_pBuilder->GetInsertBlock(), Scope.pPostScopeBB);
  3314. // Insert switch branch.
  3315. m_pBuilder->SetInsertPoint(Scope.pPreScopeBB);
  3316. BasicBlock *pDefaultBB = Scope.pDefaultBB != nullptr ? Scope.pDefaultBB : Scope.pPostScopeBB;
  3317. SwitchInst *pSwitch = m_pBuilder->CreateSwitch(Scope.pSelector, pDefaultBB);
  3318. for (size_t i = 0; i < Scope.SwitchCases.size(); i++) {
  3319. auto &Case = Scope.SwitchCases[i];
  3320. if (Case.second == Scope.pDefaultBB) continue;
  3321. pSwitch->addCase(m_pBuilder->getInt32(Case.first), Case.second);
  3322. }
  3323. // Rename casegroups BBs.
  3324. SwitchInst *pSwI = dyn_cast<SwitchInst>(Scope.pPreScopeBB->getTerminator());
  3325. DXASSERT_NOMSG(pSwI != nullptr);
  3326. BasicBlock *pPrevCaseBB = nullptr;
  3327. unsigned CaseGroupIdx = 0;
  3328. for (auto itCase = pSwI->case_begin(), endCase = pSwI->case_end(); itCase != endCase; ++itCase) {
  3329. BasicBlock *pCaseBB = itCase.getCaseSuccessor();
  3330. if (pCaseBB != pPrevCaseBB) {
  3331. pCaseBB->setName(Twine("switch") + Twine(Scope.NameIndex) + Twine(".casegroup") + Twine(CaseGroupIdx++));
  3332. pPrevCaseBB = pCaseBB;
  3333. }
  3334. }
  3335. // Rename default BB.
  3336. if (Scope.pDefaultBB != nullptr) {
  3337. Scope.pDefaultBB->setName(Twine("switch") + Twine(Scope.NameIndex) + Twine(".default"));
  3338. }
  3339. // Set endswitch BB as active.
  3340. pFunction->getBasicBlockList().push_back(Scope.pPostScopeBB);
  3341. m_pBuilder->SetInsertPoint(Scope.pPostScopeBB);
  3342. // Finish Switch-scope.
  3343. m_ScopeStack.Pop();
  3344. break;
  3345. }
  3346. case D3D10_SB_OPCODE_CONTINUE: {
  3347. DXASSERT_DXBC(Inst.m_NumOperands == 0);
  3348. // Find parent scope.
  3349. Scope &Scope = m_ScopeStack.FindParentLoop();
  3350. // Create a new basic block.
  3351. BasicBlock *pNextBB = BasicBlock::Create(m_Ctx, Twine("loop") + Twine(Scope.NameIndex) +
  3352. Twine(".continue") + Twine(Scope.ContinueIndex++), pFunction);
  3353. // Insert branch to Loop BB.
  3354. m_pBuilder->CreateBr(Scope.pLoopBB);
  3355. // Set Next BB as active.
  3356. m_pBuilder->SetInsertPoint(pNextBB);
  3357. break;
  3358. }
  3359. case D3D10_SB_OPCODE_CONTINUEC: {
  3360. DXASSERT_DXBC(Inst.m_NumOperands == 1);
  3361. // Prepare condition.
  3362. Value *pCond = LoadZNZCondition(Inst, 0);
  3363. // Find parent scope.
  3364. Scope &Scope = m_ScopeStack.FindParentLoop();
  3365. // Create a new basic block.
  3366. BasicBlock *pNextBB = BasicBlock::Create(m_Ctx, Twine("loop") + Twine(Scope.NameIndex) +
  3367. Twine(".continuec") + Twine(Scope.ContinueIndex++), pFunction);
  3368. // Insert cbranch to Loop and Next BBs.
  3369. m_pBuilder->CreateCondBr(pCond, Scope.pLoopBB, pNextBB);
  3370. // Set Next BB as active.
  3371. m_pBuilder->SetInsertPoint(pNextBB);
  3372. break;
  3373. }
  3374. case D3D10_SB_OPCODE_BREAK: {
  3375. DXASSERT_DXBC(Inst.m_NumOperands == 0);
  3376. // Find parent scope.
  3377. Scope &Scope = m_ScopeStack.FindParentLoopOrSwitch();
  3378. // Create a new basic block.
  3379. BasicBlock *pNextBB;
  3380. if (Scope.Kind == Scope::Loop) {
  3381. pNextBB = BasicBlock::Create(m_Ctx, Twine("loop") + Twine(Scope.NameIndex) +
  3382. Twine(".break") + Twine(Scope.LoopBreakIndex++), pFunction);
  3383. } else {
  3384. if (m_ScopeStack.Top().Kind == Scope::Switch) {
  3385. pNextBB = BasicBlock::Create(m_Ctx, Twine("switch") + Twine(Scope.NameIndex) +
  3386. Twine(".tmpcasegroup") + Twine(Scope.CaseGroupIndex++), pFunction);
  3387. } else {
  3388. pNextBB = BasicBlock::Create(m_Ctx, Twine("switch") + Twine(Scope.NameIndex) +
  3389. Twine(".break") + Twine(Scope.SwitchBreakIndex++), pFunction);
  3390. }
  3391. }
  3392. // Insert branch to PostScope BB.
  3393. m_pBuilder->CreateBr(Scope.pPostScopeBB);
  3394. // Set Next BB as active.
  3395. m_pBuilder->SetInsertPoint(pNextBB);
  3396. break;
  3397. }
  3398. case D3D10_SB_OPCODE_BREAKC: {
  3399. DXASSERT_DXBC(Inst.m_NumOperands == 1);
  3400. // Prepare condition.
  3401. Value *pCond = LoadZNZCondition(Inst, 0);
  3402. // Find parent scope.
  3403. Scope &Scope = m_ScopeStack.FindParentLoopOrSwitch();
  3404. // Create a new basic block.
  3405. BasicBlock *pNextBB;
  3406. if (Scope.Kind == Scope::Loop) {
  3407. pNextBB = BasicBlock::Create(m_Ctx, Twine("loop") + Twine(Scope.NameIndex) +
  3408. Twine(".breakc") + Twine(Scope.LoopBreakIndex++), pFunction);
  3409. } else {
  3410. pNextBB = BasicBlock::Create(m_Ctx, Twine("switch") + Twine(Scope.NameIndex) +
  3411. Twine(".break") + Twine(Scope.SwitchBreakIndex++), pFunction);
  3412. }
  3413. // Insert cbranch to PostScope and Next BB.
  3414. m_pBuilder->CreateCondBr(pCond, Scope.pPostScopeBB, pNextBB);
  3415. // Set Next BB as active.
  3416. m_pBuilder->SetInsertPoint(pNextBB);
  3417. break;
  3418. }
  3419. case D3D10_SB_OPCODE_LABEL: {
  3420. DXASSERT_DXBC(Inst.m_NumOperands == 1);
  3421. DXASSERT_DXBC(Inst.m_Operands[0].m_Type == D3D10_SB_OPERAND_TYPE_LABEL ||
  3422. Inst.m_Operands[0].m_Type == D3D11_SB_OPERAND_TYPE_FUNCTION_BODY);
  3423. unsigned LabelIdx = Inst.m_Operands[0].m_Index[0].m_RegIndex;
  3424. const bool IsFb = Inst.m_Operands[0].m_Type == D3D11_SB_OPERAND_TYPE_FUNCTION_BODY;
  3425. auto &Label = IsFb ? m_InterfaceFunctionBodies[LabelIdx] : m_Labels[LabelIdx];
  3426. // Create entry basic block.
  3427. pFunction = Label.pFunc;
  3428. BasicBlock *pBB = BasicBlock::Create(m_Ctx, "entry", pFunction);
  3429. m_pBuilder->SetInsertPoint(pBB);
  3430. IFT(m_ScopeStack.IsEmpty());
  3431. (void)m_ScopeStack.Push(Scope::Function, nullptr);
  3432. InsertSM50ResourceHandles();
  3433. break;
  3434. }
  3435. case D3D10_SB_OPCODE_CALL: {
  3436. DXASSERT_DXBC(Inst.m_NumOperands == 1);
  3437. DXASSERT_DXBC(Inst.m_Operands[0].m_Type == D3D10_SB_OPERAND_TYPE_LABEL);
  3438. unsigned LabelIdx = Inst.m_Operands[0].m_Index[0].m_RegIndex;
  3439. auto &Label = m_Labels[LabelIdx];
  3440. // Create call instruction.
  3441. m_pBuilder->CreateCall(Label.pFunc);
  3442. break;
  3443. }
  3444. case D3D11_SB_OPCODE_INTERFACE_CALL: {
  3445. DXASSERT_DXBC(Inst.m_Operands[0].m_Type == D3D11_SB_OPERAND_TYPE_INTERFACE);
  3446. DXASSERT_DXBC(Inst.m_Operands[0].m_IndexDimension == D3D10_SB_OPERAND_INDEX_2D);
  3447. DXASSERT_DXBC(Inst.m_Operands[0].m_IndexType[0] == D3D10_SB_OPERAND_INDEX_IMMEDIATE32);
  3448. unsigned BaseIfaceIdx = Inst.m_Operands[0].m_Index[0].m_RegIndex;
  3449. unsigned CallSiteIdx = Inst.m_InterfaceCall.FunctionIndex;
  3450. Interface& Iface = m_Interfaces[BaseIfaceIdx];
  3451. DXASSERT_DXBC(Inst.m_Operands[0].m_IndexType[0] == D3D10_SB_OPERAND_INDEX_IMMEDIATE32 || Iface.bDynamicallyIndexed);
  3452. Value* pIfaceArrayIdx = LoadOperandIndex(Inst.m_Operands[0].m_Index[1], Inst.m_Operands[0].m_IndexType[1]);
  3453. Value* pIfaceIdx = m_pBuilder->CreateAdd(m_pOP->GetU32Const(BaseIfaceIdx), pIfaceArrayIdx);
  3454. // Load function table index
  3455. Value *pCBufferRetValue;
  3456. {
  3457. Value *Args[3];
  3458. Args[0] = m_pOP->GetU32Const((unsigned)OP::OpCode::CBufferLoadLegacy); // OpCode
  3459. Args[1] = CreateHandle(m_pInterfaceDataBuffer->GetClass(),
  3460. m_pInterfaceDataBuffer->GetID(),
  3461. m_pOP->GetU32Const(m_pInterfaceDataBuffer->GetLowerBound()),
  3462. false /*Nonuniform*/); // CBuffer handle
  3463. Args[2] = pIfaceIdx; // 0-based index into cbuffer instance
  3464. Function *pCBufferLoadFunc = m_pOP->GetOpFunc(OP::OpCode::CBufferLoadLegacy, Type::getInt32Ty(m_Ctx));
  3465. pCBufferRetValue = m_pBuilder->CreateCall(pCBufferLoadFunc, Args);
  3466. pCBufferRetValue = m_pBuilder->CreateExtractValue(pCBufferRetValue, 0);
  3467. }
  3468. // Switch on function table index
  3469. // Create endswitch BB.
  3470. BasicBlock* pPostSwitchBB = BasicBlock::Create(m_Ctx, Twine("fcall") + Twine(m_FcallCount) + Twine(".end"));
  3471. SwitchInst* pSwitch = m_pBuilder->CreateSwitch(pCBufferRetValue, pPostSwitchBB);
  3472. for (unsigned caseIdx = 0; caseIdx < Iface.Tables.size(); ++caseIdx) {
  3473. BasicBlock* pCaseBB = BasicBlock::Create(m_Ctx, Twine("fcall") + Twine(m_FcallCount) +
  3474. Twine(".case") + Twine(caseIdx), pFunction);
  3475. m_pBuilder->SetInsertPoint(pCaseBB);
  3476. unsigned fbIdx = m_FunctionTables[Iface.Tables[caseIdx]][CallSiteIdx];
  3477. m_pBuilder->CreateCall(m_InterfaceFunctionBodies[fbIdx].pFunc);
  3478. m_pBuilder->CreateBr(pPostSwitchBB);
  3479. pSwitch->addCase(m_pBuilder->getInt32(Iface.Tables[caseIdx]), pCaseBB);
  3480. }
  3481. pFunction->getBasicBlockList().push_back(pPostSwitchBB);
  3482. m_pBuilder->SetInsertPoint(pPostSwitchBB);
  3483. ++m_FcallCount;
  3484. break;
  3485. }
  3486. case D3D10_SB_OPCODE_CALLC: {
  3487. DXASSERT_DXBC(Inst.m_NumOperands == 2);
  3488. DXASSERT_DXBC(Inst.m_Operands[1].m_Type == D3D10_SB_OPERAND_TYPE_LABEL);
  3489. unsigned LabelIdx = Inst.m_Operands[1].m_Index[0].m_RegIndex;
  3490. auto &Label = m_Labels[LabelIdx];
  3491. // Prepare condition.
  3492. Value *pCond = LoadZNZCondition(Inst, 0);
  3493. // Create call and after-call BBs.
  3494. Function *pCurFunc = m_pBuilder->GetInsertBlock()->getParent();
  3495. BasicBlock *pCallBB = BasicBlock::Create(m_Ctx, Twine("label") + Twine(LabelIdx) + Twine(".callc"), pCurFunc);
  3496. BasicBlock *pPostCallBB = BasicBlock::Create(m_Ctx, Twine("label") + Twine(LabelIdx) + Twine(".callc"), pCurFunc);
  3497. // Create cbranch for callc.
  3498. m_pBuilder->CreateCondBr(pCond, pCallBB, pPostCallBB);
  3499. m_pBuilder->SetInsertPoint(pCallBB);
  3500. // Create call.
  3501. m_pBuilder->CreateCall(Label.pFunc);
  3502. m_pBuilder->CreateBr(pPostCallBB);
  3503. m_pBuilder->SetInsertPoint(pPostCallBB);
  3504. break;
  3505. }
  3506. case D3D10_SB_OPCODE_RET: {
  3507. // Find parent scope.
  3508. Scope &FuncScope = m_ScopeStack.FindParentFunction();
  3509. if ((FuncScope.IsEntry() && !m_bPatchConstantPhase) || !FuncScope.IsEntry()) {
  3510. m_pBuilder->CreateRetVoid();
  3511. BasicBlock *pAfterRet = BasicBlock::Create(m_Ctx, Twine("afterret"), pFunction);
  3512. m_pBuilder->SetInsertPoint(pAfterRet);
  3513. } else {
  3514. // Hull shader control point phase fork/join.
  3515. Scope &HullScope = m_ScopeStack.FindParentHullLoop();
  3516. BasicBlock *pAfterRet = BasicBlock::Create(m_Ctx, Twine("afterret"), pFunction);
  3517. if (m_ScopeStack.Top().Kind == Scope::HullLoop) {
  3518. bMustCloseHullLoop = true;
  3519. m_pBuilder->CreateBr(pAfterRet);
  3520. } else {
  3521. // A non-terminating return.
  3522. m_pBuilder->CreateBr(HullScope.pPostScopeBB);
  3523. }
  3524. m_pBuilder->SetInsertPoint(pAfterRet);
  3525. }
  3526. break;
  3527. }
  3528. case D3D10_SB_OPCODE_RETC: {
  3529. DXASSERT_DXBC(Inst.m_NumOperands == 1);
  3530. // Find parent scope.
  3531. Scope &FuncScope = m_ScopeStack.FindParentFunction();
  3532. // Prepare condition.
  3533. Value *pCond = LoadZNZCondition(Inst, 0);
  3534. if ((FuncScope.IsEntry() && !m_bPatchConstantPhase) || !FuncScope.IsEntry()) {
  3535. // Create retc and after-retc BB.
  3536. BasicBlock *pRetc = BasicBlock::Create(m_Ctx, Twine("label") + Twine(FuncScope.LabelIdx) +
  3537. Twine(".callc") + Twine(FuncScope.CallIdx) +
  3538. Twine(".retc") + Twine(FuncScope.ReturnIndex), pFunction);
  3539. BasicBlock *pAfterRetc = BasicBlock::Create(m_Ctx, Twine("label") + Twine(FuncScope.LabelIdx) +
  3540. Twine(".callc") + Twine(FuncScope.CallIdx) +
  3541. Twine(".afterretc") + Twine(FuncScope.ReturnIndex++), pFunction);
  3542. // Create cbranch for retc.
  3543. m_pBuilder->CreateCondBr(pCond, pRetc, pAfterRetc);
  3544. // Emit return.
  3545. m_pBuilder->SetInsertPoint(pRetc);
  3546. m_pBuilder->CreateRetVoid();
  3547. m_pBuilder->SetInsertPoint(pAfterRetc);
  3548. } else {
  3549. // Hull shader control point phase fork/join.
  3550. Scope &HullScope = m_ScopeStack.FindParentHullLoop();
  3551. // Create HullLoopBreak and AfterHullLoopBreak BB.
  3552. BasicBlock *pAfterHullBreakc = BasicBlock::Create(m_Ctx, Twine("hullloop") + Twine(FuncScope.NameIndex) +
  3553. Twine(".retc") + Twine(FuncScope.HullLoopBreakIndex) +
  3554. Twine(".afterretc"), pFunction);
  3555. // Create cbranch for retc (HullLoopBreak).
  3556. m_pBuilder->CreateCondBr(pCond, HullScope.pPostScopeBB, pAfterHullBreakc);
  3557. m_pBuilder->SetInsertPoint(pAfterHullBreakc);
  3558. }
  3559. break;
  3560. }
  3561. case D3D11_SB_OPCODE_HS_CONTROL_POINT_PHASE:
  3562. IFTBOOL(m_ScopeStack.FindParentFunction().IsEntry(), E_FAIL);
  3563. m_bControlPointPhase = true;
  3564. break;
  3565. case D3D11_SB_OPCODE_HS_FORK_PHASE:
  3566. case D3D11_SB_OPCODE_HS_JOIN_PHASE: {
  3567. if (!m_bPatchConstantPhase) {
  3568. if (!m_bControlPointPhase) {
  3569. // This is a pass-through CP HS.
  3570. bPasshThroughCP = true;
  3571. }
  3572. m_bControlPointPhase = false;
  3573. m_bPatchConstantPhase = true;
  3574. // Start patch constant function.
  3575. (void)m_ScopeStack.Push(Scope::Function, nullptr);
  3576. m_ScopeStack.Top().SetEntry(true);
  3577. pFunction = Function::Create(pEntryFuncType, GlobalValue::LinkageTypes::ExternalLinkage,
  3578. "pc_main", m_pModule.get());
  3579. pFunction->setCallingConv(CallingConv::C);
  3580. m_pPR->SetPatchConstantFunction(pFunction);
  3581. BasicBlock *pBB = BasicBlock::Create(m_Ctx, "entry", pFunction);
  3582. m_pBuilder->SetInsertPoint(pBB);
  3583. // Swap active x-registers.
  3584. m_IndexableRegs.swap(m_PatchConstantIndexableRegs);
  3585. DeclareIndexableRegisters();
  3586. // Create HullLoop induction variable.
  3587. pHullLoopInductionVar = m_pBuilder->CreateAlloca(Type::getInt32Ty(m_Ctx), nullptr, "InstanceID");
  3588. InsertSM50ResourceHandles();
  3589. }
  3590. // Create HullLoop-scope.
  3591. Scope &Scope = m_ScopeStack.Push(Scope::HullLoop, m_pBuilder->GetInsertBlock());
  3592. // Initialize HullLoop induction variable.
  3593. Scope.pInductionVar = pHullLoopInductionVar;
  3594. m_pBuilder->CreateStore(m_pOP->GetI32Const(0), Scope.pInductionVar);
  3595. Scope.HullLoopTripCount = m_PatchConstantPhaseInstanceCounts[ForkJoinPhaseIndex];
  3596. ForkJoinPhaseIndex++;
  3597. // Create HullLoop and EndHullLoop BBs.
  3598. Scope.pHullLoopBB = BasicBlock::Create(m_Ctx, Twine("hullloop") + Twine(Scope.NameIndex), pFunction);
  3599. Scope.pPostScopeBB = BasicBlock::Create(m_Ctx, Twine("hullloop") + Twine(Scope.NameIndex) + Twine(".end"));
  3600. // Insert branch to Loop BB.
  3601. m_pBuilder->CreateBr(Scope.pLoopBB);
  3602. // Set Loop BB as active.
  3603. m_pBuilder->SetInsertPoint(Scope.pLoopBB);
  3604. break;
  3605. }
  3606. case D3D11_SB_OPCODE_DCL_HS_FORK_PHASE_INSTANCE_COUNT:
  3607. case D3D11_SB_OPCODE_DCL_HS_JOIN_PHASE_INSTANCE_COUNT:
  3608. break;
  3609. //
  3610. // Pixel shader.
  3611. //
  3612. case D3D10_1_SB_OPCODE_LOD: {
  3613. OP::OpCode OpCode = OP::OpCode::CalculateLOD;
  3614. const unsigned uOpOutput = 0;
  3615. const unsigned uOpCoord = uOpOutput + 1;
  3616. const unsigned uOpSRV = uOpCoord + 1;
  3617. const unsigned uOpSampler = uOpSRV + 1;
  3618. DXASSERT_DXBC(Inst.m_Operands[uOpSRV].m_Type == D3D10_SB_OPERAND_TYPE_RESOURCE);
  3619. OperandValue InCoord, InSRV, InSampler;
  3620. // Resource.
  3621. const DxilResource &R = LoadSRVOperand(InSRV, Inst, uOpSRV, CMask::MakeXMask(), CompType::getInvalid());
  3622. // Coordinates.
  3623. CMask CoordMask = CMask::MakeFirstNCompMask(DXBC::GetNumResOffsets(R.GetKind()));
  3624. LoadOperand(InCoord, Inst, uOpCoord, CoordMask, CompType::getF32());
  3625. // Sampler.
  3626. LoadOperand(InSampler, Inst, uOpSampler, CMask::MakeXMask(), CompType::getInvalid());
  3627. // Create CalculateLOD call.
  3628. Value *Args[7];
  3629. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  3630. Args[1] = InSRV[0]; // Resource handle
  3631. Args[2] = InSampler[0]; // Sampler handle
  3632. Args[3] = CoordMask.IsSet(0) ? InCoord[0] : m_pUnusedF32;
  3633. Args[4] = CoordMask.IsSet(1) ? InCoord[1] : m_pUnusedF32;
  3634. Args[5] = CoordMask.IsSet(2) ? InCoord[2] : m_pUnusedF32;
  3635. CompType DstType = DXBC::GetCompTypeWithMinPrec(CompType::getF32(), Inst.m_Operands[uOpOutput].m_MinPrecision);
  3636. Type *pDstType = DstType.GetLLVMType(m_Ctx);
  3637. Function *F = m_pOP->GetOpFunc(OpCode, pDstType);
  3638. // Create unclamped CalculateLOD.
  3639. Args[6] = m_pOP->GetI1Const(false); // Unclamped
  3640. Value *pOpRetUnclamped = m_pBuilder->CreateCall(F, Args);
  3641. // Create clamped CalculateLOD.
  3642. Args[6] = m_pOP->GetI1Const(true); // Clamped
  3643. Value *pOpRetClamped = m_pBuilder->CreateCall(F, Args);
  3644. CMask OutputMask = CMask::FromDXBC(Inst.m_Operands[uOpOutput].m_WriteMask);
  3645. OperandValue Out;
  3646. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  3647. if (!OutputMask.IsSet(c)) continue;
  3648. // Respect swizzle: resource swizzle == return value swizzle.
  3649. BYTE Comp = Inst.m_Operands[uOpSRV].m_Swizzle[c];
  3650. switch (Comp) {
  3651. case 0: Out[c] = pOpRetClamped; break;
  3652. case 1: Out[c] = pOpRetUnclamped; break;
  3653. case 2: __fallthrough;
  3654. case 3: Out[c] = m_pOP->GetFloatConst(0.f); break;
  3655. default: DXASSERT_DXBC(false);
  3656. }
  3657. }
  3658. StoreOperand(Out, Inst, uOpOutput, OutputMask, DstType);
  3659. break;
  3660. }
  3661. case D3D10_SB_OPCODE_DISCARD: {
  3662. OP::OpCode OpCode = OP::OpCode::Discard;
  3663. Value *Args[2];
  3664. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  3665. Args[1] = LoadZNZCondition(Inst, 0); // Condition
  3666. Function *F = m_pOP->GetOpFunc(OpCode, Type::getVoidTy(m_Ctx));
  3667. MarkPrecise(m_pBuilder->CreateCall(F, Args));
  3668. break;
  3669. }
  3670. case D3D10_SB_OPCODE_DERIV_RTX: __fallthrough;
  3671. case D3D11_SB_OPCODE_DERIV_RTX_COARSE: ConvertUnary(OP::OpCode::DerivCoarseX, CompType::getF32(), Inst); break;
  3672. case D3D10_SB_OPCODE_DERIV_RTY: __fallthrough;
  3673. case D3D11_SB_OPCODE_DERIV_RTY_COARSE: ConvertUnary(OP::OpCode::DerivCoarseY, CompType::getF32(), Inst); break;
  3674. case D3D11_SB_OPCODE_DERIV_RTX_FINE: ConvertUnary(OP::OpCode::DerivFineX, CompType::getF32(), Inst); break;
  3675. case D3D11_SB_OPCODE_DERIV_RTY_FINE: ConvertUnary(OP::OpCode::DerivFineY, CompType::getF32(), Inst); break;
  3676. case D3D11_SB_OPCODE_EVAL_SNAPPED: {
  3677. OP::OpCode OpCode = OP::OpCode::EvalSnapped;
  3678. const unsigned uOpOutput = 0;
  3679. const unsigned uOpInput = uOpOutput + 1;
  3680. const unsigned uOpOffset = uOpInput + 1;
  3681. OperandValue InOffset;
  3682. CMask OutputMask = CMask::FromDXBC(Inst.m_Operands[uOpOutput].m_WriteMask);
  3683. LoadOperand(InOffset, Inst, uOpOffset, CMask::MakeFirstNCompMask(2), CompType::getI32());
  3684. const D3D10ShaderBinary::COperandBase &OpInput = Inst.m_Operands[uOpInput];
  3685. DXASSERT_NOMSG(Inst.m_Operands[uOpInput].m_IndexDimension == D3D10_SB_OPERAND_INDEX_1D);
  3686. unsigned Register = OpInput.m_Index[0].m_RegIndex;
  3687. Value *pRowIndexValue = LoadOperandIndex(OpInput.m_Index[0], OpInput.m_IndexType[0]);
  3688. CompType DstType = DXBC::GetCompTypeWithMinPrec(CompType::getF32(), Inst.m_Operands[uOpOutput].m_MinPrecision);
  3689. Type *pDstType = DstType.GetLLVMType(m_Ctx);
  3690. Function *F = m_pOP->GetOpFunc(OpCode, pDstType);
  3691. Value *Args[6];
  3692. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  3693. Args[4] = InOffset[0]; // Offset X
  3694. Args[5] = InOffset[1]; // Offset Y
  3695. OperandValue Out;
  3696. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  3697. if (!OutputMask.IsSet(c)) continue;
  3698. BYTE Comp = OpInput.m_Swizzle[c];
  3699. // Retrieve signature element.
  3700. const DxilSignatureElement *E = m_pInputSignature->GetElement(Register, Comp);
  3701. // Make row/col index relative within element.
  3702. Value *pRowIndexValueRel = m_pBuilder->CreateSub(pRowIndexValue, m_pOP->GetU32Const(E->GetStartRow()));
  3703. Args[1] = m_pOP->GetU32Const(E->GetID()); // Input signature element ID
  3704. Args[2] = pRowIndexValueRel; // Row, relative to the element
  3705. Args[3] = m_pOP->GetU8Const(Comp - E->GetStartCol()); // Col, relative to the element
  3706. Out[c] = m_pBuilder->CreateCall(F, Args);
  3707. }
  3708. StoreOperand(Out, Inst, uOpOutput, OutputMask, DstType);
  3709. break;
  3710. }
  3711. case D3D11_SB_OPCODE_EVAL_SAMPLE_INDEX: {
  3712. OP::OpCode OpCode = OP::OpCode::EvalSampleIndex;
  3713. const unsigned uOpOutput = 0;
  3714. const unsigned uOpInput = uOpOutput + 1;
  3715. const unsigned uOpSampleIndex = uOpInput + 1;
  3716. CMask OutputMask = CMask::FromDXBC(Inst.m_Operands[uOpOutput].m_WriteMask);
  3717. OperandValue InSampleIndex;
  3718. LoadOperand(InSampleIndex, Inst, uOpSampleIndex, CMask::MakeXMask(), CompType::getI32());
  3719. const D3D10ShaderBinary::COperandBase &OpInput = Inst.m_Operands[uOpInput];
  3720. DXASSERT_NOMSG(Inst.m_Operands[uOpInput].m_IndexDimension == D3D10_SB_OPERAND_INDEX_1D);
  3721. unsigned Register = OpInput.m_Index[0].m_RegIndex;
  3722. Value *pRowIndexValue = LoadOperandIndex(OpInput.m_Index[0], OpInput.m_IndexType[0]);
  3723. CompType DstType = DXBC::GetCompTypeWithMinPrec(CompType::getF32(), Inst.m_Operands[uOpOutput].m_MinPrecision);
  3724. Type *pDstType = DstType.GetLLVMType(m_Ctx);
  3725. Function *F = m_pOP->GetOpFunc(OpCode, pDstType);
  3726. Value *Args[5];
  3727. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  3728. Args[4] = InSampleIndex[0]; // Sample index
  3729. OperandValue Out;
  3730. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  3731. if (!OutputMask.IsSet(c)) continue;
  3732. BYTE Comp = OpInput.m_Swizzle[c];
  3733. // Retrieve signature element.
  3734. const DxilSignatureElement *E = m_pInputSignature->GetElement(Register, Comp);
  3735. // Make row/col index relative within element.
  3736. Value *pRowIndexValueRel = m_pBuilder->CreateSub(pRowIndexValue, m_pOP->GetU32Const(E->GetStartRow()));
  3737. Args[1] = m_pOP->GetU32Const(E->GetID()); // Input signature element ID
  3738. Args[2] = pRowIndexValueRel; // Row, relative to the element
  3739. Args[3] = m_pOP->GetU8Const(Comp - E->GetStartCol()); // Col, relative to the element
  3740. Out[c] = m_pBuilder->CreateCall(F, Args);
  3741. }
  3742. StoreOperand(Out, Inst, uOpOutput, OutputMask, DstType);
  3743. break;
  3744. }
  3745. case D3D11_SB_OPCODE_EVAL_CENTROID: {
  3746. OP::OpCode OpCode = OP::OpCode::EvalCentroid;
  3747. const unsigned uOpOutput = 0;
  3748. const unsigned uOpInput = uOpOutput + 1;
  3749. CMask OutputMask = CMask::FromDXBC(Inst.m_Operands[uOpOutput].m_WriteMask);
  3750. const D3D10ShaderBinary::COperandBase &OpInput = Inst.m_Operands[uOpInput];
  3751. DXASSERT_NOMSG(Inst.m_Operands[uOpInput].m_IndexDimension == D3D10_SB_OPERAND_INDEX_1D);
  3752. unsigned Register = OpInput.m_Index[0].m_RegIndex;
  3753. Value *pRowIndexValue = LoadOperandIndex(OpInput.m_Index[0], OpInput.m_IndexType[0]);
  3754. CompType DstType = DXBC::GetCompTypeWithMinPrec(CompType::getF32(), Inst.m_Operands[uOpOutput].m_MinPrecision);
  3755. Type *pDstType = DstType.GetLLVMType(m_Ctx);
  3756. Function *F = m_pOP->GetOpFunc(OpCode, pDstType);
  3757. Value *Args[4];
  3758. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  3759. OperandValue Out;
  3760. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  3761. if (!OutputMask.IsSet(c)) continue;
  3762. BYTE Comp = OpInput.m_Swizzle[c];
  3763. // Retrieve signature element.
  3764. const DxilSignatureElement *E = m_pInputSignature->GetElement(Register, Comp);
  3765. // Make row/col index relative within element.
  3766. Value *pRowIndexValueRel = m_pBuilder->CreateSub(pRowIndexValue, m_pOP->GetU32Const(E->GetStartRow()));
  3767. Args[1] = m_pOP->GetU32Const(E->GetID()); // Input signature element ID
  3768. Args[2] = pRowIndexValueRel; // Row, relative to the element
  3769. Args[3] = m_pOP->GetU8Const(Comp - E->GetStartCol()); // Col, relative to the element
  3770. Out[c] = m_pBuilder->CreateCall(F, Args);
  3771. }
  3772. StoreOperand(Out, Inst, uOpOutput, OutputMask, DstType);
  3773. break;
  3774. }
  3775. case D3D10_SB_OPCODE_EMIT:
  3776. case D3D11_SB_OPCODE_EMIT_STREAM: {
  3777. OP::OpCode OpCode = OP::OpCode::EmitStream;
  3778. BYTE StreamId = 0;
  3779. if (Inst.OpCode() == D3D11_SB_OPCODE_EMIT_STREAM) {
  3780. StreamId = (BYTE)Inst.m_Operands[0].m_Index[0].m_RegIndex;
  3781. }
  3782. // For GS with multiple streams, capture the values of output registers at the emit point.
  3783. if (m_pPR->HasMultipleOutputStreams()) {
  3784. EmitGSOutputRegisterStore(StreamId);
  3785. }
  3786. // Create EmitStream call.
  3787. Value *Args[2];
  3788. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  3789. Args[1] = m_pOP->GetU8Const(StreamId); // Stream ID
  3790. Function *F = m_pOP->GetOpFunc(OpCode, Type::getVoidTy(m_Ctx));
  3791. MarkPrecise(m_pBuilder->CreateCall(F, Args));
  3792. break;
  3793. }
  3794. case D3D10_SB_OPCODE_CUT:
  3795. case D3D11_SB_OPCODE_CUT_STREAM: {
  3796. OP::OpCode OpCode = OP::OpCode::CutStream;
  3797. BYTE StreamId = 0;
  3798. if (Inst.OpCode() == D3D11_SB_OPCODE_CUT_STREAM) {
  3799. StreamId = (BYTE)Inst.m_Operands[0].m_Index[0].m_RegIndex;
  3800. }
  3801. // Create CutStream call.
  3802. Value *Args[2];
  3803. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  3804. Args[1] = m_pOP->GetU8Const(StreamId); // Stream ID
  3805. Function *F = m_pOP->GetOpFunc(OpCode, Type::getVoidTy(m_Ctx));
  3806. MarkPrecise(m_pBuilder->CreateCall(F, Args));
  3807. break;
  3808. }
  3809. case D3D10_SB_OPCODE_EMITTHENCUT:
  3810. case D3D11_SB_OPCODE_EMITTHENCUT_STREAM: {
  3811. OP::OpCode OpCode = OP::OpCode::EmitThenCutStream;
  3812. BYTE StreamId = 0;
  3813. if (Inst.OpCode() == D3D11_SB_OPCODE_EMITTHENCUT_STREAM) {
  3814. StreamId = (BYTE)Inst.m_Operands[0].m_Index[0].m_RegIndex;
  3815. }
  3816. // For GS with multiple streams, capture the values of output registers at the emit point.
  3817. if (m_pPR->HasMultipleOutputStreams()) {
  3818. EmitGSOutputRegisterStore(StreamId);
  3819. }
  3820. // Create EmitThenCutStream call.
  3821. Value *Args[2];
  3822. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  3823. Args[1] = m_pOP->GetU8Const(StreamId); // Stream ID
  3824. Function *F = m_pOP->GetOpFunc(OpCode, Type::getVoidTy(m_Ctx));
  3825. MarkPrecise(m_pBuilder->CreateCall(F, Args));
  3826. break;
  3827. }
  3828. case D3D10_SB_OPCODE_NOP:
  3829. break;
  3830. default:
  3831. HandleUnknownInstruction(Inst);
  3832. break;
  3833. }
  3834. }
  3835. DXASSERT_NOMSG(m_ScopeStack.IsEmpty());
  3836. if (bPasshThroughCP) {
  3837. Function *Entry = m_pPR->GetEntryFunction();
  3838. m_pPR->SetEntryFunction(nullptr);
  3839. Entry->eraseFromParent();
  3840. m_pPR->SetEntryFunctionName("");
  3841. }
  3842. CleanupIndexableRegisterDecls(m_IndexableRegs);
  3843. CleanupIndexableRegisterDecls(m_PatchConstantIndexableRegs);
  3844. RemoveUnreachableBasicBlocks();
  3845. CleanupGEP();
  3846. }
  3847. void DxbcConverter::LogConvertResult(bool InDriver, _In_ const LARGE_INTEGER *pQPCConvertStart,
  3848. _In_ const LARGE_INTEGER *pQPCConvertEnd, _In_reads_bytes_(DxbcSize) LPCVOID pDxbc, _In_ UINT32 DxbcSize,
  3849. _In_opt_z_ LPCWSTR pExtraOptions, _In_reads_bytes_(ConvertedSize) LPCVOID pConverted, _In_opt_ UINT32 ConvertedSize,
  3850. HRESULT hr) {
  3851. // intentionaly empty - override to report conversion results
  3852. }
  3853. HRESULT DxbcConverter::PreConvertHook(const CShaderToken *pByteCode) {
  3854. return S_OK;
  3855. }
  3856. HRESULT DxbcConverter::PostConvertHook(const CShaderToken *pByteCode) {
  3857. return S_OK;
  3858. }
  3859. void DxbcConverter::HandleUnknownInstruction(D3D10ShaderBinary::CInstruction &Inst) {
  3860. DXASSERT_ARGS(false, "OpCode %u is not yet implemented", Inst.OpCode());
  3861. }
  3862. unsigned DxbcConverter::GetResourceSlot(D3D10ShaderBinary::CInstruction &Inst) {
  3863. return DXBC::GetResourceSlot(Inst.OpCode());
  3864. }
  3865. void DxbcConverter::AdvanceDxbcInstructionStream(D3D10ShaderBinary::CShaderCodeParser &Parser,
  3866. D3D10ShaderBinary::CInstruction &Inst,
  3867. bool &bDoneParsing) {
  3868. if (bDoneParsing)
  3869. return;
  3870. if (!Parser.EndOfShader()) {
  3871. DXASSERT_NOMSG(!bDoneParsing);
  3872. Parser.ParseInstruction(&Inst);
  3873. } else {
  3874. IFTBOOL(!bDoneParsing, E_FAIL);
  3875. bDoneParsing = true;
  3876. }
  3877. }
  3878. bool DxbcConverter::GetNextDxbcInstruction(D3D10ShaderBinary::CShaderCodeParser &Parser,
  3879. D3D10ShaderBinary::CInstruction &NextInst) {
  3880. if (Parser.EndOfShader()) {
  3881. return false;
  3882. }
  3883. UINT CurPos = Parser.CurrentTokenOffset();
  3884. Parser.ParseInstruction(&NextInst);
  3885. Parser.SetCurrentTokenOffset(CurPos);
  3886. return true;
  3887. }
  3888. void DxbcConverter::InsertSM50ResourceHandles() {
  3889. // Create resource handles for SM5.0- to reduce the number of call instructions (to reduce IR size).
  3890. // Later: it may be worthwhile to implement a pass to hoist handle creation for SM5.1 here when the index into range is constant and used more than once within the shader.
  3891. if (!IsSM51Plus()) {
  3892. for (size_t i = 0; i < m_pPR->GetSRVs().size(); ++i) {
  3893. DxilResource &R = m_pPR->GetSRV(i);
  3894. SetCachedHandle(R);
  3895. }
  3896. for (size_t i = 0; i < m_pPR->GetUAVs().size(); ++i) {
  3897. DxilResource &R = m_pPR->GetUAV(i);
  3898. SetCachedHandle(R);
  3899. }
  3900. for (size_t i = 0; i < m_pPR->GetCBuffers().size(); ++i) {
  3901. DxilCBuffer &R = m_pPR->GetCBuffer(i);
  3902. SetCachedHandle(R);
  3903. }
  3904. for (size_t i = 0; i < m_pPR->GetSamplers().size(); ++i) {
  3905. DxilSampler &R = m_pPR->GetSampler(i);
  3906. SetCachedHandle(R);
  3907. }
  3908. }
  3909. }
  3910. void DxbcConverter::InsertInterfacesResourceDecls() {
  3911. // Insert decls for:
  3912. // 1. CB14 containing interface table selections, along with "this pointer" information,
  3913. // 2. 14 CBs in space 1,
  3914. // 3. 32 samplers in space 1 and 32 comparison samplers in space 2
  3915. // SRVs will be inserted dynamically as needed
  3916. if (m_pInterfaceDataBuffer) {
  3917. return;
  3918. }
  3919. m_pPR->m_ShaderFlags.SetAllResourcesBound(false);
  3920. // Create interface data buffer
  3921. {
  3922. unsigned ID = m_pPR->AddCBuffer(unique_ptr<DxilCBuffer>(new DxilCBuffer));
  3923. DxilCBuffer &R = m_pPR->GetCBuffer(ID); // R == record
  3924. m_pInterfaceDataBuffer = &R;
  3925. R.SetID(ID);
  3926. // Root signature bindings.
  3927. unsigned CBufferSize = D3D11_SHADER_MAX_INTERFACES * 8 /*UINTs per interface*/ * sizeof(UINT);
  3928. R.SetLowerBound(D3D11_COMMONSHADER_CONSTANT_BUFFER_API_SLOT_COUNT); // 14
  3929. R.SetRangeSize(1);
  3930. R.SetSpaceID(0);
  3931. // Declare global variable.
  3932. R.SetGlobalName(SynthesizeResGVName("CB", R.GetID()));
  3933. StructType *pResType = GetStructResElemType(CBufferSize);
  3934. R.SetGlobalSymbol(DeclareUndefPtr(pResType, DXIL::kCBufferAddrSpace));
  3935. // CBuffer-specific state.
  3936. R.SetSize(CBufferSize);
  3937. }
  3938. // Create CB array for class instances
  3939. {
  3940. unsigned ID = m_pPR->AddCBuffer(unique_ptr<DxilCBuffer>(new DxilCBuffer));
  3941. DxilCBuffer &R = m_pPR->GetCBuffer(ID); // R == record
  3942. m_pClassInstanceCBuffers = &R;
  3943. R.SetID(ID);
  3944. // Root signature bindings.
  3945. unsigned CBufferSize = DXIL::kMaxCBufferSize * DXBC::kWidth * 4;
  3946. R.SetLowerBound(0);
  3947. R.SetRangeSize(D3D11_COMMONSHADER_CONSTANT_BUFFER_API_SLOT_COUNT); // 14
  3948. R.SetSpaceID(1);
  3949. // Declare global variable.
  3950. R.SetGlobalName(SynthesizeResGVName("CB", R.GetID()));
  3951. StructType *pResType = GetStructResElemType(CBufferSize);
  3952. R.SetGlobalSymbol(DeclareUndefPtr(pResType, DXIL::kCBufferAddrSpace));
  3953. // CBuffer-specific state.
  3954. R.SetSize(CBufferSize);
  3955. }
  3956. // Create sampler arrays for class instances
  3957. for (unsigned i = 0; i < 2; ++i) {
  3958. unsigned ID = m_pPR->AddSampler(unique_ptr<DxilSampler>(new DxilSampler));
  3959. DxilSampler &R = m_pPR->GetSampler(ID); // R == record
  3960. R.SetID(ID);
  3961. // Root signature bindings.
  3962. R.SetLowerBound(0);
  3963. R.SetRangeSize(D3D11_COMMONSHADER_SAMPLER_SLOT_COUNT);
  3964. R.SetSpaceID(i + 1);
  3965. // Declare global variable.
  3966. R.SetGlobalName(SynthesizeResGVName("S", R.GetID()));
  3967. string ResTypeName("dx.types.Sampler");
  3968. StructType *pResType = m_pModule->getTypeByName(ResTypeName);
  3969. if (pResType == nullptr) {
  3970. pResType = StructType::create(m_Ctx, ResTypeName);
  3971. }
  3972. R.SetGlobalSymbol(DeclareUndefPtr(pResType, DXIL::kDeviceMemoryAddrSpace));
  3973. // Sampler-specific state.
  3974. R.SetSamplerKind(i == 0 ? DXIL::SamplerKind::Default : DXIL::SamplerKind::Comparison);
  3975. DxilSampler*& pSampler = (i == 0 ? m_pClassInstanceSamplers : m_pClassInstanceComparisonSamplers);
  3976. pSampler = &R;
  3977. }
  3978. }
  3979. const DxilResource& DxbcConverter::GetInterfacesSRVDecl(D3D10ShaderBinary::CInstruction &Inst) {
  3980. InterfaceShaderResourceKey Key = {};
  3981. DXASSERT_DXBC(Inst.m_ExtendedOpCodeCount == 2); // Extended resource dimension and return type
  3982. Key.Kind = DXBC::GetResourceKind(Inst.m_ResourceDimEx);
  3983. if (Inst.m_ResourceDimEx == D3D11_SB_RESOURCE_DIMENSION_STRUCTURED_BUFFER) {
  3984. Key.StructureByteStride = Inst.m_ResourceDimStructureStrideEx;
  3985. }
  3986. else if (Inst.m_ResourceDimEx != D3D11_SB_RESOURCE_DIMENSION_RAW_BUFFER) {
  3987. Key.TypedSRVRet = DXBC::GetDeclResCompType(Inst.m_ResourceReturnTypeEx[0]).GetKind();
  3988. }
  3989. auto iter = m_ClassInstanceSRVs.find(Key);
  3990. if (iter != m_ClassInstanceSRVs.end()) {
  3991. return m_pPR->GetSRV(iter->second);
  3992. }
  3993. unsigned ID = m_pPR->AddSRV(unique_ptr<DxilResource>(new DxilResource));
  3994. DxilResource &R = m_pPR->GetSRV(ID); // R == record
  3995. R.SetID(ID);
  3996. R.SetRW(false);
  3997. // Root signature bindings.
  3998. R.SetLowerBound(0);
  3999. R.SetRangeSize(D3D11_COMMONSHADER_INPUT_RESOURCE_SLOT_COUNT);
  4000. R.SetSpaceID(m_ClassInstanceSRVs.size() + 1);
  4001. unsigned SampleCount =
  4002. (Key.Kind == DXIL::ResourceKind::Texture2DMS ||
  4003. Key.Kind == DXIL::ResourceKind::Texture2DMSArray) ? 4 : 0;
  4004. DXASSERT_DXBC(SampleCount == 0); // Don't expect to actually see this used within interfaces...
  4005. // Resource-specific state.
  4006. StructType *pResType = nullptr;
  4007. switch (Inst.m_ResourceDimEx) {
  4008. default: {
  4009. R.SetKind(DXBC::GetResourceKind(Inst.m_ResourceDimEx));
  4010. const unsigned kTypedBufferElementSizeInBytes = 4;
  4011. R.SetElementStride(kTypedBufferElementSizeInBytes);
  4012. R.SetSampleCount(SampleCount);
  4013. CompType DeclCT = DXBC::GetDeclResCompType(Inst.m_ResourceReturnTypeEx[0]);
  4014. if (DeclCT.IsInvalid()) DeclCT = CompType::getU32();
  4015. R.SetCompType(DeclCT);
  4016. pResType = GetTypedResElemType(DeclCT);
  4017. break;
  4018. }
  4019. case D3D11_SB_RESOURCE_DIMENSION_RAW_BUFFER: {
  4020. R.SetKind(DxilResource::Kind::RawBuffer);
  4021. const unsigned kRawBufferElementSizeInBytes = 1;
  4022. R.SetElementStride(kRawBufferElementSizeInBytes);
  4023. pResType = GetTypedResElemType(CompType::getU32());
  4024. break;
  4025. }
  4026. case D3D11_SB_RESOURCE_DIMENSION_STRUCTURED_BUFFER: {
  4027. R.SetKind(DxilResource::Kind::StructuredBuffer);
  4028. unsigned Stride = Inst.m_ResourceDimStructureStrideEx;
  4029. R.SetElementStride(Stride);
  4030. pResType = GetStructResElemType(Stride);
  4031. break;
  4032. }
  4033. }
  4034. // Declare global variable.
  4035. R.SetGlobalName(SynthesizeResGVName("T", R.GetID()));
  4036. R.SetGlobalSymbol(DeclareUndefPtr(pResType, DXIL::kDeviceMemoryAddrSpace));
  4037. m_ClassInstanceSRVs[Key] = ID;
  4038. return R;
  4039. }
  4040. void DxbcConverter::DeclareIndexableRegisters() {
  4041. // Reserve storage for x-registers.
  4042. if (!HasLabels()) {
  4043. // Only main subroutine: use alloca, as optimization.
  4044. for (auto &IR : m_IndexableRegs) {
  4045. DXASSERT_NOMSG(IR.second.pValue32 == nullptr && IR.second.pValue16 == nullptr);
  4046. Type *pType32 = ArrayType::get(Type::getFloatTy(m_Ctx), IR.second.NumRegs * IR.second.NumComps);
  4047. AllocaInst *pAlloca32 = m_pBuilder->CreateAlloca(pType32, nullptr, Twine("dx.v32.x") + Twine(IR.first));
  4048. pAlloca32->setAlignment(kRegCompAlignment);
  4049. IR.second.pValue32 = pAlloca32;
  4050. Type *pType16 = ArrayType::get(Type::getHalfTy(m_Ctx), IR.second.NumRegs * IR.second.NumComps);
  4051. AllocaInst *pAlloca16 = m_pBuilder->CreateAlloca(pType16, nullptr, Twine("dx.v16.x") + Twine(IR.first));
  4052. pAlloca16->setAlignment(kRegCompAlignment);
  4053. IR.second.pValue16 = pAlloca16;
  4054. IR.second.bIsAlloca = true;
  4055. }
  4056. } else {
  4057. // Several subroutines: use global storage.
  4058. for (auto &IR : m_IndexableRegs) {
  4059. Type *pType32 = ArrayType::get(Type::getFloatTy(m_Ctx), IR.second.NumRegs * IR.second.NumComps);
  4060. GlobalVariable *pGV32 = new GlobalVariable(*m_pModule, pType32,
  4061. false, GlobalValue::InternalLinkage,
  4062. UndefValue::get(pType32),
  4063. Twine("dx.v32.x") + Twine(IR.first), nullptr,
  4064. GlobalVariable::NotThreadLocal, DXIL::kDefaultAddrSpace);
  4065. IR.second.pValue32 = pGV32;
  4066. Type *pType16 = ArrayType::get(Type::getHalfTy(m_Ctx), IR.second.NumRegs * IR.second.NumComps);
  4067. GlobalVariable *pGV16 = new GlobalVariable(*m_pModule, pType16,
  4068. false, GlobalValue::InternalLinkage,
  4069. UndefValue::get(pType16),
  4070. Twine("dx.v16.x") + Twine(IR.first), nullptr,
  4071. GlobalVariable::NotThreadLocal, DXIL::kDefaultAddrSpace);
  4072. IR.second.pValue16 = pGV16;
  4073. IR.second.bIsAlloca = false;
  4074. }
  4075. }
  4076. }
  4077. void DxbcConverter::CleanupIndexableRegisterDecls(map<unsigned, IndexableReg> &IdxRegMap) {
  4078. for (auto &IR : IdxRegMap) {
  4079. if (IR.second.pValue32 && !IR.second.pValue32->hasNUsesOrMore(1)) {
  4080. if (IR.second.bIsAlloca)
  4081. cast<Instruction>(IR.second.pValue32)->eraseFromParent();
  4082. else
  4083. cast<GlobalVariable>(IR.second.pValue32)->eraseFromParent();
  4084. }
  4085. if (IR.second.pValue16 && !IR.second.pValue16->hasNUsesOrMore(1)) {
  4086. if (IR.second.bIsAlloca)
  4087. cast<Instruction>(IR.second.pValue16)->eraseFromParent();
  4088. else
  4089. cast<GlobalVariable>(IR.second.pValue16)->eraseFromParent();
  4090. }
  4091. }
  4092. }
  4093. void DxbcConverter::RemoveUnreachableBasicBlocks() {
  4094. for (auto itFn = m_pModule->begin(), endFn = m_pModule->end(); itFn != endFn; ++itFn) {
  4095. Function *F = itFn;
  4096. vector<BasicBlock *> NoPredSet;
  4097. // 1. Detect basic blocks without predecessors.
  4098. for (auto itBB = ++(F->begin()), endBB = F->end(); itBB != endBB; ++itBB) {
  4099. BasicBlock *B = itBB;
  4100. if (pred_begin(B) == pred_end(B)) {
  4101. NoPredSet.emplace_back(B);
  4102. }
  4103. }
  4104. // 2. Remove BBs with no predecessors.
  4105. while (!NoPredSet.empty()) {
  4106. BasicBlock *B = NoPredSet.back();
  4107. NoPredSet.pop_back();
  4108. TerminatorInst *pTI = B->getTerminator();
  4109. vector<BasicBlock*> Successors(pTI->getNumSuccessors());
  4110. for (unsigned i = 0; i < pTI->getNumSuccessors(); i++) {
  4111. Successors[i] = pTI->getSuccessor(i);
  4112. }
  4113. B->eraseFromParent();
  4114. for (auto S : Successors) {
  4115. if (pred_begin(S) == pred_end(S)) {
  4116. NoPredSet.emplace_back(S);
  4117. }
  4118. }
  4119. }
  4120. }
  4121. }
  4122. class GEPVisitor : public InstVisitor<GEPVisitor> {
  4123. public:
  4124. void visitInstruction(Instruction &I) {
  4125. for (Instruction::op_iterator itOp = I.op_begin(), endOp = I.op_end(); itOp != endOp; ++itOp) {
  4126. Value *V1 = itOp->get()->stripPointerCasts();
  4127. if (GEPOperator *pGEP = dyn_cast<GEPOperator>(V1)) {
  4128. bool bReplace = false;
  4129. SmallVector<Value *, 4> GEPIndices;
  4130. for (GEPOperator::op_iterator itOp = pGEP->idx_begin(), endOp = pGEP->idx_end(); itOp != endOp; ++itOp) {
  4131. Value *V = itOp->get();
  4132. GEPIndices.push_back(V);
  4133. if (ConstantInt *C = dyn_cast<ConstantInt>(V)) {
  4134. LLVMContext &Ctx = C->getContext();
  4135. if (C->getType() != Type::getInt32Ty(Ctx)) {
  4136. uint64_t n = C->getZExtValue();
  4137. if (n <= (uint64_t)(UINT32_MAX)) {
  4138. GEPIndices.back() = Constant::getIntegerValue(IntegerType::get(Ctx, 32), APInt(32, (unsigned)n));
  4139. bReplace = true;
  4140. }
  4141. }
  4142. }
  4143. }
  4144. if (bReplace) {
  4145. Constant *pGEP2 = ConstantExpr::getGetElementPtr(pGEP->getPointerOperandType()->getPointerElementType(),
  4146. dyn_cast<Constant>(pGEP->getPointerOperand()),
  4147. GEPIndices);
  4148. pGEP->replaceAllUsesWith(pGEP2);
  4149. }
  4150. }
  4151. }
  4152. }
  4153. };
  4154. // GEPOperators may get i64 constant index values.
  4155. // We replace them here with i32 values, if possible, to avoid 64-bit values in DXIL.
  4156. void DxbcConverter::CleanupGEP() {
  4157. GEPVisitor a;
  4158. a.visit(*m_pModule);
  4159. }
  4160. void DxbcConverter::ConvertUnary(OP::OpCode OpCode,
  4161. const CompType &ElementType,
  4162. D3D10ShaderBinary::CInstruction &Inst,
  4163. const unsigned DstIdx,
  4164. const unsigned SrcIdx) {
  4165. DXASSERT_NOMSG(OP::GetOpCodeClass(OpCode) == OP::OpCodeClass::Unary ||
  4166. OP::GetOpCodeClass(OpCode) == OP::OpCodeClass::UnaryBits);
  4167. CMask WriteMask = CMask::FromDXBC(Inst.m_Operands[DstIdx].m_WriteMask);
  4168. CompType OperationType = DXBC::GetCompTypeWithMinPrec(ElementType, Inst.m_Operands[DstIdx].m_MinPrecision);
  4169. Type *pOperationType = OperationType.GetLLVMType(m_Ctx);
  4170. Function *pFunc = m_pOP->GetOpFunc(OpCode, pOperationType);
  4171. OperandValue In, Out;
  4172. LoadOperand(In, Inst, SrcIdx, WriteMask, OperationType);
  4173. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  4174. if (!WriteMask.IsSet(c)) continue;
  4175. Out[c] = m_pBuilder->CreateCall(pFunc, { m_pOP->GetU32Const((unsigned)OpCode), In[c] });
  4176. }
  4177. StoreOperand(Out, Inst, DstIdx, WriteMask, OperationType);
  4178. }
  4179. void DxbcConverter::ConvertBinary(OP::OpCode OpCode,
  4180. const CompType &ElementType,
  4181. D3D10ShaderBinary::CInstruction &Inst,
  4182. const unsigned DstIdx,
  4183. const unsigned SrcIdx1,
  4184. const unsigned SrcIdx2) {
  4185. DXASSERT_NOMSG(OP::GetOpCodeClass(OpCode) == OP::OpCodeClass::Binary);
  4186. CMask WriteMask = CMask::FromDXBC(Inst.m_Operands[DstIdx].m_WriteMask);
  4187. CompType OperationType = DXBC::GetCompTypeWithMinPrec(ElementType, Inst.m_Operands[DstIdx].m_MinPrecision);
  4188. Type *pOperationType = OperationType.GetLLVMType(m_Ctx);
  4189. Function *pFunc = m_pOP->GetOpFunc(OpCode, pOperationType);
  4190. OperandValue In1, In2, Out;
  4191. LoadOperand(In1, Inst, SrcIdx1, WriteMask, OperationType);
  4192. LoadOperand(In2, Inst, SrcIdx2, WriteMask, OperationType);
  4193. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  4194. if (!WriteMask.IsSet(c)) continue;
  4195. Out[c] = m_pBuilder->CreateCall(pFunc, { m_pOP->GetU32Const((unsigned)OpCode), In1[c], In2[c] });
  4196. if (ElementType.GetKind() == CompType::Kind::F64) {
  4197. c++;
  4198. }
  4199. }
  4200. StoreOperand(Out, Inst, DstIdx, WriteMask, OperationType);
  4201. }
  4202. void DxbcConverter::ConvertBinary(Instruction::BinaryOps OpCode,
  4203. const CompType &ElementType,
  4204. D3D10ShaderBinary::CInstruction &Inst,
  4205. const unsigned DstIdx,
  4206. const unsigned SrcIdx1,
  4207. const unsigned SrcIdx2) {
  4208. CMask WriteMask = CMask::FromDXBC(Inst.m_Operands[DstIdx].m_WriteMask);
  4209. CompType OperationType = DXBC::GetCompTypeWithMinPrec(ElementType, Inst.m_Operands[DstIdx].m_MinPrecision);
  4210. OperandValue In1, In2, Out;
  4211. LoadOperand(In1, Inst, SrcIdx1, WriteMask, OperationType);
  4212. LoadOperand(In2, Inst, SrcIdx2, WriteMask, OperationType);
  4213. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  4214. if (!WriteMask.IsSet(c)) continue;
  4215. Value *pVal2 = In2[c];
  4216. // Limit shift amount to 5 bits.
  4217. switch (OpCode) {
  4218. case Instruction::Shl:
  4219. case Instruction::AShr:
  4220. case Instruction::LShr:
  4221. pVal2 = m_pBuilder->CreateAnd(pVal2, 0x0000001F);
  4222. }
  4223. Out[c] = m_pBuilder->CreateBinOp(OpCode, In1[c], pVal2);
  4224. if (ElementType.GetKind() == CompType::Kind::F64) {
  4225. c++;
  4226. }
  4227. }
  4228. StoreOperand(Out, Inst, DstIdx, WriteMask, OperationType);
  4229. }
  4230. void DxbcConverter::ConvertBinaryWithTwoOuts(OP::OpCode OpCode,
  4231. D3D10ShaderBinary::CInstruction &Inst,
  4232. const unsigned DstIdx1, const unsigned DstIdx2,
  4233. const unsigned SrcIdx1, const unsigned SrcIdx2) {
  4234. DXASSERT_NOMSG(OP::GetOpCodeClass(OpCode) == OP::OpCodeClass::BinaryWithTwoOuts);
  4235. CMask WriteMask = CMask::FromDXBC(Inst.m_Operands[DstIdx1].m_WriteMask | Inst.m_Operands[DstIdx2].m_WriteMask);
  4236. if (WriteMask.ToByte() == 0) {
  4237. // No-op if both destinations are null
  4238. DXASSERT_NOMSG(Inst.m_Operands[DstIdx1].m_Type == D3D10_SB_OPERAND_TYPE_NULL &&
  4239. Inst.m_Operands[DstIdx2].m_Type == D3D10_SB_OPERAND_TYPE_NULL);
  4240. return;
  4241. }
  4242. CMask Dst1Mask = CMask::FromDXBC(Inst.m_Operands[DstIdx1].m_WriteMask);
  4243. CMask Dst2Mask = CMask::FromDXBC(Inst.m_Operands[DstIdx2].m_WriteMask);
  4244. CompType OperationType = CompType::getI32();
  4245. Type *pOperationType = OperationType.GetLLVMType(m_Ctx);
  4246. Function *pFunc = m_pOP->GetOpFunc(OpCode, pOperationType);
  4247. OperandValue In1, In2, Out1, Out2;
  4248. LoadOperand(In1, Inst, SrcIdx1, WriteMask, OperationType);
  4249. LoadOperand(In2, Inst, SrcIdx2, WriteMask, OperationType);
  4250. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  4251. if (!WriteMask.IsSet(c)) continue;
  4252. Value *pRes = m_pBuilder->CreateCall(pFunc, { m_pOP->GetU32Const((unsigned)OpCode), In1[c], In2[c] });
  4253. pRes = MarkPrecise(pRes, c);
  4254. Out1[c] = m_pBuilder->CreateExtractValue(pRes, 0);
  4255. Out2[c] = m_pBuilder->CreateExtractValue(pRes, 1);
  4256. }
  4257. StoreOperand(Out1, Inst, DstIdx1, Dst1Mask, OperationType);
  4258. StoreOperand(Out2, Inst, DstIdx2, Dst2Mask, OperationType);
  4259. }
  4260. void DxbcConverter::ConvertBinaryWithCarry(OP::OpCode OpCode,
  4261. D3D10ShaderBinary::CInstruction &Inst,
  4262. const unsigned DstIdx1, const unsigned DstIdx2,
  4263. const unsigned SrcIdx1, const unsigned SrcIdx2) {
  4264. DXASSERT_NOMSG(OP::GetOpCodeClass(OpCode) == OP::OpCodeClass::BinaryWithCarryOrBorrow);
  4265. CMask WriteMask = CMask::FromDXBC(Inst.m_Operands[DstIdx1].m_WriteMask | Inst.m_Operands[DstIdx2].m_WriteMask);
  4266. CompType OperationType = CompType::getI32();
  4267. Type *pOperationType = OperationType.GetLLVMType(m_Ctx);
  4268. Function *pFunc = m_pOP->GetOpFunc(OpCode, pOperationType);
  4269. OperandValue In1, In2, Out1, Out2;
  4270. LoadOperand(In1, Inst, SrcIdx1, WriteMask, OperationType);
  4271. LoadOperand(In2, Inst, SrcIdx2, WriteMask, OperationType);
  4272. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  4273. if (!WriteMask.IsSet(c)) continue;
  4274. Value *pRes = m_pBuilder->CreateCall(pFunc, { m_pOP->GetU32Const((unsigned)OpCode), In1[c], In2[c] });
  4275. pRes = MarkPrecise(pRes, c);
  4276. Out1[c] = m_pBuilder->CreateExtractValue(pRes, 0);
  4277. Out2[c] = m_pBuilder->CreateExtractValue(pRes, 1);
  4278. Out2[c] = m_pBuilder->CreateZExt(Out2[c], Type::getInt32Ty(m_Ctx));
  4279. }
  4280. StoreOperand(Out1, Inst, DstIdx1, WriteMask, OperationType);
  4281. StoreOperand(Out2, Inst, DstIdx2, WriteMask, CompType::getI32());
  4282. }
  4283. void DxbcConverter::ConvertTertiary(OP::OpCode OpCode,
  4284. const CompType &ElementType,
  4285. D3D10ShaderBinary::CInstruction &Inst,
  4286. const unsigned DstIdx,
  4287. const unsigned SrcIdx1,
  4288. const unsigned SrcIdx2,
  4289. const unsigned SrcIdx3) {
  4290. DXASSERT_NOMSG(OP::GetOpCodeClass(OpCode) == OP::OpCodeClass::Tertiary);
  4291. CMask WriteMask = CMask::FromDXBC(Inst.m_Operands[DstIdx].m_WriteMask);
  4292. CompType OperationType = DXBC::GetCompTypeWithMinPrec(ElementType, Inst.m_Operands[DstIdx].m_MinPrecision);
  4293. Type *pOperationType = OperationType.GetLLVMType(m_Ctx);
  4294. if (!m_pOP->IsOverloadLegal(OpCode, pOperationType)) {
  4295. if (pOperationType == Type::getInt16Ty(m_Ctx)) {
  4296. pOperationType = Type::getInt32Ty(m_Ctx);
  4297. OperationType = ElementType;
  4298. }
  4299. }
  4300. Function *pFunc = m_pOP->GetOpFunc(OpCode, pOperationType);
  4301. OperandValue In1, In2, In3, Out;
  4302. LoadOperand(In1, Inst, SrcIdx1, WriteMask, OperationType);
  4303. LoadOperand(In2, Inst, SrcIdx2, WriteMask, OperationType);
  4304. LoadOperand(In3, Inst, SrcIdx3, WriteMask, OperationType);
  4305. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  4306. if (!WriteMask.IsSet(c)) continue;
  4307. Out[c] = m_pBuilder->CreateCall(pFunc, { m_pOP->GetU32Const((unsigned)OpCode), In1[c], In2[c], In3[c] });
  4308. if (ElementType.GetKind() == CompType::Kind::F64) {
  4309. c++;
  4310. }
  4311. }
  4312. StoreOperand(Out, Inst, DstIdx, WriteMask, OperationType);
  4313. }
  4314. void DxbcConverter::ConvertQuaternary(OP::OpCode OpCode,
  4315. const CompType &ElementType,
  4316. D3D10ShaderBinary::CInstruction &Inst,
  4317. const unsigned DstIdx,
  4318. const unsigned SrcIdx1, const unsigned SrcIdx2,
  4319. const unsigned SrcIdx3, const unsigned SrcIdx4) {
  4320. DXASSERT_NOMSG(OP::GetOpCodeClass(OpCode) == OP::OpCodeClass::Quaternary);
  4321. CMask WriteMask = CMask::FromDXBC(Inst.m_Operands[DstIdx].m_WriteMask);
  4322. CompType OperationType = DXBC::GetCompTypeWithMinPrec(ElementType, Inst.m_Operands[DstIdx].m_MinPrecision);
  4323. Type *pOperationType = OperationType.GetLLVMType(m_Ctx);
  4324. Function *pFunc = m_pOP->GetOpFunc(OpCode, pOperationType);
  4325. OperandValue In1, In2, In3, In4, Out;
  4326. LoadOperand(In1, Inst, SrcIdx1, WriteMask, OperationType);
  4327. LoadOperand(In2, Inst, SrcIdx2, WriteMask, OperationType);
  4328. LoadOperand(In3, Inst, SrcIdx3, WriteMask, OperationType);
  4329. LoadOperand(In4, Inst, SrcIdx4, WriteMask, OperationType);
  4330. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  4331. if (!WriteMask.IsSet(c)) continue;
  4332. Out[c] = m_pBuilder->CreateCall(pFunc, { m_pOP->GetU32Const((unsigned)OpCode), In1[c], In2[c], In3[c], In4[c] });
  4333. }
  4334. StoreOperand(Out, Inst, DstIdx, WriteMask, OperationType);
  4335. }
  4336. void DxbcConverter::ConvertComparison(CmpInst::Predicate Predicate,
  4337. const CompType &ElementType,
  4338. D3D10ShaderBinary::CInstruction &Inst,
  4339. const unsigned DstIdx,
  4340. const unsigned SrcIdx1,
  4341. const unsigned SrcIdx2) {
  4342. CMask WriteMask = CMask::FromDXBC(Inst.m_Operands[DstIdx].m_WriteMask);
  4343. CompType OperationType = DXBC::GetCompTypeWithMinPrec(ElementType,
  4344. GetHigherPrecision(Inst.m_Operands[SrcIdx1].m_MinPrecision,
  4345. Inst.m_Operands[SrcIdx2].m_MinPrecision));
  4346. if (ElementType.GetKind() != CompType::Kind::F64) {
  4347. OperandValue In1, In2, Out;
  4348. LoadOperand(In1, Inst, SrcIdx1, WriteMask, OperationType);
  4349. LoadOperand(In2, Inst, SrcIdx2, WriteMask, OperationType);
  4350. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  4351. if (!WriteMask.IsSet(c)) continue;
  4352. switch (Predicate) {
  4353. case CmpInst::FCMP_OEQ:
  4354. case CmpInst::FCMP_UNE:
  4355. case CmpInst::FCMP_OLT:
  4356. case CmpInst::FCMP_OGE:
  4357. Out[c] = m_pBuilder->CreateFCmp(Predicate, In1[c], In2[c]);
  4358. break;
  4359. case CmpInst::ICMP_EQ:
  4360. case CmpInst::ICMP_NE:
  4361. case CmpInst::ICMP_SLT:
  4362. case CmpInst::ICMP_SGE:
  4363. case CmpInst::ICMP_ULT:
  4364. case CmpInst::ICMP_UGE:
  4365. Out[c] = m_pBuilder->CreateICmp(Predicate, In1[c], In2[c]);
  4366. break;
  4367. default:
  4368. DXASSERT_NOMSG(false);
  4369. }
  4370. }
  4371. StoreOperand(Out, Inst, DstIdx, WriteMask, CompType::getI1());
  4372. } else {
  4373. // Double-precision comparison.
  4374. CMask Mask = CMask::GetMaskForDoubleOperation(WriteMask);
  4375. OperandValue In1, In2, Out;
  4376. LoadOperand(In1, Inst, SrcIdx1, Mask, OperationType);
  4377. LoadOperand(In2, Inst, SrcIdx2, Mask, OperationType);
  4378. BYTE OperationComp = 0;
  4379. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  4380. if (!WriteMask.IsSet(c)) continue;
  4381. switch (Predicate) {
  4382. case CmpInst::FCMP_OEQ:
  4383. case CmpInst::FCMP_UNE:
  4384. case CmpInst::FCMP_OLT:
  4385. case CmpInst::FCMP_OGE:
  4386. Out[c] = m_pBuilder->CreateFCmp(Predicate, In1[OperationComp], In2[OperationComp]);
  4387. break;
  4388. default:
  4389. DXASSERT_NOMSG(false);
  4390. }
  4391. OperationComp += 2;
  4392. }
  4393. StoreOperand(Out, Inst, DstIdx, WriteMask, CompType::getI1());
  4394. }
  4395. }
  4396. void DxbcConverter::ConvertDotProduct(OP::OpCode OpCode,
  4397. const BYTE NumComps,
  4398. const CMask &LoadMask,
  4399. D3D10ShaderBinary::CInstruction &Inst) {
  4400. CMask WriteMask = CMask::FromDXBC(Inst.m_Operands[0].m_WriteMask);
  4401. CompType OperationType = DXBC::GetCompTypeWithMinPrec(CompType::getF32(), Inst.m_Operands[0].m_MinPrecision);
  4402. Type *pOperationType = OperationType.GetLLVMType(m_Ctx);
  4403. Function *pFunc = m_pOP->GetOpFunc(OpCode, pOperationType);
  4404. OperandValue In1, In2, Out;
  4405. LoadOperand(In1, Inst, 1, LoadMask, OperationType);
  4406. LoadOperand(In2, Inst, 2, LoadMask, OperationType);
  4407. vector<Value*> Args;
  4408. Args.resize(1 + NumComps*2);
  4409. Args[0] = m_pOP->GetU32Const((unsigned)OpCode);
  4410. for (BYTE c = 0; c < NumComps; c++) {
  4411. Args[1 + c] = In1[c];
  4412. Args[1 + NumComps + c] = In2[c];
  4413. }
  4414. Value *pValue = m_pBuilder->CreateCall(pFunc, Args);
  4415. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  4416. if (!WriteMask.IsSet(c)) continue;
  4417. Out[c] = pValue;
  4418. }
  4419. StoreOperand(Out, Inst, 0, WriteMask, OperationType);
  4420. }
  4421. static Value *SafeConvertCast(IRBuilder<> &Builder, Value *pSrc, Type *pDstType, CompType::Kind SrcKind, CompType::Kind DstKind) {
  4422. // Prevent undef or nullptr values from getting through
  4423. Value *pResult = nullptr;
  4424. switch (SrcKind) {
  4425. case CompType::Kind::F32:
  4426. switch (DstKind) {
  4427. case CompType::Kind::I32: pResult = Builder.CreateFPToSI(pSrc, pDstType); break;
  4428. case CompType::Kind::U32: pResult = Builder.CreateFPToUI(pSrc, pDstType); break;
  4429. case CompType::Kind::F16: pResult = Builder.CreateFPTrunc(pSrc, pDstType); break;
  4430. case CompType::Kind::F64: pResult = Builder.CreateFPExt(pSrc, pDstType); break;
  4431. }
  4432. break;
  4433. case CompType::Kind::I32:
  4434. switch (DstKind) {
  4435. case CompType::Kind::F32:
  4436. case CompType::Kind::F64: pResult = Builder.CreateSIToFP(pSrc, pDstType); break;
  4437. }
  4438. break;
  4439. case CompType::Kind::U32:
  4440. switch (DstKind) {
  4441. case CompType::Kind::F32:
  4442. case CompType::Kind::F64: pResult = Builder.CreateUIToFP(pSrc, pDstType); break;
  4443. }
  4444. break;
  4445. case CompType::Kind::F16:
  4446. switch (DstKind) {
  4447. case CompType::Kind::F32:
  4448. case CompType::Kind::F64: pResult = Builder.CreateFPExt(pSrc, pDstType); break;
  4449. }
  4450. break;
  4451. }
  4452. // Note: Conversion from F64 uses ConvertFromDouble instead.
  4453. DXASSERT(pResult != nullptr, "otherwise the caller passed incorrect type combination");
  4454. // nullptr result indicates an error, but undef result may also occur with out-of-range constants
  4455. // Rescue null or undef result by converting to max/min(u)int/+-infinity, or 0xfefefefe/+-nan, to prevent invalid IR.
  4456. if (!pResult || isa<UndefValue>(pResult)) {
  4457. bool bSrcNegative = false;
  4458. bool bInvalid = !pResult;
  4459. // Get src sign:
  4460. if (ConstantFP *pConstFP = dyn_cast<ConstantFP>(pSrc)) {
  4461. bSrcNegative = pConstFP->getValueAPF().isNegative();
  4462. }
  4463. else if (ConstantInt *pConstInt = dyn_cast<ConstantInt>(pSrc)) {
  4464. bSrcNegative = pConstInt->getValue().isNegative();
  4465. } else {
  4466. DXASSERT(false, "unhandled case for SafeConvertCast failure.");
  4467. bInvalid = true;
  4468. }
  4469. if (pDstType->isIntegerTy()) {
  4470. DXASSERT(pDstType->getScalarSizeInBits() == 32, "otherwise, int dest type is not expected size");
  4471. APInt API(32, 0xFEFEFEFE);
  4472. if (!bInvalid) {
  4473. switch (DstKind) {
  4474. case CompType::Kind::I32: API = bSrcNegative ? APInt::getSignedMinValue(32) : APInt::getSignedMaxValue(32); break;
  4475. case CompType::Kind::U32: API = bSrcNegative ? APInt::getNullValue(32) : APInt::getMaxValue(32); break;
  4476. }
  4477. }
  4478. pResult = ConstantInt::get(pDstType->getContext(), API);
  4479. } else {
  4480. if (bInvalid) {
  4481. pResult = ConstantFP::getNaN(pDstType, bSrcNegative);
  4482. } else {
  4483. pResult = ConstantFP::getInfinity(pDstType, bSrcNegative);
  4484. }
  4485. }
  4486. }
  4487. return pResult;
  4488. }
  4489. void DxbcConverter::ConvertCast(const CompType &SrcElementType,
  4490. const CompType &DstElementType,
  4491. D3D10ShaderBinary::CInstruction &Inst,
  4492. const unsigned DstIdx,
  4493. const unsigned SrcIdx) {
  4494. CMask WriteMask = CMask::FromDXBC(Inst.m_Operands[DstIdx].m_WriteMask);
  4495. Type *pDstType = DstElementType.GetLLVMType(m_Ctx);
  4496. OperandValue In, Out;
  4497. LoadOperand(In, Inst, SrcIdx, WriteMask, SrcElementType);
  4498. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  4499. if (!WriteMask.IsSet(c)) continue;
  4500. Out[c] = SafeConvertCast(*m_pBuilder, In[c], pDstType, SrcElementType.GetKind(), DstElementType.GetKind());
  4501. }
  4502. StoreOperand(Out, Inst, DstIdx, WriteMask, DstElementType);
  4503. }
  4504. void DxbcConverter::ConvertToDouble(const CompType &SrcElementType, D3D10ShaderBinary::CInstruction &Inst) {
  4505. const unsigned DstIdx = 0;
  4506. const unsigned SrcIdx = 1;
  4507. CMask WriteMask = CMask::FromDXBC(Inst.m_Operands[DstIdx].m_WriteMask);
  4508. CompType DstElementType = CompType::getF64();
  4509. Type *pDstType = DstElementType.GetLLVMType(m_Ctx);
  4510. CMask Mask;
  4511. BYTE OutputComp;
  4512. switch (WriteMask.ToByte()) {
  4513. case 0x0: return;
  4514. case 0x3: Mask = CMask(1,0,0,0); OutputComp = 0; break;
  4515. case 0xC: Mask = CMask(1,0,0,0); OutputComp = 2; break;
  4516. case 0xF: Mask = CMask(1,1,0,0); OutputComp = 0; break;
  4517. default: DXASSERT_DXBC(false);
  4518. }
  4519. OperandValue In, Out;
  4520. LoadOperand(In, Inst, SrcIdx, Mask, SrcElementType);
  4521. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  4522. if (!Mask.IsSet(c)) continue;
  4523. Out[OutputComp] = SafeConvertCast(*m_pBuilder, In[c], pDstType, SrcElementType.GetKind(), DstElementType.GetKind());
  4524. OutputComp += 2;
  4525. }
  4526. StoreOperand(Out, Inst, DstIdx, WriteMask, DstElementType);
  4527. }
  4528. void DxbcConverter::ConvertFromDouble(const CompType &DstElementType, D3D10ShaderBinary::CInstruction &Inst) {
  4529. const unsigned DstIdx = 0;
  4530. const unsigned SrcIdx = 1;
  4531. CMask WriteMask = CMask::FromDXBC(Inst.m_Operands[DstIdx].m_WriteMask);
  4532. CompType SrcElementType = CompType::getF64();
  4533. CMask Mask = CMask::GetMaskForDoubleOperation(WriteMask);
  4534. OperandValue In, Out;
  4535. LoadOperand(In, Inst, SrcIdx, Mask, SrcElementType);
  4536. BYTE OperationComp = 0;
  4537. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  4538. if (!WriteMask.IsSet(c)) continue;
  4539. OP::OpCode OpCode = OP::OpCode(0);
  4540. switch (DstElementType.GetKind()) {
  4541. case CompType::Kind::I32: OpCode = OP::OpCode::LegacyDoubleToSInt32; break;
  4542. case CompType::Kind::U32: OpCode = OP::OpCode::LegacyDoubleToUInt32; break;
  4543. case CompType::Kind::F32: OpCode = OP::OpCode::LegacyDoubleToFloat; break;
  4544. default: DXASSERT_NOMSG(false);
  4545. }
  4546. // Create call.
  4547. Function *F = F = m_pOP->GetOpFunc(OpCode, Type::getVoidTy(m_Ctx));
  4548. Value *Args[2];
  4549. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  4550. Args[1] = In[OperationComp]; // Double value
  4551. Out[c] = MarkPrecise(m_pBuilder->CreateCall(F, Args));
  4552. OperationComp += 2;
  4553. }
  4554. StoreOperand(Out, Inst, DstIdx, WriteMask, DstElementType);
  4555. }
  4556. void DxbcConverter::LoadCommonSampleInputs(D3D10ShaderBinary::CInstruction &Inst, Value *pArgs[], bool bSetOffsets) {
  4557. bool bHasFeedback = DXBC::HasFeedback(Inst.OpCode());
  4558. const unsigned uOpOutput = 0;
  4559. const unsigned uOpStatus = 1;
  4560. const unsigned uOpCoord = uOpStatus + (bHasFeedback ? 1 : 0);
  4561. const unsigned uOpSRV = DXBC::GetResourceSlot(Inst.OpCode());
  4562. const unsigned uOpSampler = uOpSRV + 1;
  4563. DXASSERT_DXBC(Inst.m_Operands[uOpSRV].m_Type == D3D10_SB_OPERAND_TYPE_RESOURCE);
  4564. DXASSERT_DXBC(Inst.m_Operands[uOpSampler].m_Type == D3D10_SB_OPERAND_TYPE_SAMPLER);
  4565. OperandValue InSRV, InSampler, InCoord;
  4566. // Resource.
  4567. const DxilResource &R = LoadSRVOperand(InSRV, Inst, uOpSRV, CMask::MakeXMask(), CompType::getInvalid());
  4568. // Coordinates.
  4569. CMask CoordMask = CMask::MakeFirstNCompMask(DXBC::GetNumResCoords(R.GetKind()));
  4570. LoadOperand(InCoord, Inst, uOpCoord, CoordMask, CompType::getF32());
  4571. // Sampler.
  4572. LoadOperand(InSampler, Inst, uOpSampler, CMask::MakeXMask(), CompType::getInvalid());
  4573. // Create Sample call's common arguments.
  4574. pArgs[1] = InSRV[0]; // SRV handle
  4575. pArgs[2] = InSampler[0]; // Sampler handle
  4576. pArgs[3] = CoordMask.IsSet(0) ? InCoord[0] : m_pUnusedF32; // Coordinate 0
  4577. pArgs[4] = CoordMask.IsSet(1) ? InCoord[1] : m_pUnusedF32; // Coordinate 1
  4578. pArgs[5] = CoordMask.IsSet(2) ? InCoord[2] : m_pUnusedF32; // Coordinate 2
  4579. pArgs[6] = CoordMask.IsSet(3) ? InCoord[3] : m_pUnusedF32; // Coordinate 3
  4580. // Offsets.
  4581. if (bSetOffsets) {
  4582. CMask ResOffsetMask = CMask::MakeFirstNCompMask(DXBC::GetNumResOffsets(R.GetKind()));
  4583. pArgs[7] = ResOffsetMask.IsSet(0) ? m_pOP->GetU32Const(Inst.m_TexelOffset[0]) : m_pUnusedI32; // Offset 0
  4584. pArgs[8] = ResOffsetMask.IsSet(1) ? m_pOP->GetU32Const(Inst.m_TexelOffset[1]) : m_pUnusedI32; // Offset 1
  4585. pArgs[9] = ResOffsetMask.IsSet(2) ? m_pOP->GetU32Const(Inst.m_TexelOffset[2]) : m_pUnusedI32; // Offset 2
  4586. }
  4587. }
  4588. void DxbcConverter::StoreResRetOutputAndStatus(D3D10ShaderBinary::CInstruction &Inst,
  4589. Value *pResRet,
  4590. CompType DstType) {
  4591. bool bHasFeedback = DXBC::HasFeedback(Inst.OpCode());
  4592. const unsigned uOpOutput = 0;
  4593. const unsigned uOpStatus = 1;
  4594. const unsigned uOpRes = GetResourceSlot(Inst);
  4595. MarkPrecise(pResRet);
  4596. // Store output.
  4597. CMask OutputMask = CMask::FromDXBC(Inst.m_Operands[uOpOutput].m_WriteMask);
  4598. if (!OutputMask.IsZero()) {
  4599. OperandValue Out;
  4600. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  4601. if (!OutputMask.IsSet(c)) continue;
  4602. // Respect swizzle: resource swizzle == return value swizzle.
  4603. BYTE Comp = Inst.m_Operands[uOpRes].m_Swizzle[c];
  4604. Out[c] = m_pBuilder->CreateExtractValue(pResRet, Comp);
  4605. }
  4606. StoreOperand(Out, Inst, uOpOutput, OutputMask, DstType);
  4607. }
  4608. // Store status.
  4609. if (bHasFeedback) {
  4610. CMask StatusMask = CMask::FromDXBC(Inst.m_Operands[uOpStatus].m_WriteMask);
  4611. if (!StatusMask.IsZero()) {
  4612. OperandValue Status;
  4613. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  4614. if (!StatusMask.IsSet(c)) continue;
  4615. const unsigned uStatusField = 4;
  4616. Status[c] = m_pBuilder->CreateExtractValue(pResRet, uStatusField);
  4617. }
  4618. StoreOperand(Status, Inst, uOpStatus, StatusMask, CompType::getU32());
  4619. }
  4620. }
  4621. }
  4622. void DxbcConverter::StoreGetDimensionsOutput(D3D10ShaderBinary::CInstruction &Inst, Value *pGetDimRet) {
  4623. const unsigned uOpOutput = 0;
  4624. const unsigned uOpRes = DXBC::GetResourceSlot(Inst.OpCode());
  4625. CMask OutputMask = CMask::FromDXBC(Inst.m_Operands[uOpOutput].m_WriteMask);
  4626. if (OutputMask.IsZero())
  4627. return;
  4628. // Resource.
  4629. const DxilResource *R;
  4630. if (Inst.m_Operands[uOpRes].m_Type == D3D10_SB_OPERAND_TYPE_RESOURCE) {
  4631. R = &GetSRVFromOperand(Inst, uOpRes);
  4632. } else {
  4633. unsigned RangeID = Inst.m_Operands[uOpRes].m_Index[0].m_RegIndex;
  4634. R = &m_pPR->GetUAV(m_UAVRangeMap[RangeID]);
  4635. }
  4636. // Return type.
  4637. CompType RetType = DXBC::GetCompTypeWithMinPrec(CompType::getI32(), Inst.m_Operands[uOpOutput].m_MinPrecision);
  4638. // Value type.
  4639. CompType ValueType = CompType::getI32();
  4640. bool bRcp = false;
  4641. switch (Inst.m_ResInfoReturnType) {
  4642. case D3D10_SB_RESINFO_INSTRUCTION_RETURN_FLOAT:
  4643. ValueType = CompType::getF32();
  4644. break;
  4645. case D3D10_SB_RESINFO_INSTRUCTION_RETURN_RCPFLOAT:
  4646. ValueType = CompType::getF32();
  4647. bRcp = true;
  4648. break;
  4649. case D3D10_SB_RESINFO_INSTRUCTION_RETURN_UINT:
  4650. ValueType = CompType::getI32();
  4651. break;
  4652. default:
  4653. DXASSERT_DXBC(false);
  4654. }
  4655. OperandValue Out;
  4656. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  4657. if (!OutputMask.IsSet(c)) continue;
  4658. // Respect swizzle: resource swizzle == return value swizzle.
  4659. BYTE Comp = Inst.m_Operands[uOpRes].m_Swizzle[c];
  4660. Value *pCompVal = m_pBuilder->CreateExtractValue(pGetDimRet, Comp);
  4661. if (ValueType.IsFloatTy()) {
  4662. pCompVal = m_pBuilder->CreateCast(Instruction::CastOps::UIToFP, pCompVal, Type::getFloatTy(m_Ctx));
  4663. }
  4664. if (bRcp) {
  4665. if (Comp < DxilResource::GetNumDimensions(R->GetKind())) {
  4666. pCompVal = m_pBuilder->CreateBinOp(Instruction::BinaryOps::FDiv, m_pOP->GetFloatConst(1.0f), pCompVal);
  4667. }
  4668. }
  4669. Out[c] = pCompVal;
  4670. }
  4671. StoreOperand(Out, Inst, uOpOutput, OutputMask, ValueType);
  4672. }
  4673. void DxbcConverter::StoreSamplePosOutput(D3D10ShaderBinary::CInstruction &Inst, Value *pSamplePosVal) {
  4674. const unsigned uOpOutput = 0;
  4675. const unsigned uOpRes = DXBC::GetResourceSlot(Inst.OpCode());
  4676. CompType DstType = DXBC::GetCompTypeWithMinPrec(CompType::getF32(), Inst.m_Operands[uOpOutput].m_MinPrecision);
  4677. // Store output.
  4678. CMask OutputMask = CMask::FromDXBC(Inst.m_Operands[uOpOutput].m_WriteMask);
  4679. if (!OutputMask.IsZero()) {
  4680. OperandValue Out;
  4681. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  4682. if (!OutputMask.IsSet(c)) continue;
  4683. BYTE Comp = Inst.m_Operands[uOpRes].m_Swizzle[c];
  4684. if (Comp < 2) {
  4685. Out[c] = m_pBuilder->CreateExtractValue(pSamplePosVal, Comp);
  4686. } else {
  4687. Out[c] = m_pOP->GetFloatConst(0);
  4688. }
  4689. }
  4690. StoreOperand(Out, Inst, uOpOutput, OutputMask, DstType);
  4691. }
  4692. }
  4693. void DxbcConverter::StoreBroadcastOutput(D3D10ShaderBinary::CInstruction &Inst, Value *pValue, CompType DstType) {
  4694. const unsigned uOpOutput = 0;
  4695. CMask OutputMask = CMask::FromDXBC(Inst.m_Operands[uOpOutput].m_WriteMask);
  4696. if (!OutputMask.IsZero()) {
  4697. OperandValue Out;
  4698. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  4699. if (!OutputMask.IsSet(c)) continue;
  4700. Out[c] = pValue;
  4701. }
  4702. StoreOperand(Out, Inst, uOpOutput, OutputMask, DstType);
  4703. }
  4704. }
  4705. Value *DxbcConverter::GetCoordValue(D3D10ShaderBinary::CInstruction &Inst, const unsigned uCoordIdx) {
  4706. BYTE CoordComp = Inst.m_Operands[uCoordIdx].m_ComponentName;
  4707. OperandValue InCoord;
  4708. CMask CoordMask = CMask::MakeCompMask(CoordComp);
  4709. LoadOperand(InCoord, Inst, uCoordIdx, CoordMask, CompType::getI32());
  4710. return InCoord[CoordComp];
  4711. }
  4712. Value *DxbcConverter::GetByteOffset(D3D10ShaderBinary::CInstruction &Inst, const unsigned Idx1,
  4713. const unsigned Idx2, const unsigned Stride) {
  4714. const unsigned uOpElementOffset = Idx1;
  4715. const unsigned uOpStructByteOffset = Idx2;
  4716. OperandValue InElementOffset, InStructByteOffset;
  4717. // Element offset.
  4718. BYTE ElementOffsetComp = Inst.m_Operands[uOpElementOffset].m_ComponentName;
  4719. CMask CoordMask = CMask::MakeCompMask(ElementOffsetComp);
  4720. LoadOperand(InElementOffset, Inst, uOpElementOffset, CoordMask, CompType::getI32());
  4721. // Byte offset into the structure.
  4722. BYTE StructByteOffsetComp = Inst.m_Operands[uOpStructByteOffset].m_ComponentName;
  4723. CMask StructByteOffsetMask = CMask::MakeCompMask(StructByteOffsetComp);
  4724. LoadOperand(InStructByteOffset, Inst, uOpStructByteOffset, StructByteOffsetMask, CompType::getI32());
  4725. // Calculate byte offset.
  4726. Value *pOffset1 = InElementOffset[ElementOffsetComp];
  4727. Value *pOffset2 = InStructByteOffset[StructByteOffsetComp];
  4728. Value *pMul = pOffset1;
  4729. if (Stride > 1) {
  4730. Value *pStride = m_pOP->GetU32Const(Stride);
  4731. pMul = m_pBuilder->CreateMul(pOffset1, pStride);
  4732. }
  4733. Value *pByteOffset = m_pBuilder->CreateAdd(pMul, pOffset2);
  4734. return pByteOffset;
  4735. }
  4736. void DxbcConverter::ConvertLoadTGSM(D3D10ShaderBinary::CInstruction &Inst, const unsigned uOpTGSM,
  4737. const unsigned uOpOutput, CompType SrcType, Value *pByteOffset) {
  4738. DXASSERT_DXBC(Inst.m_Operands[uOpTGSM].m_Type == D3D11_SB_OPERAND_TYPE_THREAD_GROUP_SHARED_MEMORY);
  4739. const TGSMEntry &R = m_TGSMMap[Inst.m_Operands[uOpTGSM].m_Index[0].m_RegIndex];
  4740. CMask OutputMask = CMask::FromDXBC(Inst.m_Operands[uOpOutput].m_WriteMask);
  4741. if (OutputMask.IsZero())
  4742. return;
  4743. OperandValue Out;
  4744. CompType DstType = DXBC::GetCompTypeFromMinPrec(Inst.m_Operands[uOpOutput].m_MinPrecision, CompType::getF32());
  4745. Type *pSrcType = SrcType.GetLLVMPtrType(m_Ctx, DXIL::kTGSMAddrSpace);
  4746. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  4747. if (!OutputMask.IsSet(c)) continue;
  4748. // Swizzle.
  4749. BYTE Comp = Inst.m_Operands[uOpTGSM].m_Swizzle[c];
  4750. // Adjust index for component.
  4751. Value *pValueIndex = pByteOffset;
  4752. if (Comp > 0) {
  4753. pValueIndex = m_pBuilder->CreateAdd(pByteOffset, m_pOP->GetU32Const(Comp * kRegCompAlignment));
  4754. }
  4755. // Create GEP.
  4756. Value *pGEPIndices[2] = { m_pOP->GetU32Const(0), pValueIndex };
  4757. Value *pPtrI8 = m_pBuilder->CreateGEP(R.pVar, pGEPIndices);
  4758. // Create load.
  4759. Value *pPtr = m_pBuilder->CreatePointerCast(pPtrI8, pSrcType);
  4760. LoadInst *pLoad = m_pBuilder->CreateLoad(pPtr);
  4761. pLoad->setAlignment(kRegCompAlignment);
  4762. Out[c] = CastDxbcValue(pLoad, SrcType, DstType);
  4763. }
  4764. StoreOperand(Out, Inst, uOpOutput, OutputMask, DstType);
  4765. }
  4766. void DxbcConverter::ConvertStoreTGSM(D3D10ShaderBinary::CInstruction &Inst, const unsigned uOpTGSM,
  4767. const unsigned uOpValue, CompType BaseValueType, Value *pByteOffset) {
  4768. DXASSERT_DXBC(Inst.m_Operands[uOpTGSM].m_Type == D3D11_SB_OPERAND_TYPE_THREAD_GROUP_SHARED_MEMORY);
  4769. const TGSMEntry &R = m_TGSMMap[Inst.m_Operands[uOpTGSM].m_Index[0].m_RegIndex];
  4770. // Value type.
  4771. CompType ValueType = DXBC::GetCompTypeFromMinPrec(Inst.m_Operands[uOpValue].m_MinPrecision, BaseValueType);
  4772. // Store TGSM value.
  4773. CMask OutputMask = CMask::FromDXBC(Inst.m_Operands[uOpTGSM].m_WriteMask);
  4774. if (OutputMask.IsZero())
  4775. return;
  4776. // Value.
  4777. OperandValue InValue;
  4778. LoadOperand(InValue, Inst, uOpValue, OutputMask, ValueType);
  4779. CompType DstType = BaseValueType;
  4780. Type *pDstType = DstType.GetLLVMPtrType(m_Ctx, DXIL::kTGSMAddrSpace);
  4781. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  4782. if (!OutputMask.IsSet(c)) continue;
  4783. // Adjust index for component.
  4784. Value *pValueIndex = pByteOffset;
  4785. if (c > 0) {
  4786. pValueIndex = m_pBuilder->CreateAdd(pByteOffset, m_pOP->GetU32Const(c * kRegCompAlignment));
  4787. }
  4788. // Cast value to the right type.
  4789. Value *pValue = CastDxbcValue(InValue[c], ValueType, DstType);
  4790. // Create GEP.
  4791. Value *pGEPIndices[2] = { m_pOP->GetU32Const(0), pValueIndex };
  4792. Value *pPtrI8 = m_pBuilder->CreateGEP(R.pVar, pGEPIndices);
  4793. // Create store.
  4794. Value *pPtr = m_pBuilder->CreatePointerCast(pPtrI8, pDstType);
  4795. StoreInst *pStore = m_pBuilder->CreateStore(pValue, pPtr);
  4796. pStore->setAlignment(kRegCompAlignment);
  4797. (void)MarkPrecise(pStore, c);
  4798. }
  4799. }
  4800. void DxbcConverter::EmitGSOutputRegisterStore(unsigned StreamId) {
  4801. const auto &Sig = m_pOutputSignature->m_Signature.GetElements();
  4802. // For each output decl for stream StreamID.
  4803. for (size_t i = 0; i < Sig.size(); i++) {
  4804. DxilSignatureElement &SE = m_pOutputSignature->m_Signature.GetElement(i);
  4805. if (SE.GetOutputStream() != StreamId)
  4806. continue;
  4807. DXASSERT(SE.GetRows() == 1, "to support indexable output in GS with multiple output streams");
  4808. unsigned TempReg = GetGSTempRegForOutputReg(SE.GetStartRow());
  4809. CompType DxbcValueType = SE.GetCompType();
  4810. Type *pDxbcValueType = DxbcValueType.GetLLVMType(m_Ctx);
  4811. for (BYTE c = 0; c < SE.GetCols(); c++) {
  4812. BYTE Comp = SE.GetStartCol() + c;
  4813. Value *pValue;
  4814. // 1. Load value from the corresponding temp reg.
  4815. {
  4816. Value *Args[2];
  4817. Args[0] = m_pOP->GetU32Const((unsigned)OP::OpCode::TempRegLoad); // OpCode
  4818. Args[1] = m_pOP->GetU32Const(DXBC::GetRegIndex(TempReg, Comp)); // Linearized register index
  4819. Function *F = m_pOP->GetOpFunc(OP::OpCode::TempRegLoad, pDxbcValueType);
  4820. pValue = m_pBuilder->CreateCall(F, Args);
  4821. }
  4822. // 2. Store the value to the output reg.
  4823. {
  4824. Value *Args[5];
  4825. Args[0] = m_pOP->GetU32Const((unsigned)OP::OpCode::StoreOutput); // OpCode
  4826. Args[1] = m_pOP->GetU32Const(SE.GetID()); // Output signature element ID
  4827. Args[2] = m_pOP->GetU32Const(0); // Row, relative to the element
  4828. Args[3] = m_pOP->GetU8Const(c); // Col, relative to the element
  4829. Args[4] = pValue; // Value
  4830. Function *F = m_pOP->GetOpFunc(OP::OpCode::StoreOutput, pDxbcValueType);
  4831. m_pBuilder->CreateCall(F, Args);
  4832. }
  4833. }
  4834. }
  4835. }
  4836. Value *DxbcConverter::CreateHandle(DxilResourceBase::Class Class, unsigned RangeID,
  4837. Value *pIndex, bool bNonUniformIndex) {
  4838. DXASSERT(pIndex->getType() == Type::getInt32Ty(m_Ctx), "index should be i32 type");
  4839. OP::OpCode OpCode = OP::OpCode::CreateHandle;
  4840. Value *Args[5];
  4841. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  4842. Args[1] = m_pOP->GetU8Const((BYTE)Class); // Resource class (SRV, UAV, CBuffer, Sampler)
  4843. Args[2] = m_pOP->GetU32Const(RangeID); // Range ID
  4844. Args[3] = pIndex; // 0-based index into the range
  4845. Args[4] = m_pOP->GetI1Const(bNonUniformIndex); // Non-uniform resource index
  4846. Function *pCreateHandleFunc = m_pOP->GetOpFunc(OpCode, Type::getVoidTy(m_Ctx));
  4847. return m_pBuilder->CreateCall(pCreateHandleFunc, Args);
  4848. }
  4849. void DxbcConverter::SetCachedHandle(const DxilResourceBase &R) {
  4850. DXASSERT(!IsSM51Plus(), "must not cache handles on SM 5.1");
  4851. if (R.GetSpaceID() == 0) {
  4852. // Note: Even though space should normally be 0 for SM 5.0 and below,
  4853. // the interfaces implementation uses non-zero space when converting from SM 5.0.
  4854. m_HandleMap[std::make_pair((unsigned)R.GetClass(), (unsigned)R.GetLowerBound())] =
  4855. CreateHandle(R.GetClass(), R.GetID(), m_pOP->GetU32Const(R.GetLowerBound()), false);
  4856. }
  4857. }
  4858. Value *DxbcConverter::GetCachedHandle(const DxilResourceBase &R) {
  4859. if (IsSM51Plus() || R.GetSpaceID() != 0)
  4860. return nullptr;
  4861. auto it = m_HandleMap.find(std::make_pair((unsigned)R.GetClass(), (unsigned)R.GetLowerBound()));
  4862. if (it != m_HandleMap.end())
  4863. return it->second;
  4864. return nullptr;
  4865. }
  4866. Value *DxbcConverter::LoadConstFloat(float& fVal) {
  4867. unsigned uVal = *(unsigned *)&fVal;
  4868. APFloat V(fVal);
  4869. float fVal2 = V.convertToFloat();
  4870. if ((*(unsigned *)&fVal2) == uVal) {
  4871. return m_pOP->GetFloatConst(fVal);
  4872. } else {
  4873. OP::OpCode OpCode = OP::OpCode::BitcastI32toF32;
  4874. Value *Args[2];
  4875. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  4876. Args[1] = m_pOP->GetU32Const(uVal); // Input
  4877. Function *F = m_pOP->GetOpFunc(OpCode, Type::getVoidTy(m_Ctx));
  4878. return m_pBuilder->CreateCall(F, Args);
  4879. }
  4880. }
  4881. void DxbcConverter::SetHasCounter(D3D10ShaderBinary::CInstruction &Inst, const unsigned uOpUAV) {
  4882. D3D10ShaderBinary::COperandBase &O = Inst.m_Operands[uOpUAV];
  4883. DXASSERT_DXBC(O.m_Type == D3D11_SB_OPERAND_TYPE_UNORDERED_ACCESS_VIEW);
  4884. // Retrieve UAV range ID and record.
  4885. DXASSERT_DXBC(O.m_IndexType[0] == D3D10_SB_OPERAND_INDEX_IMMEDIATE32);
  4886. unsigned RangeID = O.m_Index[0].m_RegIndex;
  4887. unsigned RecIdx = m_UAVRangeMap[RangeID];
  4888. DxilResource &R = m_pPR->GetUAV(RecIdx);
  4889. R.SetHasCounter(true);
  4890. }
  4891. void DxbcConverter::LoadOperand(OperandValue &SrcVal,
  4892. D3D10ShaderBinary::CInstruction &Inst,
  4893. const unsigned OpIdx,
  4894. const CMask &Mask,
  4895. const CompType &ValueType) {
  4896. D3D10ShaderBinary::COperandBase &O = Inst.m_Operands[OpIdx];
  4897. switch (O.m_Type) {
  4898. case D3D10_SB_OPERAND_TYPE_IMMEDIATE32:
  4899. DXASSERT_DXBC(O.m_Modifier == D3D10_SB_OPERAND_MODIFIER_NONE);
  4900. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  4901. if (!Mask.IsSet(c)) continue;
  4902. bool bVec4 = O.m_NumComponents == D3D10_SB_OPERAND_4_COMPONENT;
  4903. BYTE Comp = bVec4 ? c : 0;
  4904. switch (ValueType.GetKind()) {
  4905. case CompType::Kind::F32:
  4906. SrcVal[c] = LoadConstFloat(O.m_Valuef[Comp]);
  4907. break;
  4908. case CompType::Kind::F16:
  4909. SrcVal[c] = CastDxbcValue(LoadConstFloat(O.m_Valuef[Comp]), CompType::Kind::F32, CompType::Kind::F16);
  4910. break;
  4911. case CompType::Kind::I32: __fallthrough;
  4912. case CompType::Kind::U32:
  4913. SrcVal[c] = m_pOP->GetU32Const(O.m_Value[Comp]);
  4914. break;
  4915. case CompType::Kind::I16: __fallthrough;
  4916. case CompType::Kind::U16:
  4917. SrcVal[c] = CastDxbcValue(m_pOP->GetU32Const(O.m_Value[Comp]), CompType::Kind::U32, CompType::Kind::I16);
  4918. break;
  4919. case CompType::Kind::I1:
  4920. SrcVal[c] = CastDxbcValue(m_pOP->GetU32Const(O.m_Value[Comp]), CompType::Kind::U32, CompType::Kind::I1);
  4921. break;
  4922. default:
  4923. DXASSERT_DXBC(false);
  4924. }
  4925. }
  4926. break;
  4927. case D3D10_SB_OPERAND_TYPE_IMMEDIATE64:
  4928. DXASSERT_NOMSG(ValueType.GetKind() == CompType::Kind::F64);
  4929. for (BYTE c = 0; c < DXBC::kWidth; c += 2) {
  4930. if (!Mask.IsSet(c)) continue;
  4931. SrcVal[c] = m_pOP->GetDoubleConst(O.m_Valued[c]);
  4932. }
  4933. break;
  4934. case D3D10_SB_OPERAND_TYPE_TEMP: {
  4935. DXASSERT_DXBC(O.m_IndexDimension == D3D10_SB_OPERAND_INDEX_1D);
  4936. unsigned Reg = O.m_Index[0].m_RegIndex;
  4937. CompType DxbcValueType = DXBC::GetCompTypeFromMinPrec(O.m_MinPrecision, ValueType);
  4938. if (DxbcValueType.IsBoolTy()) {
  4939. DxbcValueType = CompType::getI32();
  4940. }
  4941. Type *pDxbcValueType = DxbcValueType.GetLLVMType(m_Ctx);
  4942. if (DxbcValueType.GetKind() != CompType::Kind::F64)
  4943. {
  4944. for (OperandValueHelper OVH(SrcVal, Mask, O); !OVH.IsDone(); OVH.Advance()) {
  4945. BYTE Comp = OVH.GetComp();
  4946. Value *Args[2];
  4947. Args[0] = m_pOP->GetU32Const((unsigned)OP::OpCode::TempRegLoad); // OpCode
  4948. Args[1] = m_pOP->GetU32Const(DXBC::GetRegIndex(Reg, Comp)); // Linearized register index
  4949. Function *F = m_pOP->GetOpFunc(OP::OpCode::TempRegLoad, pDxbcValueType);
  4950. Value *pValue = m_pBuilder->CreateCall(F, Args);
  4951. pValue = CastDxbcValue(pValue, DxbcValueType, ValueType);
  4952. pValue = ApplyOperandModifiers(pValue, O);
  4953. OVH.SetValue(pValue);
  4954. }
  4955. } else {
  4956. DXASSERT_DXBC(CMask::IsValidDoubleMask(Mask));
  4957. for (OperandValueHelper OVH(SrcVal, Mask, O); !OVH.IsDone(); OVH.Advance()) {
  4958. BYTE Comp = OVH.GetComp();
  4959. Value *pValue1, *pValue2;
  4960. {
  4961. Value *Args[2];
  4962. Args[0] = m_pOP->GetU32Const((unsigned)OP::OpCode::TempRegLoad); // OpCode
  4963. Args[1] = m_pOP->GetU32Const(DXBC::GetRegIndex(Reg, Comp)); // Linearized register index1
  4964. Function *F = m_pOP->GetOpFunc(OP::OpCode::TempRegLoad, CompType::getU32().GetLLVMType(m_Ctx));
  4965. pValue1 = m_pBuilder->CreateCall(F, Args);
  4966. Args[1] = m_pOP->GetU32Const(DXBC::GetRegIndex(Reg, Comp+1)); // Linearized register index2
  4967. pValue2 = m_pBuilder->CreateCall(F, Args);
  4968. }
  4969. Value *pValue;
  4970. {
  4971. Value *Args[3];
  4972. Function *F = m_pOP->GetOpFunc(OP::OpCode::MakeDouble, pDxbcValueType);
  4973. Args[0] = m_pOP->GetU32Const((unsigned)OP::OpCode::MakeDouble); // OpCode
  4974. Args[1] = pValue1; // Lo part
  4975. Args[2] = pValue2; // Hi part
  4976. pValue = m_pBuilder->CreateCall(F, Args);
  4977. pValue = ApplyOperandModifiers(pValue, O);
  4978. }
  4979. OVH.SetValue(pValue);
  4980. OVH.Advance();
  4981. }
  4982. }
  4983. break;
  4984. }
  4985. case D3D10_SB_OPERAND_TYPE_INDEXABLE_TEMP: {
  4986. DXASSERT_DXBC(O.m_IndexDimension == D3D10_SB_OPERAND_INDEX_2D);
  4987. DXASSERT_DXBC(O.m_IndexType[0] == D3D10_SB_OPERAND_INDEX_IMMEDIATE32);
  4988. unsigned Reg = O.m_Index[0].m_RegIndex;
  4989. IndexableReg &IRRec = m_IndexableRegs[Reg];
  4990. Value *pXRegIndex = LoadOperandIndex(O.m_Index[1], O.m_IndexType[1]);
  4991. Value *pRegIndex = m_pBuilder->CreateMul(pXRegIndex, m_pOP->GetI32Const(IRRec.NumComps));
  4992. CompType DxbcValueType = DXBC::GetCompTypeFromMinPrec(O.m_MinPrecision, ValueType);
  4993. if (DxbcValueType.IsBoolTy()) {
  4994. DxbcValueType = CompType::getI32();
  4995. }
  4996. if (DxbcValueType.GetKind() != CompType::Kind::F64) {
  4997. for (OperandValueHelper OVH(SrcVal, Mask, O); !OVH.IsDone(); OVH.Advance()) {
  4998. BYTE Comp = OVH.GetComp();
  4999. Value *pValue = nullptr;
  5000. // Create GEP.
  5001. Value *pIndex = m_pBuilder->CreateAdd(pRegIndex, m_pOP->GetU32Const(Comp));
  5002. Value *pGEPIndices[2] = { m_pOP->GetU32Const(0), pIndex };
  5003. if (!DxbcValueType.HasMinPrec()) {
  5004. Value *pBasePtr = m_IndexableRegs[Reg].pValue32;
  5005. Value *pPtr = m_pBuilder->CreateGEP(pBasePtr, pGEPIndices);
  5006. pValue = m_pBuilder->CreateAlignedLoad(pPtr, kRegCompAlignment);
  5007. pValue = CastDxbcValue(pValue, CompType::getF32(), ValueType);
  5008. } else {
  5009. // Create GEP.
  5010. Value *pBasePtr = m_IndexableRegs[Reg].pValue16;
  5011. Value *pPtr = m_pBuilder->CreateGEP(pBasePtr, pGEPIndices);
  5012. pValue = m_pBuilder->CreateAlignedLoad(pPtr, kRegCompAlignment/2);
  5013. pValue = CastDxbcValue(pValue, CompType::getF16(), ValueType);
  5014. }
  5015. pValue = ApplyOperandModifiers(pValue, O);
  5016. OVH.SetValue(pValue);
  5017. }
  5018. } else {
  5019. // Double precision.
  5020. for (OperandValueHelper OVH(SrcVal, Mask, O); !OVH.IsDone(); OVH.Advance()) {
  5021. BYTE Comp = OVH.GetComp();
  5022. Value *pValue = nullptr;
  5023. // Create GEP.
  5024. Value *pIndex = m_pBuilder->CreateAdd(pRegIndex, m_pOP->GetU32Const(Comp));
  5025. Value *pGEPIndices[1] = { pIndex };
  5026. Value *pBasePtr = m_pBuilder->CreateBitCast(m_IndexableRegs[Reg].pValue32, Type::getDoublePtrTy(m_Ctx));
  5027. Value *pPtr = m_pBuilder->CreateGEP(pBasePtr, pGEPIndices);
  5028. pValue = m_pBuilder->CreateAlignedLoad(pPtr, kRegCompAlignment*2);
  5029. pValue = ApplyOperandModifiers(pValue, O);
  5030. OVH.SetValue(pValue);
  5031. OVH.Advance();
  5032. OVH.SetValue(pValue);
  5033. }
  5034. }
  5035. break;
  5036. }
  5037. case D3D10_SB_OPERAND_TYPE_INPUT:
  5038. case D3D11_SB_OPERAND_TYPE_INPUT_CONTROL_POINT: {
  5039. OP::OpCode OpCode = OP::OpCode::LoadInput;
  5040. unsigned Register; // Starting index of the register range.
  5041. Value *pUnitIndexValue; // Vertex/point index expression.
  5042. Value *pRowIndexValue; // Row index expression.
  5043. switch (O.m_IndexDimension) {
  5044. case D3D10_SB_OPERAND_INDEX_1D:
  5045. Register = O.m_Index[0].m_RegIndex;
  5046. pUnitIndexValue = m_pUnusedI32;
  5047. pRowIndexValue = LoadOperandIndex(O.m_Index[0], O.m_IndexType[0]);
  5048. break;
  5049. case D3D10_SB_OPERAND_INDEX_2D:
  5050. // 2D input register index: <index1, input register index>.
  5051. // index1: GS -- vertex index, DS -- input control point index.
  5052. Register = O.m_Index[1].m_RegIndex;
  5053. pUnitIndexValue = LoadOperandIndex(O.m_Index[0], O.m_IndexType[0]);
  5054. pRowIndexValue = LoadOperandIndex(O.m_Index[1], O.m_IndexType[1]);
  5055. break;
  5056. default:
  5057. DXASSERT(false, "there should no other index dimensions");
  5058. }
  5059. for (OperandValueHelper OVH(SrcVal, Mask, O); !OVH.IsDone(); OVH.Advance()) {
  5060. BYTE Comp = OVH.GetComp();
  5061. // Retrieve signature element.
  5062. const DxilSignatureElement *E = m_pInputSignature->GetElement(Register, Comp);
  5063. CompType DxbcValueType = E->GetCompType();
  5064. if (DxbcValueType.IsBoolTy()) {
  5065. DxbcValueType = CompType::getI32();
  5066. }
  5067. Type *pDxbcValueType = DxbcValueType.GetLLVMType(m_Ctx);
  5068. MutableArrayRef<Value *> Args;
  5069. Value *Args1[1];
  5070. Value *Args5[5];
  5071. if (E->GetKind() == DXIL::SemanticKind::SampleIndex) {
  5072. // Use SampleIndex intrinsic instead of LoadInput
  5073. Args = Args1;
  5074. OpCode = OP::OpCode::SampleIndex;
  5075. } else {
  5076. Args = Args5;
  5077. // Make row/col index relative within element.
  5078. Value *pRowIndexValueRel = m_pBuilder->CreateSub(pRowIndexValue, m_pOP->GetU32Const(E->GetStartRow()));
  5079. Args[1] = m_pOP->GetU32Const(E->GetID()); // Input signature element ID
  5080. Args[2] = pRowIndexValueRel; // Row, relative to the element
  5081. Args[3] = m_pOP->GetU8Const(Comp - E->GetStartCol()); // Col, relative to the element
  5082. Args[4] = pUnitIndexValue; // Vertex/point index
  5083. }
  5084. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  5085. Function *F = m_pOP->GetOpFunc(OpCode, pDxbcValueType);
  5086. Value *pValue = m_pBuilder->CreateCall(F, Args);
  5087. pValue = CastDxbcValue(pValue, DxbcValueType, ValueType);
  5088. pValue = ApplyOperandModifiers(pValue, O);
  5089. OVH.SetValue(pValue);
  5090. }
  5091. break;
  5092. }
  5093. case D3D10_SB_OPERAND_TYPE_CONSTANT_BUFFER: {
  5094. // Upconvert operand to SM5.1.
  5095. if (O.m_IndexDimension == D3D10_SB_OPERAND_INDEX_2D) {
  5096. O.m_IndexDimension = D3D10_SB_OPERAND_INDEX_3D;
  5097. O.m_IndexType[2] = O.m_IndexType[1];
  5098. O.m_Index[2] = O.m_Index[1];
  5099. O.m_IndexType[1] = O.m_IndexType[0];
  5100. O.m_Index[1] = O.m_Index[0];
  5101. }
  5102. // Retrieve cbuffer range ID and record.
  5103. const DxilCBuffer* pR = m_pClassInstanceCBuffers;
  5104. if (O.m_IndexType[0] == D3D10_SB_OPERAND_INDEX_IMMEDIATE32) {
  5105. unsigned RangeID = O.m_Index[0].m_RegIndex;
  5106. unsigned RecIdx = m_CBufferRangeMap[RangeID];
  5107. pR = &m_pPR->GetCBuffer(RecIdx);
  5108. }
  5109. const DxilCBuffer &R = *pR;
  5110. // Setup cbuffer handle.
  5111. Value *pHandle = GetCachedHandle(R);
  5112. if (pHandle == nullptr) {
  5113. // Create dynamic-index handle.
  5114. pHandle = CreateHandle(R.GetClass(), R.GetID(), LoadOperandIndex(O.m_Index[1], O.m_IndexType[1]), O.m_Nonuniform);
  5115. }
  5116. // Load values for unique components.
  5117. Value *pRegIndexValue = LoadOperandIndex(O.m_Index[2], O.m_IndexType[2]);
  5118. CompType DxbcValueType = ValueType.GetBaseCompType();
  5119. if (DxbcValueType.IsBoolTy()) {
  5120. DxbcValueType = CompType::getI32();
  5121. }
  5122. Type *pDxbcValueType = DxbcValueType.GetLLVMType(m_Ctx);
  5123. DXASSERT_NOMSG(m_bLegacyCBufferLoad);
  5124. Value *Args[3];
  5125. Args[0] = m_pOP->GetU32Const((unsigned)OP::OpCode::CBufferLoadLegacy); // OpCode
  5126. Args[1] = pHandle; // CBuffer handle
  5127. Args[2] = pRegIndexValue; // 0-based index into cbuffer instance
  5128. Function *pCBufferLoadFunc = m_pOP->GetOpFunc(OP::OpCode::CBufferLoadLegacy, pDxbcValueType);
  5129. Value *pCBufferRetValue = m_pBuilder->CreateCall(pCBufferLoadFunc, Args);
  5130. if (ValueType.GetKind() != CompType::Kind::F64) {
  5131. for (OperandValueHelper OVH(SrcVal, Mask, O); !OVH.IsDone(); OVH.Advance()) {
  5132. BYTE Comp = OVH.GetComp();
  5133. Value *pValue = m_pBuilder->CreateExtractValue(pCBufferRetValue, Comp);
  5134. pValue = CastDxbcValue(pValue, DxbcValueType, ValueType);
  5135. pValue = ApplyOperandModifiers(pValue, O);
  5136. OVH.SetValue(pValue);
  5137. }
  5138. } else {
  5139. for (OperandValueHelper OVH(SrcVal, Mask, O); !OVH.IsDone(); OVH.Advance()) {
  5140. BYTE Comp = OVH.GetComp() / 2;
  5141. Value *pValue = m_pBuilder->CreateExtractValue(pCBufferRetValue, Comp);
  5142. pValue = CastDxbcValue(pValue, DxbcValueType, ValueType);
  5143. pValue = ApplyOperandModifiers(pValue, O);
  5144. OVH.SetValue(pValue);
  5145. OVH.Advance();
  5146. OVH.SetValue(pValue);
  5147. }
  5148. }
  5149. break;
  5150. }
  5151. case D3D10_SB_OPERAND_TYPE_IMMEDIATE_CONSTANT_BUFFER: {
  5152. DXASSERT_DXBC(O.m_IndexDimension == D3D10_SB_OPERAND_INDEX_1D);
  5153. Value *pRegIndex = LoadOperandIndex(O.m_Index[0], O.m_IndexType[0]);
  5154. if (ValueType.GetKind() != CompType::Kind::F64) {
  5155. for (OperandValueHelper OVH(SrcVal, Mask, O); !OVH.IsDone(); OVH.Advance()) {
  5156. BYTE Comp = OVH.GetComp();
  5157. Value *pValueIndex = m_pBuilder->CreateMul(pRegIndex, m_pOP->GetI32Const(DXBC::kWidth));
  5158. pValueIndex = m_pBuilder->CreateAdd(pValueIndex, m_pOP->GetI32Const(Comp));
  5159. // Create GEP.
  5160. Value *pGEPIndices[2] = { m_pOP->GetU32Const(0), pValueIndex };
  5161. Value *pPtr = m_pBuilder->CreateGEP(m_pIcbGV, pGEPIndices);
  5162. LoadInst *pLoad = m_pBuilder->CreateLoad(pPtr);
  5163. pLoad->setAlignment(kRegCompAlignment);
  5164. Value *pValue = CastDxbcValue(pLoad, CompType::getF32(), ValueType);
  5165. pValue = ApplyOperandModifiers(pValue, O);
  5166. OVH.SetValue(pValue);
  5167. }
  5168. } else {
  5169. // Double precision ICB.
  5170. for (OperandValueHelper OVH(SrcVal, Mask, O); !OVH.IsDone(); OVH.Advance()) {
  5171. BYTE Comp = OVH.GetComp();
  5172. Value *pValueIndex = m_pBuilder->CreateMul(pRegIndex, m_pOP->GetI32Const(DXBC::kWidth));
  5173. pValueIndex = m_pBuilder->CreateAdd(pValueIndex, m_pOP->GetI32Const(Comp));
  5174. // Bitcast pointer.
  5175. Value *pPtrBase = m_pBuilder->CreateBitCast(m_pIcbGV, Type::getDoublePtrTy(m_Ctx));
  5176. // Create GEP.
  5177. Value *pGEPIndices[1] = { pValueIndex };
  5178. Value *pPtr = m_pBuilder->CreateGEP(pPtrBase, pGEPIndices);
  5179. LoadInst *pLoad = m_pBuilder->CreateLoad(pPtr);
  5180. pLoad->setAlignment(kRegCompAlignment*2);
  5181. Value *pValue = pLoad;
  5182. pValue = ApplyOperandModifiers(pValue, O);
  5183. OVH.SetValue(pValue);
  5184. OVH.Advance();
  5185. OVH.SetValue(pValue);
  5186. }
  5187. }
  5188. break;
  5189. }
  5190. case D3D10_SB_OPERAND_TYPE_SAMPLER: {
  5191. // Upconvert operand to SM5.1.
  5192. if (O.m_IndexDimension == D3D10_SB_OPERAND_INDEX_1D) {
  5193. O.m_IndexDimension = D3D10_SB_OPERAND_INDEX_2D;
  5194. O.m_IndexType[1] = O.m_IndexType[0];
  5195. O.m_Index[1] = O.m_Index[0];
  5196. }
  5197. // Retrieve sampler range ID and record.
  5198. const DxilSampler* pR = nullptr;
  5199. if (O.m_IndexType[0] == D3D10_SB_OPERAND_INDEX_IMMEDIATE32) {
  5200. unsigned RangeID = O.m_Index[0].m_RegIndex;
  5201. unsigned RecIdx = m_SamplerRangeMap[RangeID];
  5202. pR = &m_pPR->GetSampler(RecIdx);
  5203. }
  5204. else {
  5205. switch (Inst.OpCode()) {
  5206. case D3D10_SB_OPCODE_SAMPLE_C:
  5207. case D3D10_SB_OPCODE_SAMPLE_C_LZ:
  5208. case D3DWDDM1_3_SB_OPCODE_SAMPLE_C_CLAMP_FEEDBACK:
  5209. case D3DWDDM1_3_SB_OPCODE_SAMPLE_C_LZ_FEEDBACK:
  5210. case D3D11_SB_OPCODE_GATHER4_PO_C:
  5211. case D3DWDDM1_3_SB_OPCODE_GATHER4_PO_C_FEEDBACK:
  5212. pR = m_pClassInstanceComparisonSamplers;
  5213. break;
  5214. default:
  5215. pR = m_pClassInstanceSamplers;
  5216. break;
  5217. }
  5218. }
  5219. const DxilSampler &R = *pR;
  5220. // Setup sampler handle.
  5221. Value *pHandle = GetCachedHandle(R);
  5222. if (pHandle == nullptr) {
  5223. // Create dynamic-index handle.
  5224. pHandle = CreateHandle(R.GetClass(), R.GetID(), LoadOperandIndex(O.m_Index[1], O.m_IndexType[1]), O.m_Nonuniform);
  5225. }
  5226. // Replicate handle values.
  5227. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  5228. if (Mask.IsSet(c))
  5229. SrcVal[c] = pHandle;
  5230. }
  5231. break;
  5232. }
  5233. case D3D10_SB_OPERAND_TYPE_RESOURCE: {
  5234. (void)LoadSRVOperand(SrcVal, Inst, OpIdx, Mask, ValueType);
  5235. break;
  5236. }
  5237. case D3D11_SB_OPERAND_TYPE_UNORDERED_ACCESS_VIEW: {
  5238. // Upconvert operand to SM5.1.
  5239. if (O.m_IndexDimension == D3D10_SB_OPERAND_INDEX_1D) {
  5240. DXASSERT_DXBC(O.m_IndexType[0] == D3D10_SB_OPERAND_INDEX_IMMEDIATE32);
  5241. O.m_IndexDimension = D3D10_SB_OPERAND_INDEX_2D;
  5242. O.m_IndexType[1] = O.m_IndexType[0];
  5243. O.m_Index[1] = O.m_Index[0];
  5244. }
  5245. // Retrieve UAV range ID and record.
  5246. DXASSERT_DXBC(O.m_IndexType[0] == D3D10_SB_OPERAND_INDEX_IMMEDIATE32);
  5247. unsigned RangeID = O.m_Index[0].m_RegIndex;
  5248. unsigned RecIdx = m_UAVRangeMap[RangeID];
  5249. const DxilResource &R = m_pPR->GetUAV(RecIdx);
  5250. // Setup UAV handle.
  5251. Value *pHandle = GetCachedHandle(R);
  5252. if (pHandle == nullptr) {
  5253. DXASSERT(IsSM51Plus(), "otherwise did not initialize handles on entry to main");
  5254. // Create dynamic-index handle.
  5255. pHandle = CreateHandle(R.GetClass(), R.GetID(), LoadOperandIndex(O.m_Index[1], O.m_IndexType[1]), O.m_Nonuniform);
  5256. }
  5257. // Replicate handle values.
  5258. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  5259. if (Mask.IsSet(c))
  5260. SrcVal[c] = pHandle;
  5261. }
  5262. break;
  5263. }
  5264. case D3D10_SB_OPERAND_TYPE_RASTERIZER: {
  5265. DXASSERT_DXBC(O.m_IndexDimension == D3D10_SB_OPERAND_INDEX_0D);
  5266. DXASSERT_DXBC(false); // "rasterizer" register is not used in DXIL.
  5267. break;
  5268. }
  5269. case D3D11_SB_OPERAND_TYPE_INPUT_THREAD_ID:
  5270. case D3D11_SB_OPERAND_TYPE_INPUT_THREAD_GROUP_ID:
  5271. case D3D11_SB_OPERAND_TYPE_INPUT_THREAD_ID_IN_GROUP: {
  5272. OP::OpCode OpCode;
  5273. switch (O.m_Type) {
  5274. case D3D11_SB_OPERAND_TYPE_INPUT_THREAD_ID: OpCode = OP::OpCode::ThreadId; break;
  5275. case D3D11_SB_OPERAND_TYPE_INPUT_THREAD_GROUP_ID: OpCode = OP::OpCode::GroupId; break;
  5276. case D3D11_SB_OPERAND_TYPE_INPUT_THREAD_ID_IN_GROUP: OpCode = OP::OpCode::ThreadIdInGroup; break;
  5277. }
  5278. CompType DxbcValueType = CompType::Kind::I32;
  5279. Type *pDxbcValueType = DxbcValueType.GetLLVMType(m_Ctx);
  5280. Function *F = m_pOP->GetOpFunc(OpCode, pDxbcValueType);
  5281. for (OperandValueHelper OVH(SrcVal, Mask, O); !OVH.IsDone(); OVH.Advance()) {
  5282. BYTE Comp = OVH.GetComp();
  5283. Value *Args[2];
  5284. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  5285. Args[1] = m_pOP->GetU32Const(Comp); // Component: x,y,z
  5286. Value *pValue = m_pBuilder->CreateCall(F, Args);
  5287. pValue = CastDxbcValue(pValue, DxbcValueType, ValueType);
  5288. pValue = ApplyOperandModifiers(pValue, O);
  5289. OVH.SetValue(pValue);
  5290. }
  5291. break;
  5292. }
  5293. case D3D11_SB_OPERAND_TYPE_INPUT_THREAD_ID_IN_GROUP_FLATTENED: {
  5294. OP::OpCode OpCode = OP::OpCode::FlattenedThreadIdInGroup;
  5295. CompType DxbcValueType = CompType::Kind::I32;
  5296. Type *pDxbcValueType = DxbcValueType.GetLLVMType(m_Ctx);
  5297. Function *F = m_pOP->GetOpFunc(OpCode, pDxbcValueType);
  5298. for (OperandValueHelper OVH(SrcVal, Mask, O); !OVH.IsDone(); OVH.Advance()) {
  5299. Value *Args[1];
  5300. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  5301. Value *pValue = m_pBuilder->CreateCall(F, Args);
  5302. pValue = CastDxbcValue(pValue, DxbcValueType, ValueType);
  5303. pValue = ApplyOperandModifiers(pValue, O);
  5304. OVH.SetValue(pValue);
  5305. }
  5306. break;
  5307. }
  5308. case D3D11_SB_OPERAND_TYPE_INPUT_PATCH_CONSTANT: {
  5309. DXASSERT_DXBC(O.m_IndexDimension == D3D10_SB_OPERAND_INDEX_1D);
  5310. unsigned Register = O.m_Index[0].m_RegIndex;
  5311. Value *pRowIndexValue = LoadOperandIndex(O.m_Index[0], O.m_IndexType[0]);
  5312. for (OperandValueHelper OVH(SrcVal, Mask, O); !OVH.IsDone(); OVH.Advance()) {
  5313. BYTE Comp = OVH.GetComp();
  5314. // Retrieve signature element.
  5315. const DxilSignatureElement *E = m_pPatchConstantSignature->GetElement(Register, Comp);
  5316. CompType DxbcValueType = E->GetCompType();
  5317. if (DxbcValueType.IsBoolTy()) {
  5318. DxbcValueType = CompType::getI32();
  5319. }
  5320. Type *pDxbcValueType = DxbcValueType.GetLLVMType(m_Ctx);
  5321. // Make row/col index relative within element.
  5322. Value *pRowIndexValueRel = m_pBuilder->CreateSub(pRowIndexValue, m_pOP->GetU32Const(E->GetStartRow()));
  5323. Value *Args[4];
  5324. Args[0] = m_pOP->GetU32Const((unsigned)OP::OpCode::LoadPatchConstant); // OpCode
  5325. Args[1] = m_pOP->GetU32Const(E->GetID()); // Patch constant signature element ID
  5326. Args[2] = pRowIndexValueRel; // Row, relative to the element
  5327. Args[3] = m_pOP->GetU8Const(Comp - E->GetStartCol()); // Col, relative to the element
  5328. Function *F = m_pOP->GetOpFunc(OP::OpCode::LoadPatchConstant, pDxbcValueType);
  5329. Value *pValue = m_pBuilder->CreateCall(F, Args);
  5330. pValue = CastDxbcValue(pValue, DxbcValueType, ValueType);
  5331. pValue = ApplyOperandModifiers(pValue, O);
  5332. OVH.SetValue(pValue);
  5333. }
  5334. break;
  5335. }
  5336. case D3D11_SB_OPERAND_TYPE_OUTPUT_CONTROL_POINT: {
  5337. DXASSERT_DXBC(O.m_IndexDimension == D3D10_SB_OPERAND_INDEX_2D);
  5338. OP::OpCode OpCode = OP::OpCode::LoadOutputControlPoint;
  5339. unsigned Register = O.m_Index[1].m_RegIndex; // Starting index of the register range.
  5340. Value *pUnitIndexValue = LoadOperandIndex(O.m_Index[0], O.m_IndexType[0]); // Vertex/point index expression.
  5341. Value *pRowIndexValue = LoadOperandIndex(O.m_Index[1], O.m_IndexType[1]); // Row index expression.
  5342. for (OperandValueHelper OVH(SrcVal, Mask, O); !OVH.IsDone(); OVH.Advance()) {
  5343. BYTE Comp = OVH.GetComp();
  5344. // Retrieve signature element.
  5345. const DxilSignatureElement *E = m_pOutputSignature->GetElement(Register, Comp);
  5346. CompType DxbcValueType = E->GetCompType();
  5347. if (DxbcValueType.IsBoolTy()) {
  5348. DxbcValueType = CompType::getI32();
  5349. }
  5350. Type *pDxbcValueType = DxbcValueType.GetLLVMType(m_Ctx);
  5351. // Make row/col index relative within element.
  5352. Value *pRowIndexValueRel = m_pBuilder->CreateSub(pRowIndexValue, m_pOP->GetU32Const(E->GetStartRow()));
  5353. Value *Args[5];
  5354. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  5355. Args[1] = m_pOP->GetU32Const(E->GetID()); // Output signature element ID
  5356. Args[2] = pRowIndexValueRel; // Row, relative to the element
  5357. Args[3] = m_pOP->GetU8Const(Comp - E->GetStartCol()); // Col, relative to the element
  5358. Args[4] = pUnitIndexValue; // Vertex/point index
  5359. Function *F = m_pOP->GetOpFunc(OpCode, pDxbcValueType);
  5360. Value *pValue = m_pBuilder->CreateCall(F, Args);
  5361. pValue = CastDxbcValue(pValue, DxbcValueType, ValueType);
  5362. pValue = ApplyOperandModifiers(pValue, O);
  5363. OVH.SetValue(pValue);
  5364. }
  5365. break;
  5366. }
  5367. case D3D11_SB_OPERAND_TYPE_INPUT_DOMAIN_POINT: {
  5368. OP::OpCode OpCode = OP::OpCode::DomainLocation;
  5369. CompType DxbcValueType = CompType::Kind::F32;
  5370. Type *pDxbcValueType = DxbcValueType.GetLLVMType(m_Ctx);
  5371. Function *F = m_pOP->GetOpFunc(OpCode, pDxbcValueType);
  5372. for (OperandValueHelper OVH(SrcVal, Mask, O); !OVH.IsDone(); OVH.Advance()) {
  5373. BYTE Comp = OVH.GetComp();
  5374. Value *Args[2];
  5375. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  5376. Args[1] = m_pOP->GetU8Const(Comp); // Component
  5377. Value *pValue = m_pBuilder->CreateCall(F, Args);
  5378. pValue = CastDxbcValue(pValue, DxbcValueType, ValueType);
  5379. pValue = ApplyOperandModifiers(pValue, O);
  5380. OVH.SetValue(pValue);
  5381. }
  5382. break;
  5383. }
  5384. case D3D11_SB_OPERAND_TYPE_OUTPUT_CONTROL_POINT_ID:
  5385. case D3D10_SB_OPERAND_TYPE_INPUT_PRIMITIVEID:
  5386. case D3D11_SB_OPERAND_TYPE_INPUT_GS_INSTANCE_ID:
  5387. case D3D11_SB_OPERAND_TYPE_INPUT_COVERAGE_MASK:
  5388. case D3D11_SB_OPERAND_TYPE_INNER_COVERAGE: {
  5389. OP::OpCode OpCode;
  5390. switch (O.m_Type) {
  5391. case D3D11_SB_OPERAND_TYPE_OUTPUT_CONTROL_POINT_ID: OpCode = OP::OpCode::OutputControlPointID; break;
  5392. case D3D10_SB_OPERAND_TYPE_INPUT_PRIMITIVEID: OpCode = OP::OpCode::PrimitiveID; break;
  5393. case D3D11_SB_OPERAND_TYPE_INPUT_GS_INSTANCE_ID: OpCode = OP::OpCode::GSInstanceID; break;
  5394. case D3D11_SB_OPERAND_TYPE_INPUT_COVERAGE_MASK: OpCode = OP::OpCode::Coverage; break;
  5395. case D3D11_SB_OPERAND_TYPE_INNER_COVERAGE: OpCode = OP::OpCode::InnerCoverage; break;
  5396. }
  5397. CompType DxbcValueType = CompType::Kind::I32;
  5398. Type *pDxbcValueType = DxbcValueType.GetLLVMType(m_Ctx);
  5399. Function *F = m_pOP->GetOpFunc(OpCode, pDxbcValueType);
  5400. Value *Args[1];
  5401. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  5402. Value *pValue = m_pBuilder->CreateCall(F, Args);
  5403. pValue = CastDxbcValue(pValue, DxbcValueType, ValueType);
  5404. pValue = ApplyOperandModifiers(pValue, O);
  5405. for (OperandValueHelper OVH(SrcVal, Mask, O); !OVH.IsDone(); OVH.Advance()) {
  5406. OVH.SetValue(pValue);
  5407. }
  5408. break;
  5409. }
  5410. case D3D11_SB_OPERAND_TYPE_CYCLE_COUNTER: {
  5411. OP::OpCode OpCode = OP::OpCode::CycleCounterLegacy;
  5412. Function *F = m_pOP->GetOpFunc(OpCode, Type::getVoidTy(m_Ctx));
  5413. Value *Args[1];
  5414. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  5415. Value *pValue = m_pBuilder->CreateCall(F, Args);
  5416. for (OperandValueHelper OVH(SrcVal, Mask, O); !OVH.IsDone(); OVH.Advance()) {
  5417. BYTE c = OVH.GetComp();
  5418. switch (c) {
  5419. case 0: {
  5420. Value *pLo32 = m_pBuilder->CreateExtractValue(pValue, 0);
  5421. pLo32 = CastDxbcValue(pLo32, CompType::Kind::I32, ValueType);
  5422. OVH.SetValue(pLo32);
  5423. break;
  5424. }
  5425. case 1: {
  5426. Value *pHi32 = m_pBuilder->CreateExtractValue(pValue, 1);
  5427. pHi32 = CastDxbcValue(pHi32, CompType::Kind::I32, ValueType);
  5428. OVH.SetValue(pHi32);
  5429. break;
  5430. }
  5431. default:
  5432. OVH.SetValue(m_pOP->GetU32Const(0));
  5433. }
  5434. }
  5435. break;
  5436. }
  5437. case D3D11_SB_OPERAND_TYPE_INPUT_FORK_INSTANCE_ID:
  5438. case D3D11_SB_OPERAND_TYPE_INPUT_JOIN_INSTANCE_ID: {
  5439. Scope &HullScope = m_ScopeStack.FindParentHullLoop();
  5440. Value *pValue = m_pBuilder->CreateLoad(HullScope.pInductionVar);
  5441. pValue = ApplyOperandModifiers(pValue, O);
  5442. for (OperandValueHelper OVH(SrcVal, Mask, O); !OVH.IsDone(); OVH.Advance()) {
  5443. OVH.SetValue(pValue);
  5444. }
  5445. break;
  5446. }
  5447. case D3D11_SB_OPERAND_TYPE_THIS_POINTER: {
  5448. Value *pIfaceIdx = LoadOperandIndex(O.m_Index[0], O.m_IndexType[0]);
  5449. // The CBuffer layout here is a UINT for the interface class type selection, then 3 UINTs padding, per interface.
  5450. // After that, there's another 4 UINTs per interface which defines the "this" pointer data.
  5451. // Note, legacy CBuffer loads address their data in number of 4-float constants, not bytes or single elements.
  5452. // Since the "this" data comes after 4 UINTs per interface, adjust the CB offset just by the number of interfaces.
  5453. Value* pCBOffset = m_pBuilder->CreateAdd(m_pOP->GetU32Const(m_NumIfaces), pIfaceIdx);
  5454. Value *Args[3];
  5455. Args[0] = m_pOP->GetU32Const((unsigned)OP::OpCode::CBufferLoadLegacy); // OpCode
  5456. Args[1] = CreateHandle(m_pInterfaceDataBuffer->GetClass(),
  5457. m_pInterfaceDataBuffer->GetID(),
  5458. m_pOP->GetU32Const(m_pInterfaceDataBuffer->GetLowerBound()),
  5459. false /*Nonuniform*/); // CBuffer handle
  5460. Args[2] = pCBOffset; // 0-based index into cbuffer instance
  5461. Function *pCBufferLoadFunc = m_pOP->GetOpFunc(OP::OpCode::CBufferLoadLegacy, Type::getInt32Ty(m_Ctx));
  5462. Value* pCBufferRetValue = m_pBuilder->CreateCall(pCBufferLoadFunc, Args);
  5463. for (OperandValueHelper OVH(SrcVal, Mask, O); !OVH.IsDone(); OVH.Advance()) {
  5464. BYTE Comp = OVH.GetComp();
  5465. Value *pValue = m_pBuilder->CreateExtractValue(pCBufferRetValue, Comp);
  5466. pValue = CastDxbcValue(pValue, CompType::Kind::I32, ValueType);
  5467. pValue = ApplyOperandModifiers(pValue, O);
  5468. OVH.SetValue(pValue);
  5469. }
  5470. break;
  5471. }
  5472. default:
  5473. DXASSERT_ARGS(false, "Operand type %u is not yet implemented", O.m_Type);
  5474. }
  5475. }
  5476. const DxilResource& DxbcConverter::LoadSRVOperand(OperandValue &SrcVal,
  5477. D3D10ShaderBinary::CInstruction &Inst,
  5478. const unsigned OpIdx,
  5479. const CMask &Mask,
  5480. const CompType &ValueType) {
  5481. D3D10ShaderBinary::COperandBase &O = Inst.m_Operands[OpIdx];
  5482. DXASSERT(O.m_Type == D3D10_SB_OPERAND_TYPE_RESOURCE, "LoadSRVOperand should only be called for SRV operands.");
  5483. const DxilResource &R = GetSRVFromOperand(Inst, OpIdx);
  5484. // Setup SRV handle.
  5485. Value *pHandle = GetCachedHandle(R);
  5486. if (pHandle == nullptr) {
  5487. // Create dynamic-index handle.
  5488. pHandle = CreateHandle(R.GetClass(), R.GetID(), LoadOperandIndex(O.m_Index[1], O.m_IndexType[1]), O.m_Nonuniform);
  5489. }
  5490. // Replicate handle values.
  5491. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  5492. if (Mask.IsSet(c))
  5493. SrcVal[c] = pHandle;
  5494. }
  5495. return R;
  5496. }
  5497. const DxilResource& DxbcConverter::GetSRVFromOperand(D3D10ShaderBinary::CInstruction &Inst,
  5498. const unsigned OpIdx) {
  5499. D3D10ShaderBinary::COperandBase &O = Inst.m_Operands[OpIdx];
  5500. DXASSERT(O.m_Type == D3D10_SB_OPERAND_TYPE_RESOURCE, "GetSRVFromOperand should only be called for SRV operands.");
  5501. // Upconvert operand to SM5.1.
  5502. if (O.m_IndexDimension == D3D10_SB_OPERAND_INDEX_1D) {
  5503. O.m_IndexDimension = D3D10_SB_OPERAND_INDEX_2D;
  5504. O.m_IndexType[1] = O.m_IndexType[0];
  5505. O.m_Index[1] = O.m_Index[0];
  5506. }
  5507. // Retrieve SRV range ID and record.
  5508. if (O.m_IndexType[0] == D3D10_SB_OPERAND_INDEX_IMMEDIATE32) {
  5509. unsigned RangeID = O.m_Index[0].m_RegIndex;
  5510. unsigned RecIdx = m_SRVRangeMap[RangeID];
  5511. return m_pPR->GetSRV(RecIdx);
  5512. }
  5513. else {
  5514. return GetInterfacesSRVDecl(Inst);
  5515. }
  5516. }
  5517. void DxbcConverter::StoreOperand(OperandValue &DstVal,
  5518. const D3D10ShaderBinary::CInstruction &Inst,
  5519. const unsigned OpIdx,
  5520. const CMask &Mask,
  5521. const CompType &ValueType) {
  5522. const D3D10ShaderBinary::COperandBase &O = Inst.m_Operands[OpIdx];
  5523. // Mark value as precise, if needed.
  5524. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  5525. Value *pValue = DstVal[c];
  5526. if (pValue != nullptr)
  5527. DstVal[c] = MarkPrecise(DstVal[c], c);
  5528. }
  5529. ApplyInstructionModifiers(DstVal, Inst);
  5530. switch (O.m_Type) {
  5531. case D3D10_SB_OPERAND_TYPE_TEMP: {
  5532. DXASSERT_DXBC(O.m_IndexDimension == D3D10_SB_OPERAND_INDEX_1D);
  5533. unsigned Reg = O.m_Index[0].m_RegIndex;
  5534. CompType DxbcValueType = DXBC::GetCompTypeFromMinPrec(O.m_MinPrecision, ValueType);
  5535. if (DxbcValueType.IsBoolTy()) {
  5536. DxbcValueType = CompType::getI32();
  5537. }
  5538. Type *pDxbcValueType = DxbcValueType.GetLLVMType(m_Ctx);
  5539. if (DxbcValueType.GetKind() != CompType::Kind::F64) {
  5540. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  5541. if (!Mask.IsSet(c)) continue;
  5542. Value *Args[3];
  5543. Args[0] = m_pOP->GetU32Const((unsigned)OP::OpCode::TempRegStore); // OpCode
  5544. Args[1] = m_pOP->GetU32Const(DXBC::GetRegIndex(Reg, c)); // Linearized register index
  5545. Args[2] = MarkPrecise(CastDxbcValue(DstVal[c], ValueType, DxbcValueType), c); // Value
  5546. Function *F = m_pOP->GetOpFunc(OP::OpCode::TempRegStore, pDxbcValueType);
  5547. MarkPrecise(m_pBuilder->CreateCall(F, Args));
  5548. }
  5549. } else {
  5550. for (BYTE c = 0; c < DXBC::kWidth; c += 2) {
  5551. if (!Mask.IsSet(c)) continue;
  5552. Value *pSDT; // Split double type.
  5553. {
  5554. Value *Args[2];
  5555. Args[0] = m_pOP->GetU32Const((unsigned)OP::OpCode::SplitDouble); // OpCode
  5556. Args[1] = DstVal[c]; // Double value
  5557. Function *F = m_pOP->GetOpFunc(OP::OpCode::SplitDouble, pDxbcValueType);
  5558. pSDT = MarkPrecise(m_pBuilder->CreateCall(F, Args), c);
  5559. }
  5560. Value *Args[3];
  5561. Args[0] = m_pOP->GetU32Const((unsigned)OP::OpCode::TempRegStore); // OpCode
  5562. Args[1] = m_pOP->GetU32Const(DXBC::GetRegIndex(Reg, c)); // Linearized register index 1
  5563. Args[2] = MarkPrecise(m_pBuilder->CreateExtractValue(pSDT, 0), c); // Value to store
  5564. Function *F = m_pOP->GetOpFunc(OP::OpCode::TempRegStore, Type::getInt32Ty(m_Ctx));
  5565. Value *pVal = m_pBuilder->CreateCall(F, Args);
  5566. MarkPrecise(pVal, c);
  5567. Args[1] = m_pOP->GetU32Const(DXBC::GetRegIndex(Reg, c+1)); // Linearized register index 2
  5568. Args[2] = MarkPrecise(m_pBuilder->CreateExtractValue(pSDT, 1), c+1);// Value to store
  5569. MarkPrecise(m_pBuilder->CreateCall(F, Args));
  5570. }
  5571. }
  5572. break;
  5573. }
  5574. case D3D10_SB_OPERAND_TYPE_INDEXABLE_TEMP: {
  5575. DXASSERT_DXBC(O.m_IndexDimension == D3D10_SB_OPERAND_INDEX_2D);
  5576. DXASSERT_DXBC(O.m_IndexType[0] == D3D10_SB_OPERAND_INDEX_IMMEDIATE32);
  5577. unsigned Reg = O.m_Index[0].m_RegIndex;
  5578. IndexableReg &IRRec = m_IndexableRegs[Reg];
  5579. Value *pXRegIndex = LoadOperandIndex(O.m_Index[1], O.m_IndexType[1]);
  5580. Value *pRegIndex = m_pBuilder->CreateMul(pXRegIndex, m_pOP->GetI32Const(IRRec.NumComps));
  5581. CompType DxbcValueType = DXBC::GetCompTypeFromMinPrec(O.m_MinPrecision, ValueType);
  5582. if (DxbcValueType.IsBoolTy()) {
  5583. DxbcValueType = CompType::getI32();
  5584. }
  5585. if (DxbcValueType.GetKind() != CompType::Kind::F64) {
  5586. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  5587. if (!Mask.IsSet(c)) continue;
  5588. // Create GEP.
  5589. Value *pIndex = m_pBuilder->CreateAdd(pRegIndex, m_pOP->GetU32Const(c));
  5590. Value *pGEPIndices[2] = { m_pOP->GetU32Const(0), pIndex };
  5591. if (!DxbcValueType.HasMinPrec()) {
  5592. Value *pBasePtr = m_IndexableRegs[Reg].pValue32;
  5593. Value *pPtr = m_pBuilder->CreateGEP(pBasePtr, pGEPIndices);
  5594. Value *pValue = MarkPrecise(CastDxbcValue(DstVal[c], ValueType, CompType::getF32()), c);
  5595. MarkPrecise(m_pBuilder->CreateAlignedStore(pValue, pPtr, kRegCompAlignment), c);
  5596. } else {
  5597. Value *pBasePtr = m_IndexableRegs[Reg].pValue16;
  5598. Value *pPtr = m_pBuilder->CreateGEP(pBasePtr, pGEPIndices);
  5599. Value *pValue = MarkPrecise(CastDxbcValue(DstVal[c], ValueType, CompType::getF16()), c);
  5600. MarkPrecise(m_pBuilder->CreateAlignedStore(pValue, pPtr, kRegCompAlignment/2), c);
  5601. }
  5602. }
  5603. } else {
  5604. // Double precision.
  5605. for (BYTE c = 0; c < DXBC::kWidth; c += 2) {
  5606. if (!Mask.IsSet(c)) continue;
  5607. // Create GEP.
  5608. Value *pIndex = m_pBuilder->CreateAdd(pRegIndex, m_pOP->GetU32Const(c));
  5609. Value *pGEPIndices[] = { pIndex };
  5610. Value *pBasePtr = m_pBuilder->CreateBitCast(m_IndexableRegs[Reg].pValue32, Type::getDoublePtrTy(m_Ctx));
  5611. Value *pPtr = m_pBuilder->CreateGEP(pBasePtr, pGEPIndices);
  5612. MarkPrecise(m_pBuilder->CreateAlignedStore(DstVal[c], pPtr, kRegCompAlignment*2));
  5613. }
  5614. }
  5615. break;
  5616. }
  5617. case D3D10_SB_OPERAND_TYPE_OUTPUT: {
  5618. unsigned Reg = O.m_Index[0].m_RegIndex;
  5619. // Row index expression.
  5620. Value *pRowIndexValue = LoadOperandIndex(O.m_Index[0], O.m_IndexType[0]);
  5621. bool bStoreOutputReg = !(m_pSM->IsGS() && m_pPR->HasMultipleOutputStreams());
  5622. if (bStoreOutputReg) {
  5623. for (unsigned c = 0; c < DXBC::kWidth; c++) {
  5624. if (!Mask.IsSet(c)) continue;
  5625. // Retrieve signature element.
  5626. OP::OpCode OpCode;
  5627. const DxilSignatureElement *E;
  5628. if (!m_bPatchConstantPhase) {
  5629. E = m_pOutputSignature->GetElementWithStream(Reg, c, m_pPR->GetOutputStream());
  5630. OpCode = OP::OpCode::StoreOutput;
  5631. } else {
  5632. E = m_pPatchConstantSignature->GetElementWithStream(Reg, c, m_pPR->GetOutputStream());
  5633. OpCode = OP::OpCode::StorePatchConstant;
  5634. }
  5635. CompType DxbcValueType = E->GetCompType();
  5636. if (DxbcValueType.IsBoolTy()) {
  5637. DxbcValueType = CompType::getI32();
  5638. }
  5639. Type *pLlvmDxbcValueType = DxbcValueType.GetLLVMType(m_Ctx);
  5640. // Make row index relative within element.
  5641. Value *pRowIndexValueRel = m_pBuilder->CreateSub(pRowIndexValue, m_pOP->GetU32Const(E->GetStartRow()));
  5642. Value *Args[5];
  5643. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  5644. Args[1] = m_pOP->GetU32Const(E->GetID()); // Output signature element ID
  5645. Args[2] = pRowIndexValueRel; // Row, relative to the element
  5646. Args[3] = m_pOP->GetU8Const(c - E->GetStartCol()); // Col, relative to the element
  5647. Args[4] = MarkPrecise(CastDxbcValue(DstVal[c], ValueType, DxbcValueType), c); // Value
  5648. Function *F = m_pOP->GetOpFunc(OpCode, pLlvmDxbcValueType);
  5649. MarkPrecise(m_pBuilder->CreateCall(F, Args));
  5650. }
  5651. } else {
  5652. // In GS with multiple streams, output register file is shared among the streams.
  5653. // Store the values into additional temp registers, and later, store these at the emit points.
  5654. CompType DxbcValueType = DXBC::GetCompTypeFromMinPrec(O.m_MinPrecision, ValueType);
  5655. if (DxbcValueType.IsBoolTy()) {
  5656. DxbcValueType = CompType::getI32();
  5657. }
  5658. Type *pDxbcValueType = DxbcValueType.GetLLVMType(m_Ctx);
  5659. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  5660. if (!Mask.IsSet(c)) continue;
  5661. Value *Args[3];
  5662. Args[0] = m_pOP->GetU32Const((unsigned)OP::OpCode::TempRegStore); // OpCode
  5663. unsigned TempReg = GetGSTempRegForOutputReg(Reg);
  5664. Args[1] = m_pOP->GetU32Const(DXBC::GetRegIndex(TempReg, c)); // Linearized register index
  5665. Args[2] = MarkPrecise(CastDxbcValue(DstVal[c], ValueType, DxbcValueType), c); // Value to store
  5666. Function *F = m_pOP->GetOpFunc(OP::OpCode::TempRegStore, pDxbcValueType);
  5667. MarkPrecise(m_pBuilder->CreateCall(F, Args));
  5668. }
  5669. }
  5670. break;
  5671. }
  5672. case D3D10_SB_OPERAND_TYPE_OUTPUT_DEPTH:
  5673. case D3D11_SB_OPERAND_TYPE_OUTPUT_DEPTH_GREATER_EQUAL:
  5674. case D3D11_SB_OPERAND_TYPE_OUTPUT_DEPTH_LESS_EQUAL:
  5675. case D3D11_SB_OPERAND_TYPE_OUTPUT_STENCIL_REF:
  5676. case D3D10_SB_OPERAND_TYPE_OUTPUT_COVERAGE_MASK: {
  5677. DXASSERT_DXBC(O.m_IndexDimension == D3D10_SB_OPERAND_INDEX_0D);
  5678. for (unsigned c = 0; c < DXBC::kWidth; c++) {
  5679. if (!Mask.IsSet(c)) continue;
  5680. // Retrieve signature element.
  5681. DXASSERT(m_pSM->IsPS(), "PS has only one output stream.");
  5682. const DxilSignatureElement *E = m_pOutputSignature->GetElement(O.m_Type);
  5683. CompType DxbcValueType = E->GetCompType();
  5684. Type *pLlvmDxbcValueType = DxbcValueType.GetLLVMType(m_Ctx);
  5685. Value *Args[5];
  5686. Args[0] = m_pOP->GetU32Const((unsigned)OP::OpCode::StoreOutput); // OpCode
  5687. Args[1] = m_pOP->GetU32Const(E->GetID()); // Output signature element ID
  5688. Args[2] = m_pOP->GetU32Const(0); // Row, relative to the element
  5689. Args[3] = m_pOP->GetU8Const(c - E->GetStartCol()); // Col, relative to the element
  5690. Args[4] = MarkPrecise(CastDxbcValue(DstVal[c], ValueType, DxbcValueType), c); // Value
  5691. Function *F = m_pOP->GetOpFunc(OP::OpCode::StoreOutput, pLlvmDxbcValueType);
  5692. MarkPrecise(m_pBuilder->CreateCall(F, Args));
  5693. }
  5694. break;
  5695. }
  5696. case D3D10_SB_OPERAND_TYPE_NULL:
  5697. break;
  5698. default:
  5699. DXASSERT_ARGS(false, "Operand type %u is not yet implemented", O.m_Type);
  5700. }
  5701. }
  5702. Value *DxbcConverter::LoadOperandIndex(const D3D10ShaderBinary::COperandIndex &OpIndex,
  5703. const D3D10_SB_OPERAND_INDEX_REPRESENTATION IndexType) {
  5704. Value *pValue = nullptr;
  5705. switch (IndexType) {
  5706. case D3D10_SB_OPERAND_INDEX_IMMEDIATE32:
  5707. DXASSERT_DXBC(OpIndex.m_RelRegType == D3D10_SB_OPERAND_TYPE_IMMEDIATE32);
  5708. pValue = m_pOP->GetU32Const(OpIndex.m_RegIndex);
  5709. break;
  5710. case D3D10_SB_OPERAND_INDEX_IMMEDIATE64:
  5711. DXASSERT_DXBC(false);
  5712. break;
  5713. case D3D10_SB_OPERAND_INDEX_RELATIVE:
  5714. pValue = LoadOperandIndexRelative(OpIndex);
  5715. break;
  5716. case D3D10_SB_OPERAND_INDEX_IMMEDIATE32_PLUS_RELATIVE: {
  5717. unsigned Offset = OpIndex.m_RegIndex;
  5718. pValue = LoadOperandIndexRelative(OpIndex);
  5719. if (Offset != 0) {
  5720. pValue = m_pBuilder->CreateAdd(pValue, m_pOP->GetU32Const(Offset));
  5721. }
  5722. break;
  5723. }
  5724. case D3D10_SB_OPERAND_INDEX_IMMEDIATE64_PLUS_RELATIVE:
  5725. DXASSERT_DXBC(false);
  5726. break;
  5727. default:
  5728. DXASSERT_DXBC(false);
  5729. break;
  5730. }
  5731. return pValue;
  5732. }
  5733. Value *DxbcConverter::LoadOperandIndexRelative(const D3D10ShaderBinary::COperandIndex &OpIndex) {
  5734. Value *pValue = nullptr;
  5735. switch (OpIndex.m_RelRegType) {
  5736. case D3D10_SB_OPERAND_TYPE_TEMP: {
  5737. unsigned Reg = OpIndex.m_RelIndex;
  5738. unsigned Comp = OpIndex.m_ComponentName;
  5739. Value *Args[2];
  5740. Args[0] = m_pOP->GetU32Const((unsigned)OP::OpCode::TempRegLoad); // OpCode
  5741. Args[1] = m_pOP->GetU32Const(DXBC::GetRegIndex(Reg, Comp)); // Linearized register index
  5742. Function *F = m_pOP->GetOpFunc(OP::OpCode::TempRegLoad, Type::getInt32Ty(m_Ctx));
  5743. pValue = m_pBuilder->CreateCall(F, Args);
  5744. break;
  5745. }
  5746. case D3D10_SB_OPERAND_TYPE_INDEXABLE_TEMP: {
  5747. unsigned Reg = OpIndex.m_RelIndex;
  5748. unsigned RegIdx = OpIndex.m_RelIndex1;
  5749. unsigned Comp = OpIndex.m_ComponentName;
  5750. IndexableReg &IRRec = m_IndexableRegs[Reg];
  5751. Value *pGEPIndices[2] = { m_pOP->GetU32Const(0), m_pOP->GetU32Const(RegIdx*IRRec.NumComps + Comp) };
  5752. Value *pBasePtr = m_IndexableRegs[Reg].pValue32;
  5753. Value *pPtr = m_pBuilder->CreateGEP(pBasePtr, pGEPIndices);
  5754. pValue = m_pBuilder->CreateAlignedLoad(pPtr, kRegCompAlignment);
  5755. DXASSERT(pValue->getType()->isFloatTy(), "otherwise broke the assumption that alloca locations are floats");
  5756. pValue = CastDxbcValue(pValue, CompType::getF32(), CompType::getI32());
  5757. break;
  5758. }
  5759. default:
  5760. DXASSERT_DXBC(false);
  5761. }
  5762. return pValue;
  5763. }
  5764. Value *DxbcConverter::CastDxbcValue(Value *pValue, const CompType &SrcType, const CompType &DstType) {
  5765. if (SrcType == DstType)
  5766. return pValue;
  5767. DXASSERT(SrcType.GetLLVMType(m_Ctx) == pValue->getType(), "otherwise caller passed incorrect args");
  5768. switch (SrcType.GetKind()) {
  5769. case CompType::Kind::I1:
  5770. switch (DstType.GetKind()) {
  5771. case CompType::Kind::I1:
  5772. return pValue;
  5773. case CompType::Kind::I16:
  5774. case CompType::Kind::U16:
  5775. return m_pBuilder->CreateSExt(pValue, Type::getInt16Ty(m_Ctx));
  5776. case CompType::Kind::I32:
  5777. case CompType::Kind::U32:
  5778. return m_pBuilder->CreateSExt(pValue, Type::getInt32Ty(m_Ctx));
  5779. case CompType::Kind::F16:
  5780. return m_pBuilder->CreateBitCast(m_pBuilder->CreateSExt(pValue, Type::getInt16Ty(m_Ctx)), Type::getHalfTy(m_Ctx));
  5781. case CompType::Kind::F32:
  5782. return m_pBuilder->CreateBitCast(m_pBuilder->CreateSExt(pValue, Type::getInt32Ty(m_Ctx)), Type::getFloatTy(m_Ctx));
  5783. default: __fallthrough;
  5784. }
  5785. break;
  5786. case CompType::Kind::I16:
  5787. switch (DstType.GetKind()) {
  5788. case CompType::Kind::I1:
  5789. return m_pBuilder->CreateICmpNE(pValue, m_pOP->GetI16Const(0));
  5790. case CompType::Kind::U16:
  5791. DXASSERT_DXBC(false);
  5792. return pValue;
  5793. case CompType::Kind::I32:
  5794. case CompType::Kind::U32:
  5795. return m_pBuilder->CreateSExt(pValue, Type::getInt32Ty(m_Ctx));
  5796. case CompType::Kind::F16: {
  5797. DXASSERT_DXBC(false);
  5798. pValue = m_pBuilder->CreateSExt(pValue, Type::getInt32Ty(m_Ctx));
  5799. pValue = CreateBitCast(pValue, CompType::getI32(), CompType::getF32());
  5800. return m_pBuilder->CreateFPTrunc(pValue, Type::getHalfTy(m_Ctx));
  5801. }
  5802. case CompType::Kind::F32: { // mov
  5803. pValue = m_pBuilder->CreateSExt(pValue, Type::getInt32Ty(m_Ctx));
  5804. return CreateBitCast(pValue, CompType::getI32(), CompType::getF32());
  5805. }
  5806. default: __fallthrough;
  5807. }
  5808. break;
  5809. case CompType::Kind::U16:
  5810. switch (DstType.GetKind()) {
  5811. case CompType::Kind::I1:
  5812. return m_pBuilder->CreateICmpNE(pValue, m_pOP->GetU16Const(0));
  5813. case CompType::Kind::I16:
  5814. DXASSERT_DXBC(false);
  5815. return pValue;
  5816. case CompType::Kind::I32:
  5817. case CompType::Kind::U32:
  5818. return m_pBuilder->CreateZExt(pValue, Type::getInt32Ty(m_Ctx));
  5819. case CompType::Kind::F16: {
  5820. DXASSERT_DXBC(false);
  5821. pValue = m_pBuilder->CreateZExt(pValue, Type::getInt32Ty(m_Ctx));
  5822. pValue = CreateBitCast(pValue, CompType::getI32(), CompType::getF32());
  5823. return m_pBuilder->CreateFPTrunc(pValue, Type::getHalfTy(m_Ctx));
  5824. }
  5825. case CompType::Kind::F32: { // mov
  5826. pValue = m_pBuilder->CreateZExt(pValue, Type::getInt32Ty(m_Ctx));
  5827. return CreateBitCast(pValue, CompType::getI32(), CompType::getF32());
  5828. }
  5829. default: __fallthrough;
  5830. }
  5831. break;
  5832. case CompType::Kind::I32:
  5833. case CompType::Kind::U32:
  5834. switch (DstType.GetKind()) {
  5835. case CompType::Kind::I1:
  5836. return m_pBuilder->CreateICmpNE(pValue, m_pOP->GetI32Const(0));
  5837. case CompType::Kind::I16:
  5838. case CompType::Kind::U16:
  5839. return m_pBuilder->CreateTrunc(pValue, Type::getInt16Ty(m_Ctx));
  5840. case CompType::Kind::I32:
  5841. case CompType::Kind::U32:
  5842. return pValue;
  5843. case CompType::Kind::F16: {
  5844. DXASSERT_DXBC(false);
  5845. pValue = CreateBitCast(pValue, CompType::getI32(), CompType::getF32());
  5846. return m_pBuilder->CreateFPTrunc(pValue, Type::getHalfTy(m_Ctx));
  5847. }
  5848. case CompType::Kind::F32:
  5849. return CreateBitCast(pValue, CompType::getI32(), CompType::getF32());
  5850. default: __fallthrough;
  5851. }
  5852. break;
  5853. case CompType::Kind::F16:
  5854. switch (DstType.GetKind()) {
  5855. case CompType::Kind::I16:
  5856. case CompType::Kind::U16: {
  5857. DXASSERT_DXBC(false);
  5858. pValue = m_pBuilder->CreateFPExt(pValue, Type::getFloatTy(m_Ctx));
  5859. pValue = CreateBitCast(pValue, CompType::getF32(), CompType::getI32());
  5860. return m_pBuilder->CreateTrunc(pValue, Type::getInt16Ty(m_Ctx));
  5861. }
  5862. case CompType::Kind::I32:
  5863. case CompType::Kind::U32: { // mov
  5864. pValue = m_pBuilder->CreateFPExt(pValue, Type::getFloatTy(m_Ctx));
  5865. return CreateBitCast(pValue, CompType::getF32(), CompType::getI32());
  5866. }
  5867. case CompType::Kind::F32:
  5868. return m_pBuilder->CreateFPExt(pValue, Type::getFloatTy(m_Ctx));
  5869. default: __fallthrough;
  5870. }
  5871. break;
  5872. case CompType::Kind::F32:
  5873. switch (DstType.GetKind()) {
  5874. case CompType::Kind::I1: {
  5875. pValue = CreateBitCast(pValue, CompType::getF32(), CompType::getI32());
  5876. return m_pBuilder->CreateICmpNE(pValue, m_pOP->GetI32Const(0));
  5877. }
  5878. case CompType::Kind::I16:
  5879. case CompType::Kind::U16: { // min-prec for TGSM load.
  5880. pValue = CreateBitCast(pValue, CompType::getF32(), CompType::getI32());
  5881. return m_pBuilder->CreateTrunc(pValue, Type::getInt16Ty(m_Ctx));
  5882. }
  5883. case CompType::Kind::I32:
  5884. case CompType::Kind::U32:
  5885. return CreateBitCast(pValue, CompType::getF32(), CompType::getI32());
  5886. case CompType::Kind::F16:
  5887. return m_pBuilder->CreateFPTrunc(pValue, Type::getHalfTy(m_Ctx));
  5888. default: __fallthrough;
  5889. }
  5890. break;
  5891. default: __fallthrough;
  5892. }
  5893. DXASSERT(false, "unsupported cast combination");
  5894. return nullptr;
  5895. }
  5896. Value *DxbcConverter::CreateBitCast(Value *pValue, const CompType &SrcType, const CompType &DstType) {
  5897. DXASSERT(SrcType.GetLLVMType(m_Ctx) == pValue->getType(), "otherwise caller passed incorrect args");
  5898. OP::OpCode OpCode = (OP::OpCode)(-1);
  5899. switch (SrcType.GetKind()) {
  5900. case CompType::Kind::I16:
  5901. switch (DstType.GetKind()) {
  5902. case CompType::Kind::F16: OpCode = OP::OpCode::BitcastI16toF16; break;
  5903. }
  5904. break;
  5905. case CompType::Kind::I32:
  5906. switch (DstType.GetKind()) {
  5907. case CompType::Kind::F32: OpCode = OP::OpCode::BitcastI32toF32; break;
  5908. }
  5909. break;
  5910. case CompType::Kind::I64:
  5911. switch (DstType.GetKind()) {
  5912. case CompType::Kind::F64: OpCode = OP::OpCode::BitcastI64toF64; break;
  5913. }
  5914. break;
  5915. case CompType::Kind::F16:
  5916. switch (DstType.GetKind()) {
  5917. case CompType::Kind::I16: OpCode = OP::OpCode::BitcastF16toI16; break;
  5918. }
  5919. break;
  5920. case CompType::Kind::F32:
  5921. switch (DstType.GetKind()) {
  5922. case CompType::Kind::I32: OpCode = OP::OpCode::BitcastF32toI32; break;
  5923. }
  5924. break;
  5925. case CompType::Kind::F64:
  5926. switch (DstType.GetKind()) {
  5927. case CompType::Kind::I64: OpCode = OP::OpCode::BitcastF64toI64; break;
  5928. }
  5929. break;
  5930. }
  5931. Value *Args[2];
  5932. Args[0] = m_pOP->GetU32Const((unsigned)OpCode); // OpCode
  5933. Args[1] = pValue; // Input
  5934. Function *F = m_pOP->GetOpFunc(OpCode, Type::getVoidTy(m_Ctx));
  5935. return m_pBuilder->CreateCall(F, Args);
  5936. }
  5937. Value *DxbcConverter::ApplyOperandModifiers(Value *pValue, const D3D10ShaderBinary::COperandBase &O) {
  5938. bool bAbsModifier = (O.m_Modifier & D3D10_SB_OPERAND_MODIFIER_ABS) != 0;
  5939. bool bNegModifier = (O.m_Modifier & D3D10_SB_OPERAND_MODIFIER_NEG) != 0;
  5940. if (bAbsModifier) {
  5941. DXASSERT_DXBC(pValue->getType()->isFloatingPointTy());
  5942. Function *F = m_pOP->GetOpFunc(OP::OpCode::FAbs, pValue->getType());
  5943. Value *Args[2];
  5944. Args[0] = m_pOP->GetU32Const((unsigned)OP::OpCode::FAbs);
  5945. Args[1] = pValue;
  5946. pValue = m_pBuilder->CreateCall(F, Args);
  5947. }
  5948. if (bNegModifier) {
  5949. if (pValue->getType()->isFloatingPointTy()) {
  5950. pValue = m_pBuilder->CreateFNeg(pValue);
  5951. } else {
  5952. DXASSERT_DXBC(pValue->getType()->isIntegerTy());
  5953. pValue = m_pBuilder->CreateNeg(pValue);
  5954. }
  5955. }
  5956. return pValue;
  5957. }
  5958. void DxbcConverter::ApplyInstructionModifiers(OperandValue &DstVal,
  5959. const D3D10ShaderBinary::CInstruction &Inst) {
  5960. if (Inst.m_bSaturate) {
  5961. map<Value *, Value *> M;
  5962. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  5963. Value *pValue = DstVal[c];
  5964. if (pValue == nullptr) continue;
  5965. auto const &it = M.find(pValue);
  5966. if (it != M.end()) {
  5967. DstVal[c] = it->second;
  5968. } else {
  5969. Value *Args[2];
  5970. Args[0] = m_pOP->GetU32Const((unsigned)OP::OpCode::Saturate); // OpCode
  5971. Args[1] = pValue; // Value
  5972. Function *F = m_pOP->GetOpFunc(OP::OpCode::Saturate, pValue->getType());
  5973. Value *pSaturatedValue = MarkPrecise(m_pBuilder->CreateCall(F, Args), c);
  5974. DstVal[c] = pSaturatedValue;
  5975. M[pValue] = pSaturatedValue;
  5976. }
  5977. if (pValue->getType() == Type::getDoubleTy(m_Ctx)) {
  5978. c++;
  5979. }
  5980. }
  5981. }
  5982. }
  5983. CompType DxbcConverter::InferOperandType(const D3D10ShaderBinary::CInstruction &Inst,
  5984. const unsigned OpIdx,
  5985. const CMask &Mask) {
  5986. const D3D10ShaderBinary::COperandBase &O = Inst.m_Operands[OpIdx];
  5987. for (BYTE c = 0; c < DXBC::kWidth; c++) {
  5988. if (!Mask.IsSet(c)) continue;
  5989. switch (O.m_Type) {
  5990. case D3D10_SB_OPERAND_TYPE_INPUT: {
  5991. unsigned Reg = O.m_Index[(m_pSM->IsGS() || m_pSM->IsHS()) ? 1 : 0].m_RegIndex;
  5992. unsigned Comp = O.m_ComponentName;
  5993. if (O.m_ComponentSelection == D3D10_SB_OPERAND_4_COMPONENT_SWIZZLE_MODE)
  5994. Comp = O.m_Swizzle[c];
  5995. const DxilSignatureElement *E = m_pInputSignature->GetElement(Reg, Comp);
  5996. return E->GetCompType();
  5997. }
  5998. case D3D10_SB_OPERAND_TYPE_OUTPUT: {
  5999. unsigned Reg = O.m_Index[0].m_RegIndex;
  6000. if (!m_pSM->IsGS()) {
  6001. if (!m_bPatchConstantPhase) {
  6002. const DxilSignatureElement *E = m_pOutputSignature->GetElement(Reg, c);
  6003. return E->GetCompType();
  6004. } else {
  6005. const DxilSignatureElement *E = m_pPatchConstantSignature->GetElement(Reg, c);
  6006. return E->GetCompType();
  6007. }
  6008. } else {
  6009. CompType CT;
  6010. bool bCTInitialized = false;
  6011. for (unsigned Stream = 0; Stream < DXIL::kNumOutputStreams; Stream++) {
  6012. const DxilSignatureElement *E = m_pOutputSignature->GetElement(Reg, c);
  6013. if (E == nullptr)
  6014. continue;
  6015. if (!bCTInitialized) {
  6016. bCTInitialized = true;
  6017. CT = E->GetCompType();
  6018. } else {
  6019. if (CT.GetKind() != E->GetCompType().GetKind())
  6020. return CompType::getInvalid();
  6021. }
  6022. }
  6023. return CT;
  6024. }
  6025. }
  6026. default: __fallthrough;
  6027. }
  6028. }
  6029. if (O.m_MinPrecision != D3D11_SB_OPERAND_MIN_PRECISION_DEFAULT) {
  6030. return DXBC::GetCompTypeFromMinPrec(O.m_MinPrecision, CompType::getInvalid());
  6031. }
  6032. return CompType::getInvalid();
  6033. }
  6034. void DxbcConverter::CheckDxbcString(const char *pStr, const void *pMaxPtrInclusive) {
  6035. for (;; pStr++) {
  6036. if (pStr > pMaxPtrInclusive) IFT(DXC_E_INCORRECT_DXBC);
  6037. if (*pStr == '\0')
  6038. break;
  6039. }
  6040. }
  6041. void DxbcConverter::Optimize() {
  6042. class PassManager PassManager;
  6043. #if DXBCCONV_DBG
  6044. IFTBOOL(!verifyModule(*m_pModule), DXC_E_IR_VERIFICATION_FAILED); // verifyModule returns true for failure
  6045. #endif
  6046. // Verify that CFG is reducible.
  6047. IFTBOOL(IsReducible(*m_pModule, IrreducibilityAction::ThrowException), DXC_E_IRREDUCIBLE_CFG);
  6048. if (m_bRunDxilCleanup) {
  6049. PassManager.add(createDxilCleanupPass());
  6050. PassManager.run(*m_pModule);
  6051. }
  6052. #if DXBCCONV_DBG
  6053. IFTBOOL(!verifyModule(*m_pModule), DXC_E_IR_VERIFICATION_FAILED);
  6054. #endif
  6055. }
  6056. void DxbcConverter::CreateBranchIfNeeded(BasicBlock *pBB, BasicBlock *pTargetBB) {
  6057. bool bNeedBranch = true;
  6058. if (!pBB->empty()) {
  6059. Instruction *pLastInst = &pBB->getInstList().back();
  6060. if (pLastInst->getOpcode() == Instruction::Br || pLastInst->getOpcode() == Instruction::Ret)
  6061. bNeedBranch = false;
  6062. else
  6063. DXASSERT(!pLastInst->isTerminator(), "otherwise broke possible assumptions of control flow");
  6064. }
  6065. if (bNeedBranch)
  6066. m_pBuilder->CreateBr(pTargetBB);
  6067. }
  6068. Value *DxbcConverter::LoadZNZCondition(D3D10ShaderBinary::CInstruction &Inst,
  6069. const unsigned OpIdx) {
  6070. D3D10ShaderBinary::COperandBase &O = Inst.m_Operands[OpIdx];
  6071. D3D10_SB_INSTRUCTION_TEST_BOOLEAN TestType = Inst.m_Test;
  6072. BYTE Comp = (BYTE)O.m_ComponentName;
  6073. CMask ReadMask = CMask::MakeCompMask(Comp);
  6074. OperandValue In1;
  6075. LoadOperand(In1, Inst, 0, ReadMask, CompType::getI32());
  6076. Value *pCond = In1[Comp];
  6077. if (TestType == D3D10_SB_INSTRUCTION_TEST_NONZERO) {
  6078. pCond = m_pBuilder->CreateICmpNE(pCond, m_pOP->GetI32Const(0));
  6079. } else {
  6080. pCond = m_pBuilder->CreateICmpEQ(pCond, m_pOP->GetI32Const(0));
  6081. }
  6082. return pCond;
  6083. }
  6084. D3D11_SB_OPERAND_MIN_PRECISION DxbcConverter::GetHigherPrecision(
  6085. D3D11_SB_OPERAND_MIN_PRECISION p1,
  6086. D3D11_SB_OPERAND_MIN_PRECISION p2) {
  6087. if (p1 == D3D11_SB_OPERAND_MIN_PRECISION_FLOAT_2_8) p1 = D3D11_SB_OPERAND_MIN_PRECISION_FLOAT_16;
  6088. if (p2 == D3D11_SB_OPERAND_MIN_PRECISION_FLOAT_2_8) p2 = D3D11_SB_OPERAND_MIN_PRECISION_FLOAT_16;
  6089. if (p1 == p2) return p1;
  6090. return D3D11_SB_OPERAND_MIN_PRECISION_DEFAULT;
  6091. }
  6092. unsigned DxbcConverter::GetGSTempRegForOutputReg(unsigned OutputReg) const {
  6093. return m_NumTempRegs + OutputReg;
  6094. }
  6095. //------------------------------------------------------------------------------
  6096. //
  6097. // DxbcConverter::ScopeStack methods.
  6098. //
  6099. DxbcConverter::ScopeStack::ScopeStack()
  6100. : m_FuncCount(0)
  6101. , m_IfCount(0)
  6102. , m_LoopCount(0)
  6103. , m_SwitchCount(0)
  6104. , m_HullLoopCount(0) {
  6105. }
  6106. DxbcConverter::Scope &DxbcConverter::ScopeStack::Top() {
  6107. IFTBOOL(!m_Scopes.empty(), E_FAIL);
  6108. return m_Scopes.back();
  6109. }
  6110. DxbcConverter::Scope &DxbcConverter::ScopeStack::Push(enum Scope::Kind Kind, BasicBlock *pPreScopeBB) {
  6111. Scope S;
  6112. DXASSERT(Kind < Scope::LastKind, "otherwise the caller passed incorrect scope kind value");
  6113. S.Kind = Kind;
  6114. S.pPreScopeBB = pPreScopeBB;
  6115. switch (Kind) {
  6116. case Scope::Function: S.NameIndex = m_FuncCount++; break;
  6117. case Scope::If: S.NameIndex = m_IfCount++; break;
  6118. case Scope::Loop: S.NameIndex = m_LoopCount++; break;
  6119. case Scope::Switch: S.NameIndex = m_SwitchCount++; break;
  6120. case Scope::HullLoop: S.NameIndex = m_HullLoopCount++; break;
  6121. }
  6122. m_Scopes.emplace_back(S);
  6123. return Top();
  6124. }
  6125. void DxbcConverter::ScopeStack::Pop() {
  6126. m_Scopes.pop_back();
  6127. }
  6128. bool DxbcConverter::ScopeStack::IsEmpty() const {
  6129. return m_Scopes.empty();
  6130. }
  6131. DxbcConverter::Scope &DxbcConverter::ScopeStack::FindParentLoop() {
  6132. for (auto it = m_Scopes.rbegin(); it != m_Scopes.rend(); ++it) {
  6133. Scope &Scope = *it;
  6134. if (Scope.Kind == Scope::Loop)
  6135. return Scope;
  6136. }
  6137. DXASSERT(false, "otherwise was not able to find the parent enclosing scope");
  6138. IFTBOOL(false, E_FAIL);
  6139. return Top();
  6140. }
  6141. DxbcConverter::Scope &DxbcConverter::ScopeStack::FindParentLoopOrSwitch() {
  6142. for (auto it = m_Scopes.rbegin(); it != m_Scopes.rend(); ++it) {
  6143. Scope &Scope = *it;
  6144. if (Scope.Kind == Scope::Loop || Scope.Kind == Scope::Switch)
  6145. return Scope;
  6146. }
  6147. DXASSERT(false, "otherwise was not able to find the parent enclosing scope");
  6148. IFTBOOL(false, E_FAIL);
  6149. return Top();
  6150. }
  6151. DxbcConverter::Scope &DxbcConverter::ScopeStack::FindParentFunction() {
  6152. for (auto it = m_Scopes.rbegin(); it != m_Scopes.rend(); ++it) {
  6153. Scope &Scope = *it;
  6154. if (Scope.Kind == Scope::Function)
  6155. return Scope;
  6156. }
  6157. DXASSERT(false, "otherwise was not able to find the parent enclosing scope");
  6158. IFTBOOL(false, E_FAIL);
  6159. return Top();
  6160. }
  6161. DxbcConverter::Scope &DxbcConverter::ScopeStack::FindParentHullLoop() {
  6162. for (auto it = m_Scopes.rbegin(); it != m_Scopes.rend(); ++it) {
  6163. Scope &Scope = *it;
  6164. if (Scope.Kind == Scope::HullLoop)
  6165. return Scope;
  6166. }
  6167. DXASSERT(false, "otherwise was not able to find the parent enclosing scope");
  6168. IFTBOOL(false, E_FAIL);
  6169. return Top();
  6170. }
  6171. string DxbcConverter::SynthesizeResGVName(const char *pNamePrefix, unsigned ID) {
  6172. string GVName;
  6173. raw_string_ostream GVNameStream(GVName);
  6174. (GVNameStream << pNamePrefix << ID).flush();
  6175. return GVName;
  6176. }
  6177. StructType *DxbcConverter::GetStructResElemType(unsigned StructSizeInBytes) {
  6178. string GVTypeName;
  6179. raw_string_ostream GVTypeNameStream(GVTypeName);
  6180. (GVTypeNameStream << "dx.types.i8x" << StructSizeInBytes).flush();
  6181. StructType *pGVType = m_pModule->getTypeByName(GVTypeName);
  6182. if (pGVType == nullptr) {
  6183. pGVType = StructType::create(m_Ctx, ArrayType::get(Type::getInt8Ty(m_Ctx), StructSizeInBytes), GVTypeName);
  6184. }
  6185. return pGVType;
  6186. }
  6187. StructType *DxbcConverter::GetTypedResElemType(CompType CT) {
  6188. string GVTypeName;
  6189. raw_string_ostream GVTypeNameStream(GVTypeName);
  6190. (GVTypeNameStream << "dx.types." << CT.GetName()).flush();
  6191. StructType *pGVType = m_pModule->getTypeByName(GVTypeName);
  6192. if (pGVType == nullptr) {
  6193. Type *pElemType = nullptr;
  6194. if (CT.GetKind() == CompType::Kind::SNormF32) {
  6195. pElemType = m_pPR->GetTypeSystem().GetSNormF32Type(1);
  6196. } else if (CT.GetKind() == CompType::Kind::UNormF32) {
  6197. pElemType = m_pPR->GetTypeSystem().GetUNormF32Type(1);
  6198. } else {
  6199. pElemType = CT.GetLLVMType(m_Ctx);
  6200. }
  6201. if (!pElemType->isStructTy()) {
  6202. pGVType = StructType::create(m_Ctx, pElemType, GVTypeName);
  6203. } else {
  6204. pGVType = dyn_cast<StructType>(pElemType);
  6205. }
  6206. }
  6207. return pGVType;
  6208. }
  6209. UndefValue *DxbcConverter::DeclareUndefPtr(Type *pType, unsigned AddrSpace) {
  6210. Type *pPtrType = PointerType::get(pType, AddrSpace);
  6211. UndefValue *pUV = UndefValue::get(pPtrType);
  6212. return pUV;
  6213. }
  6214. Value *DxbcConverter::MarkPrecise(Value *pVal, BYTE Comp) {
  6215. if ((Comp == BYTE(-1) && !m_PreciseMask.IsZero()) || (Comp != BYTE(-1) && m_PreciseMask.IsSet(Comp))) {
  6216. if (Instruction *pInst = dyn_cast<Instruction>(pVal)) {
  6217. bool bAttachPreciseMD = true;
  6218. if (dyn_cast<FPMathOperator>(pInst) != nullptr && dyn_cast<CallInst>(pInst) == nullptr) {
  6219. FastMathFlags FMF;
  6220. pInst->copyFastMathFlags(FMF);
  6221. bAttachPreciseMD = false;
  6222. }
  6223. if (bAttachPreciseMD) {
  6224. MDNode *pMD = MDNode::get(m_Ctx, ConstantAsMetadata::get(m_pOP->GetI32Const(1)));
  6225. pInst->setMetadata(DxilMDHelper::kDxilPreciseAttributeMDName, pMD);
  6226. }
  6227. }
  6228. }
  6229. return pVal;
  6230. }
  6231. void DxbcConverter::SerializeDxil(SmallVectorImpl<char> &DxilBitcode) {
  6232. raw_svector_ostream DxilStream(DxilBitcode);
  6233. // a. Reserve header.
  6234. DxilProgramHeader Header = { 0 };
  6235. DxilStream.write((char*)&Header, sizeof(Header));
  6236. // b. Bitcode.
  6237. WriteBitcodeToFile(m_pModule.get(), DxilStream);
  6238. DxilStream.flush();
  6239. // c. Fix header.
  6240. uint32_t bitcodeSize = (uint32_t)DxilBitcode.size_in_bytes() - sizeof(DxilProgramHeader);
  6241. DxilProgramHeader *pHeader = (DxilProgramHeader*)DxilBitcode.data();
  6242. InitProgramHeader(*pHeader, EncodeVersion(m_pSM->GetKind(), m_pSM->GetMajor(), m_pSM->GetMinor()), DXIL::MakeDxilVersion(1, 0), bitcodeSize);
  6243. // d. Trailer. Pad to 16 bytes.
  6244. while (DxilBitcode.size() & 0xF) {
  6245. DxilBitcode.push_back(0);
  6246. }
  6247. IFTBOOL(DxilBitcode.size_in_bytes() < UINT_MAX && (DxilBitcode.size_in_bytes() & 0xF) == 0, DXC_E_DATA_TOO_LARGE);
  6248. }
  6249. } // namespace hlsl
  6250. HRESULT CreateDxbcConverter(_In_ REFIID riid, _Out_ LPVOID *ppv) {
  6251. try {
  6252. CComPtr<hlsl::DxbcConverter> result(hlsl::DxbcConverter::Alloc(DxcGetThreadMallocNoRef()));
  6253. IFROOM(result.p);
  6254. return result.p->QueryInterface(riid, ppv);
  6255. }
  6256. CATCH_CPP_RETURN_HRESULT();
  6257. }