e9bda4d9470b596a54c4c7e5722d3660b6fddd46
[jabaws.git] / webservices / compbio / ws / server / TcoffeeWS.java
1 /* Copyright (c) 2011 Peter Troshin\r
2  *  \r
3  *  JAva Bioinformatics Analysis Web Services (JABAWS) @version: 2.0     \r
4  * \r
5  *  This library is free software; you can redistribute it and/or modify it under the terms of the\r
6  *  Apache License version 2 as published by the Apache Software Foundation\r
7  * \r
8  *  This library is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without\r
9  *  even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the Apache \r
10  *  License for more details.\r
11  * \r
12  *  A copy of the license is in apache_license.txt. It is also available here:\r
13  * @see: http://www.apache.org/licenses/LICENSE-2.0.txt\r
14  * \r
15  * Any republication or derived work distributed in source code form\r
16  * must include this copyright and license notice.\r
17  */\r
18 \r
19 package compbio.ws.server;\r
20 \r
21 import java.io.File;\r
22 import java.util.List;\r
23 \r
24 import javax.jws.WebService;\r
25 \r
26 import org.apache.log4j.Logger;\r
27 \r
28 import compbio.data.msa.JABAService;\r
29 import compbio.data.msa.MsaWS;\r
30 import compbio.data.sequence.Alignment;\r
31 import compbio.data.sequence.FastaSequence;\r
32 import compbio.engine.AsyncExecutor;\r
33 import compbio.engine.Configurator;\r
34 import compbio.engine.client.ConfiguredExecutable;\r
35 import compbio.engine.client.Executable;\r
36 import compbio.engine.client.SkeletalExecutable;\r
37 import compbio.metadata.ChunkHolder;\r
38 import compbio.metadata.JobStatus;\r
39 import compbio.metadata.JobSubmissionException;\r
40 import compbio.metadata.Limit;\r
41 import compbio.metadata.LimitsManager;\r
42 import compbio.metadata.Option;\r
43 import compbio.metadata.Preset;\r
44 import compbio.metadata.PresetManager;\r
45 import compbio.metadata.ResultNotAvailableException;\r
46 import compbio.metadata.RunnerConfig;\r
47 import compbio.metadata.WrongParameterException;\r
48 import compbio.runner.Util;\r
49 import compbio.runner.msa.Tcoffee;\r
50 \r
51 @WebService(endpointInterface = "compbio.data.msa.MsaWS", targetNamespace = JABAService.SERVICE_NAMESPACE, serviceName = "TcoffeeWS")\r
52 public class TcoffeeWS implements MsaWS<Tcoffee> {\r
53 \r
54         private static Logger log = Logger.getLogger(TcoffeeWS.class);\r
55 \r
56         private static final RunnerConfig<Tcoffee> tcoffeeOptions = Util\r
57                         .getSupportedOptions(Tcoffee.class);\r
58 \r
59         private static final PresetManager<Tcoffee> tcoffeePresets = Util\r
60                         .getPresets(Tcoffee.class);\r
61 \r
62         private static final LimitsManager<Tcoffee> limitMan = compbio.engine.client.Util\r
63                         .getLimits(new Tcoffee().getType());\r
64 \r
65         @Override\r
66         public String align(List<FastaSequence> sequences)\r
67                         throws JobSubmissionException {\r
68                 WSUtil.validateFastaInput(sequences);\r
69                 ConfiguredExecutable<Tcoffee> confTcoffee = init(sequences);\r
70                 return WSUtil\r
71                                 .align(sequences, confTcoffee, null, "align", getLimit(""));\r
72         }\r
73 \r
74         ConfiguredExecutable<Tcoffee> init(List<FastaSequence> sequences)\r
75                         throws JobSubmissionException {\r
76                 Tcoffee tcoffee = new Tcoffee();\r
77                 tcoffee.setInput(SkeletalExecutable.INPUT)\r
78                                 .setOutput(SkeletalExecutable.OUTPUT)\r
79                                 .setError(SkeletalExecutable.ERROR);\r
80 \r
81                 ConfiguredExecutable<Tcoffee> confCoffee = Configurator\r
82                                 .configureExecutable(tcoffee, sequences);\r
83                 if (confCoffee.getExecProvider() == Executable.ExecProvider.Cluster) {\r
84                         int clusterCpuNum = SkeletalExecutable\r
85                                         .getClusterCpuNum(Tcoffee.class);\r
86                         if (clusterCpuNum != 0) {\r
87                                 tcoffee.setNCore(clusterCpuNum);\r
88                         }\r
89                 }\r
90                 return confCoffee;\r
91         }\r
92 \r
93         @Override\r
94         public String customAlign(List<FastaSequence> sequences,\r
95                         List<Option<Tcoffee>> options) throws JobSubmissionException,\r
96                         WrongParameterException {\r
97 \r
98                 WSUtil.validateFastaInput(sequences);\r
99                 ConfiguredExecutable<Tcoffee> confTcoffee = init(sequences);\r
100                 List<String> params = WSUtil.getCommands(options,\r
101                                 Tcoffee.KEY_VALUE_SEPARATOR);\r
102                 log.info("Setting parameters:" + params);\r
103                 confTcoffee.addParameters(params);\r
104                 return WSUtil.align(sequences, confTcoffee, null, "customAlign",\r
105                                 getLimit(""));\r
106         }\r
107 \r
108         @Override\r
109         public String presetAlign(List<FastaSequence> sequences,\r
110                         Preset<Tcoffee> preset) throws JobSubmissionException,\r
111                         WrongParameterException {\r
112                 WSUtil.validateFastaInput(sequences);\r
113                 if (preset == null) {\r
114                         throw new WrongParameterException("Preset must be provided!");\r
115                 }\r
116                 ConfiguredExecutable<Tcoffee> confTcoffee = init(sequences);\r
117                 confTcoffee.addParameters(preset.getOptions());\r
118                 Limit<Tcoffee> limit = getLimit(preset.getName());\r
119                 return WSUtil.align(sequences, confTcoffee, null, "presetAlign", limit);\r
120         }\r
121 \r
122         @SuppressWarnings("unchecked")\r
123         @Override\r
124         public Alignment getResult(String jobId) throws ResultNotAvailableException {\r
125                 WSUtil.validateJobId(jobId);\r
126                 AsyncExecutor asyncEngine = Configurator.getAsyncEngine(jobId);\r
127                 ConfiguredExecutable<Tcoffee> tcoffee = (ConfiguredExecutable<Tcoffee>) asyncEngine\r
128                                 .getResults(jobId);\r
129                 Alignment al = tcoffee.getResults();\r
130                 // log(jobId, "getResults");\r
131                 return al;\r
132         }\r
133 \r
134         @Override\r
135         public Limit<Tcoffee> getLimit(String presetName) {\r
136                 if (limitMan == null) {\r
137                         // Limit is not defined\r
138                         return null;\r
139                 }\r
140                 return limitMan.getLimitByName(presetName);\r
141         }\r
142 \r
143         @Override\r
144         public LimitsManager<Tcoffee> getLimits() {\r
145                 return limitMan;\r
146         }\r
147 \r
148         @Override\r
149         public ChunkHolder pullExecStatistics(String jobId, long position) {\r
150                 WSUtil.validateJobId(jobId);\r
151                 String file = Configurator.getWorkDirectory(jobId) + File.separator\r
152                                 + new Tcoffee().getError();\r
153                 return WSUtil.pullFile(file, position);\r
154         }\r
155 \r
156         @Override\r
157         public boolean cancelJob(String jobId) {\r
158                 WSUtil.validateJobId(jobId);\r
159                 return WSUtil.cancelJob(jobId);\r
160         }\r
161 \r
162         @Override\r
163         public JobStatus getJobStatus(String jobId) {\r
164                 WSUtil.validateJobId(jobId);\r
165                 return WSUtil.getJobStatus(jobId);\r
166         }\r
167 \r
168         @Override\r
169         public PresetManager<Tcoffee> getPresets() {\r
170                 return tcoffeePresets;\r
171         }\r
172 \r
173         @Override\r
174         public RunnerConfig<Tcoffee> getRunnerOptions() {\r
175                 return tcoffeeOptions;\r
176         }\r
177 \r
178 }\r