QueryUtils.java

/*
 * Copyright (c) 2007 Genome Research Limited.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU Library General Public License as published
 * by  the Free Software Foundation; either version 2 of the License or
 * (at your option) 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 Library General Public License for more details.
 *
 * You should have received a copy of the GNU Library General Public License
 * along with this program; see the file COPYING.LIB.  If not, write to
 * the Free Software Foundation Inc., 59 Temple Place - Suite 330,
 * Boston, MA  02111-1307 USA
 */

package org.genedb.querying.core;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.apache.log4j.Logger;
import org.genedb.db.taxon.TaxonNameType;
import org.genedb.db.taxon.TaxonNode;
import org.genedb.db.taxon.TaxonNodeList;
import org.genedb.query.sql.SqlQuery;
import org.genedb.querying.tmpquery.OrganismHqlQuery;
import org.genedb.querying.tmpquery.OrganismLuceneQuery;
import org.genedb.querying.tmpquery.ProteinMatchClusterOrthologueQuery;
import org.springframework.beans.BeanWrapperImpl;
import org.springframework.util.StringUtils;

public class QueryUtils {
	
	private static final Logger logger = Logger.getLogger(QueryUtils.class);
	
    public static String makeParseableDescription(String name, String[] paramNames, Object[] paramValues) {
        StringBuilder ret = new StringBuilder();
        ret.append(name);
        ret.append('{');
        if (paramNames != null) {
            for (int i = 0; i < paramNames.length; i++) {
                if (i>0) {
                    ret.append(';');
                }
                ret.append(paramNames[i]);
                ret.append('=');
                ret.append(paramValues[i]);
            }
        }
        ret.append('}');
        return ret.toString();
    }
    
    public static String makeParseableDescription(String name, String[] paramNames, Object target) {
        return makeParseableDescription(name, paramNames, getParamValues(target, paramNames));
    }
    
    public static Object[] getParamValues(Object target, String[] paramNames) {
        Object[] ret = new Object[paramNames.length];
        BeanWrapperImpl wrapper = new BeanWrapperImpl(target);
        for (int i = 0; i < paramNames.length; i++) {
            String name = paramNames[i];
            ret[i] = wrapper.getPropertyValue(name);
        }
        return ret;
    }
    
    public static Map<String, String> getParameterMap(Query query) {
    	Map<String, String> map = new HashMap<String, String>();
    	
    	String[] paramNames = null;
    	
    	TaxonNodeList taxons = null;
    	
    	if (query instanceof LuceneQuery) {
    		LuceneQuery luceneQuery = (LuceneQuery) query;
    		paramNames = luceneQuery.getParamNames();
    		
    		if (luceneQuery instanceof OrganismLuceneQuery) {
    			taxons = ((OrganismLuceneQuery) luceneQuery).getTaxons();
    		}
    		
    		
    	} else if (query instanceof HqlQuery) {
    		HqlQuery hqlQuery = (HqlQuery) query;
    		paramNames = hqlQuery.getParamNames();
    		
    		if (hqlQuery instanceof OrganismHqlQuery) {
    			taxons = ((OrganismHqlQuery) hqlQuery).getTaxons();
    		}
    	
    	} else if (query instanceof ProteinMatchClusterOrthologueQuery) {
    		ProteinMatchClusterOrthologueQuery pQuery = (ProteinMatchClusterOrthologueQuery) query;
    		paramNames = pQuery.getParamNames();
    		
    	} else if (query instanceof SqlQuery) {
    		logger.warn ("skipping sql query for now...");
    	}
    	
    	Object[] paramValues = getParamValues(query, paramNames);
    	
    	for (int i = 0; i < paramNames.length; i++) {
    		map.put(paramNames[i], paramValues[i].toString());
    	}
    	
    	if (taxons != null) {
    		if (taxons.getNodeCount() > 0) {
        		TaxonNode node = taxons.getNodes().get(0);
        		String nodeName = node.getName(TaxonNameType.HTML_FULL);
        		if (nodeName == null) {
        			nodeName = node.getName(TaxonNameType.FULL);
        		}
        		
        		map.put("taxon", nodeName);
        	} else {
        		map.put("taxon", "--");
        	}
    	}
    	
    	
    	    	
		return map;
    }

}