123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771377237733774377537763777377837793780378137823783378437853786378737883789379037913792379337943795379637973798379938003801380238033804380538063807380838093810381138123813381438153816381738183819382038213822382338243825382638273828382938303831383238333834383538363837383838393840384138423843384438453846384738483849385038513852385338543855385638573858385938603861386238633864386538663867386838693870387138723873387438753876387738783879388038813882388338843885388638873888388938903891389238933894389538963897389838993900390139023903390439053906390739083909391039113912391339143915391639173918391939203921392239233924392539263927392839293930393139323933393439353936393739383939394039413942394339443945394639473948394939503951395239533954395539563957395839593960396139623963396439653966396739683969397039713972397339743975397639773978397939803981398239833984398539863987398839893990399139923993399439953996399739983999400040014002400340044005400640074008400940104011401240134014401540164017401840194020402140224023402440254026402740284029403040314032403340344035403640374038403940404041404240434044404540464047404840494050405140524053405440554056405740584059406040614062406340644065406640674068406940704071407240734074407540764077407840794080408140824083408440854086408740884089409040914092409340944095409640974098409941004101410241034104410541064107410841094110411141124113411441154116411741184119412041214122412341244125412641274128412941304131413241334134413541364137413841394140414141424143414441454146414741484149415041514152415341544155415641574158415941604161416241634164416541664167416841694170417141724173417441754176417741784179418041814182418341844185418641874188418941904191419241934194419541964197419841994200420142024203420442054206420742084209421042114212421342144215421642174218421942204221422242234224422542264227422842294230423142324233423442354236423742384239424042414242424342444245424642474248424942504251425242534254425542564257425842594260426142624263426442654266426742684269427042714272427342744275427642774278427942804281428242834284428542864287428842894290429142924293429442954296429742984299430043014302430343044305430643074308430943104311431243134314431543164317431843194320432143224323432443254326432743284329433043314332433343344335433643374338433943404341434243434344434543464347434843494350435143524353435443554356435743584359436043614362436343644365436643674368436943704371437243734374437543764377437843794380438143824383438443854386438743884389439043914392439343944395439643974398439944004401440244034404440544064407440844094410441144124413441444154416441744184419442044214422442344244425 |
- // Copyright 2008-2016 Conrad Sanderson (http://conradsanderson.id.au)
- // Copyright 2008-2016 National ICT Australia (NICTA)
- //
- // Licensed under the Apache License, Version 2.0 (the "License");
- // you may not use this file except in compliance with the License.
- // You may obtain a copy of the License at
- // http://www.apache.org/licenses/LICENSE-2.0
- //
- // Unless required by applicable law or agreed to in writing, software
- // distributed under the License is distributed on an "AS IS" BASIS,
- // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- // See the License for the specific language governing permissions and
- // limitations under the License.
- // ------------------------------------------------------------------------
- //! \addtogroup subview
- //! @{
- template<typename eT>
- inline
- subview<eT>::~subview()
- {
- arma_extra_debug_sigprint();
- }
- template<typename eT>
- inline
- subview<eT>::subview(const Mat<eT>& in_m, const uword in_row1, const uword in_col1, const uword in_n_rows, const uword in_n_cols)
- : m(in_m)
- , aux_row1(in_row1)
- , aux_col1(in_col1)
- , n_rows(in_n_rows)
- , n_cols(in_n_cols)
- , n_elem(in_n_rows*in_n_cols)
- {
- arma_extra_debug_sigprint();
- }
- template<typename eT>
- inline
- void
- subview<eT>::operator= (const eT val)
- {
- arma_extra_debug_sigprint();
-
- if(n_elem != 1)
- {
- arma_debug_assert_same_size(n_rows, n_cols, 1, 1, "copy into submatrix");
- }
-
- Mat<eT>& X = const_cast< Mat<eT>& >(m);
-
- X.at(aux_row1, aux_col1) = val;
- }
- template<typename eT>
- template<typename op_type>
- inline
- void
- subview<eT>::inplace_op(const eT val)
- {
- arma_extra_debug_sigprint();
-
- subview<eT>& s = *this;
-
- const uword s_n_rows = s.n_rows;
- const uword s_n_cols = s.n_cols;
-
- if(s_n_rows == 1)
- {
- Mat<eT>& A = const_cast< Mat<eT>& >(s.m);
-
- const uword A_n_rows = A.n_rows;
-
- eT* Aptr = &(A.at(s.aux_row1,s.aux_col1));
-
- uword jj;
- for(jj=1; jj < s_n_cols; jj+=2)
- {
- if(is_same_type<op_type, op_internal_plus >::yes) { (*Aptr) += val; Aptr += A_n_rows; (*Aptr) += val; Aptr += A_n_rows; }
- if(is_same_type<op_type, op_internal_minus>::yes) { (*Aptr) -= val; Aptr += A_n_rows; (*Aptr) -= val; Aptr += A_n_rows; }
- if(is_same_type<op_type, op_internal_schur>::yes) { (*Aptr) *= val; Aptr += A_n_rows; (*Aptr) *= val; Aptr += A_n_rows; }
- if(is_same_type<op_type, op_internal_div >::yes) { (*Aptr) /= val; Aptr += A_n_rows; (*Aptr) /= val; Aptr += A_n_rows; }
- }
-
- if((jj-1) < s_n_cols)
- {
- if(is_same_type<op_type, op_internal_plus >::yes) { (*Aptr) += val; }
- if(is_same_type<op_type, op_internal_minus>::yes) { (*Aptr) -= val; }
- if(is_same_type<op_type, op_internal_schur>::yes) { (*Aptr) *= val; }
- if(is_same_type<op_type, op_internal_div >::yes) { (*Aptr) /= val; }
- }
- }
- else
- {
- for(uword ucol=0; ucol < s_n_cols; ++ucol)
- {
- if(is_same_type<op_type, op_internal_plus >::yes) { arrayops::inplace_plus ( colptr(ucol), val, s_n_rows ); }
- if(is_same_type<op_type, op_internal_minus>::yes) { arrayops::inplace_minus( colptr(ucol), val, s_n_rows ); }
- if(is_same_type<op_type, op_internal_schur>::yes) { arrayops::inplace_mul ( colptr(ucol), val, s_n_rows ); }
- if(is_same_type<op_type, op_internal_div >::yes) { arrayops::inplace_div ( colptr(ucol), val, s_n_rows ); }
- }
- }
- }
- template<typename eT>
- template<typename op_type, typename T1>
- inline
- void
- subview<eT>::inplace_op(const Base<eT,T1>& in, const char* identifier)
- {
- arma_extra_debug_sigprint();
-
- const Proxy<T1> P(in.get_ref());
-
- subview<eT>& s = *this;
-
- const uword s_n_rows = s.n_rows;
- const uword s_n_cols = s.n_cols;
-
- arma_debug_assert_same_size(s, P, identifier);
-
- const bool use_mp = arma_config::cxx11 && arma_config::openmp && Proxy<T1>::use_mp && mp_gate<eT>::eval(s.n_elem);
- const bool has_overlap = P.has_overlap(s);
-
- if(has_overlap) { arma_extra_debug_print("aliasing or overlap detected"); }
-
- if( (is_Mat<typename Proxy<T1>::stored_type>::value) || (use_mp) || (has_overlap) )
- {
- const unwrap_check<typename Proxy<T1>::stored_type> tmp(P.Q, has_overlap);
- const Mat<eT>& B = tmp.M;
-
- if(s_n_rows == 1)
- {
- Mat<eT>& A = const_cast< Mat<eT>& >(m);
-
- const uword A_n_rows = A.n_rows;
-
- eT* Aptr = &(A.at(aux_row1,aux_col1));
- const eT* Bptr = B.memptr();
-
- uword jj;
- for(jj=1; jj < s_n_cols; jj+=2)
- {
- const eT tmp1 = (*Bptr); Bptr++;
- const eT tmp2 = (*Bptr); Bptr++;
-
- if(is_same_type<op_type, op_internal_equ >::yes) { (*Aptr) = tmp1; Aptr += A_n_rows; (*Aptr) = tmp2; Aptr += A_n_rows; }
- if(is_same_type<op_type, op_internal_plus >::yes) { (*Aptr) += tmp1; Aptr += A_n_rows; (*Aptr) += tmp2; Aptr += A_n_rows; }
- if(is_same_type<op_type, op_internal_minus>::yes) { (*Aptr) -= tmp1; Aptr += A_n_rows; (*Aptr) -= tmp2; Aptr += A_n_rows; }
- if(is_same_type<op_type, op_internal_schur>::yes) { (*Aptr) *= tmp1; Aptr += A_n_rows; (*Aptr) *= tmp2; Aptr += A_n_rows; }
- if(is_same_type<op_type, op_internal_div >::yes) { (*Aptr) /= tmp1; Aptr += A_n_rows; (*Aptr) /= tmp2; Aptr += A_n_rows; }
- }
-
- if((jj-1) < s_n_cols)
- {
- if(is_same_type<op_type, op_internal_equ >::yes) { (*Aptr) = (*Bptr); }
- if(is_same_type<op_type, op_internal_plus >::yes) { (*Aptr) += (*Bptr); }
- if(is_same_type<op_type, op_internal_minus>::yes) { (*Aptr) -= (*Bptr); }
- if(is_same_type<op_type, op_internal_schur>::yes) { (*Aptr) *= (*Bptr); }
- if(is_same_type<op_type, op_internal_div >::yes) { (*Aptr) /= (*Bptr); }
- }
- }
- else // not a row vector
- {
- if( (is_same_type<op_type, op_internal_equ>::yes) && (s.aux_row1 == 0) && (s_n_rows == s.m.n_rows) )
- {
- arrayops::copy( s.colptr(0), B.memptr(), s.n_elem );
- }
- else
- {
- for(uword ucol=0; ucol < s_n_cols; ++ucol)
- {
- if(is_same_type<op_type, op_internal_equ >::yes) { arrayops::copy ( s.colptr(ucol), B.colptr(ucol), s_n_rows ); }
- if(is_same_type<op_type, op_internal_plus >::yes) { arrayops::inplace_plus ( s.colptr(ucol), B.colptr(ucol), s_n_rows ); }
- if(is_same_type<op_type, op_internal_minus>::yes) { arrayops::inplace_minus( s.colptr(ucol), B.colptr(ucol), s_n_rows ); }
- if(is_same_type<op_type, op_internal_schur>::yes) { arrayops::inplace_mul ( s.colptr(ucol), B.colptr(ucol), s_n_rows ); }
- if(is_same_type<op_type, op_internal_div >::yes) { arrayops::inplace_div ( s.colptr(ucol), B.colptr(ucol), s_n_rows ); }
- }
- }
- }
- }
- else // use the Proxy
- {
- if(s_n_rows == 1)
- {
- Mat<eT>& A = const_cast< Mat<eT>& >(m);
-
- const uword A_n_rows = A.n_rows;
-
- eT* Aptr = &(A.at(aux_row1,aux_col1));
-
- uword jj;
- for(jj=1; jj < s_n_cols; jj+=2)
- {
- const uword ii = (jj-1);
-
- const eT tmp1 = (Proxy<T1>::use_at) ? P.at(0,ii) : P[ii];
- const eT tmp2 = (Proxy<T1>::use_at) ? P.at(0,jj) : P[jj];
-
- if(is_same_type<op_type, op_internal_equ >::yes) { (*Aptr) = tmp1; Aptr += A_n_rows; (*Aptr) = tmp2; Aptr += A_n_rows; }
- if(is_same_type<op_type, op_internal_plus >::yes) { (*Aptr) += tmp1; Aptr += A_n_rows; (*Aptr) += tmp2; Aptr += A_n_rows; }
- if(is_same_type<op_type, op_internal_minus>::yes) { (*Aptr) -= tmp1; Aptr += A_n_rows; (*Aptr) -= tmp2; Aptr += A_n_rows; }
- if(is_same_type<op_type, op_internal_schur>::yes) { (*Aptr) *= tmp1; Aptr += A_n_rows; (*Aptr) *= tmp2; Aptr += A_n_rows; }
- if(is_same_type<op_type, op_internal_div >::yes) { (*Aptr) /= tmp1; Aptr += A_n_rows; (*Aptr) /= tmp2; Aptr += A_n_rows; }
- }
-
- const uword ii = (jj-1);
- if(ii < s_n_cols)
- {
- if(is_same_type<op_type, op_internal_equ >::yes) { (*Aptr) = (Proxy<T1>::use_at) ? P.at(0,ii) : P[ii]; }
- if(is_same_type<op_type, op_internal_plus >::yes) { (*Aptr) += (Proxy<T1>::use_at) ? P.at(0,ii) : P[ii]; }
- if(is_same_type<op_type, op_internal_minus>::yes) { (*Aptr) -= (Proxy<T1>::use_at) ? P.at(0,ii) : P[ii]; }
- if(is_same_type<op_type, op_internal_schur>::yes) { (*Aptr) *= (Proxy<T1>::use_at) ? P.at(0,ii) : P[ii]; }
- if(is_same_type<op_type, op_internal_div >::yes) { (*Aptr) /= (Proxy<T1>::use_at) ? P.at(0,ii) : P[ii]; }
- }
- }
- else // not a row vector
- {
- if(Proxy<T1>::use_at)
- {
- for(uword ucol=0; ucol < s_n_cols; ++ucol)
- {
- eT* s_col_data = s.colptr(ucol);
-
- uword jj;
- for(jj=1; jj < s_n_rows; jj+=2)
- {
- const uword ii = (jj-1);
-
- const eT tmp1 = P.at(ii,ucol);
- const eT tmp2 = P.at(jj,ucol);
-
- if(is_same_type<op_type, op_internal_equ >::yes) { (*s_col_data) = tmp1; s_col_data++; (*s_col_data) = tmp2; s_col_data++; }
- if(is_same_type<op_type, op_internal_plus >::yes) { (*s_col_data) += tmp1; s_col_data++; (*s_col_data) += tmp2; s_col_data++; }
- if(is_same_type<op_type, op_internal_minus>::yes) { (*s_col_data) -= tmp1; s_col_data++; (*s_col_data) -= tmp2; s_col_data++; }
- if(is_same_type<op_type, op_internal_schur>::yes) { (*s_col_data) *= tmp1; s_col_data++; (*s_col_data) *= tmp2; s_col_data++; }
- if(is_same_type<op_type, op_internal_div >::yes) { (*s_col_data) /= tmp1; s_col_data++; (*s_col_data) /= tmp2; s_col_data++; }
- }
-
- const uword ii = (jj-1);
- if(ii < s_n_rows)
- {
- if(is_same_type<op_type, op_internal_equ >::yes) { (*s_col_data) = P.at(ii,ucol); }
- if(is_same_type<op_type, op_internal_plus >::yes) { (*s_col_data) += P.at(ii,ucol); }
- if(is_same_type<op_type, op_internal_minus>::yes) { (*s_col_data) -= P.at(ii,ucol); }
- if(is_same_type<op_type, op_internal_schur>::yes) { (*s_col_data) *= P.at(ii,ucol); }
- if(is_same_type<op_type, op_internal_div >::yes) { (*s_col_data) /= P.at(ii,ucol); }
- }
- }
- }
- else
- {
- typename Proxy<T1>::ea_type Pea = P.get_ea();
-
- uword count = 0;
-
- for(uword ucol=0; ucol < s_n_cols; ++ucol)
- {
- eT* s_col_data = s.colptr(ucol);
-
- uword jj;
- for(jj=1; jj < s_n_rows; jj+=2)
- {
- const eT tmp1 = Pea[count]; count++;
- const eT tmp2 = Pea[count]; count++;
-
- if(is_same_type<op_type, op_internal_equ >::yes) { (*s_col_data) = tmp1; s_col_data++; (*s_col_data) = tmp2; s_col_data++; }
- if(is_same_type<op_type, op_internal_plus >::yes) { (*s_col_data) += tmp1; s_col_data++; (*s_col_data) += tmp2; s_col_data++; }
- if(is_same_type<op_type, op_internal_minus>::yes) { (*s_col_data) -= tmp1; s_col_data++; (*s_col_data) -= tmp2; s_col_data++; }
- if(is_same_type<op_type, op_internal_schur>::yes) { (*s_col_data) *= tmp1; s_col_data++; (*s_col_data) *= tmp2; s_col_data++; }
- if(is_same_type<op_type, op_internal_div >::yes) { (*s_col_data) /= tmp1; s_col_data++; (*s_col_data) /= tmp2; s_col_data++; }
- }
-
- if((jj-1) < s_n_rows)
- {
- if(is_same_type<op_type, op_internal_equ >::yes) { (*s_col_data) = Pea[count]; count++; }
- if(is_same_type<op_type, op_internal_plus >::yes) { (*s_col_data) += Pea[count]; count++; }
- if(is_same_type<op_type, op_internal_minus>::yes) { (*s_col_data) -= Pea[count]; count++; }
- if(is_same_type<op_type, op_internal_schur>::yes) { (*s_col_data) *= Pea[count]; count++; }
- if(is_same_type<op_type, op_internal_div >::yes) { (*s_col_data) /= Pea[count]; count++; }
- }
- }
- }
- }
- }
- }
- template<typename eT>
- template<typename op_type>
- inline
- void
- subview<eT>::inplace_op(const subview<eT>& x, const char* identifier)
- {
- arma_extra_debug_sigprint();
-
- if(check_overlap(x))
- {
- const Mat<eT> tmp(x);
-
- if(is_same_type<op_type, op_internal_equ >::yes) { (*this).operator= (tmp); }
- if(is_same_type<op_type, op_internal_plus >::yes) { (*this).operator+=(tmp); }
- if(is_same_type<op_type, op_internal_minus>::yes) { (*this).operator-=(tmp); }
- if(is_same_type<op_type, op_internal_schur>::yes) { (*this).operator%=(tmp); }
- if(is_same_type<op_type, op_internal_div >::yes) { (*this).operator/=(tmp); }
-
- return;
- }
-
- subview<eT>& s = *this;
-
- arma_debug_assert_same_size(s, x, identifier);
-
- const uword s_n_cols = s.n_cols;
- const uword s_n_rows = s.n_rows;
-
- if(s_n_rows == 1)
- {
- Mat<eT>& A = const_cast< Mat<eT>& >(s.m);
- const Mat<eT>& B = x.m;
-
- const uword A_n_rows = A.n_rows;
- const uword B_n_rows = B.n_rows;
-
- eT* Aptr = &(A.at(s.aux_row1,s.aux_col1));
- const eT* Bptr = &(B.at(x.aux_row1,x.aux_col1));
-
- uword jj;
- for(jj=1; jj < s_n_cols; jj+=2)
- {
- const eT tmp1 = (*Bptr); Bptr += B_n_rows;
- const eT tmp2 = (*Bptr); Bptr += B_n_rows;
-
- if(is_same_type<op_type, op_internal_equ >::yes) { (*Aptr) = tmp1; Aptr += A_n_rows; (*Aptr) = tmp2; Aptr += A_n_rows; }
- if(is_same_type<op_type, op_internal_plus >::yes) { (*Aptr) += tmp1; Aptr += A_n_rows; (*Aptr) += tmp2; Aptr += A_n_rows; }
- if(is_same_type<op_type, op_internal_minus>::yes) { (*Aptr) -= tmp1; Aptr += A_n_rows; (*Aptr) -= tmp2; Aptr += A_n_rows; }
- if(is_same_type<op_type, op_internal_schur>::yes) { (*Aptr) *= tmp1; Aptr += A_n_rows; (*Aptr) *= tmp2; Aptr += A_n_rows; }
- if(is_same_type<op_type, op_internal_div >::yes) { (*Aptr) /= tmp1; Aptr += A_n_rows; (*Aptr) /= tmp2; Aptr += A_n_rows; }
- }
-
- if((jj-1) < s_n_cols)
- {
- if(is_same_type<op_type, op_internal_equ >::yes) { (*Aptr) = (*Bptr); }
- if(is_same_type<op_type, op_internal_plus >::yes) { (*Aptr) += (*Bptr); }
- if(is_same_type<op_type, op_internal_minus>::yes) { (*Aptr) -= (*Bptr); }
- if(is_same_type<op_type, op_internal_schur>::yes) { (*Aptr) *= (*Bptr); }
- if(is_same_type<op_type, op_internal_div >::yes) { (*Aptr) /= (*Bptr); }
- }
- }
- else
- {
- for(uword ucol=0; ucol < s_n_cols; ++ucol)
- {
- if(is_same_type<op_type, op_internal_equ >::yes) { arrayops::copy ( s.colptr(ucol), x.colptr(ucol), s_n_rows ); }
- if(is_same_type<op_type, op_internal_plus >::yes) { arrayops::inplace_plus ( s.colptr(ucol), x.colptr(ucol), s_n_rows ); }
- if(is_same_type<op_type, op_internal_minus>::yes) { arrayops::inplace_minus( s.colptr(ucol), x.colptr(ucol), s_n_rows ); }
- if(is_same_type<op_type, op_internal_schur>::yes) { arrayops::inplace_mul ( s.colptr(ucol), x.colptr(ucol), s_n_rows ); }
- if(is_same_type<op_type, op_internal_div >::yes) { arrayops::inplace_div ( s.colptr(ucol), x.colptr(ucol), s_n_rows ); }
- }
- }
- }
- template<typename eT>
- inline
- void
- subview<eT>::operator+= (const eT val)
- {
- arma_extra_debug_sigprint();
-
- inplace_op<op_internal_plus>(val);
- }
- template<typename eT>
- inline
- void
- subview<eT>::operator-= (const eT val)
- {
- arma_extra_debug_sigprint();
-
- inplace_op<op_internal_minus>(val);
- }
- template<typename eT>
- inline
- void
- subview<eT>::operator*= (const eT val)
- {
- arma_extra_debug_sigprint();
-
- inplace_op<op_internal_schur>(val);
- }
- template<typename eT>
- inline
- void
- subview<eT>::operator/= (const eT val)
- {
- arma_extra_debug_sigprint();
-
- inplace_op<op_internal_div>(val);
- }
- template<typename eT>
- inline
- void
- subview<eT>::operator= (const subview<eT>& x)
- {
- arma_extra_debug_sigprint();
-
- inplace_op<op_internal_equ>(x, "copy into submatrix");
- }
- template<typename eT>
- inline
- void
- subview<eT>::operator+= (const subview<eT>& x)
- {
- arma_extra_debug_sigprint();
-
- inplace_op<op_internal_plus>(x, "addition");
- }
- template<typename eT>
- inline
- void
- subview<eT>::operator-= (const subview<eT>& x)
- {
- arma_extra_debug_sigprint();
-
- inplace_op<op_internal_minus>(x, "subtraction");
- }
- template<typename eT>
- inline
- void
- subview<eT>::operator%= (const subview& x)
- {
- arma_extra_debug_sigprint();
-
- inplace_op<op_internal_schur>(x, "element-wise multiplication");
- }
- template<typename eT>
- inline
- void
- subview<eT>::operator/= (const subview& x)
- {
- arma_extra_debug_sigprint();
-
- inplace_op<op_internal_div>(x, "element-wise division");
- }
- template<typename eT>
- template<typename T1>
- inline
- void
- subview<eT>::operator= (const Base<eT,T1>& in)
- {
- arma_extra_debug_sigprint();
-
- inplace_op<op_internal_equ>(in, "copy into submatrix");
- }
- template<typename eT>
- template<typename T1>
- inline
- void
- subview<eT>::operator+= (const Base<eT,T1>& in)
- {
- arma_extra_debug_sigprint();
-
- inplace_op<op_internal_plus>(in, "addition");
- }
- template<typename eT>
- template<typename T1>
- inline
- void
- subview<eT>::operator-= (const Base<eT,T1>& in)
- {
- arma_extra_debug_sigprint();
-
- inplace_op<op_internal_minus>(in, "subtraction");
- }
- template<typename eT>
- template<typename T1>
- inline
- void
- subview<eT>::operator%= (const Base<eT,T1>& in)
- {
- arma_extra_debug_sigprint();
-
- inplace_op<op_internal_schur>(in, "element-wise multiplication");
- }
- template<typename eT>
- template<typename T1>
- inline
- void
- subview<eT>::operator/= (const Base<eT,T1>& in)
- {
- arma_extra_debug_sigprint();
-
- inplace_op<op_internal_div>(in, "element-wise division");
- }
- template<typename eT>
- template<typename T1>
- inline
- void
- subview<eT>::operator=(const SpBase<eT, T1>& x)
- {
- arma_extra_debug_sigprint();
-
- const SpProxy<T1> p(x.get_ref());
-
- arma_debug_assert_same_size(n_rows, n_cols, p.get_n_rows(), p.get_n_cols(), "copy into submatrix");
-
- // Clear the subview.
- zeros();
-
- // Iterate through the sparse subview and set the nonzero values appropriately.
- typename SpProxy<T1>::const_iterator_type cit = p.begin();
- typename SpProxy<T1>::const_iterator_type cit_end = p.end();
-
- while(cit != cit_end)
- {
- at(cit.row(), cit.col()) = *cit;
- ++cit;
- }
- }
- template<typename eT>
- template<typename T1>
- inline
- void
- subview<eT>::operator+=(const SpBase<eT, T1>& x)
- {
- arma_extra_debug_sigprint();
-
- const SpProxy<T1> p(x.get_ref());
-
- arma_debug_assert_same_size(n_rows, n_cols, p.get_n_rows(), p.get_n_cols(), "addition");
-
- // Iterate through the sparse subview and add its values.
- typename SpProxy<T1>::const_iterator_type cit = p.begin();
- typename SpProxy<T1>::const_iterator_type cit_end = p.end();
-
- while(cit != cit_end)
- {
- at(cit.row(), cit.col()) += *cit;
- ++cit;
- }
- }
- template<typename eT>
- template<typename T1>
- inline
- void
- subview<eT>::operator-=(const SpBase<eT, T1>& x)
- {
- arma_extra_debug_sigprint();
-
- const SpProxy<T1> p(x.get_ref());
-
- arma_debug_assert_same_size(n_rows, n_cols, p.get_n_rows(), p.get_n_cols(), "subtraction");
-
- // Iterate through the sparse subview and subtract its values.
- typename SpProxy<T1>::const_iterator_type cit = p.begin();
- typename SpProxy<T1>::const_iterator_type cit_end = p.end();
-
- while(cit != cit_end)
- {
- at(cit.row(), cit.col()) -= *cit;
- ++cit;
- }
- }
- template<typename eT>
- template<typename T1>
- inline
- void
- subview<eT>::operator%=(const SpBase<eT, T1>& x)
- {
- arma_extra_debug_sigprint();
-
- const uword s_n_rows = (*this).n_rows;
- const uword s_n_cols = (*this).n_cols;
-
- const SpProxy<T1> p(x.get_ref());
-
- arma_debug_assert_same_size(s_n_rows, s_n_cols, p.get_n_rows(), p.get_n_cols(), "element-wise multiplication");
-
- if(n_elem == 0) { return; }
-
- if(p.get_n_nonzero() == 0) { (*this).zeros(); return; }
-
- // Iterate over nonzero values.
- // Any zero values in the sparse expression will result in a zero in our subview.
- typename SpProxy<T1>::const_iterator_type cit = p.begin();
- typename SpProxy<T1>::const_iterator_type cit_end = p.end();
-
- uword r = 0;
- uword c = 0;
-
- while(cit != cit_end)
- {
- const uword cit_row = cit.row();
- const uword cit_col = cit.col();
-
- while( ((r == cit_row) && (c == cit_col)) == false )
- {
- at(r,c) = eT(0);
-
- r++; if(r >= s_n_rows) { r = 0; c++; }
- }
-
- at(r, c) *= (*cit);
-
- ++cit;
- r++; if(r >= s_n_rows) { r = 0; c++; }
- }
- }
- template<typename eT>
- template<typename T1>
- inline
- void
- subview<eT>::operator/=(const SpBase<eT, T1>& x)
- {
- arma_extra_debug_sigprint();
-
- const SpProxy<T1> p(x.get_ref());
-
- arma_debug_assert_same_size(n_rows, n_cols, p.get_n_rows(), p.get_n_cols(), "element-wise division");
-
- // This is probably going to fill your subview with a bunch of NaNs,
- // so I'm not going to bother to implement it fast.
- // You can have slow NaNs. They're fine too.
- for (uword c = 0; c < n_cols; ++c)
- for (uword r = 0; r < n_rows; ++r)
- {
- at(r, c) /= p.at(r, c);
- }
- }
- template<typename eT>
- template<typename T1, typename gen_type>
- inline
- typename enable_if2< is_same_type<typename T1::elem_type, eT>::value, void>::result
- subview<eT>::operator= (const Gen<T1,gen_type>& in)
- {
- arma_extra_debug_sigprint();
-
- arma_debug_assert_same_size(n_rows, n_cols, in.n_rows, in.n_cols, "copy into submatrix");
-
- in.apply(*this);
- }
- //! apply a functor to each element
- template<typename eT>
- template<typename functor>
- inline
- void
- subview<eT>::for_each(functor F)
- {
- arma_extra_debug_sigprint();
-
- Mat<eT>& X = const_cast< Mat<eT>& >(m);
-
- if(n_rows == 1)
- {
- const uword urow = aux_row1;
- const uword start_col = aux_col1;
- const uword end_col_plus1 = start_col + n_cols;
-
- for(uword ucol = start_col; ucol < end_col_plus1; ++ucol)
- {
- F( X.at(urow, ucol) );
- }
- }
- else
- {
- const uword start_col = aux_col1;
- const uword start_row = aux_row1;
-
- const uword end_col_plus1 = start_col + n_cols;
- const uword end_row_plus1 = start_row + n_rows;
-
- for(uword ucol = start_col; ucol < end_col_plus1; ++ucol)
- for(uword urow = start_row; urow < end_row_plus1; ++urow)
- {
- F( X.at(urow, ucol) );
- }
- }
- }
- template<typename eT>
- template<typename functor>
- inline
- void
- subview<eT>::for_each(functor F) const
- {
- arma_extra_debug_sigprint();
-
- const Mat<eT>& X = m;
-
- if(n_rows == 1)
- {
- const uword urow = aux_row1;
- const uword start_col = aux_col1;
- const uword end_col_plus1 = start_col + n_cols;
-
- for(uword ucol = start_col; ucol < end_col_plus1; ++ucol)
- {
- F( X.at(urow, ucol) );
- }
- }
- else
- {
- const uword start_col = aux_col1;
- const uword start_row = aux_row1;
-
- const uword end_col_plus1 = start_col + n_cols;
- const uword end_row_plus1 = start_row + n_rows;
-
- for(uword ucol = start_col; ucol < end_col_plus1; ++ucol)
- for(uword urow = start_row; urow < end_row_plus1; ++urow)
- {
- F( X.at(urow, ucol) );
- }
- }
- }
- //! transform each element in the subview using a functor
- template<typename eT>
- template<typename functor>
- inline
- void
- subview<eT>::transform(functor F)
- {
- arma_extra_debug_sigprint();
-
- Mat<eT>& X = const_cast< Mat<eT>& >(m);
-
- if(n_rows == 1)
- {
- const uword urow = aux_row1;
- const uword start_col = aux_col1;
- const uword end_col_plus1 = start_col + n_cols;
-
- for(uword ucol = start_col; ucol < end_col_plus1; ++ucol)
- {
- X.at(urow, ucol) = eT( F( X.at(urow, ucol) ) );
- }
- }
- else
- {
- const uword start_col = aux_col1;
- const uword start_row = aux_row1;
-
- const uword end_col_plus1 = start_col + n_cols;
- const uword end_row_plus1 = start_row + n_rows;
-
- for(uword ucol = start_col; ucol < end_col_plus1; ++ucol)
- for(uword urow = start_row; urow < end_row_plus1; ++urow)
- {
- X.at(urow, ucol) = eT( F( X.at(urow, ucol) ) );
- }
- }
- }
- //! imbue (fill) the subview with values provided by a functor
- template<typename eT>
- template<typename functor>
- inline
- void
- subview<eT>::imbue(functor F)
- {
- arma_extra_debug_sigprint();
-
- Mat<eT>& X = const_cast< Mat<eT>& >(m);
-
- if(n_rows == 1)
- {
- const uword urow = aux_row1;
- const uword start_col = aux_col1;
- const uword end_col_plus1 = start_col + n_cols;
-
- for(uword ucol = start_col; ucol < end_col_plus1; ++ucol)
- {
- X.at(urow, ucol) = eT( F() );
- }
- }
- else
- {
- const uword start_col = aux_col1;
- const uword start_row = aux_row1;
-
- const uword end_col_plus1 = start_col + n_cols;
- const uword end_row_plus1 = start_row + n_rows;
-
- for(uword ucol = start_col; ucol < end_col_plus1; ++ucol)
- for(uword urow = start_row; urow < end_row_plus1; ++urow)
- {
- X.at(urow, ucol) = eT( F() );
- }
- }
- }
- template<typename eT>
- inline
- void
- subview<eT>::replace(const eT old_val, const eT new_val)
- {
- arma_extra_debug_sigprint();
-
- subview<eT>& s = *this;
-
- const uword s_n_cols = s.n_cols;
- const uword s_n_rows = s.n_rows;
-
- if(s_n_rows == 1)
- {
- Mat<eT>& A = const_cast< Mat<eT>& >(s.m);
-
- const uword A_n_rows = A.n_rows;
-
- eT* Aptr = &(A.at(s.aux_row1,s.aux_col1));
-
- if(arma_isnan(old_val))
- {
- for(uword ucol=0; ucol < s_n_cols; ++ucol)
- {
- (*Aptr) = (arma_isnan(*Aptr)) ? new_val : (*Aptr);
-
- Aptr += A_n_rows;
- }
- }
- else
- {
- for(uword ucol=0; ucol < s_n_cols; ++ucol)
- {
- (*Aptr) = ((*Aptr) == old_val) ? new_val : (*Aptr);
-
- Aptr += A_n_rows;
- }
- }
- }
- else
- {
- for(uword ucol=0; ucol < s_n_cols; ++ucol)
- {
- arrayops::replace(s.colptr(ucol), s_n_rows, old_val, new_val);
- }
- }
- }
- template<typename eT>
- inline
- void
- subview<eT>::clean(const typename get_pod_type<eT>::result threshold)
- {
- arma_extra_debug_sigprint();
-
- subview<eT>& s = *this;
-
- const uword s_n_cols = s.n_cols;
- const uword s_n_rows = s.n_rows;
-
- for(uword ucol=0; ucol < s_n_cols; ++ucol)
- {
- arrayops::clean( s.colptr(ucol), s_n_rows, threshold );
- }
- }
- template<typename eT>
- inline
- void
- subview<eT>::fill(const eT val)
- {
- arma_extra_debug_sigprint();
-
- subview<eT>& s = *this;
-
- const uword s_n_cols = s.n_cols;
- const uword s_n_rows = s.n_rows;
-
- if(s_n_rows == 1)
- {
- Mat<eT>& A = const_cast< Mat<eT>& >(s.m);
-
- const uword A_n_rows = A.n_rows;
-
- eT* Aptr = &(A.at(s.aux_row1,s.aux_col1));
-
- uword jj;
- for(jj=1; jj < s_n_cols; jj+=2)
- {
- (*Aptr) = val; Aptr += A_n_rows;
- (*Aptr) = val; Aptr += A_n_rows;
- }
-
- if((jj-1) < s_n_cols)
- {
- (*Aptr) = val;
- }
- }
- else
- {
- if( (s.aux_row1 == 0) && (s_n_rows == s.m.n_rows) )
- {
- arrayops::inplace_set( s.colptr(0), val, s.n_elem );
-
- return;
- }
-
- for(uword ucol=0; ucol < s_n_cols; ++ucol)
- {
- arrayops::inplace_set( s.colptr(ucol), val, s_n_rows );
- }
- }
- }
- template<typename eT>
- inline
- void
- subview<eT>::zeros()
- {
- arma_extra_debug_sigprint();
-
- (*this).fill(eT(0));
- }
- template<typename eT>
- inline
- void
- subview<eT>::ones()
- {
- arma_extra_debug_sigprint();
-
- (*this).fill(eT(1));
- }
- template<typename eT>
- inline
- void
- subview<eT>::eye()
- {
- arma_extra_debug_sigprint();
-
- (*this).zeros();
-
- const uword N = (std::min)(n_rows, n_cols);
-
- for(uword ii=0; ii < N; ++ii)
- {
- at(ii,ii) = eT(1);
- }
- }
- template<typename eT>
- inline
- void
- subview<eT>::randu()
- {
- arma_extra_debug_sigprint();
-
- const uword local_n_rows = n_rows;
- const uword local_n_cols = n_cols;
-
- if(local_n_rows == 1)
- {
- for(uword ii=0; ii < local_n_cols; ++ii)
- {
- at(0,ii) = eT(arma_rng::randu<eT>());
- }
- }
- else
- {
- for(uword ii=0; ii < local_n_cols; ++ii)
- {
- arma_rng::randu<eT>::fill( colptr(ii), local_n_rows );
- }
- }
- }
- template<typename eT>
- inline
- void
- subview<eT>::randn()
- {
- arma_extra_debug_sigprint();
-
- const uword local_n_rows = n_rows;
- const uword local_n_cols = n_cols;
-
- if(local_n_rows == 1)
- {
- for(uword ii=0; ii < local_n_cols; ++ii)
- {
- at(0,ii) = eT(arma_rng::randn<eT>());
- }
- }
- else
- {
- for(uword ii=0; ii < local_n_cols; ++ii)
- {
- arma_rng::randn<eT>::fill( colptr(ii), local_n_rows );
- }
- }
- }
- template<typename eT>
- inline
- eT
- subview<eT>::at_alt(const uword ii) const
- {
- return operator[](ii);
- }
- template<typename eT>
- inline
- eT&
- subview<eT>::operator[](const uword ii)
- {
- const uword in_col = ii / n_rows;
- const uword in_row = ii % n_rows;
-
- const uword index = (in_col + aux_col1)*m.n_rows + aux_row1 + in_row;
-
- return access::rw( (const_cast< Mat<eT>& >(m)).mem[index] );
- }
- template<typename eT>
- inline
- eT
- subview<eT>::operator[](const uword ii) const
- {
- const uword in_col = ii / n_rows;
- const uword in_row = ii % n_rows;
-
- const uword index = (in_col + aux_col1)*m.n_rows + aux_row1 + in_row;
-
- return m.mem[index];
- }
- template<typename eT>
- inline
- eT&
- subview<eT>::operator()(const uword ii)
- {
- arma_debug_check( (ii >= n_elem), "subview::operator(): index out of bounds");
-
- const uword in_col = ii / n_rows;
- const uword in_row = ii % n_rows;
-
- const uword index = (in_col + aux_col1)*m.n_rows + aux_row1 + in_row;
-
- return access::rw( (const_cast< Mat<eT>& >(m)).mem[index] );
- }
- template<typename eT>
- inline
- eT
- subview<eT>::operator()(const uword ii) const
- {
- arma_debug_check( (ii >= n_elem), "subview::operator(): index out of bounds");
-
- const uword in_col = ii / n_rows;
- const uword in_row = ii % n_rows;
-
- const uword index = (in_col + aux_col1)*m.n_rows + aux_row1 + in_row;
-
- return m.mem[index];
- }
- template<typename eT>
- inline
- eT&
- subview<eT>::operator()(const uword in_row, const uword in_col)
- {
- arma_debug_check( ((in_row >= n_rows) || (in_col >= n_cols)), "subview::operator(): index out of bounds");
-
- const uword index = (in_col + aux_col1)*m.n_rows + aux_row1 + in_row;
-
- return access::rw( (const_cast< Mat<eT>& >(m)).mem[index] );
- }
- template<typename eT>
- inline
- eT
- subview<eT>::operator()(const uword in_row, const uword in_col) const
- {
- arma_debug_check( ((in_row >= n_rows) || (in_col >= n_cols)), "subview::operator(): index out of bounds");
-
- const uword index = (in_col + aux_col1)*m.n_rows + aux_row1 + in_row;
-
- return m.mem[index];
- }
- template<typename eT>
- inline
- eT&
- subview<eT>::at(const uword in_row, const uword in_col)
- {
- const uword index = (in_col + aux_col1)*m.n_rows + aux_row1 + in_row;
-
- return access::rw( (const_cast< Mat<eT>& >(m)).mem[index] );
- }
- template<typename eT>
- inline
- eT
- subview<eT>::at(const uword in_row, const uword in_col) const
- {
- const uword index = (in_col + aux_col1)*m.n_rows + aux_row1 + in_row;
-
- return m.mem[index];
- }
- template<typename eT>
- inline
- eT&
- subview<eT>::front()
- {
- const uword index = aux_col1*m.n_rows + aux_row1;
-
- return access::rw( (const_cast< Mat<eT>& >(m)).mem[index] );
- }
- template<typename eT>
- inline
- eT
- subview<eT>::front() const
- {
- const uword index = aux_col1*m.n_rows + aux_row1;
-
- return m.mem[index];
- }
- template<typename eT>
- inline
- eT&
- subview<eT>::back()
- {
- const uword in_row = n_rows - 1;
- const uword in_col = n_cols - 1;
-
- const uword index = (in_col + aux_col1)*m.n_rows + aux_row1 + in_row;
-
- return access::rw( (const_cast< Mat<eT>& >(m)).mem[index] );
- }
- template<typename eT>
- inline
- eT
- subview<eT>::back() const
- {
- const uword in_row = n_rows - 1;
- const uword in_col = n_cols - 1;
-
- const uword index = (in_col + aux_col1)*m.n_rows + aux_row1 + in_row;
-
- return m.mem[index];
- }
- template<typename eT>
- arma_inline
- eT*
- subview<eT>::colptr(const uword in_col)
- {
- return & access::rw((const_cast< Mat<eT>& >(m)).mem[ (in_col + aux_col1)*m.n_rows + aux_row1 ]);
- }
- template<typename eT>
- arma_inline
- const eT*
- subview<eT>::colptr(const uword in_col) const
- {
- return & m.mem[ (in_col + aux_col1)*m.n_rows + aux_row1 ];
- }
- template<typename eT>
- template<typename eT2>
- inline
- bool
- subview<eT>::check_overlap(const subview<eT2>& x) const
- {
- if(is_same_type<eT,eT2>::value == false) { return false; }
-
- const subview<eT>& s = (*this);
-
- if(void_ptr(&(s.m)) != void_ptr(&(x.m))) { return false; }
-
- if( (s.n_elem == 0) || (x.n_elem == 0) ) { return false; }
-
- const uword s_row_start = s.aux_row1;
- const uword s_row_end_p1 = s_row_start + s.n_rows;
-
- const uword s_col_start = s.aux_col1;
- const uword s_col_end_p1 = s_col_start + s.n_cols;
-
-
- const uword x_row_start = x.aux_row1;
- const uword x_row_end_p1 = x_row_start + x.n_rows;
-
- const uword x_col_start = x.aux_col1;
- const uword x_col_end_p1 = x_col_start + x.n_cols;
-
-
- const bool outside_rows = ( (x_row_start >= s_row_end_p1) || (s_row_start >= x_row_end_p1) );
- const bool outside_cols = ( (x_col_start >= s_col_end_p1) || (s_col_start >= x_col_end_p1) );
-
- return ( (outside_rows == false) && (outside_cols == false) );
- }
- template<typename eT>
- inline
- arma_warn_unused
- bool
- subview<eT>::is_vec() const
- {
- return ( (n_rows == 1) || (n_cols == 1) );
- }
- template<typename eT>
- inline
- arma_warn_unused
- bool
- subview<eT>::is_finite() const
- {
- arma_extra_debug_sigprint();
-
- const uword local_n_rows = n_rows;
- const uword local_n_cols = n_cols;
-
- for(uword ii=0; ii<local_n_cols; ++ii)
- {
- if(arrayops::is_finite(colptr(ii), local_n_rows) == false) { return false; }
- }
-
- return true;
- }
- template<typename eT>
- inline
- arma_warn_unused
- bool
- subview<eT>::is_zero(const typename get_pod_type<eT>::result tol) const
- {
- arma_extra_debug_sigprint();
-
- const uword local_n_rows = n_rows;
- const uword local_n_cols = n_cols;
-
- for(uword ii=0; ii<local_n_cols; ++ii)
- {
- if(arrayops::is_zero(colptr(ii), local_n_rows, tol) == false) { return false; }
- }
-
- return true;
- }
- template<typename eT>
- inline
- arma_warn_unused
- bool
- subview<eT>::has_inf() const
- {
- arma_extra_debug_sigprint();
-
- const uword local_n_rows = n_rows;
- const uword local_n_cols = n_cols;
-
- for(uword ii=0; ii<local_n_cols; ++ii)
- {
- if(arrayops::has_inf(colptr(ii), local_n_rows)) { return true; }
- }
-
- return false;
- }
- template<typename eT>
- inline
- arma_warn_unused
- bool
- subview<eT>::has_nan() const
- {
- arma_extra_debug_sigprint();
-
- const uword local_n_rows = n_rows;
- const uword local_n_cols = n_cols;
-
- for(uword ii=0; ii<local_n_cols; ++ii)
- {
- if(arrayops::has_nan(colptr(ii), local_n_rows)) { return true; }
- }
-
- return false;
- }
- //! X = Y.submat(...)
- template<typename eT>
- inline
- void
- subview<eT>::extract(Mat<eT>& out, const subview<eT>& in)
- {
- arma_extra_debug_sigprint();
-
- // NOTE: we're assuming that the matrix has already been set to the correct size and there is no aliasing;
- // size setting and alias checking is done by either the Mat contructor or operator=()
-
- const uword n_rows = in.n_rows; // number of rows in the subview
- const uword n_cols = in.n_cols; // number of columns in the subview
-
- arma_extra_debug_print(arma_str::format("out.n_rows = %d out.n_cols = %d in.m.n_rows = %d in.m.n_cols = %d") % out.n_rows % out.n_cols % in.m.n_rows % in.m.n_cols );
-
-
- if(in.is_vec() == true)
- {
- if(n_cols == 1) // a column vector
- {
- arma_extra_debug_print("subview::extract(): copying col (going across rows)");
-
- // in.colptr(0) the first column of the subview, taking into account any row offset
- arrayops::copy( out.memptr(), in.colptr(0), n_rows );
- }
- else // a row vector (possibly empty)
- {
- arma_extra_debug_print("subview::extract(): copying row (going across columns)");
-
- eT* out_mem = out.memptr();
-
- const uword X_n_rows = in.m.n_rows;
-
- const eT* Xptr = &(in.m.at(in.aux_row1,in.aux_col1));
-
- uword j;
-
- for(j=1; j < n_cols; j+=2)
- {
- const eT tmp1 = (*Xptr); Xptr += X_n_rows;
- const eT tmp2 = (*Xptr); Xptr += X_n_rows;
-
- (*out_mem) = tmp1; out_mem++;
- (*out_mem) = tmp2; out_mem++;
- }
-
- if((j-1) < n_cols)
- {
- (*out_mem) = (*Xptr);
- }
- }
- }
- else // general submatrix
- {
- arma_extra_debug_print("subview::extract(): general submatrix");
-
- if( (in.aux_row1 == 0) && (n_rows == in.m.n_rows) )
- {
- arrayops::copy( out.memptr(), in.colptr(0), in.n_elem );
-
- return;
- }
-
- for(uword col=0; col < n_cols; ++col)
- {
- arrayops::copy( out.colptr(col), in.colptr(col), n_rows );
- }
- }
- }
- //! X += Y.submat(...)
- template<typename eT>
- inline
- void
- subview<eT>::plus_inplace(Mat<eT>& out, const subview<eT>& in)
- {
- arma_extra_debug_sigprint();
-
- arma_debug_assert_same_size(out, in, "addition");
-
- const uword n_rows = in.n_rows;
- const uword n_cols = in.n_cols;
-
- if(n_rows == 1)
- {
- eT* out_mem = out.memptr();
-
- const Mat<eT>& X = in.m;
-
- const uword row = in.aux_row1;
- const uword start_col = in.aux_col1;
-
- uword i,j;
- for(i=0, j=1; j < n_cols; i+=2, j+=2)
- {
- const eT tmp1 = X.at(row, start_col+i);
- const eT tmp2 = X.at(row, start_col+j);
-
- out_mem[i] += tmp1;
- out_mem[j] += tmp2;
- }
-
- if(i < n_cols)
- {
- out_mem[i] += X.at(row, start_col+i);
- }
- }
- else
- {
- for(uword col=0; col < n_cols; ++col)
- {
- arrayops::inplace_plus(out.colptr(col), in.colptr(col), n_rows);
- }
- }
- }
- //! X -= Y.submat(...)
- template<typename eT>
- inline
- void
- subview<eT>::minus_inplace(Mat<eT>& out, const subview<eT>& in)
- {
- arma_extra_debug_sigprint();
-
- arma_debug_assert_same_size(out, in, "subtraction");
-
- const uword n_rows = in.n_rows;
- const uword n_cols = in.n_cols;
-
- if(n_rows == 1)
- {
- eT* out_mem = out.memptr();
-
- const Mat<eT>& X = in.m;
-
- const uword row = in.aux_row1;
- const uword start_col = in.aux_col1;
-
- uword i,j;
- for(i=0, j=1; j < n_cols; i+=2, j+=2)
- {
- const eT tmp1 = X.at(row, start_col+i);
- const eT tmp2 = X.at(row, start_col+j);
-
- out_mem[i] -= tmp1;
- out_mem[j] -= tmp2;
- }
-
- if(i < n_cols)
- {
- out_mem[i] -= X.at(row, start_col+i);
- }
- }
- else
- {
- for(uword col=0; col < n_cols; ++col)
- {
- arrayops::inplace_minus(out.colptr(col), in.colptr(col), n_rows);
- }
- }
- }
- //! X %= Y.submat(...)
- template<typename eT>
- inline
- void
- subview<eT>::schur_inplace(Mat<eT>& out, const subview<eT>& in)
- {
- arma_extra_debug_sigprint();
-
- arma_debug_assert_same_size(out, in, "element-wise multiplication");
-
- const uword n_rows = in.n_rows;
- const uword n_cols = in.n_cols;
-
- if(n_rows == 1)
- {
- eT* out_mem = out.memptr();
-
- const Mat<eT>& X = in.m;
-
- const uword row = in.aux_row1;
- const uword start_col = in.aux_col1;
-
- uword i,j;
- for(i=0, j=1; j < n_cols; i+=2, j+=2)
- {
- const eT tmp1 = X.at(row, start_col+i);
- const eT tmp2 = X.at(row, start_col+j);
-
- out_mem[i] *= tmp1;
- out_mem[j] *= tmp2;
- }
-
- if(i < n_cols)
- {
- out_mem[i] *= X.at(row, start_col+i);
- }
- }
- else
- {
- for(uword col=0; col < n_cols; ++col)
- {
- arrayops::inplace_mul(out.colptr(col), in.colptr(col), n_rows);
- }
- }
- }
- //! X /= Y.submat(...)
- template<typename eT>
- inline
- void
- subview<eT>::div_inplace(Mat<eT>& out, const subview<eT>& in)
- {
- arma_extra_debug_sigprint();
-
- arma_debug_assert_same_size(out, in, "element-wise division");
-
- const uword n_rows = in.n_rows;
- const uword n_cols = in.n_cols;
-
- if(n_rows == 1)
- {
- eT* out_mem = out.memptr();
-
- const Mat<eT>& X = in.m;
-
- const uword row = in.aux_row1;
- const uword start_col = in.aux_col1;
-
- uword i,j;
- for(i=0, j=1; j < n_cols; i+=2, j+=2)
- {
- const eT tmp1 = X.at(row, start_col+i);
- const eT tmp2 = X.at(row, start_col+j);
-
- out_mem[i] /= tmp1;
- out_mem[j] /= tmp2;
- }
-
- if(i < n_cols)
- {
- out_mem[i] /= X.at(row, start_col+i);
- }
- }
- else
- {
- for(uword col=0; col < n_cols; ++col)
- {
- arrayops::inplace_div(out.colptr(col), in.colptr(col), n_rows);
- }
- }
- }
- //! creation of subview (row vector)
- template<typename eT>
- inline
- subview_row<eT>
- subview<eT>::row(const uword row_num)
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check( row_num >= n_rows, "subview::row(): out of bounds" );
-
- const uword base_row = aux_row1 + row_num;
-
- return subview_row<eT>(m, base_row, aux_col1, n_cols);
- }
- //! creation of subview (row vector)
- template<typename eT>
- inline
- const subview_row<eT>
- subview<eT>::row(const uword row_num) const
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check( row_num >= n_rows, "subview::row(): out of bounds" );
-
- const uword base_row = aux_row1 + row_num;
-
- return subview_row<eT>(m, base_row, aux_col1, n_cols);
- }
- template<typename eT>
- inline
- subview_row<eT>
- subview<eT>::operator()(const uword row_num, const span& col_span)
- {
- arma_extra_debug_sigprint();
-
- const bool col_all = col_span.whole;
-
- const uword local_n_cols = n_cols;
-
- const uword in_col1 = col_all ? 0 : col_span.a;
- const uword in_col2 = col_span.b;
- const uword submat_n_cols = col_all ? local_n_cols : in_col2 - in_col1 + 1;
-
- const uword base_col1 = aux_col1 + in_col1;
- const uword base_row = aux_row1 + row_num;
-
- arma_debug_check
- (
- (row_num >= n_rows)
- ||
- ( col_all ? false : ((in_col1 > in_col2) || (in_col2 >= local_n_cols)) )
- ,
- "subview::operator(): indices out of bounds or incorrectly used"
- );
-
- return subview_row<eT>(m, base_row, base_col1, submat_n_cols);
- }
- template<typename eT>
- inline
- const subview_row<eT>
- subview<eT>::operator()(const uword row_num, const span& col_span) const
- {
- arma_extra_debug_sigprint();
-
- const bool col_all = col_span.whole;
-
- const uword local_n_cols = n_cols;
-
- const uword in_col1 = col_all ? 0 : col_span.a;
- const uword in_col2 = col_span.b;
- const uword submat_n_cols = col_all ? local_n_cols : in_col2 - in_col1 + 1;
-
- const uword base_col1 = aux_col1 + in_col1;
- const uword base_row = aux_row1 + row_num;
-
- arma_debug_check
- (
- (row_num >= n_rows)
- ||
- ( col_all ? false : ((in_col1 > in_col2) || (in_col2 >= local_n_cols)) )
- ,
- "subview::operator(): indices out of bounds or incorrectly used"
- );
-
- return subview_row<eT>(m, base_row, base_col1, submat_n_cols);
- }
- //! creation of subview (column vector)
- template<typename eT>
- inline
- subview_col<eT>
- subview<eT>::col(const uword col_num)
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check( col_num >= n_cols, "subview::col(): out of bounds");
-
- const uword base_col = aux_col1 + col_num;
-
- return subview_col<eT>(m, base_col, aux_row1, n_rows);
- }
- //! creation of subview (column vector)
- template<typename eT>
- inline
- const subview_col<eT>
- subview<eT>::col(const uword col_num) const
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check( col_num >= n_cols, "subview::col(): out of bounds");
-
- const uword base_col = aux_col1 + col_num;
-
- return subview_col<eT>(m, base_col, aux_row1, n_rows);
- }
- template<typename eT>
- inline
- subview_col<eT>
- subview<eT>::operator()(const span& row_span, const uword col_num)
- {
- arma_extra_debug_sigprint();
-
- const bool row_all = row_span.whole;
-
- const uword local_n_rows = n_rows;
-
- const uword in_row1 = row_all ? 0 : row_span.a;
- const uword in_row2 = row_span.b;
- const uword submat_n_rows = row_all ? local_n_rows : in_row2 - in_row1 + 1;
-
- const uword base_row1 = aux_row1 + in_row1;
- const uword base_col = aux_col1 + col_num;
-
- arma_debug_check
- (
- (col_num >= n_cols)
- ||
- ( row_all ? false : ((in_row1 > in_row2) || (in_row2 >= local_n_rows)) )
- ,
- "subview::operator(): indices out of bounds or incorrectly used"
- );
-
- return subview_col<eT>(m, base_col, base_row1, submat_n_rows);
- }
- template<typename eT>
- inline
- const subview_col<eT>
- subview<eT>::operator()(const span& row_span, const uword col_num) const
- {
- arma_extra_debug_sigprint();
-
- const bool row_all = row_span.whole;
-
- const uword local_n_rows = n_rows;
-
- const uword in_row1 = row_all ? 0 : row_span.a;
- const uword in_row2 = row_span.b;
- const uword submat_n_rows = row_all ? local_n_rows : in_row2 - in_row1 + 1;
-
- const uword base_row1 = aux_row1 + in_row1;
- const uword base_col = aux_col1 + col_num;
-
- arma_debug_check
- (
- (col_num >= n_cols)
- ||
- ( row_all ? false : ((in_row1 > in_row2) || (in_row2 >= local_n_rows)) )
- ,
- "subview::operator(): indices out of bounds or incorrectly used"
- );
-
- return subview_col<eT>(m, base_col, base_row1, submat_n_rows);
- }
- //! create a Col object which uses memory from an existing matrix object.
- //! this approach is currently not alias safe
- //! and does not take into account that the parent matrix object could be deleted.
- //! if deleted memory is accessed by the created Col object,
- //! it will cause memory corruption and/or a crash
- template<typename eT>
- inline
- Col<eT>
- subview<eT>::unsafe_col(const uword col_num)
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check( col_num >= n_cols, "subview::unsafe_col(): out of bounds");
-
- return Col<eT>(colptr(col_num), n_rows, false, true);
- }
- //! create a Col object which uses memory from an existing matrix object.
- //! this approach is currently not alias safe
- //! and does not take into account that the parent matrix object could be deleted.
- //! if deleted memory is accessed by the created Col object,
- //! it will cause memory corruption and/or a crash
- template<typename eT>
- inline
- const Col<eT>
- subview<eT>::unsafe_col(const uword col_num) const
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check( col_num >= n_cols, "subview::unsafe_col(): out of bounds");
-
- return Col<eT>(const_cast<eT*>(colptr(col_num)), n_rows, false, true);
- }
- //! creation of subview (submatrix comprised of specified row vectors)
- template<typename eT>
- inline
- subview<eT>
- subview<eT>::rows(const uword in_row1, const uword in_row2)
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check
- (
- (in_row1 > in_row2) || (in_row2 >= n_rows),
- "subview::rows(): indices out of bounds or incorrectly used"
- );
-
- const uword subview_n_rows = in_row2 - in_row1 + 1;
- const uword base_row1 = aux_row1 + in_row1;
-
- return subview<eT>(m, base_row1, aux_col1, subview_n_rows, n_cols );
- }
- //! creation of subview (submatrix comprised of specified row vectors)
- template<typename eT>
- inline
- const subview<eT>
- subview<eT>::rows(const uword in_row1, const uword in_row2) const
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check
- (
- (in_row1 > in_row2) || (in_row2 >= n_rows),
- "subview::rows(): indices out of bounds or incorrectly used"
- );
-
- const uword subview_n_rows = in_row2 - in_row1 + 1;
- const uword base_row1 = aux_row1 + in_row1;
-
- return subview<eT>(m, base_row1, aux_col1, subview_n_rows, n_cols );
- }
- //! creation of subview (submatrix comprised of specified column vectors)
- template<typename eT>
- inline
- subview<eT>
- subview<eT>::cols(const uword in_col1, const uword in_col2)
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check
- (
- (in_col1 > in_col2) || (in_col2 >= n_cols),
- "subview::cols(): indices out of bounds or incorrectly used"
- );
-
- const uword subview_n_cols = in_col2 - in_col1 + 1;
- const uword base_col1 = aux_col1 + in_col1;
-
- return subview<eT>(m, aux_row1, base_col1, n_rows, subview_n_cols);
- }
- //! creation of subview (submatrix comprised of specified column vectors)
- template<typename eT>
- inline
- const subview<eT>
- subview<eT>::cols(const uword in_col1, const uword in_col2) const
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check
- (
- (in_col1 > in_col2) || (in_col2 >= n_cols),
- "subview::cols(): indices out of bounds or incorrectly used"
- );
-
- const uword subview_n_cols = in_col2 - in_col1 + 1;
- const uword base_col1 = aux_col1 + in_col1;
-
- return subview<eT>(m, aux_row1, base_col1, n_rows, subview_n_cols);
- }
- //! creation of subview (submatrix)
- template<typename eT>
- inline
- subview<eT>
- subview<eT>::submat(const uword in_row1, const uword in_col1, const uword in_row2, const uword in_col2)
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check
- (
- (in_row1 > in_row2) || (in_col1 > in_col2) || (in_row2 >= n_rows) || (in_col2 >= n_cols),
- "subview::submat(): indices out of bounds or incorrectly used"
- );
-
- const uword subview_n_rows = in_row2 - in_row1 + 1;
- const uword subview_n_cols = in_col2 - in_col1 + 1;
-
- const uword base_row1 = aux_row1 + in_row1;
- const uword base_col1 = aux_col1 + in_col1;
-
- return subview<eT>(m, base_row1, base_col1, subview_n_rows, subview_n_cols);
- }
- //! creation of subview (generic submatrix)
- template<typename eT>
- inline
- const subview<eT>
- subview<eT>::submat(const uword in_row1, const uword in_col1, const uword in_row2, const uword in_col2) const
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check
- (
- (in_row1 > in_row2) || (in_col1 > in_col2) || (in_row2 >= n_rows) || (in_col2 >= n_cols),
- "subview::submat(): indices out of bounds or incorrectly used"
- );
-
- const uword subview_n_rows = in_row2 - in_row1 + 1;
- const uword subview_n_cols = in_col2 - in_col1 + 1;
-
- const uword base_row1 = aux_row1 + in_row1;
- const uword base_col1 = aux_col1 + in_col1;
-
- return subview<eT>(m, base_row1, base_col1, subview_n_rows, subview_n_cols);
- }
- //! creation of subview (submatrix)
- template<typename eT>
- inline
- subview<eT>
- subview<eT>::submat(const span& row_span, const span& col_span)
- {
- arma_extra_debug_sigprint();
-
- const bool row_all = row_span.whole;
- const bool col_all = col_span.whole;
-
- const uword local_n_rows = n_rows;
- const uword local_n_cols = n_cols;
-
- const uword in_row1 = row_all ? 0 : row_span.a;
- const uword in_row2 = row_span.b;
- const uword submat_n_rows = row_all ? local_n_rows : in_row2 - in_row1 + 1;
-
- const uword in_col1 = col_all ? 0 : col_span.a;
- const uword in_col2 = col_span.b;
- const uword submat_n_cols = col_all ? local_n_cols : in_col2 - in_col1 + 1;
-
- arma_debug_check
- (
- ( row_all ? false : ((in_row1 > in_row2) || (in_row2 >= local_n_rows)) )
- ||
- ( col_all ? false : ((in_col1 > in_col2) || (in_col2 >= local_n_cols)) )
- ,
- "subview::submat(): indices out of bounds or incorrectly used"
- );
-
- const uword base_row1 = aux_row1 + in_row1;
- const uword base_col1 = aux_col1 + in_col1;
-
- return subview<eT>(m, base_row1, base_col1, submat_n_rows, submat_n_cols);
- }
- //! creation of subview (generic submatrix)
- template<typename eT>
- inline
- const subview<eT>
- subview<eT>::submat(const span& row_span, const span& col_span) const
- {
- arma_extra_debug_sigprint();
-
- const bool row_all = row_span.whole;
- const bool col_all = col_span.whole;
-
- const uword local_n_rows = n_rows;
- const uword local_n_cols = n_cols;
-
- const uword in_row1 = row_all ? 0 : row_span.a;
- const uword in_row2 = row_span.b;
- const uword submat_n_rows = row_all ? local_n_rows : in_row2 - in_row1 + 1;
-
- const uword in_col1 = col_all ? 0 : col_span.a;
- const uword in_col2 = col_span.b;
- const uword submat_n_cols = col_all ? local_n_cols : in_col2 - in_col1 + 1;
-
- arma_debug_check
- (
- ( row_all ? false : ((in_row1 > in_row2) || (in_row2 >= local_n_rows)) )
- ||
- ( col_all ? false : ((in_col1 > in_col2) || (in_col2 >= local_n_cols)) )
- ,
- "subview::submat(): indices out of bounds or incorrectly used"
- );
-
- const uword base_row1 = aux_row1 + in_row1;
- const uword base_col1 = aux_col1 + in_col1;
-
- return subview<eT>(m, base_row1, base_col1, submat_n_rows, submat_n_cols);
- }
- template<typename eT>
- inline
- subview<eT>
- subview<eT>::operator()(const span& row_span, const span& col_span)
- {
- arma_extra_debug_sigprint();
-
- return (*this).submat(row_span, col_span);
- }
- template<typename eT>
- inline
- const subview<eT>
- subview<eT>::operator()(const span& row_span, const span& col_span) const
- {
- arma_extra_debug_sigprint();
-
- return (*this).submat(row_span, col_span);
- }
- template<typename eT>
- inline
- subview_each1< subview<eT>, 0 >
- subview<eT>::each_col()
- {
- arma_extra_debug_sigprint();
-
- return subview_each1< subview<eT>, 0 >(*this);
- }
- template<typename eT>
- inline
- subview_each1< subview<eT>, 1 >
- subview<eT>::each_row()
- {
- arma_extra_debug_sigprint();
-
- return subview_each1< subview<eT>, 1 >(*this);
- }
- template<typename eT>
- template<typename T1>
- inline
- subview_each2< subview<eT>, 0, T1 >
- subview<eT>::each_col(const Base<uword,T1>& indices)
- {
- arma_extra_debug_sigprint();
-
- return subview_each2< subview<eT>, 0, T1 >(*this, indices);
- }
- template<typename eT>
- template<typename T1>
- inline
- subview_each2< subview<eT>, 1, T1 >
- subview<eT>::each_row(const Base<uword,T1>& indices)
- {
- arma_extra_debug_sigprint();
-
- return subview_each2< subview<eT>, 1, T1 >(*this, indices);
- }
- #if defined(ARMA_USE_CXX11)
-
- //! apply a lambda function to each column, where each column is interpreted as a column vector
- template<typename eT>
- inline
- void
- subview<eT>::each_col(const std::function< void(Col<eT>&) >& F)
- {
- arma_extra_debug_sigprint();
-
- for(uword ii=0; ii < n_cols; ++ii)
- {
- Col<eT> tmp(colptr(ii), n_rows, false, true);
- F(tmp);
- }
- }
-
-
-
- template<typename eT>
- inline
- void
- subview<eT>::each_col(const std::function< void(const Col<eT>&) >& F) const
- {
- arma_extra_debug_sigprint();
-
- for(uword ii=0; ii < n_cols; ++ii)
- {
- const Col<eT> tmp(colptr(ii), n_rows, false, true);
- F(tmp);
- }
- }
-
-
-
- //! apply a lambda function to each row, where each row is interpreted as a row vector
- template<typename eT>
- inline
- void
- subview<eT>::each_row(const std::function< void(Row<eT>&) >& F)
- {
- arma_extra_debug_sigprint();
-
- podarray<eT> array1(n_cols);
- podarray<eT> array2(n_cols);
-
- Row<eT> tmp1( array1.memptr(), n_cols, false, true );
- Row<eT> tmp2( array2.memptr(), n_cols, false, true );
-
- eT* tmp1_mem = tmp1.memptr();
- eT* tmp2_mem = tmp2.memptr();
-
- uword ii, jj;
-
- for(ii=0, jj=1; jj < n_rows; ii+=2, jj+=2)
- {
- for(uword col_id = 0; col_id < n_cols; ++col_id)
- {
- const eT* col_mem = colptr(col_id);
-
- tmp1_mem[col_id] = col_mem[ii];
- tmp2_mem[col_id] = col_mem[jj];
- }
-
- F(tmp1);
- F(tmp2);
-
- for(uword col_id = 0; col_id < n_cols; ++col_id)
- {
- eT* col_mem = colptr(col_id);
-
- col_mem[ii] = tmp1_mem[col_id];
- col_mem[jj] = tmp2_mem[col_id];
- }
- }
-
- if(ii < n_rows)
- {
- tmp1 = (*this).row(ii);
-
- F(tmp1);
-
- (*this).row(ii) = tmp1;
- }
- }
-
-
-
- template<typename eT>
- inline
- void
- subview<eT>::each_row(const std::function< void(const Row<eT>&) >& F) const
- {
- arma_extra_debug_sigprint();
-
- podarray<eT> array1(n_cols);
- podarray<eT> array2(n_cols);
-
- Row<eT> tmp1( array1.memptr(), n_cols, false, true );
- Row<eT> tmp2( array2.memptr(), n_cols, false, true );
-
- eT* tmp1_mem = tmp1.memptr();
- eT* tmp2_mem = tmp2.memptr();
-
- uword ii, jj;
-
- for(ii=0, jj=1; jj < n_rows; ii+=2, jj+=2)
- {
- for(uword col_id = 0; col_id < n_cols; ++col_id)
- {
- const eT* col_mem = colptr(col_id);
-
- tmp1_mem[col_id] = col_mem[ii];
- tmp2_mem[col_id] = col_mem[jj];
- }
-
- F(tmp1);
- F(tmp2);
- }
-
- if(ii < n_rows)
- {
- tmp1 = (*this).row(ii);
-
- F(tmp1);
- }
- }
-
- #endif
- //! creation of diagview (diagonal)
- template<typename eT>
- inline
- diagview<eT>
- subview<eT>::diag(const sword in_id)
- {
- arma_extra_debug_sigprint();
-
- const uword row_offset = (in_id < 0) ? uword(-in_id) : 0;
- const uword col_offset = (in_id > 0) ? uword( in_id) : 0;
-
- arma_debug_check
- (
- ((row_offset > 0) && (row_offset >= n_rows)) || ((col_offset > 0) && (col_offset >= n_cols)),
- "subview::diag(): requested diagonal out of bounds"
- );
-
- const uword len = (std::min)(n_rows - row_offset, n_cols - col_offset);
-
- const uword base_row_offset = aux_row1 + row_offset;
- const uword base_col_offset = aux_col1 + col_offset;
-
- return diagview<eT>(m, base_row_offset, base_col_offset, len);
- }
- //! creation of diagview (diagonal)
- template<typename eT>
- inline
- const diagview<eT>
- subview<eT>::diag(const sword in_id) const
- {
- arma_extra_debug_sigprint();
-
- const uword row_offset = uword( (in_id < 0) ? -in_id : 0 );
- const uword col_offset = uword( (in_id > 0) ? in_id : 0 );
-
- arma_debug_check
- (
- ((row_offset > 0) && (row_offset >= n_rows)) || ((col_offset > 0) && (col_offset >= n_cols)),
- "subview::diag(): requested diagonal out of bounds"
- );
-
- const uword len = (std::min)(n_rows - row_offset, n_cols - col_offset);
-
- const uword base_row_offset = aux_row1 + row_offset;
- const uword base_col_offset = aux_col1 + col_offset;
-
- return diagview<eT>(m, base_row_offset, base_col_offset, len);
- }
- template<typename eT>
- inline
- void
- subview<eT>::swap_rows(const uword in_row1, const uword in_row2)
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check
- (
- (in_row1 >= n_rows) || (in_row2 >= n_rows),
- "subview::swap_rows(): out of bounds"
- );
-
- eT* mem = (const_cast< Mat<eT>& >(m)).memptr();
-
- if(n_elem > 0)
- {
- const uword m_n_rows = m.n_rows;
-
- for(uword ucol=0; ucol < n_cols; ++ucol)
- {
- const uword offset = (aux_col1 + ucol) * m_n_rows;
- const uword pos1 = aux_row1 + in_row1 + offset;
- const uword pos2 = aux_row1 + in_row2 + offset;
-
- std::swap( access::rw(mem[pos1]), access::rw(mem[pos2]) );
- }
- }
- }
- template<typename eT>
- inline
- void
- subview<eT>::swap_cols(const uword in_col1, const uword in_col2)
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check
- (
- (in_col1 >= n_cols) || (in_col2 >= n_cols),
- "subview::swap_cols(): out of bounds"
- );
-
- if(n_elem > 0)
- {
- eT* ptr1 = colptr(in_col1);
- eT* ptr2 = colptr(in_col2);
-
- for(uword urow=0; urow < n_rows; ++urow)
- {
- std::swap( ptr1[urow], ptr2[urow] );
- }
- }
- }
- template<typename eT>
- inline
- typename subview<eT>::iterator
- subview<eT>::begin()
- {
- return iterator(*this, aux_row1, aux_col1);
- }
- template<typename eT>
- inline
- typename subview<eT>::const_iterator
- subview<eT>::begin() const
- {
- return const_iterator(*this, aux_row1, aux_col1);
- }
- template<typename eT>
- inline
- typename subview<eT>::const_iterator
- subview<eT>::cbegin() const
- {
- return const_iterator(*this, aux_row1, aux_col1);
- }
- template<typename eT>
- inline
- typename subview<eT>::iterator
- subview<eT>::end()
- {
- return iterator(*this, aux_row1, aux_col1 + n_cols);
- }
- template<typename eT>
- inline
- typename subview<eT>::const_iterator
- subview<eT>::end() const
- {
- return const_iterator(*this, aux_row1, aux_col1 + n_cols);
- }
- template<typename eT>
- inline
- typename subview<eT>::const_iterator
- subview<eT>::cend() const
- {
- return const_iterator(*this, aux_row1, aux_col1 + n_cols);
- }
- //
- //
- //
- template<typename eT>
- inline
- subview<eT>::iterator::iterator()
- : M (NULL)
- , current_ptr(NULL)
- , current_row(0 )
- , current_col(0 )
- , aux_row1 (0 )
- , aux_row2_p1(0 )
- {
- arma_extra_debug_sigprint();
- // Technically this iterator is invalid (it does not point to a valid element)
- }
- template<typename eT>
- inline
- subview<eT>::iterator::iterator(const iterator& X)
- : M (X.M )
- , current_ptr(X.current_ptr)
- , current_row(X.current_row)
- , current_col(X.current_col)
- , aux_row1 (X.aux_row1 )
- , aux_row2_p1(X.aux_row2_p1)
- {
- arma_extra_debug_sigprint();
- }
- template<typename eT>
- inline
- subview<eT>::iterator::iterator(subview<eT>& in_sv, const uword in_row, const uword in_col)
- : M (&(const_cast< Mat<eT>& >(in_sv.m)))
- , current_ptr(&(M->at(in_row,in_col)) )
- , current_row(in_row )
- , current_col(in_col )
- , aux_row1 (in_sv.aux_row1 )
- , aux_row2_p1(in_sv.aux_row1 + in_sv.n_rows )
- {
- arma_extra_debug_sigprint();
- }
- template<typename eT>
- inline
- arma_warn_unused
- eT&
- subview<eT>::iterator::operator*()
- {
- return (*current_ptr);
- }
- template<typename eT>
- inline
- typename subview<eT>::iterator&
- subview<eT>::iterator::operator++()
- {
- current_row++;
-
- if(current_row == aux_row2_p1)
- {
- current_row = aux_row1;
- current_col++;
-
- current_ptr = &( (*M).at(current_row,current_col) );
- }
- else
- {
- current_ptr++;
- }
-
- return *this;
- }
- template<typename eT>
- inline
- arma_warn_unused
- typename subview<eT>::iterator
- subview<eT>::iterator::operator++(int)
- {
- typename subview<eT>::iterator temp(*this);
-
- ++(*this);
-
- return temp;
- }
- template<typename eT>
- inline
- arma_warn_unused
- bool
- subview<eT>::iterator::operator==(const iterator& rhs) const
- {
- return (current_ptr == rhs.current_ptr);
- }
- template<typename eT>
- inline
- arma_warn_unused
- bool
- subview<eT>::iterator::operator!=(const iterator& rhs) const
- {
- return (current_ptr != rhs.current_ptr);
- }
- template<typename eT>
- inline
- arma_warn_unused
- bool
- subview<eT>::iterator::operator==(const const_iterator& rhs) const
- {
- return (current_ptr == rhs.current_ptr);
- }
- template<typename eT>
- inline
- arma_warn_unused
- bool
- subview<eT>::iterator::operator!=(const const_iterator& rhs) const
- {
- return (current_ptr != rhs.current_ptr);
- }
- //
- //
- //
- template<typename eT>
- inline
- subview<eT>::const_iterator::const_iterator()
- : M (NULL)
- , current_ptr(NULL)
- , current_row(0 )
- , current_col(0 )
- , aux_row1 (0 )
- , aux_row2_p1(0 )
- {
- arma_extra_debug_sigprint();
- // Technically this iterator is invalid (it does not point to a valid element)
- }
- template<typename eT>
- inline
- subview<eT>::const_iterator::const_iterator(const iterator& X)
- : M (X.M )
- , current_ptr(X.current_ptr)
- , current_row(X.current_row)
- , current_col(X.current_col)
- , aux_row1 (X.aux_row1 )
- , aux_row2_p1(X.aux_row2_p1)
- {
- arma_extra_debug_sigprint();
- }
- template<typename eT>
- inline
- subview<eT>::const_iterator::const_iterator(const const_iterator& X)
- : M (X.M )
- , current_ptr(X.current_ptr)
- , current_row(X.current_row)
- , current_col(X.current_col)
- , aux_row1 (X.aux_row1 )
- , aux_row2_p1(X.aux_row2_p1)
- {
- arma_extra_debug_sigprint();
- }
- template<typename eT>
- inline
- subview<eT>::const_iterator::const_iterator(const subview<eT>& in_sv, const uword in_row, const uword in_col)
- : M (&(in_sv.m) )
- , current_ptr(&(M->at(in_row,in_col)) )
- , current_row(in_row )
- , current_col(in_col )
- , aux_row1 (in_sv.aux_row1 )
- , aux_row2_p1(in_sv.aux_row1 + in_sv.n_rows)
- {
- arma_extra_debug_sigprint();
- }
- template<typename eT>
- inline
- arma_warn_unused
- const eT&
- subview<eT>::const_iterator::operator*()
- {
- return (*current_ptr);
- }
- template<typename eT>
- inline
- typename subview<eT>::const_iterator&
- subview<eT>::const_iterator::operator++()
- {
- current_row++;
-
- if(current_row == aux_row2_p1)
- {
- current_row = aux_row1;
- current_col++;
-
- current_ptr = &( (*M).at(current_row,current_col) );
- }
- else
- {
- current_ptr++;
- }
-
- return *this;
- }
- template<typename eT>
- inline
- arma_warn_unused
- typename subview<eT>::const_iterator
- subview<eT>::const_iterator::operator++(int)
- {
- typename subview<eT>::const_iterator temp(*this);
-
- ++(*this);
-
- return temp;
- }
- template<typename eT>
- inline
- arma_warn_unused
- bool
- subview<eT>::const_iterator::operator==(const iterator& rhs) const
- {
- return (current_ptr == rhs.current_ptr);
- }
- template<typename eT>
- inline
- arma_warn_unused
- bool
- subview<eT>::const_iterator::operator!=(const iterator& rhs) const
- {
- return (current_ptr != rhs.current_ptr);
- }
- template<typename eT>
- inline
- arma_warn_unused
- bool
- subview<eT>::const_iterator::operator==(const const_iterator& rhs) const
- {
- return (current_ptr == rhs.current_ptr);
- }
- template<typename eT>
- inline
- arma_warn_unused
- bool
- subview<eT>::const_iterator::operator!=(const const_iterator& rhs) const
- {
- return (current_ptr != rhs.current_ptr);
- }
- //
- //
- //
- template<typename eT>
- inline
- subview<eT>::row_iterator::row_iterator()
- : M (NULL)
- , current_row(0 )
- , current_col(0 )
- , aux_col1 (0 )
- , aux_col2_p1(0 )
- {
- arma_extra_debug_sigprint();
- // Technically this iterator is invalid (it does not point to a valid element)
- }
- template<typename eT>
- inline
- subview<eT>::row_iterator::row_iterator(const row_iterator& X)
- : M (X.M )
- , current_row(X.current_row)
- , current_col(X.current_col)
- , aux_col1 (X.aux_col1 )
- , aux_col2_p1(X.aux_col2_p1)
- {
- arma_extra_debug_sigprint();
- }
- template<typename eT>
- inline
- subview<eT>::row_iterator::row_iterator(subview<eT>& in_sv, const uword in_row, const uword in_col)
- : M (&(const_cast< Mat<eT>& >(in_sv.m)))
- , current_row(in_row )
- , current_col(in_col )
- , aux_col1 (in_sv.aux_col1 )
- , aux_col2_p1(in_sv.aux_col1 + in_sv.n_cols )
- {
- arma_extra_debug_sigprint();
- }
- template<typename eT>
- inline
- arma_warn_unused
- eT&
- subview<eT>::row_iterator::operator*()
- {
- return M->at(current_row,current_col);
- }
- template<typename eT>
- inline
- typename subview<eT>::row_iterator&
- subview<eT>::row_iterator::operator++()
- {
- current_col++;
-
- if(current_col == aux_col2_p1)
- {
- current_col = aux_col1;
- current_row++;
- }
-
- return *this;
- }
- template<typename eT>
- inline
- arma_warn_unused
- typename subview<eT>::row_iterator
- subview<eT>::row_iterator::operator++(int)
- {
- typename subview<eT>::row_iterator temp(*this);
-
- ++(*this);
-
- return temp;
- }
- template<typename eT>
- inline
- arma_warn_unused
- bool
- subview<eT>::row_iterator::operator==(const row_iterator& rhs) const
- {
- return ( (current_row == rhs.current_row) && (current_col == rhs.current_col) );
- }
- template<typename eT>
- inline
- arma_warn_unused
- bool
- subview<eT>::row_iterator::operator!=(const row_iterator& rhs) const
- {
- return ( (current_row != rhs.current_row) || (current_col != rhs.current_col) );
- }
- template<typename eT>
- inline
- arma_warn_unused
- bool
- subview<eT>::row_iterator::operator==(const const_row_iterator& rhs) const
- {
- return ( (current_row == rhs.current_row) && (current_col == rhs.current_col) );
- }
- template<typename eT>
- inline
- arma_warn_unused
- bool
- subview<eT>::row_iterator::operator!=(const const_row_iterator& rhs) const
- {
- return ( (current_row != rhs.current_row) || (current_col != rhs.current_col) );
- }
- //
- //
- //
- template<typename eT>
- inline
- subview<eT>::const_row_iterator::const_row_iterator()
- : M (NULL)
- , current_row(0 )
- , current_col(0 )
- , aux_col1 (0 )
- , aux_col2_p1(0 )
- {
- arma_extra_debug_sigprint();
- // Technically this iterator is invalid (it does not point to a valid element)
- }
- template<typename eT>
- inline
- subview<eT>::const_row_iterator::const_row_iterator(const row_iterator& X)
- : M (X.M )
- , current_row(X.current_row)
- , current_col(X.current_col)
- , aux_col1 (X.aux_col1 )
- , aux_col2_p1(X.aux_col2_p1)
- {
- arma_extra_debug_sigprint();
- }
- template<typename eT>
- inline
- subview<eT>::const_row_iterator::const_row_iterator(const const_row_iterator& X)
- : M (X.M )
- , current_row(X.current_row)
- , current_col(X.current_col)
- , aux_col1 (X.aux_col1 )
- , aux_col2_p1(X.aux_col2_p1)
- {
- arma_extra_debug_sigprint();
- }
- template<typename eT>
- inline
- subview<eT>::const_row_iterator::const_row_iterator(const subview<eT>& in_sv, const uword in_row, const uword in_col)
- : M (&(in_sv.m) )
- , current_row(in_row )
- , current_col(in_col )
- , aux_col1 (in_sv.aux_col1 )
- , aux_col2_p1(in_sv.aux_col1 + in_sv.n_cols)
- {
- arma_extra_debug_sigprint();
- }
- template<typename eT>
- inline
- arma_warn_unused
- const eT&
- subview<eT>::const_row_iterator::operator*() const
- {
- return M->at(current_row,current_col);
- }
- template<typename eT>
- inline
- typename subview<eT>::const_row_iterator&
- subview<eT>::const_row_iterator::operator++()
- {
- current_col++;
-
- if(current_col == aux_col2_p1)
- {
- current_col = aux_col1;
- current_row++;
- }
-
- return *this;
- }
- template<typename eT>
- inline
- arma_warn_unused
- typename subview<eT>::const_row_iterator
- subview<eT>::const_row_iterator::operator++(int)
- {
- typename subview<eT>::const_row_iterator temp(*this);
-
- ++(*this);
-
- return temp;
- }
- template<typename eT>
- inline
- arma_warn_unused
- bool
- subview<eT>::const_row_iterator::operator==(const row_iterator& rhs) const
- {
- return ( (current_row == rhs.current_row) && (current_col == rhs.current_col) );
- }
- template<typename eT>
- inline
- arma_warn_unused
- bool
- subview<eT>::const_row_iterator::operator!=(const row_iterator& rhs) const
- {
- return ( (current_row != rhs.current_row) || (current_col != rhs.current_col) );
- }
- template<typename eT>
- inline
- arma_warn_unused
- bool
- subview<eT>::const_row_iterator::operator==(const const_row_iterator& rhs) const
- {
- return ( (current_row == rhs.current_row) && (current_col == rhs.current_col) );
- }
- template<typename eT>
- inline
- arma_warn_unused
- bool
- subview<eT>::const_row_iterator::operator!=(const const_row_iterator& rhs) const
- {
- return ( (current_row != rhs.current_row) || (current_col != rhs.current_col) );
- }
- //
- //
- //
- template<typename eT>
- inline
- subview_col<eT>::subview_col(const Mat<eT>& in_m, const uword in_col)
- : subview<eT>(in_m, 0, in_col, in_m.n_rows, 1)
- , colmem(subview<eT>::colptr(0))
- {
- arma_extra_debug_sigprint();
- }
- template<typename eT>
- inline
- subview_col<eT>::subview_col(const Mat<eT>& in_m, const uword in_col, const uword in_row1, const uword in_n_rows)
- : subview<eT>(in_m, in_row1, in_col, in_n_rows, 1)
- , colmem(subview<eT>::colptr(0))
- {
- arma_extra_debug_sigprint();
- }
- template<typename eT>
- inline
- void
- subview_col<eT>::operator=(const subview<eT>& X)
- {
- arma_extra_debug_sigprint();
-
- subview<eT>::operator=(X);
- }
- template<typename eT>
- inline
- void
- subview_col<eT>::operator=(const subview_col<eT>& X)
- {
- arma_extra_debug_sigprint();
-
- subview<eT>::operator=(X); // interprets 'subview_col' as 'subview'
- }
- template<typename eT>
- inline
- void
- subview_col<eT>::operator=(const eT val)
- {
- arma_extra_debug_sigprint();
-
- if(subview<eT>::n_elem != 1)
- {
- arma_debug_assert_same_size(subview<eT>::n_rows, subview<eT>::n_cols, 1, 1, "copy into submatrix");
- }
-
- access::rw( colmem[0] ) = val;
- }
- template<typename eT>
- template<typename T1>
- inline
- void
- subview_col<eT>::operator=(const Base<eT,T1>& X)
- {
- arma_extra_debug_sigprint();
-
- subview<eT>::operator=(X);
- }
- template<typename eT>
- template<typename T1, typename gen_type>
- inline
- typename enable_if2< is_same_type<typename T1::elem_type, eT>::value, void>::result
- subview_col<eT>::operator= (const Gen<T1,gen_type>& in)
- {
- arma_extra_debug_sigprint();
-
- arma_debug_assert_same_size(subview<eT>::n_rows, uword(1), in.n_rows, (in.is_col ? uword(1) : in.n_cols), "copy into submatrix");
-
- in.apply(*this);
- }
- template<typename eT>
- arma_inline
- const Op<subview_col<eT>,op_htrans>
- subview_col<eT>::t() const
- {
- return Op<subview_col<eT>,op_htrans>(*this);
- }
- template<typename eT>
- arma_inline
- const Op<subview_col<eT>,op_htrans>
- subview_col<eT>::ht() const
- {
- return Op<subview_col<eT>,op_htrans>(*this);
- }
- template<typename eT>
- arma_inline
- const Op<subview_col<eT>,op_strans>
- subview_col<eT>::st() const
- {
- return Op<subview_col<eT>,op_strans>(*this);
- }
- template<typename eT>
- arma_inline
- const Op<subview_col<eT>,op_strans>
- subview_col<eT>::as_row() const
- {
- return Op<subview_col<eT>,op_strans>(*this);
- }
- template<typename eT>
- inline
- void
- subview_col<eT>::fill(const eT val)
- {
- arma_extra_debug_sigprint();
-
- arrayops::inplace_set( access::rwp(colmem), val, subview<eT>::n_rows );
- }
- template<typename eT>
- inline
- void
- subview_col<eT>::zeros()
- {
- arma_extra_debug_sigprint();
-
- arrayops::fill_zeros( access::rwp(colmem), subview<eT>::n_rows );
- }
- template<typename eT>
- inline
- void
- subview_col<eT>::ones()
- {
- arma_extra_debug_sigprint();
-
- arrayops::inplace_set( access::rwp(colmem), eT(1), subview<eT>::n_rows );
- }
- template<typename eT>
- arma_inline
- eT
- subview_col<eT>::at_alt(const uword ii) const
- {
- const eT* colmem_aligned = colmem;
- memory::mark_as_aligned(colmem_aligned);
-
- return colmem_aligned[ii];
- }
- template<typename eT>
- arma_inline
- eT&
- subview_col<eT>::operator[](const uword ii)
- {
- return access::rw( colmem[ii] );
- }
- template<typename eT>
- arma_inline
- eT
- subview_col<eT>::operator[](const uword ii) const
- {
- return colmem[ii];
- }
- template<typename eT>
- inline
- eT&
- subview_col<eT>::operator()(const uword ii)
- {
- arma_debug_check( (ii >= subview<eT>::n_elem), "subview::operator(): index out of bounds");
-
- return access::rw( colmem[ii] );
- }
- template<typename eT>
- inline
- eT
- subview_col<eT>::operator()(const uword ii) const
- {
- arma_debug_check( (ii >= subview<eT>::n_elem), "subview::operator(): index out of bounds");
-
- return colmem[ii];
- }
- template<typename eT>
- inline
- eT&
- subview_col<eT>::operator()(const uword in_row, const uword in_col)
- {
- arma_debug_check( ((in_row >= subview<eT>::n_rows) || (in_col > 0)), "subview::operator(): index out of bounds");
-
- return access::rw( colmem[in_row] );
- }
- template<typename eT>
- inline
- eT
- subview_col<eT>::operator()(const uword in_row, const uword in_col) const
- {
- arma_debug_check( ((in_row >= subview<eT>::n_rows) || (in_col > 0)), "subview::operator(): index out of bounds");
-
- return colmem[in_row];
- }
- template<typename eT>
- inline
- eT&
- subview_col<eT>::at(const uword in_row, const uword)
- {
- return access::rw( colmem[in_row] );
- }
- template<typename eT>
- inline
- eT
- subview_col<eT>::at(const uword in_row, const uword) const
- {
- return colmem[in_row];
- }
- template<typename eT>
- arma_inline
- eT*
- subview_col<eT>::colptr(const uword)
- {
- return const_cast<eT*>(colmem);
- }
-
-
- template<typename eT>
- arma_inline
- const eT*
- subview_col<eT>::colptr(const uword) const
- {
- return colmem;
- }
- template<typename eT>
- inline
- subview_col<eT>
- subview_col<eT>::rows(const uword in_row1, const uword in_row2)
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check( ( (in_row1 > in_row2) || (in_row2 >= subview<eT>::n_rows) ), "subview_col::rows(): indices out of bounds or incorrectly used");
-
- const uword subview_n_rows = in_row2 - in_row1 + 1;
-
- const uword base_row1 = this->aux_row1 + in_row1;
-
- return subview_col<eT>(this->m, this->aux_col1, base_row1, subview_n_rows);
- }
- template<typename eT>
- inline
- const subview_col<eT>
- subview_col<eT>::rows(const uword in_row1, const uword in_row2) const
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check( ( (in_row1 > in_row2) || (in_row2 >= subview<eT>::n_rows) ), "subview_col::rows(): indices out of bounds or incorrectly used");
-
- const uword subview_n_rows = in_row2 - in_row1 + 1;
-
- const uword base_row1 = this->aux_row1 + in_row1;
-
- return subview_col<eT>(this->m, this->aux_col1, base_row1, subview_n_rows);
- }
- template<typename eT>
- inline
- subview_col<eT>
- subview_col<eT>::subvec(const uword in_row1, const uword in_row2)
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check( ( (in_row1 > in_row2) || (in_row2 >= subview<eT>::n_rows) ), "subview_col::subvec(): indices out of bounds or incorrectly used");
-
- const uword subview_n_rows = in_row2 - in_row1 + 1;
-
- const uword base_row1 = this->aux_row1 + in_row1;
-
- return subview_col<eT>(this->m, this->aux_col1, base_row1, subview_n_rows);
- }
- template<typename eT>
- inline
- const subview_col<eT>
- subview_col<eT>::subvec(const uword in_row1, const uword in_row2) const
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check( ( (in_row1 > in_row2) || (in_row2 >= subview<eT>::n_rows) ), "subview_col::subvec(): indices out of bounds or incorrectly used");
-
- const uword subview_n_rows = in_row2 - in_row1 + 1;
-
- const uword base_row1 = this->aux_row1 + in_row1;
-
- return subview_col<eT>(this->m, this->aux_col1, base_row1, subview_n_rows);
- }
- template<typename eT>
- inline
- subview_col<eT>
- subview_col<eT>::subvec(const uword start_row, const SizeMat& s)
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check( (s.n_cols != 1), "subview_col::subvec(): given size does not specify a column vector" );
-
- arma_debug_check( ( (start_row >= subview<eT>::n_rows) || ((start_row + s.n_rows) > subview<eT>::n_rows) ), "subview_col::subvec(): size out of bounds" );
-
- const uword base_row1 = this->aux_row1 + start_row;
-
- return subview_col<eT>(this->m, this->aux_col1, base_row1, s.n_rows);
- }
- template<typename eT>
- inline
- const subview_col<eT>
- subview_col<eT>::subvec(const uword start_row, const SizeMat& s) const
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check( (s.n_cols != 1), "subview_col::subvec(): given size does not specify a column vector" );
-
- arma_debug_check( ( (start_row >= subview<eT>::n_rows) || ((start_row + s.n_rows) > subview<eT>::n_rows) ), "subview_col::subvec(): size out of bounds" );
-
- const uword base_row1 = this->aux_row1 + start_row;
-
- return subview_col<eT>(this->m, this->aux_col1, base_row1, s.n_rows);
- }
- template<typename eT>
- inline
- subview_col<eT>
- subview_col<eT>::head(const uword N)
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check( (N > subview<eT>::n_rows), "subview_col::head(): size out of bounds");
-
- return subview_col<eT>(this->m, this->aux_col1, this->aux_row1, N);
- }
- template<typename eT>
- inline
- const subview_col<eT>
- subview_col<eT>::head(const uword N) const
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check( (N > subview<eT>::n_rows), "subview_col::head(): size out of bounds");
-
- return subview_col<eT>(this->m, this->aux_col1, this->aux_row1, N);
- }
- template<typename eT>
- inline
- subview_col<eT>
- subview_col<eT>::tail(const uword N)
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check( (N > subview<eT>::n_rows), "subview_col::tail(): size out of bounds");
-
- const uword start_row = subview<eT>::aux_row1 + subview<eT>::n_rows - N;
-
- return subview_col<eT>(this->m, this->aux_col1, start_row, N);
- }
- template<typename eT>
- inline
- const subview_col<eT>
- subview_col<eT>::tail(const uword N) const
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check( (N > subview<eT>::n_rows), "subview_col::tail(): size out of bounds");
-
- const uword start_row = subview<eT>::aux_row1 + subview<eT>::n_rows - N;
-
- return subview_col<eT>(this->m, this->aux_col1, start_row, N);
- }
- template<typename eT>
- inline
- arma_warn_unused
- eT
- subview_col<eT>::min() const
- {
- arma_extra_debug_sigprint();
-
- if(subview<eT>::n_elem == 0)
- {
- arma_debug_check(true, "min(): object has no elements");
-
- return Datum<eT>::nan;
- }
-
- return op_min::direct_min(colmem, subview<eT>::n_elem);
- }
- template<typename eT>
- inline
- arma_warn_unused
- eT
- subview_col<eT>::max() const
- {
- arma_extra_debug_sigprint();
-
- if(subview<eT>::n_elem == 0)
- {
- arma_debug_check(true, "max(): object has no elements");
-
- return Datum<eT>::nan;
- }
-
- return op_max::direct_max(colmem, subview<eT>::n_elem);
- }
- template<typename eT>
- inline
- eT
- subview_col<eT>::min(uword& index_of_min_val) const
- {
- arma_extra_debug_sigprint();
-
- if(subview<eT>::n_elem == 0)
- {
- arma_debug_check(true, "min(): object has no elements");
-
- index_of_min_val = uword(0);
-
- return Datum<eT>::nan;
- }
- else
- {
- return op_min::direct_min(colmem, subview<eT>::n_elem, index_of_min_val);
- }
- }
- template<typename eT>
- inline
- eT
- subview_col<eT>::max(uword& index_of_max_val) const
- {
- arma_extra_debug_sigprint();
-
- if(subview<eT>::n_elem == 0)
- {
- arma_debug_check(true, "max(): object has no elements");
-
- index_of_max_val = uword(0);
-
- return Datum<eT>::nan;
- }
- else
- {
- return op_max::direct_max(colmem, subview<eT>::n_elem, index_of_max_val);
- }
- }
- template<typename eT>
- inline
- arma_warn_unused
- uword
- subview_col<eT>::index_min() const
- {
- arma_extra_debug_sigprint();
-
- uword index = 0;
-
- if(subview<eT>::n_elem == 0)
- {
- arma_debug_check(true, "index_min(): object has no elements");
- }
- else
- {
- op_min::direct_min(colmem, subview<eT>::n_elem, index);
- }
-
- return index;
- }
- template<typename eT>
- inline
- arma_warn_unused
- uword
- subview_col<eT>::index_max() const
- {
- arma_extra_debug_sigprint();
-
- uword index = 0;
-
- if(subview<eT>::n_elem == 0)
- {
- arma_debug_check(true, "index_max(): object has no elements");
- }
- else
- {
- op_max::direct_max(colmem, subview<eT>::n_elem, index);
- }
-
- return index;
- }
- //
- //
- //
- template<typename eT>
- inline
- subview_row<eT>::subview_row(const Mat<eT>& in_m, const uword in_row)
- : subview<eT>(in_m, in_row, 0, 1, in_m.n_cols)
- {
- arma_extra_debug_sigprint();
- }
- template<typename eT>
- inline
- subview_row<eT>::subview_row(const Mat<eT>& in_m, const uword in_row, const uword in_col1, const uword in_n_cols)
- : subview<eT>(in_m, in_row, in_col1, 1, in_n_cols)
- {
- arma_extra_debug_sigprint();
- }
- template<typename eT>
- inline
- void
- subview_row<eT>::operator=(const subview<eT>& X)
- {
- arma_extra_debug_sigprint();
-
- subview<eT>::operator=(X);
- }
- template<typename eT>
- inline
- void
- subview_row<eT>::operator=(const subview_row<eT>& X)
- {
- arma_extra_debug_sigprint();
-
- subview<eT>::operator=(X); // interprets 'subview_row' as 'subview'
- }
- template<typename eT>
- inline
- void
- subview_row<eT>::operator=(const eT val)
- {
- arma_extra_debug_sigprint();
-
- subview<eT>::operator=(val); // interprets 'subview_row' as 'subview'
- }
- template<typename eT>
- template<typename T1>
- inline
- void
- subview_row<eT>::operator=(const Base<eT,T1>& X)
- {
- arma_extra_debug_sigprint();
-
- subview<eT>::operator=(X);
- }
- template<typename eT>
- template<typename T1, typename gen_type>
- inline
- typename enable_if2< is_same_type<typename T1::elem_type, eT>::value, void>::result
- subview_row<eT>::operator= (const Gen<T1,gen_type>& in)
- {
- arma_extra_debug_sigprint();
-
- arma_debug_assert_same_size(uword(1), subview<eT>::n_cols, (in.is_row ? uword(1) : in.n_rows), in.n_cols, "copy into submatrix");
-
- in.apply(*this);
- }
- template<typename eT>
- arma_inline
- const Op<subview_row<eT>,op_htrans>
- subview_row<eT>::t() const
- {
- return Op<subview_row<eT>,op_htrans>(*this);
- }
- template<typename eT>
- arma_inline
- const Op<subview_row<eT>,op_htrans>
- subview_row<eT>::ht() const
- {
- return Op<subview_row<eT>,op_htrans>(*this);
- }
- template<typename eT>
- arma_inline
- const Op<subview_row<eT>,op_strans>
- subview_row<eT>::st() const
- {
- return Op<subview_row<eT>,op_strans>(*this);
- }
- template<typename eT>
- arma_inline
- const Op<subview_row<eT>,op_strans>
- subview_row<eT>::as_col() const
- {
- return Op<subview_row<eT>,op_strans>(*this);
- }
- template<typename eT>
- inline
- eT
- subview_row<eT>::at_alt(const uword ii) const
- {
- const uword index = (ii + (subview<eT>::aux_col1))*(subview<eT>::m).n_rows + (subview<eT>::aux_row1);
-
- return subview<eT>::m.mem[index];
- }
- template<typename eT>
- inline
- eT&
- subview_row<eT>::operator[](const uword ii)
- {
- const uword index = (ii + (subview<eT>::aux_col1))*(subview<eT>::m).n_rows + (subview<eT>::aux_row1);
-
- return access::rw( (const_cast< Mat<eT>& >(subview<eT>::m)).mem[index] );
- }
- template<typename eT>
- inline
- eT
- subview_row<eT>::operator[](const uword ii) const
- {
- const uword index = (ii + (subview<eT>::aux_col1))*(subview<eT>::m).n_rows + (subview<eT>::aux_row1);
-
- return subview<eT>::m.mem[index];
- }
- template<typename eT>
- inline
- eT&
- subview_row<eT>::operator()(const uword ii)
- {
- arma_debug_check( (ii >= subview<eT>::n_elem), "subview::operator(): index out of bounds");
-
- const uword index = (ii + (subview<eT>::aux_col1))*(subview<eT>::m).n_rows + (subview<eT>::aux_row1);
-
- return access::rw( (const_cast< Mat<eT>& >(subview<eT>::m)).mem[index] );
- }
- template<typename eT>
- inline
- eT
- subview_row<eT>::operator()(const uword ii) const
- {
- arma_debug_check( (ii >= subview<eT>::n_elem), "subview::operator(): index out of bounds");
-
- const uword index = (ii + (subview<eT>::aux_col1))*(subview<eT>::m).n_rows + (subview<eT>::aux_row1);
-
- return subview<eT>::m.mem[index];
- }
- template<typename eT>
- inline
- eT&
- subview_row<eT>::operator()(const uword in_row, const uword in_col)
- {
- arma_debug_check( ((in_row > 0) || (in_col >= subview<eT>::n_cols)), "subview::operator(): index out of bounds");
-
- const uword index = (in_col + (subview<eT>::aux_col1))*(subview<eT>::m).n_rows + (subview<eT>::aux_row1);
-
- return access::rw( (const_cast< Mat<eT>& >(subview<eT>::m)).mem[index] );
- }
- template<typename eT>
- inline
- eT
- subview_row<eT>::operator()(const uword in_row, const uword in_col) const
- {
- arma_debug_check( ((in_row > 0) || (in_col >= subview<eT>::n_cols)), "subview::operator(): index out of bounds");
-
- const uword index = (in_col + (subview<eT>::aux_col1))*(subview<eT>::m).n_rows + (subview<eT>::aux_row1);
-
- return subview<eT>::m.mem[index];
- }
- template<typename eT>
- inline
- eT&
- subview_row<eT>::at(const uword, const uword in_col)
- {
- const uword index = (in_col + (subview<eT>::aux_col1))*(subview<eT>::m).n_rows + (subview<eT>::aux_row1);
-
- return access::rw( (const_cast< Mat<eT>& >(subview<eT>::m)).mem[index] );
- }
- template<typename eT>
- inline
- eT
- subview_row<eT>::at(const uword, const uword in_col) const
- {
- const uword index = (in_col + (subview<eT>::aux_col1))*(subview<eT>::m).n_rows + (subview<eT>::aux_row1);
-
- return subview<eT>::m.mem[index];
- }
- template<typename eT>
- inline
- subview_row<eT>
- subview_row<eT>::cols(const uword in_col1, const uword in_col2)
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check( ( (in_col1 > in_col2) || (in_col2 >= subview<eT>::n_cols) ), "subview_row::cols(): indices out of bounds or incorrectly used" );
-
- const uword subview_n_cols = in_col2 - in_col1 + 1;
-
- const uword base_col1 = this->aux_col1 + in_col1;
-
- return subview_row<eT>(this->m, this->aux_row1, base_col1, subview_n_cols);
- }
- template<typename eT>
- inline
- const subview_row<eT>
- subview_row<eT>::cols(const uword in_col1, const uword in_col2) const
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check( ( (in_col1 > in_col2) || (in_col2 >= subview<eT>::n_cols) ), "subview_row::cols(): indices out of bounds or incorrectly used");
-
- const uword subview_n_cols = in_col2 - in_col1 + 1;
-
- const uword base_col1 = this->aux_col1 + in_col1;
-
- return subview_row<eT>(this->m, this->aux_row1, base_col1, subview_n_cols);
- }
- template<typename eT>
- inline
- subview_row<eT>
- subview_row<eT>::subvec(const uword in_col1, const uword in_col2)
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check( ( (in_col1 > in_col2) || (in_col2 >= subview<eT>::n_cols) ), "subview_row::subvec(): indices out of bounds or incorrectly used");
-
- const uword subview_n_cols = in_col2 - in_col1 + 1;
-
- const uword base_col1 = this->aux_col1 + in_col1;
-
- return subview_row<eT>(this->m, this->aux_row1, base_col1, subview_n_cols);
- }
- template<typename eT>
- inline
- const subview_row<eT>
- subview_row<eT>::subvec(const uword in_col1, const uword in_col2) const
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check( ( (in_col1 > in_col2) || (in_col2 >= subview<eT>::n_cols) ), "subview_row::subvec(): indices out of bounds or incorrectly used");
-
- const uword subview_n_cols = in_col2 - in_col1 + 1;
-
- const uword base_col1 = this->aux_col1 + in_col1;
-
- return subview_row<eT>(this->m, this->aux_row1, base_col1, subview_n_cols);
- }
- template<typename eT>
- inline
- subview_row<eT>
- subview_row<eT>::subvec(const uword start_col, const SizeMat& s)
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check( (s.n_rows != 1), "subview_row::subvec(): given size does not specify a row vector" );
-
- arma_debug_check( ( (start_col >= subview<eT>::n_cols) || ((start_col + s.n_cols) > subview<eT>::n_cols) ), "subview_row::subvec(): size out of bounds" );
-
- const uword base_col1 = this->aux_col1 + start_col;
-
- return subview_row<eT>(this->m, this->aux_row1, base_col1, s.n_cols);
- }
- template<typename eT>
- inline
- const subview_row<eT>
- subview_row<eT>::subvec(const uword start_col, const SizeMat& s) const
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check( (s.n_rows != 1), "subview_row::subvec(): given size does not specify a row vector" );
-
- arma_debug_check( ( (start_col >= subview<eT>::n_cols) || ((start_col + s.n_cols) > subview<eT>::n_cols) ), "subview_row::subvec(): size out of bounds" );
-
- const uword base_col1 = this->aux_col1 + start_col;
-
- return subview_row<eT>(this->m, this->aux_row1, base_col1, s.n_cols);
- }
- template<typename eT>
- inline
- subview_row<eT>
- subview_row<eT>::head(const uword N)
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check( (N > subview<eT>::n_cols), "subview_row::head(): size out of bounds");
-
- return subview_row<eT>(this->m, this->aux_row1, this->aux_col1, N);
- }
- template<typename eT>
- inline
- const subview_row<eT>
- subview_row<eT>::head(const uword N) const
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check( (N > subview<eT>::n_cols), "subview_row::head(): size out of bounds");
-
- return subview_row<eT>(this->m, this->aux_row1, this->aux_col1, N);
- }
- template<typename eT>
- inline
- subview_row<eT>
- subview_row<eT>::tail(const uword N)
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check( (N > subview<eT>::n_cols), "subview_row::tail(): size out of bounds");
-
- const uword start_col = subview<eT>::aux_col1 + subview<eT>::n_cols - N;
-
- return subview_row<eT>(this->m, this->aux_row1, start_col, N);
- }
- template<typename eT>
- inline
- const subview_row<eT>
- subview_row<eT>::tail(const uword N) const
- {
- arma_extra_debug_sigprint();
-
- arma_debug_check( (N > subview<eT>::n_cols), "subview_row::tail(): size out of bounds");
-
- const uword start_col = subview<eT>::aux_col1 + subview<eT>::n_cols - N;
-
- return subview_row<eT>(this->m, this->aux_row1, start_col, N);
- }
- template<typename eT>
- inline
- arma_warn_unused
- uword
- subview_row<eT>::index_min() const
- {
- const Proxy< subview_row<eT> > P(*this);
-
- uword index = 0;
-
- if(P.get_n_elem() == 0)
- {
- arma_debug_check(true, "index_min(): object has no elements");
- }
- else
- {
- op_min::min_with_index(P, index);
- }
-
- return index;
- }
- template<typename eT>
- inline
- arma_warn_unused
- uword
- subview_row<eT>::index_max() const
- {
- const Proxy< subview_row<eT> > P(*this);
-
- uword index = 0;
-
- if(P.get_n_elem() == 0)
- {
- arma_debug_check(true, "index_max(): object has no elements");
- }
- else
- {
- op_max::max_with_index(P, index);
- }
-
- return index;
- }
- template<typename eT>
- inline
- typename subview<eT>::row_iterator
- subview_row<eT>::begin()
- {
- return typename subview<eT>::row_iterator(*this, subview<eT>::aux_row1, subview<eT>::aux_col1);
- }
- template<typename eT>
- inline
- typename subview<eT>::const_row_iterator
- subview_row<eT>::begin() const
- {
- return typename subview<eT>::const_row_iterator(*this, subview<eT>::aux_row1, subview<eT>::aux_col1);
- }
- template<typename eT>
- inline
- typename subview<eT>::const_row_iterator
- subview_row<eT>::cbegin() const
- {
- return typename subview<eT>::const_row_iterator(*this, subview<eT>::aux_row1, subview<eT>::aux_col1);
- }
- template<typename eT>
- inline
- typename subview<eT>::row_iterator
- subview_row<eT>::end()
- {
- return typename subview<eT>::row_iterator(*this, subview<eT>::aux_row1 + subview<eT>::n_rows, subview<eT>::aux_col1);
- }
- template<typename eT>
- inline
- typename subview<eT>::const_row_iterator
- subview_row<eT>::end() const
- {
- return typename subview<eT>::const_row_iterator(*this, subview<eT>::aux_row1 + subview<eT>::n_rows, subview<eT>::aux_col1);
- }
- template<typename eT>
- inline
- typename subview<eT>::const_row_iterator
- subview_row<eT>::cend() const
- {
- return typename subview<eT>::const_row_iterator(*this, subview<eT>::aux_row1 + subview<eT>::n_rows, subview<eT>::aux_col1);
- }
- //
- //
- //
- template<typename eT>
- inline
- subview_row_strans<eT>::subview_row_strans(const subview_row<eT>& in_sv_row)
- : sv_row(in_sv_row )
- , n_rows(in_sv_row.n_cols)
- , n_elem(in_sv_row.n_elem)
- {
- arma_extra_debug_sigprint();
- }
- template<typename eT>
- inline
- void
- subview_row_strans<eT>::extract(Mat<eT>& out) const
- {
- arma_extra_debug_sigprint();
-
- // NOTE: this function assumes that matrix 'out' has already been set to the correct size
-
- const Mat<eT>& X = sv_row.m;
-
- eT* out_mem = out.memptr();
-
- const uword row = sv_row.aux_row1;
- const uword start_col = sv_row.aux_col1;
- const uword sv_row_n_cols = sv_row.n_cols;
-
- uword ii,jj;
-
- for(ii=0, jj=1; jj < sv_row_n_cols; ii+=2, jj+=2)
- {
- const eT tmp1 = X.at(row, start_col+ii);
- const eT tmp2 = X.at(row, start_col+jj);
-
- out_mem[ii] = tmp1;
- out_mem[jj] = tmp2;
- }
-
- if(ii < sv_row_n_cols)
- {
- out_mem[ii] = X.at(row, start_col+ii);
- }
- }
- template<typename eT>
- inline
- eT
- subview_row_strans<eT>::at_alt(const uword ii) const
- {
- return sv_row[ii];
- }
- template<typename eT>
- inline
- eT
- subview_row_strans<eT>::operator[](const uword ii) const
- {
- return sv_row[ii];
- }
- template<typename eT>
- inline
- eT
- subview_row_strans<eT>::operator()(const uword ii) const
- {
- return sv_row(ii);
- }
- template<typename eT>
- inline
- eT
- subview_row_strans<eT>::operator()(const uword in_row, const uword in_col) const
- {
- return sv_row(in_col, in_row); // deliberately swapped
- }
- template<typename eT>
- inline
- eT
- subview_row_strans<eT>::at(const uword in_row, const uword) const
- {
- return sv_row.at(0, in_row); // deliberately swapped
- }
- //
- //
- //
- template<typename eT>
- inline
- subview_row_htrans<eT>::subview_row_htrans(const subview_row<eT>& in_sv_row)
- : sv_row(in_sv_row )
- , n_rows(in_sv_row.n_cols)
- , n_elem(in_sv_row.n_elem)
- {
- arma_extra_debug_sigprint();
- }
- template<typename eT>
- inline
- void
- subview_row_htrans<eT>::extract(Mat<eT>& out) const
- {
- arma_extra_debug_sigprint();
-
- // NOTE: this function assumes that matrix 'out' has already been set to the correct size
-
- const Mat<eT>& X = sv_row.m;
-
- eT* out_mem = out.memptr();
-
- const uword row = sv_row.aux_row1;
- const uword start_col = sv_row.aux_col1;
- const uword sv_row_n_cols = sv_row.n_cols;
-
- for(uword ii=0; ii < sv_row_n_cols; ++ii)
- {
- out_mem[ii] = access::alt_conj( X.at(row, start_col+ii) );
- }
- }
- template<typename eT>
- inline
- eT
- subview_row_htrans<eT>::at_alt(const uword ii) const
- {
- return access::alt_conj( sv_row[ii] );
- }
- template<typename eT>
- inline
- eT
- subview_row_htrans<eT>::operator[](const uword ii) const
- {
- return access::alt_conj( sv_row[ii] );
- }
- template<typename eT>
- inline
- eT
- subview_row_htrans<eT>::operator()(const uword ii) const
- {
- return access::alt_conj( sv_row(ii) );
- }
- template<typename eT>
- inline
- eT
- subview_row_htrans<eT>::operator()(const uword in_row, const uword in_col) const
- {
- return access::alt_conj( sv_row(in_col, in_row) ); // deliberately swapped
- }
- template<typename eT>
- inline
- eT
- subview_row_htrans<eT>::at(const uword in_row, const uword) const
- {
- return access::alt_conj( sv_row.at(0, in_row) ); // deliberately swapped
- }
- //! @}
|