CityDao.java

/*******************************************************************************
 *   Gisgraphy Project 
 * 
 *   This library is free software; you can redistribute it and/or
 *   modify it under the terms of the GNU Lesser General Public
 *   License as published by the Free Software Foundation; either
 *   version 2.1 of the License, or (at your option) any later version.
 * 
 *   This library 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
 *   Lesser General Public License for more details.
 * 
 *   You should have received a copy of the GNU Lesser General Public
 *   License along with this library; if not, write to the Free Software
 *   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA
 * 
 *  Copyright 2008  Gisgraphy project 
 *  David Masclet <davidmasclet@gisgraphy.com>
 *  
 *  
 *******************************************************************************/
package com.gisgraphy.domain.repository;

import java.util.ArrayList;
import java.util.List;
import java.util.Set;

import javax.persistence.PersistenceException;

import org.hibernate.Query;
import org.hibernate.Session;
import org.springframework.orm.hibernate3.HibernateCallback;
import org.springframework.stereotype.Repository;
import org.springframework.util.Assert;

import com.gisgraphy.domain.geoloc.entity.Adm;
import com.gisgraphy.domain.geoloc.entity.City;
import com.gisgraphy.domain.geoloc.entity.OpenStreetMap;
import com.gisgraphy.domain.geoloc.entity.ZipCode;
import com.gisgraphy.domain.valueobject.SRID;
import com.vividsolutions.jts.geom.Point;

/**
 * A data access object for {@link City}
 * 
 * @author <a href="mailto:david.masclet@gisgraphy.com">David Masclet</a>
 */
@Repository
public class CityDao extends GenericGisDao<City> implements ICityDao {

    /**
     * Default constructor
     */
    public CityDao() {
	super(City.class);
    }

    /*
     * (non-Javadoc)
     * 
     * @see com.gisgraphy.domain.repository.ICityDao#ListByZipCode(int,
     *      java.lang.String)
     */
    @SuppressWarnings("unchecked")
    public List<City> listByZipCode(final String zipcode, final String countrycode) {
	return (List<City>) this.getHibernateTemplate().execute(
		new HibernateCallback() {

		    public Object doInHibernate(Session session)
			    throws PersistenceException {
			String queryString = "from "
				+ City.class.getSimpleName()
				+ " as c left outer join c.zipCodes z where z.code = ?";
			if (countrycode != null) {
			    queryString = queryString + " and c.countryCode=?";
			}

			Query qry = session.createQuery(queryString);
			qry.setCacheable(true);

			qry.setParameter(0, zipcode);
			if (countrycode != null) {
			    qry.setParameter(1, countrycode.toUpperCase());
			}

			List<City> results = (List<City>) qry.list();
			if (results == null) {
			    results = new ArrayList<City>();
			}
			return results;
		    }
		});
    }
    
   
    

	@SuppressWarnings({ "unchecked", "rawtypes" })
	public City getByShape(final Point location,final String countryCode,final boolean filterMunicipality) {
		Assert.notNull(location);
		return (City) this.getHibernateTemplate().execute(new HibernateCallback() {

		    public Object doInHibernate(Session session)
			    throws PersistenceException {
		    	//select name,municipality,source,openstreetmapid from city c 
		    	//where st_contains(c.shape,ST_GeometryFromText('POINT(2.349 48.868)',4326))=true limit 1
		    String pointAsString = "ST_GeometryFromText('POINT("+location.getX()+" "+location.getY()+")',"+SRID.WGS84_SRID.getSRID()+")";
			String queryString = "from " + persistentClass.getSimpleName()
				+ " as c where st_contains(c.shape,"+pointAsString+")=true ";
			if (filterMunicipality){
				queryString+=" and c.municipality=true";
			}
			if (countryCode!=null ){
				queryString+=" and c.countryCode='"+countryCode+"'";
			}
			queryString = queryString+ " order by st_area(c.shape)";
			//we need to sort by shape due to error in osm data 
			//eg : if we search for the nearest city of http://www.openstreetmap.org/way/27904415
			// we can have 2 cities : http://www.openstreetmap.org/way/75509282 vs http://www.openstreetmap.org/relation/388250
			//cause there is the city and the district

			Query qry = session.createQuery(queryString).setMaxResults(1);

			//qry.setParameter("point2", location);
			City result = (City) qry.uniqueResult();
			//initialize
			if (result!=null){
				Set<ZipCode> zipCodes = result.getZipCodes();
				if (zipCodes!=null){
					zipCodes.size();
				}
				Adm adm = result.getAdm();
				if (adm!=null){
					adm.getName();
				}
			}

			return result;
		    }
		});
	}
	
	@SuppressWarnings({ "rawtypes", "unchecked" })
	public int fixPolygons(){
		return (Integer) this.getHibernateTemplate().execute(
				 new HibernateCallback() {

				    public Object doInHibernate(Session session)
					    throws PersistenceException {
					session.flush();
					logger.info("will fix polygons for city ");
					String fixPolygon= "update city set shape=ST_MakePolygon(shape) where ST_GeometryType(shape)='ST_LineString'";
					Query fixPolygonQuery = session.createSQLQuery(fixPolygon);
					int nbModify = fixPolygonQuery.executeUpdate();
					
					return nbModify;
				    }
				});
		
	}

	
}