DxbcConverter.cpp 291 KB

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