+ sqlite.input: whitespace fix

+ occi.input: added table_from_sql to allow specify SELECT statements in table parameter
+ occi.input: added use_spatial_index parameter (on by default)
+ occi.input: improved determination of query columns type

plugins/input/occi/occi_datasource.cpp

@@ -28,6 +28,7 @@
 #include <mapnik/ptree_helpers.hpp>
 
 // boost
+#include <boost/algorithm/string.hpp>
 #include <boost/lexical_cast.hpp>
 #include <boost/tokenizer.hpp>
 
@@ -64,9 +65,32 @@ using oracle::occi::SQLException;
 using oracle::occi::Type;
 using oracle::occi::StatelessConnectionPool;
 
-
 DATASOURCE_PLUGIN(occi_datasource)
 
+
+std::string table_from_sql(std::string const& sql)
+{
+   std::string table_name = boost::algorithm::to_lower_copy(sql);
+   boost::algorithm::replace_all(table_name,"\n"," ");
+   
+   std::string::size_type idx = table_name.rfind("from");
+   if (idx!=std::string::npos)
+   {
+      
+      idx=table_name.find_first_not_of(" ",idx+4);
+      if (idx != std::string::npos)
+      {
+         table_name=table_name.substr(idx);
+      }
+      idx=table_name.find_first_of(" ),");
+      if (idx != std::string::npos)
+      {
+         table_name = table_name.substr(0,idx);
+      }
+   }
+   return table_name;
+}
+
 occi_datasource::occi_datasource(parameters const& params)
    : datasource (params),
      table_(*params.get<std::string>("table","")),
@@ -80,6 +104,7 @@ occi_datasource::occi_datasource(parameters const& params)
    boost::optional<int> max_size = params_.get<int>("max_size",10);
 
    multiple_geometries_ = *params_.get<mapnik::boolean>("multiple_geometries",false);
+   use_spatial_index_ = *params_.get<mapnik::boolean>("use_spatial_index",true);
 
    boost::optional<std::string> ext = params_.get<std::string>("extent");
    if (ext)
@@ -134,13 +159,15 @@ occi_datasource::occi_datasource(parameters const& params)
        throw datasource_exception(ex.getMessage());
    }
 
+   std::string table_name = table_from_sql(table_);
+
    // get SRID from geometry metadata
    {
        occi_connection_ptr conn (pool_);
 
        std::ostringstream s;
        s << "select srid from " << SDO_GEOMETRY_METADATA_TABLE << " where";
-       s << " lower(table_name) = lower('" << table_ << "') and";
+       s << " lower(table_name) = lower('" << table_name << "') and";
        s << " lower(column_name) = lower('" << geometry_field_ << "')";
 
        try
@@ -157,10 +184,20 @@ occi_datasource::occi_datasource(parameters const& params)
        }
    }
 
-   // get table metadata
+   // get columns description
+   {
+       std::string::size_type idx = table_.find(table_name);
+       std::ostringstream s;
+       s << "select * from (" << table_.substr(0,idx + table_name.length()) << ") where rownum < 1";
+
        occi_connection_ptr conn (pool_);
-   MetaData metadata = (*conn)->getMetaData(table_.c_str(), MetaData::PTYPE_TABLE);
+
-   vector<MetaData> listOfColumns = metadata.getVector(MetaData::ATTR_LIST_COLUMNS);
+       try
+       {
+           ResultSet* rs = conn.execute_query (s.str());
+           if (rs)
+           {
+               std::vector<MetaData> listOfColumns = rs->getColumnListMetaData();
 
                for (unsigned int i=0;i<listOfColumns.size();++i)
                {
@@ -225,6 +262,13 @@ occi_datasource::occi_datasource(parameters const& params)
                    }	  
                }
            }
+       }
+       catch (SQLException &ex)
+       {
+           throw datasource_exception(ex.getMessage());
+       }
+   }        
+}
 
 occi_datasource::~occi_datasource()
 {
@@ -285,14 +329,15 @@ Envelope<double> occi_datasource::envelope() const
             throw datasource_exception(ex.getMessage());
         }
     }
