You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
					
					
						
							395 lines
						
					
					
						
							12 KiB
						
					
					
				
			
		
		
	
	
							395 lines
						
					
					
						
							12 KiB
						
					
					
				| /*
 | |
|  * Copyright 2013 The Android Open Source Project
 | |
|  *
 | |
|  * 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.
 | |
|  */
 | |
| 
 | |
| #ifndef UI_MAT4_H
 | |
| #define UI_MAT4_H
 | |
| 
 | |
| #include <stdint.h>
 | |
| #include <sys/types.h>
 | |
| 
 | |
| #include <ui/vec4.h>
 | |
| #include <utils/String8.h>
 | |
| 
 | |
| #define TMAT_IMPLEMENTATION
 | |
| #include <ui/TMatHelpers.h>
 | |
| 
 | |
| #define PURE __attribute__((pure))
 | |
| 
 | |
| namespace android {
 | |
| // -------------------------------------------------------------------------------------
 | |
| 
 | |
| template <typename T>
 | |
| class tmat44 :  public TVecUnaryOperators<tmat44, T>,
 | |
|                 public TVecComparisonOperators<tmat44, T>,
 | |
|                 public TVecAddOperators<tmat44, T>,
 | |
|                 public TMatProductOperators<tmat44, T>,
 | |
|                 public TMatSquareFunctions<tmat44, T>,
 | |
|                 public TMatDebug<tmat44, T>
 | |
| {
 | |
| public:
 | |
|     enum no_init { NO_INIT };
 | |
|     typedef T value_type;
 | |
|     typedef T& reference;
 | |
|     typedef T const& const_reference;
 | |
|     typedef size_t size_type;
 | |
|     typedef tvec4<T> col_type;
 | |
|     typedef tvec4<T> row_type;
 | |
| 
 | |
|     // size of a column (i.e.: number of rows)
 | |
|     enum { COL_SIZE = col_type::SIZE };
 | |
|     static inline size_t col_size() { return COL_SIZE; }
 | |
| 
 | |
|     // size of a row (i.e.: number of columns)
 | |
|     enum { ROW_SIZE = row_type::SIZE };
 | |
|     static inline size_t row_size() { return ROW_SIZE; }
 | |
|     static inline size_t size()     { return row_size(); }  // for TVec*<>
 | |
| 
 | |
| private:
 | |
| 
 | |
|     /*
 | |
|      *  <--  N columns  -->
 | |
|      *
 | |
|      *  a00 a10 a20 ... aN0    ^
 | |
|      *  a01 a11 a21 ... aN1    |
 | |
|      *  a02 a12 a22 ... aN2  M rows
 | |
|      *  ...                    |
 | |
|      *  a0M a1M a2M ... aNM    v
 | |
|      *
 | |
|      *  COL_SIZE = M
 | |
|      *  ROW_SIZE = N
 | |
|      *  m[0] = [a00 a01 a02 ... a01M]
 | |
|      */
 | |
| 
 | |
|     col_type mValue[ROW_SIZE];
 | |
| 
 | |
| public:
 | |
|     // array access
 | |
|     inline col_type const& operator [] (size_t i) const { return mValue[i]; }
 | |
|     inline col_type&       operator [] (size_t i)       { return mValue[i]; }
 | |
| 
 | |
|     T const* asArray() const { return &mValue[0][0]; }
 | |
| 
 | |
|     // -----------------------------------------------------------------------
 | |
|     // we don't provide copy-ctor and operator= on purpose
 | |
|     // because we want the compiler generated versions
 | |
| 
 | |
|     /*
 | |
|      *  constructors
 | |
|      */
 | |
| 
 | |
|     // leaves object uninitialized. use with caution.
 | |
|     explicit tmat44(no_init) { }
 | |
| 
 | |
|     // initialize to identity
 | |
|     tmat44();
 | |
| 
 | |
|     // initialize to Identity*scalar.
 | |
|     template<typename U>
 | |
|     explicit tmat44(U v);
 | |
| 
 | |
|     // sets the diagonal to the passed vector
 | |
|     template <typename U>
 | |
|     explicit tmat44(const tvec4<U>& rhs);
 | |
| 
 | |
|     // construct from another matrix of the same size
 | |
|     template <typename U>
 | |
|     explicit tmat44(const tmat44<U>& rhs);
 | |
| 
 | |
|     // construct from 4 column vectors
 | |
|     template <typename A, typename B, typename C, typename D>
 | |
|     tmat44(const tvec4<A>& v0, const tvec4<B>& v1, const tvec4<C>& v2, const tvec4<D>& v3);
 | |
| 
 | |
|     // construct from 16 scalars
 | |
|     template <
 | |
|         typename A, typename B, typename C, typename D,
 | |
|         typename E, typename F, typename G, typename H,
 | |
|         typename I, typename J, typename K, typename L,
 | |
|         typename M, typename N, typename O, typename P>
 | |
|     tmat44( A m00, B m01, C m02, D m03,
 | |
|             E m10, F m11, G m12, H m13,
 | |
|             I m20, J m21, K m22, L m23,
 | |
|             M m30, N m31, O m32, P m33);
 | |
| 
 | |
|     // construct from a C array
 | |
|     template <typename U>
 | |
|     explicit tmat44(U const* rawArray);
 | |
| 
 | |
|     /*
 | |
|      *  helpers
 | |
|      */
 | |
| 
 | |
|     static tmat44 ortho(T left, T right, T bottom, T top, T near, T far);
 | |
| 
 | |
|     static tmat44 frustum(T left, T right, T bottom, T top, T near, T far);
 | |
| 
 | |
|     template <typename A, typename B, typename C>
 | |
|     static tmat44 lookAt(const tvec3<A>& eye, const tvec3<B>& center, const tvec3<C>& up);
 | |
| 
 | |
|     template <typename A>
 | |
|     static tmat44 translate(const tvec4<A>& t);
 | |
| 
 | |
|     template <typename A>
 | |
|     static tmat44 scale(const tvec4<A>& s);
 | |
| 
 | |
|     template <typename A, typename B>
 | |
|     static tmat44 rotate(A radian, const tvec3<B>& about);
 | |
| };
 | |
| 
 | |
| // ----------------------------------------------------------------------------------------
 | |
| // Constructors
 | |
| // ----------------------------------------------------------------------------------------
 | |
| 
 | |
| /*
 | |
|  * Since the matrix code could become pretty big quickly, we don't inline most
 | |
|  * operations.
 | |
|  */
 | |
| 
 | |
| template <typename T>
 | |
| tmat44<T>::tmat44() {
 | |
|     mValue[0] = col_type(1,0,0,0);
 | |
|     mValue[1] = col_type(0,1,0,0);
 | |
|     mValue[2] = col_type(0,0,1,0);
 | |
|     mValue[3] = col_type(0,0,0,1);
 | |
| }
 | |
| 
 | |
| template <typename T>
 | |
| template <typename U>
 | |
| tmat44<T>::tmat44(U v) {
 | |
|     mValue[0] = col_type(v,0,0,0);
 | |
|     mValue[1] = col_type(0,v,0,0);
 | |
|     mValue[2] = col_type(0,0,v,0);
 | |
|     mValue[3] = col_type(0,0,0,v);
 | |
| }
 | |
| 
 | |
| template<typename T>
 | |
| template<typename U>
 | |
| tmat44<T>::tmat44(const tvec4<U>& v) {
 | |
|     mValue[0] = col_type(v.x,0,0,0);
 | |
|     mValue[1] = col_type(0,v.y,0,0);
 | |
|     mValue[2] = col_type(0,0,v.z,0);
 | |
|     mValue[3] = col_type(0,0,0,v.w);
 | |
| }
 | |
| 
 | |
| // construct from 16 scalars
 | |
| template<typename T>
 | |
| template <
 | |
|     typename A, typename B, typename C, typename D,
 | |
|     typename E, typename F, typename G, typename H,
 | |
|     typename I, typename J, typename K, typename L,
 | |
|     typename M, typename N, typename O, typename P>
 | |
| tmat44<T>::tmat44(  A m00, B m01, C m02, D m03,
 | |
|                     E m10, F m11, G m12, H m13,
 | |
|                     I m20, J m21, K m22, L m23,
 | |
|                     M m30, N m31, O m32, P m33) {
 | |
|     mValue[0] = col_type(m00, m01, m02, m03);
 | |
|     mValue[1] = col_type(m10, m11, m12, m13);
 | |
|     mValue[2] = col_type(m20, m21, m22, m23);
 | |
|     mValue[3] = col_type(m30, m31, m32, m33);
 | |
| }
 | |
| 
 | |
| template <typename T>
 | |
| template <typename U>
 | |
| tmat44<T>::tmat44(const tmat44<U>& rhs) {
 | |
|     for (size_t r=0 ; r<row_size() ; r++)
 | |
|         mValue[r] = rhs[r];
 | |
| }
 | |
| 
 | |
| template <typename T>
 | |
| template <typename A, typename B, typename C, typename D>
 | |
| tmat44<T>::tmat44(const tvec4<A>& v0, const tvec4<B>& v1, const tvec4<C>& v2, const tvec4<D>& v3) {
 | |
|     mValue[0] = v0;
 | |
|     mValue[1] = v1;
 | |
|     mValue[2] = v2;
 | |
|     mValue[3] = v3;
 | |
| }
 | |
| 
 | |
| template <typename T>
 | |
| template <typename U>
 | |
| tmat44<T>::tmat44(U const* rawArray) {
 | |
|     for (size_t r=0 ; r<row_size() ; r++)
 | |
|         for (size_t c=0 ; c<col_size() ; c++)
 | |
|             mValue[r][c] = *rawArray++;
 | |
| }
 | |
| 
 | |
| // ----------------------------------------------------------------------------------------
 | |
| // Helpers
 | |
| // ----------------------------------------------------------------------------------------
 | |
| 
 | |
| template <typename T>
 | |
| tmat44<T> tmat44<T>::ortho(T left, T right, T bottom, T top, T near, T far) {
 | |
|     tmat44<T> m;
 | |
|     m[0][0] =  2 / (right - left);
 | |
|     m[1][1] =  2 / (top   - bottom);
 | |
|     m[2][2] = -2 / (far   - near);
 | |
|     m[3][0] = -(right + left)   / (right - left);
 | |
|     m[3][1] = -(top   + bottom) / (top   - bottom);
 | |
|     m[3][2] = -(far   + near)   / (far   - near);
 | |
|     return m;
 | |
| }
 | |
| 
 | |
| template <typename T>
 | |
| tmat44<T> tmat44<T>::frustum(T left, T right, T bottom, T top, T near, T far) {
 | |
|     tmat44<T> m;
 | |
|     T A = (right + left)   / (right - left);
 | |
|     T B = (top   + bottom) / (top   - bottom);
 | |
|     T C = (far   + near)   / (far   - near);
 | |
|     T D = (2 * far * near) / (far   - near);
 | |
|     m[0][0] = (2 * near) / (right - left);
 | |
|     m[1][1] = (2 * near) / (top   - bottom);
 | |
|     m[2][0] = A;
 | |
|     m[2][1] = B;
 | |
|     m[2][2] = C;
 | |
|     m[2][3] =-1;
 | |
|     m[3][2] = D;
 | |
|     m[3][3] = 0;
 | |
|     return m;
 | |
| }
 | |
| 
 | |
| template <typename T>
 | |
| template <typename A, typename B, typename C>
 | |
| tmat44<T> tmat44<T>::lookAt(const tvec3<A>& eye, const tvec3<B>& center, const tvec3<C>& up) {
 | |
|     tvec3<T> L(normalize(center - eye));
 | |
|     tvec3<T> S(normalize( cross(L, up) ));
 | |
|     tvec3<T> U(cross(S, L));
 | |
|     return tmat44<T>(
 | |
|             tvec4<T>( S, 0),
 | |
|             tvec4<T>( U, 0),
 | |
|             tvec4<T>(-L, 0),
 | |
|             tvec4<T>(-eye, 1));
 | |
| }
 | |
| 
 | |
| template <typename T>
 | |
| template <typename A>
 | |
| tmat44<T> tmat44<T>::translate(const tvec4<A>& t) {
 | |
|     tmat44<T> r;
 | |
|     r[3] = t;
 | |
|     return r;
 | |
| }
 | |
| 
 | |
| template <typename T>
 | |
| template <typename A>
 | |
| tmat44<T> tmat44<T>::scale(const tvec4<A>& s) {
 | |
|     tmat44<T> r;
 | |
|     r[0][0] = s[0];
 | |
|     r[1][1] = s[1];
 | |
|     r[2][2] = s[2];
 | |
|     r[3][3] = s[3];
 | |
|     return r;
 | |
| }
 | |
| 
 | |
| template <typename T>
 | |
| template <typename A, typename B>
 | |
| tmat44<T> tmat44<T>::rotate(A radian, const tvec3<B>& about) {
 | |
|     tmat44<T> rotation;
 | |
|     T* r = const_cast<T*>(rotation.asArray());
 | |
|     T c = cos(radian);
 | |
|     T s = sin(radian);
 | |
|     if (about.x==1 && about.y==0 && about.z==0) {
 | |
|         r[5] = c;   r[10]= c;
 | |
|         r[6] = s;   r[9] = -s;
 | |
|     } else if (about.x==0 && about.y==1 && about.z==0) {
 | |
|         r[0] = c;   r[10]= c;
 | |
|         r[8] = s;   r[2] = -s;
 | |
|     } else if (about.x==0 && about.y==0 && about.z==1) {
 | |
|         r[0] = c;   r[5] = c;
 | |
|         r[1] = s;   r[4] = -s;
 | |
|     } else {
 | |
|         tvec3<B> nabout = normalize(about);
 | |
|         B x = nabout.x;
 | |
|         B y = nabout.y;
 | |
|         B z = nabout.z;
 | |
|         T nc = 1 - c;
 | |
|         T xy = x * y;
 | |
|         T yz = y * z;
 | |
|         T zx = z * x;
 | |
|         T xs = x * s;
 | |
|         T ys = y * s;
 | |
|         T zs = z * s;
 | |
|         r[ 0] = x*x*nc +  c;    r[ 4] =  xy*nc - zs;    r[ 8] =  zx*nc + ys;
 | |
|         r[ 1] =  xy*nc + zs;    r[ 5] = y*y*nc +  c;    r[ 9] =  yz*nc - xs;
 | |
|         r[ 2] =  zx*nc - ys;    r[ 6] =  yz*nc + xs;    r[10] = z*z*nc +  c;
 | |
|     }
 | |
|     return rotation;
 | |
| }
 | |
| 
 | |
| // ----------------------------------------------------------------------------------------
 | |
| // Arithmetic operators outside of class
 | |
| // ----------------------------------------------------------------------------------------
 | |
| 
 | |
| /* We use non-friend functions here to prevent the compiler from using
 | |
|  * implicit conversions, for instance of a scalar to a vector. The result would
 | |
|  * not be what the caller expects.
 | |
|  *
 | |
|  * Also note that the order of the arguments in the inner loop is important since
 | |
|  * it determines the output type (only relevant when T != U).
 | |
|  */
 | |
| 
 | |
| // matrix * vector, result is a vector of the same type than the input vector
 | |
| template <typename T, typename U>
 | |
| typename tmat44<U>::col_type PURE operator *(const tmat44<T>& lv, const tvec4<U>& rv) {
 | |
|     typename tmat44<U>::col_type result;
 | |
|     for (size_t r=0 ; r<tmat44<T>::row_size() ; r++)
 | |
|         result += rv[r]*lv[r];
 | |
|     return result;
 | |
| }
 | |
| 
 | |
| // vector * matrix, result is a vector of the same type than the input vector
 | |
| template <typename T, typename U>
 | |
| typename tmat44<U>::row_type PURE operator *(const tvec4<U>& rv, const tmat44<T>& lv) {
 | |
|     typename tmat44<U>::row_type result(tmat44<U>::row_type::NO_INIT);
 | |
|     for (size_t r=0 ; r<tmat44<T>::row_size() ; r++)
 | |
|         result[r] = dot(rv, lv[r]);
 | |
|     return result;
 | |
| }
 | |
| 
 | |
| // matrix * scalar, result is a matrix of the same type than the input matrix
 | |
| template <typename T, typename U>
 | |
| tmat44<T> PURE operator *(const tmat44<T>& lv, U rv) {
 | |
|     tmat44<T> result(tmat44<T>::NO_INIT);
 | |
|     for (size_t r=0 ; r<tmat44<T>::row_size() ; r++)
 | |
|         result[r] = lv[r]*rv;
 | |
|     return result;
 | |
| }
 | |
| 
 | |
| // scalar * matrix, result is a matrix of the same type than the input matrix
 | |
| template <typename T, typename U>
 | |
| tmat44<T> PURE operator *(U rv, const tmat44<T>& lv) {
 | |
|     tmat44<T> result(tmat44<T>::NO_INIT);
 | |
|     for (size_t r=0 ; r<tmat44<T>::row_size() ; r++)
 | |
|         result[r] = lv[r]*rv;
 | |
|     return result;
 | |
| }
 | |
| 
 | |
| // ----------------------------------------------------------------------------------------
 | |
| 
 | |
| /* FIXME: this should go into TMatSquareFunctions<> but for some reason
 | |
|  * BASE<T>::col_type is not accessible from there (???)
 | |
|  */
 | |
| template<typename T>
 | |
| typename tmat44<T>::col_type PURE diag(const tmat44<T>& m) {
 | |
|     return matrix::diag(m);
 | |
| }
 | |
| 
 | |
| // ----------------------------------------------------------------------------------------
 | |
| 
 | |
| typedef tmat44<float> mat4;
 | |
| 
 | |
| // ----------------------------------------------------------------------------------------
 | |
| }; // namespace android
 | |
| 
 | |
| #undef PURE
 | |
| 
 | |
| #endif /* UI_MAT4_H */
 | |
| 
 |