mapnik/plugins/input/shape/shape_io.cpp

/* This file is part of Mapnik (c++ mapping toolkit)
 * Copyright (C) 2005 Artem Pavlenko
 *
 * Mapnik 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 any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 */

//$Id: shape_io.cc 26 2005-03-29 19:18:59Z pavlenko $

#include "shape_io.hpp"
#include "shape.hpp"


const std::string shape_io::SHP = ".shp";
const std::string shape_io::DBF = ".dbf";

shape_io::shape_io(const std::string& shape_name)
    : type_(shape_null),
      reclength_(0),
      id_(0)
{
    bool ok = (shp_.open(shape_name + SHP) &&
	       dbf_.open(shape_name + DBF));
    if (!ok)
    { 
	throw datasource_exception("cannot read shape file");
    }
}

shape_io::~shape_io()
{
    shp_.close();
    dbf_.close();
}


void shape_io::move_to (int pos)
{
    shp_.seek(pos);
    id_ = shp_.read_xdr_integer();
    reclength_ = shp_.read_xdr_integer();
    type_ = shp_.read_ndr_integer();

    if (type_ != shape_point && type_ != shape_pointm && type_ != shape_pointz)
    {
        shp_.read_envelope(cur_extent_);
    }
}

int shape_io::type() const
{
    return type_;
}

const Envelope<double>& shape_io::current_extent() const
{
    return cur_extent_;
}

shape_file& shape_io::shp()
{
    return shp_;
}

shape_file& shape_io::shx()
{
    return shx_;
}


dbf_file& shape_io::dbf()
{
    return dbf_;
}


geometry_ptr shape_io::read_polyline()
{    
    shape_record record(reclength_*2-36);
    shp_.read_record(record);
    int num_parts=record.read_ndr_integer();
    int num_points=record.read_ndr_integer();
    geometry_ptr line(new line_string_impl(-1));
    line->set_capacity(num_points + num_parts);
    if (num_parts == 1)
    {
	line->set_capacity(num_points + 1);
	record.skip(4);
	double x=record.read_double();
	double y=record.read_double();
	line->move_to(x,y);
	for (int i=1;i<num_points;++i)
	{
	    x=record.read_double();
	    y=record.read_double();
	    line->line_to(x,y);
	}
    }
    else
    {
	std::vector<int> parts(num_parts);
	for (int i=0;i<num_parts;++i)
	{
	    parts[i]=record.read_ndr_integer();
	}

	int start,end;
	for (int k=0;k<num_parts;++k)
	{
	    start=parts[k];
	    if (k==num_parts-1)
		end=num_points;
	    else
		end=parts[k+1];
	    
	    double x=record.read_double();
	    double y=record.read_double();
	    line->move_to(x,y);
	    
	    for (int j=start+1;j<end;++j)
	    {
		x=record.read_double();
		y=record.read_double();
		line->line_to(x,y);
	    }
	}
    }
    return line;
}

geometry_ptr shape_io::read_polylinem()
{    
    shape_record record(reclength_*2-36);
    shp_.read_record(record);
    int num_parts=record.read_ndr_integer();
    int num_points=record.read_ndr_integer();
    geometry_ptr line(new line_string_impl(-1));
    line->set_capacity(num_points + num_parts);
    if (num_parts == 1)
    {
	record.skip(4);
	double x=record.read_double();
	double y=record.read_double();
	line->move_to(x,y);
	for (int i=1;i<num_points;++i)
	{
	    x=record.read_double();
	    y=record.read_double();
	    line->line_to(x,y);
	}
    }
    else
    {
	std::vector<int> parts(num_parts);
	for (int i=0;i<num_parts;++i)
	{
	    parts[i]=record.read_ndr_integer();
	}

	int start,end;
	for (int k=0;k<num_parts;++k)
	{
	    start=parts[k];
	    if (k==num_parts-1)
		end=num_points;
	    else
		end=parts[k+1];
	    
	    double x=record.read_double();
	    double y=record.read_double();
	    line->move_to(x,y);
	    
	    for (int j=start+1;j<end;++j)
	    {
		x=record.read_double();
		y=record.read_double();
		line->line_to(x,y);
	    }
	}
    }
    // m-range
    //double m0=record.read_double();
    //double m1=record.read_double();
    
    //for (int i=0;i<num_points;++i)
    //{
    //   double m=record.read_double();
    //}
    
    return line;
}

