michael@0: 
michael@0: /*
michael@0:  * Copyright 2006 The Android Open Source Project
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: 
michael@0: #include "SkDrawGradient.h"
michael@0: #include "SkAnimateMaker.h"
michael@0: #include "SkAnimatorScript.h"
michael@0: #include "SkGradientShader.h"
michael@0: #include "SkUnitMapper.h"
michael@0: 
michael@0: static SkScalar SkUnitToScalar(U16CPU x) {
michael@0:     return x / 65535.0f;
michael@0: }
michael@0: 
michael@0: static U16CPU SkScalarToUnit(SkScalar x) {
michael@0:     SkScalar pin =  SkScalarPin(x, 0, SK_Scalar1);
michael@0:     return (int) (pin * 65535.0f);
michael@0: }
michael@0: 
michael@0: class SkDrawGradientUnitMapper : public SkUnitMapper {
michael@0: public:
michael@0:     SkDrawGradientUnitMapper(SkAnimateMaker* maker, const char* script) : fMaker(maker), fScript(script) {
michael@0:     }
michael@0: 
michael@0:     SK_DECLARE_UNFLATTENABLE_OBJECT()
michael@0: 
michael@0: protected:
michael@0:     virtual uint16_t mapUnit16(uint16_t x) {
michael@0:         fUnit = SkUnitToScalar(x);
michael@0:         SkScriptValue value;
michael@0:         SkAnimatorScript engine(*fMaker, NULL, SkType_Float);
michael@0:         engine.propertyCallBack(GetUnitValue, &fUnit);
michael@0:         if (engine.evaluate(fScript, &value, SkType_Float))
michael@0:             x = SkScalarToUnit(value.fOperand.fScalar);
michael@0:         return x;
michael@0:     }
michael@0: 
michael@0:     static bool GetUnitValue(const char* token, size_t len, void* unitPtr, SkScriptValue* value) {
michael@0:         if (SK_LITERAL_STR_EQUAL("unit", token, len)) {
michael@0:             value->fOperand.fScalar = *(SkScalar*) unitPtr;
michael@0:             value->fType = SkType_Float;
michael@0:             return true;
michael@0:         }
michael@0:         return false;
michael@0:     }
michael@0: 
michael@0:     SkAnimateMaker* fMaker;
michael@0:     const char* fScript;
michael@0:     SkScalar fUnit;
michael@0: };
michael@0: 
michael@0: 
michael@0: #if SK_USE_CONDENSED_INFO == 0
michael@0: 
michael@0: const SkMemberInfo SkDrawGradient::fInfo[] = {
michael@0:     SK_MEMBER_INHERITED,
michael@0:     SK_MEMBER_ARRAY(offsets, Float),
michael@0:     SK_MEMBER(unitMapper, String)
michael@0: };
michael@0: 
michael@0: #endif
michael@0: 
michael@0: DEFINE_GET_MEMBER(SkDrawGradient);
michael@0: 
michael@0: SkDrawGradient::SkDrawGradient() : fUnitMapper(NULL) {
michael@0: }
michael@0: 
michael@0: SkDrawGradient::~SkDrawGradient() {
michael@0:     for (int index = 0; index < fDrawColors.count(); index++)
michael@0:         delete fDrawColors[index];
michael@0:     delete fUnitMapper;
michael@0: }
michael@0: 
michael@0: bool SkDrawGradient::addChild(SkAnimateMaker& , SkDisplayable* child) {
michael@0:     SkASSERT(child);
michael@0:     if (child->isColor()) {
michael@0:         SkDrawColor* color = (SkDrawColor*) child;
michael@0:         *fDrawColors.append() = color;
michael@0:         return true;
michael@0:     }
michael@0:     return false;
michael@0: }
michael@0: 
michael@0: int SkDrawGradient::addPrelude() {
michael@0:     int count = fDrawColors.count();
michael@0:     fColors.setCount(count);
michael@0:     for (int index = 0; index < count; index++)
michael@0:         fColors[index] = fDrawColors[index]->color;
michael@0:     return count;
michael@0: }
michael@0: 
michael@0: #ifdef SK_DUMP_ENABLED
michael@0: void SkDrawGradient::dumpRest(SkAnimateMaker* maker) {
michael@0:     dumpAttrs(maker);
michael@0:     //can a gradient have no colors?
michael@0:     bool closedYet = false;
michael@0:     SkDisplayList::fIndent += 4;
michael@0:     for (SkDrawColor** ptr = fDrawColors.begin(); ptr < fDrawColors.end(); ptr++) {
michael@0:         if (closedYet == false) {
michael@0:             SkDebugf(">\n");
michael@0:             closedYet = true;
michael@0:         }
michael@0:         SkDrawColor* color = *ptr;
michael@0:         color->dump(maker);
michael@0:     }
michael@0:     SkDisplayList::fIndent -= 4;
michael@0:     dumpChildren(maker, closedYet); //dumps the matrix if it has one
michael@0: }
michael@0: #endif
michael@0: 
michael@0: void SkDrawGradient::onEndElement(SkAnimateMaker& maker) {
michael@0:     if (offsets.count() != 0) {
michael@0:         if (offsets.count() != fDrawColors.count()) {
michael@0:             maker.setErrorCode(SkDisplayXMLParserError::kGradientOffsetsDontMatchColors);
michael@0:             return;
michael@0:         }
michael@0:         if (offsets[0] != 0) {
michael@0:             maker.setErrorCode(SkDisplayXMLParserError::kGradientOffsetsMustStartWithZero);
michael@0:             return;
michael@0:         }
michael@0:         if (offsets[offsets.count()-1] != SK_Scalar1) {
michael@0:             maker.setErrorCode(SkDisplayXMLParserError::kGradientOffsetsMustEndWithOne);
michael@0:             return;
michael@0:         }
michael@0:         for (int i = 1; i < offsets.count(); i++) {
michael@0:             if (offsets[i] <= offsets[i-1]) {
michael@0:                 maker.setErrorCode(SkDisplayXMLParserError::kGradientOffsetsMustIncrease);
michael@0:                 return;
michael@0:             }
michael@0:             if (offsets[i] > SK_Scalar1) {
michael@0:                 maker.setErrorCode(SkDisplayXMLParserError::kGradientOffsetsMustBeNoMoreThanOne);
michael@0:                 return;
michael@0:             }
michael@0:         }
michael@0:     }
michael@0:     if (unitMapper.size() > 0)
michael@0:         fUnitMapper = new SkDrawGradientUnitMapper(&maker, unitMapper.c_str());
michael@0:     INHERITED::onEndElement(maker);
michael@0: }
michael@0: 
michael@0: #if SK_USE_CONDENSED_INFO == 0
michael@0: 
michael@0: const SkMemberInfo SkDrawLinearGradient::fInfo[] = {
michael@0:     SK_MEMBER_INHERITED,
michael@0:     SK_MEMBER_ARRAY(points, Float),
michael@0: };
michael@0: 
michael@0: #endif
michael@0: 
michael@0: DEFINE_GET_MEMBER(SkDrawLinearGradient);
michael@0: 
michael@0: SkDrawLinearGradient::SkDrawLinearGradient() {
michael@0: }
michael@0: 
michael@0: void SkDrawLinearGradient::onEndElement(SkAnimateMaker& maker)
michael@0: {
michael@0:     if (points.count() != 4)
michael@0:         maker.setErrorCode(SkDisplayXMLParserError::kGradientPointsLengthMustBeFour);
michael@0:     INHERITED::onEndElement(maker);
michael@0: }
michael@0: 
michael@0: #ifdef SK_DUMP_ENABLED
michael@0: void SkDrawLinearGradient::dump(SkAnimateMaker* maker) {
michael@0:     dumpBase(maker);
michael@0:     dumpRest(maker);
michael@0:     }
michael@0: #endif
michael@0: 
michael@0: SkShader* SkDrawLinearGradient::getShader() {
michael@0:     if (addPrelude() == 0 || points.count() != 4)
michael@0:         return NULL;
michael@0:     SkShader* shader = SkGradientShader::CreateLinear((SkPoint*)points.begin(),
michael@0:         fColors.begin(), offsets.begin(), fColors.count(), (SkShader::TileMode) tileMode, fUnitMapper);
michael@0:     SkAutoTDelete<SkShader> autoDel(shader);
michael@0:     addPostlude(shader);
michael@0:     (void)autoDel.detach();
michael@0:     return shader;
michael@0: }
michael@0: 
michael@0: 
michael@0: #if SK_USE_CONDENSED_INFO == 0
michael@0: 
michael@0: const SkMemberInfo SkDrawRadialGradient::fInfo[] = {
michael@0:     SK_MEMBER_INHERITED,
michael@0:     SK_MEMBER(center, Point),
michael@0:     SK_MEMBER(radius, Float)
michael@0: };
michael@0: 
michael@0: #endif
michael@0: 
michael@0: DEFINE_GET_MEMBER(SkDrawRadialGradient);
michael@0: 
michael@0: SkDrawRadialGradient::SkDrawRadialGradient() : radius(0) {
michael@0:     center.set(0, 0);
michael@0: }
michael@0: 
michael@0: #ifdef SK_DUMP_ENABLED
michael@0: void SkDrawRadialGradient::dump(SkAnimateMaker* maker) {
michael@0:     dumpBase(maker);
michael@0:     dumpRest(maker);
michael@0: }
michael@0: #endif
michael@0: 
michael@0: SkShader* SkDrawRadialGradient::getShader() {
michael@0:     if (addPrelude() == 0)
michael@0:         return NULL;
michael@0:     SkShader* shader = SkGradientShader::CreateRadial(center,
michael@0:         radius, fColors.begin(), offsets.begin(), fColors.count(), (SkShader::TileMode) tileMode, fUnitMapper);
michael@0:     SkAutoTDelete<SkShader> autoDel(shader);
michael@0:     addPostlude(shader);
michael@0:     (void)autoDel.detach();
michael@0:     return shader;
michael@0: }