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
telwertowski 94a88b1d9a Added QgsFileDropEdit class which extends QLineEdit to accept files dropped on the edit field. 
git-svn-id: http://svn.osgeo.org/qgis/trunk@6513 c8812cc2-4d05-0410-92ff-de0c093fc19c
2007-02-04 09:13:34 +00:00

496 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 "../../src/core/qgscontexthelp.h"
#include "qgsmapserverexport.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 setttings");
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 setttings");
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");
QString prefixPath = QApplication::applicationDirPath();
// Setup up path to the python script directory based on platform
#ifdef Q_WS_MACX
QString dataPath = prefixPath + "/../../../../share/qgis";
#elif WIN32
QString dataPath = prefixPath + "/share/qgis";
#else
QString dataPath ( PKGDATAPATH );
#endif
dataPath = dataPath.trimmed();
QString scriptDir = dataPath + 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");
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);
// Show the return value
PyArg_Parse(pstr, "s", &cstr);
std::cout << cstr << std::endl;
Py_DECREF(pstr);
}
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->getZpos(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->visible())
{
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)->getDataProvider()->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