Change the way to deal with Limits to simplify wrapper writing and enable couping...
[jabaws.git] / webservices / compbio / ws / server / AAConWS.java
1 package compbio.ws.server;\r
2 \r
3 import java.io.File;\r
4 import java.util.Arrays;\r
5 import java.util.HashSet;\r
6 import java.util.List;\r
7 \r
8 import javax.annotation.Resource;\r
9 import javax.jws.WebService;\r
10 import javax.xml.ws.WebServiceContext;\r
11 \r
12 import org.apache.log4j.Logger;\r
13 \r
14 import compbio.data.msa.SequenceAnnotation;\r
15 import compbio.data.sequence.FastaSequence;\r
16 import compbio.data.sequence.Score;\r
17 import compbio.data.sequence.ScoreManager;\r
18 import compbio.engine.AsyncExecutor;\r
19 import compbio.engine.Configurator;\r
20 import compbio.engine.client.ConfiguredExecutable;\r
21 import compbio.metadata.ChunkHolder;\r
22 import compbio.metadata.JobStatus;\r
23 import compbio.metadata.JobSubmissionException;\r
24 import compbio.metadata.Limit;\r
25 import compbio.metadata.LimitExceededException;\r
26 import compbio.metadata.LimitsManager;\r
27 import compbio.metadata.Option;\r
28 import compbio.metadata.Preset;\r
29 import compbio.metadata.PresetManager;\r
30 import compbio.metadata.ResultNotAvailableException;\r
31 import compbio.metadata.RunnerConfig;\r
32 import compbio.metadata.UnsupportedRuntimeException;\r
33 import compbio.metadata.WrongParameterException;\r
34 import compbio.runner.Util;\r
35 import compbio.runner.conservation.AACon;\r
36 \r
37 @WebService(endpointInterface = "compbio.data.msa.SequenceAnnotation", targetNamespace = "http://msa.data.compbio/01/12/2010/", serviceName = "AAConWS")\r
38 public class AAConWS implements SequenceAnnotation<AACon> {\r
39 \r
40         // Ask for resource injection\r
41         @Resource\r
42         WebServiceContext wsContext;\r
43 \r
44         private static Logger statLog = Logger.getLogger("AAConWS-stats");\r
45 \r
46         private static Logger log = Logger.getLogger(AAConWS.class);\r
47 \r
48         private static final RunnerConfig<AACon> aaconOptions = Util\r
49                         .getSupportedOptions(AACon.class);\r
50 \r
51         private static final PresetManager<AACon> aaconPresets = Util\r
52                         .getPresets(AACon.class);\r
53 \r
54         private static final LimitsManager<AACon> limitMan = compbio.runner.Util\r
55                         .getLimits(new AACon().getType());\r
56 \r
57         ConfiguredExecutable<AACon> init(List<FastaSequence> sequences)\r
58                         throws JobSubmissionException {\r
59                 AACon aacon = new AACon();\r
60                 aacon.setInput("fasta.in").setOutput("aacon.out");\r
61                 return Configurator.configureExecutable(aacon, sequences);\r
62         }\r
63 \r
64         @Override\r
65         public ScoreManager getAnnotation(String jobId)\r
66                         throws ResultNotAvailableException {\r
67                 WSUtil.validateJobId(jobId);\r
68                 AsyncExecutor asyncEngine = Configurator.getAsyncEngine(jobId);\r
69                 ConfiguredExecutable<AACon> aacon = (ConfiguredExecutable<AACon>) asyncEngine\r
70                                 .getResults(jobId);\r
71                 HashSet<Score> mas = aacon.getResults();\r
72                 // ScoreManager result = new HashMap<String, HashSet<Score>>();\r
73                 // result.put("Alignment", mas);\r
74                 log.trace(jobId + " getConservation : " + mas);\r
75                 // FIXME\r
76                 return null;\r
77         }\r
78         /*\r
79          * @SuppressWarnings("unchecked") public JalviewAnnotation\r
80          * getJalviewAnnotation(String jobId) throws ResultNotAvailableException {\r
81          * MultiAnnotatedSequence<Method> result = getResult(jobId); // TODO //\r
82          * log(jobId, "getResults"); return result.toJalviewAnnotation(); }\r
83          */\r
84 \r
85         @Override\r
86         public Limit<AACon> getLimit(String presetName) {\r
87                 return limitMan.getLimitByName(presetName);\r
88         }\r
89 \r
90         @Override\r
91         public LimitsManager<AACon> getLimits() {\r
92                 return limitMan;\r
93         }\r
94 \r
95         @Override\r
96         public ChunkHolder pullExecStatistics(String jobId, long position) {\r
97                 WSUtil.validateJobId(jobId);\r
98                 String file = Configurator.getWorkDirectory(jobId) + File.separator\r
99                                 + AACon.getStatFile();\r
100                 return WSUtil.pullFile(file, position);\r
101         }\r
102 \r
103         @Override\r
104         public boolean cancelJob(String jobId) {\r
105                 WSUtil.validateJobId(jobId);\r
106                 return WSUtil.cancelJob(jobId);\r
107         }\r
108 \r
109         @Override\r
110         public JobStatus getJobStatus(String jobId) {\r
111                 WSUtil.validateJobId(jobId);\r
112                 return WSUtil.getJobStatus(jobId);\r
113         }\r
114 \r
115         @Override\r
116         public PresetManager<AACon> getPresets() {\r
117                 return aaconPresets;\r
118         }\r
119 \r
120         @Override\r
121         public RunnerConfig<AACon> getRunnerOptions() {\r
122                 return aaconOptions;\r
123         }\r
124 \r
125         String analize(List<FastaSequence> sequences,\r
126                         ConfiguredExecutable<AACon> confExec, Logger log, String method,\r
127                         Limit<AACon> limit) throws JobSubmissionException {\r
128                 if (limit != null && limit.isExceeded(sequences)) {\r
129                         throw LimitExceededException.newLimitExceeded(limit, sequences);\r
130                 }\r
131 \r
132                 compbio.runner.Util.writeInput(sequences, confExec);\r
133                 AsyncExecutor engine = Configurator.getAsyncEngine(confExec);\r
134                 String jobId = engine.submitJob(confExec);\r
135                 return jobId;\r
136         }\r
137 \r
138         @Override\r
139         public String analize(List<FastaSequence> sequences)\r
140                         throws UnsupportedRuntimeException, LimitExceededException,\r
141                         JobSubmissionException {\r
142                 WSUtil.validateFastaInput(sequences);\r
143                 ConfiguredExecutable<AACon> confAAcon = init(sequences);\r
144 \r
145                 // set default conservation method to fastest - SHENKIN\r
146                 // TODO: This violates encapsulation, should be moved to the runners\r
147                 // level.\r
148                 confAAcon.addParameters(Arrays.asList("-m=SHENKIN"));\r
149                 return analize(sequences, confAAcon, null, "analize", getLimit(""));\r
150         }\r
151 \r
152         @Override\r
153         public String customAnalize(List<FastaSequence> sequences,\r
154                         List<Option<AACon>> options) throws UnsupportedRuntimeException,\r
155                         LimitExceededException, JobSubmissionException,\r
156                         WrongParameterException {\r
157                 WSUtil.validateFastaInput(sequences);\r
158                 ConfiguredExecutable<AACon> confAACon = init(sequences);\r
159                 // Could not do that! Space separated values\r
160                 // will all be treated as keys! thus duplicates removed\r
161                 // String params = cbuilder.getCommand();\r
162                 List<String> params = WSUtil.getCommands(options,\r
163                                 AACon.KEY_VALUE_SEPARATOR);\r
164                 confAACon.addParameters(params);\r
165                 return analize(sequences, confAACon, null, "customAnalize",\r
166                                 getLimit(""));\r
167         }\r
168 \r
169         @Override\r
170         public String presetAnalize(List<FastaSequence> sequences,\r
171                         Preset<AACon> preset) throws UnsupportedRuntimeException,\r
172                         LimitExceededException, JobSubmissionException,\r
173                         WrongParameterException {\r
174                 WSUtil.validateFastaInput(sequences);\r
175                 if (preset == null) {\r
176                         throw new WrongParameterException("Preset must be provided!");\r
177                 }\r
178                 ConfiguredExecutable<AACon> confAAcon = init(sequences);\r
179                 confAAcon.addParameters(preset.getOptions());\r
180                 Limit<AACon> limit = getLimit(preset.getName());\r
181                 return WSUtil.align(sequences, confAAcon, null, "presetAnalize", limit);\r
182         }\r
183 \r
184 }\r