1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288 |
- #ifndef ASSIGNMENT_HPP
- #define ASSIGNMENT_HPP
- #include <boost/numeric/ublas/vector_expression.hpp>
- #include <boost/numeric/ublas/matrix_expression.hpp>
- namespace boost { namespace numeric { namespace ublas {
- template <class TV>
- class index_manipulator {
- public:
- typedef TV type;
- BOOST_UBLAS_INLINE
- const type &operator () () const {
- return *static_cast<const type *> (this);
- }
- BOOST_UBLAS_INLINE
- type &operator () () {
- return *static_cast<type *> (this);
- }
- };
- template <typename T>
- class vector_move_to_manip: public index_manipulator<vector_move_to_manip<T> > {
- public:
- BOOST_UBLAS_INLINE
- vector_move_to_manip(const T &k): i(k) { }
- template <typename V>
- BOOST_UBLAS_INLINE
- void manip(V &k) const { k=i; }
- private:
- T i;
- };
- template <typename T>
- BOOST_UBLAS_INLINE vector_move_to_manip<T> move_to(T i) {
- return vector_move_to_manip<T>(i);
- }
- template <std::size_t I>
- class static_vector_move_to_manip: public index_manipulator<static_vector_move_to_manip<I> > {
- public:
- template <typename V>
- BOOST_UBLAS_INLINE
- void manip(V &k) const { k=I; }
- };
- template <std::size_t I>
- BOOST_UBLAS_INLINE static_vector_move_to_manip<I> move_to() {
- return static_vector_move_to_manip<I>();
- }
- template <typename T>
- class vector_move_manip: public index_manipulator<vector_move_manip<T> > {
- public:
- BOOST_UBLAS_INLINE
- vector_move_manip(const T &k): i(k) { }
- template <typename V>
- BOOST_UBLAS_INLINE void manip(V &k) const { k+=i; }
- private:
- T i;
- };
- template <typename T>
- BOOST_UBLAS_INLINE vector_move_manip<T> move(T i) {
- return vector_move_manip<T>(i);
- }
- template <std::ptrdiff_t I>
- class static_vector_move_manip: public index_manipulator<static_vector_move_manip<I> > {
- public:
- template <typename V>
- BOOST_UBLAS_INLINE void manip(V &k) const {
-
-
- k = k + I;
- }
- };
- template <std::ptrdiff_t I>
- static_vector_move_manip<I> move() {
- return static_vector_move_manip<I>();
- }
- template <typename T>
- class matrix_move_to_manip: public index_manipulator<matrix_move_to_manip<T> > {
- public:
- BOOST_UBLAS_INLINE
- matrix_move_to_manip(T k, T l): i(k), j(l) { }
- template <typename V1, typename V2>
- BOOST_UBLAS_INLINE
- void manip(V1 &k, V2 &l) const {
- k=i;
- l=j;
- }
- private:
- T i, j;
- };
- template <typename T>
- BOOST_UBLAS_INLINE matrix_move_to_manip<T> move_to(T i, T j) {
- return matrix_move_to_manip<T>(i, j);
- }
- template <std::size_t I,std::size_t J>
- class static_matrix_move_to_manip: public index_manipulator<static_matrix_move_to_manip<I, J> > {
- public:
- template <typename V, typename K>
- BOOST_UBLAS_INLINE
- void manip(V &k, K &l) const {
- k=I;
- l=J;
- }
- };
- template <std::size_t I, std::size_t J>
- BOOST_UBLAS_INLINE static_matrix_move_to_manip<I, J> move_to() {
- return static_matrix_move_to_manip<I, J>();
- }
- template <typename T>
- class matrix_move_manip: public index_manipulator<matrix_move_manip<T> > {
- public:
- BOOST_UBLAS_INLINE
- matrix_move_manip(T k, T l): i(k), j(l) { }
- template <typename V, typename K>
- BOOST_UBLAS_INLINE
- void manip(V &k, K &l) const {
- k+=i;
- l+=j;
- }
- private:
- T i, j;
- };
- template <typename T>
- BOOST_UBLAS_INLINE matrix_move_manip<T> move(T i, T j) {
- return matrix_move_manip<T>(i, j);
- }
- template <std::ptrdiff_t I, std::ptrdiff_t J>
- class static_matrix_move_manip: public index_manipulator<static_matrix_move_manip<I, J> > {
- public:
- template <typename V, typename K>
- BOOST_UBLAS_INLINE
- void manip(V &k, K &l) const {
-
-
- k = k + I;
- l = l + J;
- }
- };
- template <std::ptrdiff_t I, std::ptrdiff_t J>
- BOOST_UBLAS_INLINE static_matrix_move_manip<I, J> move() {
- return static_matrix_move_manip<I, J>();
- }
- class begin1_manip: public index_manipulator<begin1_manip > {
- public:
- template <typename V, typename K>
- BOOST_UBLAS_INLINE
- void manip(V & k, K &) const {
- k=0;
- }
- };
- inline begin1_manip begin1() {
- return begin1_manip();
- }
- class begin2_manip: public index_manipulator<begin2_manip > {
- public:
- template <typename V, typename K>
- BOOST_UBLAS_INLINE
- void manip(V &, K &l) const {
- l=0;
- }
- };
- inline begin2_manip begin2() {
- return begin2_manip();
- }
- class next_row_manip: public index_manipulator<next_row_manip> {
- public:
- template <typename V, typename K>
- BOOST_UBLAS_INLINE
- void manip(V &k, K &l) const {
- k++;
- l=0;
- }
- };
- inline next_row_manip next_row() {
- return next_row_manip();
- }
- class next_column_manip: public index_manipulator<next_column_manip> {
- public:
- template <typename V, typename K>
- BOOST_UBLAS_INLINE
- void manip(V &k, K &l) const {
- k=0;
- l++;
- }
- };
- inline next_column_manip next_column() {
- return next_column_manip();
- }
- template <class T>
- class fill_policy_wrapper {
- public:
- typedef T type;
- };
- namespace fill_policy {
-
- class index_assign :public fill_policy_wrapper<index_assign> {
- public:
- template <class T, typename S, typename V>
- BOOST_UBLAS_INLINE
- static void apply(T &e, const S &i, const V &v) {
- e()(i) = v;
- }
- template <class T, typename S, typename V>
- BOOST_UBLAS_INLINE
- static void apply(T &e, const S &i, const S &j, const V &v) {
- e()(i, j) = v;
- }
- };
-
- class index_plus_assign :public fill_policy_wrapper<index_plus_assign> {
- public:
- template <class T, typename S, typename V>
- BOOST_UBLAS_INLINE
- static void apply(T &e, const S &i, const V &v) {
- e()(i) += v;
- }
- template <class T, typename S, typename V>
- BOOST_UBLAS_INLINE
- static void apply(T &e, const S &i, const S &j, const V &v) {
- e()(i, j) += v;
- }
- };
-
- class index_minus_assign :public fill_policy_wrapper<index_minus_assign> {
- public:
- template <class T, typename S, typename V>
- BOOST_UBLAS_INLINE
- static void apply(T &e, const S &i, const V &v) {
- e()(i) -= v;
- }
- template <class T, typename S, typename V>
- BOOST_UBLAS_INLINE
- static void apply(T &e, const S &i, const S &j, const V &v) {
- e()(i, j) -= v;
- }
- };
-
- class sparse_push_back :public fill_policy_wrapper<sparse_push_back > {
- public:
- template <class T, class S, class V>
- BOOST_UBLAS_INLINE
- static void apply(T &e, const S &i, const V &v) {
- e().push_back(i, v);
- }
- template <class T, class S, class V>
- BOOST_UBLAS_INLINE
- static void apply(T &e, const S &i, const S &j, const V &v) {
- e().push_back(i,j, v);
- }
- };
-
- class sparse_insert :public fill_policy_wrapper<sparse_insert> {
- public:
- template <class T, class S, class V>
- BOOST_UBLAS_INLINE
- static void apply(T &e, const S &i, const V &v) {
- e().insert_element(i, v);
- }
- template <class T, class S, class V>
- BOOST_UBLAS_INLINE
- static void apply(T &e, const S &i, const S &j, const V &v) {
- e().insert_element(i,j, v);
- }
- };
- }
- template <class T>
- class traverse_policy_wrapper {
- public:
- typedef T type;
- };
- namespace traverse_policy {
-
- struct no_wrap {
-
- template <class S1, class S2, class S3>
- BOOST_UBLAS_INLINE
- static void apply1(const S1 &/*s*/, S2 &/*i*/, S3 &/*j*/) {
- }
-
- template <class S1, class S2, class S3>
- BOOST_UBLAS_INLINE
- static void apply2(const S1 &/*s1*/, const S1 &/*s2*/, S2 &/*i1*/, S3 &/*i2*/) {
- }
- };
-
- struct wrap {
-
- template <class S1, class S2, class S3>
- BOOST_UBLAS_INLINE
- static void apply1(const S1 &s, S2 &i1, S3 &i2) {
- if (i2>=s) {
- i1++;
- i2=0;
- }
- }
-
- template <class S1, class S2, class S3>
- BOOST_UBLAS_INLINE
- static void apply2(const S1 &s1, const S1 &s2, S2 &i1, S3 &i2) {
- if (i2>=s2) i2=0;
- else i1-=s1;
- }
- };
-
- template <class Wrap = wrap>
- class by_row_policy :public traverse_policy_wrapper<by_row_policy<Wrap> > {
- public:
- template <typename S1, typename S2>
- BOOST_UBLAS_INLINE
- static void advance(S1 &, S2 &j) { j++;}
- template <class E1, class E2, typename S1, typename S2, typename S3, typename S4, typename S5>
- BOOST_UBLAS_INLINE
- static bool next(const E1 &e, const E2 &me, S1 &i, S2 &j, const S3 &/*i0*/, const S3 &j0, S4 &k, S5 &l) {
- l++; j++;
- if (l>=e().size2()) {
- l=0; k++; j=j0; i++;
-
-
-
-
- if (k>=e().size1()) {
- j=j0+e().size2();
- Wrap::apply2(e().size1(), me().size2(), i, j);
- return false;
- }
- }
- return true;
- }
- template <class E, typename S1, typename S2>
- BOOST_UBLAS_INLINE
- static void apply_wrap(const E& e, S1 &i, S2 &j) {
- Wrap::apply1(e().size2(), i, j);
- }
- };
-
- template <class Wrap = wrap>
- class by_column_policy :public traverse_policy_wrapper<by_column_policy<Wrap> > {
- public:
- template <typename S1, typename S2>
- BOOST_UBLAS_INLINE
- static void advance(S1 &i, S2 &) { i++;}
- template <class E1, class E2, typename S1, typename S2, typename S3, typename S4, typename S5>
- BOOST_UBLAS_INLINE
- static bool next(const E1 &e, const E2 &me, S1 &i, S2 &j, const S3 &i0, const S3 &/*j0*/, S4 &k, S5 &l) {
- k++; i++;
- if (k>=e().size1()) {
- k=0; l++; i=i0; j++;
-
-
-
-
- if (l>=e().size2()) {
- i=i0+e().size1();
- Wrap::apply2(e().size2(), me().size1(), j, i);
- return false;
- }
- }
- return true;
- }
- template <class E, typename S1, typename S2>
- BOOST_UBLAS_INLINE
- static void apply_wrap(const E& e, S1 &i, S2 &j) {
- Wrap::apply1(e().size1(), j, i);
- }
- };
- }
- #ifndef BOOST_UBLAS_DEFAULT_NO_WRAP_POLICY
- typedef traverse_policy::wrap DEFAULT_WRAP_POLICY;
- #else
- typedef traverse_policy::no_wrap DEFAULT_WRAP_POLICY;
- #endif
- #ifndef BOOST_UBLAS_DEFAULT_ASSIGN_BY_COLUMN
- typedef traverse_policy::by_row_policy<DEFAULT_WRAP_POLICY> DEFAULT_TRAVERSE_POLICY;
- #else
- typedef traverse_policy::by_column<DEFAULT_WRAP_POLICY> DEFAULT_TRAVERSE_POLICY;
- #endif
-
- namespace traverse_policy {
- inline by_row_policy<DEFAULT_WRAP_POLICY> by_row() {
- return by_row_policy<DEFAULT_WRAP_POLICY>();
- }
- inline by_row_policy<wrap> by_row_wrap() {
- return by_row_policy<wrap>();
- }
- inline by_row_policy<no_wrap> by_row_no_wrap() {
- return by_row_policy<no_wrap>();
- }
- inline by_column_policy<DEFAULT_WRAP_POLICY> by_column() {
- return by_column_policy<DEFAULT_WRAP_POLICY>();
- }
- inline by_column_policy<wrap> by_column_wrap() {
- return by_column_policy<wrap>();
- }
- inline by_column_policy<no_wrap> by_column_no_wrap() {
- return by_column_policy<no_wrap>();
- }
- }
- template <class E, class Fill_Policy = fill_policy::index_assign>
- class vector_expression_assigner {
- public:
- typedef typename E::expression_type::value_type value_type;
- typedef typename E::expression_type::size_type size_type;
- BOOST_UBLAS_INLINE
- vector_expression_assigner(E &e):ve(&e), i(0) {
- }
- BOOST_UBLAS_INLINE
- vector_expression_assigner(size_type k, E &e):ve(&e), i(k) {
-
-
- }
- BOOST_UBLAS_INLINE
- vector_expression_assigner(E &e, value_type val):ve(&e), i(0) {
- operator,(val);
- }
- template <class AE>
- BOOST_UBLAS_INLINE
- vector_expression_assigner(E &e, const vector_expression<AE> &nve):ve(&e), i(0) {
- operator,(nve);
- }
- template <typename T>
- BOOST_UBLAS_INLINE
- vector_expression_assigner(E &e, const index_manipulator<T> &ta):ve(&e), i(0) {
- operator,(ta);
- }
- BOOST_UBLAS_INLINE
- vector_expression_assigner &operator, (const value_type& val) {
- apply(val);
- return *this;
- }
- template <class AE>
- BOOST_UBLAS_INLINE
- vector_expression_assigner &operator, (const vector_expression<AE> &nve) {
- for (typename AE::size_type k = 0; k!= nve().size(); k++)
- operator,(nve()(k));
- return *this;
- }
- template <typename T>
- BOOST_UBLAS_INLINE
- vector_expression_assigner &operator, (const index_manipulator<T> &ta) {
- ta().manip(i);
- return *this;
- }
- template <class T>
- BOOST_UBLAS_INLINE
- vector_expression_assigner<E, T> operator, (fill_policy_wrapper<T>) const {
- return vector_expression_assigner<E, T>(i, *ve);
- }
- private:
- BOOST_UBLAS_INLINE
- vector_expression_assigner &apply(const typename E::expression_type::value_type& val) {
- Fill_Policy::apply(*ve, i++, val);
- return *this;
- }
- private:
- E *ve;
- size_type i;
- };
- template <class E>
- BOOST_UBLAS_INLINE
- vector_expression_assigner<vector_expression<E> > operator<<=(vector_expression<E> &v, const typename E::value_type &val) {
- return vector_expression_assigner<vector_expression<E> >(v,val);
- }
- template <class E1, class E2>
- BOOST_UBLAS_INLINE
- vector_expression_assigner<vector_expression<E1> > operator<<=(vector_expression<E1> &v, const vector_expression<E2> &ve) {
- return vector_expression_assigner<vector_expression<E1> >(v,ve);
- }
- template <class E, typename T>
- BOOST_UBLAS_INLINE
- vector_expression_assigner<vector_expression<E> > operator<<=(vector_expression<E> &v, const index_manipulator<T> &nv) {
- return vector_expression_assigner<vector_expression<E> >(v,nv);
- }
- template <class E, typename T>
- BOOST_UBLAS_INLINE
- vector_expression_assigner<vector_expression<E>, T> operator<<=(vector_expression<E> &v, fill_policy_wrapper<T>) {
- return vector_expression_assigner<vector_expression<E>, T>(v);
- }
- template <class E, class Fill_Policy = fill_policy::index_assign, class Traverse_Policy = DEFAULT_TRAVERSE_POLICY >
- class matrix_expression_assigner {
- public:
- typedef typename E::expression_type::size_type size_type;
- BOOST_UBLAS_INLINE
- matrix_expression_assigner(E &e): me(&e), i(0), j(0) {
- }
- BOOST_UBLAS_INLINE
- matrix_expression_assigner(E &e, size_type k, size_type l): me(&e), i(k), j(l) {
- }
- BOOST_UBLAS_INLINE
- matrix_expression_assigner(E &e, typename E::expression_type::value_type val): me(&e), i(0), j(0) {
- operator,(val);
- }
- template <class AE>
- BOOST_UBLAS_INLINE
- matrix_expression_assigner(E &e, const vector_expression<AE> &nve):me(&e), i(0), j(0) {
- operator,(nve);
- }
- template <class AE>
- BOOST_UBLAS_INLINE
- matrix_expression_assigner(E &e, const matrix_expression<AE> &nme):me(&e), i(0), j(0) {
- operator,(nme);
- }
- template <typename T>
- BOOST_UBLAS_INLINE
- matrix_expression_assigner(E &e, const index_manipulator<T> &ta):me(&e), i(0), j(0) {
- operator,(ta);
- }
- BOOST_UBLAS_INLINE
- matrix_expression_assigner &operator, (const typename E::expression_type::value_type& val) {
- Traverse_Policy::apply_wrap(*me, i ,j);
- return apply(val);
- }
- template <class AE>
- BOOST_UBLAS_INLINE
- matrix_expression_assigner &operator, (const vector_expression<AE> &nve) {
- for (typename AE::size_type k = 0; k!= nve().size(); k++) {
- operator,(nve()(k));
- }
- return *this;
- }
- template <class AE>
- BOOST_UBLAS_INLINE
- matrix_expression_assigner &operator, (const matrix_expression<AE> &nme) {
- return apply(nme);
- }
- template <typename T>
- BOOST_UBLAS_INLINE
- matrix_expression_assigner &operator, (const index_manipulator<T> &ta) {
- ta().manip(i, j);
- return *this;
- }
- template <class T>
- BOOST_UBLAS_INLINE
- matrix_expression_assigner<E, T, Traverse_Policy> operator, (fill_policy_wrapper<T>) const {
- return matrix_expression_assigner<E, T, Traverse_Policy>(*me, i, j);
- }
- template <class T>
- BOOST_UBLAS_INLINE
- matrix_expression_assigner<E, Fill_Policy, T> operator, (traverse_policy_wrapper<T>) {
- Traverse_Policy::apply_wrap(*me, i ,j);
- return matrix_expression_assigner<E, Fill_Policy, T>(*me, i, j);
- }
- private:
- BOOST_UBLAS_INLINE
- matrix_expression_assigner &apply(const typename E::expression_type::value_type& val) {
- Fill_Policy::apply(*me, i, j, val);
- Traverse_Policy::advance(i,j);
- return *this;
- }
- template <class AE>
- BOOST_UBLAS_INLINE
- matrix_expression_assigner &apply(const matrix_expression<AE> &nme) {
- size_type bi = i;
- size_type bj = j;
- typename AE::size_type k=0, l=0;
- Fill_Policy::apply(*me, i, j, nme()(k, l));
- while (Traverse_Policy::next(nme, *me, i, j, bi, bj, k, l))
- Fill_Policy::apply(*me, i, j, nme()(k, l));
- return *this;
- }
- private:
- E *me;
- size_type i, j;
- };
- template <class E>
- BOOST_UBLAS_INLINE
- matrix_expression_assigner<matrix_expression<E> > operator<<=(matrix_expression<E> &me, const typename E::value_type &val) {
- return matrix_expression_assigner<matrix_expression<E> >(me,val);
- }
- template <class E, typename T>
- BOOST_UBLAS_INLINE
- matrix_expression_assigner<matrix_expression<E>, T> operator<<=(matrix_expression<E> &me, fill_policy_wrapper<T>) {
- return matrix_expression_assigner<matrix_expression<E>, T>(me);
- }
- template <class E, typename T>
- BOOST_UBLAS_INLINE
- matrix_expression_assigner<matrix_expression<E> > operator<<=(matrix_expression<E> &me, const index_manipulator<T> &ta) {
- return matrix_expression_assigner<matrix_expression<E> >(me,ta);
- }
- template <class E, typename T>
- BOOST_UBLAS_INLINE
- matrix_expression_assigner<matrix_expression<E>, fill_policy::index_assign, T> operator<<=(matrix_expression<E> &me, traverse_policy_wrapper<T>) {
- return matrix_expression_assigner<matrix_expression<E>, fill_policy::index_assign, T>(me);
- }
- template <class E1, class E2>
- BOOST_UBLAS_INLINE
- matrix_expression_assigner<matrix_expression<E1> > operator<<=(matrix_expression<E1> &me, const vector_expression<E2> &ve) {
- return matrix_expression_assigner<matrix_expression<E1> >(me,ve);
- }
- template <class E1, class E2>
- BOOST_UBLAS_INLINE
- matrix_expression_assigner<matrix_expression<E1> > operator<<=(matrix_expression<E1> &me1, const matrix_expression<E2> &me2) {
- return matrix_expression_assigner<matrix_expression<E1> >(me1,me2);
- }
- } } }
- #endif
|