+ backports cosmetics
+ backports avoid static string as datasource name
This commit is contained in:
Lucio Asnaghi
parent
980098e2d5
commit
b4421bc39c
3 changed files with 201 additions and 200 deletions
|
|
@ -47,22 +47,22 @@ using mapnik::filter_at_point;
|
|||
using mapnik::attribute_descriptor;
|
||||
|
||||
shape_datasource::shape_datasource(const parameters ¶ms, bool bind)
|
||||
: datasource (params),
|
||||
type_(datasource::Vector),
|
||||
file_length_(0),
|
||||
indexed_(false),
|
||||
desc_(*params.get<std::string>("type"), *params.get<std::string>("encoding","utf-8"))
|
||||
: datasource (params),
|
||||
type_(datasource::Vector),
|
||||
file_length_(0),
|
||||
indexed_(false),
|
||||
desc_(*params.get<std::string>("type"), *params.get<std::string>("encoding","utf-8"))
|
||||
{
|
||||
boost::optional<std::string> file = params.get<std::string>("file");
|
||||
if (!file) throw datasource_exception("missing <file> parameter");
|
||||
boost::optional<std::string> file = params.get<std::string>("file");
|
||||
if (!file) throw datasource_exception("missing <file> parameter");
|
||||
|
||||
boost::optional<std::string> base = params.get<std::string>("base");
|
||||
if (base)
|
||||
shape_name_ = *base + "/" + *file;
|
||||
else
|
||||
shape_name_ = *file;
|
||||
boost::optional<std::string> base = params.get<std::string>("base");
|
||||
if (base)
|
||||
shape_name_ = *base + "/" + *file;
|
||||
else
|
||||
shape_name_ = *file;
|
||||
|
||||
boost::algorithm::ireplace_last(shape_name_,".shp","");
|
||||
boost::algorithm::ireplace_last(shape_name_,".shp","");
|
||||
|
||||
if (bind)
|
||||
{
|
||||
|
|
@ -74,203 +74,206 @@ void shape_datasource::bind() const
|
|||
{
|
||||
if (is_bound_) return;
|
||||
|
||||
if (!boost::filesystem::exists(shape_name_ + ".shp"))
|
||||
{
|
||||
throw datasource_exception(shape_name_ + " does not exist");
|
||||
}
|
||||
if (!boost::filesystem::exists(shape_name_ + ".shp"))
|
||||
{
|
||||
throw datasource_exception("shapefile '" + shape_name_ + ".shp' does not exist");
|
||||
}
|
||||
|
||||
try
|
||||
{
|
||||
shape_io shape(shape_name_);
|
||||
// leave ABI alone
|
||||
((shape_datasource*)this)->init(shape);
|
||||
for (int i=0;i<shape.dbf().num_fields();++i)
|
||||
{
|
||||
field_descriptor const& fd=shape.dbf().descriptor(i);
|
||||
std::string fld_name=fd.name_;
|
||||
switch (fd.type_)
|
||||
{
|
||||
if (boost::filesystem::is_directory(shape_name_ + ".shp"))
|
||||
{
|
||||
throw datasource_exception("shapefile '" + shape_name_ + ".shp' appears to be a directory not a file");
|
||||
}
|
||||
|
||||
try
|
||||
{
|
||||
shape_io shape(shape_name_);
|
||||
// leave ABI alone
|
||||
((shape_datasource*)this)->init(shape);
|
||||
for (int i=0;i<shape.dbf().num_fields();++i)
|
||||
{
|
||||
field_descriptor const& fd=shape.dbf().descriptor(i);
|
||||
std::string fld_name=fd.name_;
|
||||
switch (fd.type_)
|
||||
{
|
||||
case 'C':
|
||||
case 'D':
|
||||
case 'M':
|
||||
case 'L':
|
||||
desc_.add_descriptor(attribute_descriptor(fld_name, String));
|
||||
break;
|
||||
desc_.add_descriptor(attribute_descriptor(fld_name, String));
|
||||
break;
|
||||
case 'N':
|
||||
case 'F':
|
||||
{
|
||||
if (fd.dec_>0)
|
||||
{
|
||||
desc_.add_descriptor(attribute_descriptor(fld_name,Double,false,8));
|
||||
}
|
||||
else
|
||||
{
|
||||
desc_.add_descriptor(attribute_descriptor(fld_name,Integer,false,4));
|
||||
}
|
||||
break;
|
||||
if (fd.dec_>0)
|
||||
{
|
||||
desc_.add_descriptor(attribute_descriptor(fld_name,Double,false,8));
|
||||
}
|
||||
else
|
||||
{
|
||||
desc_.add_descriptor(attribute_descriptor(fld_name,Integer,false,4));
|
||||
}
|
||||
break;
|
||||
}
|
||||
default:
|
||||
#ifdef MAPNIK_DEBUG
|
||||
std::clog << "unknown type "<<fd.type_<<"\n";
|
||||
std::clog << "unknown type "<<fd.type_<<"\n";
|
||||
#endif
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
catch (datasource_exception& ex)
|
||||
{
|
||||
std::clog<<ex.what()<<std::endl;
|
||||
throw;
|
||||
}
|
||||
catch (...)
|
||||
{
|
||||
std::clog << " got exception ... \n";
|
||||
throw;
|
||||
}
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
catch (datasource_exception& ex)
|
||||
{
|
||||
std::clog<<ex.what()<<std::endl;
|
||||
throw;
|
||||
}
|
||||
catch (...)
|
||||
{
|
||||
std::clog << " got exception ... \n";
|
||||
throw;
|
||||
}
|
||||
|
||||
is_bound_ = true;
|
||||
}
|
||||
|
||||
shape_datasource::~shape_datasource() {}
|
||||
|
||||
const std::string shape_datasource::name_="shape";
|
||||
|
||||
void shape_datasource::init(shape_io& shape)
|
||||
{
|
||||
//first read header from *.shp
|
||||
int file_code=shape.shp().read_xdr_integer();
|
||||
if (file_code!=9994)
|
||||
{
|
||||
//invalid file code
|
||||
throw datasource_exception((boost::format("wrong file code : %d") % file_code).str());
|
||||
}
|
||||
//first read header from *.shp
|
||||
int file_code=shape.shp().read_xdr_integer();
|
||||
if (file_code!=9994)
|
||||
{
|
||||
//invalid file code
|
||||
throw datasource_exception((boost::format("wrong file code : %d") % file_code).str());
|
||||
}
|
||||
|
||||
shape.shp().skip(5*4);
|
||||
file_length_=shape.shp().read_xdr_integer();
|
||||
int version=shape.shp().read_ndr_integer();
|
||||
shape.shp().skip(5*4);
|
||||
file_length_=shape.shp().read_xdr_integer();
|
||||
int version=shape.shp().read_ndr_integer();
|
||||
|
||||
if (version!=1000)
|
||||
{
|
||||
//invalid version number
|
||||
throw datasource_exception((boost::format("invalid version number: %d") % version).str());
|
||||
}
|
||||
if (version!=1000)
|
||||
{
|
||||
//invalid version number
|
||||
throw datasource_exception((boost::format("invalid version number: %d") % version).str());
|
||||
}
|
||||
|
||||
#ifdef MAPNIK_DEBUG
|
||||
int shape_type = shape.shp().read_ndr_integer();
|
||||
int shape_type = shape.shp().read_ndr_integer();
|
||||
#else
|
||||
shape.shp().skip(4);
|
||||
shape.shp().skip(4);
|
||||
#endif
|
||||
|
||||
shape.shp().read_envelope(extent_);
|
||||
shape.shp().read_envelope(extent_);
|
||||
|
||||
#ifdef MAPNIK_DEBUG
|
||||
double zmin = shape.shp().read_double();
|
||||
double zmax = shape.shp().read_double();
|
||||
double mmin = shape.shp().read_double();
|
||||
double mmax = shape.shp().read_double();
|
||||
double zmin = shape.shp().read_double();
|
||||
double zmax = shape.shp().read_double();
|
||||
double mmin = shape.shp().read_double();
|
||||
double mmax = shape.shp().read_double();
|
||||
|
||||
std::clog << "Z min/max " << zmin << "," << zmax << "\n";
|
||||
std::clog << "M min/max " << mmin << "," << mmax << "\n";
|
||||
std::clog << "Z min/max " << zmin << "," << zmax << "\n";
|
||||
std::clog << "M min/max " << mmin << "," << mmax << "\n";
|
||||
#else
|
||||
shape.shp().skip(4*8);
|
||||
shape.shp().skip(4*8);
|
||||
#endif
|
||||
|
||||
// check if we have an index file around
|
||||
std::string index_name(shape_name_+".index");
|
||||
std::ifstream file(index_name.c_str(),std::ios::in | std::ios::binary);
|
||||
if (file)
|
||||
{
|
||||
indexed_=true;
|
||||
file.close();
|
||||
}
|
||||
// check if we have an index file around
|
||||
std::string index_name(shape_name_+".index");
|
||||
std::ifstream file(index_name.c_str(),std::ios::in | std::ios::binary);
|
||||
if (file)
|
||||
{
|
||||
indexed_=true;
|
||||
file.close();
|
||||
}
|
||||
|
||||
#ifdef MAPNIK_DEBUG
|
||||
std::clog << extent_ << std::endl;
|
||||
std::clog << "file_length=" << file_length_ << std::endl;
|
||||
std::clog << "shape_type=" << shape_type << std::endl;
|
||||
std::clog << extent_ << std::endl;
|
||||
std::clog << "file_length=" << file_length_ << std::endl;
|
||||
std::clog << "shape_type=" << shape_type << std::endl;
|
||||
#endif
|
||||
}
|
||||
|
||||
std::string shape_datasource::name()
|
||||
{
|
||||
return "shape";
|
||||
}
|
||||
|
||||
int shape_datasource::type() const
|
||||
{
|
||||
return type_;
|
||||
return type_;
|
||||
}
|
||||
|
||||
layer_descriptor shape_datasource::get_descriptor() const
|
||||
{
|
||||
if (!is_bound_) bind();
|
||||
return desc_;
|
||||
}
|
||||
|
||||
std::string shape_datasource::name()
|
||||
{
|
||||
return name_;
|
||||
return desc_;
|
||||
}
|
||||
|
||||
featureset_ptr shape_datasource::features(const query& q) const
|
||||
{
|
||||
if (!is_bound_) bind();
|
||||
|
||||
filter_in_box filter(q.get_bbox());
|
||||
if (indexed_)
|
||||
{
|
||||
return featureset_ptr
|
||||
(new shape_index_featureset<filter_in_box>(filter,
|
||||
shape_name_,
|
||||
q.property_names(),
|
||||
desc_.get_encoding()));
|
||||
}
|
||||
else
|
||||
{
|
||||
return featureset_ptr
|
||||
(new shape_featureset<filter_in_box>(filter,
|
||||
shape_name_,
|
||||
q.property_names(),
|
||||
desc_.get_encoding(),
|
||||
file_length_));
|
||||
}
|
||||
filter_in_box filter(q.get_bbox());
|
||||
|
||||
if (indexed_)
|
||||
{
|
||||
return featureset_ptr
|
||||
(new shape_index_featureset<filter_in_box>(filter,
|
||||
shape_name_,
|
||||
q.property_names(),
|
||||
desc_.get_encoding()));
|
||||
}
|
||||
else
|
||||
{
|
||||
return featureset_ptr
|
||||
(new shape_featureset<filter_in_box>(filter,
|
||||
shape_name_,
|
||||
q.property_names(),
|
||||
desc_.get_encoding(),
|
||||
file_length_));
|
||||
}
|
||||
}
|
||||
|
||||
featureset_ptr shape_datasource::features_at_point(coord2d const& pt) const
|
||||
{
|
||||
if (!is_bound_) bind();
|
||||
|
||||
filter_at_point filter(pt);
|
||||
// collect all attribute names
|
||||
std::vector<attribute_descriptor> const& desc_vector = desc_.get_descriptors();
|
||||
std::vector<attribute_descriptor>::const_iterator itr = desc_vector.begin();
|
||||
std::vector<attribute_descriptor>::const_iterator end = desc_vector.end();
|
||||
std::set<std::string> names;
|
||||
filter_at_point filter(pt);
|
||||
|
||||
while (itr != end)
|
||||
{
|
||||
names.insert(itr->get_name());
|
||||
++itr;
|
||||
}
|
||||
// collect all attribute names
|
||||
std::vector<attribute_descriptor> const& desc_vector = desc_.get_descriptors();
|
||||
std::vector<attribute_descriptor>::const_iterator itr = desc_vector.begin();
|
||||
std::vector<attribute_descriptor>::const_iterator end = desc_vector.end();
|
||||
std::set<std::string> names;
|
||||
|
||||
if (indexed_)
|
||||
{
|
||||
return featureset_ptr
|
||||
(new shape_index_featureset<filter_at_point>(filter,
|
||||
shape_name_,
|
||||
names,
|
||||
desc_.get_encoding()));
|
||||
}
|
||||
else
|
||||
{
|
||||
return featureset_ptr
|
||||
(new shape_featureset<filter_at_point>(filter,
|
||||
shape_name_,
|
||||
names,
|
||||
desc_.get_encoding(),
|
||||
file_length_));
|
||||
}
|
||||
while (itr != end)
|
||||
{
|
||||
names.insert(itr->get_name());
|
||||
++itr;
|
||||
}
|
||||
|
||||
if (indexed_)
|
||||
{
|
||||
return featureset_ptr
|
||||
(new shape_index_featureset<filter_at_point>(filter,
|
||||
shape_name_,
|
||||
names,
|
||||
desc_.get_encoding()));
|
||||
}
|
||||
else
|
||||
{
|
||||
return featureset_ptr
|
||||
(new shape_featureset<filter_at_point>(filter,
|
||||
shape_name_,
|
||||
names,
|
||||
desc_.get_encoding(),
|
||||
file_length_));
|
||||
}
|
||||
}
|
||||
|
||||
Envelope<double> shape_datasource::envelope() const
|
||||
{
|
||||
if (!is_bound_) bind();
|
||||
|
||||
return extent_;
|
||||
return extent_;
|
||||
}
|
||||
|
|
|
|||
|
|
@ -61,7 +61,6 @@ class shape_datasource : public datasource
|
|||
mutable Envelope<double> extent_;
|
||||
mutable bool indexed_;
|
||||
mutable layer_descriptor desc_;
|
||||
static const std::string name_;
|
||||
};
|
||||
|
||||
#endif //SHAPE_HPP
|
||||
|
|
|
|||
|
|
@ -94,7 +94,7 @@ struct shape_record
|
|||
|
||||
int read_ndr_integer()
|
||||
{
|
||||
boost::int32_t val;
|
||||
boost::int32_t val;
|
||||
read_int32_ndr(&data[pos],val);
|
||||
pos+=4;
|
||||
return val;
|
||||
|
|
@ -102,7 +102,7 @@ struct shape_record
|
|||
|
||||
int read_xdr_integer()
|
||||
{
|
||||
boost::int32_t val;
|
||||
boost::int32_t val;
|
||||
read_int32_xdr(&data[pos],val);
|
||||
pos+=4;
|
||||
return val;
|
||||
|
|
@ -115,6 +115,7 @@ struct shape_record
|
|||
pos+=8;
|
||||
return val;
|
||||
}
|
||||
|
||||
long remains()
|
||||
{
|
||||
return (size-pos);
|
||||
|
|
@ -124,7 +125,6 @@ struct shape_record
|
|||
{
|
||||
Tag::dealloc(data);
|
||||
}
|
||||
|
||||
};
|
||||
|
||||
using namespace boost::iostreams;
|
||||
|
|
@ -162,7 +162,6 @@ public:
|
|||
return file_.is_open();
|
||||
}
|
||||
|
||||
|
||||
inline void close()
|
||||
{
|
||||
if (file_ && file_.is_open())
|
||||
|
|
|
|||
Loading…
Reference in a new issue