123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771377237733774377537763777377837793780378137823783378437853786378737883789379037913792379337943795379637973798379938003801380238033804380538063807380838093810381138123813381438153816381738183819382038213822382338243825382638273828382938303831383238333834383538363837383838393840384138423843384438453846384738483849385038513852385338543855385638573858385938603861386238633864386538663867386838693870387138723873387438753876387738783879388038813882388338843885388638873888388938903891389238933894389538963897389838993900390139023903390439053906390739083909391039113912391339143915391639173918391939203921392239233924392539263927392839293930393139323933393439353936393739383939394039413942394339443945394639473948394939503951395239533954395539563957395839593960396139623963396439653966396739683969397039713972397339743975397639773978397939803981398239833984398539863987398839893990399139923993399439953996399739983999400040014002400340044005400640074008400940104011401240134014401540164017401840194020402140224023402440254026402740284029403040314032403340344035403640374038403940404041404240434044404540464047404840494050405140524053405440554056405740584059406040614062406340644065406640674068406940704071407240734074407540764077407840794080408140824083408440854086408740884089409040914092409340944095409640974098409941004101410241034104410541064107410841094110411141124113411441154116411741184119412041214122412341244125412641274128412941304131413241334134413541364137413841394140414141424143414441454146414741484149415041514152415341544155415641574158415941604161416241634164416541664167416841694170417141724173417441754176417741784179418041814182418341844185418641874188418941904191419241934194419541964197419841994200420142024203420442054206420742084209421042114212421342144215421642174218421942204221422242234224422542264227422842294230423142324233423442354236423742384239424042414242424342444245424642474248424942504251425242534254425542564257425842594260426142624263426442654266426742684269427042714272427342744275427642774278427942804281428242834284428542864287428842894290429142924293429442954296429742984299430043014302430343044305430643074308430943104311431243134314431543164317431843194320432143224323432443254326432743284329433043314332433343344335433643374338433943404341434243434344434543464347434843494350435143524353435443554356435743584359436043614362436343644365436643674368436943704371437243734374437543764377437843794380438143824383438443854386438743884389439043914392439343944395439643974398439944004401440244034404440544064407440844094410441144124413441444154416441744184419442044214422442344244425442644274428442944304431443244334434443544364437443844394440444144424443444444454446444744484449445044514452445344544455445644574458445944604461446244634464446544664467446844694470447144724473447444754476447744784479448044814482448344844485448644874488448944904491449244934494449544964497449844994500450145024503450445054506450745084509451045114512451345144515451645174518451945204521452245234524452545264527452845294530453145324533453445354536453745384539454045414542454345444545454645474548454945504551455245534554455545564557455845594560456145624563456445654566456745684569457045714572457345744575457645774578457945804581458245834584458545864587458845894590459145924593459445954596459745984599460046014602460346044605460646074608460946104611461246134614461546164617461846194620462146224623462446254626462746284629463046314632463346344635463646374638463946404641464246434644464546464647464846494650465146524653465446554656465746584659466046614662466346644665466646674668466946704671467246734674467546764677467846794680468146824683468446854686468746884689469046914692469346944695469646974698469947004701470247034704470547064707470847094710471147124713471447154716471747184719472047214722472347244725472647274728472947304731473247334734473547364737473847394740474147424743474447454746474747484749475047514752475347544755475647574758475947604761476247634764476547664767476847694770477147724773477447754776477747784779478047814782478347844785478647874788478947904791479247934794479547964797479847994800480148024803480448054806480748084809481048114812481348144815481648174818481948204821482248234824482548264827482848294830483148324833483448354836483748384839484048414842484348444845484648474848484948504851485248534854485548564857485848594860486148624863486448654866486748684869487048714872487348744875487648774878487948804881488248834884488548864887488848894890489148924893489448954896489748984899490049014902490349044905490649074908490949104911491249134914491549164917491849194920492149224923492449254926492749284929493049314932493349344935493649374938493949404941494249434944494549464947494849494950495149524953495449554956495749584959496049614962496349644965496649674968496949704971497249734974497549764977497849794980498149824983498449854986498749884989499049914992499349944995499649974998499950005001500250035004500550065007500850095010501150125013501450155016501750185019502050215022502350245025502650275028502950305031503250335034503550365037503850395040504150425043504450455046504750485049505050515052505350545055505650575058505950605061506250635064506550665067506850695070507150725073507450755076507750785079508050815082508350845085508650875088508950905091509250935094509550965097509850995100510151025103510451055106510751085109511051115112511351145115511651175118511951205121512251235124512551265127512851295130513151325133513451355136513751385139514051415142514351445145514651475148514951505151515251535154515551565157515851595160516151625163516451655166516751685169517051715172517351745175517651775178517951805181518251835184518551865187518851895190519151925193519451955196519751985199520052015202520352045205520652075208520952105211521252135214521552165217521852195220522152225223522452255226522752285229523052315232523352345235523652375238523952405241524252435244524552465247524852495250525152525253525452555256525752585259526052615262526352645265526652675268526952705271527252735274527552765277527852795280528152825283528452855286528752885289529052915292529352945295529652975298529953005301530253035304530553065307530853095310531153125313531453155316531753185319532053215322532353245325532653275328532953305331533253335334533553365337533853395340534153425343534453455346534753485349535053515352535353545355535653575358535953605361536253635364536553665367536853695370537153725373537453755376537753785379538053815382538353845385538653875388538953905391539253935394539553965397539853995400540154025403540454055406540754085409541054115412541354145415541654175418541954205421542254235424542554265427542854295430543154325433543454355436543754385439544054415442544354445445544654475448544954505451545254535454545554565457545854595460546154625463546454655466546754685469547054715472547354745475547654775478547954805481548254835484548554865487548854895490549154925493549454955496549754985499550055015502550355045505550655075508550955105511551255135514551555165517551855195520552155225523552455255526552755285529553055315532553355345535553655375538553955405541554255435544554555465547554855495550555155525553555455555556555755585559556055615562556355645565556655675568556955705571557255735574557555765577557855795580558155825583558455855586558755885589559055915592559355945595559655975598559956005601560256035604560556065607560856095610561156125613561456155616561756185619562056215622562356245625562656275628562956305631563256335634563556365637563856395640564156425643564456455646564756485649565056515652565356545655565656575658565956605661566256635664566556665667566856695670567156725673567456755676567756785679568056815682568356845685568656875688568956905691569256935694569556965697569856995700570157025703570457055706570757085709571057115712571357145715571657175718571957205721572257235724572557265727572857295730573157325733573457355736573757385739574057415742574357445745574657475748574957505751575257535754575557565757575857595760576157625763576457655766576757685769577057715772577357745775577657775778577957805781578257835784578557865787578857895790579157925793579457955796579757985799580058015802580358045805580658075808580958105811581258135814581558165817581858195820582158225823582458255826582758285829583058315832583358345835583658375838583958405841584258435844584558465847584858495850585158525853585458555856585758585859586058615862586358645865586658675868586958705871587258735874587558765877587858795880588158825883588458855886588758885889589058915892589358945895589658975898589959005901590259035904590559065907590859095910591159125913591459155916591759185919592059215922592359245925592659275928592959305931593259335934593559365937593859395940594159425943594459455946594759485949595059515952595359545955595659575958595959605961596259635964596559665967596859695970597159725973597459755976597759785979598059815982598359845985598659875988598959905991599259935994599559965997599859996000600160026003600460056006600760086009601060116012601360146015601660176018601960206021602260236024602560266027602860296030603160326033603460356036603760386039604060416042604360446045604660476048604960506051605260536054605560566057605860596060606160626063606460656066606760686069607060716072607360746075607660776078607960806081608260836084608560866087608860896090609160926093609460956096609760986099610061016102610361046105610661076108610961106111611261136114611561166117611861196120612161226123612461256126612761286129613061316132613361346135613661376138613961406141614261436144614561466147614861496150615161526153615461556156615761586159616061616162616361646165616661676168616961706171617261736174617561766177617861796180618161826183618461856186618761886189619061916192619361946195619661976198619962006201620262036204620562066207620862096210621162126213621462156216621762186219622062216222622362246225622662276228622962306231 |
- // Windows Template Library - WTL version 10.0
- // Copyright (C) Microsoft Corporation, WTL Team. All rights reserved.
- //
- // This file is a part of the Windows Template Library.
- // The use and distribution terms for this software are covered by the
- // Microsoft Public License (http://opensource.org/licenses/MS-PL)
- // which can be found in the file MS-PL.txt at the root folder.
- #ifndef __ATLDLGS_H__
- #define __ATLDLGS_H__
- #pragma once
- #ifndef __ATLAPP_H__
- #error atldlgs.h requires atlapp.h to be included first
- #endif
- #ifndef __ATLWIN_H__
- #error atldlgs.h requires atlwin.h to be included first
- #endif
- #include <shlobj.h>
- #if (_WIN32_WINNT >= 0x0600)
- #include <shobjidl.h>
- #endif // (_WIN32_WINNT >= 0x0600)
- ///////////////////////////////////////////////////////////////////////////////
- // Classes in this file:
- //
- // CFileDialogImpl<T>
- // CFileDialog
- // CSimpleFileDialog
- // CMultiFileDialogImpl<T>
- // CMultiFileDialog
- // CShellFileDialogImpl<T>
- // CShellFileOpenDialogImpl<T>
- // CShellFileOpenDialog
- // CShellFileSaveDialogImpl<T>
- // CShellFileSaveDialog
- // CFolderDialogImpl<T>
- // CFolderDialog
- // CFontDialogImpl<T>
- // CFontDialog
- // CRichEditFontDialogImpl<T>
- // CRichEditFontDialog
- // CColorDialogImpl<T>
- // CColorDialog
- // CPrintDialogImpl<T>
- // CPrintDialog
- // CPrintDialogExImpl<T>
- // CPrintDialogEx
- // CPageSetupDialogImpl<T>
- // CPageSetupDialog
- // CFindReplaceDialogImpl<T>
- // CFindReplaceDialog
- //
- // CDialogBaseUnits
- // CMemDlgTemplate
- // CIndirectDialogImpl<T, TDlgTemplate, TBase>
- //
- // CPropertySheetWindow
- // CPropertySheetImpl<T, TBase>
- // CPropertySheet
- // CPropertyPageWindow
- // CPropertyPageImpl<T, TBase>
- // CPropertyPage<t_wDlgTemplateID>
- // CAxPropertyPageImpl<T, TBase>
- // CAxPropertyPage<t_wDlgTemplateID>
- //
- // CWizard97SheetWindow
- // CWizard97SheetImpl<T, TBase>
- // CWizard97Sheet
- // CWizard97PageWindow
- // CWizard97PageImpl<T, TBase>
- // CWizard97ExteriorPageImpl<T, TBase>
- // CWizard97InteriorPageImpl<T, TBase>
- //
- // CAeroWizardFrameWindow
- // CAeroWizardFrameImpl<T, TBase>
- // CAeroWizardFrame
- // CAeroWizardPageWindow
- // CAeroWizardPageImpl<T, TBase>
- // CAeroWizardPage<t_wDlgTemplateID>
- // CAeroWizardAxPageImpl<T, TBase>
- // CAeroWizardAxPage<t_wDlgTemplateID>
- //
- // CTaskDialogConfig
- // CTaskDialogImpl<T>
- // CTaskDialog
- //
- // Global functions:
- // AtlTaskDialog()
- namespace WTL
- {
- ///////////////////////////////////////////////////////////////////////////////
- // CFileDialogImpl - used for File Open or File Save As
- template <class T>
- class ATL_NO_VTABLE CFileDialogImpl : public ATL::CDialogImplBase
- {
- public:
- OPENFILENAME m_ofn;
- BOOL m_bOpenFileDialog; // TRUE for file open, FALSE for file save
- TCHAR m_szFileTitle[_MAX_FNAME]; // contains file title after return
- TCHAR m_szFileName[_MAX_PATH]; // contains full path name after return
- CFileDialogImpl(BOOL bOpenFileDialog, // TRUE for FileOpen, FALSE for FileSaveAs
- LPCTSTR lpszDefExt = NULL,
- LPCTSTR lpszFileName = NULL,
- DWORD dwFlags = OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT,
- LPCTSTR lpszFilter = NULL,
- HWND hWndParent = NULL) : m_bOpenFileDialog(bOpenFileDialog)
- {
- memset(&m_ofn, 0, sizeof(m_ofn)); // initialize structure to 0/NULL
- m_ofn.lStructSize = sizeof(m_ofn);
- m_ofn.lpstrFile = m_szFileName;
- m_ofn.nMaxFile = _MAX_PATH;
- m_ofn.lpstrDefExt = lpszDefExt;
- m_ofn.lpstrFileTitle = (LPTSTR)m_szFileTitle;
- m_ofn.nMaxFileTitle = _MAX_FNAME;
- m_ofn.Flags = dwFlags | OFN_EXPLORER | OFN_ENABLEHOOK | OFN_ENABLESIZING;
- m_ofn.lpstrFilter = lpszFilter;
- m_ofn.hInstance = ModuleHelper::GetResourceInstance();
- m_ofn.lpfnHook = (LPOFNHOOKPROC)T::StartDialogProc;
- m_ofn.hwndOwner = hWndParent;
- m_szFileName[0] = _T('\0');
- m_szFileTitle[0] = _T('\0');
- // setup initial file name
- if(lpszFileName != NULL)
- ATL::Checked::tcsncpy_s(m_szFileName, _countof(m_szFileName), lpszFileName, _TRUNCATE);
- }
- INT_PTR DoModal(HWND hWndParent = ::GetActiveWindow())
- {
- ATLASSERT((m_ofn.Flags & OFN_ENABLEHOOK) != 0);
- ATLASSERT(m_ofn.lpfnHook != NULL); // can still be a user hook
- ATLASSERT((m_ofn.Flags & OFN_EXPLORER) != 0);
- if(m_ofn.hwndOwner == NULL) // set only if not specified before
- m_ofn.hwndOwner = hWndParent;
- ATLASSERT(m_hWnd == NULL);
- // Allocate the thunk structure here, where we can fail gracefully.
- BOOL bRetTh = m_thunk.Init(NULL, NULL);
- if(bRetTh == FALSE)
- {
- ::SetLastError(ERROR_OUTOFMEMORY);
- return -1;
- }
- ModuleHelper::AddCreateWndData(&m_thunk.cd, (ATL::CDialogImplBase*)this);
- BOOL bRet = (m_bOpenFileDialog != FALSE) ? ::GetOpenFileName(&m_ofn) : ::GetSaveFileName(&m_ofn);
- m_hWnd = NULL;
- return (bRet != FALSE) ? IDOK : IDCANCEL;
- }
- // Attributes
- ATL::CWindow GetFileDialogWindow() const
- {
- ATLASSERT(::IsWindow(m_hWnd));
- return ATL::CWindow(GetParent());
- }
- int GetFilePath(LPTSTR lpstrFilePath, int nLength) const
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ATLASSERT((m_ofn.Flags & OFN_EXPLORER) != 0);
- return (int)GetFileDialogWindow().SendMessage(CDM_GETFILEPATH, nLength, (LPARAM)lpstrFilePath);
- }
- int GetFolderIDList(LPVOID lpBuff, int nLength) const
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ATLASSERT((m_ofn.Flags & OFN_EXPLORER) != 0);
- return (int)GetFileDialogWindow().SendMessage(CDM_GETFOLDERIDLIST, nLength, (LPARAM)lpBuff);
- }
- int GetFolderPath(LPTSTR lpstrFolderPath, int nLength) const
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ATLASSERT((m_ofn.Flags & OFN_EXPLORER) != 0);
- return (int)GetFileDialogWindow().SendMessage(CDM_GETFOLDERPATH, nLength, (LPARAM)lpstrFolderPath);
- }
- int GetSpec(LPTSTR lpstrSpec, int nLength) const
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ATLASSERT((m_ofn.Flags & OFN_EXPLORER) != 0);
- return (int)GetFileDialogWindow().SendMessage(CDM_GETSPEC, nLength, (LPARAM)lpstrSpec);
- }
- void SetControlText(int nCtrlID, LPCTSTR lpstrText)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ATLASSERT((m_ofn.Flags & OFN_EXPLORER) != 0);
- GetFileDialogWindow().SendMessage(CDM_SETCONTROLTEXT, nCtrlID, (LPARAM)lpstrText);
- }
- void SetDefExt(LPCTSTR lpstrExt)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ATLASSERT((m_ofn.Flags & OFN_EXPLORER) != 0);
- GetFileDialogWindow().SendMessage(CDM_SETDEFEXT, 0, (LPARAM)lpstrExt);
- }
- BOOL GetReadOnlyPref() const // return TRUE if readonly checked
- {
- return ((m_ofn.Flags & OFN_READONLY) != 0) ? TRUE : FALSE;
- }
- // Operations
- void HideControl(int nCtrlID)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ATLASSERT((m_ofn.Flags & OFN_EXPLORER) != 0);
- GetFileDialogWindow().SendMessage(CDM_HIDECONTROL, nCtrlID);
- }
- // Special override for common dialogs
- BOOL EndDialog(INT_PTR /*nRetCode*/ = 0)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- GetFileDialogWindow().SendMessage(WM_COMMAND, MAKEWPARAM(IDCANCEL, 0));
- return TRUE;
- }
- // Message map and handlers
- BEGIN_MSG_MAP(CFileDialogImpl)
- NOTIFY_CODE_HANDLER(CDN_FILEOK, _OnFileOK)
- NOTIFY_CODE_HANDLER(CDN_FOLDERCHANGE, _OnFolderChange)
- NOTIFY_CODE_HANDLER(CDN_HELP, _OnHelp)
- NOTIFY_CODE_HANDLER(CDN_INITDONE, _OnInitDone)
- NOTIFY_CODE_HANDLER(CDN_SELCHANGE, _OnSelChange)
- NOTIFY_CODE_HANDLER(CDN_SHAREVIOLATION, _OnShareViolation)
- NOTIFY_CODE_HANDLER(CDN_TYPECHANGE, _OnTypeChange)
- NOTIFY_CODE_HANDLER(CDN_INCLUDEITEM, _OnIncludeItem)
- END_MSG_MAP()
- LRESULT _OnFileOK(int /*idCtrl*/, LPNMHDR pnmh, BOOL& /*bHandled*/)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- T* pT = static_cast<T*>(this);
- return !pT->OnFileOK((LPOFNOTIFY)pnmh);
- }
- LRESULT _OnFolderChange(int /*idCtrl*/, LPNMHDR pnmh, BOOL& /*bHandled*/)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- T* pT = static_cast<T*>(this);
- pT->OnFolderChange((LPOFNOTIFY)pnmh);
- return 0;
- }
- LRESULT _OnHelp(int /*idCtrl*/, LPNMHDR pnmh, BOOL& /*bHandled*/)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- T* pT = static_cast<T*>(this);
- pT->OnHelp((LPOFNOTIFY)pnmh);
- return 0;
- }
- LRESULT _OnInitDone(int /*idCtrl*/, LPNMHDR pnmh, BOOL& /*bHandled*/)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- T* pT = static_cast<T*>(this);
- pT->OnInitDone((LPOFNOTIFY)pnmh);
- return 0;
- }
- LRESULT _OnSelChange(int /*idCtrl*/, LPNMHDR pnmh, BOOL& /*bHandled*/)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- T* pT = static_cast<T*>(this);
- pT->OnSelChange((LPOFNOTIFY)pnmh);
- return 0;
- }
- LRESULT _OnShareViolation(int /*idCtrl*/, LPNMHDR pnmh, BOOL& /*bHandled*/)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- T* pT = static_cast<T*>(this);
- return pT->OnShareViolation((LPOFNOTIFY)pnmh);
- }
- LRESULT _OnTypeChange(int /*idCtrl*/, LPNMHDR pnmh, BOOL& /*bHandled*/)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- T* pT = static_cast<T*>(this);
- pT->OnTypeChange((LPOFNOTIFY)pnmh);
- return 0;
- }
- LRESULT _OnIncludeItem(int /*idCtrl*/, LPNMHDR pnmh, BOOL& /*bHandled*/)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- T* pT = static_cast<T*>(this);
- return pT->OnIncludeItem((LPOFNOTIFYEX)pnmh);
- }
- // Overrideables
- BOOL OnFileOK(LPOFNOTIFY /*lpon*/)
- {
- return TRUE;
- }
- void OnFolderChange(LPOFNOTIFY /*lpon*/)
- {
- }
- void OnHelp(LPOFNOTIFY /*lpon*/)
- {
- }
- void OnInitDone(LPOFNOTIFY /*lpon*/)
- {
- }
- void OnSelChange(LPOFNOTIFY /*lpon*/)
- {
- }
- int OnShareViolation(LPOFNOTIFY /*lpon*/)
- {
- return 0;
- }
- void OnTypeChange(LPOFNOTIFY /*lpon*/)
- {
- }
- BOOL OnIncludeItem(LPOFNOTIFYEX /*lponex*/)
- {
- return TRUE; // include item
- }
- };
- class CFileDialog : public CFileDialogImpl<CFileDialog>
- {
- public:
- CFileDialog(BOOL bOpenFileDialog, // TRUE for FileOpen, FALSE for FileSaveAs
- LPCTSTR lpszDefExt = NULL,
- LPCTSTR lpszFileName = NULL,
- DWORD dwFlags = OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT,
- LPCTSTR lpszFilter = NULL,
- HWND hWndParent = NULL)
- : CFileDialogImpl<CFileDialog>(bOpenFileDialog, lpszDefExt, lpszFileName, dwFlags, lpszFilter, hWndParent)
- { }
- // override base class map and references to handlers
- DECLARE_EMPTY_MSG_MAP()
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CSimpleFileDialog - simple class for non-customized Open/SaveAs dialogs
- class CSimpleFileDialog
- {
- public:
- OPENFILENAME m_ofn;
- BOOL m_bOpenFileDialog; // TRUE for file open, FALSE for file save
- TCHAR m_szFileTitle[_MAX_FNAME]; // contains file title after return
- TCHAR m_szFileName[_MAX_PATH]; // contains full path name after return
- CSimpleFileDialog(BOOL bOpenFileDialog, // TRUE for FileOpen, FALSE for FileSaveAs
- LPCTSTR lpszDefExt = NULL,
- LPCTSTR lpszFileName = NULL,
- DWORD dwFlags = OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT,
- LPCTSTR lpszFilter = NULL,
- HWND hWndParent = NULL) : m_bOpenFileDialog(bOpenFileDialog)
- {
- memset(&m_ofn, 0, sizeof(m_ofn)); // initialize structure to 0/NULL
- m_ofn.lStructSize = sizeof(m_ofn);
- m_ofn.lpstrFile = m_szFileName;
- m_ofn.nMaxFile = _MAX_PATH;
- m_ofn.lpstrDefExt = lpszDefExt;
- m_ofn.lpstrFileTitle = (LPTSTR)m_szFileTitle;
- m_ofn.nMaxFileTitle = _MAX_FNAME;
- m_ofn.Flags = dwFlags | OFN_EXPLORER | OFN_ENABLESIZING;
- m_ofn.lpstrFilter = lpszFilter;
- m_ofn.hInstance = ModuleHelper::GetResourceInstance();
- m_ofn.hwndOwner = hWndParent;
- m_szFileName[0] = _T('\0');
- m_szFileTitle[0] = _T('\0');
- // setup initial file name
- if(lpszFileName != NULL)
- ATL::Checked::tcsncpy_s(m_szFileName, _countof(m_szFileName), lpszFileName, _TRUNCATE);
- }
- INT_PTR DoModal(HWND hWndParent = ::GetActiveWindow())
- {
- ATLASSERT((m_ofn.Flags & OFN_EXPLORER) != 0);
- if(m_ofn.hwndOwner == NULL) // set only if not specified before
- m_ofn.hwndOwner = hWndParent;
- BOOL bRet = (m_bOpenFileDialog != FALSE) ? ::GetOpenFileName(&m_ofn) : ::GetSaveFileName(&m_ofn);
- return (bRet != FALSE) ? IDOK : IDCANCEL;
- }
- };
- ///////////////////////////////////////////////////////////////////////////////
- // Multi File Dialog - Multi-select File Open dialog
- // The class dynamically resizes the buffer as the file selection changes
- // (as described in Knowledge Base article 131462). It also expands selected
- // shortcut files to take into account the full path of the target file.
- // Note that this doesn't work on Win9x for the old style dialogs, as well as
- // on NT for non-Unicode builds.
- #ifndef _WTL_FIXED_OFN_BUFFER_LENGTH
- #define _WTL_FIXED_OFN_BUFFER_LENGTH 0x10000
- #endif
- template <class T>
- class ATL_NO_VTABLE CMultiFileDialogImpl : public CFileDialogImpl< T >
- {
- public:
- mutable LPCTSTR m_pNextFile;
- #ifndef _UNICODE
- bool m_bIsNT;
- #endif
- CMultiFileDialogImpl(
- LPCTSTR lpszDefExt = NULL,
- LPCTSTR lpszFileName = NULL,
- DWORD dwFlags = OFN_HIDEREADONLY,
- LPCTSTR lpszFilter = NULL,
- HWND hWndParent = NULL)
- : CFileDialogImpl<T>(TRUE, lpszDefExt, lpszFileName, dwFlags, lpszFilter, hWndParent),
- m_pNextFile(NULL)
- {
- this->m_ofn.Flags |= OFN_ALLOWMULTISELECT; // Force multiple selection mode
- #ifndef _UNICODE
- #ifdef _versionhelpers_H_INCLUDED_
- OSVERSIONINFOEX ovi = { sizeof(OSVERSIONINFOEX) };
- ovi.dwPlatformId = VER_PLATFORM_WIN32_NT;
- DWORDLONG const dwlConditionMask = ::VerSetConditionMask(0, VER_PLATFORMID, VER_EQUAL);
- m_bIsNT = (::VerifyVersionInfo(&ovi, VER_PLATFORMID, dwlConditionMask) != FALSE);
- #else // !_versionhelpers_H_INCLUDED_
- OSVERSIONINFO ovi = { sizeof(OSVERSIONINFO) };
- ::GetVersionEx(&ovi);
- m_bIsNT = (ovi.dwPlatformId == VER_PLATFORM_WIN32_NT);
- #endif // _versionhelpers_H_INCLUDED_
- if (m_bIsNT)
- {
- // On NT platforms, GetOpenFileNameA thunks to GetOpenFileNameW and there
- // is absolutely nothing we can do except to start off with a large buffer.
- ATLVERIFY(ResizeFilenameBuffer(_WTL_FIXED_OFN_BUFFER_LENGTH));
- }
- #endif
- }
- ~CMultiFileDialogImpl()
- {
- if (this->m_ofn.lpstrFile != this->m_szFileName) // Free the buffer if we allocated it
- delete[] this->m_ofn.lpstrFile;
- }
- // Operations
- // Get the directory that the files were chosen from.
- // The function returns the number of characters copied, not including the terminating zero.
- // If the buffer is NULL, the function returns the required size, in characters, including the terminating zero.
- // If the function fails, the return value is zero.
- int GetDirectory(LPTSTR pBuffer, int nBufLen) const
- {
- if (this->m_ofn.lpstrFile == NULL)
- return 0;
- LPCTSTR pStr = this->m_ofn.lpstrFile;
- int nLength = lstrlen(pStr);
- if (pStr[nLength + 1] == 0)
- {
- // The OFN buffer contains a single item so extract its path.
- LPCTSTR pSep = _tcsrchr(pStr, _T('\\'));
- if (pSep != NULL)
- nLength = (int)(DWORD_PTR)(pSep - pStr);
- }
- int nRet = 0;
- if (pBuffer == NULL) // If the buffer is NULL, return the required length
- {
- nRet = nLength + 1;
- }
- else if (nBufLen > nLength)
- {
- ATL::Checked::tcsncpy_s(pBuffer, nBufLen, pStr, nLength);
- nRet = nLength;
- }
- return nRet;
- }
- #ifdef __ATLSTR_H__
- bool GetDirectory(ATL::CString& strDir) const
- {
- bool bRet = false;
- int nLength = GetDirectory(NULL, 0);
- if (nLength > 0)
- {
- bRet = (GetDirectory(strDir.GetBuffer(nLength), nLength) > 0);
- strDir.ReleaseBuffer(nLength - 1);
- }
- return bRet;
- }
- #endif // __ATLSTR_H__
- // Get the first filename as a pointer into the buffer.
- LPCTSTR GetFirstFileName() const
- {
- if (this->m_ofn.lpstrFile == NULL)
- return NULL;
- m_pNextFile = NULL; // Reset internal buffer pointer
- LPCTSTR pStr = this->m_ofn.lpstrFile;
- int nLength = lstrlen(pStr);
- if (pStr[nLength + 1] != 0)
- {
- // Multiple items were selected. The first string is the directory,
- // so skip forwards to the second string.
- pStr += nLength + 1;
- // Set up m_pNext so it points to the second item (or null).
- m_pNextFile = pStr;
- GetNextFileName();
- }
- else
- {
- // A single item was selected. Skip forward past the path.
- LPCTSTR pSep = _tcsrchr(pStr, _T('\\'));
- if (pSep != NULL)
- pStr = pSep + 1;
- }
- return pStr;
- }
- // Get the next filename as a pointer into the buffer.
- LPCTSTR GetNextFileName() const
- {
- if (m_pNextFile == NULL)
- return NULL;
- LPCTSTR pStr = m_pNextFile;
- // Set "m_pNextFile" to point to the next file name, or null if we
- // have reached the last file in the list.
- int nLength = lstrlen(pStr);
- m_pNextFile = (pStr[nLength + 1] != 0) ? &pStr[nLength + 1] : NULL;
- return pStr;
- }
- // Get the first filename as a full path.
- // The function returns the number of characters copied, not including the terminating zero.
- // If the buffer is NULL, the function returns the required size, in characters, including the terminating zero.
- // If the function fails, the return value is zero.
- int GetFirstPathName(LPTSTR pBuffer, int nBufLen) const
- {
- LPCTSTR pStr = GetFirstFileName();
- int nLengthDir = GetDirectory(NULL, 0);
- if((pStr == NULL) || (nLengthDir == 0))
- return 0;
- // Figure out the required length.
- int nLengthTotal = nLengthDir + lstrlen(pStr);
- int nRet = 0;
- if(pBuffer == NULL) // If the buffer is NULL, return the required length
- {
- nRet = nLengthTotal + 1;
- }
- else if (nBufLen > nLengthTotal) // If the buffer is big enough, go ahead and construct the path
- {
- GetDirectory(pBuffer, nBufLen);
- ATL::Checked::tcscat_s(pBuffer, nBufLen, _T("\\"));
- ATL::Checked::tcscat_s(pBuffer, nBufLen, pStr);
- nRet = nLengthTotal;
- }
- return nRet;
- }
- #ifdef __ATLSTR_H__
- bool GetFirstPathName(ATL::CString& strPath) const
- {
- bool bRet = false;
- int nLength = GetFirstPathName(NULL, 0);
- if (nLength > 0)
- {
- bRet = (GetFirstPathName(strPath.GetBuffer(nLength), nLength) > 0);
- strPath.ReleaseBuffer(nLength - 1);
- }
- return bRet;
- }
- #endif // __ATLSTR_H__
- // Get the next filename as a full path.
- // The function returns the number of characters copied, not including the terminating zero.
- // If the buffer is NULL, the function returns the required size, in characters, including the terminating zero.
- // If the function fails, the return value is zero.
- // The internal position marker is moved forward only if the function succeeds and the buffer was large enough.
- int GetNextPathName(LPTSTR pBuffer, int nBufLen) const
- {
- if (m_pNextFile == NULL)
- return 0;
- int nRet = 0;
- LPCTSTR pStr = m_pNextFile;
- // Does the filename contain a backslash?
- if (_tcsrchr(pStr, _T('\\')) != NULL)
- {
- // Yes, so we'll assume it's a full path.
- int nLength = lstrlen(pStr);
- if (pBuffer == NULL) // If the buffer is NULL, return the required length
- {
- nRet = nLength + 1;
- }
- else if (nBufLen > nLength) // The buffer is big enough, so go ahead and copy the filename
- {
- ATL::Checked::tcscpy_s(pBuffer, nBufLen, GetNextFileName());
- nRet = nBufLen;
- }
- }
- else
- {
- // The filename is relative, so construct the full path.
- int nLengthDir = GetDirectory(NULL, 0);
- if (nLengthDir > 0)
- {
- // Calculate the required space.
- int nLengthTotal = nLengthDir + lstrlen(pStr);
- if(pBuffer == NULL) // If the buffer is NULL, return the required length
- {
- nRet = nLengthTotal + 1;
- }
- else if (nBufLen > nLengthTotal) // If the buffer is big enough, go ahead and construct the path
- {
- GetDirectory(pBuffer, nBufLen);
- ATL::Checked::tcscat_s(pBuffer, nBufLen, _T("\\"));
- ATL::Checked::tcscat_s(pBuffer, nBufLen, GetNextFileName());
- nRet = nLengthTotal;
- }
- }
- }
- return nRet;
- }
- #ifdef __ATLSTR_H__
- bool GetNextPathName(ATL::CString& strPath) const
- {
- bool bRet = false;
- int nLength = GetNextPathName(NULL, 0);
- if (nLength > 0)
- {
- bRet = (GetNextPathName(strPath.GetBuffer(nLength), nLength) > 0);
- strPath.ReleaseBuffer(nLength - 1);
- }
- return bRet;
- }
- #endif // __ATLSTR_H__
- // Implementation
- bool ResizeFilenameBuffer(DWORD dwLength)
- {
- if (dwLength > this->m_ofn.nMaxFile)
- {
- // Free the old buffer.
- if (this->m_ofn.lpstrFile != this->m_szFileName)
- {
- delete[] this->m_ofn.lpstrFile;
- this->m_ofn.lpstrFile = NULL;
- this->m_ofn.nMaxFile = 0;
- }
- // Allocate the new buffer.
- LPTSTR lpstrBuff = NULL;
- ATLTRY(lpstrBuff = new TCHAR[dwLength]);
- if (lpstrBuff != NULL)
- {
- this->m_ofn.lpstrFile = lpstrBuff;
- this->m_ofn.lpstrFile[0] = 0;
- this->m_ofn.nMaxFile = dwLength;
- }
- }
- return (this->m_ofn.lpstrFile != NULL);
- }
- void OnSelChange(LPOFNOTIFY /*lpon*/)
- {
- #ifndef _UNICODE
- // There is no point resizing the buffer in ANSI builds running on NT.
- if (m_bIsNT)
- return;
- #endif
- // Get the buffer length required to hold the spec.
- int nLength = this->GetSpec(NULL, 0);
- if (nLength <= 1)
- return; // no files are selected, presumably
-
- // Add room for the directory, and an extra terminating zero.
- nLength += this->GetFolderPath(NULL, 0) + 1;
- if (!ResizeFilenameBuffer(nLength))
- {
- ATLASSERT(FALSE);
- return;
- }
- // If we are not following links then our work is done.
- if ((this->m_ofn.Flags & OFN_NODEREFERENCELINKS) != 0)
- return;
- // Get the file spec, which is the text in the edit control.
- if (this->GetSpec(this->m_ofn.lpstrFile, this->m_ofn.nMaxFile) <= 0)
- return;
-
- // Get the ID-list of the current folder.
- int nBytes = this->GetFolderIDList(NULL, 0);
- #ifdef STRICT_TYPED_ITEMIDS
- ATL::CTempBuffer<ITEMIDLIST_RELATIVE> idlist;
- #else
- ATL::CTempBuffer<ITEMIDLIST> idlist;
- #endif
- idlist.AllocateBytes(nBytes);
- if ((nBytes <= 0) || (this->GetFolderIDList(idlist, nBytes) <= 0))
- return;
- // First bind to the desktop folder, then to the current folder.
- ATL::CComPtr<IShellFolder> pDesktop, pFolder;
- if (FAILED(::SHGetDesktopFolder(&pDesktop)))
- return;
- if (FAILED(pDesktop->BindToObject(idlist, NULL, IID_IShellFolder, (void**)&pFolder)))
- return;
- // Work through the file spec, looking for quoted filenames. If we find a shortcut file, then
- // we need to add enough extra buffer space to hold its target path.
- DWORD nExtraChars = 0;
- bool bInsideQuotes = false;
- LPCTSTR pAnchor = this->m_ofn.lpstrFile;
- LPCTSTR pChar = this->m_ofn.lpstrFile;
- for ( ; *pChar; ++pChar)
- {
- // Look for quotation marks.
- if (*pChar == _T('\"'))
- {
- // We are either entering or leaving a passage of quoted text.
- bInsideQuotes = !bInsideQuotes;
- // Is it an opening or closing quote?
- if (bInsideQuotes)
- {
- // We found an opening quote, so set "pAnchor" to the following character.
- pAnchor = pChar + 1;
- }
- else // closing quote
- {
- // Each quoted entity should be shorter than MAX_PATH.
- if (pChar - pAnchor >= MAX_PATH)
- return;
- // Get the ID-list and attributes of the file.
- USES_CONVERSION;
- int nFileNameLength = (int)(DWORD_PTR)(pChar - pAnchor);
- TCHAR szFileName[MAX_PATH] = {};
- ATL::Checked::tcsncpy_s(szFileName, MAX_PATH, pAnchor, nFileNameLength);
- #ifdef STRICT_TYPED_ITEMIDS
- PIDLIST_RELATIVE pidl = NULL;
- #else
- LPITEMIDLIST pidl = NULL;
- #endif
- DWORD dwAttrib = SFGAO_LINK;
- if (SUCCEEDED(pFolder->ParseDisplayName(NULL, NULL, T2W(szFileName), NULL, &pidl, &dwAttrib)))
- {
- // Is it a shortcut file?
- if (dwAttrib & SFGAO_LINK)
- {
- // Bind to its IShellLink interface.
- ATL::CComPtr<IShellLink> pLink;
- if (SUCCEEDED(pFolder->BindToObject(pidl, NULL, IID_IShellLink, (void**)&pLink)))
- {
- // Get the shortcut's target path.
- TCHAR szPath[MAX_PATH] = {};
- if (SUCCEEDED(pLink->GetPath(szPath, MAX_PATH, NULL, 0)))
- {
- // If the target path is longer than the shortcut name, then add on the number
- // of extra characters that are required.
- int nNewLength = lstrlen(szPath);
- if (nNewLength > nFileNameLength)
- nExtraChars += nNewLength - nFileNameLength;
- }
- }
- }
- // Free the ID-list returned by ParseDisplayName.
- ::CoTaskMemFree(pidl);
- }
- }
- }
- }
- // If we need more space for shortcut targets, then reallocate.
- if (nExtraChars > 0)
- ATLVERIFY(ResizeFilenameBuffer(this->m_ofn.nMaxFile + nExtraChars));
- }
- };
- class CMultiFileDialog : public CMultiFileDialogImpl<CMultiFileDialog>
- {
- public:
- CMultiFileDialog(
- LPCTSTR lpszDefExt = NULL,
- LPCTSTR lpszFileName = NULL,
- DWORD dwFlags = OFN_HIDEREADONLY,
- LPCTSTR lpszFilter = NULL,
- HWND hWndParent = NULL)
- : CMultiFileDialogImpl<CMultiFileDialog>(lpszDefExt, lpszFileName, dwFlags, lpszFilter, hWndParent)
- { }
- BEGIN_MSG_MAP(CMultiFileDialog)
- CHAIN_MSG_MAP(CMultiFileDialogImpl<CMultiFileDialog>)
- END_MSG_MAP()
- };
- ///////////////////////////////////////////////////////////////////////////////
- // Shell File Dialog - new Shell File Open and Save dialogs in Vista
- // Note: Use GetPtr() to access dialog interface methods.
- // Example:
- // CShellFileOpenDialog dlg;
- // dlg.GetPtr()->SetTitle(L"MyFileOpenDialog");
- #if (_WIN32_WINNT >= 0x0600)
- ///////////////////////////////////////////////////////////////////////////////
- // CShellFileDialogImpl - base class for CShellFileOpenDialogImpl and CShellFileSaveDialogImpl
- template <class T>
- class ATL_NO_VTABLE CShellFileDialogImpl : public IFileDialogEvents
- {
- public:
- // Operations
- INT_PTR DoModal(HWND hWndParent = ::GetActiveWindow())
- {
- INT_PTR nRet = -1;
- T* pT = static_cast<T*>(this);
- if(pT->m_spFileDlg == NULL)
- {
- ATLASSERT(FALSE);
- return nRet;
- }
- DWORD dwCookie = 0;
- pT->_Advise(dwCookie);
- HRESULT hRet = pT->m_spFileDlg->Show(hWndParent);
- if(SUCCEEDED(hRet))
- nRet = IDOK;
- else if(hRet == HRESULT_FROM_WIN32(ERROR_CANCELLED))
- nRet = IDCANCEL;
- else
- ATLASSERT(FALSE); // error
- pT->_Unadvise(dwCookie);
- return nRet;
- }
- bool IsNull() const
- {
- const T* pT = static_cast<const T*>(this);
- return (pT->m_spFileDlg == NULL);
- }
- // Operations - get file path after dialog returns
- HRESULT GetFilePath(LPWSTR lpstrFilePath, int cchLength)
- {
- T* pT = static_cast<T*>(this);
- ATLASSERT(pT->m_spFileDlg != NULL);
- ATL::CComPtr<IShellItem> spItem;
- HRESULT hRet = pT->m_spFileDlg->GetResult(&spItem);
- if(SUCCEEDED(hRet))
- hRet = GetFileNameFromShellItem(spItem, SIGDN_FILESYSPATH, lpstrFilePath, cchLength);
- return hRet;
- }
- HRESULT GetFileTitle(LPWSTR lpstrFileTitle, int cchLength)
- {
- T* pT = static_cast<T*>(this);
- ATLASSERT(pT->m_spFileDlg != NULL);
- ATL::CComPtr<IShellItem> spItem;
- HRESULT hRet = pT->m_spFileDlg->GetResult(&spItem);
- if(SUCCEEDED(hRet))
- hRet = GetFileNameFromShellItem(spItem, SIGDN_NORMALDISPLAY, lpstrFileTitle, cchLength);
- return hRet;
- }
- #ifdef __ATLSTR_H__
- HRESULT GetFilePath(ATL::CString& strFilePath)
- {
- T* pT = static_cast<T*>(this);
- ATLASSERT(pT->m_spFileDlg != NULL);
- ATL::CComPtr<IShellItem> spItem;
- HRESULT hRet = pT->m_spFileDlg->GetResult(&spItem);
- if(SUCCEEDED(hRet))
- hRet = GetFileNameFromShellItem(spItem, SIGDN_FILESYSPATH, strFilePath);
- return hRet;
- }
- HRESULT GetFileTitle(ATL::CString& strFileTitle)
- {
- T* pT = static_cast<T*>(this);
- ATLASSERT(pT->m_spFileDlg != NULL);
- ATL::CComPtr<IShellItem> spItem;
- HRESULT hRet = pT->m_spFileDlg->GetResult(&spItem);
- if(SUCCEEDED(hRet))
- hRet = GetFileNameFromShellItem(spItem, SIGDN_NORMALDISPLAY, strFileTitle);
- return hRet;
- }
- #endif // __ATLSTR_H__
- // Helpers for IShellItem
- static HRESULT GetFileNameFromShellItem(IShellItem* pShellItem, SIGDN type, LPWSTR lpstr, int cchLength)
- {
- ATLASSERT(pShellItem != NULL);
- LPWSTR lpstrName = NULL;
- HRESULT hRet = pShellItem->GetDisplayName(type, &lpstrName);
- if(SUCCEEDED(hRet))
- {
- if(lstrlenW(lpstrName) < cchLength)
- {
- ATL::Checked::wcscpy_s(lpstr, cchLength, lpstrName);
- }
- else
- {
- ATLASSERT(FALSE);
- hRet = DISP_E_BUFFERTOOSMALL;
- }
- ::CoTaskMemFree(lpstrName);
- }
- return hRet;
- }
- #ifdef __ATLSTR_H__
- static HRESULT GetFileNameFromShellItem(IShellItem* pShellItem, SIGDN type, ATL::CString& str)
- {
- ATLASSERT(pShellItem != NULL);
- LPWSTR lpstrName = NULL;
- HRESULT hRet = pShellItem->GetDisplayName(type, &lpstrName);
- if(SUCCEEDED(hRet))
- {
- str = lpstrName;
- ::CoTaskMemFree(lpstrName);
- }
- return hRet;
- }
- #endif // __ATLSTR_H__
- // Implementation
- void _Advise(DWORD& dwCookie)
- {
- T* pT = static_cast<T*>(this);
- ATLASSERT(pT->m_spFileDlg != NULL);
- HRESULT hRet = pT->m_spFileDlg->Advise((IFileDialogEvents*)this, &dwCookie);
- ATLVERIFY(SUCCEEDED(hRet));
- }
- void _Unadvise(DWORD dwCookie)
- {
- T* pT = static_cast<T*>(this);
- ATLASSERT(pT->m_spFileDlg != NULL);
- HRESULT hRet = pT->m_spFileDlg->Unadvise(dwCookie);
- ATLVERIFY(SUCCEEDED(hRet));
- }
- void _Init(LPCWSTR lpszFileName, DWORD dwOptions, LPCWSTR lpszDefExt, const COMDLG_FILTERSPEC* arrFilterSpec, UINT uFilterSpecCount)
- {
- T* pT = static_cast<T*>(this);
- ATLASSERT(pT->m_spFileDlg != NULL);
- HRESULT hRet = E_FAIL;
- if(lpszFileName != NULL)
- {
- hRet = pT->m_spFileDlg->SetFileName(lpszFileName);
- ATLASSERT(SUCCEEDED(hRet));
- }
- hRet = pT->m_spFileDlg->SetOptions(dwOptions);
- ATLASSERT(SUCCEEDED(hRet));
- if(lpszDefExt != NULL)
- {
- hRet = pT->m_spFileDlg->SetDefaultExtension(lpszDefExt);
- ATLASSERT(SUCCEEDED(hRet));
- }
- if((arrFilterSpec != NULL) && (uFilterSpecCount != 0U))
- {
- hRet = pT->m_spFileDlg->SetFileTypes(uFilterSpecCount, arrFilterSpec);
- ATLASSERT(SUCCEEDED(hRet));
- }
- }
- // Implementation - IUnknown interface
- STDMETHOD(QueryInterface)(REFIID riid, void** ppvObject)
- {
- if(ppvObject == NULL)
- return E_POINTER;
- T* pT = static_cast<T*>(this);
- if(IsEqualGUID(riid, IID_IUnknown) || IsEqualGUID(riid, IID_IFileDialogEvents))
- {
- *ppvObject = (IFileDialogEvents*)pT;
- // AddRef() not needed
- return S_OK;
- }
- return E_NOINTERFACE;
- }
- virtual ULONG STDMETHODCALLTYPE AddRef()
- {
- return 1;
- }
- virtual ULONG STDMETHODCALLTYPE Release()
- {
- return 1;
- }
- // Implementation - IFileDialogEvents interface
- virtual HRESULT STDMETHODCALLTYPE OnFileOk(IFileDialog* pfd)
- {
- T* pT = static_cast<T*>(this);
- ATLASSERT(pT->m_spFileDlg.IsEqualObject(pfd));
- (void)pfd; // avoid level 4 warning
- return pT->OnFileOk();
- }
- virtual HRESULT STDMETHODCALLTYPE OnFolderChanging(IFileDialog* pfd, IShellItem* psiFolder)
- {
- T* pT = static_cast<T*>(this);
- ATLASSERT(pT->m_spFileDlg.IsEqualObject(pfd));
- (void)pfd; // avoid level 4 warning
- return pT->OnFolderChanging(psiFolder);
- }
- virtual HRESULT STDMETHODCALLTYPE OnFolderChange(IFileDialog* pfd)
- {
- T* pT = static_cast<T*>(this);
- ATLASSERT(pT->m_spFileDlg.IsEqualObject(pfd));
- (void)pfd; // avoid level 4 warning
- return pT->OnFolderChange();
- }
- virtual HRESULT STDMETHODCALLTYPE OnSelectionChange(IFileDialog* pfd)
- {
- T* pT = static_cast<T*>(this);
- ATLASSERT(pT->m_spFileDlg.IsEqualObject(pfd));
- (void)pfd; // avoid level 4 warning
- return pT->OnSelectionChange();
- }
- virtual HRESULT STDMETHODCALLTYPE OnShareViolation(IFileDialog* pfd, IShellItem* psi, FDE_SHAREVIOLATION_RESPONSE* pResponse)
- {
- T* pT = static_cast<T*>(this);
- ATLASSERT(pT->m_spFileDlg.IsEqualObject(pfd));
- (void)pfd; // avoid level 4 warning
- return pT->OnShareViolation(psi, pResponse);
- }
- virtual HRESULT STDMETHODCALLTYPE OnTypeChange(IFileDialog* pfd)
- {
- T* pT = static_cast<T*>(this);
- ATLASSERT(pT->m_spFileDlg.IsEqualObject(pfd));
- (void)pfd; // avoid level 4 warning
- return pT->OnTypeChange();
- }
- virtual HRESULT STDMETHODCALLTYPE OnOverwrite(IFileDialog* pfd, IShellItem* psi, FDE_OVERWRITE_RESPONSE* pResponse)
- {
- T* pT = static_cast<T*>(this);
- ATLASSERT(pT->m_spFileDlg.IsEqualObject(pfd));
- (void)pfd; // avoid level 4 warning
- return pT->OnOverwrite(psi, pResponse);
- }
- // Overrideables - Event handlers
- HRESULT OnFileOk()
- {
- return E_NOTIMPL;
- }
- HRESULT OnFolderChanging(IShellItem* /*psiFolder*/)
- {
- return E_NOTIMPL;
- }
- HRESULT OnFolderChange()
- {
- return E_NOTIMPL;
- }
- HRESULT OnSelectionChange()
- {
- return E_NOTIMPL;
- }
- HRESULT OnShareViolation(IShellItem* /*psi*/, FDE_SHAREVIOLATION_RESPONSE* /*pResponse*/)
- {
- return E_NOTIMPL;
- }
- HRESULT OnTypeChange()
- {
- return E_NOTIMPL;
- }
- HRESULT OnOverwrite(IShellItem* /*psi*/, FDE_OVERWRITE_RESPONSE* /*pResponse*/)
- {
- return E_NOTIMPL;
- }
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CShellFileOpenDialogImpl - implements new Shell File Open dialog
- template <class T>
- class ATL_NO_VTABLE CShellFileOpenDialogImpl : public CShellFileDialogImpl< T >
- {
- public:
- ATL::CComPtr<IFileOpenDialog> m_spFileDlg;
- CShellFileOpenDialogImpl(LPCWSTR lpszFileName = NULL,
- DWORD dwOptions = FOS_FORCEFILESYSTEM | FOS_PATHMUSTEXIST | FOS_FILEMUSTEXIST,
- LPCWSTR lpszDefExt = NULL,
- const COMDLG_FILTERSPEC* arrFilterSpec = NULL,
- UINT uFilterSpecCount = 0U)
- {
- HRESULT hRet = m_spFileDlg.CoCreateInstance(CLSID_FileOpenDialog);
- if(SUCCEEDED(hRet))
- this->_Init(lpszFileName, dwOptions, lpszDefExt, arrFilterSpec, uFilterSpecCount);
- }
- virtual ~CShellFileOpenDialogImpl()
- { }
- IFileOpenDialog* GetPtr()
- {
- return m_spFileDlg;
- }
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CShellFileOpenDialog - new Shell File Open dialog without events
- class CShellFileOpenDialog : public CShellFileOpenDialogImpl<CShellFileOpenDialog>
- {
- public:
- CShellFileOpenDialog(LPCWSTR lpszFileName = NULL,
- DWORD dwOptions = FOS_FORCEFILESYSTEM | FOS_PATHMUSTEXIST | FOS_FILEMUSTEXIST,
- LPCWSTR lpszDefExt = NULL,
- const COMDLG_FILTERSPEC* arrFilterSpec = NULL,
- UINT uFilterSpecCount = 0U) : CShellFileOpenDialogImpl<CShellFileOpenDialog>(lpszFileName, dwOptions, lpszDefExt, arrFilterSpec, uFilterSpecCount)
- { }
- virtual ~CShellFileOpenDialog()
- { }
- // Implementation (remove _Advise/_Unadvise code using template magic)
- void _Advise(DWORD& /*dwCookie*/)
- { }
- void _Unadvise(DWORD /*dwCookie*/)
- { }
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CShellFileSaveDialogImpl - implements new Shell File Save dialog
- template <class T>
- class ATL_NO_VTABLE CShellFileSaveDialogImpl : public CShellFileDialogImpl< T >
- {
- public:
- ATL::CComPtr<IFileSaveDialog> m_spFileDlg;
- CShellFileSaveDialogImpl(LPCWSTR lpszFileName = NULL,
- DWORD dwOptions = FOS_FORCEFILESYSTEM | FOS_PATHMUSTEXIST | FOS_OVERWRITEPROMPT,
- LPCWSTR lpszDefExt = NULL,
- const COMDLG_FILTERSPEC* arrFilterSpec = NULL,
- UINT uFilterSpecCount = 0U)
- {
- HRESULT hRet = m_spFileDlg.CoCreateInstance(CLSID_FileSaveDialog);
- if(SUCCEEDED(hRet))
- this->_Init(lpszFileName, dwOptions, lpszDefExt, arrFilterSpec, uFilterSpecCount);
- }
- virtual ~CShellFileSaveDialogImpl()
- { }
- IFileSaveDialog* GetPtr()
- {
- return m_spFileDlg;
- }
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CShellFileSaveDialog - new Shell File Save dialog without events
- class CShellFileSaveDialog : public CShellFileSaveDialogImpl<CShellFileSaveDialog>
- {
- public:
- CShellFileSaveDialog(LPCWSTR lpszFileName = NULL,
- DWORD dwOptions = FOS_FORCEFILESYSTEM | FOS_PATHMUSTEXIST | FOS_OVERWRITEPROMPT,
- LPCWSTR lpszDefExt = NULL,
- const COMDLG_FILTERSPEC* arrFilterSpec = NULL,
- UINT uFilterSpecCount = 0U) : CShellFileSaveDialogImpl<CShellFileSaveDialog>(lpszFileName, dwOptions, lpszDefExt, arrFilterSpec, uFilterSpecCount)
- { }
- virtual ~CShellFileSaveDialog()
- { }
- // Implementation (remove _Advise/_Unadvise code using template magic)
- void _Advise(DWORD& /*dwCookie*/)
- { }
- void _Unadvise(DWORD /*dwCookie*/)
- { }
- };
- #endif // (_WIN32_WINNT >= 0x0600)
- ///////////////////////////////////////////////////////////////////////////////
- // CFolderDialogImpl - used for browsing for a folder
- template <class T>
- class ATL_NO_VTABLE CFolderDialogImpl
- {
- public:
- BROWSEINFO m_bi;
- LPCTSTR m_lpstrInitialFolder;
- LPCITEMIDLIST m_pidlInitialSelection;
- bool m_bExpandInitialSelection;
- TCHAR m_szFolderDisplayName[MAX_PATH];
- TCHAR m_szFolderPath[MAX_PATH];
- #ifdef STRICT_TYPED_ITEMIDS
- PIDLIST_ABSOLUTE m_pidlSelected;
- #else
- LPITEMIDLIST m_pidlSelected;
- #endif
- HWND m_hWnd; // used only in the callback function
- // Constructor
- CFolderDialogImpl(HWND hWndParent = NULL, LPCTSTR lpstrTitle = NULL, UINT uFlags = BIF_RETURNONLYFSDIRS) :
- m_lpstrInitialFolder(NULL), m_pidlInitialSelection(NULL), m_bExpandInitialSelection(false), m_pidlSelected(NULL), m_hWnd(NULL)
- {
- memset(&m_bi, 0, sizeof(m_bi)); // initialize structure to 0/NULL
- m_bi.hwndOwner = hWndParent;
- m_bi.pidlRoot = NULL;
- m_bi.pszDisplayName = m_szFolderDisplayName;
- m_bi.lpszTitle = lpstrTitle;
- m_bi.ulFlags = uFlags;
- m_bi.lpfn = BrowseCallbackProc;
- m_bi.lParam = (LPARAM)static_cast<T*>(this);
- m_szFolderPath[0] = 0;
- m_szFolderDisplayName[0] = 0;
- }
- ~CFolderDialogImpl()
- {
- ::CoTaskMemFree(m_pidlSelected);
- }
- // Operations
- INT_PTR DoModal(HWND hWndParent = ::GetActiveWindow())
- {
- if(m_bi.hwndOwner == NULL) // set only if not specified before
- m_bi.hwndOwner = hWndParent;
- // Clear out any previous results
- m_szFolderPath[0] = 0;
- m_szFolderDisplayName[0] = 0;
- ::CoTaskMemFree(m_pidlSelected);
- INT_PTR nRet = IDCANCEL;
- m_pidlSelected = ::SHBrowseForFolder(&m_bi);
- if(m_pidlSelected != NULL)
- {
- nRet = IDOK;
- // If BIF_RETURNONLYFSDIRS is set, we try to get the filesystem path.
- // Otherwise, the caller must handle the ID-list directly.
- if((m_bi.ulFlags & BIF_RETURNONLYFSDIRS) != 0)
- {
- if(::SHGetPathFromIDList(m_pidlSelected, m_szFolderPath) == FALSE)
- nRet = IDCANCEL;
- }
- }
- return nRet;
- }
- // Methods to call before DoModal
- void SetInitialFolder(LPCTSTR lpstrInitialFolder, bool bExpand = true)
- {
- // lpstrInitialFolder may be a file if BIF_BROWSEINCLUDEFILES is specified
- m_lpstrInitialFolder = lpstrInitialFolder;
- m_bExpandInitialSelection = bExpand;
- }
- void SetInitialSelection(LPCITEMIDLIST pidl, bool bExpand = true)
- {
- m_pidlInitialSelection = pidl;
- m_bExpandInitialSelection = bExpand;
- }
- #ifdef STRICT_TYPED_ITEMIDS
- void SetRootFolder(PCIDLIST_ABSOLUTE pidl)
- #else
- void SetRootFolder(LPCITEMIDLIST pidl)
- #endif
- {
- m_bi.pidlRoot = pidl;
- }
- // Methods to call after DoModal
- LPITEMIDLIST GetSelectedItem(bool bDetach = false)
- {
- LPITEMIDLIST pidl = m_pidlSelected;
- if(bDetach)
- m_pidlSelected = NULL;
- return pidl;
- }
- LPCTSTR GetFolderPath() const
- {
- return m_szFolderPath;
- }
- LPCTSTR GetFolderDisplayName() const
- {
- return m_szFolderDisplayName;
- }
- int GetFolderImageIndex() const
- {
- return m_bi.iImage;
- }
- // Callback function and overrideables
- static int CALLBACK BrowseCallbackProc(HWND hWnd, UINT uMsg, LPARAM lParam, LPARAM lpData)
- {
- int nRet = 0;
- T* pT = (T*)lpData;
- bool bClear = false;
- if(pT->m_hWnd == NULL)
- {
- pT->m_hWnd = hWnd;
- bClear = true;
- }
- else
- {
- ATLASSERT(pT->m_hWnd == hWnd);
- }
- switch(uMsg)
- {
- case BFFM_INITIALIZED:
- // Set initial selection
- // Note that m_pidlInitialSelection, if set, takes precedence over m_lpstrInitialFolder
- if(pT->m_pidlInitialSelection != NULL)
- pT->SetSelection(pT->m_pidlInitialSelection);
- else if(pT->m_lpstrInitialFolder != NULL)
- pT->SetSelection(pT->m_lpstrInitialFolder);
- // Expand initial selection if appropriate
- if(pT->m_bExpandInitialSelection && ((pT->m_bi.ulFlags & BIF_NEWDIALOGSTYLE) != 0))
- {
- if(pT->m_pidlInitialSelection != NULL)
- pT->SetExpanded(pT->m_pidlInitialSelection);
- else if(pT->m_lpstrInitialFolder != NULL)
- pT->SetExpanded(pT->m_lpstrInitialFolder);
- }
- pT->OnInitialized();
- break;
- case BFFM_SELCHANGED:
- pT->OnSelChanged((LPITEMIDLIST)lParam);
- break;
- case BFFM_VALIDATEFAILED:
- nRet = pT->OnValidateFailed((LPCTSTR)lParam);
- break;
- case BFFM_IUNKNOWN:
- pT->OnIUnknown((IUnknown*)lParam);
- break;
- default:
- ATLTRACE2(atlTraceUI, 0, _T("Unknown message received in CFolderDialogImpl::BrowseCallbackProc\n"));
- break;
- }
- if(bClear)
- pT->m_hWnd = NULL;
- return nRet;
- }
- void OnInitialized()
- {
- }
- void OnSelChanged(LPITEMIDLIST /*pItemIDList*/)
- {
- }
- int OnValidateFailed(LPCTSTR /*lpstrFolderPath*/)
- {
- return 1; // 1=continue, 0=EndDialog
- }
- void OnIUnknown(IUnknown* /*pUnknown*/)
- {
- }
- // Commands - valid to call only from handlers
- void EnableOK(BOOL bEnable)
- {
- ATLASSERT(m_hWnd != NULL);
- ::SendMessage(m_hWnd, BFFM_ENABLEOK, 0, bEnable);
- }
- void SetSelection(LPCITEMIDLIST pItemIDList)
- {
- ATLASSERT(m_hWnd != NULL);
- ::SendMessage(m_hWnd, BFFM_SETSELECTION, FALSE, (LPARAM)pItemIDList);
- }
- void SetSelection(LPCTSTR lpstrFolderPath)
- {
- ATLASSERT(m_hWnd != NULL);
- ::SendMessage(m_hWnd, BFFM_SETSELECTION, TRUE, (LPARAM)lpstrFolderPath);
- }
- void SetStatusText(LPCTSTR lpstrText)
- {
- ATLASSERT(m_hWnd != NULL);
- ::SendMessage(m_hWnd, BFFM_SETSTATUSTEXT, 0, (LPARAM)lpstrText);
- }
- void SetOKText(LPCTSTR lpstrOKText)
- {
- ATLASSERT(m_hWnd != NULL);
- USES_CONVERSION;
- LPCWSTR lpstr = T2CW(lpstrOKText);
- ::SendMessage(m_hWnd, BFFM_SETOKTEXT, 0, (LPARAM)lpstr);
- }
- void SetExpanded(LPCITEMIDLIST pItemIDList)
- {
- ATLASSERT(m_hWnd != NULL);
- ::SendMessage(m_hWnd, BFFM_SETEXPANDED, FALSE, (LPARAM)pItemIDList);
- }
- void SetExpanded(LPCTSTR lpstrFolderPath)
- {
- ATLASSERT(m_hWnd != NULL);
- USES_CONVERSION;
- LPCWSTR lpstr = T2CW(lpstrFolderPath);
- ::SendMessage(m_hWnd, BFFM_SETEXPANDED, TRUE, (LPARAM)lpstr);
- }
- };
- class CFolderDialog : public CFolderDialogImpl<CFolderDialog>
- {
- public:
- CFolderDialog(HWND hWndParent = NULL, LPCTSTR lpstrTitle = NULL, UINT uFlags = BIF_RETURNONLYFSDIRS)
- : CFolderDialogImpl<CFolderDialog>(hWndParent, lpstrTitle, uFlags)
- { }
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CCommonDialogImplBase - base class for common dialog classes
- class ATL_NO_VTABLE CCommonDialogImplBase : public ATL::CWindowImplBase
- {
- public:
- static UINT_PTR APIENTRY HookProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
- {
- if(uMsg != WM_INITDIALOG)
- return 0;
- CCommonDialogImplBase* pT = (CCommonDialogImplBase*)ModuleHelper::ExtractCreateWndData();
- ATLASSERT(pT != NULL);
- ATLASSERT(pT->m_hWnd == NULL);
- ATLASSERT(::IsWindow(hWnd));
- // subclass dialog's window
- if(!pT->SubclassWindow(hWnd))
- {
- ATLTRACE2(atlTraceUI, 0, _T("Subclassing a common dialog failed\n"));
- return 0;
- }
- // check message map for WM_INITDIALOG handler
- LRESULT lRes = 0;
- if(pT->ProcessWindowMessage(pT->m_hWnd, uMsg, wParam, lParam, lRes, 0) == FALSE)
- return 0;
- return lRes;
- }
- // Special override for common dialogs
- BOOL EndDialog(INT_PTR /*nRetCode*/ = 0)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- SendMessage(WM_COMMAND, MAKEWPARAM(IDABORT, 0));
- return TRUE;
- }
- // Implementation - try to override these, to prevent errors
- HWND Create(HWND, ATL::_U_RECT, LPCTSTR, DWORD, DWORD, ATL::_U_MENUorID, ATOM, LPVOID)
- {
- ATLASSERT(FALSE); // should not be called
- return NULL;
- }
- static LRESULT CALLBACK StartWindowProc(HWND /*hWnd*/, UINT /*uMsg*/, WPARAM /*wParam*/, LPARAM /*lParam*/)
- {
- ATLASSERT(FALSE); // should not be called
- return 0;
- }
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CFontDialogImpl - font selection dialog
- template <class T>
- class ATL_NO_VTABLE CFontDialogImpl : public CCommonDialogImplBase
- {
- public:
- enum { _cchStyleName = 64 };
- CHOOSEFONT m_cf;
- TCHAR m_szStyleName[_cchStyleName]; // contains style name after return
- LOGFONT m_lf; // default LOGFONT to store the info
- // Constructors
- CFontDialogImpl(LPLOGFONT lplfInitial = NULL,
- DWORD dwFlags = CF_EFFECTS | CF_SCREENFONTS,
- HDC hDCPrinter = NULL,
- HWND hWndParent = NULL)
- {
- memset(&m_cf, 0, sizeof(m_cf));
- memset(&m_lf, 0, sizeof(m_lf));
- memset(&m_szStyleName, 0, sizeof(m_szStyleName));
- m_cf.lStructSize = sizeof(m_cf);
- m_cf.hwndOwner = hWndParent;
- m_cf.rgbColors = RGB(0, 0, 0);
- m_cf.lpszStyle = (LPTSTR)&m_szStyleName;
- m_cf.Flags = dwFlags | CF_ENABLEHOOK;
- m_cf.lpfnHook = (LPCFHOOKPROC)T::HookProc;
- if(lplfInitial != NULL)
- {
- m_cf.lpLogFont = lplfInitial;
- m_cf.Flags |= CF_INITTOLOGFONTSTRUCT;
- m_lf = *lplfInitial;
- }
- else
- {
- m_cf.lpLogFont = &m_lf;
- }
- if(hDCPrinter != NULL)
- {
- m_cf.hDC = hDCPrinter;
- m_cf.Flags |= CF_PRINTERFONTS;
- }
- }
- // Operations
- INT_PTR DoModal(HWND hWndParent = ::GetActiveWindow())
- {
- ATLASSERT((m_cf.Flags & CF_ENABLEHOOK) != 0);
- ATLASSERT(m_cf.lpfnHook != NULL); // can still be a user hook
- if(m_cf.hwndOwner == NULL) // set only if not specified before
- m_cf.hwndOwner = hWndParent;
- ATLASSERT(m_hWnd == NULL);
- // Allocate the thunk structure here, where we can fail gracefully.
- BOOL bRetTh = m_thunk.Init(NULL, NULL);
- if(bRetTh == FALSE)
- {
- ::SetLastError(ERROR_OUTOFMEMORY);
- return -1;
- }
- ModuleHelper::AddCreateWndData(&m_thunk.cd, (CCommonDialogImplBase*)this);
- BOOL bRet = ::ChooseFont(&m_cf);
- m_hWnd = NULL;
- if(bRet) // copy logical font from user's initialization buffer (if needed)
- ATL::Checked::memcpy_s(&m_lf, sizeof(m_lf), m_cf.lpLogFont, sizeof(m_lf));
- return bRet ? IDOK : IDCANCEL;
- }
- // works only when the dialog is dislayed or after
- void GetCurrentFont(LPLOGFONT lplf) const
- {
- ATLASSERT(lplf != NULL);
- if(m_hWnd != NULL)
- ::SendMessage(m_hWnd, WM_CHOOSEFONT_GETLOGFONT, 0, (LPARAM)lplf);
- else
- *lplf = m_lf;
- }
- // works only when the dialog is dislayed or before
- void SetLogFont(LPLOGFONT lplf)
- {
- ATLASSERT(lplf != NULL);
- if(m_hWnd != NULL)
- {
- ::SendMessage(m_hWnd, WM_CHOOSEFONT_SETLOGFONT, 0, (LPARAM)lplf);
- }
- else
- {
- m_lf = *lplf;
- m_cf.Flags |= CF_INITTOLOGFONTSTRUCT;
- }
- }
- void SetFlags(DWORD dwFlags)
- {
- if(m_hWnd != NULL)
- {
- CHOOSEFONT cf = { sizeof(CHOOSEFONT) };
- cf.Flags = dwFlags;
- ::SendMessage(m_hWnd, WM_CHOOSEFONT_SETFLAGS, 0, (LPARAM)&cf);
- }
- else
- {
- m_cf.Flags = dwFlags;
- }
- }
- // Helpers for parsing information after successful return
- LPCTSTR GetFaceName() const // return the face name of the font
- {
- return (LPCTSTR)m_cf.lpLogFont->lfFaceName;
- }
- LPCTSTR GetStyleName() const // return the style name of the font
- {
- return m_cf.lpszStyle;
- }
- int GetSize() const // return the pt size of the font
- {
- return m_cf.iPointSize;
- }
- COLORREF GetColor() const // return the color of the font
- {
- return m_cf.rgbColors;
- }
- int GetWeight() const // return the chosen font weight
- {
- return (int)m_cf.lpLogFont->lfWeight;
- }
- BOOL IsStrikeOut() const // return TRUE if strikeout
- {
- return (m_cf.lpLogFont->lfStrikeOut) ? TRUE : FALSE;
- }
- BOOL IsUnderline() const // return TRUE if underline
- {
- return (m_cf.lpLogFont->lfUnderline) ? TRUE : FALSE;
- }
- BOOL IsBold() const // return TRUE if bold font
- {
- return (m_cf.lpLogFont->lfWeight == FW_BOLD) ? TRUE : FALSE;
- }
- BOOL IsItalic() const // return TRUE if italic font
- {
- return m_cf.lpLogFont->lfItalic ? TRUE : FALSE;
- }
- };
- class CFontDialog : public CFontDialogImpl<CFontDialog>
- {
- public:
- CFontDialog(LPLOGFONT lplfInitial = NULL,
- DWORD dwFlags = CF_EFFECTS | CF_SCREENFONTS,
- HDC hDCPrinter = NULL,
- HWND hWndParent = NULL)
- : CFontDialogImpl<CFontDialog>(lplfInitial, dwFlags, hDCPrinter, hWndParent)
- { }
- DECLARE_EMPTY_MSG_MAP()
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CRichEditFontDialogImpl - font selection for the Rich Edit ctrl
- #ifdef _RICHEDIT_
- template <class T>
- class ATL_NO_VTABLE CRichEditFontDialogImpl : public CFontDialogImpl< T >
- {
- public:
- CRichEditFontDialogImpl(const CHARFORMAT& charformat,
- DWORD dwFlags = CF_SCREENFONTS,
- HDC hDCPrinter = NULL,
- HWND hWndParent = NULL)
- : CFontDialogImpl< T >(NULL, dwFlags, hDCPrinter, hWndParent)
- {
- this->m_cf.Flags |= CF_INITTOLOGFONTSTRUCT;
- this->m_cf.Flags |= FillInLogFont(charformat);
- this->m_cf.lpLogFont = &this->m_lf;
- if((charformat.dwMask & CFM_COLOR) != 0)
- this->m_cf.rgbColors = charformat.crTextColor;
- }
- void GetCharFormat(CHARFORMAT& cf) const
- {
- USES_CONVERSION;
- cf.dwEffects = 0;
- cf.dwMask = 0;
- if((this->m_cf.Flags & CF_NOSTYLESEL) == 0)
- {
- cf.dwMask |= CFM_BOLD | CFM_ITALIC;
- cf.dwEffects |= this->IsBold() ? CFE_BOLD : 0;
- cf.dwEffects |= this->IsItalic() ? CFE_ITALIC : 0;
- }
- if((this->m_cf.Flags & CF_NOSIZESEL) == 0)
- {
- cf.dwMask |= CFM_SIZE;
- // GetSize() returns in tenths of points so mulitply by 2 to get twips
- cf.yHeight = this->GetSize() * 2;
- }
- if((this->m_cf.Flags & CF_NOFACESEL) == 0)
- {
- cf.dwMask |= CFM_FACE;
- cf.bPitchAndFamily = this->m_cf.lpLogFont->lfPitchAndFamily;
- ATL::Checked::tcscpy_s(cf.szFaceName, _countof(cf.szFaceName), this->GetFaceName());
- }
- if((this->m_cf.Flags & CF_EFFECTS) != 0)
- {
- cf.dwMask |= CFM_UNDERLINE | CFM_STRIKEOUT | CFM_COLOR;
- cf.dwEffects |= this->IsUnderline() ? CFE_UNDERLINE : 0;
- cf.dwEffects |= this->IsStrikeOut() ? CFE_STRIKEOUT : 0;
- cf.crTextColor = this->GetColor();
- }
- if((this->m_cf.Flags & CF_NOSCRIPTSEL) == 0)
- {
- cf.bCharSet = this->m_cf.lpLogFont->lfCharSet;
- cf.dwMask |= CFM_CHARSET;
- }
- cf.yOffset = 0;
- }
- DWORD FillInLogFont(const CHARFORMAT& cf)
- {
- USES_CONVERSION;
- DWORD dwFlags = 0;
- if((cf.dwMask & CFM_SIZE) != 0)
- {
- HDC hDC = ::CreateDC(_T("DISPLAY"), NULL, NULL, NULL);
- LONG yPerInch = ::GetDeviceCaps(hDC, LOGPIXELSY);
- this->m_lf.lfHeight = -(int)((cf.yHeight * yPerInch) / 1440);
- }
- else
- this->m_lf.lfHeight = 0;
- this->m_lf.lfWidth = 0;
- this->m_lf.lfEscapement = 0;
- this->m_lf.lfOrientation = 0;
- if((cf.dwMask & (CFM_ITALIC | CFM_BOLD)) == (CFM_ITALIC | CFM_BOLD))
- {
- this->m_lf.lfWeight = ((cf.dwEffects & CFE_BOLD) != 0) ? FW_BOLD : FW_NORMAL;
- this->m_lf.lfItalic = (BYTE)(((cf.dwEffects & CFE_ITALIC) != 0) ? TRUE : FALSE);
- }
- else
- {
- dwFlags |= CF_NOSTYLESEL;
- this->m_lf.lfWeight = FW_DONTCARE;
- this->m_lf.lfItalic = FALSE;
- }
- if((cf.dwMask & (CFM_UNDERLINE | CFM_STRIKEOUT | CFM_COLOR)) == (CFM_UNDERLINE|CFM_STRIKEOUT|CFM_COLOR))
- {
- dwFlags |= CF_EFFECTS;
- this->m_lf.lfUnderline = (BYTE)(((cf.dwEffects & CFE_UNDERLINE) != 0) ? TRUE : FALSE);
- this->m_lf.lfStrikeOut = (BYTE)(((cf.dwEffects & CFE_STRIKEOUT) != 0) ? TRUE : FALSE);
- }
- else
- {
- this->m_lf.lfUnderline = (BYTE)FALSE;
- this->m_lf.lfStrikeOut = (BYTE)FALSE;
- }
- if((cf.dwMask & CFM_CHARSET) != 0)
- this->m_lf.lfCharSet = cf.bCharSet;
- else
- dwFlags |= CF_NOSCRIPTSEL;
- this->m_lf.lfOutPrecision = OUT_DEFAULT_PRECIS;
- this->m_lf.lfClipPrecision = CLIP_DEFAULT_PRECIS;
- this->m_lf.lfQuality = DEFAULT_QUALITY;
- if((cf.dwMask & CFM_FACE) != 0)
- {
- this->m_lf.lfPitchAndFamily = cf.bPitchAndFamily;
- ATL::Checked::tcscpy_s(this->m_lf.lfFaceName, _countof(this->m_lf.lfFaceName), cf.szFaceName);
- }
- else
- {
- this->m_lf.lfPitchAndFamily = DEFAULT_PITCH|FF_DONTCARE;
- this->m_lf.lfFaceName[0] = (TCHAR)0;
- }
- return dwFlags;
- }
- };
- class CRichEditFontDialog : public CRichEditFontDialogImpl<CRichEditFontDialog>
- {
- public:
- CRichEditFontDialog(const CHARFORMAT& charformat,
- DWORD dwFlags = CF_SCREENFONTS,
- HDC hDCPrinter = NULL,
- HWND hWndParent = NULL)
- : CRichEditFontDialogImpl<CRichEditFontDialog>(charformat, dwFlags, hDCPrinter, hWndParent)
- { }
- DECLARE_EMPTY_MSG_MAP()
- };
- #endif // _RICHEDIT_
- ///////////////////////////////////////////////////////////////////////////////
- // CColorDialogImpl - color selection
- template <class T>
- class ATL_NO_VTABLE CColorDialogImpl : public CCommonDialogImplBase
- {
- public:
- CHOOSECOLOR m_cc;
- // Constructor
- CColorDialogImpl(COLORREF clrInit = 0, DWORD dwFlags = 0, HWND hWndParent = NULL)
- {
- memset(&m_cc, 0, sizeof(m_cc));
- m_cc.lStructSize = sizeof(m_cc);
- m_cc.lpCustColors = GetCustomColors();
- m_cc.hwndOwner = hWndParent;
- m_cc.Flags = dwFlags | CC_ENABLEHOOK;
- m_cc.lpfnHook = (LPCCHOOKPROC)T::HookProc;
- if(clrInit != 0)
- {
- m_cc.rgbResult = clrInit;
- m_cc.Flags |= CC_RGBINIT;
- }
- }
- // Operations
- INT_PTR DoModal(HWND hWndParent = ::GetActiveWindow())
- {
- ATLASSERT((m_cc.Flags & CC_ENABLEHOOK) != 0);
- ATLASSERT(m_cc.lpfnHook != NULL); // can still be a user hook
- if(m_cc.hwndOwner == NULL) // set only if not specified before
- m_cc.hwndOwner = hWndParent;
- ATLASSERT(m_hWnd == NULL);
- // Allocate the thunk structure here, where we can fail gracefully.
- BOOL bRetTh = m_thunk.Init(NULL, NULL);
- if(bRetTh == FALSE)
- {
- ::SetLastError(ERROR_OUTOFMEMORY);
- return -1;
- }
- ModuleHelper::AddCreateWndData(&m_thunk.cd, (CCommonDialogImplBase*)this);
- BOOL bRet = ::ChooseColor(&m_cc);
- m_hWnd = NULL;
- return bRet ? IDOK : IDCANCEL;
- }
- // Set the current color while dialog is displayed
- void SetCurrentColor(COLORREF clr)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- SendMessage(_GetSetRGBMessage(), 0, (LPARAM)clr);
- }
- // Get the selected color after DoModal returns, or in OnColorOK
- COLORREF GetColor() const
- {
- return m_cc.rgbResult;
- }
- // Special override for the color dialog
- static UINT_PTR APIENTRY HookProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
- {
- if((uMsg != WM_INITDIALOG) && (uMsg != _GetColorOKMessage()))
- return 0;
- LPCHOOSECOLOR lpCC = (LPCHOOSECOLOR)lParam;
- CCommonDialogImplBase* pT = NULL;
- if(uMsg == WM_INITDIALOG)
- {
- pT = (CCommonDialogImplBase*)ModuleHelper::ExtractCreateWndData();
- lpCC->lCustData = (LPARAM)pT;
- ATLASSERT(pT != NULL);
- ATLASSERT(pT->m_hWnd == NULL);
- ATLASSERT(::IsWindow(hWnd));
- // subclass dialog's window
- if(!pT->SubclassWindow(hWnd))
- {
- ATLTRACE2(atlTraceUI, 0, _T("Subclassing a Color common dialog failed\n"));
- return 0;
- }
- }
- else if(uMsg == _GetColorOKMessage())
- {
- pT = (CCommonDialogImplBase*)lpCC->lCustData;
- ATLASSERT(pT != NULL);
- ATLASSERT(::IsWindow(pT->m_hWnd));
- }
- else
- {
- ATLASSERT(FALSE);
- return 0;
- }
- // pass to the message map
- LRESULT lRes = 0;
- if(pT->ProcessWindowMessage(pT->m_hWnd, uMsg, wParam, lParam, lRes, 0) == FALSE)
- return 0;
- return lRes;
- }
- // Helpers
- static COLORREF* GetCustomColors()
- {
- static COLORREF rgbCustomColors[16] =
- {
- RGB(255, 255, 255), RGB(255, 255, 255),
- RGB(255, 255, 255), RGB(255, 255, 255),
- RGB(255, 255, 255), RGB(255, 255, 255),
- RGB(255, 255, 255), RGB(255, 255, 255),
- RGB(255, 255, 255), RGB(255, 255, 255),
- RGB(255, 255, 255), RGB(255, 255, 255),
- RGB(255, 255, 255), RGB(255, 255, 255),
- RGB(255, 255, 255), RGB(255, 255, 255),
- };
- return rgbCustomColors;
- }
- static UINT _GetSetRGBMessage()
- {
- static UINT uSetRGBMessage = 0;
- if(uSetRGBMessage == 0)
- {
- CStaticDataInitCriticalSectionLock lock;
- if(FAILED(lock.Lock()))
- {
- ATLTRACE2(atlTraceUI, 0, _T("ERROR : Unable to lock critical section in CColorDialogImpl::_GetSetRGBMessage.\n"));
- ATLASSERT(FALSE);
- return 0;
- }
- if(uSetRGBMessage == 0)
- uSetRGBMessage = ::RegisterWindowMessage(SETRGBSTRING);
- lock.Unlock();
- }
- ATLASSERT(uSetRGBMessage != 0);
- return uSetRGBMessage;
- }
- static UINT _GetColorOKMessage()
- {
- static UINT uColorOKMessage = 0;
- if(uColorOKMessage == 0)
- {
- CStaticDataInitCriticalSectionLock lock;
- if(FAILED(lock.Lock()))
- {
- ATLTRACE2(atlTraceUI, 0, _T("ERROR : Unable to lock critical section in CColorDialogImpl::_GetColorOKMessage.\n"));
- ATLASSERT(FALSE);
- return 0;
- }
- if(uColorOKMessage == 0)
- uColorOKMessage = ::RegisterWindowMessage(COLOROKSTRING);
- lock.Unlock();
- }
- ATLASSERT(uColorOKMessage != 0);
- return uColorOKMessage;
- }
- // Message map and handlers
- BEGIN_MSG_MAP(CColorDialogImpl)
- MESSAGE_HANDLER(_GetColorOKMessage(), _OnColorOK)
- END_MSG_MAP()
- LRESULT _OnColorOK(UINT, WPARAM, LPARAM, BOOL&)
- {
- T* pT = static_cast<T*>(this);
- return pT->OnColorOK();
- }
- // Overrideable
- BOOL OnColorOK() // validate color
- {
- return FALSE;
- }
- };
- class CColorDialog : public CColorDialogImpl<CColorDialog>
- {
- public:
- CColorDialog(COLORREF clrInit = 0, DWORD dwFlags = 0, HWND hWndParent = NULL)
- : CColorDialogImpl<CColorDialog>(clrInit, dwFlags, hWndParent)
- { }
- // override base class map and references to handlers
- DECLARE_EMPTY_MSG_MAP()
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CPrintDialogImpl - used for Print... and PrintSetup...
- // global helper
- static inline HDC _AtlCreateDC(HGLOBAL hDevNames, HGLOBAL hDevMode)
- {
- if(hDevNames == NULL)
- return NULL;
- LPDEVNAMES lpDevNames = (LPDEVNAMES)::GlobalLock(hDevNames);
- LPDEVMODE lpDevMode = (hDevMode != NULL) ? (LPDEVMODE)::GlobalLock(hDevMode) : NULL;
- if(lpDevNames == NULL)
- return NULL;
- HDC hDC = ::CreateDC((LPCTSTR)lpDevNames + lpDevNames->wDriverOffset,
- (LPCTSTR)lpDevNames + lpDevNames->wDeviceOffset,
- (LPCTSTR)lpDevNames + lpDevNames->wOutputOffset,
- lpDevMode);
- ::GlobalUnlock(hDevNames);
- if(hDevMode != NULL)
- ::GlobalUnlock(hDevMode);
- return hDC;
- }
- #pragma warning(push)
- #pragma warning(disable: 4512) // assignment operator could not be generated
- template <class T>
- class ATL_NO_VTABLE CPrintDialogImpl : public CCommonDialogImplBase
- {
- public:
- // print dialog parameter block (note this is a reference)
- PRINTDLG& m_pd;
- // Constructors
- CPrintDialogImpl(BOOL bPrintSetupOnly = FALSE, // TRUE for Print Setup, FALSE for Print Dialog
- DWORD dwFlags = PD_ALLPAGES | PD_USEDEVMODECOPIES | PD_NOPAGENUMS | PD_NOSELECTION,
- HWND hWndParent = NULL)
- : m_pd(m_pdActual)
- {
- memset(&m_pdActual, 0, sizeof(m_pdActual));
- m_pd.lStructSize = sizeof(m_pdActual);
- m_pd.hwndOwner = hWndParent;
- m_pd.Flags = (dwFlags | PD_ENABLEPRINTHOOK | PD_ENABLESETUPHOOK);
- m_pd.lpfnPrintHook = (LPPRINTHOOKPROC)T::HookProc;
- m_pd.lpfnSetupHook = (LPSETUPHOOKPROC)T::HookProc;
- if(bPrintSetupOnly)
- m_pd.Flags |= PD_PRINTSETUP;
- else
- m_pd.Flags |= PD_RETURNDC;
- m_pd.Flags &= ~PD_RETURNIC; // do not support information context
- }
- // Operations
- INT_PTR DoModal(HWND hWndParent = ::GetActiveWindow())
- {
- ATLASSERT((m_pd.Flags & PD_ENABLEPRINTHOOK) != 0);
- ATLASSERT((m_pd.Flags & PD_ENABLESETUPHOOK) != 0);
- ATLASSERT(m_pd.lpfnPrintHook != NULL); // can still be a user hook
- ATLASSERT(m_pd.lpfnSetupHook != NULL); // can still be a user hook
- ATLASSERT((m_pd.Flags & PD_RETURNDEFAULT) == 0); // use GetDefaults for this
- if(m_pd.hwndOwner == NULL) // set only if not specified before
- m_pd.hwndOwner = hWndParent;
- ATLASSERT(m_hWnd == NULL);
- // Allocate the thunk structure here, where we can fail gracefully.
- BOOL bRetTh = m_thunk.Init(NULL, NULL);
- if(bRetTh == FALSE)
- {
- ::SetLastError(ERROR_OUTOFMEMORY);
- return -1;
- }
- ModuleHelper::AddCreateWndData(&m_thunk.cd, (CCommonDialogImplBase*)this);
- BOOL bRet = ::PrintDlg(&m_pd);
- m_hWnd = NULL;
- return bRet ? IDOK : IDCANCEL;
- }
- // GetDefaults will not display a dialog but will get device defaults
- BOOL GetDefaults()
- {
- m_pd.Flags |= PD_RETURNDEFAULT;
- ATLASSERT(m_pd.hDevMode == NULL); // must be NULL
- ATLASSERT(m_pd.hDevNames == NULL); // must be NULL
- return ::PrintDlg(&m_pd);
- }
- // Helpers for parsing information after successful return num. copies requested
- int GetCopies() const
- {
- if((m_pd.Flags & PD_USEDEVMODECOPIES) != 0)
- {
- LPDEVMODE lpDevMode = GetDevMode();
- return (lpDevMode != NULL) ? lpDevMode->dmCopies : -1;
- }
- return m_pd.nCopies;
- }
- BOOL PrintCollate() const // TRUE if collate checked
- {
- return ((m_pd.Flags & PD_COLLATE) != 0) ? TRUE : FALSE;
- }
- BOOL PrintSelection() const // TRUE if printing selection
- {
- return ((m_pd.Flags & PD_SELECTION) != 0) ? TRUE : FALSE;
- }
- BOOL PrintAll() const // TRUE if printing all pages
- {
- return (!PrintRange() && !PrintSelection()) ? TRUE : FALSE;
- }
- BOOL PrintRange() const // TRUE if printing page range
- {
- return ((m_pd.Flags & PD_PAGENUMS) != 0) ? TRUE : FALSE;
- }
- BOOL PrintToFile() const // TRUE if printing to a file
- {
- return ((m_pd.Flags & PD_PRINTTOFILE) != 0) ? TRUE : FALSE;
- }
- int GetFromPage() const // starting page if valid
- {
- return PrintRange() ? m_pd.nFromPage : -1;
- }
- int GetToPage() const // ending page if valid
- {
- return PrintRange() ? m_pd.nToPage : -1;
- }
- LPDEVMODE GetDevMode() const // return DEVMODE
- {
- if(m_pd.hDevMode == NULL)
- return NULL;
- return (LPDEVMODE)::GlobalLock(m_pd.hDevMode);
- }
- LPCTSTR GetDriverName() const // return driver name
- {
- if(m_pd.hDevNames == NULL)
- return NULL;
- LPDEVNAMES lpDev = (LPDEVNAMES)::GlobalLock(m_pd.hDevNames);
- if(lpDev == NULL)
- return NULL;
- return (LPCTSTR)lpDev + lpDev->wDriverOffset;
- }
- LPCTSTR GetDeviceName() const // return device name
- {
- if(m_pd.hDevNames == NULL)
- return NULL;
- LPDEVNAMES lpDev = (LPDEVNAMES)::GlobalLock(m_pd.hDevNames);
- if(lpDev == NULL)
- return NULL;
- return (LPCTSTR)lpDev + lpDev->wDeviceOffset;
- }
- LPCTSTR GetPortName() const // return output port name
- {
- if(m_pd.hDevNames == NULL)
- return NULL;
- LPDEVNAMES lpDev = (LPDEVNAMES)::GlobalLock(m_pd.hDevNames);
- if(lpDev == NULL)
- return NULL;
- return (LPCTSTR)lpDev + lpDev->wOutputOffset;
- }
- HDC GetPrinterDC() const // return HDC (caller must delete)
- {
- ATLASSERT((m_pd.Flags & PD_RETURNDC) != 0);
- return m_pd.hDC;
- }
- // This helper creates a DC based on the DEVNAMES and DEVMODE structures.
- // This DC is returned, but also stored in m_pd.hDC as though it had been
- // returned by CommDlg. It is assumed that any previously obtained DC
- // has been/will be deleted by the user. This may be
- // used without ever invoking the print/print setup dialogs.
- HDC CreatePrinterDC()
- {
- m_pd.hDC = _AtlCreateDC(m_pd.hDevNames, m_pd.hDevMode);
- return m_pd.hDC;
- }
- // Implementation
- PRINTDLG m_pdActual; // the Print/Print Setup need to share this
- // The following handle the case of print setup... from the print dialog
- CPrintDialogImpl(PRINTDLG& pdInit) : m_pd(pdInit)
- { }
- BEGIN_MSG_MAP(CPrintDialogImpl)
- #ifdef psh1
- COMMAND_ID_HANDLER(psh1, OnPrintSetup) // print setup button when print is displayed
- #else // !psh1
- COMMAND_ID_HANDLER(0x0400, OnPrintSetup) // value from dlgs.h
- #endif // !psh1
- END_MSG_MAP()
- LRESULT OnPrintSetup(WORD wNotifyCode, WORD wID, HWND hWndCtl, BOOL& /*bHandled*/)
- {
- T dlgSetup(m_pd);
- ModuleHelper::AddCreateWndData(&dlgSetup.m_thunk.cd, (CCommonDialogImplBase*)&dlgSetup);
- return DefWindowProc(WM_COMMAND, MAKEWPARAM(wID, wNotifyCode), (LPARAM)hWndCtl);
- }
- };
- class CPrintDialog : public CPrintDialogImpl<CPrintDialog>
- {
- public:
- CPrintDialog(BOOL bPrintSetupOnly = FALSE,
- DWORD dwFlags = PD_ALLPAGES | PD_USEDEVMODECOPIES | PD_NOPAGENUMS | PD_NOSELECTION,
- HWND hWndParent = NULL)
- : CPrintDialogImpl<CPrintDialog>(bPrintSetupOnly, dwFlags, hWndParent)
- { }
- CPrintDialog(PRINTDLG& pdInit) : CPrintDialogImpl<CPrintDialog>(pdInit)
- { }
- };
- #pragma warning(pop)
- ///////////////////////////////////////////////////////////////////////////////
- // CPrintDialogExImpl - new print dialog for Windows 2000
- } // namespace WTL
- #include <atlcom.h>
- extern "C" const __declspec(selectany) IID IID_IPrintDialogCallback = {0x5852a2c3, 0x6530, 0x11d1, {0xb6, 0xa3, 0x0, 0x0, 0xf8, 0x75, 0x7b, 0xf9}};
- extern "C" const __declspec(selectany) IID IID_IPrintDialogServices = {0x509aaeda, 0x5639, 0x11d1, {0xb6, 0xa1, 0x0, 0x0, 0xf8, 0x75, 0x7b, 0xf9}};
- namespace WTL
- {
- template <class T>
- class ATL_NO_VTABLE CPrintDialogExImpl :
- public ATL::CWindow,
- public ATL::CMessageMap,
- public IPrintDialogCallback,
- public ATL::IObjectWithSiteImpl< T >
- {
- public:
- PRINTDLGEX m_pdex;
- // Constructor
- CPrintDialogExImpl(DWORD dwFlags = PD_ALLPAGES | PD_USEDEVMODECOPIES | PD_NOPAGENUMS | PD_NOSELECTION | PD_NOCURRENTPAGE,
- HWND hWndParent = NULL)
- {
- memset(&m_pdex, 0, sizeof(m_pdex));
- m_pdex.lStructSize = sizeof(PRINTDLGEX);
- m_pdex.hwndOwner = hWndParent;
- m_pdex.Flags = dwFlags;
- m_pdex.nStartPage = START_PAGE_GENERAL;
- // callback object will be set in DoModal
- m_pdex.Flags &= ~PD_RETURNIC; // do not support information context
- }
- // Operations
- HRESULT DoModal(HWND hWndParent = ::GetActiveWindow())
- {
- ATLASSERT(m_hWnd == NULL);
- ATLASSERT((m_pdex.Flags & PD_RETURNDEFAULT) == 0); // use GetDefaults for this
- if(m_pdex.hwndOwner == NULL) // set only if not specified before
- m_pdex.hwndOwner = hWndParent;
- T* pT = static_cast<T*>(this);
- m_pdex.lpCallback = (IUnknown*)(IPrintDialogCallback*)pT;
- HRESULT hResult = ::PrintDlgEx(&m_pdex);
- m_hWnd = NULL;
- return hResult;
- }
- BOOL EndDialog(INT_PTR /*nRetCode*/ = 0)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- SendMessage(WM_COMMAND, MAKEWPARAM(IDABORT, 0));
- return TRUE;
- }
- // GetDefaults will not display a dialog but will get device defaults
- HRESULT GetDefaults()
- {
- ATLASSERT(m_pdex.hDevMode == NULL); // must be NULL
- ATLASSERT(m_pdex.hDevNames == NULL); // must be NULL
- if(m_pdex.hwndOwner == NULL) // set only if not specified before
- m_pdex.hwndOwner = ::GetActiveWindow();
- m_pdex.Flags |= PD_RETURNDEFAULT;
- HRESULT hRet = ::PrintDlgEx(&m_pdex);
- m_pdex.Flags &= ~PD_RETURNDEFAULT;
- return hRet;
- }
- // Helpers for parsing information after successful return num. copies requested
- int GetCopies() const
- {
- if((m_pdex.Flags & PD_USEDEVMODECOPIES) != 0)
- {
- LPDEVMODE lpDevMode = GetDevMode();
- return (lpDevMode != NULL) ? lpDevMode->dmCopies : -1;
- }
- return m_pdex.nCopies;
- }
- BOOL PrintCollate() const // TRUE if collate checked
- {
- return ((m_pdex.Flags & PD_COLLATE) != 0) ? TRUE : FALSE;
- }
- BOOL PrintSelection() const // TRUE if printing selection
- {
- return ((m_pdex.Flags & PD_SELECTION) != 0) ? TRUE : FALSE;
- }
- BOOL PrintAll() const // TRUE if printing all pages
- {
- return (!PrintRange() && !PrintSelection()) ? TRUE : FALSE;
- }
- BOOL PrintRange() const // TRUE if printing page range
- {
- return ((m_pdex.Flags & PD_PAGENUMS) != 0) ? TRUE : FALSE;
- }
- BOOL PrintToFile() const // TRUE if printing to a file
- {
- return ((m_pdex.Flags & PD_PRINTTOFILE) != 0) ? TRUE : FALSE;
- }
- LPDEVMODE GetDevMode() const // return DEVMODE
- {
- if(m_pdex.hDevMode == NULL)
- return NULL;
- return (LPDEVMODE)::GlobalLock(m_pdex.hDevMode);
- }
- LPCTSTR GetDriverName() const // return driver name
- {
- if(m_pdex.hDevNames == NULL)
- return NULL;
- LPDEVNAMES lpDev = (LPDEVNAMES)::GlobalLock(m_pdex.hDevNames);
- if(lpDev == NULL)
- return NULL;
- return (LPCTSTR)lpDev + lpDev->wDriverOffset;
- }
- LPCTSTR GetDeviceName() const // return device name
- {
- if(m_pdex.hDevNames == NULL)
- return NULL;
- LPDEVNAMES lpDev = (LPDEVNAMES)::GlobalLock(m_pdex.hDevNames);
- if(lpDev == NULL)
- return NULL;
- return (LPCTSTR)lpDev + lpDev->wDeviceOffset;
- }
- LPCTSTR GetPortName() const // return output port name
- {
- if(m_pdex.hDevNames == NULL)
- return NULL;
- LPDEVNAMES lpDev = (LPDEVNAMES)::GlobalLock(m_pdex.hDevNames);
- if(lpDev == NULL)
- return NULL;
- return (LPCTSTR)lpDev + lpDev->wOutputOffset;
- }
- HDC GetPrinterDC() const // return HDC (caller must delete)
- {
- ATLASSERT((m_pdex.Flags & PD_RETURNDC) != 0);
- return m_pdex.hDC;
- }
- // This helper creates a DC based on the DEVNAMES and DEVMODE structures.
- // This DC is returned, but also stored in m_pdex.hDC as though it had been
- // returned by CommDlg. It is assumed that any previously obtained DC
- // has been/will be deleted by the user. This may be
- // used without ever invoking the print/print setup dialogs.
- HDC CreatePrinterDC()
- {
- m_pdex.hDC = _AtlCreateDC(m_pdex.hDevNames, m_pdex.hDevMode);
- return m_pdex.hDC;
- }
- // Implementation - interfaces
- // IUnknown
- STDMETHOD(QueryInterface)(REFIID riid, void** ppvObject)
- {
- if(ppvObject == NULL)
- return E_POINTER;
- T* pT = static_cast<T*>(this);
- if(IsEqualGUID(riid, IID_IUnknown) || IsEqualGUID(riid, IID_IPrintDialogCallback))
- {
- *ppvObject = (IPrintDialogCallback*)pT;
- // AddRef() not needed
- return S_OK;
- }
- else if(IsEqualGUID(riid, IID_IObjectWithSite))
- {
- *ppvObject = (IObjectWithSite*)pT;
- // AddRef() not needed
- return S_OK;
- }
- return E_NOINTERFACE;
- }
- virtual ULONG STDMETHODCALLTYPE AddRef()
- {
- return 1;
- }
- virtual ULONG STDMETHODCALLTYPE Release()
- {
- return 1;
- }
- // IPrintDialogCallback
- STDMETHOD(InitDone)()
- {
- return S_FALSE;
- }
- STDMETHOD(SelectionChange)()
- {
- return S_FALSE;
- }
- STDMETHOD(HandleMessage)(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam, LRESULT* plResult)
- {
- // set up m_hWnd the first time
- if(m_hWnd == NULL)
- Attach(hWnd);
- // call message map
- HRESULT hRet = ProcessWindowMessage(hWnd, uMsg, wParam, lParam, *plResult, 0) ? S_OK : S_FALSE;
- if((hRet == S_OK) && (uMsg == WM_NOTIFY)) // return in DWLP_MSGRESULT
- ::SetWindowLongPtr(GetParent(), DWLP_MSGRESULT, (LONG_PTR)*plResult);
- if((uMsg == WM_INITDIALOG) && (hRet == S_OK) && ((BOOL)*plResult != FALSE))
- hRet = S_FALSE;
- return hRet;
- }
- };
- class CPrintDialogEx : public CPrintDialogExImpl<CPrintDialogEx>
- {
- public:
- CPrintDialogEx(
- DWORD dwFlags = PD_ALLPAGES | PD_USEDEVMODECOPIES | PD_NOPAGENUMS | PD_NOSELECTION | PD_NOCURRENTPAGE,
- HWND hWndParent = NULL)
- : CPrintDialogExImpl<CPrintDialogEx>(dwFlags, hWndParent)
- { }
- DECLARE_EMPTY_MSG_MAP()
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CPageSetupDialogImpl - Page Setup dialog
- template <class T>
- class ATL_NO_VTABLE CPageSetupDialogImpl : public CCommonDialogImplBase
- {
- public:
- PAGESETUPDLG m_psd;
- ATL::CWndProcThunk m_thunkPaint;
- // Constructors
- CPageSetupDialogImpl(DWORD dwFlags = PSD_MARGINS | PSD_INWININIINTLMEASURE, HWND hWndParent = NULL)
- {
- memset(&m_psd, 0, sizeof(m_psd));
- m_psd.lStructSize = sizeof(m_psd);
- m_psd.hwndOwner = hWndParent;
- m_psd.Flags = (dwFlags | PSD_ENABLEPAGESETUPHOOK | PSD_ENABLEPAGEPAINTHOOK);
- m_psd.lpfnPageSetupHook = (LPPAGESETUPHOOK)T::HookProc;
- m_thunkPaint.Init((WNDPROC)T::PaintHookProc, this);
- m_psd.lpfnPagePaintHook = (LPPAGEPAINTHOOK)m_thunkPaint.GetWNDPROC();
- }
- DECLARE_EMPTY_MSG_MAP()
- // Attributes
- LPDEVMODE GetDevMode() const // return DEVMODE
- {
- if(m_psd.hDevMode == NULL)
- return NULL;
- return (LPDEVMODE)::GlobalLock(m_psd.hDevMode);
- }
- LPCTSTR GetDriverName() const // return driver name
- {
- if(m_psd.hDevNames == NULL)
- return NULL;
- LPDEVNAMES lpDev = (LPDEVNAMES)::GlobalLock(m_psd.hDevNames);
- return (LPCTSTR)lpDev + lpDev->wDriverOffset;
- }
- LPCTSTR GetDeviceName() const // return device name
- {
- if(m_psd.hDevNames == NULL)
- return NULL;
- LPDEVNAMES lpDev = (LPDEVNAMES)::GlobalLock(m_psd.hDevNames);
- return (LPCTSTR)lpDev + lpDev->wDeviceOffset;
- }
- LPCTSTR GetPortName() const // return output port name
- {
- if(m_psd.hDevNames == NULL)
- return NULL;
- LPDEVNAMES lpDev = (LPDEVNAMES)::GlobalLock(m_psd.hDevNames);
- return (LPCTSTR)lpDev + lpDev->wOutputOffset;
- }
- HDC CreatePrinterDC()
- {
- return _AtlCreateDC(m_psd.hDevNames, m_psd.hDevMode);
- }
- SIZE GetPaperSize() const
- {
- SIZE size = { m_psd.ptPaperSize.x, m_psd.ptPaperSize.y };
- return size;
- }
- void GetMargins(LPRECT lpRectMargins, LPRECT lpRectMinMargins) const
- {
- if(lpRectMargins != NULL)
- *lpRectMargins = m_psd.rtMargin;
- if(lpRectMinMargins != NULL)
- *lpRectMinMargins = m_psd.rtMinMargin;
- }
- // Operations
- INT_PTR DoModal(HWND hWndParent = ::GetActiveWindow())
- {
- ATLASSERT((m_psd.Flags & PSD_ENABLEPAGESETUPHOOK) != 0);
- ATLASSERT((m_psd.Flags & PSD_ENABLEPAGEPAINTHOOK) != 0);
- ATLASSERT(m_psd.lpfnPageSetupHook != NULL); // can still be a user hook
- ATLASSERT(m_psd.lpfnPagePaintHook != NULL); // can still be a user hook
- if(m_psd.hwndOwner == NULL) // set only if not specified before
- m_psd.hwndOwner = hWndParent;
- ATLASSERT(m_hWnd == NULL);
- // Allocate the thunk structure here, where we can fail gracefully.
- BOOL bRetTh = m_thunk.Init(NULL, NULL);
- if(bRetTh == FALSE)
- {
- ::SetLastError(ERROR_OUTOFMEMORY);
- return -1;
- }
- ModuleHelper::AddCreateWndData(&m_thunk.cd, (CCommonDialogImplBase*)this);
- BOOL bRet = ::PageSetupDlg(&m_psd);
- m_hWnd = NULL;
- return bRet ? IDOK : IDCANCEL;
- }
- // Implementation
- static UINT_PTR CALLBACK PaintHookProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
- {
- T* pT = (T*)hWnd;
- UINT_PTR uRet = 0;
- switch(uMsg)
- {
- case WM_PSD_PAGESETUPDLG:
- uRet = pT->PreDrawPage(LOWORD(wParam), HIWORD(wParam), (LPPAGESETUPDLG)lParam);
- break;
- case WM_PSD_FULLPAGERECT:
- case WM_PSD_MINMARGINRECT:
- case WM_PSD_MARGINRECT:
- case WM_PSD_GREEKTEXTRECT:
- case WM_PSD_ENVSTAMPRECT:
- case WM_PSD_YAFULLPAGERECT:
- uRet = pT->OnDrawPage(uMsg, (HDC)wParam, (LPRECT)lParam);
- break;
- default:
- ATLTRACE2(atlTraceUI, 0, _T("CPageSetupDialogImpl::PaintHookProc - unknown message received\n"));
- break;
- }
- return uRet;
- }
- // Overridables
- UINT_PTR PreDrawPage(WORD /*wPaper*/, WORD /*wFlags*/, LPPAGESETUPDLG /*pPSD*/)
- {
- // return 1 to prevent any more drawing
- return 0;
- }
- UINT_PTR OnDrawPage(UINT /*uMsg*/, HDC /*hDC*/, LPRECT /*lpRect*/)
- {
- return 0; // do the default
- }
- };
- class CPageSetupDialog : public CPageSetupDialogImpl<CPageSetupDialog>
- {
- public:
- CPageSetupDialog(DWORD dwFlags = PSD_MARGINS | PSD_INWININIINTLMEASURE, HWND hWndParent = NULL)
- : CPageSetupDialogImpl<CPageSetupDialog>(dwFlags, hWndParent)
- { }
- // override PaintHookProc and references to handlers
- static UINT_PTR CALLBACK PaintHookProc(HWND, UINT, WPARAM, LPARAM)
- {
- return 0;
- }
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CFindReplaceDialogImpl - Find/FindReplace modeless dialogs
- template <class T>
- class ATL_NO_VTABLE CFindReplaceDialogImpl : public CCommonDialogImplBase
- {
- public:
- enum { _cchFindReplaceBuffer = 128 };
- FINDREPLACE m_fr;
- TCHAR m_szFindWhat[_cchFindReplaceBuffer];
- TCHAR m_szReplaceWith[_cchFindReplaceBuffer];
- // Constructors
- CFindReplaceDialogImpl()
- {
- memset(&m_fr, 0, sizeof(m_fr));
- m_szFindWhat[0] = _T('\0');
- m_szReplaceWith[0] = _T('\0');
- m_fr.lStructSize = sizeof(m_fr);
- m_fr.Flags = FR_ENABLEHOOK;
- m_fr.lpfnHook = (LPFRHOOKPROC)T::HookProc;
- m_fr.lpstrFindWhat = (LPTSTR)m_szFindWhat;
- m_fr.wFindWhatLen = _cchFindReplaceBuffer;
- m_fr.lpstrReplaceWith = (LPTSTR)m_szReplaceWith;
- m_fr.wReplaceWithLen = _cchFindReplaceBuffer;
- }
- // Note: You must allocate the object on the heap.
- // If you do not, you must override OnFinalMessage()
- virtual void OnFinalMessage(HWND /*hWnd*/)
- {
- delete this;
- }
- HWND Create(BOOL bFindDialogOnly, // TRUE for Find, FALSE for FindReplace
- LPCTSTR lpszFindWhat,
- LPCTSTR lpszReplaceWith = NULL,
- DWORD dwFlags = FR_DOWN,
- HWND hWndParent = NULL)
- {
- ATLASSERT((m_fr.Flags & FR_ENABLEHOOK) != 0);
- ATLASSERT(m_fr.lpfnHook != NULL);
- m_fr.Flags |= dwFlags;
- if(hWndParent == NULL)
- m_fr.hwndOwner = ::GetActiveWindow();
- else
- m_fr.hwndOwner = hWndParent;
- ATLASSERT(m_fr.hwndOwner != NULL); // must have an owner for modeless dialog
- if(lpszFindWhat != NULL)
- ATL::Checked::tcsncpy_s(m_szFindWhat, _countof(m_szFindWhat), lpszFindWhat, _TRUNCATE);
- if(lpszReplaceWith != NULL)
- ATL::Checked::tcsncpy_s(m_szReplaceWith, _countof(m_szReplaceWith), lpszReplaceWith, _TRUNCATE);
- ATLASSERT(m_hWnd == NULL);
- // Allocate the thunk structure here, where we can fail gracefully.
- BOOL bRet = m_thunk.Init(NULL, NULL);
- if(bRet == FALSE)
- {
- ::SetLastError(ERROR_OUTOFMEMORY);
- return NULL;
- }
- ModuleHelper::AddCreateWndData(&m_thunk.cd, (CCommonDialogImplBase*)this);
- HWND hWnd = NULL;
- if(bFindDialogOnly)
- hWnd = ::FindText(&m_fr);
- else
- hWnd = ::ReplaceText(&m_fr);
- ATLASSERT(m_hWnd == hWnd);
- return hWnd;
- }
- static UINT GetFindReplaceMsg()
- {
- static const UINT nMsgFindReplace = ::RegisterWindowMessage(FINDMSGSTRING);
- return nMsgFindReplace;
- }
- // call while handling FINDMSGSTRING registered message
- // to retreive the object
- static T* PASCAL GetNotifier(LPARAM lParam)
- {
- ATLASSERT(lParam != NULL);
- T* pDlg = (T*)(lParam - offsetof(T, m_fr));
- return pDlg;
- }
- // Operations
- // Helpers for parsing information after successful return
- LPCTSTR GetFindString() const // get find string
- {
- return (LPCTSTR)m_fr.lpstrFindWhat;
- }
- LPCTSTR GetReplaceString() const // get replacement string
- {
- return (LPCTSTR)m_fr.lpstrReplaceWith;
- }
- BOOL SearchDown() const // TRUE if search down, FALSE is up
- {
- return ((m_fr.Flags & FR_DOWN) != 0) ? TRUE : FALSE;
- }
- BOOL FindNext() const // TRUE if command is find next
- {
- return ((m_fr.Flags & FR_FINDNEXT) != 0) ? TRUE : FALSE;
- }
- BOOL MatchCase() const // TRUE if matching case
- {
- return ((m_fr.Flags & FR_MATCHCASE) != 0) ? TRUE : FALSE;
- }
- BOOL MatchWholeWord() const // TRUE if matching whole words only
- {
- return ((m_fr.Flags & FR_WHOLEWORD) != 0) ? TRUE : FALSE;
- }
- BOOL ReplaceCurrent() const // TRUE if replacing current string
- {
- return ((m_fr. Flags & FR_REPLACE) != 0) ? TRUE : FALSE;
- }
- BOOL ReplaceAll() const // TRUE if replacing all occurrences
- {
- return ((m_fr.Flags & FR_REPLACEALL) != 0) ? TRUE : FALSE;
- }
- BOOL IsTerminating() const // TRUE if terminating dialog
- {
- return ((m_fr.Flags & FR_DIALOGTERM) != 0) ? TRUE : FALSE ;
- }
- };
- class CFindReplaceDialog : public CFindReplaceDialogImpl<CFindReplaceDialog>
- {
- public:
- DECLARE_EMPTY_MSG_MAP()
- };
- /////////////////////////////////////////////////////////////////////////
- // CDialogBaseUnits - Dialog Units helper
- //
- class CDialogBaseUnits
- {
- public:
- SIZE m_sizeUnits;
- // Constructors
- CDialogBaseUnits()
- {
- // The base units of the out-dated System Font
- LONG nDlgBaseUnits = ::GetDialogBaseUnits();
- m_sizeUnits.cx = LOWORD(nDlgBaseUnits);
- m_sizeUnits.cy = HIWORD(nDlgBaseUnits);
- }
- CDialogBaseUnits(HWND hWnd)
- {
- if(!InitDialogBaseUnits(hWnd)) {
- LONG nDlgBaseUnits = ::GetDialogBaseUnits();
- m_sizeUnits.cx = LOWORD(nDlgBaseUnits);
- m_sizeUnits.cy = HIWORD(nDlgBaseUnits);
- }
- }
- CDialogBaseUnits(HFONT hFont, HWND hWnd = NULL)
- {
- if(!InitDialogBaseUnits(hFont, hWnd)) {
- LONG nDlgBaseUnits = ::GetDialogBaseUnits();
- m_sizeUnits.cx = LOWORD(nDlgBaseUnits);
- m_sizeUnits.cy = HIWORD(nDlgBaseUnits);
- }
- }
- CDialogBaseUnits(const LOGFONT& lf, HWND hWnd = NULL)
- {
- if(!InitDialogBaseUnits(lf, hWnd)) {
- LONG nDlgBaseUnits = ::GetDialogBaseUnits();
- m_sizeUnits.cx = LOWORD(nDlgBaseUnits);
- m_sizeUnits.cy = HIWORD(nDlgBaseUnits);
- }
- }
- // Operations
- BOOL InitDialogBaseUnits(HWND hWnd)
- {
- ATLASSERT(::IsWindow(hWnd));
- RECT rc = { 0, 0, 4, 8 };
- if(!::MapDialogRect(hWnd, &rc)) return FALSE;
- m_sizeUnits.cx = rc.right;
- m_sizeUnits.cy = rc.bottom;
- return TRUE;
- }
- BOOL InitDialogBaseUnits(const LOGFONT& lf, HWND hWnd = NULL)
- {
- CFont font;
- font.CreateFontIndirect(&lf);
- if(font.IsNull()) return FALSE;
- return InitDialogBaseUnits(font, hWnd);
- }
- BOOL InitDialogBaseUnits(HFONT hFont, HWND hWnd = NULL)
- {
- ATLASSERT(hFont != NULL);
- CWindowDC dc = hWnd;
- TEXTMETRIC tmText = {};
- SIZE sizeText = {};
- HFONT hFontOld = dc.SelectFont(hFont);
- dc.GetTextMetrics(&tmText);
- m_sizeUnits.cy = tmText.tmHeight + tmText.tmExternalLeading;
- dc.GetTextExtent(_T("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"), 52, &sizeText);
- m_sizeUnits.cx = (sizeText.cx + 26) / 52;
- dc.SelectFont(hFontOld);
- return TRUE;
- }
- SIZE GetDialogBaseUnits() const
- {
- return m_sizeUnits;
- }
- INT MapDialogPixelsX(INT x) const
- {
- return ::MulDiv(x, 4, m_sizeUnits.cx); // Pixels X to DLU
- }
- INT MapDialogPixelsY(INT y) const
- {
- return ::MulDiv(y, 8, m_sizeUnits.cy); // Pixels Y to DLU
- }
- POINT MapDialogPixels(POINT pt) const
- {
- POINT out = { MapDialogPixelsX(pt.x), MapDialogPixelsY(pt.y) };
- return out;
- }
- SIZE MapDialogPixels(SIZE input) const
- {
- SIZE out = { MapDialogPixelsX(input.cx), MapDialogPixelsY(input.cy) };
- return out;
- }
- RECT MapDialogPixels(const RECT& input) const
- {
- RECT out = { MapDialogPixelsX(input.left), MapDialogPixelsY(input.top), MapDialogPixelsX(input.right), MapDialogPixelsY(input.bottom) };
- return out;
- }
- INT MapDialogUnitsX(INT x) const
- {
- return ::MulDiv(x, m_sizeUnits.cx, 4); // DLU to Pixels X
- }
- INT MapDialogUnitsY(INT y) const
- {
- return ::MulDiv(y, m_sizeUnits.cy, 8); // DLU to Pixels Y
- }
- POINT MapDialogUnits(POINT pt) const
- {
- POINT out = { MapDialogUnitsX(pt.x), MapDialogUnitsY(pt.y) };
- return out;
- }
- SIZE MapDialogUnits(SIZE input) const
- {
- SIZE out = { MapDialogUnitsX(input.cx), MapDialogUnitsY(input.cy) };
- return out;
- }
- RECT MapDialogUnits(const RECT& input) const
- {
- RECT out = { MapDialogUnitsX(input.left), MapDialogUnitsY(input.top), MapDialogUnitsX(input.right), MapDialogUnitsY(input.bottom) };
- return out;
- }
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CMemDlgTemplate - in-memory dialog template - DLGTEMPLATE or DLGTEMPLATEEX
- // traits suitable for dialog controls
- typedef ATL::CWinTraits<WS_CHILD | WS_VISIBLE, 0> CDlgControlWinTraits;
- template <class TWinTraits>
- class CMemDlgTemplateT
- {
- public:
- typedef ATL::_DialogSplitHelper::DLGTEMPLATEEX DLGTEMPLATEEX;
- typedef ATL::_DialogSplitHelper::DLGITEMTEMPLATEEX DLGITEMTEMPLATEEX;
- enum StdCtrlType
- {
- CTRL_BUTTON = 0x0080,
- CTRL_EDIT = 0x0081,
- CTRL_STATIC = 0x0082,
- CTRL_LISTBOX = 0x0083,
- CTRL_SCROLLBAR = 0x0084,
- CTRL_COMBOBOX = 0x0085
- };
- HANDLE m_hData;
- LPBYTE m_pData;
- LPBYTE m_pPtr;
- SIZE_T m_cAllocated;
- CMemDlgTemplateT() : m_hData(NULL), m_pData(NULL), m_pPtr(NULL), m_cAllocated(0)
- { }
- ~CMemDlgTemplateT()
- {
- Reset();
- }
- bool IsValid() const
- {
- return (m_pData != NULL);
- }
- bool IsTemplateEx() const
- {
- return (IsValid() && ((DLGTEMPLATEEX*)m_pData)->signature == 0xFFFF);
- }
- LPDLGTEMPLATE GetTemplatePtr()
- {
- return reinterpret_cast<LPDLGTEMPLATE>(m_pData);
- }
- DLGTEMPLATEEX* GetTemplateExPtr()
- {
- return reinterpret_cast<DLGTEMPLATEEX*>(m_pData);
- }
- void Reset()
- {
- if (IsValid())
- {
- ::GlobalUnlock(m_pData);
- ATLVERIFY(::GlobalFree(m_hData) == NULL);
- }
- m_hData = NULL;
- m_pData = NULL;
- m_pPtr = NULL;
- m_cAllocated = 0;
- }
- void Create(bool bDlgEx, LPCTSTR lpszCaption, const RECT& rc, DWORD dwStyle = 0, DWORD dwExStyle = 0,
- LPCTSTR lpstrFontName = NULL, WORD wFontSize = 0, WORD wWeight = 0, BYTE bItalic = 0, BYTE bCharset = 0, DWORD dwHelpID = 0,
- ATL::_U_STRINGorID ClassName = 0U, ATL::_U_STRINGorID Menu = 0U)
- {
- Create(bDlgEx, lpszCaption, (short) rc.left, (short) rc.top, (short) (rc.right - rc.left), (short) (rc.bottom - rc.top), dwStyle, dwExStyle,
- lpstrFontName, wFontSize, wWeight, bItalic, bCharset, dwHelpID, ClassName.m_lpstr, Menu.m_lpstr);
- }
- void Create(bool bDlgEx, LPCTSTR lpszCaption, short nX, short nY, short nWidth, short nHeight, DWORD dwStyle = 0, DWORD dwExStyle = 0,
- LPCTSTR lpstrFontName = NULL, WORD wFontSize = 0, WORD wWeight = 0, BYTE bItalic = 0, BYTE bCharset = 0, DWORD dwHelpID = 0,
- ATL::_U_STRINGorID ClassName = 0U, ATL::_U_STRINGorID Menu = 0U)
- {
- // Should have DS_SETFONT style to set the dialog font name and size
- if (lpstrFontName != NULL)
- {
- dwStyle |= DS_SETFONT;
- }
- else
- {
- dwStyle &= ~DS_SETFONT;
- }
- if (bDlgEx)
- {
- DLGTEMPLATEEX dlg = {1, 0xFFFF, dwHelpID, dwExStyle, dwStyle, 0, nX, nY, nWidth, nHeight};
- AddData(&dlg, sizeof(dlg));
- }
- else
- {
- DLGTEMPLATE dlg = {dwStyle, dwExStyle, 0, nX, nY, nWidth, nHeight};
- AddData(&dlg, sizeof(dlg));
- }
- if (Menu.m_lpstr == NULL)
- {
- WORD menuData = 0;
- AddData(&menuData, sizeof(WORD));
- }
- else if (IS_INTRESOURCE(Menu.m_lpstr))
- {
- WORD menuData[] = { 0xFFFF, LOWORD(Menu.m_lpstr) };
- AddData(menuData, sizeof(menuData));
- }
- else
- {
- AddString(Menu.m_lpstr);
- }
- if (ClassName.m_lpstr == NULL)
- {
- WORD classData = 0;
- AddData(&classData, sizeof(WORD));
- }
- else if (IS_INTRESOURCE(ClassName.m_lpstr))
- {
- WORD classData[] = { 0xFFFF, LOWORD(ClassName.m_lpstr) };
- AddData(classData, sizeof(classData));
- }
- else
- {
- AddString(ClassName.m_lpstr);
- }
- // Set dialog caption
- AddString(lpszCaption);
- if (lpstrFontName != NULL)
- {
- AddData(&wFontSize, sizeof(wFontSize));
- if (bDlgEx)
- {
- AddData(&wWeight, sizeof(wWeight));
- AddData(&bItalic, sizeof(bItalic));
- AddData(&bCharset, sizeof(bCharset));
- }
- AddString(lpstrFontName);
- }
- }
- void AddControl(ATL::_U_STRINGorID ClassName, WORD wId, const RECT& rc, DWORD dwStyle, DWORD dwExStyle,
- ATL::_U_STRINGorID Text, const WORD* pCreationData = NULL, WORD nCreationData = 0, DWORD dwHelpID = 0)
- {
- AddControl(ClassName.m_lpstr, wId, (short) rc.left, (short) rc.top, (short) (rc.right - rc.left), (short) (rc.bottom - rc.top), dwStyle, dwExStyle,
- Text.m_lpstr, pCreationData, nCreationData, dwHelpID);
- }
- void AddControl(ATL::_U_STRINGorID ClassName, WORD wId, short nX, short nY, short nWidth, short nHeight, DWORD dwStyle, DWORD dwExStyle,
- ATL::_U_STRINGorID Text, const WORD* pCreationData = NULL, WORD nCreationData = 0, DWORD dwHelpID = 0)
- {
- ATLASSERT(IsValid());
- // DWORD align data
- const DWORD_PTR dwDwordAlignBits = sizeof(DWORD) - 1;
- m_pPtr = (LPBYTE)(((DWORD_PTR)m_pPtr + dwDwordAlignBits) & (~dwDwordAlignBits));
- if (IsTemplateEx())
- {
- DLGTEMPLATEEX* dlg = (DLGTEMPLATEEX*)m_pData;
- dlg->cDlgItems++;
- DLGITEMTEMPLATEEX item = {dwHelpID, TWinTraits::GetWndExStyle(0) | dwExStyle, TWinTraits::GetWndStyle(0) | dwStyle, nX, nY, nWidth, nHeight, wId};
- AddData(&item, sizeof(item));
- }
- else
- {
- LPDLGTEMPLATE dlg = (LPDLGTEMPLATE)m_pData;
- dlg->cdit++;
- DLGITEMTEMPLATE item = {TWinTraits::GetWndStyle(0) | dwStyle, TWinTraits::GetWndExStyle(0) | dwExStyle, nX, nY, nWidth, nHeight, wId};
- AddData(&item, sizeof(item));
- }
- ATLASSERT(ClassName.m_lpstr != NULL);
- if (IS_INTRESOURCE(ClassName.m_lpstr))
- {
- WORD wData[] = { 0xFFFF, LOWORD(ClassName.m_lpstr) };
- AddData(wData, sizeof(wData));
- }
- else
- {
- AddString(ClassName.m_lpstr);
- }
- if (Text.m_lpstr == NULL)
- {
- WORD classData = 0;
- AddData(&classData, sizeof(WORD));
- }
- else if (IS_INTRESOURCE(Text.m_lpstr))
- {
- WORD wData[] = { 0xFFFF, LOWORD(Text.m_lpstr) };
- AddData(wData, sizeof(wData));
- }
- else
- {
- AddString(Text.m_lpstr);
- }
- AddData(&nCreationData, sizeof(nCreationData));
- if ((nCreationData != 0))
- {
- ATLASSERT(pCreationData != NULL);
- AddData(pCreationData, nCreationData * sizeof(WORD));
- }
- }
- void AddStdControl(StdCtrlType CtrlType, WORD wId, short nX, short nY, short nWidth, short nHeight,
- DWORD dwStyle, DWORD dwExStyle, ATL::_U_STRINGorID Text, const WORD* pCreationData = NULL, WORD nCreationData = 0, DWORD dwHelpID = 0)
- {
- AddControl(CtrlType, wId, nX, nY, nWidth, nHeight, dwStyle, dwExStyle, Text, pCreationData, nCreationData, dwHelpID);
- }
- void AddData(LPCVOID pData, size_t nData)
- {
- ATLASSERT(pData != NULL);
- const SIZE_T ALLOCATION_INCREMENT = 1024;
- if (m_pData == NULL)
- {
- m_cAllocated = ((nData / ALLOCATION_INCREMENT) + 1) * ALLOCATION_INCREMENT;
- m_hData = ::GlobalAlloc(GMEM_MOVEABLE | GMEM_ZEROINIT, m_cAllocated);
- ATLASSERT(m_hData != NULL);
- m_pPtr = m_pData = static_cast<LPBYTE>(::GlobalLock(m_hData));
- ATLASSERT(m_pData != NULL);
- }
- else if (((m_pPtr - m_pData) + nData) > m_cAllocated)
- {
- SIZE_T ptrPos = (m_pPtr - m_pData);
- m_cAllocated += ((nData / ALLOCATION_INCREMENT) + 1) * ALLOCATION_INCREMENT;
- ::GlobalUnlock(m_pData);
- m_hData = ::GlobalReAlloc(m_hData, m_cAllocated, GMEM_MOVEABLE | GMEM_ZEROINIT);
- ATLASSERT(m_hData != NULL);
- m_pData = static_cast<LPBYTE>(::GlobalLock(m_hData));
- ATLASSERT(m_pData != NULL);
- m_pPtr = m_pData + ptrPos;
- }
- ATL::Checked::memcpy_s(m_pPtr, m_cAllocated - (m_pPtr - m_pData), pData, nData);
- m_pPtr += nData;
- }
- void AddString(LPCTSTR lpszStr)
- {
- if (lpszStr == NULL)
- {
- WCHAR szEmpty = 0;
- AddData(&szEmpty, sizeof(szEmpty));
- }
- else
- {
- USES_CONVERSION;
- LPCWSTR lpstr = T2CW(lpszStr);
- int nSize = lstrlenW(lpstr) + 1;
- AddData(lpstr, nSize * sizeof(WCHAR));
- }
- }
- };
- typedef CMemDlgTemplateT<CDlgControlWinTraits> CMemDlgTemplate;
- ///////////////////////////////////////////////////////////////////////////////
- // Dialog and control macros for indirect dialogs
- // for DLGTEMPLATE
- #define BEGIN_DIALOG(x, y, width, height) \
- void DoInitTemplate() \
- { \
- bool bExTemplate = false; \
- short nX = x, nY = y, nWidth = width, nHeight = height; \
- LPCTSTR szCaption = NULL; \
- DWORD dwStyle = WS_POPUP | WS_BORDER | WS_SYSMENU; \
- DWORD dwExStyle = 0; \
- LPCTSTR szFontName = NULL; \
- WORD wFontSize = 0; \
- WORD wWeight = 0; \
- BYTE bItalic = 0; \
- BYTE bCharset = 0; \
- DWORD dwHelpID = 0; \
- ATL::_U_STRINGorID Menu = 0U; \
- ATL::_U_STRINGorID ClassName = 0U;
- // for DLGTEMPLATEEX
- #define BEGIN_DIALOG_EX(x, y, width, height, helpID) \
- void DoInitTemplate() \
- { \
- bool bExTemplate = true; \
- short nX = x, nY = y, nWidth = width, nHeight = height; \
- LPCTSTR szCaption = NULL; \
- DWORD dwStyle = WS_POPUP | WS_BORDER | WS_SYSMENU; \
- DWORD dwExStyle = 0; \
- LPCTSTR szFontName = NULL; \
- WORD wFontSize = 0; \
- WORD wWeight = 0; \
- BYTE bItalic = 0; \
- BYTE bCharset = 0; \
- DWORD dwHelpID = helpID; \
- ATL::_U_STRINGorID Menu = 0U; \
- ATL::_U_STRINGorID ClassName = 0U;
- #define END_DIALOG() \
- m_Template.Create(bExTemplate, szCaption, nX, nY, nWidth, nHeight, dwStyle, dwExStyle, szFontName, wFontSize, wWeight, bItalic, bCharset, dwHelpID, ClassName, Menu); \
- }
- #define DIALOG_CAPTION(caption) \
- szCaption = caption;
- #define DIALOG_STYLE(style) \
- dwStyle = style;
- #define DIALOG_EXSTYLE(exStyle) \
- dwExStyle = exStyle;
- #define DIALOG_FONT(pointSize, typeFace) \
- wFontSize = pointSize; \
- szFontName = typeFace;
- #define DIALOG_FONT_EX(pointsize, typeface, weight, italic, charset) \
- ATLASSERT(bExTemplate); \
- wFontSize = pointsize; \
- szFontName = typeface; \
- wWeight = weight; \
- bItalic = italic; \
- bCharset = charset;
- #define DIALOG_MENU(menuName) \
- Menu = menuName;
- #define DIALOG_CLASS(className) \
- ClassName = className;
- #define BEGIN_CONTROLS_MAP() \
- void DoInitControls() \
- {
- #define END_CONTROLS_MAP() \
- }
- #define CONTROL_LTEXT(text, id, x, y, width, height, style, exStyle) \
- m_Template.AddStdControl(m_Template.CTRL_STATIC, (WORD)id, x, y, width, height, style | SS_LEFT | WS_GROUP, exStyle, text, NULL, 0);
- #define CONTROL_CTEXT(text, id, x, y, width, height, style, exStyle) \
- m_Template.AddStdControl(m_Template.CTRL_STATIC, (WORD)id, x, y, width, height, style | SS_CENTER | WS_GROUP, exStyle, text, NULL, 0);
- #define CONTROL_RTEXT(text, id, x, y, width, height, style, exStyle) \
- m_Template.AddStdControl(m_Template.CTRL_STATIC, (WORD)id, x, y, width, height, style | SS_RIGHT | WS_GROUP, exStyle, text, NULL, 0);
- #define CONTROL_PUSHBUTTON(text, id, x, y, width, height, style, exStyle) \
- m_Template.AddStdControl(m_Template.CTRL_BUTTON, (WORD)id, x, y, width, height, style | BS_PUSHBUTTON | WS_TABSTOP, exStyle, text, NULL, 0);
- #define CONTROL_DEFPUSHBUTTON(text, id, x, y, width, height, style, exStyle) \
- m_Template.AddStdControl(m_Template.CTRL_BUTTON, (WORD)id, x, y, width, height, style | BS_DEFPUSHBUTTON | WS_TABSTOP, exStyle, text, NULL, 0);
- #define CONTROL_PUSHBOX(text, id, x, y, width, height, style, exStyle) \
- m_Template.AddStdControl(m_Template.CTRL_BUTTON, (WORD)id, x, y, width, height, style | BS_PUSHBOX | WS_TABSTOP, exStyle, text, NULL, 0);
- #define CONTROL_STATE3(text, id, x, y, width, height, style, exStyle) \
- m_Template.AddStdControl(m_Template.CTRL_BUTTON, (WORD)id, x, y, width, height, style | BS_3STATE | WS_TABSTOP, exStyle, text, NULL, 0);
- #define CONTROL_AUTO3STATE(text, id, x, y, width, height, style, exStyle) \
- m_Template.AddStdControl(m_Template.CTRL_BUTTON, (WORD)id, x, y, width, height, style | BS_AUTO3STATE | WS_TABSTOP, exStyle, text, NULL, 0);
- #define CONTROL_CHECKBOX(text, id, x, y, width, height, style, exStyle) \
- m_Template.AddStdControl(m_Template.CTRL_BUTTON, (WORD)id, x, y, width, height, style | BS_CHECKBOX | WS_TABSTOP, exStyle, text, NULL, 0);
- #define CONTROL_AUTOCHECKBOX(text, id, x, y, width, height, style, exStyle) \
- m_Template.AddStdControl(m_Template.CTRL_BUTTON, (WORD)id, x, y, width, height, style | BS_AUTOCHECKBOX | WS_TABSTOP, exStyle, text, NULL, 0);
- #define CONTROL_RADIOBUTTON(text, id, x, y, width, height, style, exStyle) \
- m_Template.AddStdControl(m_Template.CTRL_BUTTON, (WORD)id, x, y, width, height, style | BS_RADIOBUTTON | WS_TABSTOP, exStyle, text, NULL, 0);
- #define CONTROL_AUTORADIOBUTTON(text, id, x, y, width, height, style, exStyle) \
- m_Template.AddStdControl(m_Template.CTRL_BUTTON, (WORD)id, x, y, width, height, style | BS_AUTORADIOBUTTON | WS_TABSTOP, exStyle, text, NULL, 0);
- #define CONTROL_COMBOBOX(id, x, y, width, height, style, exStyle) \
- m_Template.AddStdControl(m_Template.CTRL_COMBOBOX, (WORD)id, x, y, width, height, style | CBS_DROPDOWN | WS_TABSTOP, exStyle, (LPCTSTR)NULL, NULL, 0);
- #define CONTROL_EDITTEXT(id, x, y, width, height, style, exStyle) \
- m_Template.AddStdControl(m_Template.CTRL_EDIT, (WORD)id, x, y, width, height, style | ES_LEFT | WS_BORDER | WS_TABSTOP, exStyle, (LPCTSTR)NULL, NULL, 0);
- #define CONTROL_GROUPBOX(text, id, x, y, width, height, style, exStyle) \
- m_Template.AddStdControl(m_Template.CTRL_BUTTON, (WORD)id, x, y, width, height, style | BS_GROUPBOX, exStyle, text, NULL, 0);
- #define CONTROL_LISTBOX(id, x, y, width, height, style, exStyle) \
- m_Template.AddStdControl(m_Template.CTRL_LISTBOX, (WORD)id, x, y, width, height, style | LBS_NOTIFY | WS_BORDER, exStyle, (LPCTSTR)NULL, NULL, 0);
- #define CONTROL_SCROLLBAR(id, x, y, width, height, style, exStyle) \
- m_Template.AddStdControl(m_Template.CTRL_SCROLLBAR, (WORD)id, x, y, width, height, style | SBS_HORZ, exStyle, (LPCTSTR)NULL, NULL, 0);
- #define CONTROL_ICON(text, id, x, y, width, height, style, exStyle) \
- m_Template.AddStdControl(m_Template.CTRL_STATIC, (WORD)id, x, y, width, height, style | SS_ICON, exStyle, text, NULL, 0);
- #define CONTROL_CONTROL(text, id, className, style, x, y, width, height, exStyle) \
- m_Template.AddControl(className, (WORD)id, x, y, width, height, style, exStyle, text, NULL, 0);
- ///////////////////////////////////////////////////////////////////////////////
- // CIndirectDialogImpl - dialogs with template in memory
- template <class T, class TDlgTemplate = CMemDlgTemplate, class TBase = ATL::CWindow>
- class ATL_NO_VTABLE CIndirectDialogImpl : public ATL::CDialogImpl< T, TBase >
- {
- public:
- enum { IDD = 0 }; // no dialog template resource
- TDlgTemplate m_Template;
- void CreateTemplate()
- {
- T* pT = static_cast<T*>(this);
- pT->DoInitTemplate();
- pT->DoInitControls();
- }
- INT_PTR DoModal(HWND hWndParent = ::GetActiveWindow(), LPARAM dwInitParam = NULL)
- {
- T* pT = static_cast<T*>(this);
- ATLASSERT(pT->m_hWnd == NULL);
- if(!m_Template.IsValid())
- CreateTemplate();
- // Allocate the thunk structure here, where we can fail gracefully.
- BOOL bRet = this->m_thunk.Init(NULL, NULL);
- if(bRet == FALSE)
- {
- ::SetLastError(ERROR_OUTOFMEMORY);
- return -1;
- }
- ModuleHelper::AddCreateWndData(&this->m_thunk.cd, (ATL::CDialogImplBaseT< TBase >*)pT);
- #ifdef _DEBUG
- this->m_bModal = true;
- #endif // _DEBUG
- return ::DialogBoxIndirectParam(ModuleHelper::GetResourceInstance(), m_Template.GetTemplatePtr(), hWndParent, (DLGPROC)T::StartDialogProc, dwInitParam);
- }
- HWND Create(HWND hWndParent, LPARAM dwInitParam = NULL)
- {
- T* pT = static_cast<T*>(this);
- ATLASSERT(pT->m_hWnd == NULL);
- if(!m_Template.IsValid())
- CreateTemplate();
- // Allocate the thunk structure here, where we can fail gracefully.
- BOOL bRet = this->m_thunk.Init(NULL, NULL);
- if(bRet == FALSE)
- {
- ::SetLastError(ERROR_OUTOFMEMORY);
- return NULL;
- }
- ModuleHelper::AddCreateWndData(&this->m_thunk.cd, (ATL::CDialogImplBaseT< TBase >*)pT);
- #ifdef _DEBUG
- this->m_bModal = false;
- #endif // _DEBUG
- HWND hWnd = ::CreateDialogIndirectParam(ModuleHelper::GetResourceInstance(), (LPCDLGTEMPLATE)m_Template.GetTemplatePtr(), hWndParent, (DLGPROC)T::StartDialogProc, dwInitParam);
- ATLASSERT(this->m_hWnd == hWnd);
- return hWnd;
- }
- // for CComControl
- HWND Create(HWND hWndParent, RECT&, LPARAM dwInitParam = NULL)
- {
- return Create(hWndParent, dwInitParam);
- }
- void DoInitTemplate()
- {
- ATLASSERT(FALSE); // MUST be defined in derived class
- }
- void DoInitControls()
- {
- ATLASSERT(FALSE); // MUST be defined in derived class
- }
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CPropertySheetWindow - client side for a property sheet
- class CPropertySheetWindow : public ATL::CWindow
- {
- public:
- // Constructors
- CPropertySheetWindow(HWND hWnd = NULL) : ATL::CWindow(hWnd)
- { }
- CPropertySheetWindow& operator =(HWND hWnd)
- {
- m_hWnd = hWnd;
- return *this;
- }
- // Attributes
- int GetPageCount() const
- {
- ATLASSERT(::IsWindow(m_hWnd));
- HWND hWndTabCtrl = GetTabControl();
- ATLASSERT(hWndTabCtrl != NULL);
- return (int)::SendMessage(hWndTabCtrl, TCM_GETITEMCOUNT, 0, 0L);
- }
- HWND GetActivePage() const
- {
- ATLASSERT(::IsWindow(m_hWnd));
- return (HWND)::SendMessage(m_hWnd, PSM_GETCURRENTPAGEHWND, 0, 0L);
- }
- int GetActiveIndex() const
- {
- ATLASSERT(::IsWindow(m_hWnd));
- HWND hWndTabCtrl = GetTabControl();
- ATLASSERT(hWndTabCtrl != NULL);
- return (int)::SendMessage(hWndTabCtrl, TCM_GETCURSEL, 0, 0L);
- }
- BOOL SetActivePage(int nPageIndex)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- return (BOOL)::SendMessage(m_hWnd, PSM_SETCURSEL, nPageIndex, 0L);
- }
- BOOL SetActivePage(HPROPSHEETPAGE hPage)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ATLASSERT(hPage != NULL);
- return (BOOL)::SendMessage(m_hWnd, PSM_SETCURSEL, 0, (LPARAM)hPage);
- }
- BOOL SetActivePageByID(int nPageID)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- return (BOOL)::SendMessage(m_hWnd, PSM_SETCURSELID, 0, nPageID);
- }
- void SetTitle(LPCTSTR lpszText, UINT nStyle = 0)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ATLASSERT((nStyle & ~PSH_PROPTITLE) == 0); // only PSH_PROPTITLE is valid
- ATLASSERT(lpszText != NULL);
- ::SendMessage(m_hWnd, PSM_SETTITLE, nStyle, (LPARAM)lpszText);
- }
- HWND GetTabControl() const
- {
- ATLASSERT(::IsWindow(m_hWnd));
- return (HWND)::SendMessage(m_hWnd, PSM_GETTABCONTROL, 0, 0L);
- }
- void SetFinishText(LPCTSTR lpszText)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ::SendMessage(m_hWnd, PSM_SETFINISHTEXT, 0, (LPARAM)lpszText);
- }
- void SetWizardButtons(DWORD dwFlags)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ::PostMessage(m_hWnd, PSM_SETWIZBUTTONS, 0, dwFlags);
- }
- // Operations
- BOOL AddPage(HPROPSHEETPAGE hPage)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ATLASSERT(hPage != NULL);
- return (BOOL)::SendMessage(m_hWnd, PSM_ADDPAGE, 0, (LPARAM)hPage);
- }
- BOOL AddPage(LPCPROPSHEETPAGE pPage)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ATLASSERT(pPage != NULL);
- HPROPSHEETPAGE hPage = ::CreatePropertySheetPage(pPage);
- if(hPage == NULL)
- return FALSE;
- return (BOOL)::SendMessage(m_hWnd, PSM_ADDPAGE, 0, (LPARAM)hPage);
- }
- BOOL InsertPage(int nNewPageIndex, HPROPSHEETPAGE hPage)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ATLASSERT(hPage != NULL);
- return (BOOL)::SendMessage(m_hWnd, PSM_INSERTPAGE, nNewPageIndex, (LPARAM)hPage);
- }
- BOOL InsertPage(int nNewPageIndex, LPCPROPSHEETPAGE pPage)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ATLASSERT(pPage != NULL);
- HPROPSHEETPAGE hPage = ::CreatePropertySheetPage(pPage);
- if(hPage == NULL)
- return FALSE;
- return (BOOL)::SendMessage(m_hWnd, PSM_INSERTPAGE, nNewPageIndex, (LPARAM)hPage);
- }
- BOOL InsertPage(HPROPSHEETPAGE hPageInsertAfter, HPROPSHEETPAGE hPage)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ATLASSERT(hPage != NULL);
- return (BOOL)::SendMessage(m_hWnd, PSM_INSERTPAGE, (WPARAM)hPageInsertAfter, (LPARAM)hPage);
- }
- BOOL InsertPage(HPROPSHEETPAGE hPageInsertAfter, LPCPROPSHEETPAGE pPage)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ATLASSERT(pPage != NULL);
- HPROPSHEETPAGE hPage = ::CreatePropertySheetPage(pPage);
- if(hPage == NULL)
- return FALSE;
- return (BOOL)::SendMessage(m_hWnd, PSM_INSERTPAGE, (WPARAM)hPageInsertAfter, (LPARAM)hPage);
- }
- void RemovePage(int nPageIndex)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ::SendMessage(m_hWnd, PSM_REMOVEPAGE, nPageIndex, 0L);
- }
- void RemovePage(HPROPSHEETPAGE hPage)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ATLASSERT(hPage != NULL);
- ::SendMessage(m_hWnd, PSM_REMOVEPAGE, 0, (LPARAM)hPage);
- }
- BOOL PressButton(int nButton)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- return (BOOL)::SendMessage(m_hWnd, PSM_PRESSBUTTON, nButton, 0L);
- }
- BOOL Apply()
- {
- ATLASSERT(::IsWindow(m_hWnd));
- return (BOOL)::SendMessage(m_hWnd, PSM_APPLY, 0, 0L);
- }
- void CancelToClose()
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ::SendMessage(m_hWnd, PSM_CANCELTOCLOSE, 0, 0L);
- }
- void SetModified(HWND hWndPage, BOOL bChanged = TRUE)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ATLASSERT(::IsWindow(hWndPage));
- UINT uMsg = bChanged ? PSM_CHANGED : PSM_UNCHANGED;
- ::SendMessage(m_hWnd, uMsg, (WPARAM)hWndPage, 0L);
- }
- LRESULT QuerySiblings(WPARAM wParam, LPARAM lParam)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- return ::SendMessage(m_hWnd, PSM_QUERYSIBLINGS, wParam, lParam);
- }
- void RebootSystem()
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ::SendMessage(m_hWnd, PSM_REBOOTSYSTEM, 0, 0L);
- }
- void RestartWindows()
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ::SendMessage(m_hWnd, PSM_RESTARTWINDOWS, 0, 0L);
- }
- BOOL IsDialogMessage(LPMSG lpMsg)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- return (BOOL)::SendMessage(m_hWnd, PSM_ISDIALOGMESSAGE, 0, (LPARAM)lpMsg);
- }
- int HwndToIndex(HWND hWnd) const
- {
- ATLASSERT(::IsWindow(m_hWnd));
- return (int)::SendMessage(m_hWnd, PSM_HWNDTOINDEX, (WPARAM)hWnd, 0L);
- }
- HWND IndexToHwnd(int nIndex) const
- {
- ATLASSERT(::IsWindow(m_hWnd));
- return (HWND)::SendMessage(m_hWnd, PSM_INDEXTOHWND, nIndex, 0L);
- }
- int PageToIndex(HPROPSHEETPAGE hPage) const
- {
- ATLASSERT(::IsWindow(m_hWnd));
- return (int)::SendMessage(m_hWnd, PSM_PAGETOINDEX, 0, (LPARAM)hPage);
- }
- HPROPSHEETPAGE IndexToPage(int nIndex) const
- {
- ATLASSERT(::IsWindow(m_hWnd));
- return (HPROPSHEETPAGE)::SendMessage(m_hWnd, PSM_INDEXTOPAGE, nIndex, 0L);
- }
- int IdToIndex(int nID) const
- {
- ATLASSERT(::IsWindow(m_hWnd));
- return (int)::SendMessage(m_hWnd, PSM_IDTOINDEX, 0, nID);
- }
- int IndexToId(int nIndex) const
- {
- ATLASSERT(::IsWindow(m_hWnd));
- return (int)::SendMessage(m_hWnd, PSM_INDEXTOID, nIndex, 0L);
- }
- int GetResult() const
- {
- ATLASSERT(::IsWindow(m_hWnd));
- return (int)::SendMessage(m_hWnd, PSM_GETRESULT, 0, 0L);
- }
- BOOL RecalcPageSizes()
- {
- ATLASSERT(::IsWindow(m_hWnd));
- return (BOOL)::SendMessage(m_hWnd, PSM_RECALCPAGESIZES, 0, 0L);
- }
- void SetHeaderTitle(int nIndex, LPCTSTR lpstrHeaderTitle)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ::SendMessage(m_hWnd, PSM_SETHEADERTITLE, nIndex, (LPARAM)lpstrHeaderTitle);
- }
- void SetHeaderSubTitle(int nIndex, LPCTSTR lpstrHeaderSubTitle)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ::SendMessage(m_hWnd, PSM_SETHEADERSUBTITLE, nIndex, (LPARAM)lpstrHeaderSubTitle);
- }
- // Implementation - override to prevent usage
- HWND Create(LPCTSTR, HWND, ATL::_U_RECT = NULL, LPCTSTR = NULL, DWORD = 0, DWORD = 0, ATL::_U_MENUorID = 0U, LPVOID = NULL)
- {
- ATLASSERT(FALSE);
- return NULL;
- }
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CPropertySheetImpl - implements a property sheet
- template <class T, class TBase = CPropertySheetWindow>
- class ATL_NO_VTABLE CPropertySheetImpl : public ATL::CWindowImplBaseT< TBase >
- {
- public:
- PROPSHEETHEADER m_psh;
- ATL::CSimpleArray<HPROPSHEETPAGE> m_arrPages;
- // Construction/Destruction
- CPropertySheetImpl(ATL::_U_STRINGorID title = (LPCTSTR)NULL, UINT uStartPage = 0, HWND hWndParent = NULL)
- {
- memset(&m_psh, 0, sizeof(PROPSHEETHEADER));
- m_psh.dwSize = sizeof(PROPSHEETHEADER);
- m_psh.dwFlags = PSH_USECALLBACK;
- m_psh.hInstance = ModuleHelper::GetResourceInstance();
- m_psh.phpage = NULL; // will be set later
- m_psh.nPages = 0; // will be set later
- m_psh.pszCaption = title.m_lpstr;
- m_psh.nStartPage = uStartPage;
- m_psh.hwndParent = hWndParent; // if NULL, will be set in DoModal/Create
- m_psh.pfnCallback = T::PropSheetCallback;
- }
- ~CPropertySheetImpl()
- {
- if(m_arrPages.GetSize() > 0) // sheet never created, destroy all pages
- {
- for(int i = 0; i < m_arrPages.GetSize(); i++)
- ::DestroyPropertySheetPage((HPROPSHEETPAGE)m_arrPages[i]);
- }
- }
- // Callback function and overrideables
- static int CALLBACK PropSheetCallback(HWND hWnd, UINT uMsg, LPARAM lParam)
- {
- (void)lParam; // avoid level 4 warning
- int nRet = 0;
- if(uMsg == PSCB_INITIALIZED)
- {
- ATLASSERT(hWnd != NULL);
- T* pT = (T*)ModuleHelper::ExtractCreateWndData();
- // subclass the sheet window
- pT->SubclassWindow(hWnd);
- // remove page handles array
- pT->_CleanUpPages();
- pT->OnSheetInitialized();
- }
- return nRet;
- }
- void OnSheetInitialized()
- {
- }
- // Create method
- HWND Create(HWND hWndParent = NULL)
- {
- ATLASSERT(this->m_hWnd == NULL);
- m_psh.dwFlags |= PSH_MODELESS;
- if(m_psh.hwndParent == NULL)
- m_psh.hwndParent = hWndParent;
- m_psh.phpage = (HPROPSHEETPAGE*)m_arrPages.GetData();
- m_psh.nPages = m_arrPages.GetSize();
- T* pT = static_cast<T*>(this);
- // Allocate the thunk structure here, where we can fail gracefully.
- BOOL bRet = pT->m_thunk.Init(NULL, NULL);
- if(bRet == FALSE)
- {
- ::SetLastError(ERROR_OUTOFMEMORY);
- return NULL;
- }
- ModuleHelper::AddCreateWndData(&pT->m_thunk.cd, pT);
- HWND hWnd = (HWND)::PropertySheet(&m_psh);
- _CleanUpPages(); // ensure clean-up, required if call failed
- ATLASSERT(this->m_hWnd == hWnd);
- return hWnd;
- }
- INT_PTR DoModal(HWND hWndParent = ::GetActiveWindow())
- {
- ATLASSERT(this->m_hWnd == NULL);
- m_psh.dwFlags &= ~PSH_MODELESS;
- if(m_psh.hwndParent == NULL)
- m_psh.hwndParent = hWndParent;
- m_psh.phpage = (HPROPSHEETPAGE*)m_arrPages.GetData();
- m_psh.nPages = m_arrPages.GetSize();
- T* pT = static_cast<T*>(this);
- // Allocate the thunk structure here, where we can fail gracefully.
- BOOL bRet = pT->m_thunk.Init(NULL, NULL);
- if(bRet == FALSE)
- {
- ::SetLastError(ERROR_OUTOFMEMORY);
- return -1;
- }
- ModuleHelper::AddCreateWndData(&pT->m_thunk.cd, pT);
- INT_PTR nRet = ::PropertySheet(&m_psh);
- _CleanUpPages(); // ensure clean-up, required if call failed
- return nRet;
- }
- // implementation helper - clean up pages array
- void _CleanUpPages()
- {
- m_psh.nPages = 0;
- m_psh.phpage = NULL;
- m_arrPages.RemoveAll();
- }
- // Attributes (extended overrides of client class methods)
- // These now can be called before the sheet is created
- // Note: Calling these after the sheet is created gives unpredictable results
- int GetPageCount() const
- {
- if(this->m_hWnd == NULL) // not created yet
- return m_arrPages.GetSize();
- return TBase::GetPageCount();
- }
- int GetActiveIndex() const
- {
- if(this->m_hWnd == NULL) // not created yet
- return m_psh.nStartPage;
- return TBase::GetActiveIndex();
- }
- HPROPSHEETPAGE GetPage(int nPageIndex) const
- {
- ATLASSERT(this->m_hWnd == NULL); // can't do this after it's created
- return (HPROPSHEETPAGE)m_arrPages[nPageIndex];
- }
- int GetPageIndex(HPROPSHEETPAGE hPage) const
- {
- ATLASSERT(this->m_hWnd == NULL); // can't do this after it's created
- return m_arrPages.Find((HPROPSHEETPAGE&)hPage);
- }
- BOOL SetActivePage(int nPageIndex)
- {
- if(this->m_hWnd == NULL) // not created yet
- {
- ATLASSERT((nPageIndex >= 0) && (nPageIndex < m_arrPages.GetSize()));
- m_psh.nStartPage = nPageIndex;
- return TRUE;
- }
- return TBase::SetActivePage(nPageIndex);
- }
- BOOL SetActivePage(HPROPSHEETPAGE hPage)
- {
- ATLASSERT(hPage != NULL);
- if(this->m_hWnd == NULL) // not created yet
- {
- int nPageIndex = GetPageIndex(hPage);
- if(nPageIndex == -1)
- return FALSE;
- return SetActivePage(nPageIndex);
- }
- return TBase::SetActivePage(hPage);
- }
- void SetTitle(LPCTSTR lpszText, UINT nStyle = 0)
- {
- ATLASSERT((nStyle & ~PSH_PROPTITLE) == 0); // only PSH_PROPTITLE is valid
- ATLASSERT(lpszText != NULL);
- if(this->m_hWnd == NULL)
- {
- // set internal state
- m_psh.pszCaption = lpszText; // must exist until sheet is created
- m_psh.dwFlags &= ~PSH_PROPTITLE;
- m_psh.dwFlags |= nStyle;
- }
- else
- {
- // set external state
- TBase::SetTitle(lpszText, nStyle);
- }
- }
- void SetWizardMode()
- {
- m_psh.dwFlags |= PSH_WIZARD;
- }
- void EnableHelp()
- {
- m_psh.dwFlags |= PSH_HASHELP;
- }
- // Operations
- BOOL AddPage(HPROPSHEETPAGE hPage)
- {
- ATLASSERT(hPage != NULL);
- BOOL bRet = FALSE;
- if(this->m_hWnd != NULL)
- bRet = TBase::AddPage(hPage);
- else // sheet not created yet, use internal data
- bRet = m_arrPages.Add((HPROPSHEETPAGE&)hPage);
- return bRet;
- }
- BOOL AddPage(LPCPROPSHEETPAGE pPage)
- {
- ATLASSERT(pPage != NULL);
- HPROPSHEETPAGE hPage = ::CreatePropertySheetPage(pPage);
- if(hPage == NULL)
- return FALSE;
- BOOL bRet = AddPage(hPage);
- if(!bRet)
- ::DestroyPropertySheetPage(hPage);
- return bRet;
- }
- BOOL RemovePage(HPROPSHEETPAGE hPage)
- {
- ATLASSERT(hPage != NULL);
- if(this->m_hWnd == NULL) // not created yet
- {
- int nPage = GetPageIndex(hPage);
- if(nPage == -1)
- return FALSE;
- return RemovePage(nPage);
- }
- TBase::RemovePage(hPage);
- return TRUE;
- }
- BOOL RemovePage(int nPageIndex)
- {
- BOOL bRet = TRUE;
- if(this->m_hWnd != NULL)
- TBase::RemovePage(nPageIndex);
- else // sheet not created yet, use internal data
- bRet = m_arrPages.RemoveAt(nPageIndex);
- return bRet;
- }
- void SetHeader(LPCTSTR szbmHeader)
- {
- ATLASSERT(this->m_hWnd == NULL); // can't do this after it's created
- m_psh.dwFlags &= ~PSH_WIZARD;
- m_psh.dwFlags |= (PSH_HEADER | PSH_WIZARD97);
- m_psh.pszbmHeader = szbmHeader;
- }
- void SetHeader(HBITMAP hbmHeader)
- {
- ATLASSERT(this->m_hWnd == NULL); // can't do this after it's created
- m_psh.dwFlags &= ~PSH_WIZARD;
- m_psh.dwFlags |= (PSH_HEADER | PSH_USEHBMHEADER | PSH_WIZARD97);
- m_psh.hbmHeader = hbmHeader;
- }
- void SetWatermark(LPCTSTR szbmWatermark, HPALETTE hplWatermark = NULL)
- {
- ATLASSERT(this->m_hWnd == NULL); // can't do this after it's created
- m_psh.dwFlags &= ~PSH_WIZARD;
- m_psh.dwFlags |= PSH_WATERMARK | PSH_WIZARD97;
- m_psh.pszbmWatermark = szbmWatermark;
- if(hplWatermark != NULL)
- {
- m_psh.dwFlags |= PSH_USEHPLWATERMARK;
- m_psh.hplWatermark = hplWatermark;
- }
- }
- void SetWatermark(HBITMAP hbmWatermark, HPALETTE hplWatermark = NULL)
- {
- ATLASSERT(this->m_hWnd == NULL); // can't do this after it's created
- m_psh.dwFlags &= ~PSH_WIZARD;
- m_psh.dwFlags |= (PSH_WATERMARK | PSH_USEHBMWATERMARK | PSH_WIZARD97);
- m_psh.hbmWatermark = hbmWatermark;
- if(hplWatermark != NULL)
- {
- m_psh.dwFlags |= PSH_USEHPLWATERMARK;
- m_psh.hplWatermark = hplWatermark;
- }
- }
- void StretchWatermark(bool bStretchWatermark)
- {
- ATLASSERT(this->m_hWnd == NULL); // can't do this after it's created
- if(bStretchWatermark)
- m_psh.dwFlags |= PSH_STRETCHWATERMARK;
- else
- m_psh.dwFlags &= ~PSH_STRETCHWATERMARK;
- }
- // Message map and handlers
- BEGIN_MSG_MAP(CPropertySheetImpl)
- MESSAGE_HANDLER(WM_COMMAND, OnCommand)
- MESSAGE_HANDLER(WM_SYSCOMMAND, OnSysCommand)
- END_MSG_MAP()
- LRESULT OnCommand(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& /*bHandled*/)
- {
- LRESULT lRet = this->DefWindowProc(uMsg, wParam, lParam);
- if((HIWORD(wParam) == BN_CLICKED) && ((LOWORD(wParam) == IDOK) || (LOWORD(wParam) == IDCANCEL)) &&
- ((m_psh.dwFlags & PSH_MODELESS) != 0) && (this->GetActivePage() == NULL))
- this->DestroyWindow();
- return lRet;
- }
- LRESULT OnSysCommand(UINT /*uMsg*/, WPARAM wParam, LPARAM /*lParam*/, BOOL& bHandled)
- {
- if(((m_psh.dwFlags & PSH_MODELESS) == PSH_MODELESS) && ((wParam & 0xFFF0) == SC_CLOSE))
- this->SendMessage(WM_CLOSE);
- else
- bHandled = FALSE;
- return 0;
- }
- };
- // for non-customized sheets
- class CPropertySheet : public CPropertySheetImpl<CPropertySheet>
- {
- public:
- CPropertySheet(ATL::_U_STRINGorID title = (LPCTSTR)NULL, UINT uStartPage = 0, HWND hWndParent = NULL)
- : CPropertySheetImpl<CPropertySheet>(title, uStartPage, hWndParent)
- { }
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CPropertyPageWindow - client side for a property page
- class CPropertyPageWindow : public ATL::CWindow
- {
- public:
- // Constructors
- CPropertyPageWindow(HWND hWnd = NULL) : ATL::CWindow(hWnd)
- { }
- CPropertyPageWindow& operator =(HWND hWnd)
- {
- m_hWnd = hWnd;
- return *this;
- }
- // Attributes
- CPropertySheetWindow GetPropertySheet() const
- {
- ATLASSERT(::IsWindow(m_hWnd));
- return CPropertySheetWindow(GetParent());
- }
- // Operations
- BOOL Apply()
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ATLASSERT(GetParent() != NULL);
- return GetPropertySheet().Apply();
- }
- void CancelToClose()
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ATLASSERT(GetParent() != NULL);
- GetPropertySheet().CancelToClose();
- }
- void SetModified(BOOL bChanged = TRUE)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ATLASSERT(GetParent() != NULL);
- GetPropertySheet().SetModified(m_hWnd, bChanged);
- }
- LRESULT QuerySiblings(WPARAM wParam, LPARAM lParam)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ATLASSERT(GetParent() != NULL);
- return GetPropertySheet().QuerySiblings(wParam, lParam);
- }
- void RebootSystem()
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ATLASSERT(GetParent() != NULL);
- GetPropertySheet().RebootSystem();
- }
- void RestartWindows()
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ATLASSERT(GetParent() != NULL);
- GetPropertySheet().RestartWindows();
- }
- void SetWizardButtons(DWORD dwFlags)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ATLASSERT(GetParent() != NULL);
- GetPropertySheet().SetWizardButtons(dwFlags);
- }
- // Implementation - overrides to prevent usage
- HWND Create(LPCTSTR, HWND, ATL::_U_RECT = NULL, LPCTSTR = NULL, DWORD = 0, DWORD = 0, ATL::_U_MENUorID = 0U, LPVOID = NULL)
- {
- ATLASSERT(FALSE);
- return NULL;
- }
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CPropertyPageImpl - implements a property page
- #if defined(_WTL_FORCE_OLD_PAGE_NOTIFY_HANDLERS) && defined(_WTL_NEW_PAGE_NOTIFY_HANDLERS)
- #error _WTL_FORCE_OLD_PAGE_NOTIFY_HANDLERS and _WTL_NEW_PAGE_NOTIFY_HANDLERS cannot be both defined
- #endif
- #if !defined(_WTL_FORCE_OLD_PAGE_NOTIFY_HANDLERS) && !defined(_WTL_NEW_PAGE_NOTIFY_HANDLERS)
- #define _WTL_NEW_PAGE_NOTIFY_HANDLERS
- #endif
- // NOTE: _WTL_NEW_PAGE_NOTIFY_HANDLERS is now defined by default.
- // It enables use of new notification handlers that
- // return direct values without any restrictions.
- // Define _WTL_FORCE_OLD_PAGE_NOTIFY_HANDLERS to use old handlers.
- template <class T, class TBase = CPropertyPageWindow>
- class ATL_NO_VTABLE CPropertyPageImpl : public ATL::CDialogImplBaseT< TBase >
- {
- public:
- PROPSHEETPAGE m_psp;
- operator PROPSHEETPAGE*() { return &m_psp; }
- // Construction
- CPropertyPageImpl(ATL::_U_STRINGorID title = (LPCTSTR)NULL)
- {
- // initialize PROPSHEETPAGE struct
- memset(&m_psp, 0, sizeof(PROPSHEETPAGE));
- m_psp.dwSize = sizeof(PROPSHEETPAGE);
- m_psp.dwFlags = PSP_USECALLBACK;
- m_psp.hInstance = ModuleHelper::GetResourceInstance();
- T* pT = static_cast<T*>(this);
- m_psp.pszTemplate = MAKEINTRESOURCE(pT->IDD);
- m_psp.pfnDlgProc = (DLGPROC)T::StartDialogProc;
- m_psp.pfnCallback = T::PropPageCallback;
- m_psp.lParam = (LPARAM)pT;
- if(title.m_lpstr != NULL)
- SetTitle(title);
- }
- // Callback function and overrideables
- static UINT CALLBACK PropPageCallback(HWND hWnd, UINT uMsg, LPPROPSHEETPAGE ppsp)
- {
- (void)hWnd; // avoid level 4 warning
- ATLASSERT(hWnd == NULL);
- T* pT = (T*)ppsp->lParam;
- UINT uRet = 0;
- switch(uMsg)
- {
- case PSPCB_CREATE:
- {
- ATL::CDialogImplBaseT< TBase >* pPage = (ATL::CDialogImplBaseT< TBase >*)pT;
- ModuleHelper::AddCreateWndData(&pPage->m_thunk.cd, pPage);
- uRet = pT->OnPageCreate() ? 1 : 0;
- }
- break;
- case PSPCB_ADDREF:
- pT->OnPageAddRef();
- break;
- case PSPCB_RELEASE:
- pT->OnPageRelease();
- break;
- default:
- break;
- }
- return uRet;
- }
- bool OnPageCreate()
- {
- return true; // true - allow page to be created, false - prevent creation
- }
- void OnPageAddRef()
- {
- }
- void OnPageRelease()
- {
- }
- // Create method
- HPROPSHEETPAGE Create()
- {
- return ::CreatePropertySheetPage(&m_psp);
- }
- // Attributes
- void SetTitle(ATL::_U_STRINGorID title)
- {
- m_psp.pszTitle = title.m_lpstr;
- m_psp.dwFlags |= PSP_USETITLE;
- }
- void SetHeaderTitle(LPCTSTR lpstrHeaderTitle)
- {
- ATLASSERT(this->m_hWnd == NULL); // can't do this after it's created
- m_psp.dwFlags |= PSP_USEHEADERTITLE;
- m_psp.pszHeaderTitle = lpstrHeaderTitle;
- }
- void SetHeaderSubTitle(LPCTSTR lpstrHeaderSubTitle)
- {
- ATLASSERT(this->m_hWnd == NULL); // can't do this after it's created
- m_psp.dwFlags |= PSP_USEHEADERSUBTITLE;
- m_psp.pszHeaderSubTitle = lpstrHeaderSubTitle;
- }
- // Operations
- void EnableHelp()
- {
- m_psp.dwFlags |= PSP_HASHELP;
- }
- // Message map and handlers
- BEGIN_MSG_MAP(CPropertyPageImpl)
- MESSAGE_HANDLER(WM_NOTIFY, OnNotify)
- END_MSG_MAP()
- LRESULT OnNotify(UINT /*uMsg*/, WPARAM /*wParam*/, LPARAM lParam, BOOL& bHandled)
- {
- ATLASSERT(::IsWindow(this->m_hWnd));
- NMHDR* pNMHDR = (NMHDR*)lParam;
- // don't handle messages not from the page/sheet itself
- if((pNMHDR->hwndFrom != this->m_hWnd) && (pNMHDR->hwndFrom != ::GetParent(this->m_hWnd)))
- {
- bHandled = FALSE;
- return 1;
- }
- T* pT = static_cast<T*>(this);
- LRESULT lResult = 0;
- switch(pNMHDR->code)
- {
- #ifdef _WTL_NEW_PAGE_NOTIFY_HANDLERS
- case PSN_SETACTIVE:
- lResult = pT->OnSetActive();
- break;
- case PSN_KILLACTIVE:
- lResult = pT->OnKillActive();
- break;
- case PSN_APPLY:
- lResult = pT->OnApply();
- break;
- case PSN_RESET:
- pT->OnReset();
- break;
- case PSN_QUERYCANCEL:
- lResult = pT->OnQueryCancel();
- break;
- case PSN_WIZNEXT:
- lResult = pT->OnWizardNext();
- break;
- case PSN_WIZBACK:
- lResult = pT->OnWizardBack();
- break;
- case PSN_WIZFINISH:
- lResult = pT->OnWizardFinish();
- break;
- case PSN_HELP:
- pT->OnHelp();
- break;
- case PSN_GETOBJECT:
- if(!pT->OnGetObject((LPNMOBJECTNOTIFY)lParam))
- bHandled = FALSE;
- break;
- case PSN_TRANSLATEACCELERATOR:
- {
- LPPSHNOTIFY lpPSHNotify = (LPPSHNOTIFY)lParam;
- lResult = pT->OnTranslateAccelerator((LPMSG)lpPSHNotify->lParam);
- }
- break;
- case PSN_QUERYINITIALFOCUS:
- {
- LPPSHNOTIFY lpPSHNotify = (LPPSHNOTIFY)lParam;
- lResult = (LRESULT)pT->OnQueryInitialFocus((HWND)lpPSHNotify->lParam);
- }
- break;
- #else // !_WTL_NEW_PAGE_NOTIFY_HANDLERS
- case PSN_SETACTIVE:
- lResult = pT->OnSetActive() ? 0 : -1;
- break;
- case PSN_KILLACTIVE:
- lResult = !pT->OnKillActive();
- break;
- case PSN_APPLY:
- lResult = pT->OnApply() ? PSNRET_NOERROR : PSNRET_INVALID_NOCHANGEPAGE;
- break;
- case PSN_RESET:
- pT->OnReset();
- break;
- case PSN_QUERYCANCEL:
- lResult = !pT->OnQueryCancel();
- break;
- case PSN_WIZNEXT:
- lResult = pT->OnWizardNext();
- break;
- case PSN_WIZBACK:
- lResult = pT->OnWizardBack();
- break;
- case PSN_WIZFINISH:
- lResult = !pT->OnWizardFinish();
- break;
- case PSN_HELP:
- pT->OnHelp();
- break;
- case PSN_GETOBJECT:
- if(!pT->OnGetObject((LPNMOBJECTNOTIFY)lParam))
- bHandled = FALSE;
- break;
- case PSN_TRANSLATEACCELERATOR:
- {
- LPPSHNOTIFY lpPSHNotify = (LPPSHNOTIFY)lParam;
- lResult = pT->OnTranslateAccelerator((LPMSG)lpPSHNotify->lParam) ? PSNRET_MESSAGEHANDLED : PSNRET_NOERROR;
- }
- break;
- case PSN_QUERYINITIALFOCUS:
- {
- LPPSHNOTIFY lpPSHNotify = (LPPSHNOTIFY)lParam;
- lResult = (LRESULT)pT->OnQueryInitialFocus((HWND)lpPSHNotify->lParam);
- }
- break;
- #endif // !_WTL_NEW_PAGE_NOTIFY_HANDLERS
- default:
- bHandled = FALSE; // not handled
- }
- return lResult;
- }
- // Overridables
- #ifdef _WTL_NEW_PAGE_NOTIFY_HANDLERS
- int OnSetActive()
- {
- // 0 = allow activate
- // -1 = go back that was active
- // page ID = jump to page
- return 0;
- }
- BOOL OnKillActive()
- {
- // FALSE = allow deactivate
- // TRUE = prevent deactivation
- return FALSE;
- }
- int OnApply()
- {
- // PSNRET_NOERROR = apply OK
- // PSNRET_INVALID = apply not OK, return to this page
- // PSNRET_INVALID_NOCHANGEPAGE = apply not OK, don't change focus
- return PSNRET_NOERROR;
- }
- void OnReset()
- {
- }
- BOOL OnQueryCancel()
- {
- // FALSE = allow cancel
- // TRUE = prevent cancel
- return FALSE;
- }
- int OnWizardBack()
- {
- // 0 = goto previous page
- // -1 = prevent page change
- // >0 = jump to page by dlg ID
- return 0;
- }
- int OnWizardNext()
- {
- // 0 = goto next page
- // -1 = prevent page change
- // >0 = jump to page by dlg ID
- return 0;
- }
- INT_PTR OnWizardFinish()
- {
- // FALSE = allow finish
- // TRUE = prevent finish
- // HWND = prevent finish and set focus to HWND (CommCtrl 5.80 only)
- return FALSE;
- }
- void OnHelp()
- {
- }
- BOOL OnGetObject(LPNMOBJECTNOTIFY /*lpObjectNotify*/)
- {
- return FALSE; // not processed
- }
- int OnTranslateAccelerator(LPMSG /*lpMsg*/)
- {
- // PSNRET_NOERROR - message not handled
- // PSNRET_MESSAGEHANDLED - message handled
- return PSNRET_NOERROR;
- }
- HWND OnQueryInitialFocus(HWND /*hWndFocus*/)
- {
- // NULL = set focus to default control
- // HWND = set focus to HWND
- return NULL;
- }
- #else // !_WTL_NEW_PAGE_NOTIFY_HANDLERS
- BOOL OnSetActive()
- {
- return TRUE;
- }
- BOOL OnKillActive()
- {
- return TRUE;
- }
- BOOL OnApply()
- {
- return TRUE;
- }
- void OnReset()
- {
- }
- BOOL OnQueryCancel()
- {
- return TRUE; // ok to cancel
- }
- int OnWizardBack()
- {
- // 0 = goto previous page
- // -1 = prevent page change
- // >0 = jump to page by dlg ID
- return 0;
- }
- int OnWizardNext()
- {
- // 0 = goto next page
- // -1 = prevent page change
- // >0 = jump to page by dlg ID
- return 0;
- }
- BOOL OnWizardFinish()
- {
- return TRUE;
- }
- void OnHelp()
- {
- }
- BOOL OnGetObject(LPNMOBJECTNOTIFY /*lpObjectNotify*/)
- {
- return FALSE; // not processed
- }
- BOOL OnTranslateAccelerator(LPMSG /*lpMsg*/)
- {
- return FALSE; // not translated
- }
- HWND OnQueryInitialFocus(HWND /*hWndFocus*/)
- {
- return NULL; // default
- }
- #endif // !_WTL_NEW_PAGE_NOTIFY_HANDLERS
- };
- // for non-customized pages
- template <WORD t_wDlgTemplateID>
- class CPropertyPage : public CPropertyPageImpl<CPropertyPage<t_wDlgTemplateID> >
- {
- public:
- enum { IDD = t_wDlgTemplateID };
- CPropertyPage(ATL::_U_STRINGorID title = (LPCTSTR)NULL) : CPropertyPageImpl<CPropertyPage>(title)
- { }
- DECLARE_EMPTY_MSG_MAP()
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CAxPropertyPageImpl - property page that hosts ActiveX controls
- #ifndef _ATL_NO_HOSTING
- // Note: You must #include <atlhost.h> to use these classes
- template <class T, class TBase = CPropertyPageWindow>
- class ATL_NO_VTABLE CAxPropertyPageImpl : public CPropertyPageImpl< T, TBase >
- {
- public:
- // Data members
- HGLOBAL m_hInitData;
- HGLOBAL m_hDlgRes;
- HGLOBAL m_hDlgResSplit;
- // Constructor/destructor
- CAxPropertyPageImpl(ATL::_U_STRINGorID title = (LPCTSTR)NULL) :
- CPropertyPageImpl< T, TBase >(title),
- m_hInitData(NULL), m_hDlgRes(NULL), m_hDlgResSplit(NULL)
- {
- T* pT = static_cast<T*>(this);
- (void)pT; // avoid level 4 warning
- // initialize ActiveX hosting and modify dialog template
- ATL::AtlAxWinInit();
- HINSTANCE hInstance = ModuleHelper::GetResourceInstance();
- LPCTSTR lpTemplateName = MAKEINTRESOURCE(pT->IDD);
- HRSRC hDlg = ::FindResource(hInstance, lpTemplateName, (LPTSTR)RT_DIALOG);
- if(hDlg != NULL)
- {
- HRSRC hDlgInit = ::FindResource(hInstance, lpTemplateName, (LPTSTR)_ATL_RT_DLGINIT);
- BYTE* pInitData = NULL;
- if(hDlgInit != NULL)
- {
- m_hInitData = ::LoadResource(hInstance, hDlgInit);
- pInitData = (BYTE*)::LockResource(m_hInitData);
- }
- m_hDlgRes = ::LoadResource(hInstance, hDlg);
- DLGTEMPLATE* pDlg = (DLGTEMPLATE*)::LockResource(m_hDlgRes);
- LPCDLGTEMPLATE lpDialogTemplate = ATL::_DialogSplitHelper::SplitDialogTemplate(pDlg, pInitData);
- if(lpDialogTemplate != pDlg)
- m_hDlgResSplit = GlobalHandle(lpDialogTemplate);
- // set up property page to use in-memory dialog template
- if(lpDialogTemplate != NULL)
- {
- this->m_psp.dwFlags |= PSP_DLGINDIRECT;
- this->m_psp.pResource = lpDialogTemplate;
- }
- else
- {
- ATLASSERT(FALSE && _T("CAxPropertyPageImpl - ActiveX initializtion failed!"));
- }
- }
- else
- {
- ATLASSERT(FALSE && _T("CAxPropertyPageImpl - Cannot find dialog template!"));
- }
- }
- ~CAxPropertyPageImpl()
- {
- if(m_hInitData != NULL)
- {
- UnlockResource(m_hInitData);
- FreeResource(m_hInitData);
- }
- if(m_hDlgRes != NULL)
- {
- UnlockResource(m_hDlgRes);
- FreeResource(m_hDlgRes);
- }
- if(m_hDlgResSplit != NULL)
- {
- ::GlobalFree(m_hDlgResSplit);
- }
- }
- // Methods
- // call this one to handle keyboard message for ActiveX controls
- BOOL PreTranslateMessage(LPMSG pMsg)
- {
- if (((pMsg->message < WM_KEYFIRST) || (pMsg->message > WM_KEYLAST)) &&
- ((pMsg->message < WM_MOUSEFIRST) || (pMsg->message > WM_MOUSELAST)))
- return FALSE;
- // find a direct child of the dialog from the window that has focus
- HWND hWndCtl = ::GetFocus();
- if (this->IsChild(hWndCtl) && (::GetParent(hWndCtl) != this->m_hWnd))
- {
- do
- {
- hWndCtl = ::GetParent(hWndCtl);
- }
- while (::GetParent(hWndCtl) != this->m_hWnd);
- }
- // give controls a chance to translate this message
- return (BOOL)::SendMessage(hWndCtl, WM_FORWARDMSG, 0, (LPARAM)pMsg);
- }
- // Overridables
- // new default implementation for ActiveX hosting pages
- #ifdef _WTL_NEW_PAGE_NOTIFY_HANDLERS
- int OnTranslateAccelerator(LPMSG lpMsg)
- {
- T* pT = static_cast<T*>(this);
- return (pT->PreTranslateMessage(lpMsg) != FALSE) ? PSNRET_MESSAGEHANDLED : PSNRET_NOERROR;
- }
- #else // !_WTL_NEW_PAGE_NOTIFY_HANDLERS
- BOOL OnTranslateAccelerator(LPMSG lpMsg)
- {
- T* pT = static_cast<T*>(this);
- return pT->PreTranslateMessage(lpMsg);
- }
- #endif // !_WTL_NEW_PAGE_NOTIFY_HANDLERS
- int GetIDD()
- {
- return( static_cast<T*>(this)->IDD );
- }
- virtual DLGPROC GetDialogProc()
- {
- return DialogProc;
- }
- static INT_PTR CALLBACK DialogProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
- {
- CAxPropertyPageImpl< T, TBase >* pThis = (CAxPropertyPageImpl< T, TBase >*)hWnd;
- if (uMsg == WM_INITDIALOG)
- {
- HRESULT hr;
- if (FAILED(hr = pThis->CreateActiveXControls(pThis->GetIDD())))
- {
- ATLASSERT(FALSE);
- return FALSE;
- }
- }
- return CPropertyPageImpl< T, TBase >::DialogProc(hWnd, uMsg, wParam, lParam);
- }
- // ActiveX controls creation
- virtual HRESULT CreateActiveXControls(UINT nID)
- {
- // Load dialog template and InitData
- HRSRC hDlgInit = ::FindResource(ATL::_AtlBaseModule.GetResourceInstance(), MAKEINTRESOURCE(nID), (LPTSTR)_ATL_RT_DLGINIT);
- BYTE* pInitData = NULL;
- HGLOBAL hData = NULL;
- HRESULT hr = S_OK;
- if (hDlgInit != NULL)
- {
- hData = ::LoadResource(ATL::_AtlBaseModule.GetResourceInstance(), hDlgInit);
- if (hData != NULL)
- pInitData = (BYTE*) ::LockResource(hData);
- }
- HRSRC hDlg = ::FindResource(ATL::_AtlBaseModule.GetResourceInstance(), MAKEINTRESOURCE(nID), (LPTSTR)RT_DIALOG);
- if (hDlg != NULL)
- {
- HGLOBAL hResource = ::LoadResource(ATL::_AtlBaseModule.GetResourceInstance(), hDlg);
- DLGTEMPLATE* pDlg = NULL;
- if (hResource != NULL)
- {
- pDlg = (DLGTEMPLATE*) ::LockResource(hResource);
- if (pDlg != NULL)
- {
- // Get first control on the template
- BOOL bDialogEx = ATL::_DialogSplitHelper::IsDialogEx(pDlg);
- WORD nItems = ATL::_DialogSplitHelper::DlgTemplateItemCount(pDlg);
- // Get first control on the dialog
- DLGITEMTEMPLATE* pItem = ATL::_DialogSplitHelper::FindFirstDlgItem(pDlg);
- HWND hWndPrev = this->GetWindow(GW_CHILD);
- // Create all ActiveX cotnrols in the dialog template and place them in the correct tab order (z-order)
- for (WORD nItem = 0; nItem < nItems; nItem++)
- {
- DWORD wID = bDialogEx ? ((ATL::_DialogSplitHelper::DLGITEMTEMPLATEEX*)pItem)->id : pItem->id;
- if (ATL::_DialogSplitHelper::IsActiveXControl(pItem, bDialogEx))
- {
- BYTE* pData = NULL;
- DWORD dwLen = ATL::_DialogSplitHelper::FindCreateData(wID, pInitData, &pData);
- ATL::CComPtr<IStream> spStream;
- if (dwLen != 0)
- {
- HGLOBAL h = GlobalAlloc(GHND, dwLen);
- if (h != NULL)
- {
- BYTE* pBytes = (BYTE*) GlobalLock(h);
- BYTE* pSource = pData;
- ATL::Checked::memcpy_s(pBytes, dwLen, pSource, dwLen);
- GlobalUnlock(h);
- CreateStreamOnHGlobal(h, TRUE, &spStream);
- }
- else
- {
- hr = E_OUTOFMEMORY;
- break;
- }
- }
- ATL::CComBSTR bstrLicKey;
- hr = ATL::_DialogSplitHelper::ParseInitData(spStream, &bstrLicKey.m_str);
- if (SUCCEEDED(hr))
- {
- ATL::CAxWindow2 wnd;
- // Get control caption.
- LPWSTR pszClassName =
- bDialogEx ?
- (LPWSTR)(((ATL::_DialogSplitHelper::DLGITEMTEMPLATEEX*)pItem) + 1) :
- (LPWSTR)(pItem + 1);
- // Get control rect.
- RECT rect = {};
- rect.left = bDialogEx ? ((ATL::_DialogSplitHelper::DLGITEMTEMPLATEEX*)pItem)->x : pItem->x;
- rect.top = bDialogEx ? ((ATL::_DialogSplitHelper::DLGITEMTEMPLATEEX*)pItem)->y : pItem->y;
- rect.right = rect.left + (bDialogEx ? ((ATL::_DialogSplitHelper::DLGITEMTEMPLATEEX*)pItem)->cx : pItem->cx);
- rect.bottom = rect.top + (bDialogEx ? ((ATL::_DialogSplitHelper::DLGITEMTEMPLATEEX*)pItem)->cy : pItem->cy);
- // Convert from dialog units to screen units
- this->MapDialogRect(&rect);
- // Create AxWindow with a NULL caption.
- wnd.Create(this->m_hWnd,
- &rect,
- NULL,
- (bDialogEx ?
- ((ATL::_DialogSplitHelper::DLGITEMTEMPLATEEX*)pItem)->style :
- pItem->style) | WS_TABSTOP,
- bDialogEx ?
- ((ATL::_DialogSplitHelper::DLGITEMTEMPLATEEX*)pItem)->exStyle :
- 0,
- bDialogEx ?
- ((ATL::_DialogSplitHelper::DLGITEMTEMPLATEEX*)pItem)->id :
- pItem->id,
- NULL);
- if (wnd != NULL)
- {
- // Set the Help ID
- if (bDialogEx && ((ATL::_DialogSplitHelper::DLGITEMTEMPLATEEX*)pItem)->helpID != 0)
- wnd.SetWindowContextHelpId(((ATL::_DialogSplitHelper::DLGITEMTEMPLATEEX*)pItem)->helpID);
- // Try to create the ActiveX control.
- hr = wnd.CreateControlLic(pszClassName, spStream, NULL, bstrLicKey);
- if (FAILED(hr))
- break;
- // Set the correct tab position.
- if (nItem == 0)
- hWndPrev = HWND_TOP;
- wnd.SetWindowPos(hWndPrev, 0,0,0,0,SWP_NOACTIVATE | SWP_NOMOVE | SWP_NOSIZE);
- hWndPrev = wnd;
- }
- else
- {
- hr = ATL::AtlHresultFromLastError();
- }
- }
- }
- else
- {
- if (nItem != 0)
- hWndPrev = ::GetWindow(hWndPrev, GW_HWNDNEXT);
- }
- pItem = ATL::_DialogSplitHelper::FindNextDlgItem(pItem, bDialogEx);
- }
- }
- else
- hr = ATL::AtlHresultFromLastError();
- }
- else
- hr = ATL::AtlHresultFromLastError();
- }
- return hr;
- }
- // Event handling support
- HRESULT AdviseSinkMap(bool bAdvise)
- {
- if(!bAdvise && (this->m_hWnd == NULL))
- {
- // window is gone, controls are already unadvised
- ATLTRACE2(atlTraceUI, 0, _T("CAxPropertyPageImpl::AdviseSinkMap called after the window was destroyed\n"));
- return S_OK;
- }
- HRESULT hRet = E_NOTIMPL;
- __if_exists(T::_GetSinkMapFinder)
- {
- T* pT = static_cast<T*>(this);
- hRet = AtlAdviseSinkMap(pT, bAdvise);
- }
- return hRet;
- }
- // Message map and handlers
- typedef CPropertyPageImpl< T, TBase> _baseClass;
- BEGIN_MSG_MAP(CAxPropertyPageImpl)
- MESSAGE_HANDLER(WM_INITDIALOG, OnInitDialog)
- MESSAGE_HANDLER(WM_DESTROY, OnDestroy)
- CHAIN_MSG_MAP(_baseClass)
- END_MSG_MAP()
- LRESULT OnInitDialog(UINT /*uMsg*/, WPARAM /*wParam*/, LPARAM /*lParam*/, BOOL& bHandled)
- {
- // initialize controls in dialog with DLGINIT resource section
- this->ExecuteDlgInit(static_cast<T*>(this)->IDD);
- AdviseSinkMap(true);
- bHandled = FALSE;
- return 1;
- }
- LRESULT OnDestroy(UINT /*uMsg*/, WPARAM /*wParam*/, LPARAM /*lParam*/, BOOL& bHandled)
- {
- AdviseSinkMap(false);
- bHandled = FALSE;
- return 1;
- }
- };
- // for non-customized pages
- template <WORD t_wDlgTemplateID>
- class CAxPropertyPage : public CAxPropertyPageImpl<CAxPropertyPage<t_wDlgTemplateID> >
- {
- public:
- enum { IDD = t_wDlgTemplateID };
- CAxPropertyPage(ATL::_U_STRINGorID title = (LPCTSTR)NULL) : CAxPropertyPageImpl<CAxPropertyPage>(title)
- { }
- BEGIN_MSG_MAP(CAxPropertyPage)
- CHAIN_MSG_MAP(CAxPropertyPageImpl<CAxPropertyPage<t_wDlgTemplateID> >)
- END_MSG_MAP()
- };
- #endif // _ATL_NO_HOSTING
- ///////////////////////////////////////////////////////////////////////////////
- // Wizard97 Support
- // Sample wizard dialog resources:
- //
- // IDD_WIZ97_INTERIOR_BLANK DIALOG 0, 0, 317, 143
- // STYLE DS_SETFONT | WS_CHILD | WS_DISABLED | WS_CAPTION
- // CAPTION "Wizard97 Property Page - Interior"
- // FONT 8, "MS Shell Dlg"
- // BEGIN
- // END
- //
- // IDD_WIZ97_EXTERIOR_BLANK DIALOGEX 0, 0, 317, 193
- // STYLE DS_SETFONT | DS_FIXEDSYS | WS_CHILD | WS_DISABLED | WS_CAPTION
- // CAPTION "Wizard97 Property Page - Welcome/Complete"
- // FONT 8, "MS Shell Dlg", 0, 0, 0x0
- // BEGIN
- // LTEXT "Welcome to the X Wizard",IDC_WIZ97_EXTERIOR_TITLE,115,8,
- // 195,24
- // LTEXT "Wizard Explanation\r\n(The height of the static text should be in multiples of 8 dlus)",
- // IDC_STATIC,115,40,195,16
- // LTEXT "h",IDC_WIZ97_BULLET1,118,64,8,8
- // LTEXT "List Item 1 (the h is turned into a bullet)",IDC_STATIC,
- // 127,63,122,8
- // LTEXT "h",IDC_WIZ97_BULLET2,118,79,8,8
- // LTEXT "List Item 2. Keep 7 dlus between paragraphs",IDC_STATIC,
- // 127,78,33,8
- // CONTROL "&Do not show this Welcome page again",
- // IDC_WIZ97_WELCOME_NOTAGAIN,"Button",BS_AUTOCHECKBOX |
- // WS_TABSTOP,115,169,138,10
- // END
- //
- // GUIDELINES DESIGNINFO
- // BEGIN
- // IDD_WIZ97_INTERIOR_BLANK, DIALOG
- // BEGIN
- // LEFTMARGIN, 7
- // RIGHTMARGIN, 310
- // VERTGUIDE, 21
- // VERTGUIDE, 31
- // VERTGUIDE, 286
- // VERTGUIDE, 296
- // TOPMARGIN, 7
- // BOTTOMMARGIN, 136
- // HORZGUIDE, 8
- // END
- //
- // IDD_WIZ97_EXTERIOR_BLANK, DIALOG
- // BEGIN
- // RIGHTMARGIN, 310
- // VERTGUIDE, 115
- // VERTGUIDE, 118
- // VERTGUIDE, 127
- // TOPMARGIN, 7
- // BOTTOMMARGIN, 186
- // HORZGUIDE, 8
- // HORZGUIDE, 32
- // HORZGUIDE, 40
- // HORZGUIDE, 169
- // END
- // END
- ///////////////////////////////////////////////////////////////////////////////
- // CWizard97SheetWindow - client side for a Wizard 97 style wizard sheet
- class CWizard97SheetWindow : public CPropertySheetWindow
- {
- public:
- // Constructors
- CWizard97SheetWindow(HWND hWnd = NULL) : CPropertySheetWindow(hWnd)
- { }
- CWizard97SheetWindow& operator =(HWND hWnd)
- {
- m_hWnd = hWnd;
- return *this;
- }
- // Operations
- HFONT GetExteriorPageTitleFont(void)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- return (HFONT)::SendMessage(m_hWnd, GetMessage_GetExteriorPageTitleFont(), 0, 0L);
- }
- HFONT GetBulletFont(void)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- return (HFONT)::SendMessage(m_hWnd, GetMessage_GetBulletFont(), 0, 0L);
- }
- // Helpers
- static UINT GetMessage_GetExteriorPageTitleFont()
- {
- static UINT uGetExteriorPageTitleFont = 0;
- if(uGetExteriorPageTitleFont == 0)
- {
- CStaticDataInitCriticalSectionLock lock;
- if(FAILED(lock.Lock()))
- {
- ATLTRACE2(atlTraceUI, 0, _T("ERROR : Unable to lock critical section in CWizard97SheetWindow::GetMessage_GetExteriorPageTitleFont().\n"));
- ATLASSERT(FALSE);
- return 0;
- }
- if(uGetExteriorPageTitleFont == 0)
- uGetExteriorPageTitleFont = ::RegisterWindowMessage(_T("GetExteriorPageTitleFont_531AF056-B8BE-4c4c-B786-AC608DF0DF12"));
- lock.Unlock();
- }
- ATLASSERT(uGetExteriorPageTitleFont != 0);
- return uGetExteriorPageTitleFont;
- }
- static UINT GetMessage_GetBulletFont()
- {
- static UINT uGetBulletFont = 0;
- if(uGetBulletFont == 0)
- {
- CStaticDataInitCriticalSectionLock lock;
- if(FAILED(lock.Lock()))
- {
- ATLTRACE2(atlTraceUI, 0, _T("ERROR : Unable to lock critical section in CWizard97SheetWindow::GetMessage_GetBulletFont().\n"));
- ATLASSERT(FALSE);
- return 0;
- }
- if(uGetBulletFont == 0)
- uGetBulletFont = ::RegisterWindowMessage(_T("GetBulletFont_AD347D08-8F65-45ef-982E-6352E8218AD5"));
- lock.Unlock();
- }
- ATLASSERT(uGetBulletFont != 0);
- return uGetBulletFont;
- }
- // Implementation - override to prevent usage
- HWND Create(LPCTSTR, HWND, ATL::_U_RECT = NULL, LPCTSTR = NULL, DWORD = 0, DWORD = 0, ATL::_U_MENUorID = 0U, LPVOID = NULL)
- {
- ATLASSERT(FALSE);
- return NULL;
- }
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CWizard97SheetImpl - implements a Wizard 97 style wizard sheet
- template <class T, class TBase = CWizard97SheetWindow>
- class ATL_NO_VTABLE CWizard97SheetImpl : public CPropertySheetImpl< T, TBase >
- {
- protected:
- // Typedefs
- typedef CWizard97SheetImpl< T, TBase > thisClass;
- typedef CPropertySheetImpl< T, TBase > baseClass;
- // Member variables
- CFont m_fontExteriorPageTitle; // Welcome and Completion page title font
- CFont m_fontBullet; // Bullet font (used on static text 'h' to produce a small bullet)
- bool m_bReceivedFirstSizeMessage;
- public:
- CWizard97SheetImpl(ATL::_U_STRINGorID title, ATL::_U_STRINGorID headerBitmap, ATL::_U_STRINGorID watermarkBitmap, UINT uStartPage = 0, HWND hWndParent = NULL) :
- baseClass(title, uStartPage, hWndParent),
- m_bReceivedFirstSizeMessage(false)
- {
- this->m_psh.dwFlags &= ~(PSH_NOCONTEXTHELP);
- this->m_psh.dwFlags &= ~(PSH_WIZARD | PSH_WIZARD_LITE);
- this->m_psh.dwFlags |= (PSH_HASHELP | PSH_WIZARDCONTEXTHELP);
- this->m_psh.dwFlags |= PSH_WIZARD97;
- baseClass::SetHeader(headerBitmap.m_lpstr);
- baseClass::SetWatermark(watermarkBitmap.m_lpstr);
- }
- // Overrides from base class
- void OnSheetInitialized()
- {
- T* pT = static_cast<T*>(this);
- pT->_InitializeFonts();
- // We'd like to center the wizard here, but its too early.
- // Instead, we'll do CenterWindow upon our first WM_SIZE message
- }
- // Initialization
- void _InitializeFonts()
- {
- // Setup the Title and Bullet Font
- // (Property pages can send the "get external page title font" and "get bullet font" messages)
- // The derived class needs to do the actual SetFont for the dialog items)
- CFontHandle fontThisDialog = this->GetFont();
- CClientDC dcScreen(NULL);
- LOGFONT titleLogFont = {};
- LOGFONT bulletLogFont = {};
- fontThisDialog.GetLogFont(&titleLogFont);
- fontThisDialog.GetLogFont(&bulletLogFont);
- // The Wizard 97 Spec recommends to do the Title Font
- // as Verdana Bold, 12pt.
- titleLogFont.lfCharSet = DEFAULT_CHARSET;
- titleLogFont.lfWeight = FW_BOLD;
- ATL::Checked::tcscpy_s(titleLogFont.lfFaceName, _countof(titleLogFont.lfFaceName), _T("Verdana Bold"));
- INT titleFontPointSize = 12;
- titleLogFont.lfHeight = -::MulDiv(titleFontPointSize, dcScreen.GetDeviceCaps(LOGPIXELSY), 72);
- m_fontExteriorPageTitle.CreateFontIndirect(&titleLogFont);
- // The Wizard 97 Spec recommends to do Bullets by having
- // static text of "h" in the Marlett font.
- bulletLogFont.lfCharSet = DEFAULT_CHARSET;
- bulletLogFont.lfWeight = FW_NORMAL;
- ATL::Checked::tcscpy_s(bulletLogFont.lfFaceName, _countof(bulletLogFont.lfFaceName), _T("Marlett"));
- INT bulletFontSize = 8;
- bulletLogFont.lfHeight = -::MulDiv(bulletFontSize, dcScreen.GetDeviceCaps(LOGPIXELSY), 72);
- m_fontBullet.CreateFontIndirect(&bulletLogFont);
- }
- // Message Handling
- BEGIN_MSG_MAP(thisClass)
- MESSAGE_HANDLER(CWizard97SheetWindow::GetMessage_GetExteriorPageTitleFont(), OnGetExteriorPageTitleFont)
- MESSAGE_HANDLER(CWizard97SheetWindow::GetMessage_GetBulletFont(), OnGetBulletFont)
- MESSAGE_HANDLER(WM_SIZE, OnSize)
- CHAIN_MSG_MAP(baseClass)
- END_MSG_MAP()
- LRESULT OnGetExteriorPageTitleFont(UINT /*uMsg*/, WPARAM /*wParam*/, LPARAM /*lParam*/, BOOL& /*bHandled*/)
- {
- return (LRESULT)(HFONT)m_fontExteriorPageTitle;
- }
- LRESULT OnGetBulletFont(UINT /*uMsg*/, WPARAM /*wParam*/, LPARAM /*lParam*/, BOOL& /*bHandled*/)
- {
- return (LRESULT)(HFONT)m_fontBullet;
- }
- LRESULT OnSize(UINT /*uMsg*/, WPARAM /*wParam*/, LPARAM /*lParam*/, BOOL& bHandled)
- {
- if(!m_bReceivedFirstSizeMessage)
- {
- m_bReceivedFirstSizeMessage = true;
- this->CenterWindow();
- }
- bHandled = FALSE;
- return 0;
- }
- };
- // for non-customized sheets
- class CWizard97Sheet : public CWizard97SheetImpl<CWizard97Sheet>
- {
- protected:
- // Typedefs
- typedef CWizard97Sheet thisClass;
- typedef CWizard97SheetImpl<CWizard97Sheet> baseClass;
- public:
- CWizard97Sheet(ATL::_U_STRINGorID title, ATL::_U_STRINGorID headerBitmap, ATL::_U_STRINGorID watermarkBitmap, UINT uStartPage = 0, HWND hWndParent = NULL) :
- baseClass(title, headerBitmap, watermarkBitmap, uStartPage, hWndParent)
- { }
- BEGIN_MSG_MAP(thisClass)
- CHAIN_MSG_MAP(baseClass)
- END_MSG_MAP()
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CWizard97PageWindow - client side for a Wizard 97 style wizard page
- #define WIZARD97_EXTERIOR_CXDLG 317
- #define WIZARD97_EXTERIOR_CYDLG 193
- #define WIZARD97_INTERIOR_CXDLG 317
- #define WIZARD97_INTERIOR_CYDLG 143
- class CWizard97PageWindow : public CPropertyPageWindow
- {
- public:
- // Constructors
- CWizard97PageWindow(HWND hWnd = NULL) : CPropertyPageWindow(hWnd)
- { }
- CWizard97PageWindow& operator =(HWND hWnd)
- {
- m_hWnd = hWnd;
- return *this;
- }
- // Attributes
- CWizard97SheetWindow GetPropertySheet() const
- {
- ATLASSERT(::IsWindow(m_hWnd));
- return CWizard97SheetWindow(GetParent());
- }
- // Operations
- HFONT GetExteriorPageTitleFont(void)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- return GetPropertySheet().GetExteriorPageTitleFont();
- }
- HFONT GetBulletFont(void)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- return GetPropertySheet().GetBulletFont();
- }
- // Implementation - overrides to prevent usage
- HWND Create(LPCTSTR, HWND, ATL::_U_RECT = NULL, LPCTSTR = NULL, DWORD = 0, DWORD = 0, ATL::_U_MENUorID = 0U, LPVOID = NULL)
- {
- ATLASSERT(FALSE);
- return NULL;
- }
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CWizard97PageImpl - implements a Wizard 97 style wizard page
- template <class T, class TBase = CWizard97PageWindow>
- class ATL_NO_VTABLE CWizard97PageImpl : public CPropertyPageImpl< T, TBase >
- {
- protected:
- // Typedefs
- typedef CWizard97PageImpl< T, TBase > thisClass;
- typedef CPropertyPageImpl< T, TBase > baseClass;
- public:
- CWizard97PageImpl(ATL::_U_STRINGorID title = (LPCTSTR)NULL) : baseClass(title)
- { }
- // Message Handling
- BEGIN_MSG_MAP(thisClass)
- CHAIN_MSG_MAP(baseClass)
- END_MSG_MAP()
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CWizard97ExteriorPageImpl - implements a Wizard 97 style exterior wizard page
- template <class T, class TBase = CWizard97PageWindow>
- class ATL_NO_VTABLE CWizard97ExteriorPageImpl : public CPropertyPageImpl< T, TBase >
- {
- protected:
- // Typedefs
- typedef CWizard97ExteriorPageImpl< T, TBase > thisClass;
- typedef CPropertyPageImpl< T, TBase > baseClass;
- public:
- // Constructors
- CWizard97ExteriorPageImpl(ATL::_U_STRINGorID title = (LPCTSTR)NULL) : baseClass(title)
- {
- this->m_psp.dwFlags |= PSP_HASHELP;
- this->m_psp.dwFlags |= PSP_HIDEHEADER;
- }
- // Message Handling
- BEGIN_MSG_MAP(thisClass)
- CHAIN_MSG_MAP(baseClass)
- END_MSG_MAP()
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CWizard97InteriorPageImpl - implements a Wizard 97 style interior wizard page
- template <class T, class TBase = CWizard97PageWindow>
- class ATL_NO_VTABLE CWizard97InteriorPageImpl : public CPropertyPageImpl< T, TBase >
- {
- protected:
- // Typedefs
- typedef CWizard97InteriorPageImpl< T, TBase > thisClass;
- typedef CPropertyPageImpl< T, TBase > baseClass;
- public:
- // Constructors
- CWizard97InteriorPageImpl(ATL::_U_STRINGorID title = (LPCTSTR)NULL) : baseClass(title)
- {
- this->m_psp.dwFlags |= PSP_HASHELP;
- this->m_psp.dwFlags &= ~PSP_HIDEHEADER;
- this->m_psp.dwFlags |= PSP_USEHEADERTITLE | PSP_USEHEADERSUBTITLE;
- // Be sure to have the derived class define this in the constructor.
- // We'll default it to something obvious in case its forgotten.
- baseClass::SetHeaderTitle(_T("Call SetHeaderTitle in Derived Class"));
- baseClass::SetHeaderSubTitle(_T("Call SetHeaderSubTitle in the constructor of the Derived Class."));
- }
- // Message Handling
- BEGIN_MSG_MAP(thisClass)
- CHAIN_MSG_MAP(baseClass)
- END_MSG_MAP()
- };
- ///////////////////////////////////////////////////////////////////////////////
- // Aero Wizard support
- #if (_WIN32_WINNT >= 0x0600)
- ///////////////////////////////////////////////////////////////////////////////
- // CAeroWizardFrameWindow - client side for an Aero Wizard frame window
- class CAeroWizardFrameWindow : public CPropertySheetWindow
- {
- public:
- // Constructors
- CAeroWizardFrameWindow(HWND hWnd = NULL) : CPropertySheetWindow(hWnd)
- { }
- CAeroWizardFrameWindow& operator =(HWND hWnd)
- {
- m_hWnd = hWnd;
- return *this;
- }
- // Operations - new, Aero Wizard only
- void SetNextText(LPCWSTR lpszText)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ::SendMessage(m_hWnd, PSM_SETNEXTTEXT, 0, (LPARAM)lpszText);
- }
- void ShowWizardButtons(DWORD dwButtons, DWORD dwStates)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ::PostMessage(m_hWnd, PSM_SHOWWIZBUTTONS, (WPARAM)dwStates, (LPARAM)dwButtons);
- }
- void EnableWizardButtons(DWORD dwButtons, DWORD dwStates)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ::PostMessage(m_hWnd, PSM_ENABLEWIZBUTTONS, (WPARAM)dwStates, (LPARAM)dwButtons);
- }
- void SetButtonText(DWORD dwButton, LPCWSTR lpszText)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ::SendMessage(m_hWnd, PSM_SETBUTTONTEXT, (WPARAM)dwButton, (LPARAM)lpszText);
- }
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CAeroWizardFrameImpl - implements an Aero Wizard frame
- template <class T, class TBase = CAeroWizardFrameWindow>
- class ATL_NO_VTABLE CAeroWizardFrameImpl : public CPropertySheetImpl<T, TBase >
- {
- public:
- // Constructor
- CAeroWizardFrameImpl(ATL::_U_STRINGorID title = (LPCTSTR)NULL, UINT uStartPage = 0, HWND hWndParent = NULL) :
- CPropertySheetImpl<T, TBase >(title, uStartPage, hWndParent)
- {
- this->m_psh.dwFlags |= PSH_WIZARD | PSH_AEROWIZARD;
- }
- // Operations
- void EnableResizing()
- {
- ATLASSERT(this->m_hWnd == NULL); // can't do this after it's created
- this->m_psh.dwFlags |= PSH_RESIZABLE;
- }
- void UseHeaderBitmap()
- {
- ATLASSERT(this->m_hWnd == NULL); // can't do this after it's created
- this->m_psh.dwFlags |= PSH_HEADERBITMAP;
- }
- void SetNoMargin()
- {
- ATLASSERT(this->m_hWnd == NULL); // can't do this after it's created
- this->m_psh.dwFlags |= PSH_NOMARGIN;
- }
- // Override to prevent use
- HWND Create(HWND /*hWndParent*/ = NULL)
- {
- ATLASSERT(FALSE); // not supported for Aero Wizard
- return NULL;
- }
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CAeroWizardFrame - for non-customized frames
- class CAeroWizardFrame : public CAeroWizardFrameImpl<CAeroWizardFrame>
- {
- public:
- CAeroWizardFrame(ATL::_U_STRINGorID title = (LPCTSTR)NULL, UINT uStartPage = 0, HWND hWndParent = NULL)
- : CAeroWizardFrameImpl<CAeroWizardFrame>(title, uStartPage, hWndParent)
- { }
- BEGIN_MSG_MAP(CAeroWizardFrame)
- MESSAGE_HANDLER(WM_COMMAND, CAeroWizardFrameImpl<CAeroWizardFrame>::OnCommand)
- END_MSG_MAP()
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CAeroWizardPageWindow - client side for an Aero Wizard page
- class CAeroWizardPageWindow : public CPropertyPageWindow
- {
- public:
- // Constructors
- CAeroWizardPageWindow(HWND hWnd = NULL) : CPropertyPageWindow(hWnd)
- { }
- CAeroWizardPageWindow& operator =(HWND hWnd)
- {
- m_hWnd = hWnd;
- return *this;
- }
- // Attributes
- CAeroWizardFrameWindow GetAeroWizardFrame() const
- {
- ATLASSERT(::IsWindow(m_hWnd));
- // This is not really top-level frame window, but it processes all frame messages
- return CAeroWizardFrameWindow(GetParent());
- }
- // Operations - new, Aero Wizard only
- void SetNextText(LPCWSTR lpszText)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ATLASSERT(GetParent() != NULL);
- GetAeroWizardFrame().SetNextText(lpszText);
- }
- void ShowWizardButtons(DWORD dwButtons, DWORD dwStates)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ATLASSERT(GetParent() != NULL);
- GetAeroWizardFrame().ShowWizardButtons(dwButtons, dwStates);
- }
- void EnableWizardButtons(DWORD dwButtons, DWORD dwStates)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ATLASSERT(GetParent() != NULL);
- GetAeroWizardFrame().EnableWizardButtons(dwButtons, dwStates);
- }
- void SetButtonText(DWORD dwButton, LPCWSTR lpszText)
- {
- ATLASSERT(::IsWindow(m_hWnd));
- ATLASSERT(GetParent() != NULL);
- GetAeroWizardFrame().SetButtonText(dwButton, lpszText);
- }
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CAeroWizardPageImpl - implements an Aero Wizard page
- template <class T, class TBase = CAeroWizardPageWindow>
- class ATL_NO_VTABLE CAeroWizardPageImpl : public CPropertyPageImpl<T, TBase >
- {
- public:
- CAeroWizardPageImpl(ATL::_U_STRINGorID title = (LPCTSTR)NULL) : CPropertyPageImpl<T, TBase >(title)
- { }
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CAeroWizardPage - for non-customized pages
- template <WORD t_wDlgTemplateID>
- class CAeroWizardPage : public CAeroWizardPageImpl<CAeroWizardPage<t_wDlgTemplateID> >
- {
- public:
- enum { IDD = t_wDlgTemplateID };
- CAeroWizardPage(ATL::_U_STRINGorID title = (LPCTSTR)NULL) : CAeroWizardPageImpl<CAeroWizardPage>(title)
- { }
- DECLARE_EMPTY_MSG_MAP()
- };
- #ifndef _ATL_NO_HOSTING
- // Note: You must #include <atlhost.h> to use these classes
- ///////////////////////////////////////////////////////////////////////////////
- // CAeroWizardAxPageImpl - Aero Wizard page that hosts ActiveX controls
- template <class T, class TBase = CAeroWizardPageWindow>
- class ATL_NO_VTABLE CAeroWizardAxPageImpl : public CAxPropertyPageImpl< T, TBase >
- {
- public:
- CAeroWizardAxPageImpl(ATL::_U_STRINGorID title = (LPCTSTR)NULL) : CAxPropertyPageImpl< T, TBase >(title)
- { }
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CAeroWizardAxPage - for non-customized pages
- template <WORD t_wDlgTemplateID>
- class CAeroWizardAxPage : public CAeroWizardAxPageImpl<CAeroWizardAxPage<t_wDlgTemplateID> >
- {
- public:
- enum { IDD = t_wDlgTemplateID };
- CAeroWizardAxPage(ATL::_U_STRINGorID title = (LPCTSTR)NULL) : CAeroWizardAxPageImpl<CAeroWizardAxPage>(title)
- { }
- BEGIN_MSG_MAP(CAeroWizardAxPage)
- CHAIN_MSG_MAP(CAeroWizardAxPageImpl<CAeroWizardAxPage<t_wDlgTemplateID> >)
- END_MSG_MAP()
- };
- #endif // _ATL_NO_HOSTING
- #endif // (_WIN32_WINNT >= 0x0600)
- ///////////////////////////////////////////////////////////////////////////////
- // TaskDialog support
- #if (_WIN32_WINNT >= 0x0600) || defined(_WTL_TASKDIALOG)
- ///////////////////////////////////////////////////////////////////////////////
- // AtlTaskDialog - support for TaskDialog() function
- inline int AtlTaskDialog(HWND hWndParent,
- ATL::_U_STRINGorID WindowTitle, ATL::_U_STRINGorID MainInstructionText, ATL::_U_STRINGorID ContentText,
- TASKDIALOG_COMMON_BUTTON_FLAGS dwCommonButtons = 0U, ATL::_U_STRINGorID Icon = (LPCTSTR)NULL)
- {
- int nRet = -1;
- #ifdef _WTL_TASKDIALOG_DIRECT
- USES_CONVERSION;
- HRESULT hRet = ::TaskDialog(hWndParent, ModuleHelper::GetResourceInstance(),
- IS_INTRESOURCE(WindowTitle.m_lpstr) ? (LPCWSTR) WindowTitle.m_lpstr : T2CW(WindowTitle.m_lpstr),
- IS_INTRESOURCE(MainInstructionText.m_lpstr) ? (LPCWSTR) MainInstructionText.m_lpstr : T2CW(MainInstructionText.m_lpstr),
- IS_INTRESOURCE(ContentText.m_lpstr) ? (LPCWSTR) ContentText.m_lpstr : T2CW(ContentText.m_lpstr),
- dwCommonButtons,
- IS_INTRESOURCE(Icon.m_lpstr) ? (LPCWSTR) Icon.m_lpstr : T2CW(Icon.m_lpstr),
- &nRet);
- ATLVERIFY(SUCCEEDED(hRet));
- #else
- // This allows apps to run on older versions of Windows
- typedef HRESULT (STDAPICALLTYPE *PFN_TaskDialog)(HWND hwndParent, HINSTANCE hInstance, PCWSTR pszWindowTitle, PCWSTR pszMainInstruction, PCWSTR pszContent, TASKDIALOG_COMMON_BUTTON_FLAGS dwCommonButtons, PCWSTR pszIcon, int* pnButton);
- HMODULE m_hCommCtrlDLL = ::LoadLibrary(_T("comctl32.dll"));
- if(m_hCommCtrlDLL != NULL)
- {
- PFN_TaskDialog pfnTaskDialog = (PFN_TaskDialog)::GetProcAddress(m_hCommCtrlDLL, "TaskDialog");
- if(pfnTaskDialog != NULL)
- {
- USES_CONVERSION;
- HRESULT hRet = pfnTaskDialog(hWndParent, ModuleHelper::GetResourceInstance(),
- IS_INTRESOURCE(WindowTitle.m_lpstr) ? (LPCWSTR) WindowTitle.m_lpstr : T2CW(WindowTitle.m_lpstr),
- IS_INTRESOURCE(MainInstructionText.m_lpstr) ? (LPCWSTR) MainInstructionText.m_lpstr : T2CW(MainInstructionText.m_lpstr),
- IS_INTRESOURCE(ContentText.m_lpstr) ? (LPCWSTR) ContentText.m_lpstr : T2CW(ContentText.m_lpstr),
- dwCommonButtons,
- IS_INTRESOURCE(Icon.m_lpstr) ? (LPCWSTR) Icon.m_lpstr : T2CW(Icon.m_lpstr),
- &nRet);
- ATLVERIFY(SUCCEEDED(hRet));
- }
- ::FreeLibrary(m_hCommCtrlDLL);
- }
- #endif
- return nRet;
- }
- ///////////////////////////////////////////////////////////////////////////////
- // CTaskDialogConfig - TASKDIALOGCONFIG wrapper
- class CTaskDialogConfig : public TASKDIALOGCONFIG
- {
- public:
- // Constructor
- CTaskDialogConfig()
- {
- Init();
- }
- void Init()
- {
- memset(this, 0, sizeof(TASKDIALOGCONFIG)); // initialize structure to 0/NULL
- this->cbSize = sizeof(TASKDIALOGCONFIG);
- this->hInstance = ModuleHelper::GetResourceInstance();
- }
- // Operations - setting values
- // common buttons
- void SetCommonButtons(TASKDIALOG_COMMON_BUTTON_FLAGS dwCommonButtonsArg)
- {
- this->dwCommonButtons = dwCommonButtonsArg;
- }
- // window title text
- void SetWindowTitle(UINT nID)
- {
- this->pszWindowTitle = MAKEINTRESOURCEW(nID);
- }
- void SetWindowTitle(LPCWSTR lpstrWindowTitle)
- {
- this->pszWindowTitle = lpstrWindowTitle;
- }
- // main icon
- void SetMainIcon(HICON hIcon)
- {
- this->dwFlags |= TDF_USE_HICON_MAIN;
- this->hMainIcon = hIcon;
- }
- void SetMainIcon(UINT nID)
- {
- this->dwFlags &= ~TDF_USE_HICON_MAIN;
- this->pszMainIcon = MAKEINTRESOURCEW(nID);
- }
- void SetMainIcon(LPCWSTR lpstrMainIcon)
- {
- this->dwFlags &= ~TDF_USE_HICON_MAIN;
- this->pszMainIcon = lpstrMainIcon;
- }
- // main instruction text
- void SetMainInstructionText(UINT nID)
- {
- this->pszMainInstruction = MAKEINTRESOURCEW(nID);
- }
- void SetMainInstructionText(LPCWSTR lpstrMainInstruction)
- {
- this->pszMainInstruction = lpstrMainInstruction;
- }
- // content text
- void SetContentText(UINT nID)
- {
- this->pszContent = MAKEINTRESOURCEW(nID);
- }
- void SetContentText(LPCWSTR lpstrContent)
- {
- this->pszContent = lpstrContent;
- }
- // buttons
- void SetButtons(const TASKDIALOG_BUTTON* pButtonsArg, UINT cButtonsArg, int nDefaultButtonArg = 0)
- {
- this->pButtons = pButtonsArg;
- this->cButtons = cButtonsArg;
- if(nDefaultButtonArg != 0)
- this->nDefaultButton = nDefaultButtonArg;
- }
- void SetDefaultButton(int nDefaultButtonArg)
- {
- this->nDefaultButton = nDefaultButtonArg;
- }
- // radio buttons
- void SetRadioButtons(const TASKDIALOG_BUTTON* pRadioButtonsArg, UINT cRadioButtonsArg, int nDefaultRadioButtonArg = 0)
- {
- this->pRadioButtons = pRadioButtonsArg;
- this->cRadioButtons = cRadioButtonsArg;
- if(nDefaultRadioButtonArg != 0)
- this->nDefaultRadioButton = nDefaultRadioButtonArg;
- }
- void SetDefaultRadioButton(int nDefaultRadioButtonArg)
- {
- this->nDefaultRadioButton = nDefaultRadioButtonArg;
- }
- // verification text
- void SetVerificationText(UINT nID)
- {
- this->pszVerificationText = MAKEINTRESOURCEW(nID);
- }
- void SetVerificationText(LPCWSTR lpstrVerificationText)
- {
- this->pszVerificationText = lpstrVerificationText;
- }
- // expanded information text
- void SetExpandedInformationText(UINT nID)
- {
- this->pszExpandedInformation = MAKEINTRESOURCEW(nID);
- }
- void SetExpandedInformationText(LPCWSTR lpstrExpandedInformation)
- {
- this->pszExpandedInformation = lpstrExpandedInformation;
- }
- // expanded control text
- void SetExpandedControlText(UINT nID)
- {
- this->pszExpandedControlText = MAKEINTRESOURCEW(nID);
- }
- void SetExpandedControlText(LPCWSTR lpstrExpandedControlText)
- {
- this->pszExpandedControlText = lpstrExpandedControlText;
- }
- // collapsed control text
- void SetCollapsedControlText(UINT nID)
- {
- this->pszCollapsedControlText = MAKEINTRESOURCEW(nID);
- }
- void SetCollapsedControlText(LPCWSTR lpstrCollapsedControlText)
- {
- this->pszCollapsedControlText = lpstrCollapsedControlText;
- }
- // footer icon
- void SetFooterIcon(HICON hIcon)
- {
- this->dwFlags |= TDF_USE_HICON_FOOTER;
- this->hFooterIcon = hIcon;
- }
- void SetFooterIcon(UINT nID)
- {
- this->dwFlags &= ~TDF_USE_HICON_FOOTER;
- this->pszFooterIcon = MAKEINTRESOURCEW(nID);
- }
- void SetFooterIcon(LPCWSTR lpstrFooterIcon)
- {
- this->dwFlags &= ~TDF_USE_HICON_FOOTER;
- this->pszFooterIcon = lpstrFooterIcon;
- }
- // footer text
- void SetFooterText(UINT nID)
- {
- this->pszFooter = MAKEINTRESOURCEW(nID);
- }
- void SetFooterText(LPCWSTR lpstrFooterText)
- {
- this->pszFooter = lpstrFooterText;
- }
- // width (in DLUs)
- void SetWidth(UINT cxWidthArg)
- {
- this->cxWidth = cxWidthArg;
- }
- // modify flags
- void ModifyFlags(DWORD dwRemove, DWORD dwAdd)
- {
- this->dwFlags = (this->dwFlags & ~dwRemove) | dwAdd;
- }
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CTaskDialogImpl - implements a Task Dialog
- template <class T>
- class ATL_NO_VTABLE CTaskDialogImpl
- {
- public:
- CTaskDialogConfig m_tdc;
- HWND m_hWnd; // used only in callback functions
- // Constructor
- CTaskDialogImpl(HWND hWndParent = NULL) : m_hWnd(NULL)
- {
- m_tdc.hwndParent = hWndParent;
- m_tdc.pfCallback = T::TaskDialogCallback;
- m_tdc.lpCallbackData = (LONG_PTR)static_cast<T*>(this);
- }
- // Operations
- HRESULT DoModal(HWND hWndParent = ::GetActiveWindow(), int* pnButton = NULL, int* pnRadioButton = NULL, BOOL* pfVerificationFlagChecked = NULL)
- {
- if(m_tdc.hwndParent == NULL)
- m_tdc.hwndParent = hWndParent;
- #ifdef _WTL_TASKDIALOG_DIRECT
- return ::TaskDialogIndirect(&m_tdc, pnButton, pnRadioButton, pfVerificationFlagChecked);
- #else
- // This allows apps to run on older versions of Windows
- typedef HRESULT (STDAPICALLTYPE *PFN_TaskDialogIndirect)(const TASKDIALOGCONFIG* pTaskConfig, int* pnButton, int* pnRadioButton, BOOL* pfVerificationFlagChecked);
- HRESULT hRet = E_UNEXPECTED;
- HMODULE m_hCommCtrlDLL = ::LoadLibrary(_T("comctl32.dll"));
- if(m_hCommCtrlDLL != NULL)
- {
- PFN_TaskDialogIndirect pfnTaskDialogIndirect = (PFN_TaskDialogIndirect)::GetProcAddress(m_hCommCtrlDLL, "TaskDialogIndirect");
- if(pfnTaskDialogIndirect != NULL)
- hRet = pfnTaskDialogIndirect(&m_tdc, pnButton, pnRadioButton, pfVerificationFlagChecked);
- ::FreeLibrary(m_hCommCtrlDLL);
- }
- return hRet;
- #endif
- }
- // Operations - setting values of TASKDIALOGCONFIG
- // common buttons
- void SetCommonButtons(TASKDIALOG_COMMON_BUTTON_FLAGS dwCommonButtons)
- { m_tdc.SetCommonButtons(dwCommonButtons); }
- // window title text
- void SetWindowTitle(UINT nID)
- { m_tdc.SetWindowTitle(nID); }
- void SetWindowTitle(LPCWSTR lpstrWindowTitle)
- { m_tdc.SetWindowTitle(lpstrWindowTitle); }
- // main icon
- void SetMainIcon(HICON hIcon)
- { m_tdc.SetMainIcon(hIcon); }
- void SetMainIcon(UINT nID)
- { m_tdc.SetMainIcon(nID); }
- void SetMainIcon(LPCWSTR lpstrMainIcon)
- { m_tdc.SetMainIcon(lpstrMainIcon); }
- // main instruction text
- void SetMainInstructionText(UINT nID)
- { m_tdc.SetMainInstructionText(nID); }
- void SetMainInstructionText(LPCWSTR lpstrMainInstruction)
- { m_tdc.SetMainInstructionText(lpstrMainInstruction); }
- // content text
- void SetContentText(UINT nID)
- { m_tdc.SetContentText(nID); }
- void SetContentText(LPCWSTR lpstrContent)
- { m_tdc.SetContentText(lpstrContent); }
- // buttons
- void SetButtons(const TASKDIALOG_BUTTON* pButtons, UINT cButtons, int nDefaultButton = 0)
- { m_tdc.SetButtons(pButtons, cButtons, nDefaultButton); }
- void SetDefaultButton(int nDefaultButton)
- { m_tdc.SetDefaultButton(nDefaultButton); }
- // radio buttons
- void SetRadioButtons(const TASKDIALOG_BUTTON* pRadioButtons, UINT cRadioButtons, int nDefaultRadioButton = 0)
- { m_tdc.SetRadioButtons(pRadioButtons, cRadioButtons, nDefaultRadioButton); }
- void SetDefaultRadioButton(int nDefaultRadioButton)
- { m_tdc.SetDefaultRadioButton(nDefaultRadioButton); }
- // verification text
- void SetVerificationText(UINT nID)
- { m_tdc.SetVerificationText(nID); }
- void SetVerificationText(LPCWSTR lpstrVerificationText)
- { m_tdc.SetVerificationText(lpstrVerificationText); }
- // expanded information text
- void SetExpandedInformationText(UINT nID)
- { m_tdc.SetExpandedInformationText(nID); }
- void SetExpandedInformationText(LPCWSTR lpstrExpandedInformation)
- { m_tdc.SetExpandedInformationText(lpstrExpandedInformation); }
- // expanded control text
- void SetExpandedControlText(UINT nID)
- { m_tdc.SetExpandedControlText(nID); }
- void SetExpandedControlText(LPCWSTR lpstrExpandedControlText)
- { m_tdc.SetExpandedControlText(lpstrExpandedControlText); }
- // collapsed control text
- void SetCollapsedControlText(UINT nID)
- { m_tdc.SetCollapsedControlText(nID); }
- void SetCollapsedControlText(LPCWSTR lpstrCollapsedControlText)
- { m_tdc.SetCollapsedControlText(lpstrCollapsedControlText); }
- // footer icon
- void SetFooterIcon(HICON hIcon)
- { m_tdc.SetFooterIcon(hIcon); }
- void SetFooterIcon(UINT nID)
- { m_tdc.SetFooterIcon(nID); }
- void SetFooterIcon(LPCWSTR lpstrFooterIcon)
- { m_tdc.SetFooterIcon(lpstrFooterIcon); }
- // footer text
- void SetFooterText(UINT nID)
- { m_tdc.SetFooterText(nID); }
- void SetFooterText(LPCWSTR lpstrFooterText)
- { m_tdc.SetFooterText(lpstrFooterText); }
- // width (in DLUs)
- void SetWidth(UINT cxWidth)
- { m_tdc.SetWidth(cxWidth); }
- // modify flags
- void ModifyFlags(DWORD dwRemove, DWORD dwAdd)
- { m_tdc.ModifyFlags(dwRemove, dwAdd); }
- // Implementation
- static HRESULT CALLBACK TaskDialogCallback(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam, LONG_PTR lpRefData)
- {
- T* pT = (T*)lpRefData;
- ATLASSERT((pT->m_hWnd == NULL) || (pT->m_hWnd == hWnd));
- BOOL bRet = FALSE;
- switch(uMsg)
- {
- case TDN_DIALOG_CONSTRUCTED:
- pT->m_hWnd = hWnd;
- pT->OnDialogConstructed();
- break;
- case TDN_CREATED:
- pT->OnCreated();
- break;
- case TDN_BUTTON_CLICKED:
- bRet = pT->OnButtonClicked((int)wParam);
- break;
- case TDN_RADIO_BUTTON_CLICKED:
- pT->OnRadioButtonClicked((int)wParam);
- break;
- case TDN_HYPERLINK_CLICKED:
- pT->OnHyperlinkClicked((LPCWSTR)lParam);
- break;
- case TDN_EXPANDO_BUTTON_CLICKED:
- pT->OnExpandoButtonClicked((wParam != 0));
- break;
- case TDN_VERIFICATION_CLICKED:
- pT->OnVerificationClicked((wParam != 0));
- break;
- case TDN_HELP:
- pT->OnHelp();
- break;
- case TDN_TIMER:
- bRet = pT->OnTimer((DWORD)wParam);
- break;
- case TDN_NAVIGATED:
- pT->OnNavigated();
- break;
- case TDN_DESTROYED:
- pT->OnDestroyed();
- pT->m_hWnd = NULL;
- break;
- default:
- ATLTRACE2(atlTraceUI, 0, _T("Unknown notification received in CTaskDialogImpl::TaskDialogCallback\n"));
- break;
- }
- return (bRet != FALSE) ? S_OK : S_FALSE;
- }
- // Overrideables - notification handlers
- void OnDialogConstructed()
- {
- }
- void OnCreated()
- {
- }
- BOOL OnButtonClicked(int /*nButton*/)
- {
- return FALSE; // don't prevent dialog to close
- }
- void OnRadioButtonClicked(int /*nRadioButton*/)
- {
- }
- void OnHyperlinkClicked(LPCWSTR /*pszHREF*/)
- {
- }
- void OnExpandoButtonClicked(bool /*bExpanded*/)
- {
- }
- void OnVerificationClicked(bool /*bChecked*/)
- {
- }
- void OnHelp()
- {
- }
- BOOL OnTimer(DWORD /*dwTickCount*/)
- {
- return FALSE; // don't reset counter
- }
- void OnNavigated()
- {
- }
- void OnDestroyed()
- {
- }
- // Commands - valid to call only from handlers
- void NavigatePage(TASKDIALOGCONFIG& tdc)
- {
- ATLASSERT(m_hWnd != NULL);
- tdc.cbSize = sizeof(TASKDIALOGCONFIG);
- if(tdc.hwndParent == NULL)
- tdc.hwndParent = m_tdc.hwndParent;
- tdc.pfCallback = m_tdc.pfCallback;
- tdc.lpCallbackData = m_tdc.lpCallbackData;
- (TASKDIALOGCONFIG)m_tdc = tdc;
- ::SendMessage(m_hWnd, TDM_NAVIGATE_PAGE, 0, (LPARAM)&tdc);
- }
- // modify TASKDIALOGCONFIG values, then call this to update task dialog
- void NavigatePage()
- {
- ATLASSERT(m_hWnd != NULL);
- ::SendMessage(m_hWnd, TDM_NAVIGATE_PAGE, 0, (LPARAM)&m_tdc);
- }
- void ClickButton(int nButton)
- {
- ATLASSERT(m_hWnd != NULL);
- ::SendMessage(m_hWnd, TDM_CLICK_BUTTON, nButton, 0L);
- }
- void SetMarqueeProgressBar(BOOL bMarquee)
- {
- ATLASSERT(m_hWnd != NULL);
- ::SendMessage(m_hWnd, TDM_SET_MARQUEE_PROGRESS_BAR, bMarquee, 0L);
- }
- BOOL SetProgressBarState(int nNewState)
- {
- ATLASSERT(m_hWnd != NULL);
- return (BOOL)::SendMessage(m_hWnd, TDM_SET_PROGRESS_BAR_STATE, nNewState, 0L);
- }
- DWORD SetProgressBarRange(int nMinRange, int nMaxRange)
- {
- ATLASSERT(m_hWnd != NULL);
- return (DWORD)::SendMessage(m_hWnd, TDM_SET_PROGRESS_BAR_RANGE, 0, MAKELPARAM(nMinRange, nMaxRange));
- }
- int SetProgressBarPos(int nNewPos)
- {
- ATLASSERT(m_hWnd != NULL);
- return (int)::SendMessage(m_hWnd, TDM_SET_PROGRESS_BAR_POS, nNewPos, 0L);
- }
- BOOL SetProgressBarMarquee(BOOL bMarquee, UINT uSpeed)
- {
- ATLASSERT(m_hWnd != NULL);
- return (BOOL)::SendMessage(m_hWnd, TDM_SET_PROGRESS_BAR_MARQUEE, bMarquee, uSpeed);
- }
- void SetElementText(TASKDIALOG_ELEMENTS element, LPCWSTR lpstrText)
- {
- ATLASSERT(m_hWnd != NULL);
- ::SendMessage(m_hWnd, TDM_SET_ELEMENT_TEXT, element, (LPARAM)lpstrText);
- }
- void ClickRadioButton(int nRadioButton)
- {
- ATLASSERT(m_hWnd != NULL);
- ::SendMessage(m_hWnd, TDM_CLICK_RADIO_BUTTON, nRadioButton, 0L);
- }
- void EnableButton(int nButton, BOOL bEnable)
- {
- ATLASSERT(m_hWnd != NULL);
- ::SendMessage(m_hWnd, TDM_ENABLE_BUTTON, nButton, bEnable);
- }
- void EnableRadioButton(int nButton, BOOL bEnable)
- {
- ATLASSERT(m_hWnd != NULL);
- ::SendMessage(m_hWnd, TDM_ENABLE_RADIO_BUTTON, nButton, bEnable);
- }
- void ClickVerification(BOOL bCheck, BOOL bFocus)
- {
- ATLASSERT(m_hWnd != NULL);
- ::SendMessage(m_hWnd, TDM_CLICK_VERIFICATION, bCheck, bFocus);
- }
- void UpdateElementText(TASKDIALOG_ELEMENTS element, LPCWSTR lpstrText)
- {
- ATLASSERT(m_hWnd != NULL);
- ::SendMessage(m_hWnd, TDM_UPDATE_ELEMENT_TEXT, element, (LPARAM)lpstrText);
- }
- void SetButtonElevationRequiredState(int nButton, BOOL bElevation)
- {
- ATLASSERT(m_hWnd != NULL);
- ::SendMessage(m_hWnd, TDM_SET_BUTTON_ELEVATION_REQUIRED_STATE, nButton, bElevation);
- }
- void UpdateIcon(TASKDIALOG_ICON_ELEMENTS element, HICON hIcon)
- {
- ATLASSERT(m_hWnd != NULL);
- #ifdef _DEBUG
- if(element == TDIE_ICON_MAIN)
- ATLASSERT((m_tdc.dwFlags & TDF_USE_HICON_MAIN) != 0);
- else if(element == TDIE_ICON_FOOTER)
- ATLASSERT((m_tdc.dwFlags & TDF_USE_HICON_FOOTER) != 0);
- #endif // _DEBUG
- ::SendMessage(m_hWnd, TDM_UPDATE_ICON, element, (LPARAM)hIcon);
- }
- void UpdateIcon(TASKDIALOG_ICON_ELEMENTS element, LPCWSTR lpstrIcon)
- {
- ATLASSERT(m_hWnd != NULL);
- #ifdef _DEBUG
- if(element == TDIE_ICON_MAIN)
- ATLASSERT((m_tdc.dwFlags & TDF_USE_HICON_MAIN) == 0);
- else if(element == TDIE_ICON_FOOTER)
- ATLASSERT((m_tdc.dwFlags & TDF_USE_HICON_FOOTER) == 0);
- #endif // _DEBUG
- ::SendMessage(m_hWnd, TDM_UPDATE_ICON, element, (LPARAM)lpstrIcon);
- }
- };
- ///////////////////////////////////////////////////////////////////////////////
- // CTaskDialog - for non-customized task dialogs
- class CTaskDialog : public CTaskDialogImpl<CTaskDialog>
- {
- public:
- CTaskDialog(HWND hWndParent = NULL) : CTaskDialogImpl<CTaskDialog>(hWndParent)
- {
- m_tdc.pfCallback = NULL;
- }
- };
- #endif // (_WIN32_WINNT >= 0x0600) || defined(_WTL_TASKDIALOG)
- } // namespace WTL
- #endif // __ATLDLGS_H__
|