Improve presentation of job on the web (servlet ServletSequenceProtein)
[proteocache.git] / server / compbio / statistic / CassandraRequester.java
index ec74c6f..214bf6f 100755 (executable)
@@ -9,12 +9,12 @@ import java.util.Iterator;
 import java.util.List;
 import java.util.Map;
 
+import compbio.cassandra.AnnotatedProteinSequenceBean;
 import compbio.cassandra.CassandraNativeConnector;
 import compbio.cassandra.CassandraReader;
 import compbio.cassandra.DataBase;
 import compbio.cassandra.Pair;
 import compbio.cassandra.StructureJobLog;
-import compbio.cassandra.StructureProteinPrediction;
 
 public class CassandraRequester {
        private CassandraReader db = new CassandraReader();
@@ -156,35 +156,32 @@ public class CassandraRequester {
                System.out.println("StatisticsProt.readLength: total number of dates = " + query.size());
                return query;
        }
+
        /* 
         * query: protein sequence
         * */
        public List<DataBase> readProteins(String protIn, String flag) {
                query = new ArrayList<DataBase>();
-               List<StructureProteinPrediction> res;
+               List<AnnotatedProteinSequenceBean> res;
                if (flag.equals("whole")) 
                        res = db.ReadWholeSequence(protIn);
                 else 
                        res = db.ReadPartOfSequence(protIn);
                if (res == null)
                        return null;
-               for (StructureProteinPrediction entry : res) {
-                       Map<String,String> pred = entry.getPrediction();
-                       Iterator it = pred.entrySet().iterator();
-                       while (it.hasNext()) {
-                               DataBase db = new DataBase();
-                               db.setProt(entry.getSequence());
-                               Map.Entry pairs = (Map.Entry)it.next();
-                               db.setId(entry.getJobid());
-                               db.setJpred(pairs.getValue().toString());
-                               if (flag.equals("part"))
-                                       db.setSubProt(CreateSubprot (entry.getSequence(), protIn));                             
-                               query.add(db);
+               for (AnnotatedProteinSequenceBean entry : res) {
+                       DataBase db = new DataBase();
+                       db.setProt(entry.getSequence());
+                       db.setPredictions(entry);
+                       db.setId(entry.getJobid());
+                       if (flag.equals("part")) {
+                               db.setSubProt(CreateSubprot (entry.getSequence(), protIn));
                        }
+                       query.add(db);
                }
                return query;
        }
-       
+
        /* 
         * query protein sequences with number of jobs
         */