1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463346434653466346734683469347034713472347334743475347634773478347934803481348234833484348534863487348834893490349134923493349434953496349734983499350035013502350335043505350635073508350935103511351235133514351535163517351835193520352135223523352435253526352735283529353035313532353335343535353635373538353935403541354235433544354535463547354835493550355135523553355435553556355735583559356035613562356335643565356635673568356935703571357235733574357535763577357835793580358135823583358435853586358735883589359035913592359335943595359635973598359936003601360236033604360536063607360836093610361136123613361436153616361736183619362036213622362336243625362636273628362936303631363236333634363536363637363836393640364136423643364436453646364736483649365036513652365336543655365636573658365936603661366236633664366536663667366836693670367136723673367436753676367736783679368036813682368336843685368636873688368936903691369236933694369536963697369836993700370137023703370437053706370737083709371037113712371337143715371637173718371937203721372237233724372537263727372837293730373137323733373437353736373737383739374037413742374337443745374637473748374937503751375237533754375537563757375837593760376137623763376437653766376737683769377037713772377337743775377637773778377937803781378237833784378537863787378837893790379137923793379437953796379737983799380038013802380338043805380638073808380938103811381238133814381538163817381838193820382138223823382438253826382738283829383038313832383338343835383638373838383938403841384238433844384538463847384838493850385138523853385438553856385738583859386038613862386338643865386638673868386938703871387238733874387538763877387838793880388138823883388438853886388738883889389038913892389338943895389638973898389939003901390239033904390539063907390839093910391139123913391439153916391739183919392039213922392339243925392639273928392939303931393239333934393539363937393839393940394139423943394439453946394739483949395039513952395339543955395639573958395939603961396239633964396539663967396839693970397139723973397439753976397739783979398039813982398339843985398639873988398939903991399239933994399539963997399839994000400140024003400440054006400740084009401040114012401340144015401640174018401940204021402240234024402540264027402840294030403140324033403440354036403740384039404040414042404340444045404640474048404940504051405240534054405540564057405840594060406140624063406440654066406740684069407040714072407340744075407640774078407940804081408240834084408540864087408840894090409140924093409440954096409740984099410041014102410341044105410641074108410941104111411241134114411541164117411841194120412141224123412441254126412741284129413041314132413341344135413641374138413941404141414241434144414541464147414841494150415141524153415441554156415741584159416041614162416341644165416641674168416941704171417241734174417541764177417841794180418141824183418441854186418741884189419041914192419341944195419641974198419942004201420242034204420542064207420842094210421142124213421442154216421742184219422042214222422342244225422642274228422942304231423242334234423542364237423842394240424142424243424442454246424742484249425042514252425342544255425642574258425942604261426242634264426542664267426842694270427142724273427442754276427742784279428042814282428342844285428642874288428942904291429242934294429542964297429842994300430143024303430443054306430743084309431043114312431343144315431643174318431943204321432243234324432543264327432843294330433143324333433443354336433743384339434043414342434343444345434643474348434943504351435243534354435543564357435843594360436143624363436443654366436743684369437043714372437343744375437643774378437943804381438243834384438543864387438843894390439143924393439443954396439743984399440044014402440344044405440644074408440944104411441244134414441544164417441844194420442144224423442444254426442744284429443044314432443344344435443644374438443944404441444244434444444544464447444844494450445144524453445444554456445744584459446044614462446344644465446644674468446944704471447244734474447544764477447844794480448144824483448444854486448744884489449044914492449344944495449644974498449945004501450245034504450545064507450845094510451145124513451445154516451745184519452045214522452345244525452645274528452945304531453245334534453545364537453845394540454145424543454445454546454745484549455045514552455345544555455645574558455945604561456245634564456545664567456845694570457145724573457445754576457745784579458045814582458345844585458645874588458945904591459245934594459545964597459845994600460146024603460446054606460746084609461046114612461346144615461646174618461946204621462246234624462546264627462846294630463146324633463446354636463746384639464046414642464346444645464646474648464946504651465246534654465546564657465846594660466146624663466446654666466746684669467046714672467346744675467646774678467946804681468246834684468546864687468846894690469146924693469446954696469746984699470047014702470347044705470647074708470947104711 |
- #ifdef NICE_USELIB_OPENMP
- #include <omp.h>
- #endif
- #include "msImageProcessor.h"
- #include <math.h>
- #include <stdio.h>
- #include <assert.h>
- #include <string.h>
- #include <stdlib.h>
- #include <iostream>
- using namespace std;
- msImageProcessor::msImageProcessor ( void )
- {
- clog << "[log] msImageProcessor::msImageProcessor: use Edison ";
- #ifdef NICE_USELIB_OPENMP
- clog << "parallel!" << endl;
- omp_set_dynamic ( 0 );
- #else
- clog << "seriell!" << endl;
- #endif
-
-
- modeTable = NULL;
- pointList = NULL;
- pointCount = 0;
-
- regionList = NULL;
-
- msRawData = NULL;
- labels = NULL;
- modes = NULL;
- modePointCounts = NULL;
- regionCount = 0;
-
-
- indexTable = NULL;
- LUV_data = NULL;
-
- raList = NULL;
- freeRAList = NULL;
- raPool = NULL;
-
- visitTable = NULL;
-
-
-
- epsilon = 1.0;
-
-
-
- class_state.OUTPUT_DEFINED = false;
- LUV_treshold = 0.1;
- }
- msImageProcessor::~msImageProcessor ( void )
- {
-
- if ( class_state.OUTPUT_DEFINED ) DestroyOutput();
- if ( regionList ) delete regionList;
- regionList = NULL;
-
- }
- void msImageProcessor::DefineImage ( byte *data_, imageType type, int height_, int width_ )
- {
-
-
- int dim;
- if ( type == COLOR )
- dim = 3;
- else
- dim = 1;
-
- int i;
- float *luv = new float [height_*width_*dim];
- if ( dim == 1 )
- {
- for ( i = 0; i < height_*width_; i++ )
- luv[i] = ( float ) ( data_[i] );
- }
- else
- {
- for ( i = 0; i < height_*width_; i++ )
- {
- RGBtoLUV ( &data_[dim*i], &luv[dim*i] );
- }
- }
-
- DefineLInput ( luv, height_, width_, dim );
-
-
- if ( !h )
- {
-
- kernelType k[2] = {Uniform, Uniform};
- int P[2] = {2, N};
- float tempH[2] = {1.0 , 1.0};
-
- DefineKernel ( k, tempH, P, 2 );
- }
-
- delete [] luv;
-
- return;
- }
- void msImageProcessor::DefineBgImage ( byte* data_, imageType type, int height_, int width_ )
- {
-
- int dim;
- if ( type == COLOR )
- dim = 3;
- else
- dim = 1;
-
- int i;
- float *luv = new float [height_*width_*dim];
- if ( dim == 1 )
- {
- for ( i = 0; i < height_*width_; i++ )
- luv[i] = ( float ) ( data_[i] );
- }
- else
- {
- for ( i = 0; i < height_*width_; i++ )
- RGBtoLUV ( &data_[dim*i], &luv[dim*i] );
- }
-
- DefineLInput ( luv, height_, width_, dim );
-
-
- if ( !h )
- {
-
- kernelType k[2] = {Uniform, Uniform};
- int P[2] = {2, N};
- float tempH[2] = {1.0 , 1.0};
-
- DefineKernel ( k, tempH, P, 2 );
- }
-
- delete [] luv;
-
- return;
- }
- void msImageProcessor::SetWeightMap ( float *wm, float eps )
- {
-
- SetLatticeWeightMap ( wm );
-
- if ( ( epsilon = eps ) < 0 )
- ErrorHandler ( ( char* ) "msImageProcessor", ( char* ) "SetWeightMap", ( char* ) "Threshold is negative." );
-
- return;
- }
- void msImageProcessor::RemoveWeightMap ( void )
- {
-
- RemoveLatticeWeightMap();
-
- epsilon = 0;
-
- return;
- }
- void msImageProcessor::Filter ( int sigmaS, float sigmaR, SpeedUpLevel speedUpLevel )
- {
-
-
-
-
-
-
-
-
- classConsistencyCheck ( N + 2, true );
- if ( ErrorStatus == EL_ERROR )
- return;
-
- if ( ( ErrorStatus = msSys.Progress ( ( float ) ( 0.0 ) ) ) == EL_HALT )
- {
- return;
- }
-
-
-
- if ( class_state.OUTPUT_DEFINED == false )
- {
- InitializeOutput();
-
- if ( ErrorStatus == EL_ERROR )
- return;
- }
-
-
- if ( ( ! ( modeTable = new unsigned char [L] ) ) || ( ! ( pointList = new int [L] ) ) )
- {
- ErrorHandler ( ( char* ) "msImageProcessor", ( char* ) "Allocate", ( char* ) "Not enough memory." );
- return;
- }
-
- #ifdef PROMPT
- double timer;
- msSys.StartTimer();
- #endif
-
-
- switch ( speedUpLevel )
- {
-
- case NO_SPEEDUP:
-
- NewNonOptimizedFilter ( ( float ) ( sigmaS ), sigmaR );
- break;
-
- case MED_SPEEDUP:
-
- NewOptimizedFilter1 ( ( float ) ( sigmaS ), sigmaR );
- break;
-
- case HIGH_SPEEDUP:
-
- NewOptimizedFilter2 ( ( float ) ( sigmaS ), sigmaR );
- break;
-
- }
-
-
- delete [] modeTable;
- delete [] pointList;
-
- modeTable = NULL;
- pointList = NULL;
- pointCount = 0;
-
-
-
- if ( ( ErrorStatus = msSys.Progress ( ( float ) ( 0.8 ) ) ) == EL_HALT )
- {
- DestroyOutput();
- return;
- }
-
-
-
-
- int i;
- for ( i = 0; i < L*N; i++ )
- {
- LUV_data[i] = msRawData[i];
- }
- #ifdef PROMPT
- timer = msSys.ElapsedTime();
- printf ( ( char* ) "(%6.2f sec)\nConnecting regions ...", timer );
- msSys.StartTimer();
- #endif
-
- Connect();
- #ifdef PROMPT
- timer = msSys.ElapsedTime();
- printf ( ( char* ) "done. (%6.2f seconds, numRegions = %6d)\n", timer, regionCount );
- msSys.StartTimer();
- #endif
-
- return;
- }
- void msImageProcessor::FuseRegions ( float sigmaS, int minRegion )
- {
-
-
-
-
-
-
-
-
- classConsistencyCheck ( N + 2, true );
- if ( ErrorStatus == EL_ERROR )
- return;
-
-
- if ( ( ErrorStatus = msSys.Progress ( ( float ) ( 0.8 ) ) ) == EL_HALT )
- {
- if ( class_state.OUTPUT_DEFINED ) DestroyOutput();
- return;
- }
-
-
- if ( ( h[1] = sigmaS ) <= 0 )
- {
- ErrorHandler ( ( char* ) "msImageProcessor", ( char* ) "FuseRegions", ( char* ) "The feature radius must be greater than or equal to zero." );
- return;
- }
-
-
- if ( ! ( class_state.OUTPUT_DEFINED ) )
- {
-
-
- InitializeOutput();
-
- if ( ErrorStatus == EL_ERROR )
- return;
-
-
-
- int i;
- for ( i = 0; i < L*N; i++ )
- {
- LUV_data[i] = data[i];
- }
- #ifdef PROMPT
- printf ( ( char* ) "Connecting regions ..." );
- msSys.StartTimer();
- #endif
-
- Connect();
-
- if ( ErrorStatus == EL_ERROR )
- return;
- #ifdef PROMPT
- double timer = msSys.ElapsedTime();
- printf ( ( char* ) "done. (%6.2f seconds, numRegions = %6d)\n", timer, regionCount );
- #endif
- }
-
-
- if ( ( ErrorStatus = msSys.Progress ( ( float ) ( 0.85 ) ) ) == EL_HALT )
- {
- DestroyOutput();
- return;
- }
- #ifdef PROMPT
- printf ( ( char* ) "Applying transitive closure..." );
- msSys.StartTimer();
- #endif
-
- visitTable = new unsigned char [L];
-
-
-
- rR2 = ( float ) ( h[1] * h[1] * 0.25 );
- TransitiveClosure();
- int oldRC = regionCount;
- int deltaRC, counter = 0;
- do {
- TransitiveClosure();
- deltaRC = oldRC - regionCount;
- oldRC = regionCount;
- counter++;
- } while ( ( deltaRC <= 0 ) && ( counter < 10 ) );
-
- delete [] visitTable;
- visitTable = NULL;
-
-
- if ( ( ErrorStatus = msSys.Progress ( ( float ) ( 1.0 ) ) ) == EL_HALT )
- {
- DestroyRAM();
- DestroyOutput();
- return;
- }
- #ifdef PROMPT
- double timer = msSys.ElapsedTime();
- printf ( ( char* ) "done. (%6.2f seconds, numRegions = %6d)\nPruning spurious regions ...", timer, regionCount );
- msSys.StartTimer();
- #endif
-
-
- Prune ( minRegion );
- #ifdef PROMPT
- timer = msSys.ElapsedTime();
- printf ( ( char* ) "done. (%6.2f seconds, numRegions = %6d)\n", timer, regionCount );
- msSys.StartTimer();
- #endif
-
-
- if ( ( ErrorStatus = msSys.Progress ( ( float ) ( 1.0 ) ) ) == EL_HALT )
- {
- DestroyRAM();
- DestroyOutput();
- return;
- }
-
- DestroyRAM();
-
- int i, j, label;
- for ( i = 0; i < L; i++ )
- {
- label = labels[i];
- for ( j = 0; j < N; j++ )
- {
- msRawData[N*i+j] = modes[N*label+j];
- }
- }
-
- return;
- }
- void msImageProcessor::Segment ( int sigmaS, float sigmaR, int minRegion, SpeedUpLevel speedUpLevel )
- {
-
- if ( ( !h ) || ( kp < 2 ) )
- {
- ErrorHandler ( ( char* ) "msImageProcessor", ( char* ) "Segment", ( char* ) "Kernel corrupt or undefined." );
- return;
- }
-
- Filter ( sigmaS, sigmaR, speedUpLevel );
-
- if ( ErrorStatus == EL_ERROR )
- return;
-
- if ( ErrorStatus == EL_HALT )
- return;
-
-
- if ( ( ErrorStatus = msSys.Progress ( ( float ) ( 0.85 ) ) ) == EL_HALT )
- {
- DestroyOutput();
- return;
- }
- #ifdef PROMPT
- printf ( ( char* ) "Applying transitive closure..." );
- msSys.StartTimer();
- #endif
-
- visitTable = new unsigned char [L];
-
-
-
- rR2 = ( float ) ( h[1] * h[1] * 0.25 );
- TransitiveClosure();
- int oldRC = regionCount;
- int deltaRC, counter = 0;
- do {
- TransitiveClosure();
- deltaRC = oldRC - regionCount;
- oldRC = regionCount;
- counter++;
- } while ( ( deltaRC <= 0 ) && ( counter < 10 ) );
-
- delete [] visitTable;
- visitTable = NULL;
-
-
- if ( ( ErrorStatus = msSys.Progress ( ( float ) ( 0.95 ) ) ) == EL_HALT )
- {
- DestroyRAM();
- DestroyOutput();
- return;
- }
- #ifdef PROMPT
- double timer = msSys.ElapsedTime();
- printf ( ( char* ) "done. (%6.2f seconds, numRegions = %6d).\nPruning spurious regions\t... ", timer, regionCount );
- msSys.StartTimer();
- #endif
-
-
- Prune ( minRegion );
- #ifdef PROMPT
- timer = msSys.ElapsedTime();
- printf ( ( char* ) "done. (%6.2f seconds, numRegions = %6d)\nPruning spurious regions ...", timer, regionCount );
- msSys.StartTimer();
- #endif
-
-
- if ( ( ErrorStatus = msSys.Progress ( 1.0 ) ) == EL_HALT )
- {
- DestroyRAM();
- DestroyOutput();
- return;
- }
-
- DestroyRAM();
-
- int j, i, label;
- for ( i = 0; i < L; i++ )
- {
- label = labels[i];
- for ( j = 0; j < N; j++ )
- {
- msRawData[N*i+j] = modes[N*label+j];
- }
- }
-
- return;
- }
- void msImageProcessor::RGBtoLUV ( byte *rgbVal, float *luvVal )
- {
-
- double x, y, z, L0, u_prime, v_prime, constant;
-
- x = XYZ[0][0] * rgbVal[0] + XYZ[0][1] * rgbVal[1] + XYZ[0][2] * rgbVal[2];
- y = XYZ[1][0] * rgbVal[0] + XYZ[1][1] * rgbVal[1] + XYZ[1][2] * rgbVal[2];
- z = XYZ[2][0] * rgbVal[0] + XYZ[2][1] * rgbVal[1] + XYZ[2][2] * rgbVal[2];
-
-
- L0 = y / ( 255.0 * Yn );
- if ( L0 > Lt )
- luvVal[0] = ( float ) ( 116.0 * ( pow ( L0, 1.0 / 3.0 ) ) - 16.0 );
- else
- luvVal[0] = ( float ) ( 903.3 * L0 );
-
- constant = x + 15 * y + 3 * z;
- if ( constant != 0 )
- {
- u_prime = ( 4 * x ) / constant;
- v_prime = ( 9 * y ) / constant;
- }
- else
- {
- u_prime = 4.0;
- v_prime = 9.0 / 15.0;
- }
-
- luvVal[1] = ( float ) ( 13 * luvVal[0] * ( u_prime - Un_prime ) );
- luvVal[2] = ( float ) ( 13 * luvVal[0] * ( v_prime - Vn_prime ) );
-
- return;
- }
- inline int my_round ( double in_x )
- {
- if ( in_x < 0 )
- return ( int ) ( in_x - 0.5 );
- else
- return ( int ) ( in_x + 0.5 );
- }
- void msImageProcessor::LUVtoRGB ( float *luvVal, byte *rgbVal )
- {
-
- int r, g, b;
- double x, y, z, u_prime, v_prime;
-
- if ( luvVal[0] < 0.1 )
- r = g = b = 0;
- else
- {
-
- if ( luvVal[0] < 8.0 )
- y = Yn * luvVal[0] / 903.3;
- else
- {
- y = ( luvVal[0] + 16.0 ) / 116.0;
- y *= Yn * y * y;
- }
- u_prime = luvVal[1] / ( 13 * luvVal[0] ) + Un_prime;
- v_prime = luvVal[2] / ( 13 * luvVal[0] ) + Vn_prime;
- x = 9 * u_prime * y / ( 4 * v_prime );
- z = ( 12 - 3 * u_prime - 20 * v_prime ) * y / ( 4 * v_prime );
-
-
- r = my_round ( ( RGB[0][0] * x + RGB[0][1] * y + RGB[0][2] * z ) * 255.0 );
- g = my_round ( ( RGB[1][0] * x + RGB[1][1] * y + RGB[1][2] * z ) * 255.0 );
- b = my_round ( ( RGB[2][0] * x + RGB[2][1] * y + RGB[2][2] * z ) * 255.0 );
-
- if ( r < 0 ) r = 0;
- if ( r > 255 ) r = 255;
- if ( g < 0 ) g = 0;
- if ( g > 255 ) g = 255;
- if ( b < 0 ) b = 0;
- if ( b > 255 ) b = 255;
- }
-
- rgbVal[0] = r;
- rgbVal[1] = g;
- rgbVal[2] = b;
-
- return;
- }
- void msImageProcessor::GetRawData ( float *outputImageData )
- {
-
- if ( !outputImageData )
- {
- ErrorHandler ( ( char* ) "msImageProcessor", ( char* ) "GetRawData", ( char* ) "Output image data buffer is NULL." );
- return;
- }
-
- int i;
- for ( i = 0; i < L*N; i++ )
- outputImageData[i] = msRawData[i];
-
- return;
- }
- void msImageProcessor::GetResults ( byte *outputImage )
- {
-
- if ( !outputImage )
- {
- ErrorHandler ( ( char* ) "msImageProcessor", ( char* ) "GetResults", ( char* ) "Output image buffer is NULL." );
- return;
- }
-
-
- if ( N == 1 )
- {
-
-
- int i, pxValue;
- for ( i = 0; i < L; i++ )
- {
-
- pxValue = ( int ) ( msRawData[i] + 0.5 );
-
- if ( pxValue < 0 )
- outputImage[i] = ( byte ) ( 0 );
- else if ( pxValue > 255 )
- outputImage[i] = ( byte ) ( 255 );
- else
- outputImage[i] = ( byte ) ( pxValue );
- }
- }
- else if ( N == 3 )
- {
-
-
- int i;
- for ( i = 0; i < L; i++ )
- LUVtoRGB ( &msRawData[N*i], &outputImage[N*i] );
- }
- else
-
- ErrorHandler ( ( char* ) "msImageProcessor", ( char* ) "GetResults", ( char* ) "Unknown image type. Try using MeanShift::GetRawData()." );
-
- return;
- }
- RegionList *msImageProcessor::GetBoundaries ( void )
- {
-
- if ( class_state.OUTPUT_DEFINED )
- DefineBoundaries();
-
- return regionList;
- }
- int msImageProcessor::GetRegions ( int **labels_out, float **modes_out, int **MPC_out )
- {
-
- if ( class_state.OUTPUT_DEFINED == false )
- return -1;
-
-
- int *labels_ = *labels_out, *MPC_out_ = *MPC_out;
- float *modes_ = *modes_out;
- if ( ! ( labels_ = new int [L] ) )
- {
- ErrorHandler ( ( char* ) "msImageProcessor", ( char* ) "GetRegions", ( char* ) "Not enough memory." );
- return -1;
- }
- if ( ! ( modes_ = new float [regionCount*N] ) )
- {
- ErrorHandler ( ( char* ) "msImageProcessor", ( char* ) "GetRegions", ( char* ) "Not enough memory." );
- return -1;
- }
- if ( ! ( MPC_out_ = new int [regionCount] ) )
- {
- ErrorHandler ( ( char* ) "msImageProcessor", ( char* ) "GetRegions", ( char* ) "Not enough memory." );
- return -1;
- }
-
- int i;
- for ( i = 0; i < L; i++ )
- labels_[i] = labels[i];
-
-
- for ( i = 0; i < regionCount*N; i++ )
- modes_[i] = modes[i];
- for ( i = 0; i < regionCount; i++ )
- MPC_out_[i] = modePointCounts[i];
-
- return regionCount;
- }
- void msImageProcessor::NonOptimizedFilter ( float sigmaS, float sigmaR )
- {
-
- int iterationCount, i, j;
- double mvAbs;
-
-
- if ( !height )
- {
- ErrorHandler ( ( char* ) "msImageProcessor", ( char* ) "LFilter", ( char* ) "Lattice height and width are undefined." );
- return;
- }
-
- if ( ( ( h[0] = sigmaS ) <= 0 ) || ( ( h[1] = sigmaR ) <= 0 ) )
- {
- ErrorHandler ( ( char* ) "msImageProcessor", ( char* ) "Segment", ( char* ) "sigmaS and/or sigmaR is zero or negative." );
- return;
- }
-
- int lN = N + 2;
-
-
-
- double *yk = new double [lN];
-
- double *Mh = new double [lN];
-
- #ifdef PROMPT
- printf ( ( char* ) "done.\nApplying mean shift (Using Lattice)... " );
- #ifdef SHOW_PROGRESS
- printf ( ( char* ) "\n 0%%" );
- #endif
- #endif
- for ( i = 0; i < L; i++ )
- {
-
-
-
- yk[0] = i % width;
- yk[1] = i / width;
- for ( j = 0; j < N; j++ )
- yk[j+2] = data[N*i+j];
-
- LatticeMSVector ( Mh, yk );
-
- mvAbs = 0;
- for ( j = 0; j < lN; j++ )
- mvAbs += Mh[j] * Mh[j];
-
-
-
-
-
- iterationCount = 1;
- while ( ( mvAbs >= EPSILON2 ) && ( iterationCount < LIMIT ) )
- {
-
- for ( j = 0; j < lN; j++ )
- yk[j] += Mh[j];
-
-
- LatticeMSVector ( Mh, yk );
-
- mvAbs = 0;
- for ( j = 0; j < lN; j++ )
- mvAbs += Mh[j] * Mh[j];
-
- iterationCount++;
- }
-
- for ( j = 0; j < lN; j++ )
- yk[j] += Mh[j];
-
- for ( j = 0; j < N; j++ )
- msRawData[N*i+j] = ( float ) ( yk[j+2] );
-
- #ifdef SHOW_PROGRESS
- percent_complete = ( float ) ( i / ( float ) ( L ) ) * 100;
- printf ( ( char* ) "\r%2d%%", ( int ) ( percent_complete + 0.5 ) );
- #endif
-
- if ( ( i % PROGRESS_RATE == 0 ) && ( ( ErrorStatus = msSys.Progress ( ( float ) ( i / ( float ) ( L ) ) * ( float ) ( 0.8 ) ) ) ) == EL_HALT )
- break;
- }
-
- #ifdef PROMPT
- #ifdef SHOW_PROGRESS
- printf ( ( char* ) "\r" );
- #endif
- printf ( ( char* ) "done." );
- #endif
-
- delete [] yk;
- delete [] Mh;
-
- return;
- }
- void msImageProcessor::OptimizedFilter1 ( float sigmaS, float sigmaR )
- {
-
- int iterationCount, i, j, k, s, p, modeCandidateX, modeCandidateY, modeCandidate_i;
- float *modeCandidatePoint;
- double mvAbs, diff, el;
-
-
- if ( !height )
- {
- ErrorHandler ( ( char* ) "msImageProcessor", ( char* ) "LFilter", ( char* ) "Lattice height and width are undefined." );
- return;
- }
-
- if ( ( ( h[0] = sigmaS ) <= 0 ) || ( ( h[1] = sigmaR ) <= 0 ) )
- {
- ErrorHandler ( ( char* ) "msImageProcessor", ( char* ) "Segment", ( char* ) "sigmaS and/or sigmaR is zero or negative." );
- return;
- }
-
- int lN = N + 2;
-
-
-
- double *yk = new double [lN];
-
- double *Mh = new double [lN];
-
- memset ( modeTable, 0, width*height );
-
-
- modeCandidatePoint = new float [N];
-
- #ifdef PROMPT
- printf ( ( char* ) "done.\nApplying mean shift (Using Lattice) ... " );
- #ifdef SHOW_PROGRESS
- printf ( ( char* ) "\n 0%%" );
- #endif
- #endif
- for ( i = 0; i < L; i++ )
- {
-
-
-
- if ( modeTable[i] == 1 )
- continue;
-
- pointCount = 0;
-
-
-
- yk[0] = i % width;
- yk[1] = i / width;
- for ( j = 0; j < N; j++ )
- yk[j+2] = data[N*i+j];
-
- LatticeMSVector ( Mh, yk );
-
- mvAbs = 0;
- for ( j = 0; j < lN; j++ )
- mvAbs += Mh[j] * Mh[j];
-
-
-
-
-
- iterationCount = 1;
- while ( ( mvAbs >= EPSILON2 ) && ( iterationCount < LIMIT ) )
- {
-
- for ( j = 0; j < lN; j++ )
- yk[j] += Mh[j];
-
-
-
- modeCandidateX = ( int ) ( yk[0] + 0.5 );
- modeCandidateY = ( int ) ( yk[1] + 0.5 );
- modeCandidate_i = modeCandidateY * width + modeCandidateX;
-
-
-
-
-
-
-
-
-
-
-
- if ( ( modeTable[modeCandidate_i] != 2 ) && ( modeCandidate_i != i ) )
- {
-
-
-
- for ( j = 0; j < N; j++ )
- modeCandidatePoint[j] = data[N*modeCandidate_i + j];
-
- k = 1;
- s = 0;
- diff = 0;
- while ( ( diff < TC_DIST_FACTOR ) && ( k < kp ) )
- {
- diff = 0;
- for ( p = 0; p < P[k]; p++ )
- {
- el = ( modeCandidatePoint[p+s] - yk[p+s+2] ) / h[k];
- diff += el * el;
- }
- s += P[k];
- k++;
- }
-
-
-
-
- if ( diff < TC_DIST_FACTOR )
- {
-
-
-
-
- if ( modeTable[modeCandidate_i] == 0 )
- {
-
-
- pointList[pointCount++] = modeCandidate_i;
- modeTable[modeCandidate_i] = 2;
- } else
- {
-
-
-
-
-
- for ( j = 0; j < N; j++ )
- yk[j+2] = msRawData[modeCandidate_i*N+j];
-
-
-
- modeTable[i] = 1;
-
-
- mvAbs = -1;
-
- break;
- }
- }
- }
-
-
- LatticeMSVector ( Mh, yk );
-
- mvAbs = 0;
- for ( j = 0; j < lN; j++ )
- mvAbs += Mh[j] * Mh[j];
-
- iterationCount++;
- }
-
-
- if ( mvAbs >= 0 )
- {
-
- for ( j = 0; j < lN; j++ )
- yk[j] += Mh[j];
-
-
-
- modeTable[i] = 1;
- }
-
-
-
- for ( j = 0; j < pointCount; j++ )
- {
-
-
- modeCandidate_i = pointList[j];
-
- modeTable[modeCandidate_i] = 1;
-
- for ( k = 0; k < N; k++ )
- msRawData[N*modeCandidate_i+k] = ( float ) ( yk[k+2] );
- }
-
- for ( j = 0; j < N; j++ )
- msRawData[N*i+j] = ( float ) ( yk[j+2] );
-
- #ifdef SHOW_PROGRESS
- percent_complete = ( float ) ( i / ( float ) ( L ) ) * 100;
- printf ( ( char* ) "\r%2d%%", ( int ) ( percent_complete + 0.5 ) );
- #endif
-
- if ( ( i % PROGRESS_RATE == 0 ) && ( ( ErrorStatus = msSys.Progress ( ( float ) ( i / ( float ) ( L ) ) * ( float ) ( 0.8 ) ) ) ) == EL_HALT )
- break;
- }
-
- #ifdef PROMPT
- #ifdef SHOW_PROGRESS
- printf ( ( char* ) "\r" );
- #endif
- printf ( ( char* ) "done." );
- #endif
-
- delete [] modeCandidatePoint;
- delete [] yk;
- delete [] Mh;
-
- return;
- }
- void msImageProcessor::OptimizedFilter2 ( float sigmaS, float sigmaR )
- {
-
- if ( !weightMap )
- {
- weightMap = new float [L];
- int i;
- for ( i = 0; i < L; i++ )
- weightMap[i] = 0;
- }
-
- int iterationCount, i, j, k, s, p, modeCandidateX, modeCandidateY, modeCandidate_i;
- float *modeCandidatePoint;
- double mvAbs, diff, el;
-
-
- if ( !height )
- {
- ErrorHandler ( ( char* ) "msImageProcessor", ( char* ) "LFilter", ( char* ) "Lattice height and width are undefined." );
- return;
- }
-
- if ( ( ( h[0] = sigmaS ) <= 0 ) || ( ( h[1] = sigmaR ) <= 0 ) )
- {
- ErrorHandler ( ( char* ) "msImageProcessor", ( char* ) "Segment", ( char* ) "sigmaS and/or sigmaR is zero or negative." );
- return;
- }
-
- int lN = N + 2;
-
-
-
- double *yk = new double [lN];
-
- double *Mh = new double [lN];
-
- memset ( modeTable, 0, width*height );
-
-
- modeCandidatePoint = new float [N];
-
- #ifdef PROMPT
- printf ( ( char* ) "done.\nApplying mean shift (Using Lattice)... " );
- #ifdef SHOW_PROGRESS
- printf ( ( char* ) "\n 0%%" );
- #endif
- #endif
- for ( i = 0; i < L; i++ )
- {
-
-
-
- if ( modeTable[i] == 1 )
- continue;
-
- pointCount = 0;
-
-
-
- yk[0] = i % width;
- yk[1] = i / width;
- for ( j = 0; j < N; j++ )
- yk[j+2] = data[N*i+j];
-
- OptLatticeMSVector ( Mh, yk );
-
- mvAbs = 0;
- for ( j = 0; j < lN; j++ )
- mvAbs += Mh[j] * Mh[j];
-
-
-
-
-
- iterationCount = 1;
- while ( ( mvAbs >= EPSILON2 ) && ( iterationCount < LIMIT ) )
- {
-
- for ( j = 0; j < lN; j++ )
- yk[j] += Mh[j];
-
-
-
- modeCandidateX = ( int ) ( yk[0] + 0.5 );
- modeCandidateY = ( int ) ( yk[1] + 0.5 );
- modeCandidate_i = modeCandidateY * width + modeCandidateX;
-
-
-
-
-
-
-
-
-
-
-
- if ( ( modeTable[modeCandidate_i] != 2 ) && ( modeCandidate_i != i ) )
- {
-
-
-
- for ( j = 0; j < N; j++ )
- modeCandidatePoint[j] = data[N*modeCandidate_i + j];
-
- k = 1;
- s = 0;
- diff = 0;
- while ( ( diff < TC_DIST_FACTOR ) && ( k < kp ) )
- {
- diff = 0;
- for ( p = 0; p < P[k]; p++ )
- {
- el = ( modeCandidatePoint[p+s] - yk[p+s+2] ) / h[k];
- diff += el * el;
- }
- s += P[k];
- k++;
- }
-
-
-
-
- if ( diff < TC_DIST_FACTOR )
- {
-
-
-
-
- if ( modeTable[modeCandidate_i] == 0 )
- {
-
-
- pointList[pointCount++] = modeCandidate_i;
- modeTable[modeCandidate_i] = 2;
- } else
- {
-
-
-
-
-
- for ( j = 0; j < N; j++ )
- yk[j+2] = msRawData[modeCandidate_i*N+j];
-
-
-
- modeTable[i] = 1;
-
-
- mvAbs = -1;
-
- break;
- }
- }
- }
-
-
- OptLatticeMSVector ( Mh, yk );
-
- mvAbs = 0;
- for ( j = 0; j < lN; j++ )
- mvAbs += Mh[j] * Mh[j];
-
- iterationCount++;
- }
-
-
-
- if ( mvAbs >= 0 )
- {
-
- for ( j = 0; j < lN; j++ )
- yk[j] += Mh[j];
-
-
-
- modeTable[i] = 1;
- }
-
-
-
- for ( j = 0; j < pointCount; j++ )
- {
-
-
- modeCandidate_i = pointList[j];
-
- modeTable[modeCandidate_i] = 1;
-
- for ( k = 0; k < N; k++ )
- msRawData[N*modeCandidate_i+k] = ( float ) ( yk[k+2] );
- }
-
- for ( j = 0; j < N; j++ )
- msRawData[N*i+j] = ( float ) ( yk[j+2] );
-
- #ifdef SHOW_PROGRESS
- percent_complete = ( float ) ( i / ( float ) ( L ) ) * 100;
- printf ( ( char* ) "\r%2d%%", ( int ) ( percent_complete + 0.5 ) );
- #endif
-
- if ( ( i % PROGRESS_RATE == 0 ) && ( ( ErrorStatus = msSys.Progress ( ( float ) ( i / ( float ) ( L ) ) * ( float ) ( 0.8 ) ) ) ) == EL_HALT )
- break;
- }
-
- #ifdef PROMPT
- #ifdef SHOW_PROGRESS
- printf ( ( char* ) "\r" );
- #endif
- printf ( ( char* ) "done." );
- #endif
-
- delete [] modeCandidatePoint;
- delete [] yk;
- delete [] Mh;
-
- return;
- }
- void msImageProcessor::Connect ( void )
- {
-
- neigh[0] = 1;
- neigh[1] = 1 - width;
- neigh[2] = -width;
- neigh[3] = - ( 1 + width );
- neigh[4] = -1;
- neigh[5] = width - 1;
- neigh[6] = width;
- neigh[7] = width + 1;
-
- int i;
- for ( i = 0; i < width*height; i++ )
- {
- labels[i] = -1;
- modePointCounts[i] = 0;
- }
-
- int k, label = -1;
- for ( i = 0; i < height*width; i++ )
- {
-
- if ( labels[i] < 0 )
- {
-
- labels[i] = ++label;
-
- for ( k = 0; k < N; k++ )
- modes[ ( N*label ) +k] = LUV_data[ ( N*i ) +k];
-
-
- Fill ( i, label );
- }
- }
-
- regionCount = label + 1;
-
- return;
- }
- void msImageProcessor::Fill ( int regionLoc, int label )
- {
-
- int i, k, neighLoc, neighborsFound, imageSize = width * height;
-
-
-
- int index = 0;
- indexTable[0] = regionLoc;
-
-
- modePointCounts[label]++;
- while ( true )
- {
-
- neighborsFound = 0;
-
-
-
- for ( i = 0; i < 8; i++ )
- {
-
-
-
- neighLoc = regionLoc + neigh[i];
- if ( ( neighLoc >= 0 ) && ( neighLoc < imageSize ) && ( labels[neighLoc] < 0 ) )
- {
- for ( k = 0; k < N; k++ )
- {
- if ( fabs ( LUV_data[ ( regionLoc*N ) +k] - LUV_data[ ( neighLoc*N ) +k] ) >= LUV_treshold )
- break;
- }
-
-
-
- if ( k == N )
- {
-
- labels[neighLoc] = label;
-
- modePointCounts[label]++;
-
- indexTable[++index] = neighLoc;
-
-
- neighborsFound = 1;
- }
- }
- }
-
-
-
- if ( neighborsFound )
- regionLoc = indexTable[index];
- else if ( index > 1 )
- regionLoc = indexTable[--index];
- else
- break;
- }
-
- return;
- }
- void msImageProcessor::BuildRAM ( void )
- {
-
- if ( ( !raList ) && ( ( ! ( raList = new RAList [regionCount] ) ) || ( ! ( raPool = new RAList [NODE_MULTIPLE*regionCount] ) ) ) )
- {
- ErrorHandler ( ( char* ) "msImageProcessor", ( char* ) "Allocate", ( char* ) "Not enough memory." );
- return;
- }
-
- int i;
- for ( i = 0; i < regionCount; i++ )
- {
- raList[i].edgeStrength = 0;
- raList[i].edgePixelCount = 0;
- raList[i].label = i;
- raList[i].next = NULL;
- }
-
- freeRAList = raPool;
- for ( i = 0; i < NODE_MULTIPLE*regionCount - 1; i++ )
- {
- raPool[i].edgeStrength = 0;
- raPool[i].edgePixelCount = 0;
- raPool[i].next = &raPool[i+1];
- }
- raPool[NODE_MULTIPLE*regionCount-1].next = NULL;
-
-
-
-
-
- int j, curLabel, rightLabel, bottomLabel, exists;
- RAList *raNode1, *raNode2, *oldRAFreeList;
- for ( i = 0; i < height - 1; i++ )
- {
-
-
- for ( j = 0; j < width - 1; j++ )
- {
-
- curLabel = labels[i*width+j ];
- rightLabel = labels[i*width+j+1 ];
- bottomLabel = labels[ ( i+1 ) *width+j];
-
-
-
-
- if ( curLabel != rightLabel )
- {
-
-
- raNode1 = freeRAList;
- raNode2 = freeRAList->next;
-
-
-
- oldRAFreeList = freeRAList;
-
- freeRAList = freeRAList->next->next;
-
- raNode1->label = curLabel;
- raNode2->label = rightLabel;
-
- exists = 0;
- raList[curLabel ].Insert ( raNode2 );
- exists = raList[rightLabel].Insert ( raNode1 );
-
-
-
- if ( exists )
- freeRAList = oldRAFreeList;
- }
-
-
-
-
- if ( curLabel != bottomLabel )
- {
-
-
- raNode1 = freeRAList;
- raNode2 = freeRAList->next;
-
-
-
- oldRAFreeList = freeRAList;
-
- freeRAList = freeRAList->next->next;
-
- raNode1->label = curLabel;
- raNode2->label = bottomLabel;
-
- exists = 0;
- raList[curLabel ].Insert ( raNode2 );
- exists = raList[bottomLabel].Insert ( raNode1 );
-
-
-
- if ( exists )
- freeRAList = oldRAFreeList;
- }
- }
-
-
-
- curLabel = labels[i*width+j ];
- bottomLabel = labels[ ( i+1 ) *width+j];
-
-
-
-
- if ( curLabel != bottomLabel )
- {
-
-
- raNode1 = freeRAList;
- raNode2 = freeRAList->next;
-
-
-
- oldRAFreeList = freeRAList;
-
- freeRAList = freeRAList->next->next;
-
- raNode1->label = curLabel;
- raNode2->label = bottomLabel;
-
- exists = 0;
- raList[curLabel ].Insert ( raNode2 );
- exists = raList[bottomLabel].Insert ( raNode1 );
-
-
-
- if ( exists )
- freeRAList = oldRAFreeList;
- }
- }
-
-
-
- for ( j = 0; j < width - 1; j++ )
- {
-
- curLabel = labels[i*width+j ];
- rightLabel = labels[i*width+j+1 ];
-
-
-
-
- if ( curLabel != rightLabel )
- {
-
-
- raNode1 = freeRAList;
- raNode2 = freeRAList->next;
-
-
-
- oldRAFreeList = freeRAList;
-
- freeRAList = freeRAList->next->next;
-
- raNode1->label = curLabel;
- raNode2->label = rightLabel;
-
- exists = 0;
- raList[curLabel ].Insert ( raNode2 );
- exists = raList[rightLabel].Insert ( raNode1 );
-
-
-
- if ( exists )
- freeRAList = oldRAFreeList;
- }
- }
-
- return;
- }
- void msImageProcessor::DestroyRAM ( void )
- {
-
- if ( raList ) delete [] raList;
- if ( raPool ) delete [] raPool;
-
- raList = NULL;
- freeRAList = NULL;
- raPool = NULL;
-
- return;
- }
- void msImageProcessor::TransitiveClosure ( void )
- {
-
-
-
- BuildRAM();
-
-
-
- if ( weightMapDefined ) ComputeEdgeStrengths();
-
-
-
-
-
-
-
-
-
- int i, iCanEl, neighCanEl;
- float threshold;
- RAList *neighbor;
- for ( i = 0; i < regionCount; i++ )
- {
-
-
- neighbor = raList[i].next;
-
-
- if ( epsilon > raList[i].edgeStrength )
- threshold = epsilon;
- else
- threshold = raList[i].edgeStrength;
-
-
-
- while ( neighbor )
- {
-
- if ( ( InWindow ( i, neighbor->label ) ) && ( neighbor->edgeStrength < epsilon ) )
- {
-
-
-
- iCanEl = i;
- while ( raList[iCanEl].label != iCanEl )
- iCanEl = raList[iCanEl].label;
-
- neighCanEl = neighbor->label;
- while ( raList[neighCanEl].label != neighCanEl )
- neighCanEl = raList[neighCanEl].label;
-
-
-
- if ( iCanEl < neighCanEl )
- raList[neighCanEl].label = iCanEl;
- else
- {
-
-
- raList[raList[iCanEl].label].label = neighCanEl;
-
- raList[iCanEl].label = neighCanEl;
- }
- }
-
- neighbor = neighbor->next;
- }
- }
-
-
- for ( i = 0; i < regionCount; i++ )
- {
- iCanEl = i;
- while ( raList[iCanEl].label != iCanEl )
- iCanEl = raList[iCanEl].label;
- raList[i].label = iCanEl;
- }
-
-
-
-
-
-
- float *modes_buffer = new float [N*regionCount];
- int *MPC_buffer = new int [regionCount];
-
- for ( i = 0; i < regionCount; i++ )
- MPC_buffer[i] = 0;
- for ( i = 0; i < N*regionCount; i++ )
- modes_buffer[i] = 0;
-
-
- int k, iMPC;
- for ( i = 0; i < regionCount; i++ )
- {
-
- iCanEl = raList[i].label;
-
- iMPC = modePointCounts[i];
-
- for ( k = 0; k < N; k++ )
- modes_buffer[ ( N*iCanEl ) +k] += iMPC * modes[ ( N*i ) +k];
-
- MPC_buffer[iCanEl] += iMPC;
- }
-
-
-
-
-
-
-
- int *label_buffer = new int [regionCount];
-
- for ( i = 0; i < regionCount; i++ )
- label_buffer[i] = -1;
-
- int label = -1;
- for ( i = 0; i < regionCount; i++ )
- {
-
- iCanEl = raList[i].label;
- if ( label_buffer[iCanEl] < 0 )
- {
-
-
- label_buffer[iCanEl] = ++label;
-
- iMPC = MPC_buffer[iCanEl];
- for ( k = 0; k < N; k++ )
- modes[ ( N*label ) +k] = ( modes_buffer[ ( N*iCanEl ) +k] ) / ( iMPC );
-
-
- modePointCounts[label] = MPC_buffer[iCanEl];
- }
- }
-
-
- regionCount = label + 1;
-
-
-
- for ( i = 0; i < height*width; i++ )
- labels[i] = label_buffer[raList[labels[i]].label];
-
- delete [] modes_buffer;
- delete [] MPC_buffer;
- delete [] label_buffer;
-
- return;
- }
- void msImageProcessor::ComputeEdgeStrengths ( void )
- {
-
-
-
-
- memset ( visitTable, 0, L*sizeof ( unsigned char ) );
-
-
- int x, y, dp, curLabel, rightLabel, bottomLabel;
- RAList *curRegion;
- for ( y = 1; y < height - 1; y++ )
- {
- for ( x = 1; x < width - 1; x++ )
- {
-
- dp = y * width + x;
-
- curLabel = labels[dp ];
- rightLabel = labels[dp+1 ];
- bottomLabel = labels[dp+width];
-
-
-
-
- if ( curLabel != rightLabel )
- {
-
- curRegion = &raList[curLabel];
- while ( ( curRegion ) && ( curRegion->label != rightLabel ) )
- curRegion = curRegion->next;
-
- assert ( curRegion );
-
- curRegion->edgeStrength += weightMap[dp] + weightMap[dp+1];
- curRegion->edgePixelCount += 2;
- }
- if ( curLabel != bottomLabel )
- {
-
- curRegion = &raList[curLabel];
- while ( ( curRegion ) && ( curRegion->label != bottomLabel ) )
- curRegion = curRegion->next;
-
- assert ( curRegion );
-
- if ( curLabel == rightLabel )
- {
- curRegion->edgeStrength += weightMap[dp] + weightMap[dp+width];
- curRegion->edgePixelCount += 2;
- }
- else
- {
- curRegion->edgeStrength += weightMap[dp+width];
- curRegion->edgePixelCount += 1;
- }
- }
- }
- }
-
- RAList *neighborRegion;
- float edgeStrength;
- int edgePixelCount;
- for ( x = 0; x < regionCount; x++ )
- {
-
- curRegion = &raList[x];
- curRegion = curRegion->next;
- while ( curRegion )
- {
-
-
-
-
-
-
- curLabel = curRegion->label;
- if ( curLabel > x )
- {
-
-
- neighborRegion = &raList[curLabel];
- while ( ( neighborRegion ) && ( neighborRegion->label != x ) )
- neighborRegion = neighborRegion->next;
-
- assert ( neighborRegion );
-
-
- if ( ( edgePixelCount = curRegion->edgePixelCount + neighborRegion->edgePixelCount ) != 0 )
- {
-
- edgeStrength = curRegion->edgeStrength + neighborRegion->edgeStrength;
- edgeStrength /= edgePixelCount;
-
- curRegion->edgeStrength = neighborRegion->edgeStrength = edgeStrength;
- curRegion->edgePixelCount = neighborRegion->edgePixelCount = edgePixelCount;
- }
- }
-
-
- curRegion = curRegion->next;
- }
- }
-
-
- int numNeighbors;
- for ( x = 0; x < regionCount; x++ )
- {
-
-
- curRegion = &raList[x];
- curRegion = curRegion->next;
- edgeStrength = 0;
- numNeighbors = 0;
- while ( curRegion )
- {
- numNeighbors++;
- edgeStrength += curRegion->edgeStrength;
- curRegion = curRegion->next;
- }
-
-
- if ( numNeighbors ) edgeStrength /= numNeighbors;
-
-
- raList[x].edgeStrength = edgeStrength;
- }
-
-
-
- return;
- }
- void msImageProcessor::Prune ( int minRegion )
- {
-
-
- float *modes_buffer = new float [N*regionCount];
- int *MPC_buffer = new int [regionCount];
-
- int *label_buffer = new int [regionCount];
-
- int i, k, candidate, iCanEl, neighCanEl, iMPC, label, oldRegionCount, minRegionCount;
- double minSqDistance, neighborDistance;
- RAList *neighbor;
-
-
- do
- {
-
- minRegionCount = 0;
-
-
-
- BuildRAM();
-
-
-
-
-
-
-
-
- for ( i = 0; i < regionCount; i++ )
- {
-
-
-
-
-
-
-
-
-
- if ( modePointCounts[i] < minRegion )
- {
-
-
- minRegionCount++;
-
-
- neighbor = raList[i].next;
-
-
- candidate = neighbor->label;
- minSqDistance = SqDistance ( i, candidate );
-
-
- neighbor = neighbor->next;
- while ( neighbor )
- {
-
-
- neighborDistance = SqDistance ( i, neighbor->label );
-
-
-
- if ( neighborDistance < minSqDistance )
- {
- minSqDistance = neighborDistance;
- candidate = neighbor->label;
- }
-
- neighbor = neighbor->next;
- }
-
-
- iCanEl = i;
- while ( raList[iCanEl].label != iCanEl )
- iCanEl = raList[iCanEl].label;
-
- neighCanEl = candidate;
- while ( raList[neighCanEl].label != neighCanEl )
- neighCanEl = raList[neighCanEl].label;
-
-
-
- if ( iCanEl < neighCanEl )
- raList[neighCanEl].label = iCanEl;
- else
- {
-
-
- raList[raList[iCanEl].label].label = neighCanEl;
-
- raList[iCanEl].label = neighCanEl;
- }
- }
- }
-
-
- for ( i = 0; i < regionCount; i++ )
- {
- iCanEl = i;
- while ( raList[iCanEl].label != iCanEl )
- iCanEl = raList[iCanEl].label;
- raList[i].label = iCanEl;
- }
-
-
-
-
-
- for ( i = 0; i < regionCount; i++ )
- MPC_buffer[i] = 0;
- for ( i = 0; i < N*regionCount; i++ )
- modes_buffer[i] = 0;
-
-
- for ( i = 0; i < regionCount; i++ )
- {
-
- iCanEl = raList[i].label;
-
- iMPC = modePointCounts[i];
-
- for ( k = 0; k < N; k++ )
- modes_buffer[ ( N*iCanEl ) +k] += iMPC * modes[ ( N*i ) +k];
-
- MPC_buffer[iCanEl] += iMPC;
- }
-
-
-
-
-
-
-
- for ( i = 0; i < regionCount; i++ )
- label_buffer[i] = -1;
-
- label = -1;
- for ( i = 0; i < regionCount; i++ )
- {
-
- iCanEl = raList[i].label;
- if ( label_buffer[iCanEl] < 0 )
- {
-
-
- label_buffer[iCanEl] = ++label;
-
- iMPC = MPC_buffer[iCanEl];
- for ( k = 0; k < N; k++ )
- modes[ ( N*label ) +k] = ( modes_buffer[ ( N*iCanEl ) +k] ) / ( iMPC );
-
-
- modePointCounts[label] = MPC_buffer[iCanEl];
- }
- }
-
- oldRegionCount = regionCount;
- regionCount = label + 1;
-
-
-
- for ( i = 0; i < height*width; i++ )
- labels[i] = label_buffer[raList[labels[i]].label];
- } while ( minRegionCount > 0 );
-
- delete [] modes_buffer;
- delete [] MPC_buffer;
- delete [] label_buffer;
-
- return;
- }
- void msImageProcessor::DefineBoundaries ( void )
- {
-
- int *boundaryMap, *boundaryCount;
- if ( ( ! ( boundaryMap = new int [L] ) ) || ( ! ( boundaryCount = new int [regionCount] ) ) )
- ErrorHandler ( ( char* ) "msImageProcessor", ( char* ) "DefineBoundaries", ( char* ) "Not enough memory." );
-
- int i;
- for ( i = 0; i < L; i++ )
- boundaryMap[i] = -1;
- for ( i = 0; i < regionCount; i++ )
- boundaryCount[i] = 0;
-
-
-
- int totalBoundaryCount = 0;
-
-
-
-
-
-
- int j, label, dataPoint;
-
- for ( i = 0; i < width; i++ )
- {
- boundaryMap[i] = label = labels[i];
- boundaryCount[label]++;
- totalBoundaryCount++;
- }
-
-
- for ( i = 1; i < height - 1; i++ )
- {
-
- dataPoint = i * width;
- boundaryMap[dataPoint] = label = labels[dataPoint];
- boundaryCount[label]++;
- totalBoundaryCount++;
- for ( j = 1; j < width - 1; j++ )
- {
-
-
- dataPoint = i * width + j;
-
-
- label = labels[dataPoint];
- if ( ( label != labels[dataPoint-1] ) || ( label != labels[dataPoint+1] ) ||
- ( label != labels[dataPoint-width] ) || ( label != labels[dataPoint+width] ) )
- {
- boundaryMap[dataPoint] = label = labels[dataPoint];
- boundaryCount[label]++;
- totalBoundaryCount++;
- }
- }
-
- dataPoint = ( i + 1 ) * width - 1;
- boundaryMap[dataPoint] = label = labels[dataPoint];
- boundaryCount[label]++;
- totalBoundaryCount++;
- }
-
- register int start = ( height - 1 ) * width, stop = height * width;
- for ( i = start; i < stop; i++ )
- {
- boundaryMap[i] = label = labels[i];
- boundaryCount[label]++;
- totalBoundaryCount++;
- }
-
-
-
-
-
-
- int *boundaryBuffer = new int [totalBoundaryCount], *boundaryIndex = new int [regionCount];
-
- int counter = 0;
- for ( i = 0; i < regionCount; i++ )
- {
- boundaryIndex[i] = counter;
- counter += boundaryCount[i];
- }
-
-
- for ( i = 0; i < L; i++ )
- {
-
-
- if ( ( label = boundaryMap[i] ) >= 0 )
- {
- boundaryBuffer[boundaryIndex[label]] = i;
- boundaryIndex[label]++;
- }
- }
-
-
-
-
-
-
-
- if ( regionList ) delete regionList;
-
- if ( ! ( regionList = new RegionList ( regionCount, totalBoundaryCount, N ) ) )
- ErrorHandler ( ( char* ) "msImageProcessor", ( char* ) "DefineBoundaries", ( char* ) "Not enough memory." );
-
-
- counter = 0;
- for ( i = 0; i < regionCount; i++ )
- {
- regionList->AddRegion ( i, boundaryCount[i], &boundaryBuffer[counter] );
- counter += boundaryCount[i];
- }
-
-
-
- delete [] boundaryMap;
- delete [] boundaryCount;
- delete [] boundaryBuffer;
- delete [] boundaryIndex;
-
- return;
- }
- bool msImageProcessor::InWindow ( int mode1, int mode2 )
- {
- int k = 1, s = 0, p;
- double diff = 0, el;
- while ( ( diff < 0.25 ) && ( k != kp ) )
- {
-
- diff = 0;
- for ( p = 0; p < P[k]; p++ )
- {
- el = ( modes[mode1*N+p+s] - modes[mode2*N+p+s] ) / ( h[k] * offset[k] );
- if ( ( !p ) && ( k == 1 ) && ( modes[mode1*N] > 80 ) )
- diff += 4 * el * el;
- else
- diff += el * el;
- }
-
- s += P[k];
- k++;
- }
- return ( bool ) ( diff < 0.25 );
- }
- float msImageProcessor::SqDistance ( int mode1, int mode2 )
- {
- int k = 1, s = 0, p;
- float dist = 0, el;
- for ( k = 1; k < kp; k++ )
- {
-
- for ( p = 0; p < P[k]; p++ )
- {
- el = ( modes[mode1*N+p+s] - modes[mode2*N+p+s] ) / ( h[k] * offset[k] );
- dist += el * el;
- }
-
- s += P[k];
- k++;
- }
-
-
- return dist;
- }
- void msImageProcessor::InitializeOutput ( void )
- {
-
- DestroyOutput();
-
- if ( ! ( msRawData = new float [L*N] ) )
- {
- ErrorHandler ( ( char* ) "msImageProcessor", ( char* ) "Allocate", ( char* ) "Not enough memory." );
- return;
- }
-
- if ( ( ! ( modes = new float [L* ( N+2 ) ] ) ) || ( ! ( labels = new int [L] ) ) || ( ! ( modePointCounts = new int [L] ) ) || ( ! ( indexTable = new int [L] ) ) )
- {
- ErrorHandler ( ( char* ) "msImageProcessor", ( char* ) "Allocate", ( char* ) "Not enough memory" );
- return;
- }
-
-
- if ( ! ( LUV_data = new float[N*L] ) )
- {
- ErrorHandler ( ( char* ) "msImageProcessor", ( char* ) "Allocate", ( char* ) "Not enough memory" );
- return;
- }
-
- class_state.OUTPUT_DEFINED = true;
- }
- void msImageProcessor::DestroyOutput ( void )
- {
-
- if ( msRawData ) delete [] msRawData;
-
-
- if ( modes ) delete [] modes;
- if ( labels ) delete [] labels;
- if ( modePointCounts ) delete [] modePointCounts;
- if ( indexTable ) delete [] indexTable;
-
- if ( LUV_data ) delete [] LUV_data;
-
-
- msRawData = NULL;
-
- modes = NULL;
- labels = NULL;
- modePointCounts = NULL;
- regionCount = 0;
-
- class_state.OUTPUT_DEFINED = false;
-
- return;
- }
- void msImageProcessor::NewOptimizedFilter1 ( float sigmaS, float sigmaR )
- {
-
- int iterationCount, i, j, k, modeCandidateX, modeCandidateY, modeCandidate_i;
- double mvAbs, diff, el;
-
-
- if ( !height )
- {
- ErrorHandler ( ( char* ) "msImageProcessor", ( char* ) "LFilter", ( char* ) "Lattice height and width are undefined." );
- return;
- }
-
- if ( ( ( h[0] = sigmaS ) <= 0 ) || ( ( h[1] = sigmaR ) <= 0 ) )
- {
- ErrorHandler ( ( char* ) "msImageProcessor", ( char* ) "Segment", ( char* ) "sigmaS and/or sigmaR is zero or negative." );
- return;
- }
-
- int lN = N + 2;
-
-
-
- double *yk = new double [lN];
-
- double *Mh = new double [lN];
-
- float* sdata;
- sdata = new float[lN*L];
-
- int idxs, idxd;
- idxs = idxd = 0;
- if ( N == 3 )
- {
- for ( i = 0; i < L; i++ )
- {
- sdata[idxs++] = ( i % width ) / sigmaS;
- sdata[idxs++] = ( i / width ) / sigmaS;
- sdata[idxs++] = data[idxd++] / sigmaR;
- sdata[idxs++] = data[idxd++] / sigmaR;
- sdata[idxs++] = data[idxd++] / sigmaR;
- }
- } else if ( N == 1 )
- {
- for ( i = 0; i < L; i++ )
- {
- sdata[idxs++] = ( i % width ) / sigmaS;
- sdata[idxs++] = ( i / width ) / sigmaS;
- sdata[idxs++] = data[idxd++] / sigmaR;
- }
- } else
- {
- for ( i = 0; i < L; i++ )
- {
- sdata[idxs++] = ( i % width ) / sigmaS;
- sdata[idxs++] = ( i / width ) / sigmaS;
- for ( j = 0; j < N; j++ )
- sdata[idxs++] = data[idxd++] / sigmaR;
- }
- }
-
- int* buckets;
- int* slist;
- slist = new int[L];
- int bucNeigh[27];
- float sMins;
- float sMaxs[3];
- sMaxs[0] = width / sigmaS;
- sMaxs[1] = height / sigmaS;
- sMins = sMaxs[2] = sdata[2];
- idxs = 2;
- float cval;
- for ( i = 0; i < L; i++ )
- {
- cval = sdata[idxs];
- if ( cval < sMins )
- sMins = cval;
- else if ( cval > sMaxs[2] )
- sMaxs[2] = cval;
- idxs += lN;
- }
- int nBuck1, nBuck2, nBuck3;
- int cBuck1, cBuck2, cBuck3, cBuck;
- nBuck1 = ( int ) ( sMaxs[0] + 3 );
- nBuck2 = ( int ) ( sMaxs[1] + 3 );
- nBuck3 = ( int ) ( sMaxs[2] - sMins + 3 );
- buckets = new int[nBuck1*nBuck2*nBuck3];
- for ( i = 0; i < ( nBuck1*nBuck2*nBuck3 ); i++ )
- buckets[i] = -1;
- idxs = 0;
- for ( i = 0; i < L; i++ )
- {
-
- cBuck1 = ( int ) sdata[idxs] + 1;
- cBuck2 = ( int ) sdata[idxs+1] + 1;
- cBuck3 = ( int ) ( sdata[idxs+2] - sMins ) + 1;
- idxs += lN;
- cBuck = cBuck1 + nBuck1 * ( cBuck2 + nBuck2 * cBuck3 );
- slist[i] = buckets[cBuck];
- buckets[cBuck] = i;
- }
-
- idxd = 0;
- for ( cBuck1 = -1; cBuck1 <= 1; cBuck1++ )
- {
- for ( cBuck2 = -1; cBuck2 <= 1; cBuck2++ )
- {
- for ( cBuck3 = -1; cBuck3 <= 1; cBuck3++ )
- {
- bucNeigh[idxd++] = cBuck1 + nBuck1 * ( cBuck2 + nBuck2 * cBuck3 );
- }
- }
- }
- double wsuml, weight;
- double hiLTr = 80.0 / sigmaR;
-
-
- memset ( modeTable, 0, width*height );
-
- #ifdef PROMPT
- printf ( ( char* ) "done.\nApplying mean shift (Using Lattice) ... " );
- #ifdef SHOW_PROGRESS
- printf ( ( char* ) "\n 0%%" );
- #endif
- #endif
- for ( i = 0; i < L; i++ )
- {
-
-
-
- if ( modeTable[i] == 1 )
- continue;
-
- pointCount = 0;
-
-
-
- idxs = i * lN;
- for ( j = 0; j < lN; j++ )
- yk[j] = sdata[idxs+j];
-
-
-
-
- for ( j = 0; j < lN; j++ )
- Mh[j] = 0;
- wsuml = 0;
-
-
- cBuck1 = ( int ) yk[0] + 1;
- cBuck2 = ( int ) yk[1] + 1;
- cBuck3 = ( int ) ( yk[2] - sMins ) + 1;
- cBuck = cBuck1 + nBuck1 * ( cBuck2 + nBuck2 * cBuck3 );
- for ( j = 0; j < 27; j++ )
- {
- idxd = buckets[cBuck+bucNeigh[j]];
-
- while ( idxd >= 0 )
- {
- idxs = lN * idxd;
-
- el = sdata[idxs+0] - yk[0];
- diff = el * el;
- el = sdata[idxs+1] - yk[1];
- diff += el * el;
- if ( diff < 1.0 )
- {
- el = sdata[idxs+2] - yk[2];
- if ( yk[2] > hiLTr )
- diff = 4 * el * el;
- else
- diff = el * el;
- if ( N > 1 )
- {
- el = sdata[idxs+3] - yk[3];
- diff += el * el;
- el = sdata[idxs+4] - yk[4];
- diff += el * el;
- }
- if ( diff < 1.0 )
- {
- weight = 1 - weightMap[idxd];
- for ( k = 0; k < lN; k++ )
- Mh[k] += weight * sdata[idxs+k];
- wsuml += weight;
- }
- }
- idxd = slist[idxd];
- }
- }
- if ( wsuml > 0 )
- {
- for ( j = 0; j < lN; j++ )
- Mh[j] = Mh[j] / wsuml - yk[j];
- }
- else
- {
- for ( j = 0; j < lN; j++ )
- Mh[j] = 0;
- }
-
-
-
-
-
- mvAbs = ( Mh[0] * Mh[0] + Mh[1] * Mh[1] ) * sigmaS * sigmaS;
- if ( N == 3 )
- mvAbs += ( Mh[2] * Mh[2] + Mh[3] * Mh[3] + Mh[4] * Mh[4] ) * sigmaR * sigmaR;
- else
- mvAbs += Mh[2] * Mh[2] * sigmaR * sigmaR;
-
-
-
-
-
- iterationCount = 1;
- while ( ( mvAbs >= EPSILON2 ) && ( iterationCount < LIMIT ) )
- {
-
- for ( j = 0; j < lN; j++ )
- yk[j] += Mh[j];
-
-
-
- modeCandidateX = ( int ) ( sigmaS * yk[0] + 0.5 );
- modeCandidateY = ( int ) ( sigmaS * yk[1] + 0.5 );
- modeCandidate_i = modeCandidateY * width + modeCandidateX;
-
-
-
-
-
-
-
-
-
-
-
- if ( ( modeTable[modeCandidate_i] != 2 ) && ( modeCandidate_i != i ) )
- {
-
-
-
- diff = 0;
- idxs = lN * modeCandidate_i;
- for ( k = 2; k < lN; k++ )
- {
- el = sdata[idxs+k] - yk[k];
- diff += el * el;
- }
-
-
-
-
- if ( diff < TC_DIST_FACTOR )
- {
-
-
-
-
- if ( modeTable[modeCandidate_i] == 0 )
- {
-
-
- pointList[pointCount++] = modeCandidate_i;
- modeTable[modeCandidate_i] = 2;
- } else
- {
-
-
-
-
-
- for ( j = 0; j < N; j++ )
- yk[j+2] = msRawData[modeCandidate_i*N+j] / sigmaR;
-
-
-
- modeTable[i] = 1;
-
-
- mvAbs = -1;
-
- break;
- }
- }
- }
-
-
-
-
-
-
- for ( j = 0; j < lN; j++ )
- Mh[j] = 0;
- wsuml = 0;
-
-
- cBuck1 = ( int ) yk[0] + 1;
- cBuck2 = ( int ) yk[1] + 1;
- cBuck3 = ( int ) ( yk[2] - sMins ) + 1;
- cBuck = cBuck1 + nBuck1 * ( cBuck2 + nBuck2 * cBuck3 );
- for ( j = 0; j < 27; j++ )
- {
- idxd = buckets[cBuck+bucNeigh[j]];
-
- while ( idxd >= 0 )
- {
- idxs = lN * idxd;
-
- el = sdata[idxs+0] - yk[0];
- diff = el * el;
- el = sdata[idxs+1] - yk[1];
- diff += el * el;
- if ( diff < 1.0 )
- {
- el = sdata[idxs+2] - yk[2];
- if ( yk[2] > hiLTr )
- diff = 4 * el * el;
- else
- diff = el * el;
- if ( N > 1 )
- {
- el = sdata[idxs+3] - yk[3];
- diff += el * el;
- el = sdata[idxs+4] - yk[4];
- diff += el * el;
- }
- if ( diff < 1.0 )
- {
- weight = 1 - weightMap[idxd];
- for ( k = 0; k < lN; k++ )
- Mh[k] += weight * sdata[idxs+k];
- wsuml += weight;
- }
- }
- idxd = slist[idxd];
- }
- }
- if ( wsuml > 0 )
- {
- for ( j = 0; j < lN; j++ )
- Mh[j] = Mh[j] / wsuml - yk[j];
- }
- else
- {
- for ( j = 0; j < lN; j++ )
- Mh[j] = 0;
- }
-
-
-
-
-
- mvAbs = ( Mh[0] * Mh[0] + Mh[1] * Mh[1] ) * sigmaS * sigmaS;
- if ( N == 3 )
- mvAbs += ( Mh[2] * Mh[2] + Mh[3] * Mh[3] + Mh[4] * Mh[4] ) * sigmaR * sigmaR;
- else
- mvAbs += Mh[2] * Mh[2] * sigmaR * sigmaR;
-
- iterationCount++;
- }
-
-
- if ( mvAbs >= 0 )
- {
-
- for ( j = 0; j < lN; j++ )
- yk[j] += Mh[j];
-
-
-
- modeTable[i] = 1;
- }
- for ( k = 0; k < N; k++ )
- yk[k+2] *= sigmaR;
-
-
-
- for ( j = 0; j < pointCount; j++ )
- {
-
-
- modeCandidate_i = pointList[j];
-
- modeTable[modeCandidate_i] = 1;
-
- for ( k = 0; k < N; k++ )
- msRawData[N*modeCandidate_i+k] = ( float ) ( yk[k+2] );
- }
-
- for ( j = 0; j < N; j++ )
- msRawData[N*i+j] = ( float ) ( yk[j+2] );
-
- #ifdef SHOW_PROGRESS
- percent_complete = ( float ) ( i / ( float ) ( L ) ) * 100;
- printf ( ( char* ) "\r%2d%%", ( int ) ( percent_complete + 0.5 ) );
- #endif
-
- if ( ( i % PROGRESS_RATE == 0 ) && ( ( ErrorStatus = msSys.Progress ( ( float ) ( i / ( float ) ( L ) ) * ( float ) ( 0.8 ) ) ) ) == EL_HALT )
- break;
- }
-
- #ifdef PROMPT
- #ifdef SHOW_PROGRESS
- printf ( ( char* ) "\r" );
- #endif
- printf ( ( char* ) "done." );
- #endif
-
- delete [] buckets;
- delete [] slist;
- delete [] sdata;
- delete [] yk;
- delete [] Mh;
-
- return;
- }
- void msImageProcessor::NewOptimizedFilter2 ( float sigmaS, float sigmaR )
- {
-
- int iterationCount, i, j, k, modeCandidateX, modeCandidateY, modeCandidate_i;
- double mvAbs, diff, el;
-
-
- if ( !height )
- {
- ErrorHandler ( ( char* ) "msImageProcessor", ( char* ) "LFilter", ( char* ) "Lattice height and width are undefined." );
- return;
- }
-
- if ( ( ( h[0] = sigmaS ) <= 0 ) || ( ( h[1] = sigmaR ) <= 0 ) )
- {
- ErrorHandler ( ( char* ) "msImageProcessor", ( char* ) "Segment", ( char* ) "sigmaS and/or sigmaR is zero or negative." );
- return;
- }
-
- int lN = N + 2;
-
-
-
- double *yk = new double [lN];
-
- double *Mh = new double [lN];
-
- float* sdata;
- sdata = new float[lN*L];
-
- int idxs, idxd;
- idxs = idxd = 0;
- if ( N == 3 )
- {
- for ( i = 0; i < L; i++ )
- {
- sdata[idxs++] = ( i % width ) / sigmaS;
- sdata[idxs++] = ( i / width ) / sigmaS;
- sdata[idxs++] = data[idxd++] / sigmaR;
- sdata[idxs++] = data[idxd++] / sigmaR;
- sdata[idxs++] = data[idxd++] / sigmaR;
- }
- } else if ( N == 1 )
- {
- for ( i = 0; i < L; i++ )
- {
- sdata[idxs++] = ( i % width ) / sigmaS;
- sdata[idxs++] = ( i / width ) / sigmaS;
- sdata[idxs++] = data[idxd++] / sigmaR;
- }
- } else
- {
- for ( i = 0; i < L; i++ )
- {
- sdata[idxs++] = ( i % width ) / sigmaS;
- sdata[idxs++] = ( i / width ) / sigmaS;
- for ( j = 0; j < N; j++ )
- sdata[idxs++] = data[idxd++] / sigmaR;
- }
- }
-
- int* buckets;
- int* slist;
- slist = new int[L];
- int bucNeigh[27];
- float sMins;
- float sMaxs[3];
- sMaxs[0] = width / sigmaS;
- sMaxs[1] = height / sigmaS;
- sMins = sMaxs[2] = sdata[2];
- idxs = 2;
- float cval;
- for ( i = 0; i < L; i++ )
- {
- cval = sdata[idxs];
- if ( cval < sMins )
- sMins = cval;
- else if ( cval > sMaxs[2] )
- sMaxs[2] = cval;
- idxs += lN;
- }
- int nBuck1, nBuck2, nBuck3;
- int cBuck1, cBuck2, cBuck3, cBuck;
- nBuck1 = ( int ) ( sMaxs[0] + 3 );
- nBuck2 = ( int ) ( sMaxs[1] + 3 );
- nBuck3 = ( int ) ( sMaxs[2] - sMins + 3 );
- buckets = new int[nBuck1*nBuck2*nBuck3];
- for ( i = 0; i < ( nBuck1*nBuck2*nBuck3 ); i++ )
- buckets[i] = -1;
- idxs = 0;
- for ( i = 0; i < L; i++ )
- {
-
- cBuck1 = ( int ) sdata[idxs] + 1;
- cBuck2 = ( int ) sdata[idxs+1] + 1;
- cBuck3 = ( int ) ( sdata[idxs+2] - sMins ) + 1;
- cBuck = cBuck1 + nBuck1 * ( cBuck2 + nBuck2 * cBuck3 );
- slist[i] = buckets[cBuck];
- buckets[cBuck] = i;
- idxs += lN;
- }
-
- idxd = 0;
- for ( cBuck1 = -1; cBuck1 <= 1; cBuck1++ )
- {
- for ( cBuck2 = -1; cBuck2 <= 1; cBuck2++ )
- {
- for ( cBuck3 = -1; cBuck3 <= 1; cBuck3++ )
- {
- bucNeigh[idxd++] = cBuck1 + nBuck1 * ( cBuck2 + nBuck2 * cBuck3 );
- }
- }
- }
- double wsuml, weight;
- double hiLTr = 80.0 / sigmaR;
-
-
- memset ( modeTable, 0, width*height );
-
- #ifdef PROMPT
- printf ( ( char* ) "done.\nApplying mean shift (Using Lattice) ... " );
- #ifdef SHOW_PROGRESS
- printf ( ( char* ) "\n 0%%" );
- #endif
- #endif
- for ( i = 0; i < L; i++ )
- {
-
-
-
- if ( modeTable[i] == 1 )
- continue;
-
- pointCount = 0;
-
-
-
- idxs = i * lN;
- for ( j = 0; j < lN; j++ )
- yk[j] = sdata[idxs+j];
-
-
-
-
- for ( j = 0; j < lN; j++ )
- Mh[j] = 0;
- wsuml = 0;
-
-
- cBuck1 = ( int ) yk[0] + 1;
- cBuck2 = ( int ) yk[1] + 1;
- cBuck3 = ( int ) ( yk[2] - sMins ) + 1;
- cBuck = cBuck1 + nBuck1 * ( cBuck2 + nBuck2 * cBuck3 );
- for ( j = 0; j < 27; j++ )
- {
- idxd = buckets[cBuck+bucNeigh[j]];
-
- while ( idxd >= 0 )
- {
- idxs = lN * idxd;
-
- el = sdata[idxs+0] - yk[0];
- diff = el * el;
- el = sdata[idxs+1] - yk[1];
- diff += el * el;
- if ( diff < 1.0 )
- {
- el = sdata[idxs+2] - yk[2];
- if ( yk[2] > hiLTr )
- diff = 4 * el * el;
- else
- diff = el * el;
- if ( N > 1 )
- {
- el = sdata[idxs+3] - yk[3];
- diff += el * el;
- el = sdata[idxs+4] - yk[4];
- diff += el * el;
- }
- if ( diff < 1.0 )
- {
- weight = 1 - weightMap[idxd];
- for ( k = 0; k < lN; k++ )
- Mh[k] += weight * sdata[idxs+k];
- wsuml += weight;
-
- if ( diff < speedThreshold )
- {
- if ( modeTable[idxd] == 0 )
- {
- pointList[pointCount++] = idxd;
- modeTable[idxd] = 2;
- }
- }
- }
- }
- idxd = slist[idxd];
- }
- }
- if ( wsuml > 0 )
- {
- for ( j = 0; j < lN; j++ )
- Mh[j] = Mh[j] / wsuml - yk[j];
- }
- else
- {
- for ( j = 0; j < lN; j++ )
- Mh[j] = 0;
- }
-
-
-
-
-
- mvAbs = ( Mh[0] * Mh[0] + Mh[1] * Mh[1] ) * sigmaS * sigmaS;
- if ( N == 3 )
- mvAbs += ( Mh[2] * Mh[2] + Mh[3] * Mh[3] + Mh[4] * Mh[4] ) * sigmaR * sigmaR;
- else
- mvAbs += Mh[2] * Mh[2] * sigmaR * sigmaR;
-
-
-
-
-
- iterationCount = 1;
- while ( ( mvAbs >= EPSILON2 ) && ( iterationCount < LIMIT ) )
- {
-
- for ( j = 0; j < lN; j++ )
- yk[j] += Mh[j];
-
-
-
- modeCandidateX = ( int ) ( sigmaS * yk[0] + 0.5 );
- modeCandidateY = ( int ) ( sigmaS * yk[1] + 0.5 );
- modeCandidate_i = modeCandidateY * width + modeCandidateX;
-
-
-
-
-
-
-
-
-
-
-
- if ( ( modeTable[modeCandidate_i] != 2 ) && ( modeCandidate_i != i ) )
- {
-
-
-
- diff = 0;
- idxs = lN * modeCandidate_i;
- for ( k = 2; k < lN; k++ )
- {
- el = sdata[idxs+k] - yk[k];
- diff += el * el;
- }
-
-
-
-
- if ( diff < speedThreshold )
- {
-
-
-
-
- if ( modeTable[modeCandidate_i] == 0 )
- {
-
-
- pointList[pointCount++] = modeCandidate_i;
- modeTable[modeCandidate_i] = 2;
- } else
- {
-
-
-
-
-
- for ( j = 0; j < N; j++ )
- yk[j+2] = msRawData[modeCandidate_i*N+j] / sigmaR;
-
-
-
- modeTable[i] = 1;
-
-
- mvAbs = -1;
-
- break;
- }
- }
- }
-
-
-
-
-
-
- for ( j = 0; j < lN; j++ )
- Mh[j] = 0;
- wsuml = 0;
-
-
- cBuck1 = ( int ) yk[0] + 1;
- cBuck2 = ( int ) yk[1] + 1;
- cBuck3 = ( int ) ( yk[2] - sMins ) + 1;
- cBuck = cBuck1 + nBuck1 * ( cBuck2 + nBuck2 * cBuck3 );
- for ( j = 0; j < 27; j++ )
- {
- idxd = buckets[cBuck+bucNeigh[j]];
-
- while ( idxd >= 0 )
- {
- idxs = lN * idxd;
-
- el = sdata[idxs+0] - yk[0];
- diff = el * el;
- el = sdata[idxs+1] - yk[1];
- diff += el * el;
- if ( diff < 1.0 )
- {
- el = sdata[idxs+2] - yk[2];
- if ( yk[2] > hiLTr )
- diff = 4 * el * el;
- else
- diff = el * el;
- if ( N > 1 )
- {
- el = sdata[idxs+3] - yk[3];
- diff += el * el;
- el = sdata[idxs+4] - yk[4];
- diff += el * el;
- }
- if ( diff < 1.0 )
- {
- weight = 1 - weightMap[idxd];
- for ( k = 0; k < lN; k++ )
- Mh[k] += weight * sdata[idxs+k];
- wsuml += weight;
-
- if ( diff < speedThreshold )
- {
- if ( modeTable[idxd] == 0 )
- {
- pointList[pointCount++] = idxd;
- modeTable[idxd] = 2;
- }
- }
- }
- }
- idxd = slist[idxd];
- }
- }
- if ( wsuml > 0 )
- {
- for ( j = 0; j < lN; j++ )
- Mh[j] = Mh[j] / wsuml - yk[j];
- }
- else
- {
- for ( j = 0; j < lN; j++ )
- Mh[j] = 0;
- }
-
-
-
-
-
- mvAbs = ( Mh[0] * Mh[0] + Mh[1] * Mh[1] ) * sigmaS * sigmaS;
- if ( N == 3 )
- mvAbs += ( Mh[2] * Mh[2] + Mh[3] * Mh[3] + Mh[4] * Mh[4] ) * sigmaR * sigmaR;
- else
- mvAbs += Mh[2] * Mh[2] * sigmaR * sigmaR;
-
- iterationCount++;
- }
-
-
- if ( mvAbs >= 0 )
- {
-
- for ( j = 0; j < lN; j++ )
- yk[j] += Mh[j];
-
-
-
- modeTable[i] = 1;
- }
- for ( k = 0; k < N; k++ )
- yk[k+2] *= sigmaR;
-
-
-
- for ( j = 0; j < pointCount; j++ )
- {
-
-
- modeCandidate_i = pointList[j];
-
- modeTable[modeCandidate_i] = 1;
-
- for ( k = 0; k < N; k++ )
- msRawData[N*modeCandidate_i+k] = ( float ) ( yk[k+2] );
- }
-
- for ( j = 0; j < N; j++ )
- msRawData[N*i+j] = ( float ) ( yk[j+2] );
-
- #ifdef SHOW_PROGRESS
- percent_complete = ( float ) ( i / ( float ) ( L ) ) * 100;
- printf ( ( char* ) "\r%2d%%", ( int ) ( percent_complete + 0.5 ) );
- #endif
-
- if ( ( i % PROGRESS_RATE == 0 ) && ( ( ErrorStatus = msSys.Progress ( ( float ) ( i / ( float ) ( L ) ) * ( float ) ( 0.8 ) ) ) ) == EL_HALT )
- break;
- }
-
- #ifdef PROMPT
- #ifdef SHOW_PROGRESS
- printf ( ( char* ) "\r" );
- #endif
- printf ( ( char* ) "done." );
- #endif
-
- delete [] buckets;
- delete [] slist;
- delete [] sdata;
- delete [] yk;
- delete [] Mh;
-
- return;
- }
- void msImageProcessor::NewNonOptimizedFilter ( float sigmaS, float sigmaR )
- {
-
- int iterationCount, i, j, k;
- double mvAbs, diff, el;
-
-
- if ( !height )
- {
- ErrorHandler ( ( char* ) "msImageProcessor", ( char* ) "LFilter", ( char* ) "Lattice height and width are undefined." );
- return;
- }
-
- if ( ( ( h[0] = sigmaS ) <= 0 ) || ( ( h[1] = sigmaR ) <= 0 ) )
- {
- ErrorHandler ( ( char* ) "msImageProcessor", ( char* ) "Segment", ( char* ) "sigmaS and/or sigmaR is zero or negative." );
- return;
- }
-
- int lN = N + 2;
-
-
-
- double *yk = new double [lN];
-
- double *Mh = new double [lN];
-
- double* sdata;
- sdata = new double[lN*L];
-
- int idxs, idxd;
- idxs = idxd = 0;
- if ( N == 3 )
- {
- for ( i = 0; i < L; i++ )
- {
- sdata[idxs++] = ( i % width ) / sigmaS;
- sdata[idxs++] = ( i / width ) / sigmaS;
- sdata[idxs++] = data[idxd++] / sigmaR;
- sdata[idxs++] = data[idxd++] / sigmaR;
- sdata[idxs++] = data[idxd++] / sigmaR;
- }
- } else if ( N == 1 )
- {
- for ( i = 0; i < L; i++ )
- {
- sdata[idxs++] = ( i % width ) / sigmaS;
- sdata[idxs++] = ( i / width ) / sigmaS;
- sdata[idxs++] = data[idxd++] / sigmaR;
- }
- } else
- {
- for ( i = 0; i < L; i++ )
- {
- sdata[idxs++] = ( i % width ) / sigmaS;
- sdata[idxs++] = ( i % width ) / sigmaS;
- for ( j = 0; j < N; j++ )
- sdata[idxs++] = data[idxd++] / sigmaR;
- }
- }
-
- int* buckets;
- int* slist;
- slist = new int[L];
- int bucNeigh[27];
- double sMins;
- double sMaxs[3];
- sMaxs[0] = width / sigmaS;
- sMaxs[1] = height / sigmaS;
- sMins = sMaxs[2] = sdata[2];
- idxs = 2;
- double cval;
- for ( i = 0; i < L; i++ )
- {
- cval = sdata[idxs];
- if ( cval < sMins )
- sMins = cval;
- else if ( cval > sMaxs[2] )
- sMaxs[2] = cval;
- idxs += lN;
- }
- int nBuck1, nBuck2, nBuck3;
- int cBuck1, cBuck2, cBuck3, cBuck;
- nBuck1 = ( int ) ( sMaxs[0] + 3 );
- nBuck2 = ( int ) ( sMaxs[1] + 3 );
- nBuck3 = ( int ) ( sMaxs[2] - sMins + 3 );
- buckets = new int[nBuck1*nBuck2*nBuck3];
- for ( i = 0; i < ( nBuck1*nBuck2*nBuck3 ); i++ )
- buckets[i] = -1;
- idxs = 0;
- for ( i = 0; i < L; i++ )
- {
-
- cBuck1 = ( int ) sdata[idxs] + 1;
- cBuck2 = ( int ) sdata[idxs+1] + 1;
- cBuck3 = ( int ) ( sdata[idxs+2] - sMins ) + 1;
- cBuck = cBuck1 + nBuck1 * ( cBuck2 + nBuck2 * cBuck3 );
- slist[i] = buckets[cBuck];
- buckets[cBuck] = i;
- idxs += lN;
- }
-
- idxd = 0;
- for ( cBuck1 = -1; cBuck1 <= 1; cBuck1++ )
- {
- for ( cBuck2 = -1; cBuck2 <= 1; cBuck2++ )
- {
- for ( cBuck3 = -1; cBuck3 <= 1; cBuck3++ )
- {
- bucNeigh[idxd++] = cBuck1 + nBuck1 * ( cBuck2 + nBuck2 * cBuck3 );
- }
- }
- }
- double wsuml, weight;
- double hiLTr = 80.0 / sigmaR;
-
-
- #ifdef PROMPT
- printf ( ( char* ) "done.\nApplying mean shift (Using Lattice)... " );
- #ifdef SHOW_PROGRESS
- printf ( ( char* ) "\n 0%%" );
- #endif
- #endif
- for ( i = 0; i < L; i++ )
- {
-
-
-
- idxs = i * lN;
- for ( j = 0; j < lN; j++ )
- yk[j] = sdata[idxs+j];
-
-
-
-
- for ( j = 0; j < lN; j++ )
- Mh[j] = 0;
- wsuml = 0;
-
-
- cBuck1 = ( int ) yk[0] + 1;
- cBuck2 = ( int ) yk[1] + 1;
- cBuck3 = ( int ) ( yk[2] - sMins ) + 1;
- cBuck = cBuck1 + nBuck1 * ( cBuck2 + nBuck2 * cBuck3 );
- for ( j = 0; j < 27; j++ )
- {
- idxd = buckets[cBuck+bucNeigh[j]];
-
- while ( idxd >= 0 )
- {
- idxs = lN * idxd;
-
- el = sdata[idxs+0] - yk[0];
- diff = el * el;
- el = sdata[idxs+1] - yk[1];
- diff += el * el;
- if ( diff < 1.0 )
- {
- el = sdata[idxs+2] - yk[2];
- if ( yk[2] > hiLTr )
- diff = 4 * el * el;
- else
- diff = el * el;
- if ( N > 1 )
- {
- el = sdata[idxs+3] - yk[3];
- diff += el * el;
- el = sdata[idxs+4] - yk[4];
- diff += el * el;
- }
- if ( diff < 1.0 )
- {
- weight = 1 - weightMap[idxd];
- for ( k = 0; k < lN; k++ )
- Mh[k] += weight * sdata[idxs+k];
- wsuml += weight;
- }
- }
- idxd = slist[idxd];
- }
- }
- if ( wsuml > 0 )
- {
- for ( j = 0; j < lN; j++ )
- Mh[j] = Mh[j] / wsuml - yk[j];
- }
- else
- {
- for ( j = 0; j < lN; j++ )
- Mh[j] = 0;
- }
-
-
- mvAbs = 0;
- for ( j = 0; j < lN; j++ )
- mvAbs += Mh[j] * Mh[j];
-
-
-
-
-
- iterationCount = 1;
- while ( ( mvAbs >= EPSILON2 ) && ( iterationCount < LIMIT ) )
- {
-
- for ( j = 0; j < lN; j++ )
- yk[j] += Mh[j];
-
-
-
-
-
- for ( j = 0; j < lN; j++ )
- Mh[j] = 0;
- wsuml = 0;
-
-
- cBuck1 = ( int ) yk[0] + 1;
- cBuck2 = ( int ) yk[1] + 1;
- cBuck3 = ( int ) ( yk[2] - sMins ) + 1;
- cBuck = cBuck1 + nBuck1 * ( cBuck2 + nBuck2 * cBuck3 );
- for ( j = 0; j < 27; j++ )
- {
- idxd = buckets[cBuck+bucNeigh[j]];
-
- while ( idxd >= 0 )
- {
- idxs = lN * idxd;
-
- el = sdata[idxs+0] - yk[0];
- diff = el * el;
- el = sdata[idxs+1] - yk[1];
- diff += el * el;
- if ( diff < 1.0 )
- {
- el = sdata[idxs+2] - yk[2];
- if ( yk[2] > hiLTr )
- diff = 4 * el * el;
- else
- diff = el * el;
- if ( N > 1 )
- {
- el = sdata[idxs+3] - yk[3];
- diff += el * el;
- el = sdata[idxs+4] - yk[4];
- diff += el * el;
- }
- if ( diff < 1.0 )
- {
- weight = 1 - weightMap[idxd];
- for ( k = 0; k < lN; k++ )
- Mh[k] += weight * sdata[idxs+k];
- wsuml += weight;
- }
- }
- idxd = slist[idxd];
- }
- }
- if ( wsuml > 0 )
- {
- for ( j = 0; j < lN; j++ )
- Mh[j] = Mh[j] / wsuml - yk[j];
- }
- else
- {
- for ( j = 0; j < lN; j++ )
- Mh[j] = 0;
- }
-
-
-
-
-
- mvAbs = ( Mh[0] * Mh[0] + Mh[1] * Mh[1] ) * sigmaS * sigmaS;
- if ( N == 3 )
- mvAbs += ( Mh[2] * Mh[2] + Mh[3] * Mh[3] + Mh[4] * Mh[4] ) * sigmaR * sigmaR;
- else
- mvAbs += Mh[2] * Mh[2] * sigmaR * sigmaR;
-
- iterationCount++;
- }
-
- for ( j = 0; j < lN; j++ )
- yk[j] += Mh[j];
-
- for ( j = 0; j < N; j++ )
- msRawData[N*i+j] = ( float ) ( yk[j+2] * sigmaR );
-
- #ifdef SHOW_PROGRESS
- percent_complete = ( float ) ( i / ( float ) ( L ) ) * 100;
- printf ( ( char* ) "\r%2d%%", ( int ) ( percent_complete + 0.5 ) );
- #endif
-
- if ( ( i % PROGRESS_RATE == 0 ) && ( ( ErrorStatus = msSys.Progress ( ( float ) ( i / ( float ) ( L ) ) * ( float ) ( 0.8 ) ) ) ) == EL_HALT )
- break;
- }
-
- #ifdef PROMPT
- #ifdef SHOW_PROGRESS
- printf ( ( char* ) "\r" );
- #endif
- printf ( ( char* ) "done." );
- #endif
-
- delete [] buckets;
- delete [] slist;
- delete [] sdata;
- delete [] yk;
- delete [] Mh;
-
- return;
- }
- void msImageProcessor::SetSpeedThreshold ( float speedUpThreshold )
- {
- speedThreshold = speedUpThreshold;
- }
|