sharpetronics/QGIS
1
0
Fork 0
mirror of https://github.com/qgis/QGIS.git synced 2025-02-24 00:47:57 -05:00
Code Issues Packages Projects Releases Wiki Activity
QGIS/tools/mapserver_export/qgsmapserverexport.cpp
timlinux 8e087f5b23 API Cleanups 
git-svn-id: http://svn.osgeo.org/qgis/trunk/qgis@9446 c8812cc2-4d05-0410-92ff-de0c093fc19c
2008-10-06 18:50:42 +00:00

505 lines
18 KiB
C++
Raw Blame History

/***************************************************************************
qgsmapserverexport.cpp - Export QGIS MapCanvas to MapServer
--------------------------------------
Date : 8-Nov-2003
Copyright : (C) 2003 by Gary E.Sherman
email : sherman at mrcc.com
***************************************************************************
* *
* This program is free software; you can redistribute it and/or modify *
* it under the terms of the GNU General Public License as published by *
* the Free Software Foundation; either version 2 of the License, or *
* (at your option) any later version. *
* *
***************************************************************************/
/* $Id: qgsmapserverexport.cpp 4170 2005-11-09 01:49:00Z timlinux $ */
#include <Python.h>
#include <iostream>
#include <QFileDialog>
#include <qfileinfo.h>
#include <qlineedit.h>
#include <qcheckbox.h>
#include <QComboBox>
#include <QSettings>
#include <qmessagebox.h>
#include <qcolor.h>
#include <qregexp.h>
#include <qstring.h>
#include <QWidget>
#include <QApplication>
#include "qgsmapserverexport.h"
// from CORE library
#include "qgsapplication.h"
#include "qgscontexthelp.h"
// constructor
QgsMapserverExport::QgsMapserverExport( QWidget * parent, Qt::WFlags fl )
: QDialog( parent, fl )
{
setupUi( this );
connect( this, SIGNAL( accepted() ), this, SLOT( apply() ) );
// drag and drop filter
txtQgisFilePath->setSuffixFilter( "qgs" );
// initialize python
initPy();
qDebug( "Reading settings" );
QSettings mySettings;
txtMapFilePath->setText( mySettings.value( "mapserverExport/lastMapFile", "" ).toString() );
txtQgisFilePath->setText( mySettings.value( "mapserverExport/lastQgsFile", "" ).toString() );
}
// Default destructor
QgsMapserverExport::~QgsMapserverExport()
{
// cleanup python
Py_Finalize();
}
// Get the base name for the map file
QString QgsMapserverExport::baseName()
{
QFileInfo fi( txtMapFilePath->text() );
return fi.baseName( true );
}
/** Auto-connected slots are here **/
// Choose the map file to create
void QgsMapserverExport::on_btnChooseFile_clicked()
{
mapFile = QFileDialog::getSaveFileName( this, tr( "Name for the map file" ),
".", tr( "MapServer map files (*.map);;All files (*.*)", "Filter list for selecting files from a dialog box" ) );
txtMapFilePath->setText( mapFile );
}
// Chooose the project file to process
void QgsMapserverExport::on_btnChooseProjectFile_clicked()
{
qgisProjectFile = QFileDialog::getOpenFileName( this, tr( "Choose the QGIS project file" ),
".", tr( "QGIS Project Files (*.qgs);;All files (*.*)", "Filter list for selecting files from a dialog box" ) );
txtQgisFilePath->setText( qgisProjectFile );
}
// Toggle controls based on the "layer only" checkbox
void QgsMapserverExport::on_chkExpLayersOnly_clicked( bool isChecked )
{
// disable other sections if only layer export is desired
txtMapName->setEnabled( !isChecked );
txtMapWidth->setEnabled( !isChecked );
txtMapHeight->setEnabled( !isChecked );
cmbMapUnits->setEnabled( !isChecked );
cmbMapImageType->setEnabled( !isChecked );
//txtMinScale->setEnabled(!isChecked);
//txtMaxScale->setEnabled(!isChecked);
txtWebTemplate->setEnabled( !isChecked );
txtWebHeader->setEnabled( !isChecked );
txtWebFooter->setEnabled( !isChecked );
btnChooseFooterFile->setEnabled( !isChecked );
btnChooseHeaderFile->setEnabled( !isChecked );
btnChooseTemplateFile->setEnabled( !isChecked );
}
void QgsMapserverExport::apply()
{
qDebug( "Writing settings" );
QSettings mySettings;
mySettings.setValue( "mapserverExport/lastMapFile", txtMapFilePath->text() );
mySettings.setValue( "mapserverExport/lastQgsFile", txtQgisFilePath->text() );
char *cstr;
PyObject *pstr, *pmod, *pclass, *pinst, *pmeth, *pargs;
//TODO Need to append the path to the qgis python files using the path to the
// Python files in the QGIS install directory
PyRun_SimpleString( "import sys" );
// Setup up path to the python script directory
QString scriptDir = QgsApplication::pkgDataPath() + QDir::separator() + "python";
qDebug( "Python scripts directory: " + scriptDir.toLocal8Bit() );
//QString curdir = "/home/gsherman/development/qgis_qt_port/tools/mapserver_export";
QString sysCmd = QString( "sys.path.append('%1')" ).arg( scriptDir );
PyRun_SimpleString( sysCmd.ascii() );
// Import the module
std::cout << "Importing module" << std::endl;
pmod = PyImport_ImportModule( "ms_export" );
if ( !pmod )
{
QMessageBox::warning( this, "Map Export Error", "ms_export python module not found" );
return;
}
std::cout << "Getting Qgis2Map constructor as python obj" << std::endl;
pclass = PyObject_GetAttrString( pmod, "Qgis2Map" );
Py_DECREF( pmod );
std::cout << "Creating args to pass to the constructor" << std::endl;
pargs = Py_BuildValue( "(ss)", txtQgisFilePath->text().ascii(), txtMapFilePath->text().ascii() );
//XXX for testing: pargs = Py_BuildValue("(ss)", "foo", "bar");
pinst = PyEval_CallObject( pclass, pargs );
Py_DECREF( pclass );
Py_DECREF( pargs );
// Set the various options for the conversion only if we are doing a full
// export (ie more than just layer info)
if ( !chkExpLayersOnly->isChecked() )
{
std::cout << "Initializing all options" << std::endl;
pmeth = PyObject_GetAttrString( pinst, "setOptions" );
pargs = Py_BuildValue( "(ssssssss)",
cmbMapUnits->currentText().ascii(), cmbMapImageType->currentText().ascii(),
txtMapName->text().ascii(), txtMapWidth->text().ascii(), txtMapHeight->text().ascii(),
txtWebTemplate->text().ascii(), txtWebFooter->text().ascii(), txtWebHeader->text().ascii() );
pstr = PyEval_CallObject( pmeth, pargs );
Py_DECREF( pargs );
Py_DECREF( pmeth );
}
// Get the writeMapFile method from the Qgis2Map class
pmeth = PyObject_GetAttrString( pinst, "writeMapFile" );
pargs = Py_BuildValue( "()" );
// Execute the writeMapFile method to parse the QGIS project file and create the .map file
pstr = PyEval_CallObject( pmeth, pargs );
if ( pstr )
{
// Show the return value
PyArg_Parse( pstr, "s", &cstr );
std::cout << "Result: " << std::endl << cstr << std::endl;
// Show the results to the user
QMessageBox::information( this, "Results of Export", QString( cstr ) );
Py_DECREF( pstr );
}
else
{
QMessageBox::warning( this, "Mapfile Export Error", "method call failed" );
}
}
void QgsMapserverExport::on_buttonBox_helpRequested()
{
QgsContextHelp::run( context_id );
}
/** End of Auto-connected Slots **/
// Write the map file
bool QgsMapserverExport::write()
{
//QMessageBox::information(0,"Full Path",fullPath);
QMessageBox::StandardButton okToSave = QMessageBox::Ok;
// Check for file and prompt for overwrite if it exists
if ( QFile::exists( txtMapFilePath->text() ) )
{
okToSave = QMessageBox::warning( 0, tr( "Overwrite File?" ), txtMapFilePath->text() +
tr( " exists. \nDo you want to overwrite it?",
"a fileName is prepended to this text, and appears in a dialog box" ),
QMessageBox::Ok | QMessageBox::Cancel );
}
if ( okToSave == QMessageBox::Ok )
{
// write the project information to the selected file
writeMapFile();
return true;
}
else
{
return false;
}
}
void QgsMapserverExport::setFileName( QString fn )
{
//fullPath = fn;
}
QString QgsMapserverExport::fullPathName()
{
return QString(); // return something to allow compilation
//return fullPath;
}
void QgsMapserverExport::writeMapFile()
{
/*
// write the map file, making massive assumptions about default values
#ifdef QGISDEBUG
std::cout << "Opening map file " << txtMapFilePath->text().toLocal8Bit().data() << std::endl;
#endif
std::ofstream mapFile(txtMapFilePath->text().toLocal8Bit().data());
if (!mapFile.fail())
{
// XXX So, what encoding should we use here???
mapFile << "# Map file generated by QGIS version " << QGis::qgisVersion << std::endl;
mapFile << "# Edit this file to customize for your interface" << std::endl;
mapFile << "# Not all sections are complete. See comments for details." << std::endl;
if (!chkExpLayersOnly->isChecked())
{
// header
mapFile << "NAME " << txtMapName->text().toLocal8Bit().data() << std::endl;
mapFile << "STATUS ON" << std::endl;
mapFile << "\n# Map image size. Change size as desired" << std::endl;
mapFile << "SIZE " << txtMapWidth->text().toLocal8Bit().data() << " " << txtMapHeight->text().toLocal8Bit().data() << std::endl;
// web interface definition - this is minimal!
mapFile << "#" << std::endl;
mapFile << "# Start of web interface definition. Only the TEMPLATE parameter" << std::endl;
mapFile << "# must be specified to display a map. See Mapserver documentation" << std::endl;
mapFile << "#" << std::endl;
mapFile << "WEB" << std::endl;
// if no header is supplied, write the header line but comment it out
if (txtWebHeader->text().isEmpty())
{
mapFile << " # HEADER" << std::endl;
} else
{
// header provided - write it
mapFile << " HEADER " << txtWebHeader->text().toLocal8Bit().data() << std::endl;
}
// if no template provided, write the template line but comment it out
if (txtWebTemplate->text().isEmpty())
{
mapFile << " # TEMPLATE" << std::endl;
} else
{
// template provided - write it
mapFile << " TEMPLATE " << txtWebTemplate->text().toLocal8Bit().data() << std::endl;
}
// if no footer provided, write the footer line but comment it out
if (txtWebFooter->text().isEmpty())
{
mapFile << " # FOOTER" << std::endl;
} else
{
mapFile << " FOOTER " << txtWebFooter->text().toLocal8Bit().data() << std::endl;
}
QString minScale = txtMinScale->text().isEmpty()?"#MINSCALE":"MINSCALE";
QString maxScale = txtMinScale->text().isEmpty()?" #MAXSCALE ":" MAXSCALE ";
// write min and maxscale
mapFile << minScale.toLocal8Bit().data() << txtMinScale->text().toLocal8Bit().data() << std::endl;
mapFile << maxScale.toLocal8Bit().data() << txtMaxScale->text().toLocal8Bit().data() << std::endl;
// write comments about the imagepath and image url
mapFile << "# Set IMAGEPATH to the path where mapserver should\n" <<
"# write its output\n" <<
" IMAGEPATH '/tmp/'" << std::endl;
mapFile << "# Set IMAGEURL to the url that points to IMAGEPATH" << std::endl;
mapFile << " #IMAGEURL '/map_output/'" << std::endl;
// end of web section
mapFile << "END" << std::endl;
// extent
mapFile << "\n# Extent based on full extent of QGIS view" << std::endl;
mapFile << "EXTENT ";
QgsRect extent = map->extent();
mapFile << extent.xMin() << " " << extent.yMin() << " ";
mapFile << extent.xMax() << " " << extent.yMax() << std::endl;
// units
mapFile << "UNITS " << cmbMapUnits->currentText().toLocal8Bit().data() << std::endl;
// image info
mapFile << "IMAGECOLOR 255 255 255" << std::endl;
mapFile << "IMAGETYPE " << cmbMapImageType->currentText().toLocal8Bit().data() << std::endl;
// projection information TODO: support projections :)
mapFile << "# Projection definition" << std::endl;
mapFile << "# Projections are not currenlty supported. If desired, add your own" << std::endl;
mapFile << "# projection information based on Mapserver documentation." << std::endl;
mapFile << "#" << std::endl;
} else
{
mapFile << " # This file contains layer definitions only and is not a complete" << std::endl;
mapFile << " # Mapserver map file." << std::endl;
}
// write layer definitions
for (int i = 0; i < map->layerCount(); i++)
{
bool isPolygon = false;
bool isLine = false;
QgsMapLayer *lyr = map->layer(i);
#ifdef QGISDEBUG
std::cout << "Mapsrver Export Processing Layer" << std::endl;
#endif
mapFile << "LAYER" << std::endl;
QString name = lyr->name().lower();
// MapServer NAME must be < 20 char and unique
name.replace(QRegExp(" "), "_");
name.replace(QRegExp("\\."), "_");
name.replace(QRegExp("\\("), "_");
name.replace(QRegExp("\\)"), "_");
mapFile << " NAME " << name.toLocal8Bit().data() << std::endl;
// feature type
#ifdef QGISDEBUG
std::cout << "\tMapsrver Export checking feature type" << std::endl;
#endif
mapFile << " TYPE ";
switch (lyr->featureType())
{
case QGis::WKBPoint:
case QGis::WKBMultiPoint:
mapFile << "POINT";
break;
case QGis::WKBLineString:
case QGis::WKBMultiLineString:
mapFile << "LINE";
isLine = true;
break;
case QGis::WKBPolygon:
case QGis::WKBMultiPolygon:
mapFile << "POLYGON";
isPolygon = true;
break;
}
if(lyr->type() == QgsMapLayer::RASTER)
{
mapFile << "RASTER";
}
mapFile << std::endl;
#ifdef QGISDEBUG
std::cout << "\tMapsrver Export checking visibility" << std::endl;
#endif
// set visibility (STATUS)
mapFile << " STATUS ";
if (lyr->isVisible())
{
mapFile << "ON";
} else
{
mapFile << "OFF";
}
mapFile << std::endl;
// data source (DATA)
// Data source spec depends on layer type
#ifdef QGISDEBUG
std::cout << "\tMapsrver Export checking layer type" << std::endl;
#endif
switch (lyr->type())
{
case QgsMapLayer::VECTOR:
// get the provider type
{
QString providerType =
dynamic_cast<QgsVectorLayer*>(lyr)->providerType();
if(providerType == "postgres")
{
QgsDataSourceURI *dUri =
dynamic_cast<QgsVectorLayer *>(lyr)->dataProvider()->getURI();
mapFile << "CONNECTION \"user=" << dUri->username.toLocal8Bit().data();
if(dUri->password.length() > 0)
{
mapFile << " password="<< dUri->password.toLocal8Bit().data();
}
mapFile << " dbname=" << dUri->database.toLocal8Bit().data()
<< " host=" << dUri->host.toLocal8Bit().data()
<< " port=" << dUri->port.toLocal8Bit().data()
<< "\"" << std::endl;
mapFile << "CONNECTIONTYPE postgis" << std::endl;
mapFile << "DATA \"" << dUri->geometryColumn.toLocal8Bit().data() << " from "
<< dUri->table.toLocal8Bit().data() << "\"" << std::endl;
if(dUri->sql.length() > 0)
{
mapFile << "FILTER \"" << dUri->sql.toLocal8Bit().data() << "\"" << std::endl;
}
}
else
{
// must be an ogr
mapFile << " DATA " << lyr->source().toLocal8Bit().data() << std::endl;
}
}
break;
case QgsMapLayer::RASTER:
mapFile << " DATA " << lyr->source().toLocal8Bit().data() << std::endl;
break;
}
#ifdef QGISDEBUG
std::cout << "\tMapsrver Export creating symbol entries" << std::endl;
#endif
// create a simple class entry based on red fill color and black outline color
//TODO: adapt the following section to the new symbology
mapFile << " CLASS" << std::endl;
//QListViewItem *li = map->getLegend()->currentItem();
// return li->text(0);
#ifdef QGISDEBUG
std::cout << "\tMapsrver Export symbol name" << std::endl;
#endif
mapFile << " NAME \"" << lyr->name().toLocal8Bit().data() << "\"" << std::endl;
mapFile << " # TEMPLATE" << std::endl;
if (isPolygon)
{
#ifdef QGISDEBUG
std::cout << "\tMapsrver Export symbol fill color" << std::endl;
#endif
// use random fill colors
// TODO Get fill color from the renderer
// TODO Figure out what to do for layers that are
// rendered with other than the simple symbol
// renderer.
int red = 1 + (int) (255.0 * rand() / (RAND_MAX + 1.0));
int green = 1 + (int) (255.0 * rand() / (RAND_MAX + 1.0));
int blue = 1 + (int) (255.0 * rand() / (RAND_MAX + 1.0));
mapFile << " COLOR " << red << " " << green << " " << blue << std::endl;
}
#ifdef QGISDEBUG
std::cout << "\tMapsrver Export checking for line symbol " << std::endl;
#endif
if (isPolygon || isLine)
{
// use line color
// TODO Get line color from the renderer
int red = 1 + (int) (255.0 * rand() / (RAND_MAX + 1.0));
int green = 1 + (int) (255.0 * rand() / (RAND_MAX + 1.0));
int blue = 1 + (int) (255.0 * rand() / (RAND_MAX + 1.0));
mapFile << " OUTLINECOLOR " << red << " "
<< green << " " << blue << std::endl;
}
mapFile << " END" << std::endl;
mapFile << "END" << std::endl;
#ifdef QGISDEBUG
std::cout << "\tMapsrver Export layer definition done..." << std::endl;
#endif
}
if (!chkExpLayersOnly->isChecked())
{
mapFile << "END # Map File";
}
mapFile.flush();
mapFile.close();
} else
{
}
*/
}
void QgsMapserverExport::showHelp()
{
/*
//QMessageBox::information(this, "Help","Help");
QgsHelpViewer *hv = new QgsHelpViewer(this);
// causes problems in qt3.1.x: hv->setModal(false);
hv->setCaption("QGIS Help - Mapserver Export");
hv->show();
*/
}
void QgsMapserverExport::initPy()
{
// init the python interpreter
Py_Initialize();
// spit something to stdout
PyRun_SimpleString( "print '>>>>>>>>>>>>>>>>>>> Python initialized <<<<<<<<<<<<<<<<<<<'" );
}
Reference in New Issue View Git Blame Copy Permalink