Пример #1
0
void QgsCategorizedSymbolRendererV2Widget::addCategories()
{
    QString attrName = cboCategorizedColumn->currentText();
    int idx = mLayer->fieldNameIndex( attrName );
    QList<QVariant> unique_vals;
    mLayer->uniqueValues( idx, unique_vals );

    //DlgAddCategories dlg(mStyle, createDefaultSymbol(), unique_vals, this);
    //if (!dlg.exec())
    //  return;

    QgsVectorColorRampV2* ramp = cboCategorizedColorRamp->currentColorRamp();

    if ( ramp == NULL )
    {
        if ( cboCategorizedColorRamp->count() == 0 )
            QMessageBox::critical( this, tr( "Error" ), tr( "There are no available color ramps. You can add them in Style Manager." ) );
        else
            QMessageBox::critical( this, tr( "Error" ), tr( "The selected color ramp is not available." ) );
        return;
    }

    QgsCategoryList cats;
    _createCategories( cats, unique_vals, mCategorizedSymbol, ramp );

    bool deleteExisting = false;

    if ( !mOldClassificationAttribute.isEmpty() &&
            attrName != mOldClassificationAttribute &&
            mRenderer->categories().count() > 0 )
    {
        int res = QMessageBox::question( this,
                                         tr( "Confirm Delete" ),
                                         tr( "The classification field was changed from '%1' to '%2'.\n"
                                             "Should the existing classes be deleted before classification?" )
                                         .arg( mOldClassificationAttribute ).arg( attrName ),
                                         QMessageBox::Yes | QMessageBox::No | QMessageBox::Cancel );
        if ( res == QMessageBox::Cancel )
        {
            return;
        }

        deleteExisting = ( res == QMessageBox::Yes );
    }

    if ( !deleteExisting )
    {
        QgsCategoryList prevCats = mRenderer->categories();
        for ( int i = 0; i < cats.size(); ++i )
        {
            bool contains = false;
            QVariant value = cats.at( i ).value();
            for ( int j = 0; j < prevCats.size() && !contains; ++j )
            {
                if ( prevCats.at( j ).value() == value )
                {
                    contains = true;
                    break;
                }
            }

            if ( !contains )
                prevCats.append( cats.at( i ) );
        }
        cats = prevCats;
    }

    mOldClassificationAttribute = attrName;

    // TODO: if not all categories are desired, delete some!
    /*
    if (not dlg.readAllCats.isChecked())
    {
      cats2 = {}
      for item in dlg.listCategories.selectedItems():
        for k,c in cats.iteritems():
          if item.text() == k.toString():
            break
        cats2[k] = c
      cats = cats2
    }
    */

    // recreate renderer
    QgsCategorizedSymbolRendererV2 *r = new QgsCategorizedSymbolRendererV2( attrName, cats );
    r->setSourceSymbol( mCategorizedSymbol->clone() );
    r->setSourceColorRamp( ramp->clone() );
    r->setScaleMethod( mRenderer->scaleMethod() );
    r->setSizeScaleField( mRenderer->sizeScaleField() );
    r->setRotationField( mRenderer->rotationField() );
    delete mRenderer;
    mRenderer = r;

    populateCategories();
}
QgsFeatureRendererV2* QgsCategorizedSymbolRendererV2::create( QDomElement& element )
{
  QDomElement symbolsElem = element.firstChildElement( "symbols" );
  if ( symbolsElem.isNull() )
    return NULL;

  QDomElement catsElem = element.firstChildElement( "categories" );
  if ( catsElem.isNull() )
    return NULL;

  QgsSymbolV2Map symbolMap = QgsSymbolLayerV2Utils::loadSymbols( symbolsElem );
  QgsCategoryList cats;

  QDomElement catElem = catsElem.firstChildElement();
  while ( !catElem.isNull() )
  {
    if ( catElem.tagName() == "category" )
    {
      QVariant value = QVariant( catElem.attribute( "value" ) );
      QString symbolName = catElem.attribute( "symbol" );
      QString label = catElem.attribute( "label" );
      if ( symbolMap.contains( symbolName ) )
      {
        QgsSymbolV2* symbol = symbolMap.take( symbolName );
        cats.append( QgsRendererCategoryV2( value, symbol, label ) );
      }
    }
    catElem = catElem.nextSiblingElement();
  }

  QString attrName = element.attribute( "attr" );

  QgsCategorizedSymbolRendererV2* r = new QgsCategorizedSymbolRendererV2( attrName, cats );

  // delete symbols if there are any more
  QgsSymbolLayerV2Utils::clearSymbolMap( symbolMap );

  // try to load source symbol (optional)
  QDomElement sourceSymbolElem = element.firstChildElement( "source-symbol" );
  if ( !sourceSymbolElem.isNull() )
  {
    QgsSymbolV2Map sourceSymbolMap = QgsSymbolLayerV2Utils::loadSymbols( sourceSymbolElem );
    if ( sourceSymbolMap.contains( "0" ) )
    {
      r->setSourceSymbol( sourceSymbolMap.take( "0" ) );
    }
    QgsSymbolLayerV2Utils::clearSymbolMap( sourceSymbolMap );
  }

  // try to load color ramp (optional)
  QDomElement sourceColorRampElem = element.firstChildElement( "colorramp" );
  if ( !sourceColorRampElem.isNull() && sourceColorRampElem.attribute( "name" ) == "[source]" )
  {
    r->setSourceColorRamp( QgsSymbolLayerV2Utils::loadColorRamp( sourceColorRampElem ) );
    QDomElement invertedColorRampElem = element.firstChildElement( "invertedcolorramp" );
    if ( !invertedColorRampElem.isNull() )
      r->setInvertedColorRamp( invertedColorRampElem.attribute( "value" ) == "1" );
  }

  QDomElement rotationElem = element.firstChildElement( "rotation" );
  if ( !rotationElem.isNull() )
    r->setRotationField( rotationElem.attribute( "field" ) );

  QDomElement sizeScaleElem = element.firstChildElement( "sizescale" );
  if ( !sizeScaleElem.isNull() )
  {
    r->setSizeScaleField( sizeScaleElem.attribute( "field" ) );
    r->setScaleMethod( QgsSymbolLayerV2Utils::decodeScaleMethod( sizeScaleElem.attribute( "scalemethod" ) ) );
  }

  // TODO: symbol levels
  return r;
}