ccCommandLineCommands.cpp 226 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284328532863287328832893290329132923293329432953296329732983299330033013302330333043305330633073308330933103311331233133314331533163317331833193320332133223323332433253326332733283329333033313332333333343335333633373338333933403341334233433344334533463347334833493350335133523353335433553356335733583359336033613362336333643365336633673368336933703371337233733374337533763377337833793380338133823383338433853386338733883389339033913392339333943395339633973398339934003401340234033404340534063407340834093410341134123413341434153416341734183419342034213422342334243425342634273428342934303431343234333434343534363437343834393440344134423443344434453446344734483449345034513452345334543455345634573458345934603461346234633464346534663467346834693470347134723473347434753476347734783479348034813482348334843485348634873488348934903491349234933494349534963497349834993500350135023503350435053506350735083509351035113512351335143515351635173518351935203521352235233524352535263527352835293530353135323533353435353536353735383539354035413542354335443545354635473548354935503551355235533554355535563557355835593560356135623563356435653566356735683569357035713572357335743575357635773578357935803581358235833584358535863587358835893590359135923593359435953596359735983599360036013602360336043605360636073608360936103611361236133614361536163617361836193620362136223623362436253626362736283629363036313632363336343635363636373638363936403641364236433644364536463647364836493650365136523653365436553656365736583659366036613662366336643665366636673668366936703671367236733674367536763677367836793680368136823683368436853686368736883689369036913692369336943695369636973698369937003701370237033704370537063707370837093710371137123713371437153716371737183719372037213722372337243725372637273728372937303731373237333734373537363737373837393740374137423743374437453746374737483749375037513752375337543755375637573758375937603761376237633764376537663767376837693770377137723773377437753776377737783779378037813782378337843785378637873788378937903791379237933794379537963797379837993800380138023803380438053806380738083809381038113812381338143815381638173818381938203821382238233824382538263827382838293830383138323833383438353836383738383839384038413842384338443845384638473848384938503851385238533854385538563857385838593860386138623863386438653866386738683869387038713872387338743875387638773878387938803881388238833884388538863887388838893890389138923893389438953896389738983899390039013902390339043905390639073908390939103911391239133914391539163917391839193920392139223923392439253926392739283929393039313932393339343935393639373938393939403941394239433944394539463947394839493950395139523953395439553956395739583959396039613962396339643965396639673968396939703971397239733974397539763977397839793980398139823983398439853986398739883989399039913992399339943995399639973998399940004001400240034004400540064007400840094010401140124013401440154016401740184019402040214022402340244025402640274028402940304031403240334034403540364037403840394040404140424043404440454046404740484049405040514052405340544055405640574058405940604061406240634064406540664067406840694070407140724073407440754076407740784079408040814082408340844085408640874088408940904091409240934094409540964097409840994100410141024103410441054106410741084109411041114112411341144115411641174118411941204121412241234124412541264127412841294130413141324133413441354136413741384139414041414142414341444145414641474148414941504151415241534154415541564157415841594160416141624163416441654166416741684169417041714172417341744175417641774178417941804181418241834184418541864187418841894190419141924193419441954196419741984199420042014202420342044205420642074208420942104211421242134214421542164217421842194220422142224223422442254226422742284229423042314232423342344235423642374238423942404241424242434244424542464247424842494250425142524253425442554256425742584259426042614262426342644265426642674268426942704271427242734274427542764277427842794280428142824283428442854286428742884289429042914292429342944295429642974298429943004301430243034304430543064307430843094310431143124313431443154316431743184319432043214322432343244325432643274328432943304331433243334334433543364337433843394340434143424343434443454346434743484349435043514352435343544355435643574358435943604361436243634364436543664367436843694370437143724373437443754376437743784379438043814382438343844385438643874388438943904391439243934394439543964397439843994400440144024403440444054406440744084409441044114412441344144415441644174418441944204421442244234424442544264427442844294430443144324433443444354436443744384439444044414442444344444445444644474448444944504451445244534454445544564457445844594460446144624463446444654466446744684469447044714472447344744475447644774478447944804481448244834484448544864487448844894490449144924493449444954496449744984499450045014502450345044505450645074508450945104511451245134514451545164517451845194520452145224523452445254526452745284529453045314532453345344535453645374538453945404541454245434544454545464547454845494550455145524553455445554556455745584559456045614562456345644565456645674568456945704571457245734574457545764577457845794580458145824583458445854586458745884589459045914592459345944595459645974598459946004601460246034604460546064607460846094610461146124613461446154616461746184619462046214622462346244625462646274628462946304631463246334634463546364637463846394640464146424643464446454646464746484649465046514652465346544655465646574658465946604661466246634664466546664667466846694670467146724673467446754676467746784679468046814682468346844685468646874688468946904691469246934694469546964697469846994700470147024703470447054706470747084709471047114712471347144715471647174718471947204721472247234724472547264727472847294730473147324733473447354736473747384739474047414742474347444745474647474748474947504751475247534754475547564757475847594760476147624763476447654766476747684769477047714772477347744775477647774778477947804781478247834784478547864787478847894790479147924793479447954796479747984799480048014802480348044805480648074808480948104811481248134814481548164817481848194820482148224823482448254826482748284829483048314832483348344835483648374838483948404841484248434844484548464847484848494850485148524853485448554856485748584859486048614862486348644865486648674868486948704871487248734874487548764877487848794880488148824883488448854886488748884889489048914892489348944895489648974898489949004901490249034904490549064907490849094910491149124913491449154916491749184919492049214922492349244925492649274928492949304931493249334934493549364937493849394940494149424943494449454946494749484949495049514952495349544955495649574958495949604961496249634964496549664967496849694970497149724973497449754976497749784979498049814982498349844985498649874988498949904991499249934994499549964997499849995000500150025003500450055006500750085009501050115012501350145015501650175018501950205021502250235024502550265027502850295030503150325033503450355036503750385039504050415042504350445045504650475048504950505051505250535054505550565057505850595060506150625063506450655066506750685069507050715072507350745075507650775078507950805081508250835084508550865087508850895090509150925093509450955096509750985099510051015102510351045105510651075108510951105111511251135114511551165117511851195120512151225123512451255126512751285129513051315132513351345135513651375138513951405141514251435144514551465147514851495150515151525153515451555156515751585159516051615162516351645165516651675168516951705171517251735174517551765177517851795180518151825183518451855186518751885189519051915192519351945195519651975198519952005201520252035204520552065207520852095210521152125213521452155216521752185219522052215222522352245225522652275228522952305231523252335234523552365237523852395240524152425243524452455246524752485249525052515252525352545255525652575258525952605261526252635264526552665267526852695270527152725273527452755276527752785279528052815282528352845285528652875288528952905291529252935294529552965297529852995300530153025303530453055306530753085309531053115312531353145315531653175318531953205321532253235324532553265327532853295330533153325333533453355336533753385339534053415342534353445345534653475348534953505351535253535354535553565357535853595360536153625363536453655366536753685369537053715372537353745375537653775378537953805381538253835384538553865387538853895390539153925393539453955396539753985399540054015402540354045405540654075408540954105411541254135414541554165417541854195420542154225423542454255426542754285429543054315432543354345435543654375438543954405441544254435444544554465447544854495450545154525453545454555456545754585459546054615462546354645465546654675468546954705471547254735474547554765477547854795480548154825483548454855486548754885489549054915492549354945495549654975498549955005501550255035504550555065507550855095510551155125513551455155516551755185519552055215522552355245525552655275528552955305531553255335534553555365537553855395540554155425543554455455546554755485549555055515552555355545555555655575558555955605561556255635564556555665567556855695570557155725573557455755576557755785579558055815582558355845585558655875588558955905591559255935594559555965597559855995600560156025603560456055606560756085609561056115612561356145615561656175618561956205621562256235624562556265627562856295630563156325633563456355636563756385639564056415642564356445645564656475648564956505651565256535654565556565657565856595660566156625663566456655666566756685669567056715672567356745675567656775678567956805681568256835684568556865687568856895690569156925693569456955696569756985699570057015702570357045705570657075708570957105711571257135714571557165717571857195720572157225723572457255726572757285729573057315732573357345735573657375738573957405741574257435744574557465747574857495750575157525753575457555756575757585759576057615762576357645765576657675768576957705771577257735774577557765777577857795780578157825783578457855786578757885789579057915792579357945795579657975798579958005801580258035804580558065807580858095810581158125813581458155816581758185819582058215822582358245825582658275828582958305831583258335834583558365837583858395840584158425843584458455846584758485849585058515852585358545855585658575858585958605861586258635864586558665867586858695870587158725873587458755876587758785879588058815882588358845885588658875888588958905891589258935894589558965897589858995900590159025903590459055906590759085909591059115912591359145915591659175918591959205921592259235924592559265927592859295930593159325933593459355936593759385939594059415942594359445945594659475948594959505951595259535954595559565957595859595960596159625963596459655966596759685969597059715972597359745975597659775978597959805981598259835984598559865987598859895990599159925993599459955996599759985999600060016002600360046005600660076008600960106011601260136014601560166017601860196020602160226023602460256026602760286029603060316032603360346035603660376038603960406041604260436044604560466047604860496050605160526053605460556056605760586059606060616062606360646065606660676068606960706071607260736074607560766077607860796080608160826083608460856086608760886089609060916092609360946095609660976098609961006101610261036104610561066107610861096110611161126113611461156116611761186119612061216122612361246125612661276128612961306131613261336134613561366137613861396140614161426143614461456146614761486149615061516152615361546155615661576158615961606161616261636164616561666167616861696170617161726173617461756176617761786179618061816182618361846185618661876188618961906191619261936194619561966197619861996200620162026203620462056206620762086209621062116212621362146215621662176218621962206221622262236224622562266227622862296230623162326233623462356236623762386239624062416242624362446245624662476248624962506251625262536254625562566257625862596260626162626263626462656266626762686269627062716272627362746275627662776278627962806281628262836284628562866287628862896290629162926293629462956296629762986299630063016302630363046305630663076308630963106311631263136314631563166317631863196320632163226323632463256326632763286329633063316332633363346335633663376338633963406341634263436344634563466347634863496350635163526353635463556356635763586359636063616362636363646365636663676368636963706371637263736374637563766377637863796380638163826383638463856386638763886389639063916392639363946395639663976398639964006401640264036404640564066407640864096410641164126413641464156416641764186419642064216422642364246425642664276428642964306431643264336434643564366437643864396440644164426443644464456446644764486449645064516452645364546455645664576458645964606461646264636464646564666467646864696470647164726473647464756476647764786479648064816482648364846485648664876488648964906491649264936494649564966497649864996500650165026503650465056506650765086509651065116512651365146515651665176518651965206521652265236524652565266527652865296530653165326533653465356536653765386539654065416542654365446545654665476548654965506551655265536554655565566557655865596560656165626563656465656566656765686569657065716572657365746575657665776578657965806581658265836584658565866587658865896590659165926593659465956596659765986599660066016602660366046605660666076608660966106611661266136614661566166617661866196620662166226623662466256626662766286629663066316632663366346635663666376638663966406641664266436644664566466647664866496650665166526653665466556656665766586659666066616662666366646665666666676668666966706671667266736674667566766677667866796680668166826683668466856686668766886689669066916692669366946695669666976698669967006701670267036704670567066707670867096710671167126713671467156716671767186719672067216722672367246725672667276728672967306731673267336734673567366737673867396740674167426743674467456746674767486749675067516752675367546755675667576758675967606761676267636764676567666767676867696770677167726773677467756776677767786779678067816782678367846785678667876788678967906791679267936794679567966797679867996800680168026803680468056806680768086809681068116812681368146815681668176818681968206821682268236824682568266827682868296830683168326833683468356836683768386839684068416842684368446845684668476848684968506851685268536854685568566857685868596860686168626863686468656866686768686869687068716872687368746875687668776878687968806881688268836884688568866887688868896890689168926893689468956896689768986899690069016902690369046905690669076908690969106911691269136914691569166917691869196920692169226923692469256926692769286929693069316932693369346935693669376938693969406941694269436944694569466947694869496950695169526953695469556956695769586959696069616962696369646965696669676968696969706971697269736974697569766977697869796980698169826983698469856986698769886989699069916992699369946995699669976998699970007001700270037004700570067007700870097010701170127013701470157016701770187019702070217022702370247025702670277028702970307031703270337034703570367037703870397040704170427043704470457046704770487049705070517052705370547055705670577058705970607061706270637064706570667067706870697070707170727073707470757076707770787079708070817082708370847085708670877088708970907091709270937094709570967097709870997100710171027103710471057106710771087109711071117112711371147115711671177118711971207121712271237124712571267127712871297130713171327133713471357136713771387139714071417142714371447145714671477148714971507151715271537154715571567157715871597160716171627163716471657166716771687169717071717172717371747175717671777178717971807181718271837184718571867187718871897190719171927193719471957196719771987199720072017202720372047205720672077208720972107211721272137214721572167217721872197220722172227223722472257226722772287229723072317232723372347235723672377238723972407241724272437244724572467247724872497250725172527253725472557256725772587259726072617262726372647265726672677268726972707271727272737274727572767277727872797280728172827283728472857286728772887289729072917292729372947295729672977298729973007301730273037304730573067307730873097310731173127313731473157316731773187319732073217322732373247325732673277328732973307331733273337334733573367337733873397340734173427343734473457346734773487349735073517352735373547355735673577358735973607361736273637364736573667367736873697370737173727373737473757376737773787379738073817382738373847385738673877388738973907391739273937394739573967397739873997400740174027403740474057406740774087409741074117412741374147415741674177418741974207421742274237424742574267427742874297430743174327433743474357436743774387439744074417442744374447445744674477448744974507451745274537454745574567457745874597460746174627463746474657466746774687469747074717472747374747475747674777478747974807481748274837484748574867487748874897490749174927493749474957496749774987499750075017502750375047505750675077508750975107511751275137514751575167517751875197520752175227523752475257526752775287529753075317532753375347535753675377538753975407541754275437544754575467547754875497550755175527553755475557556755775587559756075617562756375647565756675677568756975707571757275737574757575767577757875797580758175827583758475857586758775887589759075917592759375947595759675977598759976007601760276037604760576067607760876097610761176127613761476157616761776187619762076217622762376247625762676277628762976307631763276337634763576367637763876397640764176427643764476457646764776487649765076517652765376547655765676577658765976607661766276637664766576667667766876697670767176727673767476757676767776787679768076817682768376847685768676877688768976907691769276937694769576967697769876997700770177027703770477057706770777087709771077117712771377147715771677177718771977207721772277237724772577267727772877297730773177327733773477357736773777387739774077417742774377447745774677477748774977507751775277537754775577567757775877597760776177627763776477657766776777687769777077717772777377747775777677777778777977807781778277837784778577867787778877897790779177927793779477957796779777987799780078017802780378047805780678077808780978107811781278137814781578167817781878197820782178227823782478257826782778287829783078317832783378347835783678377838783978407841784278437844784578467847784878497850785178527853785478557856785778587859786078617862786378647865786678677868786978707871787278737874787578767877787878797880788178827883788478857886788778887889789078917892789378947895789678977898789979007901790279037904790579067907790879097910791179127913791479157916791779187919792079217922792379247925792679277928792979307931793279337934793579367937793879397940794179427943794479457946794779487949795079517952795379547955795679577958795979607961796279637964796579667967796879697970
  1. //CCCoreLib
  2. #include <AutoSegmentationTools.h>
  3. #include <CCConst.h>
  4. #include <CloudSamplingTools.h>
  5. #include <MeshSamplingTools.h>
  6. #include <NormalDistribution.h>
  7. #include <StatisticalTestingTools.h>
  8. #include <WeibullDistribution.h>
  9. #include <DistanceComputationTools.h>
  10. //qCC_db
  11. #include <ccHObjectCaster.h>
  12. #include <ccNormalVectors.h>
  13. #include <ccPlane.h>
  14. #include <ccPolyline.h>
  15. #include <ccProgressDialog.h>
  16. #include <ccScalarField.h>
  17. #include <ccVolumeCalcTool.h>
  18. #include <ccSubMesh.h>
  19. #include <ccPointCloudInterpolator.h>
  20. //qCC_io
  21. #include <AsciiFilter.h>
  22. #include <PlyFilter.h>
  23. //qCC
  24. #include "ccCommon.h"
  25. #include "ccComparisonDlg.h"
  26. #include "ccConsole.h"
  27. #include "ccCropTool.h"
  28. #include "ccLibAlgorithms.h"
  29. #include "ccRegistrationTools.h"
  30. #include "ccScalarFieldArithmeticsDlg.h"
  31. #include "ccColorLevelsDlg.h"
  32. //Qt
  33. #include "ccCommandLineCommands.h"
  34. //Local
  35. #include "ccEntityAction.h"
  36. #include <QDateTime>
  37. #include <QFileInfo>
  38. //commands
  39. constexpr char COMMAND_CLOUD_EXPORT_FORMAT[] = "C_EXPORT_FMT";
  40. constexpr char COMMAND_EXPORT_EXTENSION[] = "EXT";
  41. constexpr char COMMAND_ASCII_EXPORT_PRECISION[] = "PREC";
  42. constexpr char COMMAND_ASCII_EXPORT_SEPARATOR[] = "SEP";
  43. constexpr char COMMAND_ASCII_EXPORT_ADD_COL_HEADER[] = "ADD_HEADER";
  44. constexpr char COMMAND_ASCII_EXPORT_ADD_PTS_COUNT[] = "ADD_PTS_COUNT";
  45. constexpr char COMMAND_MESH_EXPORT_FORMAT[] = "M_EXPORT_FMT";
  46. constexpr char COMMAND_HIERARCHY_EXPORT_FORMAT[] = "H_EXPORT_FMT";
  47. constexpr char COMMAND_OPEN[] = "O"; //+ file name
  48. constexpr char COMMAND_OPEN_SKIP_LINES[] = "SKIP"; //+ number of lines to skip
  49. constexpr char COMMAND_OPEN_NO_LABEL[] = "NO_LABEL";
  50. constexpr char COMMAND_COMMAND_FILE[] = "COMMAND_FILE"; //+ file name
  51. constexpr char COMMAND_SUBSAMPLE[] = "SS"; //+ method (RANDOM/SPATIAL/OCTREE) + parameter (resp. point count / spatial step / octree level)
  52. constexpr char COMMAND_EXTRACT_CC[] = "EXTRACT_CC";
  53. constexpr char COMMAND_CURVATURE[] = "CURV"; //+ curvature type (MEAN/GAUSS)
  54. constexpr char COMMAND_DENSITY[] = "DENSITY"; //+ sphere radius
  55. constexpr char COMMAND_DENSITY_TYPE[] = "TYPE"; //+ density type
  56. constexpr char COMMAND_APPROX_DENSITY[] = "APPROX_DENSITY";
  57. constexpr char COMMAND_SF_GRADIENT[] = "SF_GRAD";
  58. constexpr char COMMAND_ROUGHNESS[] = "ROUGH";
  59. constexpr char COMMAND_ROUGHNESS_UP_DIR[] = "UP_DIR";
  60. constexpr char COMMAND_APPLY_TRANSFORMATION[] = "APPLY_TRANS";
  61. constexpr char COMMAND_APPLY_TRANS_TO_GLOBAL[] = "APPLY_TO_GLOBAL";
  62. constexpr char COMMAND_APPLY_TRANS_INVERSE[] = "INVERSE";
  63. constexpr char COMMAND_DROP_GLOBAL_SHIFT[] = "DROP_GLOBAL_SHIFT";
  64. constexpr char COMMAND_SF_COLOR_SCALE[] = "SF_COLOR_SCALE";
  65. constexpr char COMMAND_SF_CONVERT_TO_RGB[] = "SF_CONVERT_TO_RGB";
  66. constexpr char COMMAND_FILTER_SF_BY_VALUE[] = "FILTER_SF";
  67. constexpr char COMMAND_MERGE_CLOUDS[] = "MERGE_CLOUDS";
  68. constexpr char COMMAND_MERGE_MESHES[] = "MERGE_MESHES";
  69. constexpr char COMMAND_SET_ACTIVE_SF[] = "SET_ACTIVE_SF";
  70. constexpr char COMMAND_SET_GLOBAL_SHIFT[] = "SET_GLOBAL_SHIFT"; // + global shift {x,y,z}
  71. constexpr char COMMAND_SET_GLOBAL_SHIFT_KEEP_ORIG_FIXED[]= "KEEP_ORIG_FIXED";
  72. constexpr char COMMAND_REMOVE_ALL_SFS[] = "REMOVE_ALL_SFS";
  73. constexpr char COMMAND_REMOVE_SF[] = "REMOVE_SF";
  74. constexpr char COMMAND_REMOVE_SCAN_GRIDS[] = "REMOVE_SCAN_GRIDS";
  75. constexpr char COMMAND_REMOVE_SENSORS[] = "REMOVE_SENSORS";
  76. constexpr char COMMAND_REMOVE_RGB[] = "REMOVE_RGB";
  77. constexpr char COMMAND_REMOVE_NORMALS[] = "REMOVE_NORMALS";
  78. constexpr char COMMAND_MATCH_BB_CENTERS[] = "MATCH_CENTERS";
  79. constexpr char COMMAND_BEST_FIT_PLANE[] = "BEST_FIT_PLANE";
  80. constexpr char COMMAND_BEST_FIT_PLANE_MAKE_HORIZ[] = "MAKE_HORIZ";
  81. constexpr char COMMAND_BEST_FIT_PLANE_KEEP_LOADED[] = "KEEP_LOADED";
  82. constexpr char COMMAND_ORIENT_NORMALS[] = "ORIENT_NORMS_MST";
  83. constexpr char COMMAND_SOR_FILTER[] = "SOR";
  84. constexpr char COMMAND_NOISE_FILTER[] = "NOISE";
  85. constexpr char COMMAND_NOISE_FILTER_KNN[] = "KNN";
  86. constexpr char COMMAND_NOISE_FILTER_RADIUS[] = "RADIUS";
  87. constexpr char COMMAND_NOISE_FILTER_REL[] = "REL";
  88. constexpr char COMMAND_NOISE_FILTER_ABS[] = "ABS";
  89. constexpr char COMMAND_NOISE_FILTER_RIP[] = "RIP";
  90. constexpr char COMMAND_REMOVE_DUPLICATE_POINTS[] = "RDP";
  91. constexpr char COMMAND_SAMPLE_MESH[] = "SAMPLE_MESH";
  92. constexpr char COMMAND_COMPRESS_FWF[] = "COMPRESS_FWF";
  93. constexpr char COMMAND_CROP[] = "CROP";
  94. constexpr char COMMAND_CROP_OUTSIDE[] = "OUTSIDE";
  95. constexpr char COMMAND_CROP_2D[] = "CROP2D";
  96. constexpr char COMMAND_COLOR_BANDING[] = "CBANDING";
  97. constexpr char COMMAND_COLOR_LEVELS[] = "CLEVELS";
  98. constexpr char COMMAND_C2M_DIST[] = "C2M_DIST";
  99. constexpr char COMMAND_C2M_DIST_FLIP_NORMALS[] = "FLIP_NORMS";
  100. constexpr char COMMAND_C2M_DIST_UNSIGNED[] = "UNSIGNED";
  101. constexpr char COMMAND_C2M_DIST_NON_ROBUST[] = "NON_ROBUST";
  102. constexpr char COMMAND_C2M_NORMAL_MATCHING[] = "NORMAL_MATCH";
  103. constexpr char COMMAND_C2C_DIST[] = "C2C_DIST";
  104. constexpr char COMMAND_CLOSEST_POINT_SET[] = "CLOSEST_POINT_SET";
  105. constexpr char COMMAND_C2C_SPLIT_XYZ[] = "SPLIT_XYZ";
  106. constexpr char COMMAND_C2C_SPLIT_XY_Z[] = "SPLIT_XY_Z";
  107. constexpr char COMMAND_C2C_LOCAL_MODEL[] = "MODEL";
  108. constexpr char COMMAND_C2X_MAX_DISTANCE[] = "MAX_DIST";
  109. constexpr char COMMAND_C2X_OCTREE_LEVEL[] = "OCTREE_LEVEL";
  110. constexpr char COMMAND_STAT_TEST[] = "STAT_TEST";
  111. constexpr char COMMAND_DELAUNAY[] = "DELAUNAY";
  112. constexpr char COMMAND_DELAUNAY_AA[] = "AA";
  113. constexpr char COMMAND_DELAUNAY_BF[] = "BEST_FIT";
  114. constexpr char COMMAND_DELAUNAY_MAX_EDGE_LENGTH[] = "MAX_EDGE_LENGTH";
  115. constexpr char COMMAND_SF_ARITHMETIC[] = "SF_ARITHMETIC";
  116. constexpr char COMMAND_SF_ARITHMETIC_IN_PLACE[] = "IN_PLACE";
  117. constexpr char COMMAND_SF_OP[] = "SF_OP";
  118. constexpr char COMMAND_SF_OP_NOT_IN_PLACE[] = "NOT_IN_PLACE";
  119. constexpr char COMMAND_SF_OP_SF[] = "SF_OP_SF";
  120. constexpr char COMMAND_SF_INTERP[] = "SF_INTERP";
  121. constexpr char COMMAND_COLOR_INTERP[] = "COLOR_INTERP";
  122. constexpr char COMMAND_SF_INTERP_DEST_IS_FIRST[] = "DEST_IS_FIRST";
  123. constexpr char COMMAND_SF_ADD_CONST[] = "SF_ADD_CONST";
  124. constexpr char COMMAND_SF_ADD_ID[] = "SF_ADD_ID";
  125. constexpr char COMMAND_SF_ADD_ID_AS_INT[] = "AS_INT";
  126. constexpr char COMMAND_RENAME_ENTITIES[] = "RENAME_ENTITIES"; //+ base name
  127. constexpr char COMMAND_RENAME_SF[] = "RENAME_SF";
  128. constexpr char COMMAND_COORD_TO_SF[] = "COORD_TO_SF";
  129. constexpr char COMMAND_SF_TO_COORD[] = "SF_TO_COORD";
  130. constexpr char COMMAND_EXTRACT_VERTICES[] = "EXTRACT_VERTICES";
  131. constexpr char COMMAND_ICP[] = "ICP";
  132. constexpr char COMMAND_ICP_REFERENCE_IS_FIRST[] = "REFERENCE_IS_FIRST";
  133. constexpr char COMMAND_ICP_MIN_ERROR_DIIF[] = "MIN_ERROR_DIFF";
  134. constexpr char COMMAND_ICP_ITERATION_COUNT[] = "ITER";
  135. constexpr char COMMAND_ICP_OVERLAP[] = "OVERLAP";
  136. constexpr char COMMAND_ICP_ADJUST_SCALE[] = "ADJUST_SCALE";
  137. constexpr char COMMAND_ICP_RANDOM_SAMPLING_LIMIT[] = "RANDOM_SAMPLING_LIMIT";
  138. constexpr char COMMAND_ICP_ENABLE_FARTHEST_REMOVAL[] = "FARTHEST_REMOVAL";
  139. constexpr char COMMAND_ICP_USE_MODEL_SF_AS_WEIGHT[] = "MODEL_SF_AS_WEIGHTS";
  140. constexpr char COMMAND_ICP_USE_DATA_SF_AS_WEIGHT[] = "DATA_SF_AS_WEIGHTS";
  141. constexpr char COMMAND_ICP_ROT[] = "ROT";
  142. constexpr char COMMAND_ICP_SKIP_TX[] = "SKIP_TX";
  143. constexpr char COMMAND_ICP_SKIP_TY[] = "SKIP_TY";
  144. constexpr char COMMAND_ICP_SKIP_TZ[] = "SKIP_TZ";
  145. constexpr char COMMAND_ICP_C2M_DIST[] = "USE_C2M_DIST";
  146. constexpr char COMMAND_PLY_EXPORT_FORMAT[] = "PLY_EXPORT_FMT";
  147. constexpr char COMMAND_COMPUTE_GRIDDED_NORMALS[] = "COMPUTE_NORMALS";
  148. constexpr char COMMAND_INVERT_NORMALS[] = "INVERT_NORMALS";
  149. constexpr char COMMAND_COMPUTE_OCTREE_NORMALS[] = "OCTREE_NORMALS";
  150. constexpr char COMMAND_CONVERT_NORMALS_TO_DIP[] = "NORMALS_TO_DIP";
  151. constexpr char COMMAND_CONVERT_NORMALS_TO_SFS[] = "NORMALS_TO_SFS";
  152. constexpr char COMMAND_CONVERT_NORMALS_TO_HSV[] = "NORMALS_TO_HSV";
  153. constexpr char COMMAND_CLEAR_NORMALS[] = "CLEAR_NORMALS";
  154. constexpr char COMMAND_MESH_VOLUME[] = "MESH_VOLUME";
  155. constexpr char COMMAND_VOLUME_TO_FILE[] = "TO_FILE";
  156. constexpr char COMMAND_SAVE_CLOUDS[] = "SAVE_CLOUDS";
  157. constexpr char COMMAND_SAVE_MESHES[] = "SAVE_MESHES";
  158. constexpr char COMMAND_AUTO_SAVE[] = "AUTO_SAVE";
  159. constexpr char COMMAND_LOG_FILE[] = "LOG_FILE";
  160. constexpr char COMMAND_SELECT_ENTITIES[] = "SELECT_ENTITIES";
  161. constexpr char COMMAND_CLEAR[] = "CLEAR";
  162. constexpr char COMMAND_CLEAR_CLOUDS[] = "CLEAR_CLOUDS";
  163. constexpr char COMMAND_POP_CLOUDS[] = "POP_CLOUDS";
  164. constexpr char COMMAND_CLEAR_MESHES[] = "CLEAR_MESHES";
  165. constexpr char COMMAND_POP_MESHES[] = "POP_MESHES";
  166. constexpr char COMMAND_NO_TIMESTAMP[] = "NO_TIMESTAMP";
  167. constexpr char COMMAND_MOMENT[] = "MOMENT";
  168. constexpr char COMMAND_FEATURE[] = "FEATURE";
  169. constexpr char COMMAND_RGB_CONVERT_TO_SF[] = "RGB_CONVERT_TO_SF";
  170. constexpr char COMMAND_FLIP_TRIANGLES[] = "FLIP_TRI";
  171. constexpr char COMMAND_DEBUG[] = "DEBUG";
  172. constexpr char COMMAND_VERBOSITY[] = "VERBOSITY";
  173. constexpr char COMMAND_FILTER[] = "FILTER";
  174. //options / modifiers
  175. constexpr char COMMAND_MAX_THREAD_COUNT[] = "MAX_TCOUNT";
  176. constexpr char OPTION_ALL_AT_ONCE[] = "ALL_AT_ONCE";
  177. constexpr char OPTION_ON[] = "ON";
  178. constexpr char OPTION_OFF[] = "OFF";
  179. constexpr char OPTION_FILE_NAMES[] = "FILE";
  180. constexpr char OPTION_ORIENT[] = "ORIENT";
  181. constexpr char OPTION_MODEL[] = "MODEL";
  182. constexpr char OPTION_FIRST[] = "FIRST";
  183. constexpr char OPTION_LAST[] = "LAST";
  184. constexpr char OPTION_ALL[] = "ALL";
  185. constexpr char OPTION_REGEX[] = "REGEX";
  186. constexpr char OPTION_NOT[] = "NOT";
  187. constexpr char OPTION_CLOUD[] = "CLOUD";
  188. constexpr char OPTION_MESH[] = "MESH";
  189. constexpr char OPTION_PERCENT[] = "PERCENT";
  190. constexpr char OPTION_NUMBER_OF_POINTS[] = "NUMBER_OF_POINTS";
  191. constexpr char OPTION_FORCE[] = "FORCE";
  192. constexpr char OPTION_USE_ACTIVE_SF[] = "USE_ACTIVE_SF";
  193. constexpr char OPTION_SF[] = "SF";
  194. constexpr char OPTION_RGB[] = "RGB";
  195. constexpr char OPTION_GAUSSIAN[] = "GAUSSIAN";
  196. constexpr char OPTION_BILATERAL[] = "BILATERAL";
  197. constexpr char OPTION_MEAN[] = "MEAN";
  198. constexpr char OPTION_MEDIAN[] = "MEDIAN";
  199. constexpr char OPTION_SIGMA[] = "SIGMA";
  200. constexpr char OPTION_SIGMA_SF[] = "SIGMA_SF";
  201. constexpr char OPTION_BURNT_COLOR_THRESHOLD[] = "BURNT_COLOR_THRESHOLD";
  202. constexpr char OPTION_BLEND_GRAYSCALE[] = "BLEND_GRAYSCALE";
  203. static bool GetSFIndexOrName(ccCommandLineInterface& cmd, int& sfIndex, QString& sfName, bool allowMinusOne = false)
  204. {
  205. sfName = cmd.arguments().takeFirst();
  206. if (sfName.toUpper() == OPTION_LAST)
  207. {
  208. sfIndex = -2;
  209. cmd.print(QObject::tr("SF index: LAST"));
  210. }
  211. else
  212. {
  213. bool validInt = false;
  214. sfIndex = sfName.toInt(&validInt);
  215. if (validInt)
  216. {
  217. sfName.clear(); //user has provided an index, not a name
  218. if (sfIndex < 0)
  219. {
  220. if (allowMinusOne && sfIndex == -1)
  221. {
  222. // -1 means 'no active SF'
  223. cmd.print(QObject::tr("SF index: none"));
  224. }
  225. else
  226. {
  227. // invalid index
  228. cmd.warning(QObject::tr("Invalid SF index: %1").arg(sfIndex));
  229. return false;
  230. }
  231. }
  232. else
  233. {
  234. cmd.print(QObject::tr("SF index: %1").arg(sfIndex));
  235. }
  236. }
  237. else
  238. {
  239. cmd.print(QObject::tr("SF name: '%1'").arg(sfName));
  240. sfIndex = -1;
  241. }
  242. }
  243. return true;
  244. }
  245. int GetScalarFieldIndex(ccPointCloud* cloud, int sfIndex, const QString& sfName, bool minusOneMeansCurrent = false)
  246. {
  247. if (!cloud)
  248. {
  249. assert(false);
  250. return -1;
  251. }
  252. else if (!cloud->hasScalarFields())
  253. {
  254. return -1;
  255. }
  256. else if (sfIndex == -2)
  257. {
  258. return static_cast<int>(cloud->getNumberOfScalarFields()) - 1;
  259. }
  260. else if (sfIndex == -1)
  261. {
  262. if (!sfName.isEmpty()) // the user has provided a SF name instead of an index
  263. {
  264. //check if this cloud has a scalar field with the input name
  265. sfIndex = cloud->getScalarFieldIndexByName(sfName.toStdString());
  266. if (sfIndex < 0)
  267. {
  268. ccLog::Warning(QObject::tr("Cloud %1 has no SF named '%2'").arg(cloud->getName()).arg(sfName));
  269. return -1;
  270. }
  271. }
  272. else if (minusOneMeansCurrent)
  273. {
  274. return cloud->getCurrentInScalarFieldIndex();
  275. }
  276. else
  277. {
  278. ccLog::Warning(QObject::tr("Input scalar field index is invalid: %1").arg(sfIndex));
  279. return -1;
  280. }
  281. }
  282. else if (sfIndex >= static_cast<int>(cloud->getNumberOfScalarFields()))
  283. {
  284. ccLog::Warning(QObject::tr("Cloud %1 has less scalar fields than the SF index (%2/%3)").arg(cloud->getName()).arg(sfIndex).arg(cloud->getNumberOfScalarFields()));
  285. return -1;
  286. }
  287. return sfIndex;
  288. }
  289. CCCoreLib::ScalarField* GetScalarField(ccPointCloud* cloud, int sfIndex, const QString& sfName, bool minusOneMeansCurrent = false)
  290. {
  291. sfIndex = GetScalarFieldIndex(cloud, sfIndex, sfName, minusOneMeansCurrent);
  292. if (sfIndex < 0)
  293. {
  294. return nullptr;
  295. }
  296. else
  297. {
  298. return cloud->getScalarField(sfIndex);
  299. }
  300. }
  301. CommandChangeOutputFormat::CommandChangeOutputFormat(const QString& name, const QString& keyword)
  302. : ccCommandLineInterface::Command(name, keyword)
  303. {}
  304. QString CommandChangeOutputFormat::getFileFormatFilter(ccCommandLineInterface& cmd, QString& defaultExt)
  305. {
  306. QString fileFilter;
  307. defaultExt.clear();
  308. if (!cmd.arguments().isEmpty())
  309. {
  310. //test if the specified format corresponds to a known file type
  311. QString extension = cmd.arguments().front();
  312. QString upperExtension = extension.toUpper();
  313. cmd.arguments().pop_front();
  314. const FileIOFilter::FilterContainer& filters = FileIOFilter::GetFilters();
  315. for (const auto& filter : filters)
  316. {
  317. if (upperExtension == filter->getDefaultExtension().toUpper())
  318. {
  319. //found a matching filter
  320. fileFilter = filter->getFileFilters(false).first();
  321. defaultExt = filter->getDefaultExtension();
  322. break;
  323. }
  324. }
  325. //haven't found anything?
  326. if (fileFilter.isEmpty())
  327. {
  328. ccLog::Warning(QString("Extension '%1' was not identified as a primary extension. Let's look for secondary ones...").arg(extension));
  329. // let's look for secondary formats
  330. for (const auto& filter : filters)
  331. {
  332. QStringList outputFilters = filter->getFileFilters(false);
  333. for (const QString& outputFilter : outputFilters)
  334. {
  335. int index = outputFilter.toUpper().indexOf(upperExtension);
  336. if (index >= 2 && outputFilter.mid(index - 2, 2) == "*.")
  337. {
  338. ccLog::Print(QString("Extension '%1' found in the output formats supported by the '%2' filter").arg(extension).arg(filter->getDefaultExtension().toUpper()));
  339. fileFilter = outputFilter;
  340. defaultExt = extension;
  341. break;
  342. }
  343. }
  344. if (!fileFilter.isEmpty())
  345. {
  346. break;
  347. }
  348. }
  349. }
  350. //still haven't found anything?
  351. if (fileFilter.isEmpty())
  352. {
  353. cmd.error(QObject::tr("Unhandled format specifier (%1)").arg(extension));
  354. }
  355. }
  356. else
  357. {
  358. cmd.error(QObject::tr("Missing file format specifier!"));
  359. }
  360. return fileFilter;
  361. }
  362. CommandChangeCloudOutputFormat::CommandChangeCloudOutputFormat()
  363. : CommandChangeOutputFormat(QObject::tr("Change cloud output format"), COMMAND_CLOUD_EXPORT_FORMAT)
  364. {}
  365. bool CommandChangeCloudOutputFormat::process(ccCommandLineInterface& cmd)
  366. {
  367. QString defaultExt;
  368. QString fileFilter = getFileFormatFilter(cmd, defaultExt);
  369. if (fileFilter.isEmpty())
  370. {
  371. return false;
  372. }
  373. cmd.setCloudExportFormat(fileFilter, defaultExt);
  374. cmd.print(QObject::tr("Output export format (clouds) set to: %1").arg(defaultExt.toUpper()));
  375. //default options for ASCII output
  376. if (fileFilter == AsciiFilter::GetFileFilter())
  377. {
  378. AsciiFilter::SetOutputCoordsPrecision(cmd.numericalPrecision());
  379. AsciiFilter::SetOutputSFPrecision(cmd.numericalPrecision());
  380. AsciiFilter::SetOutputSeparatorIndex(0); //space
  381. AsciiFilter::SaveSFBeforeColor(false); //default order: point, color, SF, normal
  382. AsciiFilter::SaveColumnsNamesHeader(false);
  383. AsciiFilter::SavePointCountHeader(false);
  384. }
  385. //look for additional parameters
  386. while (!cmd.arguments().empty())
  387. {
  388. QString argument = cmd.arguments().front();
  389. if (ccCommandLineInterface::IsCommand(argument, COMMAND_EXPORT_EXTENSION))
  390. {
  391. //local option confirmed, we can move on
  392. cmd.arguments().pop_front();
  393. if (cmd.arguments().empty())
  394. {
  395. return cmd.error(QObject::tr("Missing parameter: extension after '%1'").arg(COMMAND_EXPORT_EXTENSION));
  396. }
  397. cmd.setCloudExportFormat(cmd.cloudExportFormat(), cmd.arguments().takeFirst());
  398. cmd.print(QObject::tr("New output extension for clouds: %1").arg(cmd.cloudExportExt()));
  399. }
  400. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_ASCII_EXPORT_PRECISION))
  401. {
  402. //local option confirmed, we can move on
  403. cmd.arguments().pop_front();
  404. if (cmd.arguments().empty())
  405. {
  406. return cmd.error(QObject::tr("Missing parameter: precision value after '%1'").arg(COMMAND_ASCII_EXPORT_PRECISION));
  407. }
  408. bool ok;
  409. int precision = cmd.arguments().takeFirst().toInt(&ok);
  410. if (!ok || precision < 0)
  411. {
  412. return cmd.error(QObject::tr("Invalid value for precision! (%1)").arg(COMMAND_ASCII_EXPORT_PRECISION));
  413. }
  414. if (fileFilter != AsciiFilter::GetFileFilter())
  415. {
  416. cmd.warning(QObject::tr("Argument '%1' is only applicable to ASCII format!").arg(argument));
  417. }
  418. AsciiFilter::SetOutputCoordsPrecision(precision);
  419. AsciiFilter::SetOutputSFPrecision(precision);
  420. }
  421. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_ASCII_EXPORT_SEPARATOR))
  422. {
  423. //local option confirmed, we can move on
  424. cmd.arguments().pop_front();
  425. if (cmd.arguments().empty())
  426. {
  427. return cmd.error(QObject::tr("Missing parameter: separator character after '%1'").arg(COMMAND_ASCII_EXPORT_SEPARATOR));
  428. }
  429. if (fileFilter != AsciiFilter::GetFileFilter())
  430. {
  431. cmd.warning(QObject::tr("Argument '%1' is only applicable to ASCII format!").arg(argument));
  432. }
  433. QString separatorStr = cmd.arguments().takeFirst().toUpper();
  434. //printf("%s\n",qPrintable(separatorStr));
  435. int index = -1;
  436. if (separatorStr == "SPACE")
  437. {
  438. index = 0;
  439. }
  440. else if (separatorStr == "SEMICOLON")
  441. {
  442. index = 1;
  443. }
  444. else if (separatorStr == "COMMA")
  445. {
  446. index = 2;
  447. }
  448. else if (separatorStr == "TAB")
  449. {
  450. index = 3;
  451. }
  452. else
  453. {
  454. return cmd.error(QObject::tr("Invalid separator! ('%1')").arg(separatorStr));
  455. }
  456. AsciiFilter::SetOutputSeparatorIndex(index);
  457. }
  458. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_ASCII_EXPORT_ADD_COL_HEADER))
  459. {
  460. //local option confirmed, we can move on
  461. cmd.arguments().pop_front();
  462. if (fileFilter != AsciiFilter::GetFileFilter())
  463. {
  464. cmd.warning(QObject::tr("Argument '%1' is only applicable to ASCII format!").arg(argument));
  465. }
  466. AsciiFilter::SaveColumnsNamesHeader(true);
  467. }
  468. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_ASCII_EXPORT_ADD_PTS_COUNT))
  469. {
  470. //local option confirmed, we can move on
  471. cmd.arguments().pop_front();
  472. if (fileFilter != AsciiFilter::GetFileFilter())
  473. {
  474. cmd.warning(QObject::tr("Argument '%1' is only applicable to ASCII format!").arg(argument));
  475. }
  476. AsciiFilter::SavePointCountHeader(true);
  477. }
  478. else
  479. {
  480. break; //as soon as we encounter an unrecognized argument, we break the local loop to go back to the main one!
  481. }
  482. }
  483. return true;
  484. }
  485. CommandChangeMeshOutputFormat::CommandChangeMeshOutputFormat()
  486. : CommandChangeOutputFormat(QObject::tr("Change mesh output format"), COMMAND_MESH_EXPORT_FORMAT)
  487. {}
  488. bool CommandChangeMeshOutputFormat::process(ccCommandLineInterface& cmd)
  489. {
  490. QString defaultExt;
  491. QString fileFilter = getFileFormatFilter(cmd, defaultExt);
  492. if (fileFilter.isEmpty())
  493. {
  494. return false;
  495. }
  496. cmd.setMeshExportFormat(fileFilter, defaultExt);
  497. cmd.print(QObject::tr("Output export format (meshes) set to: %1").arg(defaultExt.toUpper()));
  498. //look for additional parameters
  499. while (!cmd.arguments().empty())
  500. {
  501. QString argument = cmd.arguments().front();
  502. if (ccCommandLineInterface::IsCommand(argument, COMMAND_EXPORT_EXTENSION))
  503. {
  504. //local option confirmed, we can move on
  505. cmd.arguments().pop_front();
  506. if (cmd.arguments().empty())
  507. {
  508. return cmd.error(QObject::tr("Missing parameter: extension after '%1'").arg(COMMAND_EXPORT_EXTENSION));
  509. }
  510. cmd.setMeshExportFormat(cmd.meshExportFormat(), cmd.arguments().takeFirst());
  511. cmd.print(QObject::tr("New output extension for meshes: %1").arg(cmd.meshExportExt()));
  512. }
  513. else
  514. {
  515. break; //as soon as we encounter an unrecognized argument, we break the local loop to go back to the main one!
  516. }
  517. }
  518. return true;
  519. }
  520. CommandChangeHierarchyOutputFormat::CommandChangeHierarchyOutputFormat()
  521. : CommandChangeOutputFormat(QObject::tr("Change hierarchy output format"), COMMAND_HIERARCHY_EXPORT_FORMAT)
  522. {}
  523. bool CommandChangeHierarchyOutputFormat::process(ccCommandLineInterface& cmd)
  524. {
  525. QString defaultExt;
  526. QString fileFilter = getFileFormatFilter(cmd, defaultExt);
  527. if (fileFilter.isEmpty())
  528. {
  529. return false;
  530. }
  531. cmd.setHierarchyExportFormat(fileFilter, defaultExt);
  532. cmd.print(QObject::tr("Output export format (hierarchy) set to: %1").arg(defaultExt.toUpper()));
  533. //look for additional parameters
  534. while (!cmd.arguments().empty())
  535. {
  536. QString argument = cmd.arguments().front();
  537. if (ccCommandLineInterface::IsCommand(argument, COMMAND_EXPORT_EXTENSION))
  538. {
  539. //local option confirmed, we can move on
  540. cmd.arguments().pop_front();
  541. if (cmd.arguments().empty())
  542. {
  543. return cmd.error(QObject::tr("Missing parameter: extension after '%1'").arg(COMMAND_EXPORT_EXTENSION));
  544. }
  545. cmd.setHierarchyExportFormat(cmd.hierarchyExportFormat(), cmd.arguments().takeFirst());
  546. cmd.print(QObject::tr("New output extension for hierarchies: %1").arg(cmd.hierarchyExportExt()));
  547. }
  548. else
  549. {
  550. break; //as soon as we encounter an unrecognized argument, we break the local loop to go back to the main one!
  551. }
  552. }
  553. return true;
  554. }
  555. CommandLoad::CommandLoad()
  556. : ccCommandLineInterface::Command(QObject::tr("Load"), COMMAND_OPEN)
  557. {}
  558. bool CommandLoad::process(ccCommandLineInterface& cmd)
  559. {
  560. if (cmd.arguments().empty())
  561. {
  562. return cmd.error(QObject::tr("Missing parameter: filename after \"-%1\"").arg(COMMAND_OPEN));
  563. }
  564. //optional parameters
  565. int skipLines = 0;
  566. ccCommandLineInterface::GlobalShiftOptions globalShiftOptions;
  567. bool doNotCreateLabels = false;
  568. while (!cmd.arguments().empty())
  569. {
  570. QString argument = cmd.arguments().front();
  571. if (ccCommandLineInterface::IsCommand(argument, COMMAND_OPEN_NO_LABEL))
  572. {
  573. //local option confirmed, we can move on
  574. cmd.arguments().pop_front();
  575. cmd.print(QObject::tr("Will not load labels"));
  576. doNotCreateLabels = true;
  577. }
  578. if (ccCommandLineInterface::IsCommand(argument, COMMAND_OPEN_SKIP_LINES))
  579. {
  580. //local option confirmed, we can move on
  581. cmd.arguments().pop_front();
  582. if (cmd.arguments().empty())
  583. {
  584. return cmd.error(QObject::tr("Missing parameter: number of lines after '%1'").arg(COMMAND_OPEN_SKIP_LINES));
  585. }
  586. bool ok;
  587. skipLines = cmd.arguments().takeFirst().toInt(&ok);
  588. if (!ok)
  589. {
  590. return cmd.error(QObject::tr("Invalid parameter: number of lines after '%1'").arg(COMMAND_OPEN_SKIP_LINES));
  591. }
  592. cmd.print(QObject::tr("Will skip %1 lines").arg(skipLines));
  593. }
  594. else if (cmd.nextCommandIsGlobalShift())
  595. {
  596. //local option confirmed, we can move on
  597. cmd.arguments().pop_front();
  598. if (!cmd.processGlobalShiftCommand(globalShiftOptions))
  599. {
  600. //error message already issued
  601. return false;
  602. }
  603. }
  604. else
  605. {
  606. break;
  607. }
  608. }
  609. if (skipLines >= 0)
  610. {
  611. AsciiFilter::SetDefaultSkippedLineCount(skipLines);
  612. }
  613. AsciiFilter::SetNoLabelCreated(doNotCreateLabels);
  614. //open specified file
  615. QString filename(cmd.arguments().takeFirst());
  616. if (!cmd.importFile(filename, globalShiftOptions))
  617. {
  618. return false;
  619. }
  620. return true;
  621. }
  622. CommandLoadCommandFile::CommandLoadCommandFile()
  623. : ccCommandLineInterface::Command(QObject::tr("Load commands from file"), COMMAND_COMMAND_FILE)
  624. {}
  625. bool CommandLoadCommandFile::process(ccCommandLineInterface& cmd)
  626. {
  627. if (cmd.arguments().empty())
  628. {
  629. return cmd.error(QObject::tr("Missing parameter: filename after \"-%1\"").arg(COMMAND_COMMAND_FILE));
  630. }
  631. QString commandFilePath = cmd.arguments().takeFirst();
  632. //check if file exists
  633. if (!QFileInfo::exists(commandFilePath))
  634. {
  635. return cmd.error(QObject::tr("Command file not exists \"-%1\"").arg(commandFilePath));
  636. }
  637. QFile commandFile(commandFilePath);
  638. if (commandFile.open(QIODevice::ReadOnly))
  639. {
  640. int insertingIndex = 0;
  641. QTextStream in(&commandFile);
  642. while (!in.atEnd())
  643. {
  644. QString line = in.readLine().trimmed();
  645. //early abort whole line comments
  646. if (line.startsWith("#") || line.startsWith("//"))
  647. {
  648. cmd.print(QObject::tr("\t[COMMENT] %1").arg(line));
  649. continue;
  650. }
  651. QStringList argumentsInLine = line.split(" ");
  652. QStringList processedArgs;
  653. // 'massage' the arguments to handle single/double quotes
  654. //TODO handle escaped quotes/spaces
  655. {
  656. bool insideSingleQuoteSection = false;
  657. bool insideDoubleQuoteSection = false;
  658. QString buffer;
  659. static const QChar SingleQuote{ '\'' };
  660. static const QChar DoubleQuote{ '"' };
  661. for (int currentArgIndex = 0; currentArgIndex < argumentsInLine.size(); ++currentArgIndex)
  662. {
  663. QString arg = argumentsInLine[currentArgIndex];
  664. //handle singleQuotes
  665. {
  666. // argument starts with a single quote and not inside double quotes
  667. if (!insideSingleQuoteSection && !insideDoubleQuoteSection && arg.startsWith(SingleQuote))
  668. {
  669. if (arg.endsWith(SingleQuote))
  670. {
  671. arg.remove(SingleQuote);
  672. // nothing to do, non*truncated argument
  673. }
  674. else
  675. {
  676. // we'll collect the next pieces to get the full argument
  677. insideSingleQuoteSection = true;
  678. buffer = arg.mid(1); // remove the single quote
  679. }
  680. }
  681. else if (insideSingleQuoteSection)
  682. {
  683. buffer += QChar(' ') + arg; // append the current argument to the previous one(s)
  684. if (arg.endsWith(SingleQuote))
  685. {
  686. insideSingleQuoteSection = false;
  687. arg = buffer.left(buffer.length() - 1); // remove the single quote
  688. arg.remove(SingleQuote);
  689. }
  690. }
  691. }
  692. //handle doubleQuotes
  693. {
  694. // argument starts with a double quote and not inside single quotes
  695. if (!insideSingleQuoteSection && !insideDoubleQuoteSection && arg.startsWith(DoubleQuote))
  696. {
  697. if (arg.endsWith(DoubleQuote))
  698. {
  699. arg.remove(DoubleQuote);
  700. // nothing to do, non*truncated argument
  701. }
  702. else
  703. {
  704. // we'll collect the next pieces to get the full argument
  705. insideDoubleQuoteSection = true;
  706. buffer = arg.mid(1); // remove the double quote
  707. }
  708. }
  709. else if (insideDoubleQuoteSection)
  710. {
  711. buffer += QChar(' ') + arg; // append the current argument to the previous one(s)
  712. if (arg.endsWith(DoubleQuote))
  713. {
  714. insideDoubleQuoteSection = false;
  715. arg = buffer.left(buffer.length() - 1); // remove the double quote
  716. arg.remove(DoubleQuote);
  717. }
  718. }
  719. }
  720. if (!insideSingleQuoteSection && !insideDoubleQuoteSection)
  721. {
  722. processedArgs.append(arg);
  723. }
  724. }
  725. if (insideSingleQuoteSection || insideDoubleQuoteSection)
  726. {
  727. // the single/double quote section was not closed...
  728. cmd.warning("Probably malformed command (missing closing quote)");
  729. // ...still, we'll try to proceed
  730. processedArgs.append(buffer);
  731. }
  732. }
  733. //inject back all the arguments to the cmd.arguments()
  734. while (!processedArgs.isEmpty())
  735. {
  736. QString processedArg = processedArgs.takeFirst();
  737. if (!processedArg.isEmpty())
  738. {
  739. if (processedArg.startsWith("//") || processedArg.startsWith("#"))
  740. {
  741. //abort and keep the rest in processedArgs
  742. break;
  743. }
  744. if (!(processedArg.startsWith("/*") && processedArg.endsWith("*/")))
  745. {
  746. //standard argument
  747. cmd.arguments().insert(insertingIndex, processedArg);
  748. insertingIndex++;
  749. cmd.print(QObject::tr("\t[%1] %2").arg(insertingIndex - 1).arg(processedArg));
  750. }
  751. else
  752. {
  753. cmd.print(QObject::tr("\t[COMMENT] %1").arg(processedArg));
  754. }
  755. }
  756. }
  757. //if any arguments left, then it is a comment
  758. if (!processedArgs.isEmpty())
  759. {
  760. cmd.print(QObject::tr("\t[COMMENT] %1").arg(processedArgs.join(" ")));
  761. }
  762. }
  763. commandFile.close();
  764. }
  765. return true;
  766. }
  767. CommandClearNormals::CommandClearNormals()
  768. : ccCommandLineInterface::Command(QObject::tr("Clears normals"), COMMAND_CLEAR_NORMALS)
  769. {}
  770. bool CommandClearNormals::process(ccCommandLineInterface& cmd)
  771. {
  772. if (cmd.clouds().empty() && cmd.meshes().empty())
  773. {
  774. return cmd.error(QObject::tr("No entity loaded (be sure to open at least one file with \"-%1 [cloud filename]\" before \"-%2\")").arg(COMMAND_OPEN, COMMAND_CLEAR_NORMALS));
  775. }
  776. for (const CLCloudDesc& thisCloudDesc : cmd.clouds())
  777. {
  778. ccPointCloud* cloud = thisCloudDesc.pc;
  779. if (cloud)
  780. {
  781. cloud->unallocateNorms();
  782. }
  783. }
  784. for (const CLMeshDesc& thisMeshDesc : cmd.meshes())
  785. {
  786. ccMesh* mesh = ccHObjectCaster::ToMesh(thisMeshDesc.mesh);
  787. if (!mesh)
  788. {
  789. assert(false);
  790. continue;
  791. }
  792. mesh->clearTriNormals();
  793. if (mesh->getParent()
  794. && (mesh->getParent()->isA(CC_TYPES::MESH)/*|| mesh->getParent()->isKindOf(CC_TYPES::PRIMITIVE)*/) //TODO
  795. && ccHObjectCaster::ToMesh(mesh->getParent())->getAssociatedCloud() == mesh->getAssociatedCloud())
  796. {
  797. //Can't remove per-vertex normals on a sub mesh!
  798. }
  799. else
  800. {
  801. //mesh is alone, we can freely remove normals
  802. if (mesh->getAssociatedCloud() && mesh->getAssociatedCloud()->isA(CC_TYPES::POINT_CLOUD))
  803. {
  804. static_cast<ccPointCloud*>(mesh->getAssociatedCloud())->unallocateNorms();
  805. }
  806. }
  807. }
  808. return true;
  809. }
  810. CommandInvertNormal::CommandInvertNormal()
  811. : ccCommandLineInterface::Command(QObject::tr("Invert normals"), COMMAND_INVERT_NORMALS)
  812. {}
  813. bool CommandInvertNormal::process(ccCommandLineInterface& cmd)
  814. {
  815. if (cmd.clouds().empty() && cmd.meshes().empty())
  816. {
  817. return cmd.error(QObject::tr("No input point cloud or mesh (be sure to open one with \"-%1 [cloud filename]\" before \"-%2\")").arg(COMMAND_OPEN, COMMAND_INVERT_NORMALS));
  818. }
  819. for (CLCloudDesc& thisCloudDesc : cmd.clouds())
  820. {
  821. ccPointCloud* cloud = thisCloudDesc.pc;
  822. if (!cloud->hasNormals())
  823. {
  824. cmd.warning(QObject::tr("Cloud %1 has no normals").arg(cloud->getName()));
  825. continue;
  826. }
  827. cloud->invertNormals();
  828. if (cmd.autoSaveMode())
  829. {
  830. QString errorStr = cmd.exportEntity(thisCloudDesc, "_INVERTED_NORMALS");
  831. if (!errorStr.isEmpty())
  832. {
  833. return cmd.error(errorStr);
  834. }
  835. }
  836. }
  837. for (CLMeshDesc& thisMeshDesc : cmd.meshes())
  838. {
  839. ccMesh* mesh = ccHObjectCaster::ToMesh(thisMeshDesc.mesh);
  840. if (!mesh)
  841. {
  842. assert(false);
  843. continue;
  844. }
  845. if (!mesh->hasNormals())
  846. {
  847. cmd.warning(QObject::tr("Mesh %1 has no normals").arg(mesh->getName()));
  848. continue;
  849. }
  850. mesh->invertNormals();
  851. if (cmd.autoSaveMode())
  852. {
  853. QString errorStr = cmd.exportEntity(thisMeshDesc, "_INVERTED_NORMALS");
  854. if (!errorStr.isEmpty())
  855. {
  856. return cmd.error(errorStr);
  857. }
  858. }
  859. }
  860. return true;
  861. }
  862. CommandOctreeNormal::CommandOctreeNormal()
  863. : ccCommandLineInterface::Command(QObject::tr("Compute normals with octree"), COMMAND_COMPUTE_OCTREE_NORMALS)
  864. {}
  865. bool CommandOctreeNormal::process(ccCommandLineInterface& cmd)
  866. {
  867. if (cmd.clouds().empty())
  868. {
  869. return cmd.error(QObject::tr("No point cloud to compute normals (be sure to open one with \"-%1 [cloud filename]\" before \"-%2\")").arg(COMMAND_OPEN, COMMAND_COMPUTE_OCTREE_NORMALS));
  870. }
  871. if (cmd.arguments().empty())
  872. {
  873. return cmd.error(QObject::tr("Missing parameter: radius after \"-%1\"").arg(COMMAND_COMPUTE_OCTREE_NORMALS));
  874. }
  875. float radius = std::numeric_limits<float>::quiet_NaN(); //if this stays
  876. QString radiusArg = cmd.arguments().takeFirst();
  877. if (radiusArg.toUpper() != "AUTO")
  878. {
  879. bool ok = false;
  880. radius = radiusArg.toFloat(&ok);
  881. if (!ok)
  882. {
  883. return cmd.error(QObject::tr("Invalid radius"));
  884. }
  885. }
  886. cmd.print(QObject::tr("\tRadius: %1").arg(radiusArg));
  887. CCCoreLib::LOCAL_MODEL_TYPES model = CCCoreLib::QUADRIC;
  888. ccNormalVectors::Orientation orientation = ccNormalVectors::Orientation::UNDEFINED;
  889. while (!cmd.arguments().isEmpty())
  890. {
  891. QString argument = cmd.arguments().front().toUpper();
  892. if (ccCommandLineInterface::IsCommand(argument, OPTION_ORIENT))
  893. {
  894. cmd.arguments().takeFirst();
  895. if (!cmd.arguments().isEmpty())
  896. {
  897. QString orient_argument = cmd.arguments().takeFirst().toUpper();
  898. if (orient_argument == "PLUS_ZERO" || orient_argument == "PLUS_ORIGIN")
  899. {
  900. orientation = ccNormalVectors::Orientation::PLUS_ORIGIN;
  901. }
  902. else if (orient_argument == "MINUS_ZERO" || orient_argument == "MINUS_ORIGIN")
  903. {
  904. orientation = ccNormalVectors::Orientation::MINUS_ORIGIN;
  905. }
  906. else if (orient_argument == "PLUS_BARYCENTER")
  907. {
  908. orientation = ccNormalVectors::Orientation::PLUS_BARYCENTER;
  909. }
  910. else if (orient_argument == "MINUS_BARYCENTER")
  911. {
  912. orientation = ccNormalVectors::Orientation::MINUS_BARYCENTER;
  913. }
  914. else if (orient_argument == "PLUS_X")
  915. {
  916. orientation = ccNormalVectors::Orientation::PLUS_X;
  917. }
  918. else if (orient_argument == "MINUS_X")
  919. {
  920. orientation = ccNormalVectors::Orientation::MINUS_X;
  921. }
  922. else if (orient_argument == "PLUS_Y")
  923. {
  924. orientation = ccNormalVectors::Orientation::PLUS_Y;
  925. }
  926. else if (orient_argument == "MINUS_Y")
  927. {
  928. orientation = ccNormalVectors::Orientation::MINUS_Y;
  929. }
  930. else if (orient_argument == "PLUS_Z")
  931. {
  932. orientation = ccNormalVectors::Orientation::PLUS_Z;
  933. }
  934. else if (orient_argument == "MINUS_Z")
  935. {
  936. orientation = ccNormalVectors::Orientation::MINUS_Z;
  937. }
  938. else if (orient_argument == "PREVIOUS")
  939. {
  940. orientation = ccNormalVectors::Orientation::PREVIOUS;
  941. }
  942. else if (orient_argument == "PLUS_SENSOR_ORIGIN")
  943. {
  944. orientation = ccNormalVectors::Orientation::PLUS_SENSOR_ORIGIN;
  945. }
  946. else if (orient_argument == "MINUS_SENSOR_ORIGIN")
  947. {
  948. orientation = ccNormalVectors::Orientation::MINUS_SENSOR_ORIGIN;
  949. }
  950. else
  951. {
  952. return cmd.error(QObject::tr("Invalid parameter: unknown orientation '%1'").arg(orient_argument));
  953. }
  954. }
  955. else
  956. {
  957. return cmd.error(QObject::tr("Missing orientation"));
  958. }
  959. }
  960. else if (ccCommandLineInterface::IsCommand(argument, OPTION_MODEL))
  961. {
  962. cmd.arguments().takeFirst();
  963. if (!cmd.arguments().isEmpty())
  964. {
  965. QString model_arg = cmd.arguments().takeFirst().toUpper();
  966. if (model_arg == "LS")
  967. {
  968. model = CCCoreLib::LOCAL_MODEL_TYPES::LS;
  969. }
  970. else if (model_arg == "TRI")
  971. {
  972. model = CCCoreLib::LOCAL_MODEL_TYPES::TRI;
  973. }
  974. else if (model_arg == "QUADRIC")
  975. {
  976. model = CCCoreLib::LOCAL_MODEL_TYPES::QUADRIC;
  977. }
  978. else
  979. {
  980. return cmd.error(QObject::tr("Invalid parameter: unknown model '%1'").arg(model_arg));
  981. }
  982. }
  983. else
  984. {
  985. return cmd.error(QObject::tr("Missing model"));
  986. }
  987. }
  988. else
  989. {
  990. break;
  991. }
  992. }
  993. for (const CLCloudDesc& thisCloudDesc : cmd.clouds())
  994. {
  995. ccPointCloud* cloud = thisCloudDesc.pc;
  996. QScopedPointer<ccProgressDialog> progressDialog(nullptr);
  997. if (!cmd.silentMode())
  998. {
  999. progressDialog.reset(new ccProgressDialog(true, cmd.widgetParent()));
  1000. progressDialog->setAutoClose(false);
  1001. }
  1002. if (!cloud->getOctree())
  1003. {
  1004. if (!cloud->computeOctree(progressDialog.data()))
  1005. {
  1006. return cmd.error(QObject::tr("Failed to compute octree for cloud '%1'").arg(cloud->getName()));
  1007. }
  1008. }
  1009. float thisCloudRadius = radius;
  1010. if (std::isnan(thisCloudRadius))
  1011. {
  1012. ccOctree::BestRadiusParams params;
  1013. {
  1014. params.aimedPopulationPerCell = 16;
  1015. params.aimedPopulationRange = 4;
  1016. params.minCellPopulation = 6;
  1017. params.minAboveMinRatio = 0.97;
  1018. }
  1019. thisCloudRadius = ccOctree::GuessBestRadius(cloud, params, cloud->getOctree().data());
  1020. if (thisCloudRadius == 0)
  1021. {
  1022. return cmd.error(QObject::tr("Failed to determine best normal radius for cloud '%1'").arg(cloud->getName()));
  1023. }
  1024. cmd.print(QObject::tr("\tCloud %1 radius = %2").arg(cloud->getName()).arg(thisCloudRadius));
  1025. }
  1026. cmd.print(QObject::tr("computeNormalsWithOctree started..."));
  1027. bool success = cloud->computeNormalsWithOctree(model, orientation, thisCloudRadius, progressDialog.data());
  1028. if(success)
  1029. {
  1030. cmd.print(QObject::tr("computeNormalsWithOctree success"));
  1031. }
  1032. else
  1033. {
  1034. return cmd.error(QObject::tr("computeNormalsWithOctree failed"));
  1035. }
  1036. cloud->setName(cloud->getName() + QObject::tr(".OctreeNormal"));
  1037. if (cmd.autoSaveMode())
  1038. {
  1039. CLCloudDesc desc(cloud, thisCloudDesc.basename, thisCloudDesc.path, thisCloudDesc.indexInFile);
  1040. QString errorStr = cmd.exportEntity(desc, "OCTREE_NORMALS");
  1041. if (!errorStr.isEmpty())
  1042. {
  1043. return cmd.error(errorStr);
  1044. }
  1045. }
  1046. }
  1047. return true;
  1048. }
  1049. CommandConvertNormalsToDipAndDipDir::CommandConvertNormalsToDipAndDipDir()
  1050. : ccCommandLineInterface::Command(QObject::tr("Convert normals to dip and dip. dir."), COMMAND_CONVERT_NORMALS_TO_DIP)
  1051. {}
  1052. bool CommandConvertNormalsToDipAndDipDir::process(ccCommandLineInterface& cmd)
  1053. {
  1054. if (cmd.clouds().empty())
  1055. {
  1056. return cmd.error(QObject::tr("No input point cloud (be sure to open one with \"-%1 [cloud filename]\" before \"-%2\")").arg(COMMAND_OPEN, COMMAND_CONVERT_NORMALS_TO_DIP));
  1057. }
  1058. for (CLCloudDesc& thisCloudDesc : cmd.clouds())
  1059. {
  1060. ccPointCloud* cloud = thisCloudDesc.pc;
  1061. if (!cloud->hasNormals())
  1062. {
  1063. cmd.warning(QObject::tr("Cloud %1 has no normals").arg(cloud->getName()));
  1064. continue;
  1065. }
  1066. ccHObject::Container container;
  1067. container.push_back(cloud);
  1068. if (!ccEntityAction::convertNormalsTo(container, ccEntityAction::NORMAL_CONVERSION_DEST::DIP_DIR_SFS))
  1069. {
  1070. return cmd.error(QObject::tr("Failed to convert normals to dip and dip direction"));
  1071. }
  1072. if (cmd.autoSaveMode())
  1073. {
  1074. QString errorStr = cmd.exportEntity(thisCloudDesc, "_DIP_AND_DIP_DIR");
  1075. if (!errorStr.isEmpty())
  1076. {
  1077. return cmd.error(errorStr);
  1078. }
  1079. }
  1080. }
  1081. return true;
  1082. }
  1083. CommandConvertNormalsToSFs::CommandConvertNormalsToSFs()
  1084. : ccCommandLineInterface::Command(QObject::tr("Convert normals to scalar fields"), COMMAND_CONVERT_NORMALS_TO_SFS)
  1085. {}
  1086. bool CommandConvertNormalsToSFs::process(ccCommandLineInterface& cmd)
  1087. {
  1088. if (cmd.clouds().empty())
  1089. {
  1090. return cmd.error(QObject::tr("No input point cloud (be sure to open one with \"-%1 [cloud filename]\" before \"-%2\")").arg(COMMAND_OPEN, COMMAND_CONVERT_NORMALS_TO_SFS));
  1091. }
  1092. for (CLCloudDesc& thisCloudDesc : cmd.clouds())
  1093. {
  1094. ccPointCloud* cloud = thisCloudDesc.pc;
  1095. if (!cloud->hasNormals())
  1096. {
  1097. cmd.warning(QObject::tr("Cloud %1 has no normals").arg(cloud->getName()));
  1098. continue;
  1099. }
  1100. bool exportDims[3] = { true, true, true };
  1101. ccHObject::Container container;
  1102. container.push_back(cloud);
  1103. if (!ccEntityAction::exportNormalToSF(container, cmd.widgetParent(), exportDims))
  1104. {
  1105. return cmd.error(QObject::tr("Failed to convert normals to scalar fields"));
  1106. }
  1107. if (cmd.autoSaveMode())
  1108. {
  1109. QString errorStr = cmd.exportEntity(thisCloudDesc, "_NORM_TO_SF");
  1110. if (!errorStr.isEmpty())
  1111. {
  1112. return cmd.error(errorStr);
  1113. }
  1114. }
  1115. }
  1116. return true;
  1117. }
  1118. CommandConvertNormalsToHSV::CommandConvertNormalsToHSV()
  1119. : ccCommandLineInterface::Command(QObject::tr("Convert normals to HSV colors"), COMMAND_CONVERT_NORMALS_TO_HSV)
  1120. {}
  1121. bool CommandConvertNormalsToHSV::process(ccCommandLineInterface& cmd)
  1122. {
  1123. if (cmd.clouds().empty())
  1124. {
  1125. return cmd.error(QObject::tr("No input point cloud (be sure to open one with \"-%1 [cloud filename]\" before \"-%2\")").arg(COMMAND_OPEN, COMMAND_CONVERT_NORMALS_TO_HSV));
  1126. }
  1127. for (CLCloudDesc& thisCloudDesc : cmd.clouds())
  1128. {
  1129. ccPointCloud* cloud = thisCloudDesc.pc;
  1130. if (!cloud->hasNormals())
  1131. {
  1132. cmd.warning(QObject::tr("Cloud %1 has no normals").arg(cloud->getName()));
  1133. continue;
  1134. }
  1135. ccHObject::Container container;
  1136. container.push_back(cloud);
  1137. if (!ccEntityAction::convertNormalsTo(container, ccEntityAction::NORMAL_CONVERSION_DEST::HSV_COLORS))
  1138. {
  1139. return cmd.error(QObject::tr("Failed to convert normals to HSV colors"));
  1140. }
  1141. if (cmd.autoSaveMode())
  1142. {
  1143. QString errorStr = cmd.exportEntity(thisCloudDesc, "_HSV");
  1144. if (!errorStr.isEmpty())
  1145. {
  1146. return cmd.error(errorStr);
  1147. }
  1148. }
  1149. }
  1150. return true;
  1151. }
  1152. CommandSubsample::CommandSubsample()
  1153. : ccCommandLineInterface::Command(QObject::tr("Subsample"), COMMAND_SUBSAMPLE)
  1154. {}
  1155. bool CommandSubsample::process(ccCommandLineInterface& cmd)
  1156. {
  1157. if (cmd.clouds().empty())
  1158. {
  1159. return cmd.error(QObject::tr("No point cloud to resample (be sure to open one with \"-%1 [cloud filename]\" before \"-%2\")").arg(COMMAND_OPEN, COMMAND_SUBSAMPLE));
  1160. }
  1161. if (cmd.arguments().empty())
  1162. {
  1163. return cmd.error(QObject::tr("Missing parameter: resampling method after \"-%1\"").arg(COMMAND_SUBSAMPLE));
  1164. }
  1165. QString method = cmd.arguments().takeFirst().toUpper();
  1166. cmd.print(QObject::tr("\tMethod: ") + method);
  1167. if (method == "RANDOM")
  1168. {
  1169. if (cmd.arguments().empty())
  1170. {
  1171. return cmd.error(QObject::tr("Missing parameter: number of points or option \"%2\" after \"-%1 RANDOM \"").arg(COMMAND_SUBSAMPLE).arg(OPTION_PERCENT));
  1172. }
  1173. bool isPercent = false;
  1174. double percent = 0.0;
  1175. unsigned count = 0;
  1176. //handle percent argument
  1177. if (cmd.arguments().front() == OPTION_PERCENT)
  1178. {
  1179. //local option verified
  1180. cmd.arguments().pop_front();
  1181. if (cmd.arguments().empty())
  1182. {
  1183. return cmd.error(QObject::tr("Missing parameter: number after \"-%1 RANDOM %2\"").arg(COMMAND_SUBSAMPLE).arg(OPTION_PERCENT));
  1184. }
  1185. bool ok;
  1186. percent = cmd.arguments().takeFirst().toDouble(&ok);
  1187. if (!ok || percent < 0 || percent > 100)
  1188. {
  1189. return cmd.error(QObject::tr("Invalid parameter: number after \"-%1 RANDOM %2\" must be decimal between 0 and 100").arg(COMMAND_SUBSAMPLE).arg(OPTION_PERCENT));
  1190. }
  1191. isPercent = true;
  1192. }
  1193. else
  1194. {
  1195. bool ok;
  1196. count = cmd.arguments().takeFirst().toUInt(&ok);
  1197. if (!ok)
  1198. {
  1199. return cmd.error(QObject::tr("Invalid parameter: number of points or option \"%2\" after \"-%1 RANDOM \"").arg(COMMAND_SUBSAMPLE).arg(OPTION_PERCENT));
  1200. }
  1201. cmd.print(QObject::tr("\tOutput points: %1").arg(count));
  1202. }
  1203. for (CLCloudDesc& desc : cmd.clouds())
  1204. {
  1205. cmd.print(QObject::tr("\tProcessing cloud %1").arg(!desc.pc->getName().isEmpty() ? desc.pc->getName() : "no name"));
  1206. if (isPercent)
  1207. {
  1208. size_t nrOfPoints = desc.pc->size();
  1209. count = static_cast<unsigned>(ceil(nrOfPoints * percent / 100));
  1210. cmd.print(QObject::tr("\tOutput points: %1 * %2% = %3").arg(nrOfPoints).arg(percent).arg(count));
  1211. }
  1212. CCCoreLib::ReferenceCloud* refCloud = CCCoreLib::CloudSamplingTools::subsampleCloudRandomly(desc.pc, count, cmd.progressDialog());
  1213. if (!refCloud)
  1214. {
  1215. return cmd.error(QObject::tr("Subsampling process failed!"));
  1216. }
  1217. cmd.print(QObject::tr("\tResult: %1 points").arg(refCloud->size()));
  1218. //save output
  1219. ccPointCloud* result = desc.pc->partialClone(refCloud);
  1220. delete refCloud;
  1221. refCloud = nullptr;
  1222. if (result)
  1223. {
  1224. result->setName(desc.pc->getName() + QObject::tr(".subsampled"));
  1225. if (cmd.autoSaveMode())
  1226. {
  1227. CLCloudDesc newDesc(result, desc.basename, desc.path, desc.indexInFile);
  1228. QString errorStr = cmd.exportEntity(newDesc, "RANDOM_SUBSAMPLED");
  1229. if (!errorStr.isEmpty())
  1230. {
  1231. delete result;
  1232. return cmd.error(errorStr);
  1233. }
  1234. }
  1235. //replace current cloud by this one
  1236. delete desc.pc;
  1237. desc.pc = result;
  1238. desc.basename += "_RANDOM_SUBSAMPLED";
  1239. }
  1240. else
  1241. {
  1242. return cmd.error(QObject::tr("Not enough memory!"));
  1243. }
  1244. }
  1245. }
  1246. else if (method == "SPATIAL")
  1247. {
  1248. if (cmd.arguments().empty())
  1249. {
  1250. return cmd.error(QObject::tr("Missing parameter: spatial step after \"-%1 SPATIAL\"").arg(COMMAND_SUBSAMPLE));
  1251. }
  1252. bool ok;
  1253. double step = cmd.arguments().takeFirst().toDouble(&ok);
  1254. if (!ok || step <= 0)
  1255. {
  1256. return cmd.error(QObject::tr("Invalid step value for spatial subsampling!"));
  1257. }
  1258. cmd.print(QObject::tr("\tSpatial step: %1").arg(step));
  1259. double sfMinSpacing = 0;
  1260. double sfMaxSpacing = 0;
  1261. bool useActiveSF = false;
  1262. if (!cmd.arguments().empty())
  1263. {
  1264. if (cmd.arguments().front().toUpper() == OPTION_USE_ACTIVE_SF)
  1265. {
  1266. //enable USE_ACTIVE_SF
  1267. useActiveSF = true;
  1268. cmd.arguments().pop_front();
  1269. if (cmd.arguments().size() >= 2)
  1270. {
  1271. bool validMin = false;
  1272. sfMinSpacing = cmd.arguments().takeFirst().toDouble(&validMin);
  1273. bool validMax = false;
  1274. sfMaxSpacing = cmd.arguments().takeFirst().toDouble(&validMax);
  1275. if (!validMin || !validMax || sfMinSpacing < 0 || sfMaxSpacing < 0)
  1276. {
  1277. return cmd.error(QObject::tr("Invalid parameters: Two positive decimal number required after '%1'").arg(OPTION_USE_ACTIVE_SF));
  1278. }
  1279. }
  1280. else
  1281. {
  1282. return cmd.error(QObject::tr("Missing parameters: Two positive decimal number required after '%1'").arg(OPTION_USE_ACTIVE_SF));
  1283. }
  1284. }
  1285. }
  1286. for (CLCloudDesc& desc : cmd.clouds())
  1287. {
  1288. cmd.print(QObject::tr("\tProcessing cloud %1").arg(!desc.pc->getName().isEmpty() ? desc.pc->getName() : "no name"));
  1289. CCCoreLib::CloudSamplingTools::SFModulationParams modParams(false);
  1290. //handle Use Active SF on each cloud
  1291. if (useActiveSF)
  1292. {
  1293. //look for the min and max sf values
  1294. ccScalarField* sf = desc.pc->getCurrentDisplayedScalarField();
  1295. if (!sf)
  1296. {
  1297. //warn the user, not use active SF and keep going
  1298. cmd.warning(QObject::tr("\tCan't use 'Use active SF': no active scalar field. Set one with '-%1'").arg(COMMAND_SET_ACTIVE_SF));
  1299. }
  1300. else
  1301. {
  1302. //found active scalar field
  1303. ScalarType sfMin = CCCoreLib::NAN_VALUE;
  1304. ScalarType sfMax = CCCoreLib::NAN_VALUE;
  1305. if (sf->countValidValues() > 0)
  1306. {
  1307. if (!ccScalarField::ValidValue(sfMin) || sfMin > sf->getMin())
  1308. sfMin = sf->getMin();
  1309. if (!ccScalarField::ValidValue(sfMax) || sfMax < sf->getMax())
  1310. sfMax = sf->getMax();
  1311. if (!ccScalarField::ValidValue(sfMin) || !ccScalarField::ValidValue(sfMax))
  1312. {
  1313. //warn the user, don't use 'Use Active SF' and keep going
  1314. cmd.warning(QObject::tr("\tCan't use 'Use active SF': scalar field '%1' has invalid min/max values.").arg(QString::fromStdString(sf->getName())));
  1315. }
  1316. else
  1317. {
  1318. //everything validated use acitve SF for modulation
  1319. //implementation of modParams.a/b values come from MainWindow::doActionSubsample()
  1320. modParams.enabled = true;
  1321. double deltaSF = static_cast<double>(sfMax) - static_cast<double>(sfMin);
  1322. if (CCCoreLib::GreaterThanEpsilon(deltaSF))
  1323. {
  1324. modParams.a = (sfMaxSpacing - sfMinSpacing) / deltaSF;
  1325. modParams.b = sfMinSpacing - modParams.a * sfMin;
  1326. }
  1327. else
  1328. {
  1329. modParams.a = 0.0;
  1330. modParams.b = sfMin;
  1331. }
  1332. cmd.print(QObject::tr("\tUse active SF: enabled\n\t\tSpacing at SF min (%1): %2\n\t\tSpacing at SF max (%3): %4")
  1333. .arg(sfMin)
  1334. .arg(sfMinSpacing)
  1335. .arg(sfMax)
  1336. .arg(sfMaxSpacing));
  1337. }
  1338. }
  1339. else
  1340. {
  1341. //warn the user, not use Use Active SF and keep going
  1342. cmd.warning(QObject::tr("\tCan't use 'Use active SF': scalar field '%2' does not have any valid value.").arg(COMMAND_SET_ACTIVE_SF).arg(QString::fromStdString(sf->getName())));
  1343. }
  1344. }
  1345. }
  1346. if (useActiveSF && !modParams.enabled)
  1347. {
  1348. cmd.print(QObject::tr("\t'Use active SF' disabled. Falling back to constant spacing."));
  1349. }
  1350. CCCoreLib::ReferenceCloud* refCloud = CCCoreLib::CloudSamplingTools::resampleCloudSpatially(desc.pc, static_cast<PointCoordinateType>(step), modParams, nullptr, cmd.progressDialog());
  1351. if (!refCloud)
  1352. {
  1353. return cmd.error("Subsampling process failed!");
  1354. }
  1355. cmd.print(QObject::tr("\tResult: %1 points").arg(refCloud->size()));
  1356. //save output
  1357. ccPointCloud* result = desc.pc->partialClone(refCloud);
  1358. delete refCloud;
  1359. refCloud = nullptr;
  1360. if (result)
  1361. {
  1362. result->setName(desc.pc->getName() + QObject::tr(".subsampled"));
  1363. if (cmd.autoSaveMode())
  1364. {
  1365. CLCloudDesc newDesc(result, desc.basename, desc.path, desc.indexInFile);
  1366. QString errorStr = cmd.exportEntity(newDesc, "SPATIAL_SUBSAMPLED");
  1367. if (!errorStr.isEmpty())
  1368. {
  1369. delete result;
  1370. return cmd.error(errorStr);
  1371. }
  1372. }
  1373. //replace current cloud by this one
  1374. delete desc.pc;
  1375. desc.pc = result;
  1376. desc.basename += "_SPATIAL_SUBSAMPLED";
  1377. }
  1378. else
  1379. {
  1380. return cmd.error(QObject::tr("Not enough memory!"));
  1381. }
  1382. }
  1383. }
  1384. else if (method == "OCTREE")
  1385. {
  1386. int octreeLevel = 1;
  1387. double cellSize = 0.0;
  1388. bool byCellSize = false;
  1389. bool byMaxNumberOfPoints = false;
  1390. unsigned maxNumberOfPoints = 0;
  1391. bool isPercent = false;
  1392. double percent = 0.0;
  1393. const int maxOctreeLevel = CCCoreLib::DgmOctree::MAX_OCTREE_LEVEL;
  1394. if (!cmd.arguments().empty())
  1395. {
  1396. //params for automatic OCTREE level calculation based on cell size
  1397. if (cmd.arguments().front() == "CELL_SIZE")
  1398. {
  1399. cmd.arguments().pop_front();
  1400. if (cmd.arguments().empty())
  1401. {
  1402. return cmd.error(QObject::tr("Missing parameter: octree cell size after \"-%1 OCTREE CELL_SIZE \"").arg(COMMAND_SUBSAMPLE));
  1403. }
  1404. bool ok = false;
  1405. cellSize = cmd.arguments().takeFirst().toDouble(&ok);
  1406. if (!ok)
  1407. {
  1408. return cmd.error(QObject::tr("Invalid parameter: octree cell size after \"-%1 OCTREE CELL_SIZE \"").arg(COMMAND_SUBSAMPLE));
  1409. }
  1410. byCellSize = true;
  1411. cmd.print(QObject::tr("\tOctree cell size: %1").arg(cellSize));
  1412. }
  1413. //params for automatic OCTREE level calculation based on number of points
  1414. else if (cmd.arguments().front() == OPTION_NUMBER_OF_POINTS)
  1415. {
  1416. //local option verified
  1417. byMaxNumberOfPoints = true;
  1418. cmd.arguments().pop_front();
  1419. if (cmd.arguments().empty())
  1420. {
  1421. return cmd.error(QObject::tr("Missing parameter: number of points or option \"%3\" after \"-%1 OCTREE %2 \"").arg(COMMAND_SUBSAMPLE).arg(OPTION_NUMBER_OF_POINTS).arg(OPTION_PERCENT));
  1422. }
  1423. //handle percent argument
  1424. if (cmd.arguments().front() == OPTION_PERCENT)
  1425. {
  1426. //local option verified
  1427. cmd.arguments().pop_front();
  1428. if (cmd.arguments().empty())
  1429. {
  1430. return cmd.error(QObject::tr("Missing parameter: number after \"-%1 OCTREE %2 %3\"").arg(COMMAND_SUBSAMPLE).arg(OPTION_NUMBER_OF_POINTS).arg(OPTION_PERCENT));
  1431. }
  1432. bool ok = false;
  1433. percent = cmd.arguments().takeFirst().toDouble(&ok);
  1434. if (!ok || percent < 0 || percent > 100)
  1435. {
  1436. return cmd.error(QObject::tr("Invalid parameter: number after \"-%1 OCTREE %2 %3\" must be decimal between 0 and 100").arg(COMMAND_SUBSAMPLE).arg(OPTION_NUMBER_OF_POINTS).arg(OPTION_PERCENT));
  1437. }
  1438. isPercent = true;
  1439. }
  1440. else
  1441. {
  1442. bool ok = false;
  1443. maxNumberOfPoints = cmd.arguments().takeFirst().toUInt(&ok);
  1444. if (!ok)
  1445. {
  1446. return cmd.error(QObject::tr("Invalid parameter: number of points or option \"%3\" after \"-%1 OCTREE %2 \"").arg(COMMAND_SUBSAMPLE).arg(OPTION_NUMBER_OF_POINTS).arg(OPTION_PERCENT));
  1447. }
  1448. cmd.print(QObject::tr("\tOctree target number of points: %1").arg(maxNumberOfPoints));
  1449. }
  1450. }
  1451. //params for original version octree calculation based on given level
  1452. else
  1453. {
  1454. if (cmd.arguments().empty())
  1455. {
  1456. return cmd.error(QObject::tr("Missing parameter: octree level after \"-%1 OCTREE\"").arg(COMMAND_SUBSAMPLE));
  1457. }
  1458. bool ok = false;
  1459. octreeLevel = cmd.arguments().takeFirst().toInt(&ok);
  1460. if (!ok || octreeLevel < 1 || octreeLevel > maxOctreeLevel)
  1461. {
  1462. return cmd.error(QObject::tr("Invalid octree level!"));
  1463. }
  1464. cmd.print(QObject::tr("\tOctree level: %1").arg(octreeLevel));
  1465. }
  1466. }
  1467. QScopedPointer<ccProgressDialog> progressDialog(nullptr);
  1468. if (!cmd.silentMode())
  1469. {
  1470. progressDialog.reset(new ccProgressDialog(false, cmd.widgetParent()));
  1471. progressDialog->setAutoClose(false);
  1472. }
  1473. for (CLCloudDesc& desc : cmd.clouds())
  1474. {
  1475. //calculate octree before subsampling, it is passed to subsampling, so it won't be recalculated there.
  1476. CCCoreLib::DgmOctree* octree = desc.pc->getOctree().data();
  1477. if (!octree)
  1478. {
  1479. octree = desc.pc->computeOctree(nullptr, false).data();
  1480. }
  1481. if (!octree)
  1482. {
  1483. return cmd.error("Octree calculation failed, not enough memory?");
  1484. }
  1485. CCCoreLib::ReferenceCloud* refCloud = nullptr;
  1486. ccPointCloud* result = nullptr;
  1487. unsigned sizeOfInputCloud = desc.pc->size();
  1488. cmd.print(QObject::tr("\tProcessing cloud %1").arg(!desc.pc->getName().isEmpty() ? desc.pc->getName() : "no name"));
  1489. if (byCellSize)
  1490. {
  1491. //calculate OCTREE level for each cloud based on required octree cell size
  1492. octreeLevel = static_cast<int>(ceil(log(desc.pc->getOwnBB().getMaxBoxDim() / cellSize) / log(2.0)));
  1493. }
  1494. if (byMaxNumberOfPoints)
  1495. {
  1496. if (isPercent)
  1497. {
  1498. maxNumberOfPoints = static_cast<unsigned>(ceil(sizeOfInputCloud * percent / 100));
  1499. cmd.print(QObject::tr("\tOutput point target: %1 * %2% = %3").arg(sizeOfInputCloud).arg(percent).arg(maxNumberOfPoints));
  1500. }
  1501. //calculate OCTREE level for each cloud based on required number of points
  1502. octreeLevel = maxOctreeLevel;
  1503. unsigned numberOfPoints = sizeOfInputCloud;
  1504. //go through max->min until previous point count and current point count is different then break
  1505. for (int currentOctreeLevel = maxOctreeLevel; currentOctreeLevel > 0; --currentOctreeLevel)
  1506. {
  1507. unsigned currentNumberOfPoints = octree->getCellNumber(currentOctreeLevel);
  1508. if (currentNumberOfPoints != numberOfPoints && numberOfPoints < maxNumberOfPoints)
  1509. {
  1510. break;
  1511. }
  1512. octreeLevel = currentOctreeLevel;
  1513. numberOfPoints = currentNumberOfPoints;
  1514. }
  1515. }
  1516. //only process further if CELL_SIZE or octree level was given, or the numberOfPoints smaller than the input cloud
  1517. if (!byMaxNumberOfPoints || (byMaxNumberOfPoints && sizeOfInputCloud > maxNumberOfPoints))
  1518. {
  1519. //overwrite and print out finalized OCTREE level.
  1520. if (byCellSize || byMaxNumberOfPoints)
  1521. {
  1522. //clamp octree level
  1523. octreeLevel = std::max(std::min(octreeLevel, maxOctreeLevel), 1);
  1524. cmd.print(QObject::tr("\tCalculated octree level: %1").arg(octreeLevel));
  1525. }
  1526. refCloud = CCCoreLib::CloudSamplingTools::subsampleCloudWithOctreeAtLevel( desc.pc,
  1527. static_cast<unsigned char>(octreeLevel),
  1528. CCCoreLib::CloudSamplingTools::NEAREST_POINT_TO_CELL_CENTER,
  1529. progressDialog.data(),
  1530. octree);
  1531. if (!refCloud)
  1532. {
  1533. return cmd.error(QObject::tr("Subsampling process failed!"));
  1534. }
  1535. cmd.print(QObject::tr("\tResult: %1 points").arg(refCloud->size()));
  1536. //save output
  1537. result = desc.pc->partialClone(refCloud);
  1538. delete refCloud;
  1539. refCloud = nullptr;
  1540. }
  1541. else
  1542. {
  1543. cmd.print("\tNot subsampled, point count is smaller than max number of points");
  1544. //no subsampling happened so result="input cloud"
  1545. result = desc.pc;
  1546. //set octreeLevel to indicate it was not subsampled at all
  1547. octreeLevel = -1;
  1548. }
  1549. if (result)
  1550. {
  1551. result->setName(desc.pc->getName() + QObject::tr(".subsampled"));
  1552. QString suffix = QObject::tr("OCTREE_LEVEL_%1_SUBSAMPLED").arg(octreeLevel);
  1553. if (cmd.autoSaveMode())
  1554. {
  1555. CLCloudDesc newDesc(result, desc.basename, desc.path, desc.indexInFile);
  1556. QString errorStr = cmd.exportEntity(newDesc, suffix);
  1557. if (!errorStr.isEmpty())
  1558. {
  1559. delete result;
  1560. return cmd.error(errorStr);
  1561. }
  1562. }
  1563. if (desc.pc != result)
  1564. {
  1565. //replace current cloud by subsampled one if it was changed
  1566. delete desc.pc;
  1567. desc.pc = result;
  1568. desc.basename += '_' + suffix;
  1569. }
  1570. }
  1571. else
  1572. {
  1573. return cmd.error(QObject::tr("Not enough memory!"));
  1574. }
  1575. }
  1576. if (progressDialog)
  1577. {
  1578. progressDialog->close();
  1579. QCoreApplication::processEvents();
  1580. }
  1581. }
  1582. else
  1583. {
  1584. return cmd.error(QObject::tr("Unknown method!"));
  1585. }
  1586. return true;
  1587. }
  1588. CommandExtractCCs::CommandExtractCCs()
  1589. : ccCommandLineInterface::Command(QObject::tr("ExtractCCs"), COMMAND_EXTRACT_CC)
  1590. {}
  1591. bool CommandExtractCCs::process(ccCommandLineInterface& cmd)
  1592. {
  1593. if (cmd.clouds().empty())
  1594. {
  1595. return cmd.error(QObject::tr("No point cloud loaded (be sure to open one with \"-%1 [cloud filename]\" before \"-%2\")").arg(COMMAND_OPEN, COMMAND_EXTRACT_CC));
  1596. }
  1597. //octree level
  1598. if (cmd.arguments().empty())
  1599. {
  1600. return cmd.error(QObject::tr("Missing parameter: octree level after \"-%1\"").arg(COMMAND_EXTRACT_CC));
  1601. }
  1602. bool ok;
  1603. unsigned char octreeLevel = std::min<unsigned char>(cmd.arguments().takeFirst().toUShort(&ok), CCCoreLib::DgmOctree::MAX_OCTREE_LEVEL);
  1604. if (!ok)
  1605. {
  1606. return cmd.error(QObject::tr("Invalid octree level!"));
  1607. }
  1608. cmd.print(QObject::tr("\tOctree level: %1").arg(octreeLevel));
  1609. //min number of points
  1610. if (cmd.arguments().empty())
  1611. {
  1612. return cmd.error(QObject::tr("Missing parameter: minimum number of points per component after \"-%1 [octree level]\"").arg(COMMAND_EXTRACT_CC));
  1613. }
  1614. unsigned minPointCount = cmd.arguments().takeFirst().toUInt(&ok);
  1615. if (!ok)
  1616. {
  1617. return cmd.error(QObject::tr("Invalid min. number of points!"));
  1618. }
  1619. cmd.print(QObject::tr("\tMin number of points per component: %1").arg(minPointCount));
  1620. try
  1621. {
  1622. QScopedPointer<ccProgressDialog> progressDialog(nullptr);
  1623. if (!cmd.silentMode())
  1624. {
  1625. progressDialog.reset(new ccProgressDialog(false, cmd.widgetParent()));
  1626. progressDialog->setAutoClose(false);
  1627. }
  1628. std::vector<CLCloudDesc> inputClouds = cmd.clouds();
  1629. cmd.clouds().clear();
  1630. for (CLCloudDesc& desc : inputClouds)
  1631. {
  1632. cmd.print(QObject::tr("\tProcessing cloud %1").arg(!desc.pc->getName().isEmpty() ? desc.pc->getName() : "no name"));
  1633. //we create/activate CCs label scalar field
  1634. int sfIdx = desc.pc->getScalarFieldIndexByName(CC_CONNECTED_COMPONENTS_DEFAULT_LABEL_NAME);
  1635. if (sfIdx < 0)
  1636. {
  1637. sfIdx = desc.pc->addScalarField(CC_CONNECTED_COMPONENTS_DEFAULT_LABEL_NAME);
  1638. }
  1639. if (sfIdx < 0)
  1640. {
  1641. cmd.error(QObject::tr("Couldn't allocate a new scalar field for computing CC labels! Try to free some memory ..."));
  1642. continue;
  1643. }
  1644. desc.pc->setCurrentScalarField(sfIdx);
  1645. //try to label all CCs
  1646. int componentCount = CCCoreLib::AutoSegmentationTools::labelConnectedComponents(desc.pc,
  1647. static_cast<unsigned char>(octreeLevel),
  1648. false,
  1649. progressDialog.data());
  1650. if (componentCount == 0)
  1651. {
  1652. cmd.error(QObject::tr("No component found!"));
  1653. continue;
  1654. }
  1655. desc.pc->getCurrentInScalarField()->computeMinAndMax();
  1656. CCCoreLib::ReferenceCloudContainer components;
  1657. bool success = CCCoreLib::AutoSegmentationTools::extractConnectedComponents(desc.pc, components);
  1658. desc.pc->deleteScalarField(sfIdx);
  1659. sfIdx = -1;
  1660. if (!success)
  1661. {
  1662. cmd.warning(QObject::tr("An error occurred (failed to finish the extraction)"));
  1663. continue;
  1664. }
  1665. //we create "real" point clouds for all input components
  1666. int realIndex = 0;
  1667. for (size_t j = 0; j < components.size(); ++j)
  1668. {
  1669. CCCoreLib::ReferenceCloud* compIndexes = components[j];
  1670. //if it has enough points
  1671. if (compIndexes->size() >= minPointCount)
  1672. {
  1673. //we create a new entity
  1674. ccPointCloud* compCloud = desc.pc->partialClone(compIndexes);
  1675. if (compCloud)
  1676. {
  1677. //'shift on load' information
  1678. compCloud->copyGlobalShiftAndScale(*desc.pc);
  1679. compCloud->setName(QString(desc.pc->getName() + "_CC#%1").arg(j + 1));
  1680. QString filenameSuffix = QObject::tr("_COMPONENT_%1").arg(++realIndex);
  1681. if (desc.indexInFile >= 0)
  1682. {
  1683. // add the cloud name and its index in the file to avoid overwriting files if mutlitple clouds came from the same file
  1684. filenameSuffix.prepend(QObject::tr("_CLOUD_%1(%2)").arg(desc.pc->getName()).arg(desc.indexInFile));
  1685. }
  1686. CLCloudDesc newDesc(compCloud, desc.basename + filenameSuffix, desc.path);
  1687. if (cmd.autoSaveMode())
  1688. {
  1689. QString errorStr = cmd.exportEntity(newDesc, QString(), nullptr, ccCommandLineInterface::ExportOption::ForceNoTimestamp);
  1690. if (!errorStr.isEmpty())
  1691. {
  1692. cmd.error(errorStr);
  1693. }
  1694. }
  1695. //add newDesc to the current pool
  1696. cmd.clouds().push_back(newDesc);
  1697. }
  1698. else
  1699. {
  1700. cmd.warning(QObject::tr("Failed to create component #%1! (not enough memory)").arg(j + 1));
  1701. }
  1702. }
  1703. delete compIndexes;
  1704. compIndexes = nullptr;
  1705. }
  1706. components.clear();
  1707. if (cmd.clouds().empty())
  1708. {
  1709. cmd.error(QObject::tr("No component was created! Check the minimum size..."));
  1710. }
  1711. else
  1712. {
  1713. cmd.print(QObject::tr("%1 component(s) were created").arg(cmd.clouds().size()));
  1714. }
  1715. }
  1716. if (progressDialog)
  1717. {
  1718. progressDialog->close();
  1719. QCoreApplication::processEvents();
  1720. }
  1721. }
  1722. catch (const std::bad_alloc&)
  1723. {
  1724. cmd.error(QObject::tr("Not enough memory"));
  1725. return false;
  1726. }
  1727. return true;
  1728. }
  1729. CommandCurvature::CommandCurvature()
  1730. : ccCommandLineInterface::Command(QObject::tr("Curvature"), COMMAND_CURVATURE)
  1731. {}
  1732. bool CommandCurvature::process(ccCommandLineInterface& cmd)
  1733. {
  1734. if (cmd.arguments().empty())
  1735. {
  1736. return cmd.error(QObject::tr("Missing parameter: curvature type after \"-%1\"").arg(COMMAND_CURVATURE));
  1737. }
  1738. QString curvTypeStr = cmd.arguments().takeFirst().toUpper();
  1739. CCCoreLib::Neighbourhood::CurvatureType curvType = CCCoreLib::Neighbourhood::MEAN_CURV;
  1740. if (curvTypeStr == "MEAN")
  1741. {
  1742. //curvType = CCCoreLib::Neighbourhood::MEAN_CURV;
  1743. }
  1744. else if (curvTypeStr == "GAUSS")
  1745. {
  1746. curvType = CCCoreLib::Neighbourhood::GAUSSIAN_CURV;
  1747. }
  1748. else if (curvTypeStr == "NORMAL_CHANGE")
  1749. {
  1750. curvType = CCCoreLib::Neighbourhood::NORMAL_CHANGE_RATE;
  1751. }
  1752. else
  1753. {
  1754. return cmd.error(QObject::tr("Invalid curvature type after \"-%1\". Got '%2' instead of MEAN or GAUSS.").arg(COMMAND_CURVATURE, curvTypeStr));
  1755. }
  1756. if (cmd.arguments().empty())
  1757. {
  1758. return cmd.error(QObject::tr("Missing parameter: kernel size after curvature type"));
  1759. }
  1760. bool paramOk = false;
  1761. QString kernelStr = cmd.arguments().takeFirst();
  1762. PointCoordinateType kernelSize = static_cast<PointCoordinateType>(kernelStr.toDouble(&paramOk));
  1763. if (!paramOk)
  1764. {
  1765. return cmd.error(QObject::tr("Failed to read a numerical parameter: kernel size (after curvature type). Got '%1' instead.").arg(kernelStr));
  1766. }
  1767. cmd.print(QObject::tr("\tKernel size: %1").arg(kernelSize));
  1768. if (cmd.clouds().empty())
  1769. {
  1770. return cmd.error(QObject::tr("No point cloud on which to compute curvature! (be sure to open one with \"-%1 [cloud filename]\" before \"-%2\")").arg(COMMAND_OPEN, COMMAND_CURVATURE));
  1771. }
  1772. //Call MainWindow generic method
  1773. ccHObject::Container entities;
  1774. entities.resize(cmd.clouds().size());
  1775. for (size_t i = 0; i < cmd.clouds().size(); ++i)
  1776. {
  1777. entities[i] = cmd.clouds()[i].pc;
  1778. }
  1779. if (ccLibAlgorithms::ComputeGeomCharacteristic(CCCoreLib::GeometricalAnalysisTools::Curvature, curvType, kernelSize, entities, nullptr, cmd.widgetParent()))
  1780. {
  1781. //save output
  1782. if (cmd.autoSaveMode() && !cmd.saveClouds(QObject::tr("%1_CURVATURE_KERNEL_%2").arg(curvTypeStr).arg(kernelSize)))
  1783. {
  1784. return false;
  1785. }
  1786. }
  1787. return true;
  1788. }
  1789. static bool ReadDensityType(ccCommandLineInterface& cmd, CCCoreLib::GeometricalAnalysisTools::Density& density)
  1790. {
  1791. if (cmd.arguments().empty())
  1792. {
  1793. return cmd.error(QObject::tr("Missing parameter: density type after \"-%1\" (KNN/SURFACE/VOLUME)").arg(COMMAND_DENSITY_TYPE));
  1794. }
  1795. //read option confirmed, we can move on
  1796. QString typeArg = cmd.arguments().takeFirst().toUpper();
  1797. if (typeArg == "KNN")
  1798. {
  1799. density = CCCoreLib::GeometricalAnalysisTools::DENSITY_KNN;
  1800. }
  1801. else if (typeArg == "SURFACE")
  1802. {
  1803. density = CCCoreLib::GeometricalAnalysisTools::DENSITY_2D;
  1804. }
  1805. else if (typeArg == "VOLUME")
  1806. {
  1807. density = CCCoreLib::GeometricalAnalysisTools::DENSITY_3D;
  1808. }
  1809. else
  1810. {
  1811. return cmd.error(QObject::tr("Invalid parameter: density type is expected after \"-%1\" (KNN/SURFACE/VOLUME)").arg(COMMAND_DENSITY_TYPE));
  1812. }
  1813. return true;
  1814. }
  1815. CommandApproxDensity::CommandApproxDensity()
  1816. : ccCommandLineInterface::Command(QObject::tr("Approx Density"), COMMAND_APPROX_DENSITY)
  1817. {}
  1818. bool CommandApproxDensity::process(ccCommandLineInterface& cmd)
  1819. {
  1820. if (cmd.clouds().empty())
  1821. {
  1822. return cmd.error(QObject::tr("No point cloud on which to compute approx. density! (be sure to open one with \"-%1 [cloud filename]\" before \"-%2\")").arg(COMMAND_OPEN, COMMAND_APPROX_DENSITY));
  1823. }
  1824. //Call MainWindow generic method
  1825. ccHObject::Container entities;
  1826. entities.resize(cmd.clouds().size());
  1827. for (size_t i = 0; i < cmd.clouds().size(); ++i)
  1828. {
  1829. entities[i] = cmd.clouds()[i].pc;
  1830. }
  1831. //optional parameter: density type
  1832. CCCoreLib::GeometricalAnalysisTools::Density densityType = CCCoreLib::GeometricalAnalysisTools::DENSITY_3D;
  1833. if (!cmd.arguments().empty())
  1834. {
  1835. QString argument = cmd.arguments().front();
  1836. if (ccCommandLineInterface::IsCommand(argument, COMMAND_DENSITY_TYPE))
  1837. {
  1838. //local option confirmed, we can move on
  1839. cmd.arguments().pop_front();
  1840. if (cmd.arguments().empty())
  1841. {
  1842. return cmd.error(QObject::tr("Missing parameter: density type after \"-%1\" (KNN/SURFACE/VOLUME)").arg(COMMAND_DENSITY_TYPE));
  1843. }
  1844. //read option confirmed, we can move on
  1845. if (!ReadDensityType(cmd, densityType))
  1846. {
  1847. return false;
  1848. }
  1849. }
  1850. }
  1851. if (ccLibAlgorithms::ComputeGeomCharacteristic(CCCoreLib::GeometricalAnalysisTools::ApproxLocalDensity, densityType, 0, entities, nullptr, cmd.widgetParent()))
  1852. {
  1853. //save output
  1854. if (cmd.autoSaveMode() && !cmd.saveClouds("APPROX_DENSITY"))
  1855. {
  1856. return false;
  1857. }
  1858. }
  1859. return true;
  1860. }
  1861. CommandDensity::CommandDensity()
  1862. : ccCommandLineInterface::Command(QObject::tr("Density"), COMMAND_DENSITY)
  1863. {}
  1864. bool CommandDensity::process(ccCommandLineInterface& cmd)
  1865. {
  1866. if (cmd.arguments().empty())
  1867. {
  1868. return cmd.error(QObject::tr("Missing parameter: sphere radius after \"-%1\"").arg(COMMAND_DENSITY));
  1869. }
  1870. bool paramOk = false;
  1871. QString kernelStr = cmd.arguments().takeFirst();
  1872. PointCoordinateType kernelSize = static_cast<PointCoordinateType>(kernelStr.toDouble(&paramOk));
  1873. if (!paramOk)
  1874. {
  1875. return cmd.error(QObject::tr("Failed to read a numerical parameter: sphere radius (after \"-%1\"). Got '%2' instead.").arg(COMMAND_DENSITY, kernelStr));
  1876. }
  1877. cmd.print(QObject::tr("\tSphere radius: %1").arg(kernelSize));
  1878. //optional parameter: density type
  1879. CCCoreLib::GeometricalAnalysisTools::Density densityType = CCCoreLib::GeometricalAnalysisTools::DENSITY_3D;
  1880. if (!cmd.arguments().empty())
  1881. {
  1882. QString argument = cmd.arguments().front();
  1883. if (ccCommandLineInterface::IsCommand(argument, COMMAND_DENSITY_TYPE))
  1884. {
  1885. //local option confirmed, we can move on
  1886. cmd.arguments().pop_front();
  1887. if (cmd.arguments().empty())
  1888. {
  1889. return cmd.error(QObject::tr("Missing parameter: density type after \"-%1\" (KNN/SURFACE/VOLUME)").arg(COMMAND_DENSITY_TYPE));
  1890. }
  1891. //read option confirmed, we can move on
  1892. if (!ReadDensityType(cmd, densityType))
  1893. {
  1894. return false;
  1895. }
  1896. }
  1897. }
  1898. if (cmd.clouds().empty())
  1899. {
  1900. return cmd.error(QObject::tr("No point cloud on which to compute density! (be sure to open one with \"-%1 [cloud filename]\" before \"-%2\")").arg(COMMAND_OPEN, COMMAND_DENSITY));
  1901. }
  1902. //Call MainWindow generic method
  1903. ccHObject::Container entities;
  1904. entities.resize(cmd.clouds().size());
  1905. for (size_t i = 0; i < cmd.clouds().size(); ++i)
  1906. {
  1907. entities[i] = cmd.clouds()[i].pc;
  1908. }
  1909. if (ccLibAlgorithms::ComputeGeomCharacteristic(CCCoreLib::GeometricalAnalysisTools::LocalDensity, densityType, kernelSize, entities, nullptr, cmd.widgetParent()))
  1910. {
  1911. //save output
  1912. if (cmd.autoSaveMode() && !cmd.saveClouds("DENSITY"))
  1913. {
  1914. return false;
  1915. }
  1916. }
  1917. return true;
  1918. }
  1919. CommandSFGradient::CommandSFGradient()
  1920. : ccCommandLineInterface::Command(QObject::tr("SF gradient"), COMMAND_SF_GRADIENT)
  1921. {}
  1922. bool CommandSFGradient::process(ccCommandLineInterface& cmd)
  1923. {
  1924. if (cmd.arguments().empty())
  1925. {
  1926. return cmd.error(QObject::tr("Missing parameter: boolean (whether SF is euclidean or not) after \"-%1\"").arg(COMMAND_SF_GRADIENT));
  1927. }
  1928. QString euclideanStr = cmd.arguments().takeFirst().toUpper();
  1929. bool euclidean = false;
  1930. if (euclideanStr == "TRUE")
  1931. {
  1932. euclidean = true;
  1933. }
  1934. else if (euclideanStr != "FALSE")
  1935. {
  1936. return cmd.error(QObject::tr("Invalid boolean value after \"-%1\". Got '%2' instead of TRUE or FALSE.").arg(COMMAND_SF_GRADIENT, euclideanStr));
  1937. }
  1938. if (cmd.clouds().empty())
  1939. {
  1940. return cmd.error(QObject::tr("No point cloud on which to compute SF gradient! (be sure to open one with \"-%1 [cloud filename]\" before \"-%2\")").arg(COMMAND_OPEN, COMMAND_SF_GRADIENT));
  1941. }
  1942. //Call MainWindow generic method
  1943. void* additionalParameters[1] = { &euclidean };
  1944. ccHObject::Container entities;
  1945. entities.reserve(cmd.clouds().size());
  1946. for (CLCloudDesc& desc : cmd.clouds())
  1947. {
  1948. unsigned sfCount = desc.pc->getNumberOfScalarFields();
  1949. if (sfCount == 0)
  1950. {
  1951. cmd.warning(QObject::tr("cmd.warning: cloud '%1' has no scalar field (it will be ignored)").arg(desc.pc->getName()));
  1952. }
  1953. else
  1954. {
  1955. if (sfCount > 1)
  1956. {
  1957. cmd.warning(QObject::tr("cmd.warning: cloud '%1' has several scalar fields (the active one will be used by default, or the first one if none is active)").arg(desc.pc->getName()));
  1958. }
  1959. int activeSFIndex = desc.pc->getCurrentOutScalarFieldIndex();
  1960. if (activeSFIndex < 0)
  1961. {
  1962. activeSFIndex = 0;
  1963. }
  1964. desc.pc->setCurrentDisplayedScalarField(activeSFIndex);
  1965. entities.push_back(desc.pc);
  1966. }
  1967. }
  1968. if (ccLibAlgorithms::ApplyCCLibAlgorithm(ccLibAlgorithms::CCLIB_ALGO_SF_GRADIENT, entities, cmd.widgetParent(), additionalParameters))
  1969. {
  1970. //save output
  1971. if (cmd.autoSaveMode() && !cmd.saveClouds(euclidean ? "EUCLIDEAN_SF_GRAD" : "SF_GRAD"))
  1972. {
  1973. return false;
  1974. }
  1975. }
  1976. return true;
  1977. }
  1978. CommandRoughness::CommandRoughness()
  1979. : ccCommandLineInterface::Command(QObject::tr("Roughness"), COMMAND_ROUGHNESS)
  1980. {}
  1981. bool CommandRoughness::process(ccCommandLineInterface& cmd)
  1982. {
  1983. if (cmd.arguments().empty())
  1984. {
  1985. return cmd.error(QObject::tr("Missing parameter: kernel size after \"-%1\"").arg(COMMAND_ROUGHNESS));
  1986. }
  1987. bool paramOk = false;
  1988. QString kernelStr = cmd.arguments().takeFirst();
  1989. PointCoordinateType kernelSize = static_cast<PointCoordinateType>(kernelStr.toDouble(&paramOk));
  1990. if (!paramOk)
  1991. {
  1992. return cmd.error(QObject::tr("Failed to read a numerical parameter: kernel size (after \"-%1\"). Got '%2' instead.").arg(COMMAND_ROUGHNESS, kernelStr));
  1993. }
  1994. cmd.print(QObject::tr("\tKernel size: %1").arg(kernelSize));
  1995. // optional argument
  1996. CCVector3 roughnessUpDir;
  1997. CCVector3* _roughnessUpDir = nullptr;
  1998. if (cmd.arguments().size() >= 4)
  1999. {
  2000. QString nextArg = cmd.arguments().first();
  2001. if (nextArg.startsWith('-') && nextArg.mid(1).toUpper() == COMMAND_ROUGHNESS_UP_DIR)
  2002. {
  2003. // option confirmed
  2004. cmd.arguments().takeFirst();
  2005. QString xStr = cmd.arguments().takeFirst();
  2006. QString yStr = cmd.arguments().takeFirst();
  2007. QString zStr = cmd.arguments().takeFirst();
  2008. bool okX = false, okY = false, okZ = false;
  2009. roughnessUpDir.x = static_cast<PointCoordinateType>(xStr.toDouble(&okX));
  2010. roughnessUpDir.y = static_cast<PointCoordinateType>(yStr.toDouble(&okY));
  2011. roughnessUpDir.z = static_cast<PointCoordinateType>(zStr.toDouble(&okZ));
  2012. if (!okX || !okY || !okZ)
  2013. {
  2014. return cmd.error(QObject::tr("Invalid 'up direction' vector after option -%1 (3 coordinates expected)").arg(COMMAND_ROUGHNESS_UP_DIR));
  2015. }
  2016. _roughnessUpDir = &roughnessUpDir;
  2017. }
  2018. }
  2019. if (cmd.clouds().empty())
  2020. {
  2021. return cmd.error(QObject::tr("No point cloud on which to compute roughness! (be sure to open one with \"-%1 [cloud filename]\" before \"-%2\")").arg(COMMAND_OPEN, COMMAND_ROUGHNESS));
  2022. }
  2023. //Call MainWindow generic method
  2024. ccHObject::Container entities;
  2025. entities.resize(cmd.clouds().size());
  2026. for (size_t i = 0; i < cmd.clouds().size(); ++i)
  2027. {
  2028. entities[i] = cmd.clouds()[i].pc;
  2029. }
  2030. if (ccLibAlgorithms::ComputeGeomCharacteristic(CCCoreLib::GeometricalAnalysisTools::Roughness, 0, kernelSize, entities, _roughnessUpDir, cmd.widgetParent()))
  2031. {
  2032. //save output
  2033. if (cmd.autoSaveMode() && !cmd.saveClouds(QObject::tr("ROUGHNESS_KERNEL_%2").arg(kernelSize)))
  2034. {
  2035. return false;
  2036. }
  2037. }
  2038. return true;
  2039. }
  2040. CommandApplyTransformation::CommandApplyTransformation()
  2041. : ccCommandLineInterface::Command(QObject::tr("Apply Transformation"), COMMAND_APPLY_TRANSFORMATION)
  2042. {}
  2043. bool CommandApplyTransformation::process(ccCommandLineInterface& cmd)
  2044. {
  2045. //optional parameters
  2046. bool inverse = false;
  2047. bool applyToGlobal = false;
  2048. bool forceApplyToGlobal = false;
  2049. while (!cmd.arguments().empty())
  2050. {
  2051. QString argument = cmd.arguments().front();
  2052. if (ccCommandLineInterface::IsCommand(argument, COMMAND_APPLY_TRANS_INVERSE))
  2053. {
  2054. //local option confirmed, we can move on
  2055. inverse = true;
  2056. cmd.arguments().pop_front();
  2057. cmd.print("Transformation matrix will be inversed");
  2058. }
  2059. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_APPLY_TRANS_TO_GLOBAL))
  2060. {
  2061. //local option confirmed, we can move on
  2062. applyToGlobal = true;
  2063. cmd.arguments().pop_front();
  2064. cmd.print("Transformation will be applied to the global coordinates (Global Shift may be automatically adjusted to preserve accuracy)");
  2065. //look for the 'FORCE' sub-option
  2066. if (!cmd.arguments().empty() && cmd.arguments().front().toUpper() == OPTION_FORCE)
  2067. {
  2068. //local option confirmed, we can move on
  2069. forceApplyToGlobal = true;
  2070. cmd.arguments().pop_front();
  2071. cmd.print("Transformation will be applied to the global coordinates even if the entity already has large coordinates");
  2072. }
  2073. }
  2074. else
  2075. {
  2076. break;
  2077. }
  2078. }
  2079. if (cmd.arguments().empty())
  2080. {
  2081. return cmd.error(QObject::tr("Missing parameter: transformation file after \"-%1\"").arg(COMMAND_APPLY_TRANSFORMATION));
  2082. }
  2083. QString filename = cmd.arguments().takeFirst();
  2084. ccGLMatrixd mat;
  2085. if (!mat.fromAsciiFile(filename))
  2086. {
  2087. return cmd.error(QObject::tr("Failed to read transformation matrix file '%1'!").arg(filename));
  2088. }
  2089. if (inverse)
  2090. {
  2091. cmd.print(QObject::tr("Transformation before inversion:\n") + mat.toString());
  2092. mat = mat.inverse();
  2093. }
  2094. cmd.print(QObject::tr("Transformation:\n") + mat.toString());
  2095. if (cmd.clouds().empty() && cmd.meshes().empty())
  2096. {
  2097. return cmd.error(QObject::tr("No entity on which to apply the transformation! (be sure to open one with \"-%1 [filename]\" before \"-%2\")").arg(COMMAND_OPEN, COMMAND_APPLY_TRANSFORMATION));
  2098. }
  2099. //create an entity vector
  2100. size_t nrOfClouds = cmd.clouds().size();
  2101. size_t nrOfMeshes = cmd.meshes().size();
  2102. size_t nrOfEntities = nrOfClouds + nrOfMeshes;
  2103. std::vector<std::pair<ccShiftedObject*, CLEntityDesc*>> entities;
  2104. entities.reserve(nrOfEntities);
  2105. //add clouds to the vector
  2106. for (CLCloudDesc& desc : cmd.clouds())
  2107. {
  2108. entities.push_back({ desc.pc, &desc });
  2109. }
  2110. //add meshes to the vector
  2111. for (CLMeshDesc& desc : cmd.meshes())
  2112. {
  2113. bool isLocked = false;
  2114. ccShiftedObject* shifted = ccHObjectCaster::ToShifted(desc.mesh, &isLocked);
  2115. if (shifted && !isLocked)
  2116. {
  2117. entities.push_back({ shifted, &desc });
  2118. }
  2119. }
  2120. //if the transformation is partly converted to global shift/scale
  2121. bool autoApplyPreviousGlobalShiftAndScale = false;
  2122. double previousScale = 1.0;
  2123. CCVector3d previousShift(0, 0, 0);
  2124. //process both clouds and meshes
  2125. for (size_t i = 0; i < nrOfEntities; i++)
  2126. {
  2127. CLEntityDesc& desc = *entities[i].second;
  2128. ccShiftedObject* shiftedEntity = entities[i].first;
  2129. ccGLMatrixd transMat(mat);
  2130. if (applyToGlobal)
  2131. {
  2132. // the user wants to apply the transformation to the global coordinates
  2133. CCVector3d globalShift = shiftedEntity->getGlobalShift();
  2134. double globalScale = shiftedEntity->getGlobalScale();
  2135. // we compute the impact to the local coordinate system without changing the
  2136. // actual Global Shift & Scale parameters (for now)
  2137. CCVector3d localTranslation = globalScale * (globalShift - transMat * globalShift + 2 * transMat.getTranslationAsVec3D());
  2138. // we switch to a local transformation matrix
  2139. transMat.setTranslation(localTranslation);
  2140. //test if the translated cloud coordinates were already "too large"
  2141. ccBBox localBBox = shiftedEntity->getOwnBB();
  2142. CCVector3d Pl = localBBox.minCorner();
  2143. double Dl = localBBox.getDiagNormd();
  2144. if (forceApplyToGlobal || (!ccGlobalShiftManager::NeedShift(Pl) && !ccGlobalShiftManager::NeedRescale(Dl)))
  2145. {
  2146. //test if the translated (local) cloud coordinates are too large
  2147. ccBBox transformedLocalBox = localBBox * transMat;
  2148. CCVector3d transformedPl = transformedLocalBox.minCorner();
  2149. double transformedDl = transformedLocalBox.getDiagNormd();
  2150. bool needShift = ccGlobalShiftManager::NeedShift(transformedPl) || ccGlobalShiftManager::NeedRescale(transformedDl);
  2151. if (needShift)
  2152. {
  2153. //existing shift information
  2154. CCVector3d globalShift = shiftedEntity->getGlobalShift();
  2155. double globalScale = shiftedEntity->getGlobalScale();
  2156. //we compute the global coordinates and scale of the reference point (= the min corner of the bounding-box)
  2157. CCVector3d Pg = shiftedEntity->toGlobal3d(transformedPl);
  2158. double Dg = transformedDl / globalScale;
  2159. //let's try to find better Global Shift and Scale values
  2160. CCVector3d newShift(0.0, 0.0, 0.0);
  2161. double newScale = 1.0;
  2162. //should we try to use the previous Global Shift and Scale values?
  2163. if (autoApplyPreviousGlobalShiftAndScale)
  2164. {
  2165. if (!ccGlobalShiftManager::NeedShift(Pg + previousShift) && !ccGlobalShiftManager::NeedRescale(Dg * previousScale))
  2166. {
  2167. newScale = previousScale;
  2168. newShift = previousShift;
  2169. needShift = false;
  2170. }
  2171. }
  2172. if (needShift)
  2173. {
  2174. //don't bother the user as we are running in command line, use the best possible solution
  2175. newShift = ccGlobalShiftManager::BestShift(Pg);
  2176. newScale = ccGlobalShiftManager::BestScale(Dg);
  2177. needShift = false;
  2178. //force this shift for upcoming entities
  2179. autoApplyPreviousGlobalShiftAndScale = true;
  2180. previousShift = newShift;
  2181. previousScale = newScale;
  2182. }
  2183. //get the relative modification to existing global shift/scale info
  2184. double scaleChange = newScale / globalScale;
  2185. CCVector3d shiftChange = newShift - globalShift;
  2186. if (scaleChange != 1.0 || shiftChange.norm2() != 0)
  2187. {
  2188. //apply translation as global shift
  2189. shiftedEntity->setGlobalShift(newShift);
  2190. shiftedEntity->setGlobalScale(newScale);
  2191. cmd.warning(QObject::tr("Entity '%1' global shift/scale information has been updated: shift = (%2,%3,%4) / scale = %5")
  2192. .arg(shiftedEntity->getName())
  2193. .arg(newShift.x)
  2194. .arg(newShift.y)
  2195. .arg(newShift.z)
  2196. .arg(newScale));
  2197. transMat.scaleRotation(scaleChange);
  2198. transMat.setTranslation(transMat.getTranslationAsVec3D() + newScale * shiftChange);
  2199. }
  2200. }
  2201. }
  2202. else
  2203. {
  2204. cmd.warning(QObject::tr("Entity '%1' already has very large local coordinates. Global shift/scale won't be automatically adjusted to preserve accuracy. Consider using the -%2 option to force global shift/scale adjustment.").arg(shiftedEntity->getName()).arg(OPTION_FORCE));
  2205. }
  2206. }
  2207. else
  2208. {
  2209. // check if the transformed coordinates are too large
  2210. ccBBox transformedLocalBox = shiftedEntity->getOwnBB() * transMat;
  2211. CCVector3d transformedPl = transformedLocalBox.minCorner();
  2212. double transformedDl = transformedLocalBox.getDiagNormd();
  2213. bool needShift = ccGlobalShiftManager::NeedShift(transformedPl) || ccGlobalShiftManager::NeedRescale(transformedDl);
  2214. if (needShift)
  2215. {
  2216. cmd.warning(QObject::tr("Entity '%1' will have very large local coordinates after transformation. Consider using the -%1 option to preserve accuracy.").arg(COMMAND_APPLY_TRANS_TO_GLOBAL));
  2217. }
  2218. }
  2219. ccGLMatrix matrixToApply(transMat.data());
  2220. shiftedEntity->applyGLTransformation_recursive(&matrixToApply);
  2221. QString nameSuffix = "_TRANSFORMED";
  2222. if ((&desc)->getCLEntityType() == CL_ENTITY_TYPE::MESH)
  2223. {
  2224. //set the mesh name instead of the vertices cloud inside the mesh
  2225. ccHObject* parent = shiftedEntity->getParent();
  2226. if (parent)
  2227. {
  2228. parent->setName(QObject::tr("%1%2").arg(parent->getName()).arg(nameSuffix));
  2229. }
  2230. }
  2231. else
  2232. {
  2233. shiftedEntity->setName(QObject::tr("%1%2").arg(shiftedEntity->getName()).arg(nameSuffix));
  2234. }
  2235. desc.basename += nameSuffix;
  2236. //save it as well
  2237. if (cmd.autoSaveMode())
  2238. {
  2239. QString errorStr = cmd.exportEntity(desc);
  2240. if (!errorStr.isEmpty())
  2241. {
  2242. return cmd.error(errorStr);
  2243. }
  2244. }
  2245. }
  2246. return true;
  2247. }
  2248. CommandDropGlobalShift::CommandDropGlobalShift()
  2249. : ccCommandLineInterface::Command(QObject::tr("Drop global shift"), COMMAND_DROP_GLOBAL_SHIFT)
  2250. {}
  2251. bool CommandDropGlobalShift::process(ccCommandLineInterface& cmd)
  2252. {
  2253. if (cmd.clouds().empty() && cmd.meshes().empty())
  2254. {
  2255. return cmd.error(QObject::tr("No loaded entity! (be sure to open one with \"-%1 [filename]\" before \"-%2\")").arg(COMMAND_OPEN, COMMAND_DROP_GLOBAL_SHIFT));
  2256. }
  2257. //process clouds
  2258. for (const CLCloudDesc& desc : cmd.clouds())
  2259. {
  2260. desc.pc->setGlobalShift(0, 0, 0);
  2261. }
  2262. for (const CLMeshDesc& desc : cmd.meshes())
  2263. {
  2264. bool isLocked = false;
  2265. ccShiftedObject* shifted = ccHObjectCaster::ToShifted(desc.mesh, &isLocked);
  2266. if (shifted && !isLocked)
  2267. {
  2268. shifted->setGlobalShift(0, 0, 0);
  2269. }
  2270. }
  2271. return true;
  2272. }
  2273. CommandSFColorScale::CommandSFColorScale()
  2274. : ccCommandLineInterface::Command(QObject::tr("SF color scale"), COMMAND_SF_COLOR_SCALE)
  2275. {}
  2276. bool CommandSFColorScale::process(ccCommandLineInterface& cmd)
  2277. {
  2278. if (cmd.arguments().empty())
  2279. {
  2280. return cmd.error(QObject::tr("Missing parameter: color scale file after \"-%1\"").arg(COMMAND_SF_COLOR_SCALE));
  2281. }
  2282. QString filename = cmd.arguments().takeFirst();
  2283. ccColorScale::Shared scale = ccColorScale::LoadFromXML(filename);
  2284. if (!scale)
  2285. {
  2286. return cmd.error(QObject::tr("Failed to read color scale file '%1'!").arg(filename));
  2287. }
  2288. if (cmd.clouds().empty() && cmd.meshes().empty())
  2289. {
  2290. return cmd.error(QObject::tr("No point cloud or mesh on which to set the SF color scale! (be sure to open one with \"-%1 [cloud filename]\" before \"-%2\")").arg(COMMAND_OPEN, COMMAND_SF_COLOR_SCALE));
  2291. }
  2292. // clouds
  2293. if (!cmd.clouds().empty())
  2294. {
  2295. bool hasCandidateClouds = false;
  2296. for (auto& cloud : cmd.clouds())
  2297. {
  2298. ccScalarField* sf = static_cast<ccScalarField*>(cloud.pc->getCurrentOutScalarField());
  2299. if (sf)
  2300. {
  2301. sf->setColorScale(scale);
  2302. hasCandidateClouds = true;
  2303. }
  2304. }
  2305. if (hasCandidateClouds && cmd.autoSaveMode() && !cmd.saveClouds("COLOR_SCALE"))
  2306. {
  2307. return false;
  2308. }
  2309. }
  2310. // meshes
  2311. if (!cmd.meshes().empty())
  2312. {
  2313. bool hasCandidateMeshes = false;
  2314. for (auto& mesh : cmd.meshes())
  2315. {
  2316. ccPointCloud* vertices = dynamic_cast<ccPointCloud*>(mesh.mesh->getAssociatedCloud());
  2317. if (vertices)
  2318. {
  2319. ccScalarField* sf = static_cast<ccScalarField*>(vertices->getCurrentOutScalarField());
  2320. if (sf)
  2321. {
  2322. sf->setColorScale(scale);
  2323. hasCandidateMeshes = true;
  2324. }
  2325. }
  2326. }
  2327. if (hasCandidateMeshes && cmd.autoSaveMode() && !cmd.saveMeshes("COLOR_SCALE"))
  2328. {
  2329. return false;
  2330. }
  2331. }
  2332. return true;
  2333. }
  2334. CommandSFConvertToRGB::CommandSFConvertToRGB()
  2335. : ccCommandLineInterface::Command(QObject::tr("SF convert to RGB"), COMMAND_SF_CONVERT_TO_RGB)
  2336. {}
  2337. bool CommandSFConvertToRGB::process(ccCommandLineInterface& cmd)
  2338. {
  2339. if (cmd.arguments().empty())
  2340. {
  2341. return cmd.error(QObject::tr("Missing parameter: boolean (whether to mix with existing colors or not) after \"-%1\"").arg(COMMAND_SF_CONVERT_TO_RGB));
  2342. }
  2343. QString mixWithExistingColorsStr = cmd.arguments().takeFirst().toUpper();
  2344. bool mixWithExistingColors = false;
  2345. if (mixWithExistingColorsStr == "TRUE")
  2346. {
  2347. mixWithExistingColors = true;
  2348. }
  2349. else if (mixWithExistingColorsStr != "FALSE")
  2350. {
  2351. return cmd.error(QObject::tr("Invalid boolean value after \"-%1\". Got '%2' instead of TRUE or FALSE.").arg(COMMAND_SF_CONVERT_TO_RGB, mixWithExistingColorsStr));
  2352. }
  2353. if (cmd.clouds().empty())
  2354. {
  2355. return cmd.error(QObject::tr("No point cloud on which to convert SF to RGB! (be sure to open one with \"-%1 [cloud filename]\" before \"-%2\")").arg(COMMAND_OPEN, COMMAND_SF_CONVERT_TO_RGB));
  2356. }
  2357. for (CLCloudDesc& desc : cmd.clouds())
  2358. {
  2359. unsigned sfCount = desc.pc->getNumberOfScalarFields();
  2360. int activeSFIndex = desc.pc->getCurrentOutScalarFieldIndex();
  2361. if (sfCount == 0)
  2362. {
  2363. cmd.warning(QObject::tr("cmd.warning: cloud '%1' has no scalar field (it will be ignored)").arg(desc.pc->getName()));
  2364. }
  2365. else if (activeSFIndex < 0)
  2366. {
  2367. cmd.warning(QObject::tr("cmd.warning: cloud '%1' has no active scalar field (it will be ignored)").arg(desc.pc->getName()));
  2368. }
  2369. else
  2370. {
  2371. int displaySFIndex = desc.pc->getCurrentDisplayedScalarFieldIndex();
  2372. desc.pc->setCurrentDisplayedScalarField(activeSFIndex);
  2373. if (desc.pc->convertCurrentScalarFieldToColors(mixWithExistingColors))
  2374. {
  2375. desc.pc->showColors(true);
  2376. desc.pc->showSF(false);
  2377. }
  2378. else
  2379. {
  2380. cmd.warning(QObject::tr("cmd.warning: cloud '%1' failed to convert SF to RGB").arg(desc.pc->getName()));
  2381. }
  2382. desc.pc->setCurrentDisplayedScalarField(displaySFIndex);
  2383. }
  2384. }
  2385. if (cmd.autoSaveMode() && !cmd.saveClouds("SF_CONVERT_TO_RGB"))
  2386. {
  2387. return false;
  2388. }
  2389. return true;
  2390. }
  2391. CommandRGBConvertToSF::CommandRGBConvertToSF()
  2392. : ccCommandLineInterface::Command(QObject::tr("RGB convert to SF"), COMMAND_RGB_CONVERT_TO_SF)
  2393. {}
  2394. bool CommandRGBConvertToSF::process(ccCommandLineInterface& cmd)
  2395. {
  2396. if (cmd.clouds().empty())
  2397. {
  2398. return cmd.error(QObject::tr("No point cloud on which to convert RGB to SF! (be sure to open one with \"-%1 [cloud filename]\" before \"-%2\")").arg(COMMAND_OPEN, COMMAND_RGB_CONVERT_TO_SF));
  2399. }
  2400. for (CLCloudDesc& desc : cmd.clouds())
  2401. {
  2402. if (!desc.pc->hasColors())
  2403. {
  2404. cmd.warning(QObject::tr("Cloud %1 has no colors").arg(desc.pc->getName()));
  2405. continue;
  2406. }
  2407. ccHObject::Container container;
  2408. container.push_back(desc.pc);
  2409. if (!ccEntityAction::sfFromColor(container, /*exportR=*/true, /*exportG=*/true, /*exportB=*/true, /*exportAlpha=*/true, /*exportC=*/true)) //beta version, only composite
  2410. {
  2411. return cmd.error(QObject::tr("Failed to convert RGB to scalar fields"));
  2412. }
  2413. if (cmd.autoSaveMode())
  2414. {
  2415. QString errorStr = cmd.exportEntity(desc, "_RGB_TO_SF");
  2416. if (!errorStr.isEmpty())
  2417. {
  2418. return cmd.error(errorStr);
  2419. }
  2420. }
  2421. }
  2422. return true;
  2423. }
  2424. CommandFilterBySFValue::CommandFilterBySFValue()
  2425. : ccCommandLineInterface::Command(QObject::tr("Filter by SF value"), COMMAND_FILTER_SF_BY_VALUE)
  2426. {}
  2427. //special SF values that can be used instead of explicit ones
  2428. enum USE_SPECIAL_SF_VALUE
  2429. {
  2430. USE_NONE,
  2431. USE_MIN,
  2432. USE_DISP_MIN,
  2433. USE_SAT_MIN,
  2434. USE_N_SIGMA_MIN,
  2435. USE_MAX,
  2436. USE_DISP_MAX,
  2437. USE_SAT_MAX,
  2438. USE_N_SIGMA_MAX
  2439. };
  2440. static std::pair<ScalarType, ScalarType> GetSFRange(const CCCoreLib::ScalarField& sf,
  2441. ScalarType minVal,
  2442. USE_SPECIAL_SF_VALUE useValForMin,
  2443. ScalarType maxVal,
  2444. USE_SPECIAL_SF_VALUE useValForMax)
  2445. {
  2446. ScalarType thisMinVal = minVal;
  2447. {
  2448. switch (useValForMin)
  2449. {
  2450. case USE_MIN:
  2451. thisMinVal = sf.getMin();
  2452. break;
  2453. case USE_DISP_MIN:
  2454. thisMinVal = static_cast<const ccScalarField&>(sf).displayRange().start();
  2455. break;
  2456. case USE_SAT_MIN:
  2457. thisMinVal = static_cast<const ccScalarField&>(sf).saturationRange().start();
  2458. break;
  2459. case USE_N_SIGMA_MIN:
  2460. ScalarType mean;
  2461. ScalarType variance;
  2462. sf.computeMeanAndVariance(mean, &variance);
  2463. thisMinVal = mean - (sqrt(variance) * minVal);
  2464. break;
  2465. default:
  2466. //nothing to do
  2467. break;
  2468. }
  2469. }
  2470. ScalarType thisMaxVal = maxVal;
  2471. {
  2472. switch (useValForMax)
  2473. {
  2474. case USE_MAX:
  2475. thisMaxVal = sf.getMax();
  2476. break;
  2477. case USE_DISP_MAX:
  2478. thisMaxVal = static_cast<const ccScalarField&>(sf).displayRange().stop();
  2479. break;
  2480. case USE_SAT_MAX:
  2481. thisMaxVal = static_cast<const ccScalarField&>(sf).saturationRange().stop();
  2482. break;
  2483. case USE_N_SIGMA_MAX:
  2484. ScalarType mean;
  2485. ScalarType variance;
  2486. sf.computeMeanAndVariance(mean, &variance);
  2487. thisMaxVal = mean + (sqrt(variance) * maxVal);
  2488. break;
  2489. default:
  2490. //nothing to do
  2491. break;
  2492. }
  2493. }
  2494. return { thisMinVal, thisMaxVal };
  2495. }
  2496. static ScalarType GetSFValue(const ccPointCloud& pc, int sfIndex, ScalarType value, USE_SPECIAL_SF_VALUE useVal)
  2497. {
  2498. CCCoreLib::ScalarField* sf = pc.getScalarField(sfIndex);
  2499. //should be handled way before this point this is just safety
  2500. if (sf)
  2501. {
  2502. std::pair<ScalarType, ScalarType> range = GetSFRange(*sf, value, useVal, value, useVal);
  2503. if (useVal <= USE_N_SIGMA_MIN)
  2504. {
  2505. return range.first;
  2506. }
  2507. else
  2508. {
  2509. return range.second;
  2510. }
  2511. }
  2512. return 1.0;
  2513. }
  2514. static USE_SPECIAL_SF_VALUE ToSpecialSFValue(QString valString)
  2515. {
  2516. valString = valString.toUpper();
  2517. if (valString == "MIN")
  2518. {
  2519. return USE_MIN;
  2520. }
  2521. else if (valString == "DISP_MIN")
  2522. {
  2523. return USE_DISP_MIN;
  2524. }
  2525. else if (valString == "SAT_MIN")
  2526. {
  2527. return USE_SAT_MIN;
  2528. }
  2529. else if (valString == "N_SIGMA_MIN")
  2530. {
  2531. return USE_N_SIGMA_MIN;
  2532. }
  2533. else if (valString == "MAX")
  2534. {
  2535. return USE_MAX;
  2536. }
  2537. else if (valString == "DISP_MAX")
  2538. {
  2539. return USE_DISP_MAX;
  2540. }
  2541. else if (valString == "SAT_MAX")
  2542. {
  2543. return USE_SAT_MAX;
  2544. }
  2545. else if (valString == "N_SIGMA_MAX")
  2546. {
  2547. return USE_N_SIGMA_MAX;
  2548. }
  2549. else
  2550. {
  2551. return USE_NONE;
  2552. }
  2553. }
  2554. bool CommandFilterBySFValue::process(ccCommandLineInterface& cmd)
  2555. {
  2556. USE_SPECIAL_SF_VALUE useValForMin = USE_NONE;
  2557. ScalarType minVal = 0;
  2558. QString minValStr;
  2559. {
  2560. if (cmd.arguments().empty())
  2561. {
  2562. return cmd.error(QObject::tr("Missing parameter: min value after \"-%1\"").arg(COMMAND_FILTER_SF_BY_VALUE));
  2563. }
  2564. bool paramOk = false;
  2565. minValStr = cmd.arguments().takeFirst();
  2566. useValForMin = ToSpecialSFValue(minValStr);
  2567. if (useValForMin == USE_N_SIGMA_MIN)
  2568. {
  2569. if (cmd.arguments().empty())
  2570. {
  2571. return cmd.error(QObject::tr("Missing parameter: N value (after \"-%1 N_SIGMA_MIN\").").arg(COMMAND_FILTER_SF_BY_VALUE));
  2572. }
  2573. minValStr = cmd.arguments().takeFirst();
  2574. minVal = static_cast<ScalarType>(minValStr.toDouble(&paramOk));
  2575. if (!paramOk)
  2576. {
  2577. return cmd.error(QObject::tr("Failed to read a numerical parameter: N value (after \"N_SIGMA_MIN\"). Got '%2' instead.").arg(minValStr));
  2578. }
  2579. }
  2580. else if (useValForMin == USE_N_SIGMA_MAX)
  2581. {
  2582. if (cmd.arguments().empty())
  2583. {
  2584. return cmd.error(QObject::tr("Missing parameter: N value (after \"-%1 N_SIGMA_MAX\").").arg(COMMAND_FILTER_SF_BY_VALUE));
  2585. }
  2586. minValStr = cmd.arguments().takeFirst();
  2587. minVal = static_cast<ScalarType>(minValStr.toDouble(&paramOk));
  2588. if (!paramOk)
  2589. {
  2590. return cmd.error(QObject::tr("Failed to read a numerical parameter: N value (after \"N_SIGMA_MAX\"). Got '%2' instead.").arg(minValStr));
  2591. }
  2592. }
  2593. else if (useValForMin == USE_NONE)
  2594. {
  2595. minVal = static_cast<ScalarType>(minValStr.toDouble(&paramOk));
  2596. if (!paramOk)
  2597. {
  2598. return cmd.error(QObject::tr("Failed to read a numerical parameter: min value (after \"-%1\"). Got '%2' instead.").arg(COMMAND_FILTER_SF_BY_VALUE, minValStr));
  2599. }
  2600. }
  2601. }
  2602. USE_SPECIAL_SF_VALUE useValForMax = USE_NONE;
  2603. ScalarType maxVal = 0;
  2604. QString maxValStr;
  2605. {
  2606. if (cmd.arguments().empty())
  2607. {
  2608. return cmd.error(QObject::tr("Missing parameter: max value after \"-%1\" {min}").arg(COMMAND_FILTER_SF_BY_VALUE));
  2609. }
  2610. bool paramOk = false;
  2611. maxValStr = cmd.arguments().takeFirst();
  2612. useValForMax = ToSpecialSFValue(maxValStr);
  2613. if (useValForMax == USE_N_SIGMA_MIN)
  2614. {
  2615. if (cmd.arguments().empty())
  2616. {
  2617. return cmd.error(QObject::tr("Missing parameter: N value (after \"-%1 XXX N_SIGMA_MIN\").").arg(COMMAND_FILTER_SF_BY_VALUE));
  2618. }
  2619. maxValStr = cmd.arguments().takeFirst();
  2620. maxVal = static_cast<ScalarType>(maxValStr.toDouble(&paramOk));
  2621. if (!paramOk)
  2622. {
  2623. return cmd.error(QObject::tr("Failed to read a numerical parameter: N value (after \"N_SIGMA_MIN\"). Got '%2' instead.").arg(maxValStr));
  2624. }
  2625. }
  2626. else if (useValForMax == USE_N_SIGMA_MAX)
  2627. {
  2628. if (cmd.arguments().empty())
  2629. {
  2630. return cmd.error(QObject::tr("Missing parameter: N value (after \"-%1 XXX N_SIGMA_MAX\").").arg(COMMAND_FILTER_SF_BY_VALUE));
  2631. }
  2632. maxValStr = cmd.arguments().takeFirst();
  2633. maxVal = static_cast<ScalarType>(maxValStr.toDouble(&paramOk));
  2634. if (!paramOk)
  2635. {
  2636. return cmd.error(QObject::tr("Failed to read a numerical parameter: N value (after \"N_SIGMA_MAX\"). Got '%2' instead.").arg(maxValStr));
  2637. }
  2638. }
  2639. else if (useValForMax == USE_NONE)
  2640. {
  2641. maxVal = static_cast<ScalarType>(maxValStr.toDouble(&paramOk));
  2642. if (!paramOk)
  2643. {
  2644. return cmd.error(QObject::tr("Failed to read a numerical parameter: max value (after min value). Got '%1' instead.").arg(COMMAND_FILTER_SF_BY_VALUE, maxValStr));
  2645. }
  2646. }
  2647. }
  2648. cmd.print(QObject::tr("\tInterval: [%1 - %2]").arg(minValStr, maxValStr));
  2649. if (cmd.clouds().empty() && cmd.meshes().empty())
  2650. {
  2651. return cmd.error(QObject::tr("No point cloud nor mesh on which to filter SF! (be sure to open one or generate one with \"-%1 [cloud filename]\" before \"-%2\")").arg(COMMAND_OPEN, COMMAND_FILTER_SF_BY_VALUE));
  2652. }
  2653. // for each cloud
  2654. for (CLCloudDesc& desc : cmd.clouds())
  2655. {
  2656. CCCoreLib::ScalarField* sf = desc.pc->getCurrentOutScalarField();
  2657. if (sf)
  2658. {
  2659. std::pair<ScalarType, ScalarType> range = GetSFRange(*sf, minVal, useValForMin, maxVal, useValForMax);
  2660. ccPointCloud* fitleredCloud = desc.pc->filterPointsByScalarValue(range.first, range.second);
  2661. if (fitleredCloud)
  2662. {
  2663. cmd.print(QObject::tr("\t\tCloud '%1' --> %2/%3 points remaining").arg(desc.pc->getName()).arg(fitleredCloud->size()).arg(desc.pc->size()));
  2664. if (fitleredCloud != desc.pc)
  2665. {
  2666. //replace current cloud by this one
  2667. delete desc.pc;
  2668. desc.pc = fitleredCloud;
  2669. }
  2670. desc.basename += QObject::tr("_FILTERED_[%1_%2]").arg(range.first).arg(range.second);
  2671. if (cmd.autoSaveMode())
  2672. {
  2673. QString errorStr = cmd.exportEntity(desc);
  2674. if (!errorStr.isEmpty())
  2675. {
  2676. return cmd.error(errorStr);
  2677. }
  2678. }
  2679. }
  2680. }
  2681. }
  2682. // for each mesh
  2683. for (CLMeshDesc& desc : cmd.meshes())
  2684. {
  2685. ccGenericMesh* mesh = desc.mesh;
  2686. ccPointCloud* pc = ccHObjectCaster::ToPointCloud(mesh);
  2687. if (!pc)
  2688. {
  2689. // strange mesh
  2690. continue;
  2691. }
  2692. CCCoreLib::ScalarField* sf = pc->getCurrentOutScalarField();
  2693. if (sf)
  2694. {
  2695. std::pair<ScalarType, ScalarType> range = GetSFRange(*sf, minVal, useValForMin, maxVal, useValForMax);
  2696. pc->hidePointsByScalarValue(range.first, range.second);
  2697. ccGenericMesh* filteredMesh = nullptr;
  2698. if (mesh->isA(CC_TYPES::MESH)/*|| ent->isKindOf(CC_TYPES::PRIMITIVE)*/) //TODO
  2699. filteredMesh = ccHObjectCaster::ToMesh(mesh)->createNewMeshFromSelection(false);
  2700. else if (mesh->isA(CC_TYPES::SUB_MESH))
  2701. filteredMesh = ccHObjectCaster::ToSubMesh(mesh)->createNewSubMeshFromSelection(false);
  2702. else
  2703. {
  2704. cmd.warning("Unhandled mesh type for entitiy " + mesh->getName());
  2705. continue;
  2706. }
  2707. if (filteredMesh)
  2708. {
  2709. cmd.print(QObject::tr("\t\tMesh '%1' --> %2/%3 triangles remaining").arg(mesh->getName()).arg(filteredMesh->size()).arg(mesh->size()));
  2710. //replace current mesh by this one
  2711. if (filteredMesh != mesh) //it's technically possible to have the same pointer if all triangles were filtered (with 'createNewMeshFromSelection')
  2712. {
  2713. delete mesh;
  2714. mesh = nullptr;
  2715. desc.mesh = filteredMesh;
  2716. }
  2717. desc.basename += QObject::tr("_FILTERED_[%1_%2]").arg(range.first).arg(range.second);
  2718. if (cmd.autoSaveMode())
  2719. {
  2720. QString errorStr = cmd.exportEntity(desc);
  2721. if (!errorStr.isEmpty())
  2722. {
  2723. return cmd.error(errorStr);
  2724. }
  2725. }
  2726. }
  2727. }
  2728. }
  2729. return true;
  2730. }
  2731. CommandComputeMeshVolume::CommandComputeMeshVolume()
  2732. : ccCommandLineInterface::Command(QObject::tr("Compute mesh volume"), COMMAND_MESH_VOLUME)
  2733. {}
  2734. bool CommandComputeMeshVolume::process(ccCommandLineInterface& cmd)
  2735. {
  2736. if (cmd.meshes().empty())
  2737. {
  2738. cmd.warning(QObject::tr("No mesh loaded! Nothing to do..."));
  2739. return true;
  2740. }
  2741. //optional parameters
  2742. QString outputFilename;
  2743. if (!cmd.arguments().empty())
  2744. {
  2745. QString argument = cmd.arguments().front();
  2746. if (ccCommandLineInterface::IsCommand(argument, COMMAND_VOLUME_TO_FILE))
  2747. {
  2748. //local option confirmed, we can move on
  2749. cmd.arguments().pop_front();
  2750. if (!cmd.arguments().empty())
  2751. {
  2752. outputFilename = cmd.arguments().front();
  2753. cmd.arguments().pop_front();
  2754. cmd.print(QObject::tr("Volume report file: %1").arg(outputFilename));
  2755. }
  2756. else
  2757. {
  2758. return cmd.error(QObject::tr("Missing argument: filename after '%1'").arg(COMMAND_VOLUME_TO_FILE));
  2759. }
  2760. }
  2761. }
  2762. QFile outFile;
  2763. QTextStream outStream(&outFile);
  2764. if (!outputFilename.isEmpty())
  2765. {
  2766. outFile.setFileName(outputFilename);
  2767. if (!outFile.open(QFile::WriteOnly | QFile::Text))
  2768. {
  2769. return cmd.error(QObject::tr("Failed to create/open volume report file"));
  2770. }
  2771. }
  2772. //for each mesh
  2773. for (CLMeshDesc& desc : cmd.meshes())
  2774. {
  2775. //we compute the mesh volume
  2776. double V = CCCoreLib::MeshSamplingTools::computeMeshVolume(desc.mesh);
  2777. QString titleStr = QObject::tr("Mesh '%1'").arg(desc.basename);
  2778. if (desc.indexInFile >= 0)
  2779. {
  2780. titleStr += QObject::tr(" (#%2)").arg(desc.indexInFile);
  2781. }
  2782. cmd.print(titleStr);
  2783. QString volumeStr = QObject::tr("V = %2").arg(V, 0, 'f', 8);
  2784. cmd.print(volumeStr);
  2785. if (outFile.isOpen())
  2786. {
  2787. outStream << titleStr << endl;
  2788. outStream << volumeStr << endl;
  2789. }
  2790. }
  2791. return true;
  2792. }
  2793. CommandMergeMeshes::CommandMergeMeshes()
  2794. : ccCommandLineInterface::Command(QObject::tr("Merge meshes"), COMMAND_MERGE_MESHES)
  2795. {}
  2796. bool CommandMergeMeshes::process(ccCommandLineInterface& cmd)
  2797. {
  2798. if (cmd.meshes().size() < 2)
  2799. {
  2800. cmd.warning(QObject::tr("Less than 2 meshes are loaded! Nothing to do..."));
  2801. return true;
  2802. }
  2803. CLMeshDesc mergedMeshDesc;
  2804. bool firstValidMesh = true;
  2805. //create the destination mesh
  2806. ccPointCloud* vertices = new ccPointCloud("vertices");
  2807. QScopedPointer<ccMesh> mergedMesh(new ccMesh(vertices));
  2808. mergedMesh->setName("Merged mesh");
  2809. mergedMesh->addChild(vertices);
  2810. vertices->setEnabled(false);
  2811. //merge meshes
  2812. for (CLMeshDesc& desc : cmd.meshes())
  2813. {
  2814. //get the mesh
  2815. ccMesh* mesh = dynamic_cast<ccMesh*>(desc.mesh);
  2816. if (!mesh)
  2817. {
  2818. ccLog::Error(QObject::tr("Can't merge mesh '%1' (unhandled type)").arg(desc.basename));
  2819. }
  2820. if (mergedMesh->merge(mesh, true)) //merge it
  2821. {
  2822. if (firstValidMesh)
  2823. {
  2824. //copy the first valid mesh description
  2825. mergedMeshDesc = desc;
  2826. mergedMeshDesc.mesh = nullptr;
  2827. firstValidMesh = false;
  2828. }
  2829. }
  2830. else
  2831. {
  2832. return cmd.error(QObject::tr("Merge operation failed"));
  2833. }
  2834. delete desc.mesh;
  2835. desc.mesh = nullptr;
  2836. }
  2837. if (mergedMesh->size() == 0)
  2838. {
  2839. return cmd.error(QObject::tr("Result is empty"));
  2840. }
  2841. //clean the 'cmd.meshes()' vector
  2842. cmd.removeMeshes();
  2843. //add the new mesh
  2844. mergedMeshDesc.basename += QObject::tr("_MERGED");
  2845. mergedMeshDesc.mesh = mergedMesh.take();
  2846. cmd.meshes().push_back(mergedMeshDesc);
  2847. if (cmd.autoSaveMode())
  2848. {
  2849. QString errorStr = cmd.exportEntity(mergedMeshDesc);
  2850. if (!errorStr.isEmpty())
  2851. {
  2852. return cmd.error(errorStr);
  2853. }
  2854. }
  2855. return true;
  2856. }
  2857. CommandMergeClouds::CommandMergeClouds()
  2858. : ccCommandLineInterface::Command(QObject::tr("Merge clouds"), COMMAND_MERGE_CLOUDS)
  2859. {}
  2860. bool CommandMergeClouds::process(ccCommandLineInterface& cmd)
  2861. {
  2862. if (cmd.clouds().size() < 2)
  2863. {
  2864. cmd.warning(QObject::tr("Less than 2 clouds are loaded! Nothing to do..."));
  2865. return true;
  2866. }
  2867. //merge clouds
  2868. if (!cmd.clouds().empty())
  2869. {
  2870. for (size_t i = 1; i < cmd.clouds().size(); ++i)
  2871. {
  2872. unsigned beforePts = cmd.clouds().front().pc->size();
  2873. CLCloudDesc& desc = cmd.clouds()[i];
  2874. unsigned newPts = desc.pc->size();
  2875. *cmd.clouds().front().pc += desc.pc;
  2876. //success?
  2877. if (cmd.clouds().front().pc->size() == beforePts + newPts)
  2878. {
  2879. delete desc.pc;
  2880. desc.pc = nullptr;
  2881. }
  2882. else
  2883. {
  2884. return cmd.error(QObject::tr("Fusion failed! (not enough memory?)"));
  2885. }
  2886. }
  2887. }
  2888. //clean the 'cmd.clouds()' vector
  2889. cmd.clouds().resize(1);
  2890. //update the first one
  2891. cmd.clouds().front().basename += QObject::tr("_MERGED");
  2892. if (cmd.autoSaveMode())
  2893. {
  2894. QString errorStr = cmd.exportEntity(cmd.clouds().front());
  2895. if (!errorStr.isEmpty())
  2896. {
  2897. return cmd.error(errorStr);
  2898. }
  2899. }
  2900. return true;
  2901. }
  2902. CommandSetGlobalShift::CommandSetGlobalShift()
  2903. : ccCommandLineInterface::Command(QObject::tr("Set global shift"), COMMAND_SET_GLOBAL_SHIFT)
  2904. {}
  2905. bool CommandSetGlobalShift::process(ccCommandLineInterface& cmd)
  2906. {
  2907. if (cmd.clouds().empty() && cmd.meshes().empty())
  2908. {
  2909. return cmd.error(QObject::tr("No loaded entity! (be sure to open one with \"-%1 [filename]\" before \"-%2\")").arg(COMMAND_OPEN, COMMAND_SET_GLOBAL_SHIFT));
  2910. }
  2911. //process globalshift options first
  2912. ccCommandLineInterface::GlobalShiftOptions globalShiftOptions;
  2913. cmd.processGlobalShiftCommand(globalShiftOptions);
  2914. //if it is not a valid global shift then an error msg already issued.
  2915. if (globalShiftOptions.mode != ccCommandLineInterface::GlobalShiftOptions::Mode::CUSTOM_GLOBAL_SHIFT)
  2916. {
  2917. return cmd.error(QObject::tr("Global shift must be in the form of three coordinates 'x' 'y' 'z'"));
  2918. }
  2919. CCVector3d newShift = globalShiftOptions.customGlobalShift;
  2920. //look for additional parameters
  2921. bool keepOrigFixed = false;
  2922. while (!cmd.arguments().empty())
  2923. {
  2924. QString argument = cmd.arguments().front();
  2925. if (ccCommandLineInterface::IsCommand(argument, COMMAND_SET_GLOBAL_SHIFT_KEEP_ORIG_FIXED))
  2926. {
  2927. //local option confirmed, pop that from front
  2928. cmd.arguments().pop_front();
  2929. keepOrigFixed = true;
  2930. cmd.print(QObject::tr("[%1]").arg(COMMAND_SET_GLOBAL_SHIFT_KEEP_ORIG_FIXED));
  2931. }
  2932. else
  2933. {
  2934. break;
  2935. }
  2936. }
  2937. //create an entity vector
  2938. size_t nrOfClouds = cmd.clouds().size();
  2939. size_t nrOfMeshes = cmd.meshes().size();
  2940. size_t nrOfEntities = nrOfClouds + nrOfMeshes;
  2941. std::vector<std::pair<ccShiftedObject*, CLEntityDesc*>> entities;
  2942. entities.reserve(nrOfEntities);
  2943. //add clouds to the vector
  2944. for (CLCloudDesc& desc : cmd.clouds())
  2945. {
  2946. entities.push_back({ desc.pc, &desc });
  2947. }
  2948. //add meshes to the vector
  2949. for (CLMeshDesc& desc : cmd.meshes())
  2950. {
  2951. bool isLocked = false;
  2952. ccShiftedObject* shifted = ccHObjectCaster::ToShifted(desc.mesh, &isLocked);
  2953. if (shifted && !isLocked)
  2954. {
  2955. entities.push_back({ shifted, &desc });
  2956. }
  2957. }
  2958. //process both clouds and meshes
  2959. for (const auto& pair : entities)
  2960. {
  2961. CLEntityDesc& desc = *pair.second;
  2962. ccShiftedObject* shiftedObject = pair.first;
  2963. CCVector3d originalShift = shiftedObject->getGlobalShift();
  2964. cmd.print(QObject::tr("\t[%4 - %5] Original global shift {%1,%2,%3}")
  2965. .arg(originalShift.x)
  2966. .arg(originalShift.y)
  2967. .arg(originalShift.z)
  2968. .arg(desc.basename)
  2969. .arg(shiftedObject->getName()));
  2970. //translate entity to keep the initial global origin
  2971. if (keepOrigFixed)
  2972. {
  2973. CCVector3d T = newShift - originalShift;
  2974. ccGLMatrix transMat;
  2975. double maxCoordValue = ccGlobalShiftManager::MaxCoordinateAbsValue();
  2976. if (T.x > maxCoordValue || T.y > maxCoordValue || T.z > maxCoordValue)
  2977. {
  2978. cmd.warning(QObject::tr("\t[%5 - %6] Applied transformation is bigger {%1,%2,%3} than the threshold {%4}, precision loss may occur.")
  2979. .arg(T.x)
  2980. .arg(T.y)
  2981. .arg(T.z)
  2982. .arg(maxCoordValue)
  2983. .arg(desc.basename)
  2984. .arg(shiftedObject->getName()));
  2985. }
  2986. cmd.print(QObject::tr("\t[%4 - %5] Applied Transformation {%1,%2,%3}")
  2987. .arg(T.x)
  2988. .arg(T.y)
  2989. .arg(T.z)
  2990. .arg(desc.basename)
  2991. .arg(shiftedObject->getName()));
  2992. transMat.toIdentity();
  2993. transMat.setTranslation(T);
  2994. shiftedObject->applyGLTransformation_recursive(&transMat);
  2995. }
  2996. //apply new global shift
  2997. shiftedObject->setGlobalShift(newShift.x, newShift.y, newShift.z);
  2998. cmd.print(QObject::tr("\t[%4 - %5] Global shift set to {%1,%2,%3}")
  2999. .arg(newShift.x)
  3000. .arg(newShift.y)
  3001. .arg(newShift.z)
  3002. .arg(desc.basename)
  3003. .arg(shiftedObject->getName()));
  3004. QString nameSuffix = QObject::tr("_SHIFTED_FROM_%1_%2_%3_TO_%4_%5_%6")
  3005. .arg(originalShift.x)
  3006. .arg(originalShift.y)
  3007. .arg(originalShift.z)
  3008. .arg(newShift.x)
  3009. .arg(newShift.y)
  3010. .arg(newShift.z);
  3011. if ((&desc)->getCLEntityType() == CL_ENTITY_TYPE::MESH)
  3012. {
  3013. //set the mesh name instead of the vertices cloud inside the mesh
  3014. ccHObject* parent = shiftedObject->getParent();
  3015. if (parent)
  3016. {
  3017. parent->setName(QObject::tr("%1%2").arg(parent->getName()).arg(nameSuffix));
  3018. }
  3019. }
  3020. else
  3021. {
  3022. shiftedObject->setName(QObject::tr("%1%2").arg(shiftedObject->getName()).arg(nameSuffix));
  3023. }
  3024. desc.basename += nameSuffix;
  3025. //save it as well
  3026. if (cmd.autoSaveMode())
  3027. {
  3028. QString errorStr = cmd.exportEntity(desc);
  3029. if (!errorStr.isEmpty())
  3030. {
  3031. return cmd.error(errorStr);
  3032. }
  3033. }
  3034. }
  3035. return true;
  3036. }
  3037. CommandSetActiveSF::CommandSetActiveSF()
  3038. : ccCommandLineInterface::Command(QObject::tr("Set active SF"), COMMAND_SET_ACTIVE_SF)
  3039. {}
  3040. bool CommandSetActiveSF::process(ccCommandLineInterface& cmd)
  3041. {
  3042. if (cmd.arguments().empty())
  3043. {
  3044. return cmd.error(QObject::tr("Missing parameter: scalar field index after \"-%1\"").arg(COMMAND_SET_ACTIVE_SF));
  3045. }
  3046. int sfIndex = -1;
  3047. QString sfName;
  3048. if (!GetSFIndexOrName(cmd, sfIndex, sfName, true))
  3049. {
  3050. return false;
  3051. }
  3052. if (cmd.clouds().empty() && cmd.meshes().empty())
  3053. {
  3054. return cmd.error(QObject::tr("No point cloud nor mesh loaded! (be sure to open one with \"-%1 [cloud filename]\" before \"-%2\")").arg(COMMAND_OPEN, COMMAND_SET_ACTIVE_SF));
  3055. }
  3056. for (CLCloudDesc& desc : cmd.clouds())
  3057. {
  3058. if (desc.pc)
  3059. {
  3060. int thisSFIndex = GetScalarFieldIndex(desc.pc, sfIndex, sfName, false);
  3061. desc.pc->setCurrentScalarField(thisSFIndex);
  3062. }
  3063. }
  3064. for (CLMeshDesc& desc : cmd.meshes())
  3065. {
  3066. ccPointCloud* pc = ccHObjectCaster::ToPointCloud(desc.mesh);
  3067. if (pc)
  3068. {
  3069. int thisSFIndex = GetScalarFieldIndex(pc, sfIndex, sfName, false);
  3070. pc->setCurrentScalarField(thisSFIndex);
  3071. }
  3072. }
  3073. return true;
  3074. }
  3075. CommandRemoveAllSFs::CommandRemoveAllSFs()
  3076. : ccCommandLineInterface::Command(QObject::tr("Remove all SF"), COMMAND_REMOVE_ALL_SFS)
  3077. {}
  3078. bool CommandRemoveAllSFs::process(ccCommandLineInterface& cmd)
  3079. {
  3080. //no argument required
  3081. for (auto& desc : cmd.clouds())
  3082. {
  3083. if (desc.pc/* && desc.pc->hasScalarFields()*/)
  3084. {
  3085. desc.pc->deleteAllScalarFields();
  3086. desc.pc->showSF(false);
  3087. }
  3088. }
  3089. for (auto& desc : cmd.meshes())
  3090. {
  3091. if (desc.mesh)
  3092. {
  3093. ccGenericPointCloud* cloud = desc.mesh->getAssociatedCloud();
  3094. if (cloud->isA(CC_TYPES::POINT_CLOUD))
  3095. {
  3096. static_cast<ccPointCloud*>(cloud)->deleteAllScalarFields();
  3097. cloud->showSF(false);
  3098. }
  3099. }
  3100. }
  3101. return true;
  3102. }
  3103. CommandRemoveSF::CommandRemoveSF()
  3104. : ccCommandLineInterface::Command(QObject::tr("Remove a specific SF"), COMMAND_REMOVE_SF)
  3105. {}
  3106. bool CommandRemoveSF::removeSF(int sfIndex, ccPointCloud& pc)
  3107. {
  3108. if (sfIndex < static_cast<int>(pc.getNumberOfScalarFields()))
  3109. {
  3110. ccLog::Print("[REMOVE_SF] " + QString::fromStdString(pc.getScalarFieldName(sfIndex)));
  3111. pc.deleteScalarField(sfIndex);
  3112. if (pc.getNumberOfScalarFields() == 0)
  3113. {
  3114. pc.showSF(false);
  3115. }
  3116. else if (pc.getCurrentDisplayedScalarFieldIndex() < 0)
  3117. {
  3118. pc.setCurrentDisplayedScalarField(static_cast<int>(pc.getNumberOfScalarFields()) - 1);
  3119. }
  3120. return true;
  3121. }
  3122. else
  3123. {
  3124. return false;
  3125. }
  3126. }
  3127. bool CommandRemoveSF::process(ccCommandLineInterface& cmd)
  3128. {
  3129. if (cmd.arguments().empty())
  3130. {
  3131. return cmd.error(QObject::tr("Missing parameter: SF index after %1").arg(COMMAND_REMOVE_SF));
  3132. }
  3133. int sfIndex = -1;
  3134. QString sfName;
  3135. if (!GetSFIndexOrName(cmd, sfIndex, sfName, true))
  3136. {
  3137. return false;
  3138. }
  3139. for (auto& desc : cmd.clouds())
  3140. {
  3141. if (desc.pc)
  3142. {
  3143. int thisSFIndex = GetScalarFieldIndex(desc.pc, sfIndex, sfName, true);
  3144. if (thisSFIndex >= 0)
  3145. {
  3146. removeSF(thisSFIndex, *desc.pc);
  3147. }
  3148. }
  3149. }
  3150. for (auto& desc : cmd.meshes())
  3151. {
  3152. if (desc.mesh)
  3153. {
  3154. ccGenericPointCloud* cloud = desc.mesh->getAssociatedCloud();
  3155. if (cloud->isA(CC_TYPES::POINT_CLOUD))
  3156. {
  3157. ccPointCloud* pc = static_cast<ccPointCloud*>(cloud);
  3158. int thisSFIndex = GetScalarFieldIndex(pc, sfIndex, sfName, true);
  3159. if (thisSFIndex >= 0)
  3160. {
  3161. removeSF(thisSFIndex, *pc);
  3162. if (pc->getNumberOfScalarFields() == 0)
  3163. {
  3164. desc.mesh->showSF(false);
  3165. }
  3166. }
  3167. }
  3168. }
  3169. }
  3170. return true;
  3171. }
  3172. CommandRemoveRGB::CommandRemoveRGB()
  3173. : ccCommandLineInterface::Command(QObject::tr("Remove RGB"), COMMAND_REMOVE_RGB)
  3174. {}
  3175. bool CommandRemoveRGB::process(ccCommandLineInterface& cmd)
  3176. {
  3177. //no argument required
  3178. for (auto& desc : cmd.clouds())
  3179. {
  3180. if (desc.pc)
  3181. {
  3182. desc.pc->unallocateColors();
  3183. desc.pc->showColors(false);
  3184. }
  3185. }
  3186. for (auto& desc : cmd.meshes())
  3187. {
  3188. if (desc.mesh)
  3189. {
  3190. ccGenericPointCloud* cloud = desc.mesh->getAssociatedCloud();
  3191. if (cloud->isA(CC_TYPES::POINT_CLOUD))
  3192. {
  3193. static_cast<ccPointCloud*>(cloud)->unallocateColors();
  3194. cloud->showColors(false);
  3195. }
  3196. desc.mesh->showColors(false);
  3197. }
  3198. }
  3199. return true;
  3200. }
  3201. CommandRemoveNormals::CommandRemoveNormals()
  3202. : ccCommandLineInterface::Command(QObject::tr("Remove normals"), COMMAND_REMOVE_NORMALS)
  3203. {}
  3204. bool CommandRemoveNormals::process(ccCommandLineInterface& cmd)
  3205. {
  3206. //no argument required
  3207. for (auto& desc : cmd.clouds())
  3208. {
  3209. if (desc.pc)
  3210. {
  3211. desc.pc->unallocateNorms();
  3212. desc.pc->showNormals(false);
  3213. }
  3214. }
  3215. for (auto& desc : cmd.meshes())
  3216. {
  3217. if (desc.mesh)
  3218. {
  3219. ccGenericPointCloud* cloud = desc.mesh->getAssociatedCloud();
  3220. if (cloud->isA(CC_TYPES::POINT_CLOUD))
  3221. {
  3222. static_cast<ccPointCloud*>(cloud)->unallocateNorms();
  3223. cloud->showNormals(false);
  3224. }
  3225. if (desc.mesh->isA(CC_TYPES::MESH))
  3226. {
  3227. static_cast<ccMesh*>(desc.mesh)->clearTriNormals();
  3228. desc.mesh->showNormals(false);
  3229. }
  3230. }
  3231. }
  3232. return true;
  3233. }
  3234. CommandRemoveScanGrids::CommandRemoveScanGrids()
  3235. : ccCommandLineInterface::Command(QObject::tr("Remove scan grids"), COMMAND_REMOVE_SCAN_GRIDS)
  3236. {}
  3237. bool CommandRemoveScanGrids::process(ccCommandLineInterface& cmd)
  3238. {
  3239. //no argument required
  3240. for (CLCloudDesc& desc : cmd.clouds())
  3241. {
  3242. if (desc.pc)
  3243. {
  3244. desc.pc->removeGrids();
  3245. }
  3246. }
  3247. for (CLMeshDesc& desc : cmd.meshes())
  3248. {
  3249. if (desc.mesh)
  3250. {
  3251. ccGenericPointCloud* cloud = desc.mesh->getAssociatedCloud();
  3252. if (cloud->isA(CC_TYPES::POINT_CLOUD))
  3253. {
  3254. static_cast<ccPointCloud*>(cloud)->removeGrids();
  3255. }
  3256. }
  3257. }
  3258. return true;
  3259. }
  3260. CommandRemoveSensors::CommandRemoveSensors()
  3261. : ccCommandLineInterface::Command(QObject::tr("Remove sensors"), COMMAND_REMOVE_SENSORS)
  3262. {}
  3263. static void RemoveSensors(ccHObject* entity)
  3264. {
  3265. if (!entity)
  3266. {
  3267. assert(false);
  3268. return;
  3269. }
  3270. ccHObject::Container sensors;
  3271. entity->filterChildren(sensors, false, CC_TYPES::SENSOR, false); // direct children only
  3272. for (ccHObject* sensor : sensors)
  3273. {
  3274. ccLog::Print(QString("Removing sensor '%1' from entity '%2'").arg(sensor->getName()).arg(entity->getName()));
  3275. entity->removeChild(sensor);
  3276. }
  3277. }
  3278. bool CommandRemoveSensors::process(ccCommandLineInterface& cmd)
  3279. {
  3280. //no argument required
  3281. for (CLCloudDesc& desc : cmd.clouds())
  3282. {
  3283. if (desc.pc)
  3284. {
  3285. RemoveSensors(desc.pc);
  3286. }
  3287. }
  3288. for (CLMeshDesc& desc : cmd.meshes())
  3289. {
  3290. if (desc.mesh)
  3291. {
  3292. RemoveSensors(desc.mesh);
  3293. ccGenericPointCloud* cloud = desc.mesh->getAssociatedCloud();
  3294. RemoveSensors(cloud);
  3295. }
  3296. }
  3297. return true;
  3298. }
  3299. CommandMatchBBCenters::CommandMatchBBCenters()
  3300. : ccCommandLineInterface::Command(QObject::tr("Match B.B. centers"), COMMAND_MATCH_BB_CENTERS)
  3301. {}
  3302. bool CommandMatchBBCenters::process(ccCommandLineInterface& cmd)
  3303. {
  3304. std::vector<CLEntityDesc*> entities;
  3305. for (auto& cloud : cmd.clouds())
  3306. {
  3307. entities.push_back(&cloud);
  3308. }
  3309. for (auto& mesh : cmd.meshes())
  3310. {
  3311. entities.push_back(&mesh);
  3312. }
  3313. if (entities.empty())
  3314. {
  3315. return cmd.error("No entity loaded!");
  3316. }
  3317. else if (entities.size() == 1)
  3318. {
  3319. cmd.warning("Nothing to do: only one entity currently loaded!");
  3320. return true;
  3321. }
  3322. CCVector3 firstCenter = entities.front()->getEntity()->getOwnBB().getCenter();
  3323. for (size_t i = 1; i < entities.size(); ++i)
  3324. {
  3325. ccHObject* ent = entities[i]->getEntity();
  3326. CCVector3 center = ent->getOwnBB().getCenter();
  3327. CCVector3 T = firstCenter - center;
  3328. //transformation (used only for translation)
  3329. ccGLMatrix glTrans;
  3330. glTrans += T;
  3331. //apply translation matrix
  3332. ent->applyGLTransformation_recursive(&glTrans);
  3333. cmd.print(QObject::tr("Entity '%1' has been translated: (%2,%3,%4)").arg(ent->getName()).arg(T.x).arg(T.y).arg(T.z));
  3334. if (cmd.autoSaveMode())
  3335. {
  3336. QString errorStr = cmd.exportEntity(*entities[i]);
  3337. if (!errorStr.isEmpty())
  3338. {
  3339. return cmd.error(errorStr);
  3340. }
  3341. }
  3342. }
  3343. return true;
  3344. }
  3345. CommandMatchBestFitPlane::CommandMatchBestFitPlane()
  3346. : ccCommandLineInterface::Command(QObject::tr("Compute best fit plane"), COMMAND_BEST_FIT_PLANE)
  3347. {}
  3348. bool CommandMatchBestFitPlane::process(ccCommandLineInterface& cmd)
  3349. {
  3350. //look for local options
  3351. bool makeCloudsHoriz = false;
  3352. bool keepLoaded = false;
  3353. while (!cmd.arguments().empty())
  3354. {
  3355. QString argument = cmd.arguments().front();
  3356. if (ccCommandLineInterface::IsCommand(argument, COMMAND_BEST_FIT_PLANE_MAKE_HORIZ))
  3357. {
  3358. //local option confirmed, we can move on
  3359. cmd.arguments().pop_front();
  3360. makeCloudsHoriz = true;
  3361. }
  3362. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_BEST_FIT_PLANE_KEEP_LOADED))
  3363. {
  3364. //local option confirmed, we can move on
  3365. cmd.arguments().pop_front();
  3366. keepLoaded = true;
  3367. }
  3368. else
  3369. {
  3370. break; //as soon as we encounter an unrecognized argument, we break the local loop to go back to the main one!
  3371. }
  3372. }
  3373. if (cmd.clouds().empty())
  3374. {
  3375. return cmd.error(QObject::tr("No cloud available. Be sure to open one first!"));
  3376. }
  3377. for (CLCloudDesc& desc : cmd.clouds())
  3378. {
  3379. //try to fit plane
  3380. double rms = 0.0;
  3381. ccPlane* pPlane = ccPlane::Fit(desc.pc, &rms);
  3382. if (pPlane)
  3383. {
  3384. cmd.print(QObject::tr("Plane successfully fitted: rms = %1").arg(rms));
  3385. CCVector3 N = pPlane->getNormal();
  3386. CCVector3 C = *CCCoreLib::Neighbourhood(desc.pc).getGravityCenter();
  3387. CLMeshDesc planeDesc;
  3388. planeDesc.mesh = pPlane;
  3389. planeDesc.basename = desc.basename;
  3390. planeDesc.path = desc.path;
  3391. //save plane as a BIN file
  3392. QString outputFilename;
  3393. QString errorStr = cmd.exportEntity(planeDesc, "BEST_FIT_PLANE", &outputFilename);
  3394. if (!errorStr.isEmpty())
  3395. {
  3396. cmd.warning(errorStr);
  3397. }
  3398. //compute the transformation matrix that would make this normal points towards +Z
  3399. ccGLMatrix makeZPosMatrix = ccGLMatrix::FromToRotation(N, CCVector3(0, 0, CCCoreLib::PC_ONE));
  3400. CCVector3 Gt = C;
  3401. makeZPosMatrix.applyRotation(Gt);
  3402. makeZPosMatrix.setTranslation(C - Gt);
  3403. //open text file to save plane related information
  3404. {
  3405. QString txtFilename = QObject::tr("%1/%2_BEST_FIT_PLANE_INFO").arg(desc.path, desc.basename);
  3406. if (cmd.addTimestamp())
  3407. {
  3408. QString timestamp = QDateTime::currentDateTime().toString("yyyy-MM-dd_hh'h'mm_ss_zzz");
  3409. txtFilename += QObject::tr("_%1").arg(timestamp);
  3410. }
  3411. txtFilename += QObject::tr(".txt");
  3412. QFile txtFile(txtFilename);
  3413. if (txtFile.open(QIODevice::WriteOnly | QIODevice::Text))
  3414. {
  3415. QTextStream txtStream(&txtFile);
  3416. txtStream << QObject::tr("Filename: %1").arg(outputFilename) << endl;
  3417. txtStream << QObject::tr("Fitting RMS: %1").arg(rms) << endl;
  3418. //We always consider the normal with a positive 'Z' by default!
  3419. if (N.z < 0.0)
  3420. {
  3421. N *= -1.0;
  3422. }
  3423. int precision = cmd.numericalPrecision();
  3424. txtStream << QObject::tr("Normal: (%1,%2,%3)").arg(N.x, 0, 'f', precision).arg(N.y, 0, 'f', precision).arg(N.z, 0, 'f', precision) << endl;
  3425. //we compute strike & dip by the way
  3426. {
  3427. PointCoordinateType dip = 0;
  3428. PointCoordinateType dipDir = 0;
  3429. ccNormalVectors::ConvertNormalToDipAndDipDir(N, dip, dipDir);
  3430. txtStream << ccNormalVectors::ConvertDipAndDipDirToString(dip, dipDir) << endl;
  3431. }
  3432. txtStream << "Orientation matrix:" << endl;
  3433. txtStream << makeZPosMatrix.toString(precision, ' ') << endl;
  3434. //close the text file
  3435. txtFile.close();
  3436. }
  3437. else
  3438. {
  3439. cmd.warning("Failed to open file " + txtFilename + " for writing");
  3440. }
  3441. }
  3442. if (keepLoaded)
  3443. {
  3444. //add the resulting plane (mesh) to the main set
  3445. cmd.meshes().push_back(planeDesc);
  3446. }
  3447. if (makeCloudsHoriz)
  3448. {
  3449. //apply 'horizontal' matrix
  3450. desc.pc->applyGLTransformation_recursive(&makeZPosMatrix);
  3451. cmd.print(QObject::tr("Cloud '%1' has been transformed with the above matrix").arg(desc.pc->getName()));
  3452. desc.basename += QObject::tr("_HORIZ");
  3453. if (cmd.autoSaveMode())
  3454. {
  3455. QString errorStr = cmd.exportEntity(desc);
  3456. if (!errorStr.isEmpty())
  3457. {
  3458. cmd.warning(errorStr);
  3459. }
  3460. }
  3461. }
  3462. }
  3463. else
  3464. {
  3465. cmd.warning(QObject::tr("Failed to compute best fit plane for cloud '%1'").arg(desc.pc->getName()));
  3466. }
  3467. }
  3468. return true;
  3469. }
  3470. CommandOrientNormalsMST::CommandOrientNormalsMST()
  3471. : ccCommandLineInterface::Command(QObject::tr("Orient normals"), COMMAND_ORIENT_NORMALS)
  3472. {}
  3473. bool CommandOrientNormalsMST::process(ccCommandLineInterface& cmd)
  3474. {
  3475. if (cmd.arguments().empty())
  3476. {
  3477. return cmd.error(QObject::tr("Missing parameter: number of neighbors after \"-%1\"").arg(COMMAND_ORIENT_NORMALS));
  3478. }
  3479. QString knnStr = cmd.arguments().takeFirst();
  3480. bool ok;
  3481. int knn = knnStr.toInt(&ok);
  3482. if (!ok || knn <= 0)
  3483. {
  3484. return cmd.error(QObject::tr("Invalid parameter: number of neighbors (%1)").arg(knnStr));
  3485. }
  3486. if (cmd.clouds().empty())
  3487. {
  3488. return cmd.error(QObject::tr("No cloud available. Be sure to open one first!"));
  3489. }
  3490. QScopedPointer<ccProgressDialog> progressDialog(nullptr);
  3491. if (!cmd.silentMode())
  3492. {
  3493. progressDialog.reset(new ccProgressDialog(false, cmd.widgetParent()));
  3494. progressDialog->setAutoClose(false);
  3495. }
  3496. for (CLCloudDesc& desc : cmd.clouds())
  3497. {
  3498. assert(desc.pc);
  3499. if (!desc.pc->hasNormals())
  3500. {
  3501. continue;
  3502. }
  3503. //computation
  3504. if (desc.pc->orientNormalsWithMST(knn, progressDialog.data()))
  3505. {
  3506. desc.basename += QObject::tr("_NORMS_REORIENTED");
  3507. if (cmd.autoSaveMode())
  3508. {
  3509. QString errorStr = cmd.exportEntity(desc);
  3510. if (!errorStr.isEmpty())
  3511. {
  3512. cmd.warning(errorStr);
  3513. }
  3514. }
  3515. }
  3516. else
  3517. {
  3518. return cmd.error(QObject::tr("Failed to orient the normals of cloud '%1'!").arg(desc.pc->getName()));
  3519. }
  3520. }
  3521. if (progressDialog)
  3522. {
  3523. progressDialog->close();
  3524. QCoreApplication::processEvents();
  3525. }
  3526. return true;
  3527. }
  3528. CommandSORFilter::CommandSORFilter()
  3529. : ccCommandLineInterface::Command(QObject::tr("S.O.R. filter"), COMMAND_SOR_FILTER)
  3530. {}
  3531. bool CommandSORFilter::process(ccCommandLineInterface& cmd)
  3532. {
  3533. if (cmd.arguments().empty())
  3534. {
  3535. return cmd.error(QObject::tr("Missing parameter: number of neighbors mode after \"-%1\"").arg(COMMAND_SOR_FILTER));
  3536. }
  3537. QString knnStr = cmd.arguments().takeFirst();
  3538. bool ok;
  3539. int knn = knnStr.toInt(&ok);
  3540. if (!ok || knn <= 0)
  3541. {
  3542. return cmd.error(QObject::tr("Invalid parameter: number of neighbors (%1)").arg(knnStr));
  3543. }
  3544. if (cmd.arguments().empty())
  3545. {
  3546. return cmd.error(QObject::tr("Missing parameter: sigma multiplier after number of neighbors (SOR)"));
  3547. }
  3548. QString sigmaStr = cmd.arguments().takeFirst();
  3549. double nSigma = sigmaStr.toDouble(&ok);
  3550. if (!ok || nSigma < 0)
  3551. {
  3552. return cmd.error(QObject::tr("Invalid parameter: sigma multiplier (%1)").arg(nSigma));
  3553. }
  3554. if (cmd.clouds().empty())
  3555. {
  3556. return cmd.error(QObject::tr("No cloud available. Be sure to open one first!"));
  3557. }
  3558. QScopedPointer<ccProgressDialog> progressDialog(nullptr);
  3559. if (!cmd.silentMode())
  3560. {
  3561. progressDialog.reset(new ccProgressDialog(false, cmd.widgetParent()));
  3562. progressDialog->setAutoClose(false);
  3563. }
  3564. for (CLCloudDesc& desc : cmd.clouds())
  3565. {
  3566. assert(desc.pc);
  3567. //computation
  3568. CCCoreLib::ReferenceCloud* selection = CCCoreLib::CloudSamplingTools::sorFilter(desc.pc,
  3569. knn,
  3570. nSigma,
  3571. nullptr,
  3572. progressDialog.data());
  3573. if (selection)
  3574. {
  3575. ccPointCloud* cleanCloud = desc.pc->partialClone(selection);
  3576. if (cleanCloud)
  3577. {
  3578. cleanCloud->setName(desc.pc->getName() + QObject::tr(".clean"));
  3579. if (cmd.autoSaveMode())
  3580. {
  3581. CLCloudDesc newDesc(cleanCloud, desc.basename, desc.path, desc.indexInFile);
  3582. QString errorStr = cmd.exportEntity(newDesc, "SOR");
  3583. if (!errorStr.isEmpty())
  3584. {
  3585. delete cleanCloud;
  3586. return cmd.error(errorStr);
  3587. }
  3588. }
  3589. //replace current cloud by this one
  3590. delete desc.pc;
  3591. desc.pc = cleanCloud;
  3592. desc.basename += QObject::tr("_SOR");
  3593. //delete cleanCloud;
  3594. //cleanCloud = 0;
  3595. }
  3596. else
  3597. {
  3598. return cmd.error(QObject::tr("Not enough memory to create a clean version of cloud '%1'!").arg(desc.pc->getName()));
  3599. }
  3600. delete selection;
  3601. selection = nullptr;
  3602. }
  3603. else
  3604. {
  3605. //no points fall inside selection!
  3606. return cmd.error(QObject::tr("Failed to apply SOR filter on cloud '%1'! (empty output or not enough memory?)").arg(desc.pc->getName()));
  3607. }
  3608. }
  3609. if (progressDialog)
  3610. {
  3611. progressDialog->close();
  3612. QCoreApplication::processEvents();
  3613. }
  3614. return true;
  3615. }
  3616. CommandNoiseFilter::CommandNoiseFilter()
  3617. : ccCommandLineInterface::Command(QObject::tr("Noise filter"), COMMAND_NOISE_FILTER)
  3618. {}
  3619. bool CommandNoiseFilter::process(ccCommandLineInterface& cmd)
  3620. {
  3621. if (cmd.arguments().size() < 4)
  3622. {
  3623. return cmd.error(QObject::tr("Missing parameters: 'KNN/RADIUS {value} REL/ABS {value}' expected after \"-%1\"").arg(COMMAND_NOISE_FILTER));
  3624. }
  3625. QString firstOption = cmd.arguments().takeFirst().toUpper();
  3626. int knn = -1;
  3627. double radius = std::numeric_limits<double>::quiet_NaN();
  3628. if (firstOption == COMMAND_NOISE_FILTER_KNN)
  3629. {
  3630. QString knnStr = cmd.arguments().takeFirst();
  3631. bool ok;
  3632. knn = knnStr.toInt(&ok);
  3633. if (!ok || knn <= 0)
  3634. {
  3635. return cmd.error(QObject::tr("Invalid parameter: number of neighbors after KNN (got '%1' instead)").arg(knnStr));
  3636. }
  3637. }
  3638. else if(firstOption == COMMAND_NOISE_FILTER_RADIUS)
  3639. {
  3640. QString radiusStr = cmd.arguments().takeFirst();
  3641. bool ok;
  3642. radius = radiusStr.toDouble(&ok);
  3643. if (!ok || radius <= 0)
  3644. {
  3645. return cmd.error(QObject::tr("Invalid parameter: radius after RADIUS (got '%1' instead)").arg(radiusStr));
  3646. }
  3647. }
  3648. else
  3649. {
  3650. return cmd.error(QObject::tr("Invalid parameter: KNN or RADIUS expected after \"-%1\"").arg(COMMAND_NOISE_FILTER));
  3651. }
  3652. QString secondOption = cmd.arguments().takeFirst().toUpper();
  3653. bool absoluteError = true;
  3654. if (secondOption == COMMAND_NOISE_FILTER_REL)
  3655. {
  3656. absoluteError = false;
  3657. }
  3658. else if (secondOption == COMMAND_NOISE_FILTER_ABS)
  3659. {
  3660. absoluteError = true;
  3661. }
  3662. else
  3663. {
  3664. return cmd.error(QObject::tr("Invalid parameter: REL or ABS expected"));
  3665. }
  3666. double error = std::numeric_limits<double>::quiet_NaN();
  3667. {
  3668. QString errorStr = cmd.arguments().takeFirst();
  3669. bool ok;
  3670. error = errorStr.toDouble(&ok);
  3671. if (!ok || error <= 0)
  3672. {
  3673. return cmd.error(QObject::tr("Invalid parameter: relative or absolute error expected after KNN (got '%1' instead)").arg(errorStr));
  3674. }
  3675. }
  3676. //optional arguments
  3677. bool removeIsolatedPoints = false;
  3678. if (!cmd.arguments().empty())
  3679. {
  3680. QString argument = cmd.arguments().front();
  3681. if (argument == COMMAND_NOISE_FILTER_RIP)
  3682. {
  3683. //local option confirmed, we can move on
  3684. cmd.arguments().pop_front();
  3685. removeIsolatedPoints = true;
  3686. }
  3687. }
  3688. QScopedPointer<ccProgressDialog> progressDialog(nullptr);
  3689. if (!cmd.silentMode())
  3690. {
  3691. progressDialog.reset(new ccProgressDialog(false, cmd.widgetParent()));
  3692. progressDialog->setAutoClose(false);
  3693. }
  3694. for (CLCloudDesc& desc : cmd.clouds())
  3695. {
  3696. assert(desc.pc);
  3697. //computation
  3698. CCCoreLib::ReferenceCloud* selection = CCCoreLib::CloudSamplingTools::noiseFilter(desc.pc,
  3699. static_cast<PointCoordinateType>(radius),
  3700. error,
  3701. removeIsolatedPoints,
  3702. knn > 0,
  3703. knn,
  3704. absoluteError,
  3705. error,
  3706. nullptr,
  3707. progressDialog.data());
  3708. if (selection)
  3709. {
  3710. ccPointCloud* cleanCloud = desc.pc->partialClone(selection);
  3711. if (cleanCloud)
  3712. {
  3713. cleanCloud->setName(desc.pc->getName() + QObject::tr(".clean"));
  3714. if (cmd.autoSaveMode())
  3715. {
  3716. CLCloudDesc newDesc(cleanCloud, desc.basename, desc.path, desc.indexInFile);
  3717. QString errorStr = cmd.exportEntity(newDesc, "DENOISED");
  3718. if (!errorStr.isEmpty())
  3719. {
  3720. delete cleanCloud;
  3721. return cmd.error(errorStr);
  3722. }
  3723. }
  3724. //replace current cloud by this one
  3725. delete desc.pc;
  3726. desc.pc = cleanCloud;
  3727. desc.basename += QObject::tr("_DENOISED");
  3728. //delete cleanCloud;
  3729. //cleanCloud = 0;
  3730. }
  3731. else
  3732. {
  3733. return cmd.error(QObject::tr("Not enough memory to create a clean version of cloud '%1'!").arg(desc.pc->getName()));
  3734. }
  3735. delete selection;
  3736. selection = nullptr;
  3737. }
  3738. else
  3739. {
  3740. //no points fall inside selection!
  3741. return cmd.error(QObject::tr("Failed to apply Noise filter on cloud '%1'! (empty output or not enough memory?)").arg(desc.pc->getName()));
  3742. }
  3743. }
  3744. if (progressDialog)
  3745. {
  3746. progressDialog->close();
  3747. QCoreApplication::processEvents();
  3748. }
  3749. return true;
  3750. }
  3751. CommandRemoveDuplicatePoints::CommandRemoveDuplicatePoints()
  3752. : ccCommandLineInterface::Command(QObject::tr("Remove duplicate points"), COMMAND_REMOVE_DUPLICATE_POINTS)
  3753. {}
  3754. bool CommandRemoveDuplicatePoints::process(ccCommandLineInterface& cmd)
  3755. {
  3756. double minDistanceBetweenPoints = std::numeric_limits<double>::epsilon();
  3757. //get optional argument
  3758. if (!cmd.arguments().empty())
  3759. {
  3760. bool paramOk = false;
  3761. double arg = cmd.arguments().front().toDouble(&paramOk);
  3762. if (paramOk)
  3763. {
  3764. if (arg < minDistanceBetweenPoints)
  3765. {
  3766. return cmd.error(QObject::tr("Invalid argument: '%1'").arg(arg));
  3767. }
  3768. minDistanceBetweenPoints = arg;
  3769. cmd.arguments().pop_front();
  3770. }
  3771. }
  3772. cmd.print(QObject::tr("Minimum distance between points: '%1'").arg(minDistanceBetweenPoints));
  3773. QScopedPointer<ccProgressDialog> progressDialog(nullptr);
  3774. if (!cmd.silentMode())
  3775. {
  3776. progressDialog.reset(new ccProgressDialog(false, cmd.widgetParent()));
  3777. progressDialog->setAutoClose(false);
  3778. }
  3779. for (CLCloudDesc& desc : cmd.clouds())
  3780. {
  3781. assert(desc.pc);
  3782. ccPointCloud* filteredCloud = desc.pc->removeDuplicatePoints(minDistanceBetweenPoints, progressDialog.data());
  3783. if (!filteredCloud)
  3784. {
  3785. return cmd.error(QObject::tr("Process failed (see log)"));
  3786. }
  3787. if (filteredCloud == desc.pc)
  3788. {
  3789. // nothing to do
  3790. continue;
  3791. }
  3792. //replace current cloud by filtered one
  3793. delete desc.pc;
  3794. desc.pc = filteredCloud;
  3795. desc.basename += QObject::tr("_REMOVED_DUPLICATE_POINTS");
  3796. }
  3797. if (progressDialog)
  3798. {
  3799. progressDialog->close();
  3800. QCoreApplication::processEvents();
  3801. }
  3802. return true;
  3803. }
  3804. CommandExtractVertices::CommandExtractVertices()
  3805. : ccCommandLineInterface::Command(QObject::tr("Extract vertices"), COMMAND_EXTRACT_VERTICES)
  3806. {}
  3807. bool CommandExtractVertices::process(ccCommandLineInterface& cmd)
  3808. {
  3809. if (cmd.meshes().empty())
  3810. {
  3811. cmd.warning(QObject::tr("No mesh available. Be sure to open one first!"));
  3812. return false;
  3813. }
  3814. for (CLMeshDesc& desc : cmd.meshes())
  3815. {
  3816. ccGenericMesh* mesh = desc.mesh;
  3817. ccGenericPointCloud* cloud = mesh->getAssociatedCloud();
  3818. ccPointCloud* pc = ccHObjectCaster::ToPointCloud(cloud);
  3819. if (!pc)
  3820. {
  3821. assert(false);
  3822. continue;
  3823. }
  3824. //add the resulting cloud to the main set
  3825. cmd.clouds().emplace_back(pc, desc.basename + QObject::tr(".vertices"), desc.path);
  3826. //don't forget to detach the cloud before we delete the meshes!
  3827. assert(pc->getParent() == mesh);
  3828. if (pc->getParent())
  3829. {
  3830. pc->getParent()->detachChild(pc);
  3831. }
  3832. //save it as well
  3833. if (cmd.autoSaveMode())
  3834. {
  3835. QString errorStr = cmd.exportEntity(cmd.clouds().back());
  3836. if (!errorStr.isEmpty())
  3837. {
  3838. return cmd.error(errorStr);
  3839. }
  3840. }
  3841. }
  3842. cmd.removeMeshes(false);
  3843. return true;
  3844. }
  3845. CommandFlipTriangles::CommandFlipTriangles()
  3846. : ccCommandLineInterface::Command(QObject::tr("Flip mesh triangles"), COMMAND_FLIP_TRIANGLES)
  3847. {}
  3848. bool CommandFlipTriangles::process(ccCommandLineInterface& cmd)
  3849. {
  3850. if (cmd.meshes().empty())
  3851. {
  3852. cmd.warning(QObject::tr("No mesh available. Be sure to open one first!"));
  3853. return false;
  3854. }
  3855. for (CLMeshDesc& desc : cmd.meshes())
  3856. {
  3857. ccGenericMesh* mesh = desc.mesh;
  3858. ccMesh* ccMesh = ccHObjectCaster::ToMesh(mesh);
  3859. if (ccMesh)
  3860. {
  3861. ccMesh->flipTriangles();
  3862. }
  3863. desc.basename += QObject::tr("_FLIPPED_TRIANGLES");
  3864. //save it as well
  3865. if (cmd.autoSaveMode())
  3866. {
  3867. QString errorStr = cmd.exportEntity(desc);
  3868. if (!errorStr.isEmpty())
  3869. {
  3870. return cmd.error(errorStr);
  3871. }
  3872. }
  3873. }
  3874. cmd.removeMeshes(false);
  3875. return true;
  3876. }
  3877. CommandSampleMesh::CommandSampleMesh()
  3878. : ccCommandLineInterface::Command(QObject::tr("Sample mesh"), COMMAND_SAMPLE_MESH)
  3879. {}
  3880. bool CommandSampleMesh::process(ccCommandLineInterface& cmd)
  3881. {
  3882. if (cmd.arguments().empty())
  3883. {
  3884. return cmd.error(QObject::tr("Missing parameter: sampling mode after \"-%1\" (POINTS/DENSITY)").arg(COMMAND_SAMPLE_MESH));
  3885. }
  3886. bool useDensity = false;
  3887. double parameter = 0;
  3888. QString sampleMode = cmd.arguments().takeFirst().toUpper();
  3889. if (sampleMode == "POINTS")
  3890. {
  3891. useDensity = false;
  3892. }
  3893. else if (sampleMode == "DENSITY")
  3894. {
  3895. useDensity = true;
  3896. }
  3897. else
  3898. {
  3899. return cmd.error(QObject::tr("Invalid parameter: unknown sampling mode \"%1\"").arg(sampleMode));
  3900. }
  3901. if (cmd.arguments().empty())
  3902. {
  3903. return cmd.error(QObject::tr("Missing parameter: value after sampling mode"));
  3904. }
  3905. bool conversionOk = false;
  3906. parameter = cmd.arguments().takeFirst().toDouble(&conversionOk);
  3907. if (!conversionOk)
  3908. {
  3909. return cmd.error(QObject::tr("Invalid parameter: value after sampling mode"));
  3910. }
  3911. if (cmd.meshes().empty())
  3912. {
  3913. return cmd.error(QObject::tr("No mesh available. Be sure to open one first!"));
  3914. }
  3915. QScopedPointer<ccProgressDialog> progressDialog(nullptr);
  3916. if (!cmd.silentMode())
  3917. {
  3918. progressDialog.reset(new ccProgressDialog(false, cmd.widgetParent()));
  3919. progressDialog->setAutoClose(false);
  3920. }
  3921. for (CLMeshDesc& desc : cmd.meshes())
  3922. {
  3923. ccPointCloud* cloud = desc.mesh->samplePoints(useDensity, parameter, true, true, true, progressDialog.data());
  3924. if (!cloud)
  3925. {
  3926. return cmd.error(QObject::tr("Cloud sampling failed!"));
  3927. }
  3928. //add the resulting cloud to the main set
  3929. cmd.print(QObject::tr("Sampled cloud created: %1 points").arg(cloud->size()));
  3930. cmd.clouds().emplace_back(cloud, desc.basename + QObject::tr("_SAMPLED_POINTS"), desc.path);
  3931. //save it as well
  3932. if (cmd.autoSaveMode())
  3933. {
  3934. QString errorStr = cmd.exportEntity(cmd.clouds().back());
  3935. if (!errorStr.isEmpty())
  3936. {
  3937. return cmd.error(errorStr);
  3938. }
  3939. }
  3940. }
  3941. if (progressDialog)
  3942. {
  3943. progressDialog->close();
  3944. QCoreApplication::processEvents();
  3945. }
  3946. return true;
  3947. }
  3948. CommandCompressFWF::CommandCompressFWF()
  3949. : ccCommandLineInterface::Command(QObject::tr("Compress FWF"), COMMAND_COMPRESS_FWF)
  3950. {}
  3951. bool CommandCompressFWF::process(ccCommandLineInterface& cmd)
  3952. {
  3953. if (cmd.clouds().empty())
  3954. {
  3955. return cmd.error(QObject::tr("No point cloud available. Be sure to open or generate one first!"));
  3956. }
  3957. for (CLCloudDesc& desc : cmd.clouds())
  3958. {
  3959. desc.pc->compressFWFData();
  3960. }
  3961. return true;
  3962. }
  3963. CommandCrop::CommandCrop()
  3964. : ccCommandLineInterface::Command(QObject::tr("Crop"), COMMAND_CROP)
  3965. {}
  3966. bool CommandCrop::process(ccCommandLineInterface& cmd)
  3967. {
  3968. if (cmd.arguments().empty())
  3969. {
  3970. return cmd.error(QObject::tr("Missing parameter: box extents after \"-%1\" (Xmin:Ymin:Zmin:Xmax:Ymax:Zmax)").arg(COMMAND_CROP));
  3971. }
  3972. if (cmd.clouds().empty() && cmd.meshes().empty())
  3973. {
  3974. return cmd.error(QObject::tr("No point cloud or mesh available. Be sure to open or generate one first!"));
  3975. }
  3976. //decode box extents
  3977. CCVector3 boxMin;
  3978. CCVector3 boxMax;
  3979. {
  3980. QString boxBlock = cmd.arguments().takeFirst();
  3981. QStringList tokens = boxBlock.split(':');
  3982. if (tokens.size() != 6)
  3983. {
  3984. return cmd.error(QObject::tr("Invalid parameter: box extents (expected format is 'Xmin:Ymin:Zmin:Xmax:Ymax:Zmax')").arg(COMMAND_CROP));
  3985. }
  3986. for (int i = 0; i < 6; ++i)
  3987. {
  3988. CCVector3* vec = (i < 3 ? &boxMin : &boxMax);
  3989. bool ok = true;
  3990. vec->u[i % 3] = static_cast<PointCoordinateType>(tokens[i].toDouble(&ok));
  3991. if (!ok)
  3992. {
  3993. return cmd.error(QObject::tr("Invalid parameter: box extents (component #%1 is not a valid number)").arg(i + 1));
  3994. }
  3995. }
  3996. }
  3997. //optional parameters
  3998. bool inside = true;
  3999. while (!cmd.arguments().empty())
  4000. {
  4001. QString argument = cmd.arguments().front();
  4002. if (ccCommandLineInterface::IsCommand(argument, COMMAND_CROP_OUTSIDE))
  4003. {
  4004. //local option confirmed, we can move on
  4005. cmd.arguments().pop_front();
  4006. inside = false;
  4007. }
  4008. else
  4009. {
  4010. break;
  4011. }
  4012. }
  4013. ccBBox cropBox(boxMin, boxMax, true);
  4014. //crop clouds
  4015. {
  4016. for (CLCloudDesc& desc : cmd.clouds())
  4017. {
  4018. ccHObject* croppedCloud = ccCropTool::Crop(desc.pc, cropBox, inside);
  4019. if (croppedCloud)
  4020. {
  4021. delete desc.pc;
  4022. assert(croppedCloud->isA(CC_TYPES::POINT_CLOUD));
  4023. desc.pc = static_cast<ccPointCloud*>(croppedCloud);
  4024. desc.basename += "_CROPPED";
  4025. if (cmd.autoSaveMode())
  4026. {
  4027. QString errorStr = cmd.exportEntity(desc);
  4028. if (!errorStr.isEmpty())
  4029. {
  4030. return cmd.error(errorStr);
  4031. }
  4032. }
  4033. }
  4034. else
  4035. {
  4036. //otherwise an error message has already been issued
  4037. delete desc.pc;
  4038. desc.pc = nullptr; //will be removed after this loop
  4039. //desc.basename += "_FULLY_CROPPED";
  4040. }
  4041. }
  4042. //now clean the set of clouds in case some have been 'cropped out'
  4043. for (auto it = cmd.clouds().begin(); it != cmd.clouds().end(); )
  4044. {
  4045. if (it->pc == nullptr)
  4046. {
  4047. it = cmd.clouds().erase(it);
  4048. }
  4049. else
  4050. {
  4051. ++it;
  4052. }
  4053. }
  4054. }
  4055. //crop meshes
  4056. {
  4057. for (CLMeshDesc& desc : cmd.meshes())
  4058. {
  4059. ccHObject* croppedMesh = ccCropTool::Crop(desc.mesh, cropBox, inside);
  4060. if (croppedMesh)
  4061. {
  4062. delete desc.mesh;
  4063. assert(croppedMesh->isA(CC_TYPES::MESH));
  4064. desc.mesh = static_cast<ccMesh*>(croppedMesh);
  4065. desc.basename += "_CROPPED";
  4066. if (cmd.autoSaveMode())
  4067. {
  4068. QString errorStr = cmd.exportEntity(desc);
  4069. if (!errorStr.isEmpty())
  4070. {
  4071. return cmd.error(errorStr);
  4072. }
  4073. }
  4074. }
  4075. else
  4076. {
  4077. //otherwise an error message has already been issued
  4078. delete desc.mesh;
  4079. desc.mesh = nullptr; //will be removed after this loop
  4080. //desc.basename += "_FULLY_CROPPED";
  4081. }
  4082. }
  4083. //now clean the set of meshes in case some have been 'cropped out'
  4084. for (auto it = cmd.meshes().begin(); it != cmd.meshes().end(); )
  4085. {
  4086. if (it->mesh == nullptr)
  4087. {
  4088. it = cmd.meshes().erase(it);
  4089. }
  4090. else
  4091. {
  4092. ++it;
  4093. }
  4094. }
  4095. }
  4096. return true;
  4097. }
  4098. CommandSFToCoord::CommandSFToCoord()
  4099. : ccCommandLineInterface::Command(QObject::tr("SF to Coord"), COMMAND_SF_TO_COORD)
  4100. {}
  4101. bool CommandSFToCoord::process(ccCommandLineInterface& cmd)
  4102. {
  4103. if (cmd.arguments().size() < 2)
  4104. {
  4105. return cmd.error(QObject::tr("Missing parameter(s) after \"-%1\" (SF INDEX OR NAME) (DIMENSION)").arg(COMMAND_SF_TO_COORD));
  4106. }
  4107. if (cmd.clouds().empty())
  4108. {
  4109. return cmd.error(QObject::tr("No point cloud available. Be sure to open or generate one first!"));
  4110. }
  4111. int sfIndex = -1;
  4112. QString sfName;
  4113. if (!GetSFIndexOrName(cmd, sfIndex, sfName))
  4114. {
  4115. return false;
  4116. }
  4117. //dimension
  4118. QString dimStr = cmd.arguments().takeFirst().toUpper();
  4119. bool exportDims[3] { dimStr == "X", dimStr == "Y", dimStr == "Z" };
  4120. if (!exportDims[0] && !exportDims[1] && !exportDims[2])
  4121. {
  4122. return cmd.error(QObject::tr("Invalid parameter: dimension after \"-%1\" (expected: X, Y or Z)").arg(COMMAND_SF_TO_COORD));
  4123. }
  4124. //now we can export the corresponding coordinate
  4125. for (CLCloudDesc& desc : cmd.clouds())
  4126. {
  4127. if (desc.pc)
  4128. {
  4129. CCCoreLib::ScalarField* sf = GetScalarField(desc.pc, sfIndex, sfName, true);
  4130. if (sf)
  4131. {
  4132. if (desc.pc->setCoordFromSF(exportDims, sf, std::numeric_limits<PointCoordinateType>::quiet_NaN()))
  4133. {
  4134. desc.basename += QObject::tr("_SF_TO_COORD_%1").arg(dimStr);
  4135. if (cmd.autoSaveMode())
  4136. {
  4137. QString errorStr = cmd.exportEntity(desc);
  4138. if (!errorStr.isEmpty())
  4139. {
  4140. return cmd.error(errorStr);
  4141. }
  4142. }
  4143. }
  4144. }
  4145. else
  4146. {
  4147. return cmd.error(QObject::tr("Failed to set SF %1 as coord %2 on cloud '%3'!").arg(sfName).arg(dimStr).arg(desc.pc->getName()));
  4148. }
  4149. }
  4150. }
  4151. return true;
  4152. }
  4153. CommandCoordToSF::CommandCoordToSF()
  4154. : ccCommandLineInterface::Command(QObject::tr("Coord to SF"), COMMAND_COORD_TO_SF)
  4155. {}
  4156. bool CommandCoordToSF::process(ccCommandLineInterface& cmd)
  4157. {
  4158. if (cmd.arguments().empty())
  4159. {
  4160. return cmd.error(QObject::tr("Missing parameter after \"-%1\" (DIMENSION)").arg(COMMAND_COORD_TO_SF));
  4161. }
  4162. if (cmd.clouds().empty())
  4163. {
  4164. return cmd.error(QObject::tr("No point cloud available. Be sure to open or generate one first!"));
  4165. }
  4166. //dimension
  4167. QString dimStr = cmd.arguments().takeFirst().toUpper();
  4168. bool exportDims[3]{ dimStr == "X", dimStr == "Y", dimStr == "Z" };
  4169. if (!exportDims[0] && !exportDims[1] && !exportDims[2])
  4170. {
  4171. return cmd.error(QObject::tr("Invalid parameter: dimension after \"-%1\" (expected: X, Y or Z)").arg(COMMAND_COORD_TO_SF));
  4172. }
  4173. //now we can export the corresponding coordinate
  4174. for (CLCloudDesc& desc : cmd.clouds())
  4175. {
  4176. if (desc.pc->exportCoordToSF(exportDims))
  4177. {
  4178. desc.basename += QObject::tr("_%1_TO_SF").arg(dimStr);
  4179. if (cmd.autoSaveMode())
  4180. {
  4181. QString errorStr = cmd.exportEntity(desc);
  4182. if (!errorStr.isEmpty())
  4183. {
  4184. return cmd.error(errorStr);
  4185. }
  4186. }
  4187. }
  4188. else
  4189. {
  4190. return cmd.error(QObject::tr("Failed to export coord. %1 to SF on cloud '%2'!").arg(dimStr, desc.pc->getName()));
  4191. }
  4192. }
  4193. return true;
  4194. }
  4195. CommandCrop2D::CommandCrop2D()
  4196. : ccCommandLineInterface::Command(QObject::tr("Crop 2D"), COMMAND_CROP_2D)
  4197. {}
  4198. bool CommandCrop2D::process(ccCommandLineInterface& cmd)
  4199. {
  4200. if (cmd.arguments().size() < 6)
  4201. {
  4202. return cmd.error(QObject::tr("Missing parameter(s) after \"-%1\" (ORTHO_DIM N X1 Y1 X2 Y2 ... XN YN)").arg(COMMAND_CROP_2D));
  4203. }
  4204. if (cmd.clouds().empty())
  4205. {
  4206. return cmd.error(QObject::tr("No point cloud available. Be sure to open or generate one first!"));
  4207. }
  4208. //decode poyline extents
  4209. ccPointCloud vertices("polyline.vertices");
  4210. ccPolyline poly(&vertices);
  4211. //orthogonal dimension
  4212. unsigned char orthoDim = 2;
  4213. bool orderFlipped = false;
  4214. {
  4215. QString orthoDimStr = cmd.arguments().takeFirst().toUpper();
  4216. if (orthoDimStr.endsWith("FLIP"))
  4217. {
  4218. orderFlipped = true;
  4219. orthoDimStr = orthoDimStr.left(orthoDimStr.size() - 4);
  4220. }
  4221. if (orthoDimStr == "X")
  4222. {
  4223. orthoDim = 0;
  4224. }
  4225. else if (orthoDimStr == "Y")
  4226. {
  4227. orthoDim = 1;
  4228. }
  4229. else if (orthoDimStr == "Z")
  4230. {
  4231. orthoDim = 2;
  4232. }
  4233. else
  4234. {
  4235. return cmd.error(QObject::tr("Invalid parameter: orthogonal dimension after \"-%1\" (expected: X, Y or Z)").arg(COMMAND_CROP_2D));
  4236. }
  4237. }
  4238. ccCommandLineInterface::GlobalShiftOptions globalShiftOptions;
  4239. globalShiftOptions.mode = ccCommandLineInterface::GlobalShiftOptions::NO_GLOBAL_SHIFT;
  4240. if (cmd.arguments().size() >= 4)
  4241. {
  4242. if (cmd.nextCommandIsGlobalShift())
  4243. {
  4244. //local option confirmed, we can move on
  4245. cmd.arguments().pop_front();
  4246. if (!cmd.processGlobalShiftCommand(globalShiftOptions))
  4247. {
  4248. //error message already issued
  4249. return false;
  4250. }
  4251. cmd.setGlobalShiftOptions(globalShiftOptions);
  4252. }
  4253. }
  4254. //number of vertices
  4255. bool ok = true;
  4256. unsigned N = 0;
  4257. {
  4258. QString countStr = cmd.arguments().takeFirst();
  4259. N = countStr.toUInt(&ok);
  4260. if (!ok)
  4261. {
  4262. return cmd.error(QObject::tr("Invalid parameter: number of vertices for the 2D polyline after \"-%1\"").arg(COMMAND_CROP_2D));
  4263. }
  4264. }
  4265. //now read the vertices
  4266. {
  4267. if (!vertices.reserve(N) || !poly.addPointIndex(0, N))
  4268. {
  4269. return cmd.error(QObject::tr("Not enough memory!"));
  4270. }
  4271. assert(orthoDim < 3);
  4272. unsigned char X = ((orthoDim + 1) % 3);
  4273. unsigned char Y = ((X + 1) % 3);
  4274. unsigned char Xread = X;
  4275. unsigned char Yread = Y;
  4276. if (orderFlipped)
  4277. {
  4278. std::swap(Xread, Yread);
  4279. }
  4280. CCVector3d PShift(0, 0, 0);
  4281. for (unsigned i = 0; i < N; ++i)
  4282. {
  4283. if (cmd.arguments().size() < 2)
  4284. {
  4285. return cmd.error(QObject::tr("Missing parameter(s): vertex #%1 data and following").arg(i + 1));
  4286. }
  4287. CCVector3d Pd(0, 0, 0);
  4288. QString coordStr = cmd.arguments().takeFirst();
  4289. Pd.u[Xread] = coordStr.toDouble(&ok);
  4290. if (!ok)
  4291. {
  4292. return cmd.error(QObject::tr("Invalid parameter: X-coordinate of vertex #%1").arg(i + 1));
  4293. }
  4294. /*QString */coordStr = cmd.arguments().takeFirst();
  4295. Pd.u[Yread] = coordStr.toDouble(&ok);
  4296. if (!ok)
  4297. {
  4298. return cmd.error(QObject::tr("Invalid parameter: Y-coordinate of vertex #%1").arg(i + 1));
  4299. }
  4300. if (i == 0)
  4301. {
  4302. bool preserveCoordinateShift = false; //ignored
  4303. if (FileIOFilter::HandleGlobalShift(Pd, PShift, preserveCoordinateShift, cmd.fileLoadingParams()))
  4304. {
  4305. ccLog::Warning(QString("[%1] 2D polyline has been recentered! Translation: (%2 ; %3 ; %4)").arg(COMMAND_CROP_2D).arg(PShift.x, 0, 'f', 2).arg(PShift.y, 0, 'f', 2).arg(PShift.z, 0, 'f', 2));
  4306. }
  4307. }
  4308. CCVector3 P3D = (Pd + PShift).toPC();
  4309. // warning: the polyline must be defined in the XY plane!
  4310. CCVector3 P2D;
  4311. {
  4312. P2D.x = P3D.u[X];
  4313. P2D.y = P3D.u[Y];
  4314. P2D.z = 0;
  4315. }
  4316. vertices.addPoint(P2D);
  4317. }
  4318. poly.setClosed(true);
  4319. }
  4320. cmd.updateInteralGlobalShift(globalShiftOptions);
  4321. //optional parameters
  4322. bool inside = true;
  4323. while (!cmd.arguments().empty())
  4324. {
  4325. QString argument = cmd.arguments().front();
  4326. if (ccCommandLineInterface::IsCommand(argument, COMMAND_CROP_OUTSIDE))
  4327. {
  4328. //local option confirmed, we can move on
  4329. cmd.arguments().pop_front();
  4330. inside = false;
  4331. }
  4332. else
  4333. {
  4334. break;
  4335. }
  4336. }
  4337. //now we can crop the loaded cloud(s)
  4338. for (CLCloudDesc& desc : cmd.clouds())
  4339. {
  4340. CCCoreLib::ReferenceCloud* ref = desc.pc->crop2D(&poly, orthoDim, inside);
  4341. if (ref)
  4342. {
  4343. if (ref->size() != 0)
  4344. {
  4345. ccPointCloud* croppedCloud = desc.pc->partialClone(ref);
  4346. delete ref;
  4347. ref = nullptr;
  4348. if (croppedCloud)
  4349. {
  4350. delete desc.pc;
  4351. desc.pc = croppedCloud;
  4352. croppedCloud->setName(desc.pc->getName() + QObject::tr(".cropped"));
  4353. desc.basename += "_CROPPED";
  4354. if (cmd.autoSaveMode())
  4355. {
  4356. QString errorStr = cmd.exportEntity(desc);
  4357. if (!errorStr.isEmpty())
  4358. {
  4359. return cmd.error(errorStr);
  4360. }
  4361. }
  4362. }
  4363. else
  4364. {
  4365. return cmd.error(QObject::tr("Not enough memory to crop cloud '%1'!").arg(desc.pc->getName()));
  4366. }
  4367. }
  4368. else
  4369. {
  4370. delete ref;
  4371. ref = nullptr;
  4372. cmd.warning(QObject::tr("No point of cloud '%1' falls inside the input box!").arg(desc.pc->getName()));
  4373. }
  4374. }
  4375. else
  4376. {
  4377. return cmd.error(QObject::tr("Crop process failed! (not enough memory)"));
  4378. }
  4379. }
  4380. return true;
  4381. }
  4382. CommandColorBanding::CommandColorBanding()
  4383. : ccCommandLineInterface::Command(QObject::tr("Color banding"), COMMAND_COLOR_BANDING)
  4384. {}
  4385. bool CommandColorBanding::process(ccCommandLineInterface& cmd)
  4386. {
  4387. if (cmd.arguments().size() < 2)
  4388. {
  4389. return cmd.error(QObject::tr("Missing parameter(s) after \"-%1\" (DIM FREQUENCY)").arg(COMMAND_COLOR_BANDING));
  4390. }
  4391. if (cmd.clouds().empty() && cmd.meshes().empty())
  4392. {
  4393. return cmd.error(QObject::tr("No entity available. Be sure to open or generate one first!"));
  4394. }
  4395. //dimension
  4396. unsigned char dim = 2;
  4397. QString dimStr = "Z";
  4398. {
  4399. dimStr = cmd.arguments().takeFirst().toUpper();
  4400. if (dimStr == "X")
  4401. {
  4402. dim = 0;
  4403. }
  4404. else if (dimStr == "Y")
  4405. {
  4406. dim = 1;
  4407. }
  4408. else if (dimStr == "Z")
  4409. {
  4410. dim = 2;
  4411. }
  4412. else
  4413. {
  4414. return cmd.error(QObject::tr("Invalid parameter: dimension after \"-%1\" (expected: X, Y or Z)").arg(COMMAND_COLOR_BANDING));
  4415. }
  4416. }
  4417. //frequency
  4418. bool ok = true;
  4419. double freq = 0;
  4420. {
  4421. QString countStr = cmd.arguments().takeFirst();
  4422. freq = countStr.toDouble(&ok);
  4423. if (!ok)
  4424. {
  4425. return cmd.error(QObject::tr("Invalid parameter: frequency after \"-%1 DIM\" (in Hz, integer value)").arg(COMMAND_COLOR_BANDING));
  4426. }
  4427. }
  4428. //process clouds
  4429. if (!cmd.clouds().empty())
  4430. {
  4431. bool hasclouds = false;
  4432. for (CLCloudDesc& desc : cmd.clouds())
  4433. {
  4434. if (desc.pc)
  4435. {
  4436. if (!desc.pc->setRGBColorByBanding(dim, freq))
  4437. {
  4438. return cmd.error(QObject::tr("Not enough memory"));
  4439. }
  4440. else
  4441. {
  4442. hasclouds = true;
  4443. }
  4444. }
  4445. }
  4446. //save output
  4447. if (hasclouds && cmd.autoSaveMode() && !cmd.saveClouds(QObject::tr("COLOR_BANDING_%1_%2").arg(dimStr).arg(freq)))
  4448. {
  4449. return false;
  4450. }
  4451. }
  4452. if (!cmd.meshes().empty())
  4453. {
  4454. bool hasMeshes = false;
  4455. for (CLMeshDesc& desc : cmd.meshes())
  4456. {
  4457. ccPointCloud* cloud = ccHObjectCaster::ToPointCloud(desc.mesh);
  4458. if (cloud)
  4459. {
  4460. if (!cloud->setRGBColorByBanding(dim, freq))
  4461. {
  4462. return cmd.error(QObject::tr("Not enough memory"));
  4463. }
  4464. else
  4465. {
  4466. desc.mesh->showColors(true);
  4467. hasMeshes = true;
  4468. }
  4469. }
  4470. else
  4471. {
  4472. cmd.warning(QObject::tr("Vertices of mesh '%1' are locked (they may be shared by multiple entities for instance). Can't apply the current command on them.").arg(desc.mesh->getName()));
  4473. }
  4474. }
  4475. //save output
  4476. if (hasMeshes && cmd.autoSaveMode() && !cmd.saveMeshes(QObject::tr("COLOR_BANDING_%1_%2").arg(dimStr).arg(freq)))
  4477. {
  4478. return false;
  4479. }
  4480. }
  4481. return true;
  4482. }
  4483. CommandColorLevels::CommandColorLevels()
  4484. : ccCommandLineInterface::Command(QObject::tr("Color levels"), COMMAND_COLOR_LEVELS)
  4485. {}
  4486. bool CommandColorLevels::process(ccCommandLineInterface& cmd)
  4487. {
  4488. if (cmd.arguments().size() < 5)
  4489. {
  4490. return cmd.error(QObject::tr("Missing parameter(s) after \"-%1\" (COLOR-BANDS MIN-INPUT-LEVEL MAX-INPUT-LEVEL MIN-OUTPUT-LEVEL MAX-OUTPUT-LEVEL)").arg(COMMAND_COLOR_LEVELS));
  4491. }
  4492. if (cmd.clouds().empty() && cmd.meshes().empty())
  4493. {
  4494. return cmd.error(QObject::tr("No entity available. Be sure to open or generate one first!"));
  4495. }
  4496. //color bands
  4497. QString band = cmd.arguments().takeFirst().toUpper();
  4498. bool rgb[3] { band.contains('R'), band.contains('G'), band.contains('B') };
  4499. {
  4500. QString testBand = band;
  4501. testBand.remove('R');
  4502. testBand.remove('G');
  4503. testBand.remove('B');
  4504. if (!testBand.isEmpty())
  4505. {
  4506. return cmd.error(QObject::tr("Invalid parameter: bands after \"-%1\" (expected: any combination of R, G or B)").arg(COMMAND_COLOR_LEVELS));
  4507. }
  4508. }
  4509. //min level
  4510. int levels[4] = { 0 };
  4511. for (int i = 0; i < 4; ++i)
  4512. {
  4513. bool ok = true;
  4514. QString levelStr = cmd.arguments().takeFirst();
  4515. levels[i] = levelStr.toInt(&ok);
  4516. if (!ok || levels[i] < 0 || levels[i] > 255)
  4517. {
  4518. return cmd.error(QObject::tr("Invalid parameter: color level after \"-%1 COLOR-BANDS\" (integer value between 0 and 255 expected)").arg(COMMAND_COLOR_LEVELS));
  4519. }
  4520. }
  4521. //process clouds
  4522. if (!cmd.clouds().empty())
  4523. {
  4524. bool hasClouds = false;
  4525. for (CLCloudDesc& desc : cmd.clouds())
  4526. {
  4527. if (desc.pc && desc.pc->hasColors())
  4528. {
  4529. if (!ccColorLevelsDlg::ScaleColorFields(desc.pc, levels[0], levels[1], levels[2], levels[3], rgb))
  4530. {
  4531. cmd.warning(QObject::tr("Failed to scale the color band(s) of cloud '%1'").arg(desc.pc->getName()));
  4532. }
  4533. else
  4534. {
  4535. hasClouds = true;
  4536. }
  4537. }
  4538. }
  4539. //save output
  4540. if (hasClouds && cmd.autoSaveMode() && !cmd.saveClouds(QObject::tr("COLOR_LEVELS_%1_%2_%3").arg(band).arg(levels[2]).arg(levels[3])))
  4541. {
  4542. return false;
  4543. }
  4544. }
  4545. if (!cmd.meshes().empty())
  4546. {
  4547. bool hasMeshes = false;
  4548. for (CLMeshDesc& desc : cmd.meshes())
  4549. {
  4550. ccPointCloud* cloud = ccHObjectCaster::ToPointCloud(desc.mesh);
  4551. if (cloud && cloud->hasColors())
  4552. {
  4553. if (!ccColorLevelsDlg::ScaleColorFields(cloud, levels[0], levels[1], levels[2], levels[3], rgb))
  4554. {
  4555. cmd.warning(QObject::tr("Failed to scale the color band(s) of mesh '%1'").arg(desc.mesh->getName()));
  4556. }
  4557. else
  4558. {
  4559. hasMeshes = true;
  4560. }
  4561. }
  4562. else if (desc.mesh->hasColors())
  4563. {
  4564. cmd.warning(QObject::tr("Vertices of mesh '%1' are locked (they may be shared by multiple entities for instance). Can't apply the current command on them.").arg(desc.mesh->getName()));
  4565. }
  4566. }
  4567. //save output
  4568. if (hasMeshes && cmd.autoSaveMode() && !cmd.saveMeshes(QObject::tr("COLOR_LEVELS_%1_%2_%3").arg(band).arg(levels[2]).arg(levels[3])))
  4569. {
  4570. return false;
  4571. }
  4572. }
  4573. return true;
  4574. }
  4575. CommandDist::CommandDist(bool cloud2meshDist, const QString& name, const QString& keyword)
  4576. : ccCommandLineInterface::Command(name, keyword)
  4577. , m_cloud2meshDist(cloud2meshDist)
  4578. {}
  4579. bool CommandDist::process(ccCommandLineInterface& cmd)
  4580. {
  4581. //compared cloud
  4582. CLEntityDesc* compEntity = nullptr;
  4583. ccHObject* compCloud = nullptr;
  4584. size_t nextMeshIndex = 0;
  4585. if (cmd.clouds().empty())
  4586. {
  4587. //no cloud loaded
  4588. if (!m_cloud2meshDist || cmd.meshes().size() < 2)
  4589. {
  4590. //we would need at least two meshes
  4591. return cmd.error(QObject::tr("No point cloud available. Be sure to open or generate one first!"));
  4592. }
  4593. else
  4594. {
  4595. cmd.warning(QObject::tr("No point cloud available. Will use the first mesh vertices as compared cloud."));
  4596. compEntity = &(cmd.meshes().front());
  4597. compCloud = dynamic_cast<ccPointCloud*>(cmd.meshes()[nextMeshIndex++].mesh->getAssociatedCloud());
  4598. if (!compCloud)
  4599. {
  4600. return cmd.error(QObject::tr("Unhandled mesh vertices type"));
  4601. }
  4602. }
  4603. }
  4604. else //at least two clouds
  4605. {
  4606. if (m_cloud2meshDist && cmd.clouds().size() != 1)
  4607. {
  4608. cmd.warning(QObject::tr("[C2M] Multiple point clouds loaded! Will take the first one by default."));
  4609. }
  4610. compEntity = &(cmd.clouds().front());
  4611. compCloud = cmd.clouds().front().pc;
  4612. }
  4613. assert(compEntity && compCloud);
  4614. //reference entity
  4615. ccHObject* refEntity = nullptr;
  4616. if (m_cloud2meshDist)
  4617. {
  4618. if (cmd.meshes().size() <= nextMeshIndex)
  4619. {
  4620. return cmd.error(QObject::tr("No mesh available. Be sure to open one first!"));
  4621. }
  4622. else if (cmd.meshes().size() != nextMeshIndex + 1)
  4623. {
  4624. cmd.warning(QString("Multiple meshes loaded! We take the %1 one by default").arg(nextMeshIndex == 0 ? "first" : "second"));
  4625. }
  4626. refEntity = cmd.meshes()[nextMeshIndex].mesh;
  4627. }
  4628. else
  4629. {
  4630. if (cmd.clouds().size() < 2)
  4631. {
  4632. return cmd.error(QObject::tr("Only one point cloud available. Be sure to open or generate a second one before performing C2C distance!"));
  4633. }
  4634. else if (cmd.clouds().size() > 2)
  4635. {
  4636. cmd.warning(QObject::tr("More than 3 point clouds loaded! We take the second one as reference by default"));
  4637. }
  4638. refEntity = cmd.clouds()[1].pc;
  4639. }
  4640. //inner loop for Distance computation options
  4641. bool flipNormals = false;
  4642. bool unsignedDistances = false;
  4643. bool robust = true;
  4644. double maxDist = 0.0;
  4645. unsigned octreeLevel = 0;
  4646. int maxThreadCount = 0;
  4647. bool splitXYZ = false;
  4648. bool mergeXY = false;
  4649. int modelIndex = 0;
  4650. bool useKNN = true;
  4651. double nSize = 0;
  4652. while (!cmd.arguments().empty())
  4653. {
  4654. QString argument = cmd.arguments().front();
  4655. if (ccCommandLineInterface::IsCommand(argument, COMMAND_C2M_DIST_FLIP_NORMALS))
  4656. {
  4657. //local option confirmed, we can move on
  4658. cmd.arguments().pop_front();
  4659. flipNormals = true;
  4660. if (!m_cloud2meshDist)
  4661. {
  4662. cmd.warning(QObject::tr("Parameter \"-%1\" ignored: only for C2M distance!").arg(COMMAND_C2M_DIST_FLIP_NORMALS));
  4663. }
  4664. }
  4665. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_C2M_DIST_UNSIGNED))
  4666. {
  4667. //local option confirmed, we can move on
  4668. cmd.arguments().pop_front();
  4669. unsignedDistances = true;
  4670. if (!m_cloud2meshDist)
  4671. {
  4672. cmd.warning(QObject::tr("Parameter \"-%1\" ignored: only for C2M distance!").arg(COMMAND_C2M_DIST_UNSIGNED));
  4673. }
  4674. }
  4675. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_C2M_DIST_NON_ROBUST))
  4676. {
  4677. //local option confirmed, we can move on
  4678. cmd.arguments().pop_front();
  4679. robust = false;
  4680. if (!m_cloud2meshDist)
  4681. {
  4682. cmd.warning(QObject::tr("Parameter \"-%1\" ignored: only for C2M distance!").arg(COMMAND_C2M_DIST_NON_ROBUST));
  4683. }
  4684. }
  4685. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_C2X_MAX_DISTANCE))
  4686. {
  4687. //local option confirmed, we can move on
  4688. cmd.arguments().pop_front();
  4689. if (cmd.arguments().empty())
  4690. {
  4691. return cmd.error(QObject::tr("Missing parameter: value after \"-%1\"").arg(COMMAND_C2X_MAX_DISTANCE));
  4692. }
  4693. bool conversionOk = false;
  4694. maxDist = cmd.arguments().takeFirst().toDouble(&conversionOk);
  4695. if (!conversionOk)
  4696. {
  4697. return cmd.error(QObject::tr("Invalid parameter: value after \"-%1\"").arg(COMMAND_C2X_MAX_DISTANCE));
  4698. }
  4699. }
  4700. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_C2X_OCTREE_LEVEL))
  4701. {
  4702. //local option confirmed, we can move on
  4703. cmd.arguments().pop_front();
  4704. if (cmd.arguments().empty())
  4705. {
  4706. return cmd.error(QObject::tr("Missing parameter: value after \"-%1\"").arg(COMMAND_C2X_OCTREE_LEVEL));
  4707. }
  4708. bool conversionOk = false;
  4709. octreeLevel = cmd.arguments().takeFirst().toUInt(&conversionOk);
  4710. if (!conversionOk)
  4711. {
  4712. return cmd.error(QObject::tr("Invalid parameter: value after \"-%1\"").arg(COMMAND_C2X_OCTREE_LEVEL));
  4713. }
  4714. }
  4715. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_C2C_SPLIT_XYZ))
  4716. {
  4717. //local option confirmed, we can move on
  4718. cmd.arguments().pop_front();
  4719. splitXYZ = true;
  4720. if (m_cloud2meshDist)
  4721. {
  4722. cmd.warning(QObject::tr("Parameter \"-%1\" ignored: only for C2C distance!"));
  4723. }
  4724. }
  4725. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_C2C_SPLIT_XY_Z))
  4726. {
  4727. //local option confirmed, we can move on
  4728. cmd.arguments().pop_front();
  4729. splitXYZ = true;
  4730. mergeXY = true;
  4731. if (m_cloud2meshDist)
  4732. {
  4733. cmd.warning(QObject::tr("Parameter \"-%1\" ignored: only for C2C distance!"));
  4734. }
  4735. }
  4736. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_C2C_LOCAL_MODEL))
  4737. {
  4738. //local option confirmed, we can move on
  4739. cmd.arguments().pop_front();
  4740. if (!cmd.arguments().empty())
  4741. {
  4742. QString modelType = cmd.arguments().takeFirst().toUpper();
  4743. if (modelType == "LS")
  4744. {
  4745. modelIndex = 1;
  4746. }
  4747. else if (modelType == "TRI")
  4748. {
  4749. modelIndex = 2;
  4750. }
  4751. else if (modelType == "HF")
  4752. {
  4753. modelIndex = 3;
  4754. }
  4755. else
  4756. {
  4757. return cmd.error(QObject::tr("Invalid parameter: unknown model type \"%1\"").arg(modelType));
  4758. }
  4759. }
  4760. else
  4761. {
  4762. return cmd.error(QObject::tr("Missing parameter: model type after \"-%1\" (LS/TRI/HF)").arg(COMMAND_C2C_LOCAL_MODEL));
  4763. }
  4764. if (!cmd.arguments().empty())
  4765. {
  4766. QString nType = cmd.arguments().takeFirst().toUpper();
  4767. if (nType == "KNN")
  4768. {
  4769. useKNN = true;
  4770. }
  4771. else if (nType == "SPHERE")
  4772. {
  4773. useKNN = false;
  4774. }
  4775. else
  4776. {
  4777. return cmd.error(QObject::tr("Invalid parameter: unknown neighborhood type \"%1\"").arg(nType));
  4778. }
  4779. }
  4780. else
  4781. {
  4782. return cmd.error(QObject::tr("Missing parameter: expected neighborhood type after model type (KNN/SPHERE)"));
  4783. }
  4784. //neighborhood size
  4785. if (!cmd.arguments().empty())
  4786. {
  4787. bool conversionOk = false;
  4788. nSize = cmd.arguments().takeFirst().toDouble(&conversionOk);
  4789. if (!conversionOk)
  4790. {
  4791. return cmd.error(QObject::tr("Invalid parameter: neighborhood size"));
  4792. }
  4793. }
  4794. else
  4795. {
  4796. return cmd.error(QObject::tr("Missing parameter: expected neighborhood size after neighborhood type (neighbor count/sphere radius)"));
  4797. }
  4798. }
  4799. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_MAX_THREAD_COUNT))
  4800. {
  4801. //local option confirmed, we can move on
  4802. cmd.arguments().pop_front();
  4803. if (cmd.arguments().empty())
  4804. {
  4805. return cmd.error(QObject::tr("Missing parameter: max thread count after '%1'").arg(COMMAND_MAX_THREAD_COUNT));
  4806. }
  4807. bool ok;
  4808. maxThreadCount = cmd.arguments().takeFirst().toInt(&ok);
  4809. if (!ok || maxThreadCount < 0)
  4810. {
  4811. return cmd.error(QObject::tr("Invalid thread count! (after %1)").arg(COMMAND_MAX_THREAD_COUNT));
  4812. }
  4813. }
  4814. else
  4815. {
  4816. break; //as soon as we encounter an unrecognized argument, we break the local loop to go back to the main one!
  4817. }
  4818. }
  4819. //spawn dialog (virtually) so as to prepare the comparison process
  4820. ccComparisonDlg compDlg(compCloud,
  4821. refEntity,
  4822. m_cloud2meshDist ? ccComparisonDlg::CLOUDMESH_DIST : ccComparisonDlg::CLOUDCLOUD_DIST,
  4823. cmd.widgetParent(),
  4824. true);
  4825. if (!compDlg.initDialog())
  4826. {
  4827. return cmd.error(QObject::tr("Failed to initialize comparison dialog"));
  4828. }
  4829. //update parameters
  4830. if (maxDist > 0)
  4831. {
  4832. compDlg.maxDistCheckBox->setChecked(true);
  4833. compDlg.maxSearchDistSpinBox->setValue(maxDist);
  4834. }
  4835. if (octreeLevel > 0)
  4836. {
  4837. compDlg.octreeLevelComboBox->setCurrentIndex(octreeLevel);
  4838. }
  4839. if (maxThreadCount != 0)
  4840. {
  4841. compDlg.maxThreadCountSpinBox->setValue(maxThreadCount);
  4842. }
  4843. if (m_cloud2meshDist)
  4844. {
  4845. //C2M-only parameters
  4846. compDlg.flipNormalsCheckBox->setChecked(flipNormals);
  4847. compDlg.signedDistCheckBox->setChecked(!unsignedDistances);
  4848. compDlg.robustCheckBox->setChecked(robust);
  4849. }
  4850. else
  4851. {
  4852. //C2C-only parameters
  4853. if (splitXYZ)
  4854. {
  4855. //DGM: not true anymore
  4856. //if (maxDist > 0)
  4857. // cmd.warning("'Split XYZ' option is ignored if max distance is defined!");
  4858. compDlg.split3DCheckBox->setChecked(true);
  4859. }
  4860. if (mergeXY)
  4861. compDlg.compute2DCheckBox->setChecked(true);
  4862. if (modelIndex != 0)
  4863. {
  4864. compDlg.localModelComboBox->setCurrentIndex(modelIndex);
  4865. if (useKNN)
  4866. {
  4867. compDlg.lmKNNRadioButton->setChecked(true);
  4868. compDlg.lmKNNSpinBox->setValue(static_cast<int>(nSize));
  4869. }
  4870. else
  4871. {
  4872. compDlg.lmRadiusRadioButton->setChecked(true);
  4873. compDlg.lmRadiusDoubleSpinBox->setValue(nSize);
  4874. }
  4875. }
  4876. }
  4877. if (!compDlg.computeDistances())
  4878. {
  4879. compDlg.cancelAndExit();
  4880. return cmd.error(QObject::tr("An error occurred during distances computation!"));
  4881. }
  4882. compDlg.applyAndExit();
  4883. QString suffix(m_cloud2meshDist ? "_C2M_DIST" : "_C2C_DIST");
  4884. if (maxDist > 0)
  4885. {
  4886. suffix += QObject::tr("_MAX_DIST_%1").arg(maxDist);
  4887. }
  4888. compEntity->basename += suffix;
  4889. if (cmd.autoSaveMode())
  4890. {
  4891. QString errorStr = cmd.exportEntity(*compEntity);
  4892. if (!errorStr.isEmpty())
  4893. {
  4894. return cmd.error(errorStr);
  4895. }
  4896. }
  4897. return true;
  4898. }
  4899. CommandC2MDist::CommandC2MDist()
  4900. : CommandDist(true, QObject::tr("C2M distance"), COMMAND_C2M_DIST)
  4901. {}
  4902. CommandC2CDist::CommandC2CDist()
  4903. : CommandDist(false, QObject::tr("C2C distance"), COMMAND_C2C_DIST)
  4904. {}
  4905. CommandCPS::CommandCPS()
  4906. : ccCommandLineInterface::Command(QObject::tr("Closest Point Set"), COMMAND_CLOSEST_POINT_SET)
  4907. {}
  4908. bool CommandCPS::process(ccCommandLineInterface& cmd)
  4909. {
  4910. if (cmd.clouds().size() < 2)
  4911. {
  4912. return cmd.error(QObject::tr("At least two point clouds are needed to compute the closest point set!"));
  4913. }
  4914. else if (cmd.clouds().size() > 2)
  4915. {
  4916. cmd.warning(QObject::tr("More than 3 point clouds loaded! We take the second one as reference by default"));
  4917. }
  4918. // COMPARED CLOUD / REFERENCE CLOUD
  4919. CLCloudDesc compDesc = cmd.clouds().front();
  4920. CLCloudDesc refDesc = cmd.clouds()[1];
  4921. ccPointCloud* compPointCloud = compDesc.pc;
  4922. ccPointCloud* refPointCloud = refDesc.pc;
  4923. assert(compPointCloud && refPointCloud);
  4924. ccProgressDialog pDlg(true, nullptr);
  4925. CCCoreLib::DistanceComputationTools::Cloud2CloudDistancesComputationParams params;
  4926. CCCoreLib::ReferenceCloud closestPointSet(refPointCloud);
  4927. params.CPSet = &closestPointSet;
  4928. // COMPUTE CLOUD 2 CLOUD DISTANCE, THIS INCLUDES THE CLOSEST POINT SET GENERATION
  4929. int result = CCCoreLib::DistanceComputationTools::computeCloud2CloudDistances(compPointCloud, refPointCloud, params, &pDlg);
  4930. if (result >= 0)
  4931. {
  4932. // the extracted CPS will get the attributes of the reference cloud
  4933. ccPointCloud* newCloud = refPointCloud->partialClone(&closestPointSet);
  4934. // give to newCloud a name similar to the one generated by the GUI Closest Point Set tool
  4935. CLCloudDesc desc(
  4936. newCloud,
  4937. "[" + refDesc.basename + "]_CPSet(" + compDesc.basename + ")",
  4938. cmd.clouds()[0].path);
  4939. QString errorStr = cmd.exportEntity(desc, QString(), nullptr, ccCommandLineInterface::ExportOption::ForceNoTimestamp);
  4940. if (!errorStr.isEmpty())
  4941. {
  4942. cmd.error(errorStr);
  4943. }
  4944. //add cloud to the current pool
  4945. cmd.clouds().push_back(desc);
  4946. }
  4947. return true;
  4948. }
  4949. CommandStatTest::CommandStatTest()
  4950. : ccCommandLineInterface::Command(QObject::tr("Statistical test"), COMMAND_STAT_TEST)
  4951. {}
  4952. bool CommandStatTest::process(ccCommandLineInterface& cmd)
  4953. {
  4954. //distribution
  4955. CCCoreLib::GenericDistribution* distrib = nullptr;
  4956. {
  4957. if (cmd.arguments().empty())
  4958. {
  4959. return cmd.error(QObject::tr("Missing parameter: distribution type after \"-%1\" (GAUSS/WEIBULL)").arg(COMMAND_STAT_TEST));
  4960. }
  4961. QString distribStr = cmd.arguments().takeFirst().toUpper();
  4962. if (distribStr == "GAUSS")
  4963. {
  4964. //mu
  4965. if (cmd.arguments().empty())
  4966. {
  4967. return cmd.error(QObject::tr("Missing parameter: mean value after \"GAUSS\""));
  4968. }
  4969. bool conversionOk = false;
  4970. double mu = cmd.arguments().takeFirst().toDouble(&conversionOk);
  4971. if (!conversionOk)
  4972. {
  4973. return cmd.error(QObject::tr("Invalid parameter: mean value after \"GAUSS\""));
  4974. }
  4975. //sigma
  4976. if (cmd.arguments().empty())
  4977. {
  4978. return cmd.error(QObject::tr("Missing parameter: sigma value after \"GAUSS\" {mu}"));
  4979. }
  4980. conversionOk = false;
  4981. double sigma = cmd.arguments().takeFirst().toDouble(&conversionOk);
  4982. if (!conversionOk)
  4983. {
  4984. return cmd.error(QObject::tr("Invalid parameter: sigma value after \"GAUSS\" {mu}"));
  4985. }
  4986. CCCoreLib::NormalDistribution* N = new CCCoreLib::NormalDistribution();
  4987. N->setParameters(static_cast<ScalarType>(mu), static_cast<ScalarType>(sigma*sigma)); //warning: we input sigma2 here (not sigma)
  4988. distrib = static_cast<CCCoreLib::GenericDistribution*>(N);
  4989. }
  4990. else if (distribStr == "WEIBULL")
  4991. {
  4992. //a
  4993. if (cmd.arguments().empty())
  4994. {
  4995. return cmd.error(QObject::tr("Missing parameter: a value after \"WEIBULL\""));
  4996. }
  4997. bool conversionOk = false;
  4998. double a = cmd.arguments().takeFirst().toDouble(&conversionOk);
  4999. if (!conversionOk)
  5000. {
  5001. return cmd.error(QObject::tr("Invalid parameter: a value after \"WEIBULL\""));
  5002. }
  5003. //b
  5004. if (cmd.arguments().empty())
  5005. {
  5006. return cmd.error(QObject::tr("Missing parameter: b value after \"WEIBULL\" {a}"));
  5007. }
  5008. conversionOk = false;
  5009. double b = cmd.arguments().takeFirst().toDouble(&conversionOk);
  5010. if (!conversionOk)
  5011. {
  5012. return cmd.error(QObject::tr("Invalid parameter: b value after \"WEIBULL\" {a}"));
  5013. }
  5014. //c
  5015. if (cmd.arguments().empty())
  5016. {
  5017. return cmd.error(QObject::tr("Missing parameter: shift value after \"WEIBULL\" {a} {b}"));
  5018. }
  5019. conversionOk = false;
  5020. double shift = cmd.arguments().takeFirst().toDouble(&conversionOk);
  5021. if (!conversionOk)
  5022. {
  5023. return cmd.error(QObject::tr("Invalid parameter: shift value after \"WEIBULL\" {a} {b}"));
  5024. }
  5025. CCCoreLib::WeibullDistribution* N = new CCCoreLib::WeibullDistribution();
  5026. N->setParameters(static_cast<ScalarType>(a), static_cast<ScalarType>(b), static_cast<ScalarType>(shift));
  5027. distrib = static_cast<CCCoreLib::GenericDistribution*>(N);
  5028. }
  5029. else
  5030. {
  5031. return cmd.error(QObject::tr("Invalid parameter: unknown distribution \"%1\"").arg(distribStr));
  5032. }
  5033. }
  5034. //pValue
  5035. double pValue = 0.0005;
  5036. {
  5037. if (cmd.arguments().empty())
  5038. {
  5039. return cmd.error(QObject::tr("Missing parameter: p-value after distribution"));
  5040. }
  5041. bool conversionOk = false;
  5042. pValue = cmd.arguments().takeFirst().toDouble(&conversionOk);
  5043. if (!conversionOk)
  5044. {
  5045. return cmd.error(QObject::tr("Invalid parameter: p-value after distribution"));
  5046. }
  5047. }
  5048. //kNN
  5049. unsigned kNN = 16;
  5050. {
  5051. if (cmd.arguments().empty())
  5052. {
  5053. return cmd.error(QObject::tr("Missing parameter: neighbors after p-value"));
  5054. }
  5055. bool conversionOk = false;
  5056. kNN = cmd.arguments().takeFirst().toUInt(&conversionOk);
  5057. if (!conversionOk)
  5058. {
  5059. return cmd.error(QObject::tr("Invalid parameter: neighbors after p-value"));
  5060. }
  5061. }
  5062. if (cmd.clouds().empty())
  5063. {
  5064. return cmd.error(QObject::tr("No cloud available. Be sure to open one first!"));
  5065. }
  5066. QScopedPointer<ccProgressDialog> progressDialog(nullptr);
  5067. if (!cmd.silentMode())
  5068. {
  5069. progressDialog.reset(new ccProgressDialog(false, cmd.widgetParent()));
  5070. progressDialog->setAutoClose(false);
  5071. }
  5072. for (CLCloudDesc& desc : cmd.clouds())
  5073. {
  5074. //we apply method on currently 'output' SF
  5075. CCCoreLib::ScalarField* outSF = desc.pc->getCurrentOutScalarField();
  5076. if (outSF)
  5077. {
  5078. assert(outSF->capacity() != 0);
  5079. //force Chi2 Distances field as 'IN' field (create it by the way if necessary)
  5080. int chi2SfIdx = desc.pc->getScalarFieldIndexByName(CC_CHI2_DISTANCES_DEFAULT_SF_NAME);
  5081. if (chi2SfIdx < 0)
  5082. {
  5083. chi2SfIdx = desc.pc->addScalarField(CC_CHI2_DISTANCES_DEFAULT_SF_NAME);
  5084. }
  5085. if (chi2SfIdx < 0)
  5086. {
  5087. delete distrib;
  5088. return cmd.error(QObject::tr("Couldn't allocate a new scalar field for computing chi2 distances! Try to free some memory ..."));
  5089. }
  5090. desc.pc->setCurrentInScalarField(chi2SfIdx);
  5091. //compute octree if necessary
  5092. ccOctree::Shared theOctree = desc.pc->getOctree();
  5093. if (!theOctree)
  5094. {
  5095. theOctree = desc.pc->computeOctree(progressDialog.data());
  5096. if (!theOctree)
  5097. {
  5098. delete distrib;
  5099. cmd.error(QObject::tr("Couldn't compute octree for cloud '%1'!").arg(desc.pc->getName()));
  5100. break;
  5101. }
  5102. }
  5103. double chi2dist = CCCoreLib::StatisticalTestingTools::testCloudWithStatisticalModel(distrib, desc.pc, kNN, pValue, progressDialog.data(), theOctree.data());
  5104. cmd.print(QObject::tr("[Chi2 Test] %1 test result = %2").arg(distrib->getName()).arg(chi2dist));
  5105. //we set the theoretical Chi2 distance limit as the minimum displayed SF value so that all points below are grayed
  5106. {
  5107. ccScalarField* chi2SF = static_cast<ccScalarField*>(desc.pc->getCurrentInScalarField());
  5108. assert(chi2SF);
  5109. chi2SF->computeMinAndMax();
  5110. chi2dist *= chi2dist;
  5111. chi2SF->setMinDisplayed(static_cast<ScalarType>(chi2dist));
  5112. chi2SF->setSymmetricalScale(false);
  5113. chi2SF->setSaturationStart(static_cast<ScalarType>(chi2dist));
  5114. //chi2SF->setSaturationStop(chi2dist);
  5115. desc.pc->setCurrentDisplayedScalarField(chi2SfIdx);
  5116. desc.pc->showSF(true);
  5117. }
  5118. desc.basename += QObject::tr("_STAT_TEST_%1").arg(distrib->getName());
  5119. if (cmd.autoSaveMode())
  5120. {
  5121. QString errorStr = cmd.exportEntity(desc);
  5122. if (!errorStr.isEmpty())
  5123. {
  5124. return cmd.error(errorStr);
  5125. }
  5126. }
  5127. }
  5128. }
  5129. if (progressDialog)
  5130. {
  5131. progressDialog->close();
  5132. QCoreApplication::processEvents();
  5133. }
  5134. return true;
  5135. }
  5136. CommandDelaunayTri::CommandDelaunayTri()
  5137. : ccCommandLineInterface::Command(QObject::tr("Delaunay triangulation"), COMMAND_DELAUNAY)
  5138. {}
  5139. bool CommandDelaunayTri::process(ccCommandLineInterface& cmd)
  5140. {
  5141. bool axisAligned = true;
  5142. double maxEdgeLength = 0;
  5143. while (!cmd.arguments().empty())
  5144. {
  5145. QString argument = cmd.arguments().front();
  5146. if (ccCommandLineInterface::IsCommand(argument, COMMAND_DELAUNAY_AA))
  5147. {
  5148. //local option confirmed, we can move on
  5149. cmd.arguments().pop_front();
  5150. axisAligned = true;
  5151. }
  5152. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_DELAUNAY_BF))
  5153. {
  5154. //local option confirmed, we can move on
  5155. cmd.arguments().pop_front();
  5156. axisAligned = false;
  5157. }
  5158. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_DELAUNAY_MAX_EDGE_LENGTH))
  5159. {
  5160. //local option confirmed, we can move on
  5161. cmd.arguments().pop_front();
  5162. if (cmd.arguments().empty())
  5163. {
  5164. return cmd.error(QObject::tr("Missing parameter: max edge length value after '%1'").arg(COMMAND_DELAUNAY_MAX_EDGE_LENGTH));
  5165. }
  5166. bool ok;
  5167. maxEdgeLength = cmd.arguments().takeFirst().toDouble(&ok);
  5168. if (!ok)
  5169. {
  5170. return cmd.error(QObject::tr("Invalid value for max edge length (%1)! (after %2)").arg(maxEdgeLength).arg(COMMAND_DELAUNAY_MAX_EDGE_LENGTH));
  5171. }
  5172. }
  5173. else
  5174. {
  5175. break;
  5176. }
  5177. }
  5178. cmd.print(QObject::tr("Axis aligned: %1").arg(axisAligned ? "yes" : "no"));
  5179. //try to triangulate each cloud
  5180. for (CLCloudDesc& desc : cmd.clouds())
  5181. {
  5182. cmd.print(QObject::tr("\tProcessing cloud %1").arg(!desc.pc->getName().isEmpty() ? desc.pc->getName() : "no name"));
  5183. ccMesh* mesh = ccMesh::Triangulate(desc.pc,
  5184. axisAligned ? CCCoreLib::DELAUNAY_2D_AXIS_ALIGNED : CCCoreLib::DELAUNAY_2D_BEST_LS_PLANE,
  5185. false,
  5186. static_cast<PointCoordinateType>(maxEdgeLength),
  5187. 2 //XY plane by default
  5188. );
  5189. if (mesh)
  5190. {
  5191. cmd.print(QObject::tr("\tResulting mesh: #%1 faces, %2 vertices").arg(mesh->size()).arg(mesh->getAssociatedCloud()->size()));
  5192. CLMeshDesc meshDesc;
  5193. {
  5194. meshDesc.mesh = mesh;
  5195. meshDesc.basename = desc.basename;
  5196. meshDesc.path = desc.path;
  5197. meshDesc.indexInFile = desc.indexInFile;
  5198. }
  5199. //save mesh
  5200. if (cmd.autoSaveMode())
  5201. {
  5202. QString outputFilename;
  5203. QString errorStr = cmd.exportEntity(meshDesc, "DELAUNAY", &outputFilename);
  5204. if (!errorStr.isEmpty())
  5205. {
  5206. cmd.warning(errorStr);
  5207. }
  5208. }
  5209. //add the resulting mesh to the main set
  5210. cmd.meshes().push_back(meshDesc);
  5211. //the mesh takes ownership of the cloud.
  5212. //Therefore we have to remove all clouds from the 'cloud set'! (see below)
  5213. //(otherwise bad things will happen when we'll clear it later ;)
  5214. desc.pc->setEnabled(false);
  5215. mesh->addChild(desc.pc);
  5216. }
  5217. }
  5218. //mehses have taken ownership of the clouds!
  5219. cmd.clouds().resize(0);
  5220. return true;
  5221. }
  5222. CommandSFArithmetic::CommandSFArithmetic()
  5223. : ccCommandLineInterface::Command(QObject::tr("SF arithmetic"), COMMAND_SF_ARITHMETIC)
  5224. {}
  5225. bool CommandSFArithmetic::process(ccCommandLineInterface& cmd)
  5226. {
  5227. if (cmd.arguments().size() < 2)
  5228. {
  5229. return cmd.error(QObject::tr("Missing parameter(s): SF index and/or operation after '%1' (2 values expected)").arg(COMMAND_SF_ARITHMETIC));
  5230. }
  5231. //read SF index
  5232. int sfIndex = -1;
  5233. QString sfName;
  5234. if (!GetSFIndexOrName(cmd, sfIndex, sfName, true))
  5235. {
  5236. return false;
  5237. }
  5238. //read operation type
  5239. ccScalarFieldArithmeticsDlg::Operation operation = ccScalarFieldArithmeticsDlg::INVALID;
  5240. {
  5241. QString opName = cmd.arguments().takeFirst();
  5242. operation = ccScalarFieldArithmeticsDlg::GetOperationByName(opName);
  5243. if (operation == ccScalarFieldArithmeticsDlg::INVALID)
  5244. {
  5245. return cmd.error(QObject::tr("Unknown operation! (%1)").arg(opName));
  5246. }
  5247. else if (operation <= ccScalarFieldArithmeticsDlg::MAX || operation == ccScalarFieldArithmeticsDlg::SET)
  5248. {
  5249. return cmd.error(QObject::tr("Operation %1 can't be applied with %2. Consider using the %3 command").arg(opName, COMMAND_SF_ARITHMETIC, COMMAND_SF_OP));
  5250. }
  5251. }
  5252. bool inPlace = false;
  5253. //read the optional arguments
  5254. while (!cmd.arguments().empty())
  5255. {
  5256. QString argument = cmd.arguments().front();
  5257. if (ccCommandLineInterface::IsCommand(argument, COMMAND_SF_ARITHMETIC_IN_PLACE))
  5258. {
  5259. //local option confirmed, we can move on
  5260. cmd.arguments().pop_front();
  5261. inPlace = true;
  5262. }
  5263. else
  5264. {
  5265. break; //as soon as we encounter an unrecognized argument, we break the local loop to go back to the main one!
  5266. }
  5267. }
  5268. //apply operation on clouds
  5269. for (CLCloudDesc& desc : cmd.clouds())
  5270. {
  5271. if (desc.pc)
  5272. {
  5273. int thisSFIndex = GetScalarFieldIndex(desc.pc, sfIndex, sfName, true);
  5274. if (thisSFIndex >= 0)
  5275. {
  5276. if (!ccScalarFieldArithmeticsDlg::Apply(desc.pc, operation, thisSFIndex, inPlace))
  5277. {
  5278. return cmd.error(QObject::tr("Failed to apply operation on cloud '%1'").arg(desc.pc->getName()));
  5279. }
  5280. else if (cmd.autoSaveMode())
  5281. {
  5282. QString errorStr = cmd.exportEntity(desc, "SF_ARITHMETIC");
  5283. if (!errorStr.isEmpty())
  5284. {
  5285. return cmd.error(errorStr);
  5286. }
  5287. }
  5288. }
  5289. }
  5290. }
  5291. //and meshes!
  5292. for (size_t j = 0; j < cmd.meshes().size(); ++j)
  5293. {
  5294. bool isLocked = false;
  5295. ccGenericMesh* mesh = cmd.meshes()[j].mesh;
  5296. ccPointCloud* cloud = ccHObjectCaster::ToPointCloud(mesh, &isLocked);
  5297. if (cloud && !isLocked)
  5298. {
  5299. int thisSFIndex = GetScalarFieldIndex(cloud, sfIndex, sfName, true);
  5300. if (thisSFIndex >= 0)
  5301. {
  5302. if (!ccScalarFieldArithmeticsDlg::Apply(cloud, operation, thisSFIndex, inPlace))
  5303. {
  5304. return cmd.error(QObject::tr("Failed to apply operation on mesh '%1'").arg(mesh->getName()));
  5305. }
  5306. else if (cmd.autoSaveMode())
  5307. {
  5308. QString errorStr = cmd.exportEntity(cmd.meshes()[j], "SF_ARITHMETIC");
  5309. if (!errorStr.isEmpty())
  5310. {
  5311. return cmd.error(errorStr);
  5312. }
  5313. }
  5314. }
  5315. }
  5316. }
  5317. return true;
  5318. }
  5319. CommandSFOperation::CommandSFOperation()
  5320. : ccCommandLineInterface::Command(QObject::tr("SF operation"), COMMAND_SF_OP)
  5321. {}
  5322. bool CommandSFOperation::process(ccCommandLineInterface& cmd)
  5323. {
  5324. //in place modifier, to keep old commands intact we should keep it in place by default. However it makes the command line inconsistent, because the SF_ARITHMETIC works the other way.
  5325. bool inPlace = true;
  5326. if (!cmd.arguments().empty())
  5327. {
  5328. if (cmd.IsCommand(cmd.arguments().front(), COMMAND_SF_OP_NOT_IN_PLACE))
  5329. {
  5330. //local arg detected
  5331. inPlace = false;
  5332. cmd.arguments().pop_front();
  5333. }
  5334. }
  5335. if (cmd.arguments().size() < 3)
  5336. {
  5337. return cmd.error(QObject::tr("Missing parameter(s): SF index and/or operation and/or scalar value after '%1' (3 values expected)").arg(COMMAND_SF_OP));
  5338. }
  5339. //read SF index
  5340. int sfIndex = -1;
  5341. QString sfName;
  5342. if (!GetSFIndexOrName(cmd, sfIndex, sfName, true))
  5343. {
  5344. return false;
  5345. }
  5346. //read operation type
  5347. ccScalarFieldArithmeticsDlg::Operation operation = ccScalarFieldArithmeticsDlg::INVALID;
  5348. {
  5349. QString opName = cmd.arguments().takeFirst();
  5350. operation = ccScalarFieldArithmeticsDlg::GetOperationByName(opName);
  5351. if (operation == ccScalarFieldArithmeticsDlg::INVALID)
  5352. {
  5353. return cmd.error(QObject::tr("Unknown operation! (%1)").arg(opName));
  5354. }
  5355. else if (operation > ccScalarFieldArithmeticsDlg::MAX && operation != ccScalarFieldArithmeticsDlg::SET)
  5356. {
  5357. return cmd.error(QObject::tr("Operation %1 can't be applied with %2. Consider using the %3 command").arg(opName, COMMAND_SF_OP, COMMAND_SF_ARITHMETIC));
  5358. }
  5359. }
  5360. //read scalar value
  5361. ScalarType value = static_cast<ScalarType>(1.0);
  5362. USE_SPECIAL_SF_VALUE specialValue = USE_SPECIAL_SF_VALUE::USE_NONE;
  5363. {
  5364. QString valueStr = cmd.arguments().takeFirst();
  5365. specialValue = ToSpecialSFValue(valueStr);
  5366. if (specialValue == USE_NONE)
  5367. {
  5368. bool ok = false;
  5369. value = valueStr.toDouble(&ok);
  5370. if (!ok)
  5371. {
  5372. return cmd.error(QObject::tr("Invalid scalar value! (after %1)").arg(COMMAND_SF_OP));
  5373. }
  5374. }
  5375. }
  5376. ccScalarFieldArithmeticsDlg::SF2 sf2;
  5377. {
  5378. sf2.isConstantValue = true;
  5379. sf2.constantValue = value;
  5380. }
  5381. //apply operation on clouds
  5382. for (CLCloudDesc& desc : cmd.clouds())
  5383. {
  5384. if (desc.pc)
  5385. {
  5386. int thisSFIndex = GetScalarFieldIndex(desc.pc, sfIndex, sfName, true);
  5387. if (thisSFIndex >= 0)
  5388. {
  5389. sf2.constantValue = GetSFValue(*desc.pc, thisSFIndex, value, specialValue);
  5390. if (!ccScalarFieldArithmeticsDlg::Apply(desc.pc, operation, thisSFIndex, inPlace, &sf2))
  5391. {
  5392. return cmd.error(QObject::tr("Failed to apply operation on cloud '%1'").arg(desc.pc->getName()));
  5393. }
  5394. else if (cmd.autoSaveMode())
  5395. {
  5396. QString errorStr = cmd.exportEntity(desc, "SF_OP");
  5397. if (!errorStr.isEmpty())
  5398. {
  5399. return cmd.error(errorStr);
  5400. }
  5401. }
  5402. }
  5403. }
  5404. }
  5405. //and meshes!
  5406. for (size_t j = 0; j < cmd.meshes().size(); ++j)
  5407. {
  5408. bool isLocked = false;
  5409. ccGenericMesh* mesh = cmd.meshes()[j].mesh;
  5410. ccPointCloud* cloud = ccHObjectCaster::ToPointCloud(mesh, &isLocked);
  5411. if (cloud && !isLocked)
  5412. {
  5413. int thisSFIndex = GetScalarFieldIndex(cloud, sfIndex, sfName, true);
  5414. if (thisSFIndex >= 0)
  5415. {
  5416. sf2.constantValue = GetSFValue(*cloud, thisSFIndex, value, specialValue);
  5417. if (!ccScalarFieldArithmeticsDlg::Apply(cloud, operation, thisSFIndex, inPlace, &sf2))
  5418. {
  5419. return cmd.error(QObject::tr("Failed to apply operation on mesh '%1'").arg(mesh->getName()));
  5420. }
  5421. else if (cmd.autoSaveMode())
  5422. {
  5423. QString errorStr = cmd.exportEntity(cmd.meshes()[j], "SF_OP");
  5424. if (!errorStr.isEmpty())
  5425. {
  5426. return cmd.error(errorStr);
  5427. }
  5428. }
  5429. }
  5430. }
  5431. }
  5432. return true;
  5433. }
  5434. CommandSFOperationSF::CommandSFOperationSF()
  5435. : ccCommandLineInterface::Command(QObject::tr("SF (add, sub, mult, div) SF"), COMMAND_SF_OP_SF)
  5436. {}
  5437. bool CommandSFOperationSF::process(ccCommandLineInterface& cmd)
  5438. {
  5439. if (cmd.arguments().size() < 3)
  5440. {
  5441. return cmd.error(QObject::tr("Missing parameter(s): SF index and operation and SF index '%1' (3 values expected)").arg(COMMAND_SF_OP_SF));
  5442. }
  5443. //read SF index 1
  5444. int sfIndex = -1;
  5445. QString sfName;
  5446. if (!GetSFIndexOrName(cmd, sfIndex, sfName))
  5447. {
  5448. return false;
  5449. }
  5450. //read operation type
  5451. ccScalarFieldArithmeticsDlg::Operation operation = ccScalarFieldArithmeticsDlg::INVALID;
  5452. {
  5453. QString opName = cmd.arguments().takeFirst();
  5454. operation = ccScalarFieldArithmeticsDlg::GetOperationByName(opName);
  5455. if (operation == ccScalarFieldArithmeticsDlg::INVALID)
  5456. {
  5457. return cmd.error(QObject::tr("Unknown operation! (%1)").arg(opName));
  5458. }
  5459. else if (operation > ccScalarFieldArithmeticsDlg::MAX)
  5460. {
  5461. return cmd.error(QObject::tr("Operation %1 can't be applied with %2").arg(opName, COMMAND_SF_OP_SF));
  5462. }
  5463. }
  5464. //read SF index 2
  5465. int sfIndex2 = -1;
  5466. QString sfName2;
  5467. if (!GetSFIndexOrName(cmd, sfIndex2, sfName2))
  5468. {
  5469. return false;
  5470. }
  5471. //apply operation on clouds
  5472. for (CLCloudDesc& desc : cmd.clouds())
  5473. {
  5474. if (desc.pc)
  5475. {
  5476. int thisSFFIndex = GetScalarFieldIndex(desc.pc, sfIndex, sfName);
  5477. int thisSFFIndex2 = GetScalarFieldIndex(desc.pc, sfIndex2, sfName2);
  5478. if (thisSFFIndex >= 0 && thisSFFIndex2 >= 0)
  5479. {
  5480. ccScalarFieldArithmeticsDlg::SF2 sf2;
  5481. {
  5482. sf2.isConstantValue = false;
  5483. sf2.sfIndex = thisSFFIndex2;
  5484. }
  5485. if (!ccScalarFieldArithmeticsDlg::Apply(desc.pc, operation, thisSFFIndex, true, &sf2))
  5486. {
  5487. return cmd.error(QObject::tr("Failed top apply operation on cloud '%1'").arg(desc.pc->getName()));
  5488. }
  5489. else if (cmd.autoSaveMode())
  5490. {
  5491. QString errorStr = cmd.exportEntity(desc, "SF_OP_SF");
  5492. if (!errorStr.isEmpty())
  5493. {
  5494. return cmd.error(errorStr);
  5495. }
  5496. }
  5497. }
  5498. }
  5499. }
  5500. //and meshes!
  5501. for (size_t j = 0; j < cmd.meshes().size(); ++j)
  5502. {
  5503. bool isLocked = false;
  5504. ccGenericMesh* mesh = cmd.meshes()[j].mesh;
  5505. ccPointCloud* cloud = ccHObjectCaster::ToPointCloud(mesh, &isLocked);
  5506. if (cloud && !isLocked)
  5507. {
  5508. int thisSFFIndex = GetScalarFieldIndex(cloud, sfIndex, sfName);
  5509. int thisSFFIndex2 = GetScalarFieldIndex(cloud, sfIndex2, sfName2);
  5510. if (thisSFFIndex >= 0 && thisSFFIndex2 >= 0)
  5511. {
  5512. ccScalarFieldArithmeticsDlg::SF2 sf2;
  5513. {
  5514. sf2.isConstantValue = false;
  5515. sf2.sfIndex = thisSFFIndex2;
  5516. }
  5517. if (!ccScalarFieldArithmeticsDlg::Apply(cloud, operation, thisSFFIndex, true, &sf2))
  5518. {
  5519. return cmd.error(QObject::tr("Failed top apply operation on mesh '%1'").arg(mesh->getName()));
  5520. }
  5521. else if (cmd.autoSaveMode())
  5522. {
  5523. QString errorStr = cmd.exportEntity(cmd.meshes()[j], "SF_OP_SF");
  5524. if (!errorStr.isEmpty())
  5525. {
  5526. return cmd.error(errorStr);
  5527. }
  5528. }
  5529. }
  5530. }
  5531. }
  5532. return true;
  5533. }
  5534. CommandSFInterpolation::CommandSFInterpolation()
  5535. : ccCommandLineInterface::Command(QObject::tr("SF interpolation"), COMMAND_SF_INTERP)
  5536. {}
  5537. bool CommandSFInterpolation::process(ccCommandLineInterface& cmd)
  5538. {
  5539. if (cmd.arguments().size() < 1)
  5540. return cmd.error(QObject::tr("Missing parameter(s): SF index after '%1' (1 value expected)").arg(COMMAND_SF_INTERP));
  5541. if (cmd.clouds().size() < 2)
  5542. return cmd.error(QObject::tr("Unexpected number of clouds for '%1' (at least 2 clouds expected: first = source, second = dest)").arg(COMMAND_SF_INTERP));
  5543. //read sf index or name
  5544. int sfIndex = -1;
  5545. QString sfName;
  5546. if (!GetSFIndexOrName(cmd, sfIndex, sfName, true))
  5547. {
  5548. return false;
  5549. }
  5550. bool destIsFirst = false;
  5551. while (!cmd.arguments().empty())
  5552. {
  5553. QString argument = cmd.arguments().front();
  5554. if (ccCommandLineInterface::IsCommand(argument, COMMAND_SF_INTERP_DEST_IS_FIRST))
  5555. {
  5556. cmd.print(QObject::tr("[DEST_IS_FIRST]"));
  5557. //local option confirmed, we can move on
  5558. cmd.arguments().pop_front();
  5559. destIsFirst = true;
  5560. }
  5561. else
  5562. {
  5563. break; //as soon as we encounter an unrecognized argument, we break the local loop to go back to the main one!
  5564. }
  5565. }
  5566. ccPointCloud* source = cmd.clouds()[0].pc;
  5567. ccPointCloud* dest = cmd.clouds()[1].pc;
  5568. if (destIsFirst) // swap source and destination
  5569. {
  5570. source = cmd.clouds()[1].pc;
  5571. dest = cmd.clouds()[0].pc;
  5572. }
  5573. sfIndex = GetScalarFieldIndex(source, sfIndex, sfName, true);
  5574. if (sfIndex < 0)
  5575. {
  5576. return false;
  5577. }
  5578. cmd.print("SF to interpolate: index " + QString::number(sfIndex) + ", name " + QString::fromStdString(source->getScalarField(sfIndex)->getName()));
  5579. //semi-persistent parameters
  5580. ccPointCloudInterpolator::Parameters params;
  5581. {
  5582. params.method = ccPointCloudInterpolator::Parameters::NEAREST_NEIGHBOR; // nearest neighbor
  5583. params.algo = ccPointCloudInterpolator::Parameters::NORMAL_DIST; // normal distribution
  5584. params.knn = 6;
  5585. params.radius = static_cast<float>(dest->getOwnBB().getDiagNormd() / 100);
  5586. params.sigma = params.radius / 2.5; // see ccInterpolationDlg::onRadiusUpdated
  5587. }
  5588. return ccEntityAction::interpolateSFs(source, dest, sfIndex, params, cmd.widgetParent());
  5589. }
  5590. CommandColorInterpolation::CommandColorInterpolation()
  5591. : ccCommandLineInterface::Command(QObject::tr("Color interpolation"), COMMAND_COLOR_INTERP)
  5592. {}
  5593. bool CommandColorInterpolation::process(ccCommandLineInterface& cmd)
  5594. {
  5595. if (cmd.clouds().size() < 2)
  5596. return cmd.error(QObject::tr("Unexpected number of clouds for '%1' (at least 2 clouds expected: first = source, second = dest)").arg(COMMAND_COLOR_INTERP));
  5597. ccHObject::Container entities;
  5598. entities.push_back(cmd.clouds()[0].pc);
  5599. entities.push_back(cmd.clouds()[1].pc);
  5600. return ccEntityAction::interpolateColors(entities, cmd.widgetParent());
  5601. }
  5602. CommandFilter::CommandFilter()
  5603. : ccCommandLineInterface::Command(QObject::tr("FILTER"), COMMAND_FILTER)
  5604. {}
  5605. bool CommandFilter::process(ccCommandLineInterface& cmd)
  5606. {
  5607. bool applyToRGB = false;
  5608. bool applyToSF = false;
  5609. bool gaussian = false;
  5610. ccPointCloud::RgbFilterOptions(filterParams);
  5611. filterParams.commandLine = true;
  5612. while (!cmd.arguments().empty())
  5613. {
  5614. QString argument = cmd.arguments().front();
  5615. if (ccCommandLineInterface::IsCommand(argument, OPTION_SF))
  5616. {
  5617. cmd.arguments().pop_front();
  5618. applyToSF = true;
  5619. }
  5620. else if (ccCommandLineInterface::IsCommand(argument, OPTION_RGB))
  5621. {
  5622. cmd.arguments().pop_front();
  5623. applyToRGB = true;
  5624. }
  5625. else if (ccCommandLineInterface::IsCommand(argument, OPTION_BILATERAL))
  5626. {
  5627. cmd.arguments().pop_front();
  5628. if (filterParams.filterType == ccPointCloud::RGB_FILTER_TYPES::NONE)
  5629. {
  5630. filterParams.filterType = ccPointCloud::RGB_FILTER_TYPES::BILATERAL;
  5631. }
  5632. }
  5633. else if (ccCommandLineInterface::IsCommand(argument, OPTION_GAUSSIAN))
  5634. {
  5635. cmd.arguments().pop_front();
  5636. if (filterParams.filterType == ccPointCloud::RGB_FILTER_TYPES::NONE)
  5637. {
  5638. filterParams.filterType = ccPointCloud::RGB_FILTER_TYPES::GAUSSIAN;
  5639. }
  5640. }
  5641. else if (ccCommandLineInterface::IsCommand(argument, OPTION_MEAN))
  5642. {
  5643. cmd.arguments().pop_front();
  5644. if (filterParams.filterType == ccPointCloud::RGB_FILTER_TYPES::NONE)
  5645. {
  5646. filterParams.filterType = ccPointCloud::RGB_FILTER_TYPES::MEAN;
  5647. }
  5648. }
  5649. else if (ccCommandLineInterface::IsCommand(argument, OPTION_MEDIAN))
  5650. {
  5651. cmd.arguments().pop_front();
  5652. if (filterParams.filterType == ccPointCloud::RGB_FILTER_TYPES::NONE)
  5653. {
  5654. filterParams.filterType = ccPointCloud::RGB_FILTER_TYPES::MEDIAN;
  5655. }
  5656. }
  5657. else if (ccCommandLineInterface::IsCommand(argument, OPTION_SIGMA))
  5658. {
  5659. cmd.arguments().pop_front();
  5660. if (cmd.arguments().empty())
  5661. {
  5662. return cmd.error(QObject::tr("Missing parameter: spatial sigma after '-%1'").arg(OPTION_SIGMA));
  5663. }
  5664. bool ok = false;
  5665. filterParams.spatialSigma = cmd.arguments().takeFirst().toDouble(&ok);
  5666. if (!ok)
  5667. {
  5668. return cmd.error(QObject::tr("Invalid value for spatial sigma after '%1'!").arg(OPTION_SIGMA));
  5669. }
  5670. }
  5671. else if (ccCommandLineInterface::IsCommand(argument, OPTION_SIGMA_SF))
  5672. {
  5673. cmd.arguments().pop_front();
  5674. if (cmd.arguments().empty())
  5675. {
  5676. return cmd.error(QObject::tr("Missing parameter: spatial sigma after '-%1'").arg(OPTION_SIGMA_SF));
  5677. }
  5678. bool ok = false;
  5679. filterParams.sigmaSF = cmd.arguments().takeFirst().toDouble(&ok);
  5680. if (!ok)
  5681. {
  5682. return cmd.error(QObject::tr("Invalid value for spatial sigma after '%1'!").arg(OPTION_SIGMA_SF));
  5683. }
  5684. }
  5685. else if (ccCommandLineInterface::IsCommand(argument, OPTION_BURNT_COLOR_THRESHOLD))
  5686. {
  5687. cmd.arguments().pop_front();
  5688. if (cmd.arguments().empty())
  5689. {
  5690. return cmd.error(QObject::tr("Missing parameter: burnt color threshold after '-%1'").arg(OPTION_BURNT_COLOR_THRESHOLD));
  5691. }
  5692. bool ok = false;
  5693. uint burntOutColorThreshold = cmd.arguments().takeFirst().toUInt(&ok);
  5694. if (!ok || burntOutColorThreshold > 255)
  5695. {
  5696. return cmd.error(QObject::tr("Invalid value for burnt color threshold after '%1', must be an integer between 0 and 255!").arg(OPTION_BURNT_COLOR_THRESHOLD));
  5697. }
  5698. filterParams.burntOutColorThreshold = static_cast<unsigned char>(burntOutColorThreshold);
  5699. }
  5700. else if (ccCommandLineInterface::IsCommand(argument, OPTION_BLEND_GRAYSCALE))
  5701. {
  5702. cmd.arguments().pop_front();
  5703. if (cmd.arguments().size() < 2)
  5704. {
  5705. return cmd.error(QObject::tr("Missing parameter: blend grayscale threshold and grayscale percent after '-%1'").arg(OPTION_BLEND_GRAYSCALE));
  5706. }
  5707. bool ok = false;
  5708. uint blendGrayscale = cmd.arguments().takeFirst().toUInt(&ok);
  5709. if (!ok || blendGrayscale > 255)
  5710. {
  5711. return cmd.error(QObject::tr("Invalid value for blend grayscale threshold after '%1', must be an integer between 0 and 255!").arg(OPTION_BLEND_GRAYSCALE));
  5712. }
  5713. filterParams.blendGrayscale = true;
  5714. filterParams.blendGrayscaleThreshold = static_cast<unsigned char>(blendGrayscale);
  5715. uint grayscalePercent = cmd.arguments().takeFirst().toUInt(&ok);
  5716. if (!ok || grayscalePercent > 100)
  5717. {
  5718. return cmd.error(QObject::tr("Invalid value for grayscale percent after '%1 %2', must be an integer between 0 and 100!").arg(OPTION_BLEND_GRAYSCALE).arg(filterParams.blendGrayscaleThreshold));
  5719. }
  5720. filterParams.blendGrayscalePercent = static_cast<double>(grayscalePercent) / 100;
  5721. }
  5722. else
  5723. {
  5724. break;
  5725. }
  5726. }
  5727. if (!applyToRGB && !applyToSF)
  5728. {
  5729. return cmd.error(QObject::tr("Missing parameter -%1 and/or -%2 need to be set.").arg(OPTION_RGB).arg(OPTION_SF));
  5730. }
  5731. if (filterParams.filterType == ccPointCloud::RGB_FILTER_TYPES::NONE)
  5732. {
  5733. return cmd.error(QObject::tr("Missing parameter any of '-%1', '-%2', '-%3', '-%4' need to be set.")
  5734. .arg(OPTION_MEAN)
  5735. .arg(OPTION_GAUSSIAN)
  5736. .arg(OPTION_BILATERAL)
  5737. .arg(OPTION_MEDIAN));
  5738. }
  5739. //apply operation on clouds
  5740. ccHObject::Container selectedEntities;
  5741. for (CLCloudDesc& thisCloudDesc : cmd.clouds())
  5742. {
  5743. selectedEntities.push_back(thisCloudDesc.pc);
  5744. }
  5745. if (applyToSF && applyToRGB)
  5746. {
  5747. applyToSF = false;
  5748. filterParams.applyToSFduringRGB = true;
  5749. }
  5750. if (applyToSF)
  5751. {
  5752. return ccEntityAction::sfGaussianFilter(selectedEntities, filterParams, cmd.widgetParent());
  5753. }
  5754. else if (applyToRGB)
  5755. {
  5756. return ccEntityAction::rgbGaussianFilter(selectedEntities, filterParams, cmd.widgetParent());
  5757. }
  5758. return true;
  5759. }
  5760. CommandRenameEntities::CommandRenameEntities()
  5761. : ccCommandLineInterface::Command(QObject::tr("Rename entities"), COMMAND_RENAME_ENTITIES)
  5762. {}
  5763. bool CommandRenameEntities::process(ccCommandLineInterface& cmd)
  5764. {
  5765. if (cmd.arguments().empty())
  5766. {
  5767. return cmd.error(QObject::tr("Missing parameter: Name after \"-%1\"").arg(COMMAND_RENAME_ENTITIES));
  5768. }
  5769. QString newBaseName = cmd.arguments().takeFirst();
  5770. //Validate if the given name contains any breaking characters for NTFS filesystem at least
  5771. QRegExp rx("[^:/\\\\*?\"|<>]*");
  5772. QRegExpValidator v(rx, 0);
  5773. int pos = 0;
  5774. if (!v.validate(newBaseName, pos))
  5775. {
  5776. assert(false);
  5777. return cmd.error("Name cannot contain any of these characters: :/\\*?\"|<>");
  5778. }
  5779. //apply operation on clouds
  5780. int index = 1;
  5781. size_t nrOfEntities = cmd.clouds().size() + cmd.meshes().size();
  5782. for (CLCloudDesc& desc : cmd.clouds())
  5783. {
  5784. if (desc.pc)
  5785. {
  5786. QString currentCloudName = newBaseName;
  5787. if (nrOfEntities > 1)
  5788. {
  5789. currentCloudName += "-" + QString::number(index).rightJustified(3, '0');
  5790. }
  5791. cmd.print("Cloud '" + desc.pc->getName() + "' renamed to '" + currentCloudName + "'");
  5792. //rename the Point Cloud entity
  5793. desc.pc->setName(currentCloudName);
  5794. //rename the Root entity
  5795. desc.basename = currentCloudName;
  5796. ++index;
  5797. }
  5798. }
  5799. //apply operation on meshes
  5800. for (CLMeshDesc& desc : cmd.meshes())
  5801. {
  5802. if (desc.mesh)
  5803. {
  5804. QString currentMeshName = newBaseName;
  5805. if (nrOfEntities > 1)
  5806. {
  5807. currentMeshName += "-" + QString::number(index).rightJustified(3, '0');
  5808. }
  5809. cmd.print("Mesh '" + desc.mesh->getName() + "' renamed to '" + currentMeshName + "'");
  5810. //rename the Mesh entity
  5811. desc.mesh->setName(currentMeshName);
  5812. //rename the Root entity
  5813. desc.basename = currentMeshName;
  5814. ++index;
  5815. }
  5816. }
  5817. return true;
  5818. }
  5819. CommandSFRename::CommandSFRename()
  5820. : ccCommandLineInterface::Command(QObject::tr("Rename SF"), COMMAND_RENAME_SF)
  5821. {}
  5822. bool CommandSFRename::process(ccCommandLineInterface& cmd)
  5823. {
  5824. if (cmd.arguments().size() < 2)
  5825. {
  5826. return cmd.error(QObject::tr("Missing parameter(s): SF index and/or scalar field name after '%1' (2 values expected)").arg(COMMAND_RENAME_SF));
  5827. }
  5828. //read SF index
  5829. int sfIndex = -1;
  5830. QString sfName;
  5831. if (!GetSFIndexOrName(cmd, sfIndex, sfName, true))
  5832. {
  5833. return false;
  5834. }
  5835. //read the SF name
  5836. QString newSFName = cmd.arguments().takeFirst();
  5837. //apply operation on clouds
  5838. for (CLCloudDesc& desc : cmd.clouds())
  5839. {
  5840. if (desc.pc)
  5841. {
  5842. int thisSFIndex = GetScalarFieldIndex(desc.pc, sfIndex, sfName, true);
  5843. if (thisSFIndex >= 0)
  5844. {
  5845. int indexOfSFWithSameName = desc.pc->getScalarFieldIndexByName(newSFName.toStdString());
  5846. if (indexOfSFWithSameName >= 0 && thisSFIndex != indexOfSFWithSameName)
  5847. {
  5848. return cmd.error("A SF with the same name is already defined on cloud " + desc.pc->getName());
  5849. }
  5850. CCCoreLib::ScalarField* sf = desc.pc->getScalarField(thisSFIndex);
  5851. if (!sf)
  5852. {
  5853. assert(false);
  5854. return cmd.error("Internal error: invalid SF index");
  5855. }
  5856. sf->setName(newSFName.toStdString());
  5857. if (cmd.autoSaveMode())
  5858. {
  5859. QString errorStr = cmd.exportEntity(desc, "SF_RENAMED");
  5860. if (!errorStr.isEmpty())
  5861. {
  5862. return cmd.error(errorStr);
  5863. }
  5864. }
  5865. }
  5866. }
  5867. }
  5868. //and meshes!
  5869. for (CLMeshDesc& desc : cmd.meshes())
  5870. {
  5871. bool isLocked = false;
  5872. ccPointCloud* cloud = ccHObjectCaster::ToPointCloud(desc.mesh, &isLocked);
  5873. if (cloud && !isLocked)
  5874. {
  5875. int thisSFIndex = GetScalarFieldIndex(cloud, sfIndex, sfName, true);
  5876. if (thisSFIndex >= 0)
  5877. {
  5878. int indexOfSFWithSameName = cloud->getScalarFieldIndexByName(newSFName.toStdString());
  5879. if (indexOfSFWithSameName >= 0 && thisSFIndex != indexOfSFWithSameName)
  5880. {
  5881. return cmd.error("A SF with the same name is already defined on cloud " + cloud->getName());
  5882. }
  5883. CCCoreLib::ScalarField* sf = cloud->getScalarField(thisSFIndex);
  5884. if (!sf)
  5885. {
  5886. assert(false);
  5887. return cmd.error("Internal error: invalid SF index");
  5888. }
  5889. sf->setName(newSFName.toStdString());
  5890. if (cmd.autoSaveMode())
  5891. {
  5892. QString errorStr = cmd.exportEntity(desc, "SF_RENAMED");
  5893. if (!errorStr.isEmpty())
  5894. {
  5895. return cmd.error(errorStr);
  5896. }
  5897. }
  5898. }
  5899. }
  5900. }
  5901. return true;
  5902. }
  5903. CommandSFAddConst::CommandSFAddConst()
  5904. : ccCommandLineInterface::Command(QObject::tr("Add constant SF"), COMMAND_SF_ADD_CONST)
  5905. {}
  5906. bool CommandSFAddConst::process(ccCommandLineInterface& cmd)
  5907. {
  5908. cmd.print(QObject::tr("Note: this operation is only done on clouds"));
  5909. if (cmd.arguments().size() < 2)
  5910. {
  5911. return cmd.error(QObject::tr("Missing parameter(s): SF name and value after '%1' (2 values expected)").arg(COMMAND_SF_ADD_CONST));
  5912. }
  5913. //read the SF name
  5914. QString sfName = cmd.arguments().takeFirst();
  5915. //read constant value
  5916. bool ok = true;
  5917. ScalarType value = static_cast<ScalarType>(cmd.arguments().takeFirst().toDouble(&ok));
  5918. if (!ok)
  5919. {
  5920. return cmd.error(QObject::tr("Invalid constant value! (after %1)").arg(COMMAND_SF_ADD_CONST));
  5921. }
  5922. //apply operation on clouds
  5923. for (CLCloudDesc& desc : cmd.clouds())
  5924. {
  5925. if (desc.pc)
  5926. {
  5927. // check that there is no existing scalar field with the same name
  5928. int indexOfSFWithSameName = desc.pc->getScalarFieldIndexByName(sfName.toStdString());
  5929. if (indexOfSFWithSameName >= 0)
  5930. return cmd.error("A SF with the same name is already defined on cloud " + desc.pc->getName());
  5931. // add the new scalar field
  5932. int sfIndex = desc.pc->addScalarField(sfName.toStdString());
  5933. if (sfIndex == -1)
  5934. {
  5935. return cmd.error("Internal error: addScalarField failed");
  5936. }
  5937. CCCoreLib::ScalarField* sf = desc.pc->getScalarField(sfIndex);
  5938. assert(sf);
  5939. for (unsigned index = 0; index < desc.pc->size(); index++)
  5940. {
  5941. sf->setValue(index, value);
  5942. }
  5943. if (cmd.autoSaveMode())
  5944. {
  5945. QString errorStr = cmd.exportEntity(desc, "SF_ADDED");
  5946. if (!errorStr.isEmpty())
  5947. {
  5948. return cmd.error(errorStr);
  5949. }
  5950. }
  5951. }
  5952. }
  5953. return true;
  5954. }
  5955. CommandSFAddId::CommandSFAddId()
  5956. : ccCommandLineInterface::Command(QObject::tr("Add indexes as SF"), COMMAND_SF_ADD_ID)
  5957. {}
  5958. bool CommandSFAddId::process(ccCommandLineInterface& cmd)
  5959. {
  5960. ccHObject::Container selectedEntities;
  5961. bool addIdAsInt = false;
  5962. while (!cmd.arguments().empty())
  5963. {
  5964. QString argument = cmd.arguments().front();
  5965. if (ccCommandLineInterface::IsCommand(argument, COMMAND_SF_ADD_ID_AS_INT))
  5966. {
  5967. cmd.print(QObject::tr("[AS_INT]"));
  5968. //local option confirmed, we can move on
  5969. cmd.arguments().pop_front();
  5970. addIdAsInt = true;
  5971. }
  5972. else
  5973. {
  5974. break; //as soon as we encounter an unrecognized argument, we break the local loop to go back to the main one!
  5975. }
  5976. }
  5977. for (CLCloudDesc& desc : cmd.clouds())
  5978. {
  5979. selectedEntities.push_back(desc.getEntity());
  5980. }
  5981. return ccEntityAction::sfAddIdField(selectedEntities, addIdAsInt);
  5982. }
  5983. CommandICP::CommandICP()
  5984. : ccCommandLineInterface::Command("ICP", COMMAND_ICP)
  5985. {}
  5986. bool CommandICP::process(ccCommandLineInterface& cmd)
  5987. {
  5988. //look for local options
  5989. bool referenceIsFirst = false;
  5990. bool adjustScale = false;
  5991. bool enableFarthestPointRemoval = false;
  5992. double minErrorDiff = 1.0e-6;
  5993. unsigned iterationCount = 0;
  5994. unsigned randomSamplingLimit = 20000;
  5995. unsigned overlap = 100;
  5996. int modelWeightsSFIndex = -1;
  5997. QString modelWeightsSFIndexName;
  5998. int dataWeightsSFIndex = -1;
  5999. QString dataWeightsSFIndexName;
  6000. int maxThreadCount = 0;
  6001. int transformationFilters = CCCoreLib::RegistrationTools::SKIP_NONE;
  6002. bool useC2MDistances = false;
  6003. bool robustC2MDistances = true;
  6004. CCCoreLib::ICPRegistrationTools::NORMALS_MATCHING normalsMatching = CCCoreLib::ICPRegistrationTools::NO_NORMAL;
  6005. while (!cmd.arguments().empty())
  6006. {
  6007. QString argument = cmd.arguments().front();
  6008. if (ccCommandLineInterface::IsCommand(argument, COMMAND_ICP_REFERENCE_IS_FIRST))
  6009. {
  6010. //local option confirmed, we can move on
  6011. cmd.arguments().pop_front();
  6012. referenceIsFirst = true;
  6013. }
  6014. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_ICP_ADJUST_SCALE))
  6015. {
  6016. //local option confirmed, we can move on
  6017. cmd.arguments().pop_front();
  6018. adjustScale = true;
  6019. }
  6020. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_ICP_ENABLE_FARTHEST_REMOVAL))
  6021. {
  6022. //local option confirmed, we can move on
  6023. cmd.arguments().pop_front();
  6024. enableFarthestPointRemoval = true;
  6025. }
  6026. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_ICP_MIN_ERROR_DIIF))
  6027. {
  6028. //local option confirmed, we can move on
  6029. cmd.arguments().pop_front();
  6030. if (cmd.arguments().empty())
  6031. {
  6032. return cmd.error(QObject::tr("Missing parameter: min error difference after '%1'").arg(COMMAND_ICP_MIN_ERROR_DIIF));
  6033. }
  6034. bool ok;
  6035. minErrorDiff = cmd.arguments().takeFirst().toDouble(&ok);
  6036. if (!ok || minErrorDiff <= 0)
  6037. {
  6038. return cmd.error(QObject::tr("Invalid value for min. error difference! (after %1)").arg(COMMAND_ICP_MIN_ERROR_DIIF));
  6039. }
  6040. }
  6041. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_ICP_ITERATION_COUNT))
  6042. {
  6043. //local option confirmed, we can move on
  6044. cmd.arguments().pop_front();
  6045. if (cmd.arguments().empty())
  6046. {
  6047. return cmd.error(QObject::tr("Missing parameter: number of iterations after '%1'").arg(COMMAND_ICP_ITERATION_COUNT));
  6048. }
  6049. bool ok;
  6050. QString arg = cmd.arguments().takeFirst();
  6051. iterationCount = arg.toUInt(&ok);
  6052. if (!ok || iterationCount == 0)
  6053. return cmd.error(QObject::tr("Invalid number of iterations! (%1)").arg(arg));
  6054. }
  6055. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_ICP_OVERLAP))
  6056. {
  6057. //local option confirmed, we can move on
  6058. cmd.arguments().pop_front();
  6059. if (cmd.arguments().empty())
  6060. {
  6061. return cmd.error(QObject::tr("Missing parameter: overlap percentage after '%1'").arg(COMMAND_ICP_OVERLAP));
  6062. }
  6063. bool ok;
  6064. QString arg = cmd.arguments().takeFirst();
  6065. overlap = arg.toUInt(&ok);
  6066. if (!ok || overlap < 10 || overlap > 100)
  6067. {
  6068. return cmd.error(QObject::tr("Invalid overlap value! (%1 --> should be between 10 and 100)").arg(arg));
  6069. }
  6070. }
  6071. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_ICP_RANDOM_SAMPLING_LIMIT))
  6072. {
  6073. //local option confirmed, we can move on
  6074. cmd.arguments().pop_front();
  6075. if (cmd.arguments().empty())
  6076. {
  6077. return cmd.error(QObject::tr("Missing parameter: random sampling limit value after '%1'").arg(COMMAND_ICP_RANDOM_SAMPLING_LIMIT));
  6078. }
  6079. bool ok;
  6080. randomSamplingLimit = cmd.arguments().takeFirst().toUInt(&ok);
  6081. if (!ok || randomSamplingLimit < 3)
  6082. {
  6083. return cmd.error(QObject::tr("Invalid random sampling limit! (after %1)").arg(COMMAND_ICP_RANDOM_SAMPLING_LIMIT));
  6084. }
  6085. }
  6086. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_ICP_USE_MODEL_SF_AS_WEIGHT))
  6087. {
  6088. //local option confirmed, we can move on
  6089. cmd.arguments().pop_front();
  6090. if (cmd.arguments().empty())
  6091. {
  6092. return cmd.error(QObject::tr("Missing parameter: SF index after '%1'").arg(COMMAND_ICP_USE_MODEL_SF_AS_WEIGHT));
  6093. }
  6094. //read SF index
  6095. if (!GetSFIndexOrName(cmd, modelWeightsSFIndex, modelWeightsSFIndexName, true))
  6096. {
  6097. return false;
  6098. }
  6099. }
  6100. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_ICP_USE_DATA_SF_AS_WEIGHT))
  6101. {
  6102. //local option confirmed, we can move on
  6103. cmd.arguments().pop_front();
  6104. if (cmd.arguments().empty())
  6105. {
  6106. return cmd.error(QObject::tr("Missing parameter: SF index after '%1'").arg(COMMAND_ICP_USE_DATA_SF_AS_WEIGHT));
  6107. }
  6108. //read SF index
  6109. if (!GetSFIndexOrName(cmd, dataWeightsSFIndex, dataWeightsSFIndexName, true))
  6110. {
  6111. return false;
  6112. }
  6113. }
  6114. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_MAX_THREAD_COUNT))
  6115. {
  6116. //local option confirmed, we can move on
  6117. cmd.arguments().pop_front();
  6118. if (cmd.arguments().empty())
  6119. {
  6120. return cmd.error(QObject::tr("Missing parameter: max thread count after '%1'").arg(COMMAND_MAX_THREAD_COUNT));
  6121. }
  6122. bool ok;
  6123. maxThreadCount = cmd.arguments().takeFirst().toInt(&ok);
  6124. if (!ok || maxThreadCount < 0)
  6125. {
  6126. return cmd.error(QObject::tr("Invalid thread count! (after %1)").arg(COMMAND_MAX_THREAD_COUNT));
  6127. }
  6128. }
  6129. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_ICP_ROT))
  6130. {
  6131. //local option confirmed, we can move on
  6132. cmd.arguments().pop_front();
  6133. if (!cmd.arguments().empty())
  6134. {
  6135. QString rotation = cmd.arguments().takeFirst().toUpper();
  6136. //invalidate all previous rotations in case -ROT used twice
  6137. cmd.print(QObject::tr("[ICP] Reset rotation constraints if any. Only one -%1 argument allowed").arg(COMMAND_ICP_ROT));
  6138. transformationFilters &= (~CCCoreLib::RegistrationTools::SKIP_ROTATION);
  6139. if (rotation == "XYZ")
  6140. {
  6141. cmd.print(QObject::tr("[ICP] Use all rotations"));
  6142. }
  6143. else if (rotation == "X")
  6144. {
  6145. transformationFilters |= CCCoreLib::RegistrationTools::SKIP_RYZ;
  6146. cmd.print(QObject::tr("[ICP] Skip RYZ"));
  6147. }
  6148. else if (rotation == "Y")
  6149. {
  6150. transformationFilters |= CCCoreLib::RegistrationTools::SKIP_RXZ;
  6151. cmd.print(QObject::tr("[ICP] Skip RXZ"));
  6152. }
  6153. else if (rotation == "Z")
  6154. {
  6155. transformationFilters |= CCCoreLib::RegistrationTools::SKIP_RXY;
  6156. cmd.print(QObject::tr("[ICP] Skip RXY"));
  6157. }
  6158. else if (rotation == "NONE")
  6159. {
  6160. transformationFilters |= CCCoreLib::RegistrationTools::SKIP_ROTATION;
  6161. cmd.print(QObject::tr("[ICP] Skip rotation"));
  6162. }
  6163. else
  6164. {
  6165. return cmd.error(QObject::tr("Invalid parameter: unknown rotation filter \"%1\"").arg(rotation));
  6166. }
  6167. }
  6168. else
  6169. {
  6170. return cmd.error(QObject::tr("Missing parameter: rotation filter after \"-%1\" (XYZ/X/Y/Z/NONE)").arg(COMMAND_ICP_ROT));
  6171. }
  6172. }
  6173. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_ICP_SKIP_TX))
  6174. {
  6175. transformationFilters |= CCCoreLib::RegistrationTools::SKIP_TX;
  6176. cmd.print(QObject::tr("[ICP] Skip TX"));
  6177. //local option confirmed, we can move on
  6178. cmd.arguments().pop_front();
  6179. }
  6180. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_ICP_SKIP_TY))
  6181. {
  6182. transformationFilters |= CCCoreLib::RegistrationTools::SKIP_TY;
  6183. cmd.print(QObject::tr("[ICP] Skip TY"));
  6184. //local option confirmed, we can move on
  6185. cmd.arguments().pop_front();
  6186. }
  6187. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_ICP_SKIP_TZ))
  6188. {
  6189. transformationFilters |= CCCoreLib::RegistrationTools::SKIP_TZ;
  6190. cmd.print(QObject::tr("[ICP] Skip TZ"));
  6191. //local option confirmed, we can move on
  6192. cmd.arguments().pop_front();
  6193. }
  6194. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_ICP_C2M_DIST))
  6195. {
  6196. useC2MDistances = true;
  6197. cmd.print(QObject::tr("[ICP] Use C2M distances"));
  6198. //local option confirmed, we can move on
  6199. cmd.arguments().pop_front();
  6200. }
  6201. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_C2M_DIST_NON_ROBUST))
  6202. {
  6203. robustC2MDistances = false;
  6204. cmd.warning(QObject::tr("[ICP] Use non-robust C2M distances"));
  6205. //local option confirmed, we can move on
  6206. cmd.arguments().pop_front();
  6207. }
  6208. else if (ccCommandLineInterface::IsCommand(argument, COMMAND_C2M_NORMAL_MATCHING))
  6209. {
  6210. //local option confirmed, we can move on
  6211. cmd.arguments().pop_front();
  6212. if (cmd.arguments().empty())
  6213. {
  6214. return cmd.error(QObject::tr("Missing parameter: normals matching mode after '%1'").arg(COMMAND_C2M_NORMAL_MATCHING));
  6215. }
  6216. QString normalsMatchingOption = cmd.arguments().takeFirst().toUpper();
  6217. if (normalsMatchingOption == "OPPOSITE")
  6218. {
  6219. normalsMatching = CCCoreLib::ICPRegistrationTools::OPPOSITE_NORMALS;
  6220. cmd.print(QObject::tr("[ICP] Use opposite normals matching mode"));
  6221. }
  6222. else if (normalsMatchingOption == "SAME_SIDE")
  6223. {
  6224. normalsMatching = CCCoreLib::ICPRegistrationTools::SAME_SIDE_NORMALS;
  6225. cmd.print(QObject::tr("[ICP] Use same-side normals matching mode"));
  6226. }
  6227. else if (normalsMatchingOption == "DOUBLE_SIDED")
  6228. {
  6229. normalsMatching = CCCoreLib::ICPRegistrationTools::DOUBLE_SIDED_NORMALS;
  6230. cmd.print(QObject::tr("[ICP] Use double-sided normals matching mode"));
  6231. }
  6232. else
  6233. {
  6234. return cmd.error(QObject::tr("Unknown normal matching mode: ") + normalsMatchingOption);
  6235. }
  6236. //local option confirmed, we can move on
  6237. cmd.arguments().pop_front();
  6238. }
  6239. else
  6240. {
  6241. break; //as soon as we encounter an unrecognized argument, we break the local loop to go back to the main one!
  6242. }
  6243. }
  6244. cmd.printDebug(QObject::tr("[ICP] Transfromation filter: %1").arg(transformationFilters));
  6245. //we'll get the first two entities
  6246. CLEntityDesc* dataAndModel[2]{ nullptr, nullptr };
  6247. {
  6248. int index = 0;
  6249. if (!cmd.clouds().empty())
  6250. {
  6251. dataAndModel[index++] = &cmd.clouds()[0];
  6252. if (cmd.clouds().size() > 1)
  6253. {
  6254. dataAndModel[index++] = &cmd.clouds()[1];
  6255. }
  6256. }
  6257. if (index < 2 && !cmd.meshes().empty())
  6258. {
  6259. dataAndModel[index++] = &cmd.meshes()[0];
  6260. if (index < 2 && cmd.meshes().size() > 1)
  6261. {
  6262. dataAndModel[index++] = &cmd.meshes()[1];
  6263. }
  6264. }
  6265. if (index < 2)
  6266. {
  6267. return cmd.error(QObject::tr("Not enough loaded entities (expect at least 2!)"));
  6268. }
  6269. }
  6270. //put them in the right order (data first, model next)
  6271. if (referenceIsFirst)
  6272. {
  6273. std::swap(dataAndModel[0], dataAndModel[1]);
  6274. }
  6275. //check that the weights (scalar fields) exist
  6276. if (dataWeightsSFIndex >= 0 || !dataWeightsSFIndexName.isEmpty())
  6277. {
  6278. ccPointCloud* dataAsCloud = ccHObjectCaster::ToPointCloud(dataAndModel[0]->getEntity());
  6279. if (dataAsCloud)
  6280. {
  6281. dataWeightsSFIndex = GetScalarFieldIndex(dataAsCloud, dataWeightsSFIndex, dataWeightsSFIndexName, true);
  6282. if (dataWeightsSFIndex >= 0)
  6283. {
  6284. cmd.print(QObject::tr("[ICP] SF #%1 (data entity) will be used as weights").arg(dataWeightsSFIndex));
  6285. dataAsCloud->setCurrentDisplayedScalarField(dataWeightsSFIndex);
  6286. }
  6287. }
  6288. }
  6289. if (modelWeightsSFIndex >= 0 || !modelWeightsSFIndexName.isEmpty())
  6290. {
  6291. ccPointCloud* modelAsCloud = ccHObjectCaster::ToPointCloud(dataAndModel[1]->getEntity());
  6292. if (modelAsCloud)
  6293. {
  6294. modelWeightsSFIndex = GetScalarFieldIndex(modelAsCloud, modelWeightsSFIndex, modelWeightsSFIndexName, true);
  6295. if (modelWeightsSFIndex >= 0)
  6296. {
  6297. cmd.print(QObject::tr("[ICP] SF #%1 (model entity) will be used as weights").arg(modelWeightsSFIndex));
  6298. modelAsCloud->setCurrentDisplayedScalarField(modelWeightsSFIndex);
  6299. }
  6300. }
  6301. }
  6302. ccGLMatrix transMat;
  6303. double finalError = 0.0;
  6304. double finalScale = 1.0;
  6305. unsigned finalPointCount = 0;
  6306. CCCoreLib::ICPRegistrationTools::Parameters parameters;
  6307. {
  6308. parameters.convType = (iterationCount != 0 ? CCCoreLib::ICPRegistrationTools::MAX_ITER_CONVERGENCE : CCCoreLib::ICPRegistrationTools::MAX_ERROR_CONVERGENCE);
  6309. parameters.minRMSDecrease = minErrorDiff;
  6310. parameters.nbMaxIterations = iterationCount;
  6311. parameters.adjustScale = adjustScale;
  6312. parameters.filterOutFarthestPoints = enableFarthestPointRemoval;
  6313. parameters.samplingLimit = randomSamplingLimit;
  6314. parameters.finalOverlapRatio = overlap / 100.0;
  6315. parameters.transformationFilters = transformationFilters;
  6316. parameters.maxThreadCount = maxThreadCount;
  6317. parameters.useC2MSignedDistances = useC2MDistances;
  6318. parameters.robustC2MSignedDistances = robustC2MDistances;
  6319. parameters.normalsMatching = normalsMatching;
  6320. }
  6321. if (ccRegistrationTools::ICP( dataAndModel[0]->getEntity(),
  6322. dataAndModel[1]->getEntity(),
  6323. transMat,
  6324. finalScale,
  6325. finalError,
  6326. finalPointCount,
  6327. parameters,
  6328. dataWeightsSFIndex >= 0,
  6329. modelWeightsSFIndex >= 0,
  6330. cmd.widgetParent()))
  6331. {
  6332. ccHObject* data = dataAndModel[0]->getEntity();
  6333. data->applyGLTransformation_recursive(&transMat);
  6334. cmd.print(QObject::tr("Entity '%1' has been registered").arg(data->getName()));
  6335. cmd.print(QObject::tr("RMS: %1").arg(finalError));
  6336. cmd.print(QObject::tr("Number of points used for final step: %1").arg(finalPointCount));
  6337. //save matrix in a separate text file
  6338. {
  6339. QString txtFilename = QObject::tr("%1/%2_REGISTRATION_MATRIX").arg(dataAndModel[0]->path, dataAndModel[0]->basename);
  6340. if (cmd.addTimestamp())
  6341. {
  6342. QString timestamp = QDateTime::currentDateTime().toString("yyyy-MM-dd_hh'h'mm_ss_zzz");
  6343. txtFilename += QString("_%1").arg(timestamp);
  6344. }
  6345. txtFilename += ".txt";
  6346. QFile txtFile(txtFilename);
  6347. if (txtFile.open(QIODevice::WriteOnly | QIODevice::Text))
  6348. {
  6349. QTextStream txtStream(&txtFile);
  6350. txtStream << transMat.toString(cmd.numericalPrecision(), ' ') << endl;
  6351. txtFile.close();
  6352. }
  6353. else
  6354. {
  6355. cmd.warning("Failed to save the registration matrix to file " + txtFilename);
  6356. }
  6357. }
  6358. dataAndModel[0]->basename += QObject::tr("_REGISTERED");
  6359. if (cmd.autoSaveMode())
  6360. {
  6361. QString errorStr = cmd.exportEntity(*dataAndModel[0]);
  6362. if (!errorStr.isEmpty())
  6363. {
  6364. return cmd.error(errorStr);
  6365. }
  6366. }
  6367. }
  6368. else
  6369. {
  6370. return false;
  6371. }
  6372. return true;
  6373. }
  6374. CommandChangePLYExportFormat::CommandChangePLYExportFormat()
  6375. : ccCommandLineInterface::Command(QObject::tr("Change PLY output format"), COMMAND_PLY_EXPORT_FORMAT)
  6376. {}
  6377. bool CommandChangePLYExportFormat::process(ccCommandLineInterface& cmd)
  6378. {
  6379. if (cmd.arguments().empty())
  6380. {
  6381. return cmd.error(QObject::tr("Missing parameter: format (ASCII, BINARY_LE, or BINARY_BE) after '%1'").arg(COMMAND_PLY_EXPORT_FORMAT));
  6382. }
  6383. //if (fileFilter != PlyFilter::GetFileFilter())
  6384. // cmd.warning(QObject::tr("Argument '%1' is only applicable to PLY format!").arg(argument));
  6385. QString plyFormat = cmd.arguments().takeFirst().toUpper();
  6386. //printf("%s\n",qPrintable(plyFormat));
  6387. if (plyFormat == "ASCII")
  6388. {
  6389. PlyFilter::SetDefaultOutputFormat(PLY_ASCII);
  6390. }
  6391. else if (plyFormat == "BINARY_BE")
  6392. {
  6393. PlyFilter::SetDefaultOutputFormat(PLY_BIG_ENDIAN);
  6394. }
  6395. else if (plyFormat == "BINARY_LE")
  6396. {
  6397. PlyFilter::SetDefaultOutputFormat(PLY_LITTLE_ENDIAN);
  6398. }
  6399. else
  6400. {
  6401. return cmd.error(QObject::tr("Invalid PLY format! ('%1')").arg(plyFormat));
  6402. }
  6403. return true;
  6404. }
  6405. CommandForceNormalsComputation::CommandForceNormalsComputation()
  6406. : ccCommandLineInterface::Command(QObject::tr("Compute structured cloud normals"), COMMAND_COMPUTE_GRIDDED_NORMALS)
  6407. {}
  6408. bool CommandForceNormalsComputation::process(ccCommandLineInterface& cmd)
  6409. {
  6410. //simply change the default filter behavior
  6411. cmd.fileLoadingParams().autoComputeNormals = true;
  6412. return true;
  6413. }
  6414. CommandSave::CommandSave(const QString& name, const QString& keyword)
  6415. : ccCommandLineInterface::Command(name, keyword)
  6416. {}
  6417. bool CommandSave::ParseFileNames(ccCommandLineInterface& cmd, QStringList &fileNames)
  6418. {
  6419. //
  6420. // File list is space separated, but can use quotes to include spaces in the file names
  6421. //
  6422. auto argument = cmd.arguments().takeFirst();
  6423. while (!argument.isEmpty())
  6424. {
  6425. auto firstChar = argument.at(0);
  6426. if (firstChar == '\'' || firstChar == '\"')
  6427. {
  6428. auto end = argument.indexOf(firstChar, 1);
  6429. if (end == -1)
  6430. {
  6431. return cmd.error(QObject::tr("A file starting with %1 does not have a closing %1").arg(firstChar));
  6432. }
  6433. fileNames.push_back(argument.mid(1, end - 1));
  6434. argument.remove(0, end + 1);
  6435. if (argument.startsWith(' '))
  6436. {
  6437. argument.remove(0, 1);
  6438. }
  6439. }
  6440. else
  6441. {
  6442. auto end = argument.indexOf(' ');
  6443. if (end == -1)
  6444. {
  6445. end = argument.length();
  6446. }
  6447. fileNames.push_back(argument.left(end));
  6448. argument.remove(0, end + 1);
  6449. }
  6450. }
  6451. return true;
  6452. }
  6453. void CommandSave::SetFileDesc(CLEntityDesc &desc, const QString &fileName)
  6454. {
  6455. QFileInfo fInfo(fileName);
  6456. desc.basename = fInfo.fileName();
  6457. desc.path = fInfo.filePath().left(fInfo.filePath().length() - fInfo.fileName().length());
  6458. }
  6459. CommandSaveClouds::CommandSaveClouds()
  6460. : CommandSave(QObject::tr("Save clouds"), COMMAND_SAVE_CLOUDS)
  6461. {}
  6462. bool CommandSaveClouds::process(ccCommandLineInterface& cmd)
  6463. {
  6464. bool allAtOnce = false;
  6465. bool setFileNames = false;
  6466. QStringList fileNames;
  6467. //look for additional parameters
  6468. while (!cmd.arguments().empty())
  6469. {
  6470. QString argument = cmd.arguments().front();
  6471. if (argument.toUpper() == OPTION_ALL_AT_ONCE)
  6472. {
  6473. //local option confirmed, we can move on
  6474. cmd.arguments().pop_front();
  6475. allAtOnce = true;
  6476. }
  6477. else if (argument.startsWith(OPTION_FILE_NAMES))
  6478. {
  6479. cmd.arguments().pop_front();
  6480. setFileNames = true;
  6481. if (!ParseFileNames(cmd, fileNames))
  6482. {
  6483. return false;
  6484. }
  6485. }
  6486. else
  6487. {
  6488. break; //as soon as we encounter an unrecognized argument, we break the local loop to go back to the main one!
  6489. }
  6490. }
  6491. if (setFileNames && allAtOnce && fileNames.size() != 1)
  6492. {
  6493. return cmd.error(QObject::tr("Invalid parameter: specified %1 file names, but ALL_AT_ONCE is on").arg(fileNames.size()));
  6494. }
  6495. if (setFileNames && !allAtOnce && fileNames.size() != cmd.clouds().size())
  6496. {
  6497. return cmd.error(QObject::tr("Invalid parameter: specified %1 file names, but there are %2 clouds").arg(fileNames.size()).arg(cmd.clouds().size()));
  6498. }
  6499. QString ext = cmd.cloudExportExt();
  6500. bool autoAddTimestamp = cmd.addTimestamp();
  6501. if (setFileNames)
  6502. {
  6503. cmd.toggleAddTimestamp(false);
  6504. cmd.setCloudExportFormat(cmd.cloudExportFormat(), QString());
  6505. if (!allAtOnce)
  6506. {
  6507. for (int i = 0; i < fileNames.size(); ++i)
  6508. {
  6509. SetFileDesc(cmd.clouds()[i], fileNames[i]);
  6510. }
  6511. }
  6512. }
  6513. bool res = cmd.saveClouds(QString(), allAtOnce, allAtOnce && setFileNames ? &fileNames[0] : nullptr);
  6514. if (setFileNames)
  6515. {
  6516. cmd.toggleAddTimestamp(autoAddTimestamp);
  6517. cmd.setCloudExportFormat(cmd.cloudExportFormat(), ext);
  6518. }
  6519. return res;
  6520. }
  6521. CommandSaveMeshes::CommandSaveMeshes()
  6522. : CommandSave(QObject::tr("Save meshes"), COMMAND_SAVE_MESHES)
  6523. {}
  6524. bool CommandSaveMeshes::process(ccCommandLineInterface& cmd)
  6525. {
  6526. bool allAtOnce = false;
  6527. bool setFileNames = false;
  6528. QStringList fileNames;
  6529. //look for additional parameters
  6530. while (!cmd.arguments().empty())
  6531. {
  6532. QString argument = cmd.arguments().front();
  6533. if (argument.toUpper() == OPTION_ALL_AT_ONCE)
  6534. {
  6535. //local option confirmed, we can move on
  6536. cmd.arguments().pop_front();
  6537. allAtOnce = true;
  6538. }
  6539. else if (argument.left(sizeof(OPTION_FILE_NAMES) - 1).toUpper() == OPTION_FILE_NAMES)
  6540. {
  6541. cmd.arguments().pop_front();
  6542. setFileNames = true;
  6543. if (!ParseFileNames(cmd, fileNames))
  6544. {
  6545. return false;
  6546. }
  6547. }
  6548. else
  6549. {
  6550. break; //as soon as we encounter an unrecognized argument, we break the local loop to go back to the main one!
  6551. }
  6552. }
  6553. if (setFileNames && allAtOnce && fileNames.size() != 1)
  6554. {
  6555. return cmd.error(QObject::tr("Invalid parameter: specified %1 file names, but ALL_AT_ONCE is on").arg(fileNames.size()));
  6556. }
  6557. if (setFileNames && !allAtOnce && fileNames.size() != cmd.meshes().size())
  6558. {
  6559. return cmd.error(QObject::tr("Invalid parameter: specified %1 file names, but there are %2 meshes").arg(fileNames.size()).arg(cmd.meshes().size()));
  6560. }
  6561. QString ext = cmd.meshExportExt();
  6562. bool autoAddTimestamp = cmd.addTimestamp();
  6563. if (setFileNames)
  6564. {
  6565. cmd.toggleAddTimestamp(false);
  6566. cmd.setMeshExportFormat(cmd.meshExportFormat(), QString());
  6567. if (!allAtOnce)
  6568. {
  6569. for (int i = 0; i < fileNames.size(); ++i)
  6570. {
  6571. SetFileDesc(cmd.meshes()[i], fileNames[i]);
  6572. }
  6573. }
  6574. }
  6575. bool res = cmd.saveMeshes(QString(), allAtOnce, allAtOnce && setFileNames ? &fileNames[0] : nullptr);
  6576. if (setFileNames)
  6577. {
  6578. cmd.toggleAddTimestamp(autoAddTimestamp);
  6579. cmd.setMeshExportFormat(cmd.meshExportFormat(), ext);
  6580. }
  6581. return res;
  6582. }
  6583. CommandAutoSave::CommandAutoSave()
  6584. : ccCommandLineInterface::Command(QObject::tr("Auto save state"), COMMAND_AUTO_SAVE)
  6585. {}
  6586. bool CommandAutoSave::process(ccCommandLineInterface& cmd)
  6587. {
  6588. if (cmd.arguments().empty())
  6589. {
  6590. return cmd.error(QObject::tr("Missing parameter: option after '%1' (%2/%3)").arg(COMMAND_AUTO_SAVE, OPTION_ON, OPTION_OFF));
  6591. }
  6592. QString option = cmd.arguments().takeFirst().toUpper();
  6593. if (option == OPTION_ON)
  6594. {
  6595. cmd.print(QObject::tr("Auto-save is enabled"));
  6596. cmd.toggleAutoSaveMode(true);
  6597. }
  6598. else if (option == OPTION_OFF)
  6599. {
  6600. cmd.print(QObject::tr("Auto-save is disabled"));
  6601. cmd.toggleAutoSaveMode(false);
  6602. }
  6603. else
  6604. {
  6605. return cmd.error(QObject::tr("Unrecognized option after '%1' (%2 or %3 expected)").arg(COMMAND_AUTO_SAVE, OPTION_ON, OPTION_OFF));
  6606. }
  6607. return true;
  6608. }
  6609. CommandLogFile::CommandLogFile()
  6610. : ccCommandLineInterface::Command(QObject::tr("Set log file"), COMMAND_LOG_FILE)
  6611. {}
  6612. bool CommandLogFile::process(ccCommandLineInterface& cmd)
  6613. {
  6614. if (cmd.arguments().empty())
  6615. {
  6616. return cmd.error(QObject::tr("Missing parameter: filename after '%1'").arg(COMMAND_LOG_FILE));
  6617. }
  6618. QString filename = cmd.arguments().takeFirst();
  6619. if (!ccConsole::TheInstance(false))
  6620. {
  6621. assert(cmd.silentMode());
  6622. ccConsole::Init();
  6623. }
  6624. return ccConsole::TheInstance()->setLogFile(filename);
  6625. }
  6626. CommandSelectEntities::CommandSelectEntities()
  6627. : ccCommandLineInterface::Command(QObject::tr("Select entities"), COMMAND_SELECT_ENTITIES)
  6628. {}
  6629. bool CommandSelectEntities::process(ccCommandLineInterface& cmd)
  6630. {
  6631. //option handling
  6632. //look for additional parameters
  6633. ccCommandLineInterface::SelectEntitiesOptions options;
  6634. bool selectMeshes = false;
  6635. bool selectClouds = false;
  6636. while (!cmd.arguments().empty())
  6637. {
  6638. QString argument = cmd.arguments().front().toUpper();
  6639. if (ccCommandLineInterface::IsCommand(argument, OPTION_ALL))
  6640. {
  6641. //local option confirmed, we can move on
  6642. cmd.arguments().pop_front();
  6643. options.selectAll = true;
  6644. //no other params needed
  6645. }
  6646. else if (ccCommandLineInterface::IsCommand(argument, OPTION_FIRST))
  6647. {
  6648. //local option confirmed, we can move on
  6649. cmd.arguments().pop_front();
  6650. //it requires a number after the argument
  6651. if (cmd.arguments().empty())
  6652. {
  6653. return cmd.error(QObject::tr("Missing parameter: number of entities after %1").arg(OPTION_FIRST));
  6654. }
  6655. bool ok;
  6656. options.firstNr = cmd.arguments().takeFirst().toUInt(&ok);
  6657. if (!ok)
  6658. {
  6659. return cmd.error(QObject::tr("Invalid number after -%1").arg(OPTION_FIRST));
  6660. }
  6661. options.selectFirst = true;
  6662. }
  6663. else if (ccCommandLineInterface::IsCommand(argument, OPTION_LAST))
  6664. {
  6665. //local option confirmed, we can move on
  6666. cmd.arguments().pop_front();
  6667. //it requires a number after the argument
  6668. if (cmd.arguments().empty())
  6669. {
  6670. return cmd.error(QObject::tr("Missing parameter: number of entities after %1").arg(OPTION_LAST));
  6671. }
  6672. bool ok;
  6673. options.lastNr = cmd.arguments().takeFirst().toUInt(&ok);
  6674. if (!ok)
  6675. {
  6676. return cmd.error(QObject::tr("Invalid number after -%1").arg(OPTION_LAST));
  6677. }
  6678. options.selectLast = true;
  6679. }
  6680. else if (ccCommandLineInterface::IsCommand(argument, OPTION_REGEX))
  6681. {
  6682. //local option confirmed, we can move on
  6683. cmd.arguments().pop_front();
  6684. options.selectRegex = true;
  6685. //it requires a string after the argument
  6686. if (cmd.arguments().empty())
  6687. {
  6688. return cmd.error(QObject::tr("Missing parameter: regex string after %1").arg(OPTION_REGEX));
  6689. }
  6690. QString regexString = cmd.arguments().takeFirst();
  6691. options.regex.setPattern(regexString);
  6692. if (!options.regex.isValid())
  6693. {
  6694. return cmd.error(QObject::tr("Invalid regex pattern: %1").arg(options.regex.errorString()));
  6695. }
  6696. }
  6697. else if (ccCommandLineInterface::IsCommand(argument, OPTION_NOT))
  6698. {
  6699. //local option confirmed, we can move on
  6700. cmd.arguments().pop_front();
  6701. options.reverse = true;
  6702. //no other params needed
  6703. }
  6704. else if (ccCommandLineInterface::IsCommand(argument, OPTION_CLOUD))
  6705. {
  6706. //local option confirmed, we can move on
  6707. cmd.arguments().pop_front();
  6708. selectClouds = true;
  6709. }
  6710. else if (ccCommandLineInterface::IsCommand(argument, OPTION_MESH))
  6711. {
  6712. //local option confirmed, we can move on
  6713. cmd.arguments().pop_front();
  6714. selectMeshes = true;
  6715. }
  6716. else
  6717. {
  6718. break; //as soon as we encounter an unrecognized argument, we break the local loop to go back to the main one!
  6719. }
  6720. }
  6721. if (options.selectAll)
  6722. {
  6723. //overwrite any other mode (first/last/regex)
  6724. options.selectFirst = false;
  6725. options.selectLast = false;
  6726. options.selectRegex = false;
  6727. if (options.reverse)
  6728. {
  6729. //NONE
  6730. cmd.print("No entities will be selected.");
  6731. }
  6732. else
  6733. {
  6734. //ALL
  6735. cmd.print("All entities will be selected. Other options will be ignored except -NOT.");
  6736. }
  6737. }
  6738. if (options.selectFirst)
  6739. {
  6740. if (options.reverse)
  6741. {
  6742. if (options.selectLast)
  6743. {
  6744. //not first {nr} and not last {nr}
  6745. cmd.print(QObject::tr("First %1 and last %2 entity(ies) will not be selected").arg(options.firstNr).arg(options.lastNr));
  6746. }
  6747. else
  6748. {
  6749. //not first {nr}
  6750. cmd.print(QObject::tr("First %1 entity(ies) will not be selected").arg(options.firstNr));
  6751. }
  6752. }
  6753. else
  6754. {
  6755. //first {nr}
  6756. cmd.print(QObject::tr("First %1 entity(ies) will be selected").arg(options.firstNr));
  6757. }
  6758. }
  6759. if (options.selectLast)
  6760. {
  6761. if (options.reverse)
  6762. {
  6763. if (!options.selectFirst)
  6764. {
  6765. //not last {nr}
  6766. cmd.print(QObject::tr("Last %1 entity(ies) will not be selected").arg(options.lastNr));
  6767. }
  6768. }
  6769. else
  6770. {
  6771. //last {nr}
  6772. cmd.print(QObject::tr("Last %1 entity(ies) will be selected").arg(options.lastNr));
  6773. }
  6774. }
  6775. if (options.selectRegex)
  6776. {
  6777. if (options.reverse)
  6778. {
  6779. //regex not matches
  6780. cmd.print(QObject::tr("Entities with name matches the regex /%1/ will not be selected.").arg(options.regex.pattern()));
  6781. }
  6782. else
  6783. {
  6784. //regex matches
  6785. cmd.print(QObject::tr("Entities with name matches the regex /%1/ will be selected.").arg(options.regex.pattern()));
  6786. }
  6787. }
  6788. //no option was set
  6789. if (!options.selectFirst && !options.selectLast && !options.selectRegex && !options.selectAll)
  6790. {
  6791. return cmd.error(QObject::tr("Missing parameter(s): any of the option (%1,%2,%3,%4) expected after %5")
  6792. .arg(OPTION_ALL)
  6793. .arg(OPTION_FIRST)
  6794. .arg(OPTION_LAST)
  6795. .arg(OPTION_REGEX)
  6796. .arg(COMMAND_SELECT_ENTITIES));
  6797. }
  6798. //no entity type was selected so select both clouds and meshes
  6799. if (!selectClouds && !selectMeshes)
  6800. {
  6801. selectClouds = true;
  6802. selectMeshes = true;
  6803. }
  6804. if (selectClouds)
  6805. {
  6806. cmd.print(QObject::tr("[Select clouds]"));
  6807. if (!cmd.selectClouds(options))
  6808. {
  6809. //error message already sent
  6810. return false;
  6811. }
  6812. }
  6813. if (selectMeshes)
  6814. {
  6815. cmd.print(QObject::tr("[Select meshes]"));
  6816. if (!cmd.selectMeshes(options))
  6817. {
  6818. //error message already sent
  6819. return false;
  6820. }
  6821. }
  6822. return true;
  6823. }
  6824. CommandClear::CommandClear()
  6825. : ccCommandLineInterface::Command(QObject::tr("Clear"), COMMAND_CLEAR)
  6826. {}
  6827. bool CommandClear::process(ccCommandLineInterface& cmd)
  6828. {
  6829. cmd.removeClouds(false);
  6830. cmd.removeMeshes(false);
  6831. return true;
  6832. }
  6833. CommandClearClouds::CommandClearClouds()
  6834. : ccCommandLineInterface::Command(QObject::tr("Clear clouds"), COMMAND_CLEAR_CLOUDS)
  6835. {}
  6836. bool CommandClearClouds::process(ccCommandLineInterface& cmd)
  6837. {
  6838. cmd.removeClouds(false);
  6839. return true;
  6840. }
  6841. CommandPopClouds::CommandPopClouds()
  6842. : ccCommandLineInterface::Command(QObject::tr("Pop cloud"), COMMAND_POP_CLOUDS)
  6843. {}
  6844. bool CommandPopClouds::process(ccCommandLineInterface& cmd)
  6845. {
  6846. cmd.removeClouds(true);
  6847. return true;
  6848. }
  6849. CommandClearMeshes::CommandClearMeshes()
  6850. : ccCommandLineInterface::Command(QObject::tr("Clear meshes"), COMMAND_CLEAR_MESHES)
  6851. {}
  6852. bool CommandClearMeshes::process(ccCommandLineInterface& cmd)
  6853. {
  6854. cmd.removeMeshes(false);
  6855. return true;
  6856. }
  6857. CommandPopMeshes::CommandPopMeshes()
  6858. : ccCommandLineInterface::Command(QObject::tr("Pop mesh"), COMMAND_POP_MESHES)
  6859. {}
  6860. bool CommandPopMeshes::process(ccCommandLineInterface& cmd)
  6861. {
  6862. cmd.removeMeshes(true);
  6863. return true;
  6864. }
  6865. CommandSetNoTimestamp::CommandSetNoTimestamp()
  6866. : ccCommandLineInterface::Command(QObject::tr("No timestamp"), COMMAND_NO_TIMESTAMP)
  6867. {}
  6868. bool CommandSetNoTimestamp::process(ccCommandLineInterface& cmd)
  6869. {
  6870. cmd.toggleAddTimestamp(false);
  6871. return true;
  6872. }
  6873. CommandMoment::CommandMoment()
  6874. : ccCommandLineInterface::Command(QObject::tr("1st order moment"), COMMAND_MOMENT)
  6875. {}
  6876. bool CommandMoment::process(ccCommandLineInterface& cmd)
  6877. {
  6878. if (cmd.arguments().empty())
  6879. {
  6880. return cmd.error(QObject::tr("Missing parameter: kernel size after %1").arg(COMMAND_MOMENT));
  6881. }
  6882. bool paramOk = false;
  6883. QString kernelStr = cmd.arguments().takeFirst();
  6884. PointCoordinateType kernelSize = static_cast<PointCoordinateType>(kernelStr.toDouble(&paramOk));
  6885. if (!paramOk)
  6886. {
  6887. return cmd.error(QObject::tr("Failed to read a numerical parameter: kernel size. Got '%1' instead.").arg(kernelStr));
  6888. }
  6889. cmd.print(QObject::tr("\tKernel size: %1").arg(kernelSize));
  6890. if (cmd.clouds().empty())
  6891. {
  6892. return cmd.error(QObject::tr("No point cloud on which to compute first order moment! (be sure to open one with \"-%1 [cloud filename]\" before \"-%2\")").arg(COMMAND_OPEN, COMMAND_MOMENT));
  6893. }
  6894. //Call MainWindow generic method
  6895. ccHObject::Container entities;
  6896. entities.resize(cmd.clouds().size());
  6897. for (size_t i = 0; i < cmd.clouds().size(); ++i)
  6898. {
  6899. entities[i] = cmd.clouds()[i].pc;
  6900. }
  6901. if (ccLibAlgorithms::ComputeGeomCharacteristic(CCCoreLib::GeometricalAnalysisTools::MomentOrder1, 0, kernelSize, entities, nullptr, cmd.widgetParent()))
  6902. {
  6903. //save output
  6904. if (cmd.autoSaveMode() && !cmd.saveClouds(QObject::tr("MOMENT_KERNEL_%2").arg(kernelSize)))
  6905. {
  6906. return false;
  6907. }
  6908. }
  6909. return true;
  6910. }
  6911. CommandFeature::CommandFeature()
  6912. : ccCommandLineInterface::Command(QObject::tr("Feature"), COMMAND_FEATURE)
  6913. {}
  6914. bool CommandFeature::process(ccCommandLineInterface& cmd)
  6915. {
  6916. if (cmd.arguments().empty())
  6917. {
  6918. return cmd.error(QObject::tr("Missing parameter: feature type after \"-%1\"").arg(COMMAND_FEATURE));
  6919. }
  6920. QString featureTypeStr = cmd.arguments().takeFirst().toUpper();
  6921. CCCoreLib::Neighbourhood::GeomFeature featureType;
  6922. if (featureTypeStr == "SUM_OF_EIGENVALUES")
  6923. {
  6924. featureType = CCCoreLib::Neighbourhood::EigenValuesSum;
  6925. }
  6926. else if (featureTypeStr == "OMNIVARIANCE")
  6927. {
  6928. featureType = CCCoreLib::Neighbourhood::Omnivariance;
  6929. }
  6930. else if (featureTypeStr == "EIGENTROPY")
  6931. {
  6932. featureType = CCCoreLib::Neighbourhood::EigenEntropy;
  6933. }
  6934. else if (featureTypeStr == "ANISOTROPY")
  6935. {
  6936. featureType = CCCoreLib::Neighbourhood::Anisotropy;
  6937. }
  6938. else if (featureTypeStr == "PLANARITY")
  6939. {
  6940. featureType = CCCoreLib::Neighbourhood::Planarity;
  6941. }
  6942. else if (featureTypeStr == "LINEARITY")
  6943. {
  6944. featureType = CCCoreLib::Neighbourhood::Linearity;
  6945. }
  6946. else if (featureTypeStr == "PCA1")
  6947. {
  6948. featureType = CCCoreLib::Neighbourhood::PCA1;
  6949. }
  6950. else if (featureTypeStr == "PCA2")
  6951. {
  6952. featureType = CCCoreLib::Neighbourhood::PCA2;
  6953. }
  6954. else if (featureTypeStr == "SURFACE_VARIATION")
  6955. {
  6956. featureType = CCCoreLib::Neighbourhood::SurfaceVariation;
  6957. }
  6958. else if (featureTypeStr == "SPHERICITY")
  6959. {
  6960. featureType = CCCoreLib::Neighbourhood::Sphericity;
  6961. }
  6962. else if (featureTypeStr == "VERTICALITY")
  6963. {
  6964. featureType = CCCoreLib::Neighbourhood::Verticality;
  6965. }
  6966. else if (featureTypeStr == "EIGENVALUE1")
  6967. {
  6968. featureType = CCCoreLib::Neighbourhood::EigenValue1;
  6969. }
  6970. else if (featureTypeStr == "EIGENVALUE2")
  6971. {
  6972. featureType = CCCoreLib::Neighbourhood::EigenValue2;
  6973. }
  6974. else if (featureTypeStr == "EIGENVALUE3")
  6975. {
  6976. featureType = CCCoreLib::Neighbourhood::EigenValue3;
  6977. }
  6978. else
  6979. {
  6980. return cmd.error(QObject::tr("Invalid feature type after \"-%1\". Got '%2' instead of:\n\
  6981. - SUM_OF_EIGENVALUES\n\
  6982. - OMNIVARIANCE\n\
  6983. - EIGENTROPY\n\
  6984. - ANISOTROPY\n\
  6985. - PLANARITY\n\
  6986. - LINEARITY\n\
  6987. - PCA1\n\
  6988. - PCA2\n\
  6989. - SURFACE_VARIATION\n\
  6990. - SPHERICITY\n\
  6991. - VERTICALITY\n\
  6992. - EIGENVALUE1\n\
  6993. - EIGENVALUE2\n\
  6994. - EIGENVALUE3").arg(COMMAND_FEATURE, featureTypeStr));
  6995. }
  6996. if (cmd.arguments().empty())
  6997. {
  6998. return cmd.error(QObject::tr("Missing parameter: kernel size after feature type"));
  6999. }
  7000. bool paramOk = false;
  7001. QString kernelStr = cmd.arguments().takeFirst();
  7002. PointCoordinateType kernelSize = static_cast<PointCoordinateType>(kernelStr.toDouble(&paramOk));
  7003. if (!paramOk)
  7004. {
  7005. return cmd.error(QObject::tr("Failed to read a numerical parameter: kernel size. Got '%1' instead.").arg(kernelStr));
  7006. }
  7007. cmd.print(QObject::tr("\tKernel size: %1").arg(kernelSize));
  7008. if (cmd.clouds().empty())
  7009. {
  7010. return cmd.error(QObject::tr("No point cloud on which to compute feature! (be sure to open one with \"-%1 [cloud filename]\" before \"-%2\")").arg(COMMAND_OPEN, COMMAND_FEATURE));
  7011. }
  7012. //Call MainWindow generic method on all available clouds
  7013. ccHObject::Container entities;
  7014. {
  7015. entities.resize(cmd.clouds().size());
  7016. for (size_t i = 0; i < cmd.clouds().size(); ++i)
  7017. {
  7018. entities[i] = cmd.clouds()[i].pc;
  7019. }
  7020. }
  7021. if (!ccLibAlgorithms::ComputeGeomCharacteristic(CCCoreLib::GeometricalAnalysisTools::Feature, featureType, kernelSize, entities, nullptr, cmd.widgetParent()))
  7022. {
  7023. return cmd.error(QObject::tr("The computation of some geometric features failed."));
  7024. }
  7025. // on success, update the cloud names
  7026. QString fileNameExt = QObject::tr("%1_FEATURE_KERNEL_%2").arg(featureTypeStr).arg(kernelSize);
  7027. for (size_t i = 0; i < cmd.clouds().size(); ++i)
  7028. {
  7029. CLCloudDesc& desc = cmd.clouds()[i];
  7030. desc.basename += "_" + fileNameExt;
  7031. desc.pc->setName(entities[i]->getName() + QObject::tr(".%1_feature(%2)").arg(featureTypeStr.toLower()).arg(kernelSize));
  7032. }
  7033. //save output
  7034. if (cmd.autoSaveMode() && !cmd.saveClouds())
  7035. {
  7036. return false;
  7037. }
  7038. return true;
  7039. }
  7040. CommandDebugCmdLine::CommandDebugCmdLine()
  7041. : ccCommandLineInterface::Command(QObject::tr("Debug Command Line"), COMMAND_DEBUG)
  7042. {}
  7043. bool CommandDebugCmdLine::process(ccCommandLineInterface& cmd)
  7044. {
  7045. cmd.print("******************************************");
  7046. cmd.print("Number of selected clouds: " + QString::number(cmd.clouds().size()));
  7047. cmd.print("Number of selected meshes: " + QString::number(cmd.meshes().size()));
  7048. cmd.print("******************************************");
  7049. const QStringList& arguments = cmd.arguments();
  7050. cmd.print("Number of arguments: " + QString::number(arguments.size()));
  7051. for (int i = 0; i < arguments.size(); ++i)
  7052. {
  7053. cmd.print(QString("Argument #%1: < %2 >").arg(i + 1).arg(arguments[i]));
  7054. }
  7055. cmd.print("******************************************");
  7056. cmd.print("[Loading parameters]");
  7057. cmd.print(QObject::tr("Global shift set: ") + (cmd.fileLoadingParams().coordinatesShiftEnabled ? "yes" : "no"));
  7058. cmd.print(QObject::tr("Global shift: (%1, %2, %3)").arg(cmd.fileLoadingParams().coordinatesShift.x).arg(cmd.fileLoadingParams().coordinatesShift.y).arg(cmd.fileLoadingParams().coordinatesShift.z));
  7059. cmd.print("******************************************");
  7060. cmd.print("[Export parameters]");
  7061. cmd.print("Cloud export format: " + cmd.cloudExportFormat());
  7062. cmd.print("Mesh export format: " + cmd.meshExportFormat());
  7063. cmd.print("Group export format: " + cmd.hierarchyExportFormat());
  7064. cmd.print("******************************************");
  7065. cmd.print("[Other parameters]");
  7066. cmd.print(QObject::tr("Silent mode: ") + (cmd.silentMode() ? "ON" : "OFF"));
  7067. cmd.print(QObject::tr("Auto save: ") + (cmd.autoSaveMode() ? "ON" : "OFF"));
  7068. cmd.print(QObject::tr("Auto add timestamp: ") + (cmd.addTimestamp() ? "ON" : "OFF"));
  7069. cmd.print(QObject::tr("Numerical precision: %1").arg(cmd.numericalPrecision()));
  7070. return true;
  7071. }
  7072. CommandSetVerbosity::CommandSetVerbosity()
  7073. : ccCommandLineInterface::Command(QObject::tr("Set Verbosity"), COMMAND_VERBOSITY)
  7074. {}
  7075. bool CommandSetVerbosity::process(ccCommandLineInterface& cmd)
  7076. {
  7077. if (cmd.arguments().empty())
  7078. {
  7079. return cmd.error(QObject::tr("Missing parameter: verbosity level after: %1").arg(COMMAND_VERBOSITY));
  7080. }
  7081. bool ok = false;
  7082. int verbosityLevel = cmd.arguments().takeFirst().toInt(&ok);
  7083. if (!ok)
  7084. {
  7085. return cmd.error(QObject::tr("Invalid verbosity level %1").arg(verbosityLevel));
  7086. }
  7087. else
  7088. {
  7089. cmd.print(QObject::tr("Set verbosity level to %1").arg(verbosityLevel));
  7090. ccLog::SetVerbosityLevel(verbosityLevel);
  7091. }
  7092. return true;
  7093. }