package compbio.cassandra.readers;

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

import com.datastax.driver.core.ResultSet;
import com.datastax.driver.core.Row;


public class FeaturesReader extends CassandraReader {

	public FeaturesReader() {
		super();
	}

	/**
	 * query protein feature
	 * 
	 * @param feature
	 * 			protein SS feature
	 * 
	 * @param percent
	 * 			percent SS feature in protein sequence
	 * 
	 * @return UserBean to the controller SSFeaturesController
	 **/
	 
	 public Map<String, String> readProteinsPrediction(String feature, int percent) {
		 ResultSet results = CassandraQuery("SELECT * FROM ProteinRow;");
		 if (results.isExhausted())
				return null;
		List<Row> rows = results.all();
		Map<String, String> query = new HashMap<String, String>();
		for (Row r : rows) {
			String prot = r.getString("Protein");
			String prediction = findJnetpred(r.getMap("Predictions", String.class, String.class));
			if (prot != null && prediction != null && (!prot.equals(""))) {
				if (prediction.replaceAll("[^" + feature + "]", "").length() > prediction.length() * percent / 100 ) {
					query.put(prot, prediction);
				}
			}
		}
		return query;
	}

	 
	private String findJnetpred(Map<String, String> pred) {
		if (pred != null) {
			if (pred.containsKey("jnetpred"))
				return pred.get("jnetpred");
		}
		return null;
	}


}