-    else 
+    else if (use_spatial_index_)
     {
+        std::string table_name = table_from_sql(table_);
 
         {
             std::ostringstream s;
             s << "select dim.sdo_lb as lx, dim.sdo_ub as ux from ";
             s << SDO_GEOMETRY_METADATA_TABLE << " m, table(m.diminfo) dim ";
-            s << " where lower(m.table_name) = '" << table_ << "' and dim.sdo_dimname = 'X'";
+            s << " where lower(m.table_name) = '" << table_name << "' and dim.sdo_dimname = 'X'";
         
             try
             {
@@ -320,7 +365,7 @@ Envelope<double> occi_datasource::envelope() const
             std::ostringstream s;
             s << "select dim.sdo_lb as ly, dim.sdo_ub as uy from ";
             s << SDO_GEOMETRY_METADATA_TABLE << " m, table(m.diminfo) dim ";
-            s << " where lower(m.table_name) = '" << table_ << "' and dim.sdo_dimname = 'Y'";
+            s << " where lower(m.table_name) = '" << table_name << "' and dim.sdo_dimname = 'Y'";
         
             try
             {
@@ -372,13 +417,34 @@ featureset_ptr occi_datasource::features(query const& q) const
            s <<",\""<<*pos<<"\"";
            ++pos;
         }	 
-        s << " from " << table_ << " where sdo_filter(" << geometry_field_ << ",";
+
-        s << " mdsys.sdo_geometry(" << SDO_GTYPE_2DPOLYGON << "," << srid_ << ",NULL,";
+        s << " from ";
-        s << " mdsys.sdo_elem_info_array(1," << SDO_ETYPE_POLYGON << "," << SDO_INTERPRETATION_RECTANGLE << "),";
+
-        s << " mdsys.sdo_ordinate_array(";
+        std::string query (table_); 
-        s << std::setprecision(16);
+        
-        s << box.minx() << "," << box.miny() << ", ";
+        if (use_spatial_index_)
-        s << box.maxx() << "," << box.maxy() << ")), 'querytype=WINDOW') = 'TRUE'";      
+        {
+           std::string table_name = table_from_sql(query);
+           std::ostringstream spatial_sql;
+           spatial_sql << std::setprecision(16);
+           spatial_sql << " where sdo_filter(" << geometry_field_ << ",";
+           spatial_sql << " mdsys.sdo_geometry(" << SDO_GTYPE_2DPOLYGON << "," << srid_ << ",NULL,";
+           spatial_sql << " mdsys.sdo_elem_info_array(1," << SDO_ETYPE_POLYGON << "," << SDO_INTERPRETATION_RECTANGLE << "),";
+           spatial_sql << " mdsys.sdo_ordinate_array(";
+           spatial_sql << box.minx() << "," << box.miny() << ", ";
+           spatial_sql << box.maxx() << "," << box.maxy() << ")), 'querytype=WINDOW') = 'TRUE'";      
+
+           if (boost::algorithm::ifind_first(query,"where"))
+           {
+              boost::algorithm::ireplace_first(query, "where", spatial_sql.str() + " and");
+           }
+           else if (boost::algorithm::find_first(query,table_name))  
+           {
+              boost::algorithm::ireplace_first(query, table_name , table_name + " " + spatial_sql.str());
+           }
+        }
+        
+        s << query;
 
 #ifdef MAPNIK_DEBUG
         clog << s.str() << endl;
@@ -396,7 +462,6 @@ featureset_ptr occi_datasource::features_at_point(coord2d const& pt) const
     {
         std::ostringstream s;
         s << "select " << geometry_field_ << " as geom";
-
         std::vector<attribute_descriptor>::const_iterator itr = desc_.get_descriptors().begin();
         std::vector<attribute_descriptor>::const_iterator end = desc_.get_descriptors().end();
         unsigned size=0;
@@ -406,18 +471,30 @@ featureset_ptr occi_datasource::features_at_point(coord2d const& pt) const
            ++itr;
            ++size;
         }
-        s << " from " << table_<<" where "<<geometryColumn_<<" && setSRID('BOX3D(";
-        s << std::setprecision(16);
-        s << pt.x << " " << pt.y << ",";
-        s << pt.x << " " << pt.y << ")'::box3d,"<<srid_<<")";
 
+        s << " from ";
 
-        s << " from " << table_ << " where sdo_filter(" << geometry_field_ << ",";
+        std::string query (table_); 
-        s << " mdsys.sdo_geometry(" << SDO_GTYPE_2DPOINT << "," << srid_ << ",NULL,";
+        std::string table_name = table_from_sql(query);
-        s << " mdsys.sdo_elem_info_array(1," << SDO_ETYPE_POINT << "," << SDO_INTERPRETATION_POINT << "),";
+
-        s << " mdsys.sdo_ordinate_array(";
+        std::ostringstream spatial_sql;
-        s << std::setprecision(16);
+        spatial_sql << std::setprecision(16);
-        s << pt.x << "," << pt.y << ")), 'querytype=WINDOW') = 'TRUE'";      
+        spatial_sql << " where sdo_filter(" << geometry_field_ << ",";
+        spatial_sql << " mdsys.sdo_geometry(" << SDO_GTYPE_2DPOINT << "," << srid_ << ",NULL,";
+        spatial_sql << " mdsys.sdo_elem_info_array(1," << SDO_ETYPE_POINT << "," << SDO_INTERPRETATION_POINT << "),";
+        spatial_sql << " mdsys.sdo_ordinate_array(";
+        spatial_sql << pt.x << "," << pt.y << ")), 'querytype=WINDOW') = 'TRUE'";      
+
+        if (boost::algorithm::ifind_first(query,"where"))
+        {
+           boost::algorithm::ireplace_first(query, "where", spatial_sql.str() + " and");
+        }
+        else if (boost::algorithm::find_first(query,table_name))  
+        {
+           boost::algorithm::ireplace_first(query, table_name , table_name + " " + spatial_sql.str());
+        }
+        
+        s << query;
 
 #ifdef MAPNIK_DEBUG
         clog << s.str() << endl;

plugins/input/occi/occi_datasource.hpp

@@ -61,6 +61,7 @@ class occi_datasource : public mapnik::datasource
       mapnik::layer_descriptor desc_;
       oracle::occi::StatelessConnectionPool* pool_;
       bool multiple_geometries_;
+      bool use_spatial_index_;
       static const std::string name_;
 };
 

plugins/input/sqlite/sqlite_datasource.cpp

@@ -292,7 +292,6 @@ featureset_ptr sqlite_datasource::features(query const& q) const
         
         s << query ;
         
-        
         if (row_limit_ > 0) {
             s << " limit " << row_limit_;
         }