Filtrer la légende PyQGIS

Oct 22 2020

J'ai effectué un selectByExpression sur l'un de mes calques, puis j'ai créé un nouveau calque. couche 1: co, couche 2: co_temp

  co.selectByExpression("SECTION=1")
  co_temp = processing.run("native:saveselectedfeatures", {'INPUT': co, 'OUTPUT': 'memory:'})['OUTPUT']
  co.removeSelection()
  renderer = co.renderer().clone()
  co_temp.setRenderer(renderer)
  co_temp.renderer().filter()
  co_temp.triggerRepaint()
  co.removeSelection()

Je voudrais pouvoir créer une légende co_temp, mais uniquement pour les expressions co_temp sélectionnées. Il y a beaucoup d'informations inutiles dans co que je ne voudrais pas récupérer pour la légende co_temp.

mapt = QgsLayoutItemMap(layout)
mapt.attemptSetSceneRect(QRectF(20, 20, 80, 80))
mapt.setFrameEnabled(True)
mapt.setLayers([co_temp])
mapt.setCrs(co_temp.crs())
layout.addLayoutItem(mapt)
mapt.setExtent(co_temp.extent().buffered(250))

legend = QgsLayoutItemLegend(layout)
legend.setTitle("Legend")
legend.attemptSetSceneRect(QRectF(120, 20, 80, 80))
legend.setFrameEnabled(True)
legend.setBackgroundColor(QColor(200, 200, 200))
legend.setTitle('')
legend.setLegendFilterByMapEnabled(True)

legend.setResizeToContents(False)

layout.addLayoutItem(legend)
legend.setLinkedMap(mapt)

J'ai essayé ça mais ça ne marche pas.

co:

co légende:

co_temp:

légende co_temp:

Réponses

1 mathiasmirra Oct 28 2020 at 17:09

J'ai finalement réussi à filtrer ma légende, je partage mon code avec vous. co: couche 1 co_temp: couche 2

            co.selectByExpression("SECTION=1")
            co_temp = processing.run("native:saveselectedfeatures", {'INPUT': co, 
            'OUTPUT': 'memory:'})['OUTPUT']
            co.removeSelection()

            renderer = co.renderer().clone()
            renderer2 = co.renderer().clone()
            renderer3 = co.renderer().clone()
            renderer2.deleteAllCategories()
            renderer3.deleteAllCategories()
            co_temp.setRenderer(renderer)
            ctx = QgsRenderContext()
            table = []
            table2 = []

            for feature in co_temp.getFeatures():
                for i in renderer.categories():
                    renderer3.addCategory(i)
                    #####################################
                    renderer3.startRender(ctx, QgsFields())
                    ctx.expressionContext().setFeature(feature)
                    #####################################
                    if renderer3.willRenderFeature(feature, ctx):
                        if i.label() not in table and i.label() != "":
                            table.append(i.label())
                            table2.append(i)
                    #####################################
                    renderer3.stopRender(ctx)
                    #####################################
                    renderer3.deleteAllCategories()

            for i in table2:
                i.setLabel(i.label().lower())
                renderer2.addCategory(i)

            co_temp.setRenderer(renderer2)
            co_temp.triggerRepaint()