Add "sample locations" feature to GrProcessor

src/gpu/GrGpu.cpp

 /*
  * Copyright 2010 Google Inc.
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */
 
 
 #include "GrGpu.h"
 
@@ skipping 29 matching lines @@
     fIndexBuffer.reset(di.indexBuffer());
 
     return *this;
 }
 
 ////////////////////////////////////////////////////////////////////////////////
 
 GrGpu::GrGpu(GrContext* context)
     : fResetTimestamp(kExpiredTimestamp+1)
     , fResetBits(kAll_GrBackendState)
-    , fContext(context) {
+    , fContext(context)
+    , fNextSamplePatternID(1) {
 }
 
 GrGpu::~GrGpu() {}
 
 void GrGpu::contextAbandoned() {}
 
 ////////////////////////////////////////////////////////////////////////////////
 
 bool GrGpu::makeCopyForTextureParams(int width, int height, const GrTextureParams& textureParams,
                                      GrTextureProducer::CopyParams* copyParams) const {
@@ skipping 376 matching lines @@
     }
     return false;
 }
 
 void GrGpu::resolveRenderTarget(GrRenderTarget* target) {
     SkASSERT(target);
     this->handleDirtyContext();
     this->onResolveRenderTarget(target);
 }
 
+const GrGpu::MultisampleSpecs& GrGpu::getMultisampleSpecs(GrRenderTarget* rt,
+                                                          const GrStencilSettings& stencil) {
+    skstd::unique_ptr<MultisampleSpecs>* specsPtr;
+    bool willUseRasterMultisample = rt->hasMixedSamples() && stencil.isDisabled();
+    if (willUseRasterMultisample) {
+        int rasterSampleCnt = rt->desc().fSampleCnt;
+        SkASSERT(rasterSampleCnt > 0 && rasterSampleCnt <= this->caps()->maxRasterSamples());
+        fRasterMultisampleSpecs.resize_back(SkTMax(fRasterMultisampleSpecs.count(),
+                                                   rasterSampleCnt));
+        specsPtr = &fRasterMultisampleSpecs[rasterSampleCnt - 1];
+    } else {
+        specsPtr = rt->renderTargetPriv().accessMultisampleSpecs();
+    }
+    if (!specsPtr->get()) {
+        MultisampleSpecs* specs = new MultisampleSpecs;
+        SkSTArray<16, SkPoint, true> samplePattern;
+        this->onGetMultisampleSpecs(rt, stencil, &specs->fEffectiveSampleCnt, &samplePattern);
+        if (this->caps()->sampleLocationsSupport()) {
+            SkASSERT(!samplePattern.empty());
+            const auto& result = fKnownSamplePatterns.emplace(samplePattern, fNextSamplePatternID);
+            if (result.second) {
+                // This sample pattern didn't exist already in the map. (We don't expect to see very
+                // many unique sample patterns).
+                SkASSERT(fKnownSamplePatterns.size() < 65535);
+                do {} while (0 == ++fNextSamplePatternID);
+            }
+            specs->fSamplePatternID = result.first->second;
+            specs->fSampleLocations = result.first->first.begin();
+        }
+        specsPtr->reset(specs);
+    }
+    return *specsPtr->get();
+}
+
+bool GrGpu::SamplePatternComparator::operator()(const SamplePattern& a,
+                                                const SamplePattern& b) const {
+    if (a.count() != b.count()) {
+        return a.count() < b.count();
+    }
+    for (int i = 0; i < a.count(); ++i) {
+        // This doesn't have geometric meaning. We just need to define an ordering for std::map.
+        if (a[i].x() != b[i].x()) {
+            return a[i].x() < b[i].x();
+        }
+        if (a[i].y() != b[i].y()) {
+            return a[i].y() < b[i].y();
+        }
+    }
+    return false; // Equal.
+}
+
 ////////////////////////////////////////////////////////////////////////////////
 
 void GrGpu::draw(const DrawArgs& args, const GrVertices& vertices) {
     this->handleDirtyContext();
     if (GrXferBarrierType barrierType = args.fPipeline->xferBarrierType(*this->caps())) {
         this->xferBarrier(args.fPipeline->getRenderTarget(), barrierType);
     }
 
     GrVertices::Iterator iter;
     const GrNonInstancedVertices* verts = iter.init(vertices);
     do {
         this->onDraw(args, *verts);
         fStats.incNumDraws();
     } while ((verts = iter.next()));
 }