michael@0: /*
michael@0:  * Copyright 2013 Google Inc.
michael@0:  *
michael@0:  * Use of this source code is governed by a BSD-style license that can be
michael@0:  * found in the LICENSE file.
michael@0:  */
michael@0: 
michael@0: #ifndef GrGLSL_impl_DEFINED
michael@0: #define GrGLSL_impl_DEFINED
michael@0: 
michael@0: template<typename Self>
michael@0: template<typename T>
michael@0: inline Self GrGLSLExpr<Self>::VectorCastImpl(const T& expr) {
michael@0:     if (expr.isZeros()) {
michael@0:         return Self(0);
michael@0:     }
michael@0:     if (expr.isOnes()) {
michael@0:         return Self(1);
michael@0:     }
michael@0:     return Self(Self::CastStr(), expr.c_str());
michael@0: }
michael@0: 
michael@0: template<typename Self>
michael@0: template<typename T0, typename T1>
michael@0: inline Self GrGLSLExpr<Self>::Mul(T0 in0, T1 in1) {
michael@0:     if (in0.isZeros() || in1.isZeros()) {
michael@0:         return Self(0);
michael@0:     }
michael@0:     if (in0.isOnes()) {
michael@0:         return Self::VectorCast(in1);
michael@0:     }
michael@0:     if (in1.isOnes()) {
michael@0:         return Self::VectorCast(in0);
michael@0:     }
michael@0:     return Self("(%s * %s)", in0.c_str(), in1.c_str());
michael@0: }
michael@0: 
michael@0: template<typename Self>
michael@0: template<typename T0, typename T1>
michael@0: inline Self GrGLSLExpr<Self>::Add(T0 in0, T1 in1) {
michael@0:     if (in1.isZeros()) {
michael@0:         return Self::VectorCast(in0);
michael@0:     }
michael@0:     if (in0.isZeros()) {
michael@0:         return Self::VectorCast(in1);
michael@0:     }
michael@0:     if (in0.isOnes() && in1.isOnes()) {
michael@0:         return Self(2);
michael@0:     }
michael@0:     return Self("(%s + %s)", in0.c_str(), in1.c_str());
michael@0: }
michael@0: 
michael@0: template<typename Self>
michael@0: template<typename T0, typename T1>
michael@0: inline Self GrGLSLExpr<Self>::Sub(T0 in0, T1 in1) {
michael@0:     if (in1.isZeros()) {
michael@0:         return Self::VectorCast(in0);
michael@0:     }
michael@0:     if (in1.isOnes()) {
michael@0:         if (in0.isOnes()) {
michael@0:             return Self(0);
michael@0:         }
michael@0:     }
michael@0: 
michael@0:     return Self("(%s - %s)", in0.c_str(), in1.c_str());
michael@0: }
michael@0: 
michael@0: template <typename Self>
michael@0: template <typename T>
michael@0: T GrGLSLExpr<Self>::extractComponents(const char format[]) const {
michael@0:     if (this->isZeros()) {
michael@0:         return T(0);
michael@0:     }
michael@0:     if (this->isOnes()) {
michael@0:         return T(1);
michael@0:     }
michael@0:     return T(format, this->c_str());
michael@0: }
michael@0: 
michael@0: inline GrGLSLExpr1 GrGLSLExpr1::VectorCast(const GrGLSLExpr1& expr) {
michael@0:     return expr;
michael@0: }
michael@0: 
michael@0: inline const char* GrGLSLExpr1::ZerosStr() {
michael@0:     return "0";
michael@0: }
michael@0: 
michael@0: inline const char* GrGLSLExpr1::OnesStr() {
michael@0:     return "1.0";
michael@0: }
michael@0: 
michael@0: // GrGLSLExpr1::CastStr() is unimplemented because using them is likely an
michael@0: // error. This is now caught compile-time.
michael@0: 
michael@0: inline const char* GrGLSLExpr1::CastIntStr() {
michael@0:     return "%d";
michael@0: }
michael@0: 
michael@0: inline GrGLSLExpr1 operator*(const GrGLSLExpr1& in0, const GrGLSLExpr1& in1) {
michael@0:     return GrGLSLExpr1::Mul(in0, in1);
michael@0: }
michael@0: 
michael@0: inline GrGLSLExpr1 operator+(const GrGLSLExpr1& in0, const GrGLSLExpr1& in1) {
michael@0:     return GrGLSLExpr1::Add(in0, in1);
michael@0: }
michael@0: 
michael@0: inline GrGLSLExpr1 operator-(const GrGLSLExpr1& in0, const GrGLSLExpr1& in1) {
michael@0:     return GrGLSLExpr1::Sub(in0, in1);
michael@0: }
michael@0: 
michael@0: inline const char* GrGLSLExpr4::ZerosStr() {
michael@0:     return "vec4(0)";
michael@0: }
michael@0: 
michael@0: inline const char* GrGLSLExpr4::OnesStr() {
michael@0:     return "vec4(1)";
michael@0: }
michael@0: 
michael@0: inline const char* GrGLSLExpr4::CastStr() {
michael@0:     return "vec4(%s)";
michael@0: }
michael@0: 
michael@0: inline const char* GrGLSLExpr4::CastIntStr() {
michael@0:     return "vec4(%d)";
michael@0: }
michael@0: 
michael@0: inline GrGLSLExpr4 GrGLSLExpr4::VectorCast(const GrGLSLExpr1& expr) {
michael@0:     return INHERITED::VectorCastImpl(expr);
michael@0: }
michael@0: 
michael@0: inline GrGLSLExpr4 GrGLSLExpr4::VectorCast(const GrGLSLExpr4& expr) {
michael@0:     return expr;
michael@0: }
michael@0: 
michael@0: inline GrGLSLExpr4::AExpr GrGLSLExpr4::a() const {
michael@0:     return this->extractComponents<GrGLSLExpr4::AExpr>("%s.a");
michael@0: }
michael@0: 
michael@0: inline GrGLSLExpr4 operator*(const GrGLSLExpr1& in0, const GrGLSLExpr4& in1) {
michael@0:     return GrGLSLExpr4::Mul(in0, in1);
michael@0: }
michael@0: 
michael@0: inline GrGLSLExpr4 operator+(const GrGLSLExpr1& in0, const GrGLSLExpr4& in1) {
michael@0:     return GrGLSLExpr4::Add(in0, in1);
michael@0: }
michael@0: 
michael@0: inline GrGLSLExpr4 operator-(const GrGLSLExpr1& in0, const GrGLSLExpr4& in1) {
michael@0:     return GrGLSLExpr4::Sub(in0, in1);
michael@0: }
michael@0: 
michael@0: inline GrGLSLExpr4 operator*(const GrGLSLExpr4& in0, const GrGLSLExpr1& in1) {
michael@0:     return GrGLSLExpr4::Mul(in0, in1);
michael@0: }
michael@0: 
michael@0: inline GrGLSLExpr4 operator+(const GrGLSLExpr4& in0, const GrGLSLExpr1& in1) {
michael@0:     return GrGLSLExpr4::Add(in0, in1);
michael@0: }
michael@0: 
michael@0: inline GrGLSLExpr4 operator-(const GrGLSLExpr4& in0, const GrGLSLExpr1& in1) {
michael@0:     return GrGLSLExpr4::Sub(in0, in1);
michael@0: }
michael@0: 
michael@0: inline GrGLSLExpr4 operator*(const GrGLSLExpr4& in0, const GrGLSLExpr4& in1) {
michael@0:     return GrGLSLExpr4::Mul(in0, in1);
michael@0: }
michael@0: 
michael@0: inline GrGLSLExpr4 operator+(const GrGLSLExpr4& in0, const GrGLSLExpr4& in1) {
michael@0:     return GrGLSLExpr4::Add(in0, in1);
michael@0: }
michael@0: 
michael@0: inline GrGLSLExpr4 operator-(const GrGLSLExpr4& in0, const GrGLSLExpr4& in1) {
michael@0:     return GrGLSLExpr4::Sub(in0, in1);
michael@0: }
michael@0: 
michael@0: #endif