geometry_ptr shape_io::read_polylinez()
{    
    shape_record record(reclength_*2-36);
    shp_.read_record(record);
    int num_parts=record.read_ndr_integer();
    int num_points=record.read_ndr_integer();
    geometry_ptr line(new line_string_impl(-1));
    line->set_capacity(num_points + num_parts);
    if (num_parts == 1)
    {
	record.skip(4);
	double x=record.read_double();
	double y=record.read_double();
	line->move_to(x,y);
	for (int i=1;i<num_points;++i)
	{
	    x=record.read_double();
	    y=record.read_double();
	    line->line_to(x,y);
	}
    }
    else
    {
	std::vector<int> parts(num_parts);
	for (int i=0;i<num_parts;++i)
	{
	    parts[i]=record.read_ndr_integer();
	}

	int start,end;
	for (int k=0;k<num_parts;++k)
	{
	    start=parts[k];
	    if (k==num_parts-1)
		end=num_points;
	    else
		end=parts[k+1];
	    
	    double x=record.read_double();
	    double y=record.read_double();
	    line->move_to(x,y);
	    
	    for (int j=start+1;j<end;++j)
	    {
		x=record.read_double();
		y=record.read_double();
		line->line_to(x,y);
	    }
	}
    }
    // z-range
    //double z0=record.read_double();
    //double z1=record.read_double();
    //for (int i=0;i<num_points;++i)
    // {
    //	double z=record.read_double();
    // }
    
    // m-range
    //double m0=record.read_double();
    //double m1=record.read_double();
    
    //for (int i=0;i<num_points;++i)
    //{
    //   double m=record.read_double();
    //} 
    return line;
}

geometry_ptr shape_io::read_polygon()
{
    shape_record record(reclength_*2-36);
    shp_.read_record(record);
    int num_parts=record.read_ndr_integer();
    int num_points=record.read_ndr_integer();
    std::vector<int> parts(num_parts);
    geometry_ptr poly(new polygon_impl(-1));
    poly->set_capacity(num_points + num_parts);
    for (int i=0;i<num_parts;++i)
    {
        parts[i]=record.read_ndr_integer();
    }

    for (int k=0;k<num_parts;k++)
    {
        int start=parts[k];
        int end;
        if (k==num_parts-1)
        {
            end=num_points;
        }
        else
        {
            end=parts[k+1];
        }
	double x=record.read_double();
	double y=record.read_double();
        poly->move_to(x,y);
   
        for (int j=start+1;j<end;j++)
        {
	    x=record.read_double();
	    y=record.read_double();
            poly->line_to(x,y);
        }
    }
    return poly;
}

geometry_ptr shape_io::read_polygonm()
{
    shape_record record(reclength_*2-36);
    shp_.read_record(record);
    int num_parts=record.read_ndr_integer();
    int num_points=record.read_ndr_integer();
    std::vector<int> parts(num_parts);
    geometry_ptr poly(new polygon_impl(-1));
    poly->set_capacity(num_points + num_parts);
    for (int i=0;i<num_parts;++i)
    {
        parts[i]=record.read_ndr_integer();
    }
    
    for (int k=0;k<num_parts;k++)
    {
        int start=parts[k];
        int end;
        if (k==num_parts-1)
        {
            end=num_points;
        }
        else
        {
            end=parts[k+1];
        }
	double x=record.read_double();
	double y=record.read_double();
        poly->move_to(x,y);
   
        for (int j=start+1;j<end;j++)
        {
	    x=record.read_double();
	    y=record.read_double();
            poly->line_to(x,y);
        }
    }
    // m-range
    //double m0=record.read_double();
    //double m1=record.read_double();
    
    //for (int i=0;i<num_points;++i)
    //{
    //   double m=record.read_double();
    //} 
    return poly;
}

geometry_ptr shape_io::read_polygonz()
{
    shape_record record(reclength_*2-36);
    shp_.read_record(record);
    int num_parts=record.read_ndr_integer();
    int num_points=record.read_ndr_integer();
    std::vector<int> parts(num_parts);
    geometry_ptr poly(new polygon_impl(-1));
    poly->set_capacity(num_points + num_parts);
    for (int i=0;i<num_parts;++i)
    {
        parts[i]=record.read_ndr_integer();
    }
    
    for (int k=0;k<num_parts;k++)
    {
        int start=parts[k];
        int end;
        if (k==num_parts-1)
        {
            end=num_points;
        }
        else
        {
            end=parts[k+1];
        }
	double x=record.read_double();
	double y=record.read_double();
        poly->move_to(x,y);
   
        for (int j=start+1;j<end;j++)
        {
	    x=record.read_double();
	    y=record.read_double();
            poly->line_to(x,y);
        }
    }
    // z-range
    //double z0=record.read_double();
    //double z1=record.read_double();
    //for (int i=0;i<num_points;++i)
    //{
    //	double z=record.read_double();
    //}
    
    // m-range
    //double m0=record.read_double();
    //double m1=record.read_double();
    
    //for (int i=0;i<num_points;++i)
    //{
    //   double m=record.read_double();
    //} 
    return poly;
}