GrDrawTarget.cpp revision c4dc0ad8e252a7e30d19b47d3d0d9f2c69faf854 
1
2/*
3 * Copyright 2010 Google Inc.
4 *
5 * Use of this source code is governed by a BSD-style license that can be
6 * found in the LICENSE file.
7 */
8
9
10
11#include "GrDrawTarget.h"
12#include "GrContext.h"
13#include "GrDrawTargetCaps.h"
14#include "GrPath.h"
15#include "GrRenderTarget.h"
16#include "GrTexture.h"
17#include "GrVertexBuffer.h"
18
19#include "SkStrokeRec.h"
20
21SK_DEFINE_INST_COUNT(GrDrawTarget)
22
23////////////////////////////////////////////////////////////////////////////////
24
25GrDrawTarget::DrawInfo& GrDrawTarget::DrawInfo::operator =(const DrawInfo& di) {
26    fPrimitiveType  = di.fPrimitiveType;
27    fStartVertex    = di.fStartVertex;
28    fStartIndex     = di.fStartIndex;
29    fVertexCount    = di.fVertexCount;
30    fIndexCount     = di.fIndexCount;
31
32    fInstanceCount          = di.fInstanceCount;
33    fVerticesPerInstance    = di.fVerticesPerInstance;
34    fIndicesPerInstance     = di.fIndicesPerInstance;
35
36    if (NULL != di.fDevBounds) {
37        SkASSERT(di.fDevBounds == &di.fDevBoundsStorage);
38        fDevBoundsStorage = di.fDevBoundsStorage;
39        fDevBounds = &fDevBoundsStorage;
40    } else {
41        fDevBounds = NULL;
42    }
43
44    fDstCopy = di.fDstCopy;
45
46    return *this;
47}
48
49#ifdef SK_DEBUG
50bool GrDrawTarget::DrawInfo::isInstanced() const {
51    if (fInstanceCount > 0) {
52        SkASSERT(0 == fIndexCount % fIndicesPerInstance);
53        SkASSERT(0 == fVertexCount % fVerticesPerInstance);
54        SkASSERT(fIndexCount / fIndicesPerInstance == fInstanceCount);
55        SkASSERT(fVertexCount / fVerticesPerInstance == fInstanceCount);
56        // there is no way to specify a non-zero start index to drawIndexedInstances().
57        SkASSERT(0 == fStartIndex);
58        return true;
59    } else {
60        SkASSERT(!fVerticesPerInstance);
61        SkASSERT(!fIndicesPerInstance);
62        return false;
63    }
64}
65#endif
66
67void GrDrawTarget::DrawInfo::adjustInstanceCount(int instanceOffset) {
68    SkASSERT(this->isInstanced());
69    SkASSERT(instanceOffset + fInstanceCount >= 0);
70    fInstanceCount += instanceOffset;
71    fVertexCount = fVerticesPerInstance * fInstanceCount;
72    fIndexCount = fIndicesPerInstance * fInstanceCount;
73}
74
75void GrDrawTarget::DrawInfo::adjustStartVertex(int vertexOffset) {
76    fStartVertex += vertexOffset;
77    SkASSERT(fStartVertex >= 0);
78}
79
80void GrDrawTarget::DrawInfo::adjustStartIndex(int indexOffset) {
81    SkASSERT(this->isIndexed());
82    fStartIndex += indexOffset;
83    SkASSERT(fStartIndex >= 0);
84}
85
86////////////////////////////////////////////////////////////////////////////////
87
88#define DEBUG_INVAL_BUFFER 0xdeadcafe
89#define DEBUG_INVAL_START_IDX -1
90
91GrDrawTarget::GrDrawTarget(GrContext* context)
92    : fClip(NULL)
93    , fContext(context) {
94    SkASSERT(NULL != context);
95
96    fDrawState = &fDefaultDrawState;
97    // We assume that fDrawState always owns a ref to the object it points at.
98    fDefaultDrawState.ref();
99    GeometrySrcState& geoSrc = fGeoSrcStateStack.push_back();
100#ifdef SK_DEBUG
101    geoSrc.fVertexCount = DEBUG_INVAL_START_IDX;
102    geoSrc.fVertexBuffer = (GrVertexBuffer*)DEBUG_INVAL_BUFFER;
103    geoSrc.fIndexCount = DEBUG_INVAL_START_IDX;
104    geoSrc.fIndexBuffer = (GrIndexBuffer*)DEBUG_INVAL_BUFFER;
105#endif
106    geoSrc.fVertexSrc = kNone_GeometrySrcType;
107    geoSrc.fIndexSrc  = kNone_GeometrySrcType;
108}
109
110GrDrawTarget::~GrDrawTarget() {
111    SkASSERT(1 == fGeoSrcStateStack.count());
112    SkDEBUGCODE(GeometrySrcState& geoSrc = fGeoSrcStateStack.back());
113    SkASSERT(kNone_GeometrySrcType == geoSrc.fIndexSrc);
114    SkASSERT(kNone_GeometrySrcType == geoSrc.fVertexSrc);
115    fDrawState->unref();
116}
117
118void GrDrawTarget::releaseGeometry() {
119    int popCnt = fGeoSrcStateStack.count() - 1;
120    while (popCnt) {
121        this->popGeometrySource();
122        --popCnt;
123    }
124    this->resetVertexSource();
125    this->resetIndexSource();
126}
127
128void GrDrawTarget::setClip(const GrClipData* clip) {
129    clipWillBeSet(clip);
130    fClip = clip;
131}
132
133const GrClipData* GrDrawTarget::getClip() const {
134    return fClip;
135}
136
137void GrDrawTarget::setDrawState(GrDrawState*  drawState) {
138    SkASSERT(NULL != fDrawState);
139    if (NULL == drawState) {
140        drawState = &fDefaultDrawState;
141    }
142    if (fDrawState != drawState) {
143        fDrawState->unref();
144        drawState->ref();
145        fDrawState = drawState;
146    }
147}
148
149bool GrDrawTarget::reserveVertexSpace(size_t vertexSize,
150                                      int vertexCount,
151                                      void** vertices) {
152    GeometrySrcState& geoSrc = fGeoSrcStateStack.back();
153    bool acquired = false;
154    if (vertexCount > 0) {
155        SkASSERT(NULL != vertices);
156        this->releasePreviousVertexSource();
157        geoSrc.fVertexSrc = kNone_GeometrySrcType;
158
159        acquired = this->onReserveVertexSpace(vertexSize,
160                                              vertexCount,
161                                              vertices);
162    }
163    if (acquired) {
164        geoSrc.fVertexSrc = kReserved_GeometrySrcType;
165        geoSrc.fVertexCount = vertexCount;
166        geoSrc.fVertexSize = vertexSize;
167    } else if (NULL != vertices) {
168        *vertices = NULL;
169    }
170    return acquired;
171}
172
173bool GrDrawTarget::reserveIndexSpace(int indexCount,
174                                     void** indices) {
175    GeometrySrcState& geoSrc = fGeoSrcStateStack.back();
176    bool acquired = false;
177    if (indexCount > 0) {
178        SkASSERT(NULL != indices);
179        this->releasePreviousIndexSource();
180        geoSrc.fIndexSrc = kNone_GeometrySrcType;
181
182        acquired = this->onReserveIndexSpace(indexCount, indices);
183    }
184    if (acquired) {
185        geoSrc.fIndexSrc = kReserved_GeometrySrcType;
186        geoSrc.fIndexCount = indexCount;
187    } else if (NULL != indices) {
188        *indices = NULL;
189    }
190    return acquired;
191
192}
193
194bool GrDrawTarget::reserveVertexAndIndexSpace(int vertexCount,
195                                              int indexCount,
196                                              void** vertices,
197                                              void** indices) {
198    size_t vertexSize = this->drawState()->getVertexSize();
199    this->willReserveVertexAndIndexSpace(vertexCount, indexCount);
200    if (vertexCount) {
201        if (!this->reserveVertexSpace(vertexSize, vertexCount, vertices)) {
202            if (indexCount) {
203                this->resetIndexSource();
204            }
205            return false;
206        }
207    }
208    if (indexCount) {
209        if (!this->reserveIndexSpace(indexCount, indices)) {
210            if (vertexCount) {
211                this->resetVertexSource();
212            }
213            return false;
214        }
215    }
216    return true;
217}
218
219bool GrDrawTarget::geometryHints(int32_t* vertexCount,
220                                 int32_t* indexCount) const {
221    if (NULL != vertexCount) {
222        *vertexCount = -1;
223    }
224    if (NULL != indexCount) {
225        *indexCount = -1;
226    }
227    return false;
228}
229
230void GrDrawTarget::releasePreviousVertexSource() {
231    GeometrySrcState& geoSrc = fGeoSrcStateStack.back();
232    switch (geoSrc.fVertexSrc) {
233        case kNone_GeometrySrcType:
234            break;
235        case kArray_GeometrySrcType:
236            this->releaseVertexArray();
237            break;
238        case kReserved_GeometrySrcType:
239            this->releaseReservedVertexSpace();
240            break;
241        case kBuffer_GeometrySrcType:
242            geoSrc.fVertexBuffer->unref();
243#ifdef SK_DEBUG
244            geoSrc.fVertexBuffer = (GrVertexBuffer*)DEBUG_INVAL_BUFFER;
245#endif
246            break;
247        default:
248            GrCrash("Unknown Vertex Source Type.");
249            break;
250    }
251}
252
253void GrDrawTarget::releasePreviousIndexSource() {
254    GeometrySrcState& geoSrc = fGeoSrcStateStack.back();
255    switch (geoSrc.fIndexSrc) {
256        case kNone_GeometrySrcType:   // these two don't require
257            break;
258        case kArray_GeometrySrcType:
259            this->releaseIndexArray();
260            break;
261        case kReserved_GeometrySrcType:
262            this->releaseReservedIndexSpace();
263            break;
264        case kBuffer_GeometrySrcType:
265            geoSrc.fIndexBuffer->unref();
266#ifdef SK_DEBUG
267            geoSrc.fIndexBuffer = (GrIndexBuffer*)DEBUG_INVAL_BUFFER;
268#endif
269            break;
270        default:
271            GrCrash("Unknown Index Source Type.");
272            break;
273    }
274}
275
276void GrDrawTarget::setVertexSourceToArray(const void* vertexArray,
277                                          int vertexCount) {
278    this->releasePreviousVertexSource();
279    GeometrySrcState& geoSrc = fGeoSrcStateStack.back();
280    geoSrc.fVertexSrc = kArray_GeometrySrcType;
281    geoSrc.fVertexSize = this->drawState()->getVertexSize();
282    geoSrc.fVertexCount = vertexCount;
283    this->onSetVertexSourceToArray(vertexArray, vertexCount);
284}
285
286void GrDrawTarget::setIndexSourceToArray(const void* indexArray,
287                                         int indexCount) {
288    this->releasePreviousIndexSource();
289    GeometrySrcState& geoSrc = fGeoSrcStateStack.back();
290    geoSrc.fIndexSrc = kArray_GeometrySrcType;
291    geoSrc.fIndexCount = indexCount;
292    this->onSetIndexSourceToArray(indexArray, indexCount);
293}
294
295void GrDrawTarget::setVertexSourceToBuffer(const GrVertexBuffer* buffer) {
296    this->releasePreviousVertexSource();
297    GeometrySrcState& geoSrc = fGeoSrcStateStack.back();
298    geoSrc.fVertexSrc    = kBuffer_GeometrySrcType;
299    geoSrc.fVertexBuffer = buffer;
300    buffer->ref();
301    geoSrc.fVertexSize = this->drawState()->getVertexSize();
302}
303
304void GrDrawTarget::setIndexSourceToBuffer(const GrIndexBuffer* buffer) {
305    this->releasePreviousIndexSource();
306    GeometrySrcState& geoSrc = fGeoSrcStateStack.back();
307    geoSrc.fIndexSrc     = kBuffer_GeometrySrcType;
308    geoSrc.fIndexBuffer  = buffer;
309    buffer->ref();
310}
311
312void GrDrawTarget::resetVertexSource() {
313    this->releasePreviousVertexSource();
314    GeometrySrcState& geoSrc = fGeoSrcStateStack.back();
315    geoSrc.fVertexSrc = kNone_GeometrySrcType;
316}
317
318void GrDrawTarget::resetIndexSource() {
319    this->releasePreviousIndexSource();
320    GeometrySrcState& geoSrc = fGeoSrcStateStack.back();
321    geoSrc.fIndexSrc = kNone_GeometrySrcType;
322}
323
324void GrDrawTarget::pushGeometrySource() {
325    this->geometrySourceWillPush();
326    GeometrySrcState& newState = fGeoSrcStateStack.push_back();
327    newState.fIndexSrc = kNone_GeometrySrcType;
328    newState.fVertexSrc = kNone_GeometrySrcType;
329#ifdef SK_DEBUG
330    newState.fVertexCount  = ~0;
331    newState.fVertexBuffer = (GrVertexBuffer*)~0;
332    newState.fIndexCount   = ~0;
333    newState.fIndexBuffer = (GrIndexBuffer*)~0;
334#endif
335}
336
337void GrDrawTarget::popGeometrySource() {
338    // if popping last element then pops are unbalanced with pushes
339    SkASSERT(fGeoSrcStateStack.count() > 1);
340
341    this->geometrySourceWillPop(fGeoSrcStateStack.fromBack(1));
342    this->releasePreviousVertexSource();
343    this->releasePreviousIndexSource();
344    fGeoSrcStateStack.pop_back();
345}
346
347////////////////////////////////////////////////////////////////////////////////
348
349bool GrDrawTarget::checkDraw(GrPrimitiveType type, int startVertex,
350                             int startIndex, int vertexCount,
351                             int indexCount) const {
352    const GrDrawState& drawState = this->getDrawState();
353#ifdef SK_DEBUG
354    const GeometrySrcState& geoSrc = fGeoSrcStateStack.back();
355    int maxVertex = startVertex + vertexCount;
356    int maxValidVertex;
357    switch (geoSrc.fVertexSrc) {
358        case kNone_GeometrySrcType:
359            GrCrash("Attempting to draw without vertex src.");
360        case kReserved_GeometrySrcType: // fallthrough
361        case kArray_GeometrySrcType:
362            maxValidVertex = geoSrc.fVertexCount;
363            break;
364        case kBuffer_GeometrySrcType:
365            maxValidVertex = geoSrc.fVertexBuffer->sizeInBytes() / geoSrc.fVertexSize;
366            break;
367    }
368    if (maxVertex > maxValidVertex) {
369        GrCrash("Drawing outside valid vertex range.");
370    }
371    if (indexCount > 0) {
372        int maxIndex = startIndex + indexCount;
373        int maxValidIndex;
374        switch (geoSrc.fIndexSrc) {
375            case kNone_GeometrySrcType:
376                GrCrash("Attempting to draw indexed geom without index src.");
377            case kReserved_GeometrySrcType: // fallthrough
378            case kArray_GeometrySrcType:
379                maxValidIndex = geoSrc.fIndexCount;
380                break;
381            case kBuffer_GeometrySrcType:
382                maxValidIndex = geoSrc.fIndexBuffer->sizeInBytes() / sizeof(uint16_t);
383                break;
384        }
385        if (maxIndex > maxValidIndex) {
386            GrCrash("Index reads outside valid index range.");
387        }
388    }
389
390    SkASSERT(NULL != drawState.getRenderTarget());
391
392    for (int s = 0; s < drawState.numColorStages(); ++s) {
393        const GrEffectRef& effect = *drawState.getColorStage(s).getEffect();
394        int numTextures = effect->numTextures();
395        for (int t = 0; t < numTextures; ++t) {
396            GrTexture* texture = effect->texture(t);
397            SkASSERT(texture->asRenderTarget() != drawState.getRenderTarget());
398        }
399    }
400    for (int s = 0; s < drawState.numCoverageStages(); ++s) {
401        const GrEffectRef& effect = *drawState.getCoverageStage(s).getEffect();
402        int numTextures = effect->numTextures();
403        for (int t = 0; t < numTextures; ++t) {
404            GrTexture* texture = effect->texture(t);
405            SkASSERT(texture->asRenderTarget() != drawState.getRenderTarget());
406        }
407    }
408
409    SkASSERT(drawState.validateVertexAttribs());
410#endif
411    if (NULL == drawState.getRenderTarget()) {
412        return false;
413    }
414    return true;
415}
416
417bool GrDrawTarget::setupDstReadIfNecessary(GrDeviceCoordTexture* dstCopy, const SkRect* drawBounds) {
418    if (this->caps()->dstReadInShaderSupport() || !this->getDrawState().willEffectReadDstColor()) {
419        return true;
420    }
421    GrRenderTarget* rt = this->drawState()->getRenderTarget();
422    SkIRect copyRect;
423    const GrClipData* clip = this->getClip();
424    clip->getConservativeBounds(rt, &copyRect);
425
426    if (NULL != drawBounds) {
427        SkIRect drawIBounds;
428        drawBounds->roundOut(&drawIBounds);
429        if (!copyRect.intersect(drawIBounds)) {
430#ifdef SK_DEBUG
431            GrPrintf("Missed an early reject. Bailing on draw from setupDstReadIfNecessary.\n");
432#endif
433            return false;
434        }
435    } else {
436#ifdef SK_DEBUG
437        //GrPrintf("No dev bounds when dst copy is made.\n");
438#endif
439    }
440
441    // MSAA consideration: When there is support for reading MSAA samples in the shader we could
442    // have per-sample dst values by making the copy multisampled.
443    GrTextureDesc desc;
444    this->initCopySurfaceDstDesc(rt, &desc);
445    desc.fWidth = copyRect.width();
446    desc.fHeight = copyRect.height();
447
448    GrAutoScratchTexture ast(fContext, desc, GrContext::kApprox_ScratchTexMatch);
449
450    if (NULL == ast.texture()) {
451        GrPrintf("Failed to create temporary copy of destination texture.\n");
452        return false;
453    }
454    SkIPoint dstPoint = {0, 0};
455    if (this->copySurface(ast.texture(), rt, copyRect, dstPoint)) {
456        dstCopy->setTexture(ast.texture());
457        dstCopy->setOffset(copyRect.fLeft, copyRect.fTop);
458        return true;
459    } else {
460        return false;
461    }
462}
463
464void GrDrawTarget::drawIndexed(GrPrimitiveType type,
465                               int startVertex,
466                               int startIndex,
467                               int vertexCount,
468                               int indexCount,
469                               const SkRect* devBounds) {
470    if (indexCount > 0 && this->checkDraw(type, startVertex, startIndex, vertexCount, indexCount)) {
471        DrawInfo info;
472        info.fPrimitiveType = type;
473        info.fStartVertex   = startVertex;
474        info.fStartIndex    = startIndex;
475        info.fVertexCount   = vertexCount;
476        info.fIndexCount    = indexCount;
477
478        info.fInstanceCount         = 0;
479        info.fVerticesPerInstance   = 0;
480        info.fIndicesPerInstance    = 0;
481
482        if (NULL != devBounds) {
483            info.setDevBounds(*devBounds);
484        }
485        // TODO: We should continue with incorrect blending.
486        if (!this->setupDstReadIfNecessary(&info)) {
487            return;
488        }
489        this->onDraw(info);
490    }
491}
492
493void GrDrawTarget::drawNonIndexed(GrPrimitiveType type,
494                                  int startVertex,
495                                  int vertexCount,
496                                  const SkRect* devBounds) {
497    if (vertexCount > 0 && this->checkDraw(type, startVertex, -1, vertexCount, -1)) {
498        DrawInfo info;
499        info.fPrimitiveType = type;
500        info.fStartVertex   = startVertex;
501        info.fStartIndex    = 0;
502        info.fVertexCount   = vertexCount;
503        info.fIndexCount    = 0;
504
505        info.fInstanceCount         = 0;
506        info.fVerticesPerInstance   = 0;
507        info.fIndicesPerInstance    = 0;
508
509        if (NULL != devBounds) {
510            info.setDevBounds(*devBounds);
511        }
512        // TODO: We should continue with incorrect blending.
513        if (!this->setupDstReadIfNecessary(&info)) {
514            return;
515        }
516        this->onDraw(info);
517    }
518}
519
520void GrDrawTarget::stencilPath(const GrPath* path, const SkStrokeRec& stroke, SkPath::FillType fill) {
521    // TODO: extract portions of checkDraw that are relevant to path stenciling.
522    SkASSERT(NULL != path);
523    SkASSERT(this->caps()->pathRenderingSupport());
524    SkASSERT(!stroke.isHairlineStyle());
525    SkASSERT(!SkPath::IsInverseFillType(fill));
526    this->onStencilPath(path, stroke, fill);
527}
528
529void GrDrawTarget::fillPath(const GrPath* path, const SkStrokeRec& stroke, SkPath::FillType fill) {
530    // TODO: extract portions of checkDraw that are relevant to path rendering.
531    SkASSERT(NULL != path);
532    SkASSERT(this->caps()->pathRenderingSupport());
533    SkASSERT(!stroke.isHairlineStyle());
534    const GrDrawState* drawState = &getDrawState();
535
536    SkRect devBounds;
537    if (SkPath::IsInverseFillType(fill)) {
538        devBounds = SkRect::MakeWH(SkIntToScalar(drawState->getRenderTarget()->width()),
539                                   SkIntToScalar(drawState->getRenderTarget()->height()));
540    } else {
541        devBounds = path->getBounds();
542    }
543    SkMatrix viewM = drawState->getViewMatrix();
544    viewM.mapRect(&devBounds);
545
546    GrDeviceCoordTexture dstCopy;
547    if (!this->setupDstReadIfNecessary(&dstCopy, &devBounds)) {
548        return;
549    }
550
551    this->onFillPath(path, stroke, fill, dstCopy.texture() ? &dstCopy : NULL);
552}
553
554////////////////////////////////////////////////////////////////////////////////
555
556bool GrDrawTarget::willUseHWAALines() const {
557    // There is a conflict between using smooth lines and our use of premultiplied alpha. Smooth
558    // lines tweak the incoming alpha value but not in a premul-alpha way. So we only use them when
559    // our alpha is 0xff and tweaking the color for partial coverage is OK
560    if (!this->caps()->hwAALineSupport() ||
561        !this->getDrawState().isHWAntialiasState()) {
562        return false;
563    }
564    GrDrawState::BlendOptFlags opts = this->getDrawState().getBlendOpts();
565    return (GrDrawState::kDisableBlend_BlendOptFlag & opts) &&
566           (GrDrawState::kCoverageAsAlpha_BlendOptFlag & opts);
567}
568
569bool GrDrawTarget::canApplyCoverage() const {
570    // we can correctly apply coverage if a) we have dual source blending
571    // or b) one of our blend optimizations applies.
572    return this->caps()->dualSourceBlendingSupport() ||
573           GrDrawState::kNone_BlendOpt != this->getDrawState().getBlendOpts(true);
574}
575
576////////////////////////////////////////////////////////////////////////////////
577
578void GrDrawTarget::drawIndexedInstances(GrPrimitiveType type,
579                                        int instanceCount,
580                                        int verticesPerInstance,
581                                        int indicesPerInstance,
582                                        const SkRect* devBounds) {
583    if (!verticesPerInstance || !indicesPerInstance) {
584        return;
585    }
586
587    int maxInstancesPerDraw = this->indexCountInCurrentSource() / indicesPerInstance;
588    if (!maxInstancesPerDraw) {
589        return;
590    }
591
592    DrawInfo info;
593    info.fPrimitiveType = type;
594    info.fStartIndex = 0;
595    info.fStartVertex = 0;
596    info.fIndicesPerInstance = indicesPerInstance;
597    info.fVerticesPerInstance = verticesPerInstance;
598
599    // Set the same bounds for all the draws.
600    if (NULL != devBounds) {
601        info.setDevBounds(*devBounds);
602    }
603    // TODO: We should continue with incorrect blending.
604    if (!this->setupDstReadIfNecessary(&info)) {
605        return;
606    }
607
608    while (instanceCount) {
609        info.fInstanceCount = GrMin(instanceCount, maxInstancesPerDraw);
610        info.fVertexCount = info.fInstanceCount * verticesPerInstance;
611        info.fIndexCount = info.fInstanceCount * indicesPerInstance;
612
613        if (this->checkDraw(type,
614                            info.fStartVertex,
615                            info.fStartIndex,
616                            info.fVertexCount,
617                            info.fIndexCount)) {
618            this->onDraw(info);
619        }
620        info.fStartVertex += info.fVertexCount;
621        instanceCount -= info.fInstanceCount;
622    }
623}
624
625////////////////////////////////////////////////////////////////////////////////
626
627namespace {
628
629// position + (optional) texture coord
630extern const GrVertexAttrib gBWRectPosUVAttribs[] = {
631    {kVec2f_GrVertexAttribType, 0,               kPosition_GrVertexAttribBinding},
632    {kVec2f_GrVertexAttribType, sizeof(GrPoint), kLocalCoord_GrVertexAttribBinding}
633};
634
635void set_vertex_attributes(GrDrawState* drawState, bool hasUVs) {
636    if (hasUVs) {
637        drawState->setVertexAttribs<gBWRectPosUVAttribs>(2);
638    } else {
639        drawState->setVertexAttribs<gBWRectPosUVAttribs>(1);
640    }
641}
642
643};
644
645void GrDrawTarget::onDrawRect(const SkRect& rect,
646                              const SkMatrix* matrix,
647                              const SkRect* localRect,
648                              const SkMatrix* localMatrix) {
649
650    GrDrawState::AutoViewMatrixRestore avmr;
651    if (NULL != matrix) {
652        avmr.set(this->drawState(), *matrix);
653    }
654
655    set_vertex_attributes(this->drawState(), NULL != localRect);
656
657    AutoReleaseGeometry geo(this, 4, 0);
658    if (!geo.succeeded()) {
659        GrPrintf("Failed to get space for vertices!\n");
660        return;
661    }
662
663    size_t vsize = this->drawState()->getVertexSize();
664    geo.positions()->setRectFan(rect.fLeft, rect.fTop, rect.fRight, rect.fBottom, vsize);
665    if (NULL != localRect) {
666        GrPoint* coords = GrTCast<GrPoint*>(GrTCast<intptr_t>(geo.vertices()) +
667                                            sizeof(GrPoint));
668        coords->setRectFan(localRect->fLeft, localRect->fTop,
669                           localRect->fRight, localRect->fBottom,
670                           vsize);
671        if (NULL != localMatrix) {
672            localMatrix->mapPointsWithStride(coords, vsize, 4);
673        }
674    }
675    SkTLazy<SkRect> bounds;
676    if (this->getDrawState().willEffectReadDstColor()) {
677        bounds.init();
678        this->getDrawState().getViewMatrix().mapRect(bounds.get(), rect);
679    }
680
681    this->drawNonIndexed(kTriangleFan_GrPrimitiveType, 0, 4, bounds.getMaybeNull());
682}
683
684void GrDrawTarget::clipWillBeSet(const GrClipData* clipData) {
685}
686
687////////////////////////////////////////////////////////////////////////////////
688
689GrDrawTarget::AutoStateRestore::AutoStateRestore() {
690    fDrawTarget = NULL;
691}
692
693GrDrawTarget::AutoStateRestore::AutoStateRestore(GrDrawTarget* target,
694                                                 ASRInit init,
695                                                 const SkMatrix* vm) {
696    fDrawTarget = NULL;
697    this->set(target, init, vm);
698}
699
700GrDrawTarget::AutoStateRestore::~AutoStateRestore() {
701    if (NULL != fDrawTarget) {
702        fDrawTarget->setDrawState(fSavedState);
703        fSavedState->unref();
704    }
705}
706
707void GrDrawTarget::AutoStateRestore::set(GrDrawTarget* target, ASRInit init, const SkMatrix* vm) {
708    SkASSERT(NULL == fDrawTarget);
709    fDrawTarget = target;
710    fSavedState = target->drawState();
711    SkASSERT(fSavedState);
712    fSavedState->ref();
713    if (kReset_ASRInit == init) {
714        if (NULL == vm) {
715            // calls the default cons
716            fTempState.init();
717        } else {
718            SkNEW_IN_TLAZY(&fTempState, GrDrawState, (*vm));
719        }
720    } else {
721        SkASSERT(kPreserve_ASRInit == init);
722        if (NULL == vm) {
723            fTempState.set(*fSavedState);
724        } else {
725            SkNEW_IN_TLAZY(&fTempState, GrDrawState, (*fSavedState, *vm));
726        }
727    }
728    target->setDrawState(fTempState.get());
729}
730
731bool GrDrawTarget::AutoStateRestore::setIdentity(GrDrawTarget* target, ASRInit init) {
732    SkASSERT(NULL == fDrawTarget);
733    fDrawTarget = target;
734    fSavedState = target->drawState();
735    SkASSERT(fSavedState);
736    fSavedState->ref();
737    if (kReset_ASRInit == init) {
738        // calls the default cons
739        fTempState.init();
740    } else {
741        SkASSERT(kPreserve_ASRInit == init);
742        // calls the copy cons
743        fTempState.set(*fSavedState);
744        if (!fTempState.get()->setIdentityViewMatrix()) {
745            // let go of any resources held by the temp
746            fTempState.get()->reset();
747            fDrawTarget = NULL;
748            fSavedState->unref();
749            fSavedState = NULL;
750            return false;
751        }
752    }
753    target->setDrawState(fTempState.get());
754    return true;
755}
756
757////////////////////////////////////////////////////////////////////////////////
758
759GrDrawTarget::AutoReleaseGeometry::AutoReleaseGeometry(
760                                         GrDrawTarget*  target,
761                                         int vertexCount,
762                                         int indexCount) {
763    fTarget = NULL;
764    this->set(target, vertexCount, indexCount);
765}
766
767GrDrawTarget::AutoReleaseGeometry::AutoReleaseGeometry() {
768    fTarget = NULL;
769}
770
771GrDrawTarget::AutoReleaseGeometry::~AutoReleaseGeometry() {
772    this->reset();
773}
774
775bool GrDrawTarget::AutoReleaseGeometry::set(GrDrawTarget*  target,
776                                            int vertexCount,
777                                            int indexCount) {
778    this->reset();
779    fTarget = target;
780    bool success = true;
781    if (NULL != fTarget) {
782        fTarget = target;
783        success = target->reserveVertexAndIndexSpace(vertexCount,
784                                                     indexCount,
785                                                     &fVertices,
786                                                     &fIndices);
787        if (!success) {
788            fTarget = NULL;
789            this->reset();
790        }
791    }
792    SkASSERT(success == (NULL != fTarget));
793    return success;
794}
795
796void GrDrawTarget::AutoReleaseGeometry::reset() {
797    if (NULL != fTarget) {
798        if (NULL != fVertices) {
799            fTarget->resetVertexSource();
800        }
801        if (NULL != fIndices) {
802            fTarget->resetIndexSource();
803        }
804        fTarget = NULL;
805    }
806    fVertices = NULL;
807    fIndices = NULL;
808}
809
810GrDrawTarget::AutoClipRestore::AutoClipRestore(GrDrawTarget* target, const SkIRect& newClip) {
811    fTarget = target;
812    fClip = fTarget->getClip();
813    fStack.init();
814    fStack.get()->clipDevRect(newClip, SkRegion::kReplace_Op);
815    fReplacementClip.fClipStack = fStack.get();
816    target->setClip(&fReplacementClip);
817}
818
819namespace {
820// returns true if the read/written rect intersects the src/dst and false if not.
821bool clip_srcrect_and_dstpoint(const GrSurface* dst,
822                               const GrSurface* src,
823                               const SkIRect& srcRect,
824                               const SkIPoint& dstPoint,
825                               SkIRect* clippedSrcRect,
826                               SkIPoint* clippedDstPoint) {
827    *clippedSrcRect = srcRect;
828    *clippedDstPoint = dstPoint;
829
830    // clip the left edge to src and dst bounds, adjusting dstPoint if necessary
831    if (clippedSrcRect->fLeft < 0) {
832        clippedDstPoint->fX -= clippedSrcRect->fLeft;
833        clippedSrcRect->fLeft = 0;
834    }
835    if (clippedDstPoint->fX < 0) {
836        clippedSrcRect->fLeft -= clippedDstPoint->fX;
837        clippedDstPoint->fX = 0;
838    }
839
840    // clip the top edge to src and dst bounds, adjusting dstPoint if necessary
841    if (clippedSrcRect->fTop < 0) {
842        clippedDstPoint->fY -= clippedSrcRect->fTop;
843        clippedSrcRect->fTop = 0;
844    }
845    if (clippedDstPoint->fY < 0) {
846        clippedSrcRect->fTop -= clippedDstPoint->fY;
847        clippedDstPoint->fY = 0;
848    }
849
850    // clip the right edge to the src and dst bounds.
851    if (clippedSrcRect->fRight > src->width()) {
852        clippedSrcRect->fRight = src->width();
853    }
854    if (clippedDstPoint->fX + clippedSrcRect->width() > dst->width()) {
855        clippedSrcRect->fRight = clippedSrcRect->fLeft + dst->width() - clippedDstPoint->fX;
856    }
857
858    // clip the bottom edge to the src and dst bounds.
859    if (clippedSrcRect->fBottom > src->height()) {
860        clippedSrcRect->fBottom = src->height();
861    }
862    if (clippedDstPoint->fY + clippedSrcRect->height() > dst->height()) {
863        clippedSrcRect->fBottom = clippedSrcRect->fTop + dst->height() - clippedDstPoint->fY;
864    }
865
866    // The above clipping steps may have inverted the rect if it didn't intersect either the src or
867    // dst bounds.
868    return !clippedSrcRect->isEmpty();
869}
870}
871
872bool GrDrawTarget::copySurface(GrSurface* dst,
873                               GrSurface* src,
874                               const SkIRect& srcRect,
875                               const SkIPoint& dstPoint) {
876    SkASSERT(NULL != dst);
877    SkASSERT(NULL != src);
878
879    SkIRect clippedSrcRect;
880    SkIPoint clippedDstPoint;
881    // If the rect is outside the src or dst then we've already succeeded.
882    if (!clip_srcrect_and_dstpoint(dst,
883                                   src,
884                                   srcRect,
885                                   dstPoint,
886                                   &clippedSrcRect,
887                                   &clippedDstPoint)) {
888        SkASSERT(this->canCopySurface(dst, src, srcRect, dstPoint));
889        return true;
890    }
891
892    bool result = this->onCopySurface(dst, src, clippedSrcRect, clippedDstPoint);
893    SkASSERT(result == this->canCopySurface(dst, src, clippedSrcRect, clippedDstPoint));
894    return result;
895}
896
897bool GrDrawTarget::canCopySurface(GrSurface* dst,
898                                  GrSurface* src,
899                                  const SkIRect& srcRect,
900                                  const SkIPoint& dstPoint) {
901    SkASSERT(NULL != dst);
902    SkASSERT(NULL != src);
903
904    SkIRect clippedSrcRect;
905    SkIPoint clippedDstPoint;
906    // If the rect is outside the src or dst then we're guaranteed success
907    if (!clip_srcrect_and_dstpoint(dst,
908                                   src,
909                                   srcRect,
910                                   dstPoint,
911                                   &clippedSrcRect,
912                                   &clippedDstPoint)) {
913        return true;
914    }
915    return this->onCanCopySurface(dst, src, clippedSrcRect, clippedDstPoint);
916}
917
918bool GrDrawTarget::onCanCopySurface(GrSurface* dst,
919                                    GrSurface* src,
920                                    const SkIRect& srcRect,
921                                    const SkIPoint& dstPoint) {
922    // Check that the read/write rects are contained within the src/dst bounds.
923    SkASSERT(!srcRect.isEmpty());
924    SkASSERT(SkIRect::MakeWH(src->width(), src->height()).contains(srcRect));
925    SkASSERT(dstPoint.fX >= 0 && dstPoint.fY >= 0);
926    SkASSERT(dstPoint.fX + srcRect.width() <= dst->width() &&
927             dstPoint.fY + srcRect.height() <= dst->height());
928
929    return !dst->isSameAs(src) && NULL != dst->asRenderTarget() && NULL != src->asTexture();
930}
931
932bool GrDrawTarget::onCopySurface(GrSurface* dst,
933                                 GrSurface* src,
934                                 const SkIRect& srcRect,
935                                 const SkIPoint& dstPoint) {
936    if (!GrDrawTarget::onCanCopySurface(dst, src, srcRect, dstPoint)) {
937        return false;
938    }
939
940    GrRenderTarget* rt = dst->asRenderTarget();
941    GrTexture* tex = src->asTexture();
942
943    GrDrawTarget::AutoStateRestore asr(this, kReset_ASRInit);
944    this->drawState()->setRenderTarget(rt);
945    SkMatrix matrix;
946    matrix.setTranslate(SkIntToScalar(srcRect.fLeft - dstPoint.fX),
947                        SkIntToScalar(srcRect.fTop - dstPoint.fY));
948    matrix.postIDiv(tex->width(), tex->height());
949    this->drawState()->addColorTextureEffect(tex, matrix);
950    SkIRect dstRect = SkIRect::MakeXYWH(dstPoint.fX,
951                                        dstPoint.fY,
952                                        srcRect.width(),
953                                        srcRect.height());
954    this->drawSimpleRect(dstRect);
955    return true;
956}
957
958void GrDrawTarget::initCopySurfaceDstDesc(const GrSurface* src, GrTextureDesc* desc) {
959    // Make the dst of the copy be a render target because the default copySurface draws to the dst.
960    desc->fOrigin = kDefault_GrSurfaceOrigin;
961    desc->fFlags = kRenderTarget_GrTextureFlagBit | kNoStencil_GrTextureFlagBit;
962    desc->fConfig = src->config();
963}
964
965///////////////////////////////////////////////////////////////////////////////
966
967SK_DEFINE_INST_COUNT(GrDrawTargetCaps)
968
969void GrDrawTargetCaps::reset() {
970    f8BitPaletteSupport = false;
971    fNPOTTextureTileSupport = false;
972    fTwoSidedStencilSupport = false;
973    fStencilWrapOpsSupport = false;
974    fHWAALineSupport = false;
975    fShaderDerivativeSupport = false;
976    fGeometryShaderSupport = false;
977    fDualSourceBlendingSupport = false;
978    fBufferLockSupport = false;
979    fPathRenderingSupport = false;
980    fDstReadInShaderSupport = false;
981    fReuseScratchTextures = true;
982
983    fMaxRenderTargetSize = 0;
984    fMaxTextureSize = 0;
985    fMaxSampleCount = 0;
986}
987
988GrDrawTargetCaps& GrDrawTargetCaps::operator=(const GrDrawTargetCaps& other) {
989    f8BitPaletteSupport = other.f8BitPaletteSupport;
990    fNPOTTextureTileSupport = other.fNPOTTextureTileSupport;
991    fTwoSidedStencilSupport = other.fTwoSidedStencilSupport;
992    fStencilWrapOpsSupport = other.fStencilWrapOpsSupport;
993    fHWAALineSupport = other.fHWAALineSupport;
994    fShaderDerivativeSupport = other.fShaderDerivativeSupport;
995    fGeometryShaderSupport = other.fGeometryShaderSupport;
996    fDualSourceBlendingSupport = other.fDualSourceBlendingSupport;
997    fBufferLockSupport = other.fBufferLockSupport;
998    fPathRenderingSupport = other.fPathRenderingSupport;
999    fDstReadInShaderSupport = other.fDstReadInShaderSupport;
1000    fReuseScratchTextures = other.fReuseScratchTextures;
1001
1002    fMaxRenderTargetSize = other.fMaxRenderTargetSize;
1003    fMaxTextureSize = other.fMaxTextureSize;
1004    fMaxSampleCount = other.fMaxSampleCount;
1005
1006    return *this;
1007}
1008
1009void GrDrawTargetCaps::print() const {
1010    static const char* gNY[] = {"NO", "YES"};
1011    GrPrintf("8 Bit Palette Support       : %s\n", gNY[f8BitPaletteSupport]);
1012    GrPrintf("NPOT Texture Tile Support   : %s\n", gNY[fNPOTTextureTileSupport]);
1013    GrPrintf("Two Sided Stencil Support   : %s\n", gNY[fTwoSidedStencilSupport]);
1014    GrPrintf("Stencil Wrap Ops  Support   : %s\n", gNY[fStencilWrapOpsSupport]);
1015    GrPrintf("HW AA Lines Support         : %s\n", gNY[fHWAALineSupport]);
1016    GrPrintf("Shader Derivative Support   : %s\n", gNY[fShaderDerivativeSupport]);
1017    GrPrintf("Geometry Shader Support     : %s\n", gNY[fGeometryShaderSupport]);
1018    GrPrintf("Dual Source Blending Support: %s\n", gNY[fDualSourceBlendingSupport]);
1019    GrPrintf("Buffer Lock Support         : %s\n", gNY[fBufferLockSupport]);
1020    GrPrintf("Path Rendering Support      : %s\n", gNY[fPathRenderingSupport]);
1021    GrPrintf("Dst Read In Shader Support  : %s\n", gNY[fDstReadInShaderSupport]);
1022    GrPrintf("Reuse Scratch Textures      : %s\n", gNY[fReuseScratchTextures]);
1023    GrPrintf("Max Texture Size            : %d\n", fMaxTextureSize);
1024    GrPrintf("Max Render Target Size      : %d\n", fMaxRenderTargetSize);
1025    GrPrintf("Max Sample Count            : %d\n", fMaxSampleCount);
1026}
1027