eebc96f25b97b00add1844971a0278a9cef17216
[jabaws.git] / webservices / compbio / ws / server / JpredWS.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.predictors.Jpred;\r
50 \r
51 @WebService(endpointInterface = "compbio.data.msa.MsaWS", targetNamespace = JABAService.V2_SERVICE_NAMESPACE, serviceName = "JpredWS")\r
52 public class JpredWS implements MsaWS<Jpred> {\r
53 \r
54         private static Logger log = Logger.getLogger(JpredWS.class);\r
55 \r
56         private static final RunnerConfig<Jpred> jpredOptions = Util.getSupportedOptions(Jpred.class);\r
57 \r
58         private static final PresetManager<Jpred> jpredPresets = Util.getPresets(Jpred.class);\r
59 \r
60         private static final LimitsManager<Jpred> limitMan = compbio.engine.client.Util.getLimits(new Jpred().getType());\r
61 \r
62         @Override\r
63         public String align(List<FastaSequence> sequences)\r
64                         throws JobSubmissionException {\r
65 \r
66                 WSUtil.validateFastaInput(sequences);\r
67                 ConfiguredExecutable<Jpred> confClust = init(sequences);\r
68                 return WSUtil.align(sequences, confClust, log, "align", getLimit(""));\r
69         }\r
70 \r
71         ConfiguredExecutable<Jpred> init(List<FastaSequence> dataSet)\r
72                         throws JobSubmissionException {\r
73                 Jpred jpred = new Jpred();\r
74                 jpred.setInput(SkeletalExecutable.INPUT);\r
75                 jpred.setOutput(SkeletalExecutable.OUTPUT);\r
76                 jpred.setError(SkeletalExecutable.ERROR);\r
77                 ConfiguredExecutable<Jpred> confClust = Configurator.configureExecutable(jpred, dataSet);\r
78                 // Set the number of threads for the cluster execution from conf file\r
79                 if (confClust.getExecProvider() == Executable.ExecProvider.Cluster) {\r
80                         int clusterCpuNum = SkeletalExecutable.getClusterCpuNum(jpred.getType());\r
81                         if (clusterCpuNum != 0) {\r
82                                 jpred.setNCore(clusterCpuNum);\r
83                         }\r
84                 }\r
85                 return confClust;\r
86         }\r
87 \r
88         @Override\r
89         public String presetAlign(List<FastaSequence> sequences,\r
90                         Preset<Jpred> preset) throws JobSubmissionException,\r
91                         WrongParameterException {\r
92                 WSUtil.validateFastaInput(sequences);\r
93                 if (preset == null) {\r
94                         throw new WrongParameterException("Preset must be provided!");\r
95                 }\r
96                 Limit<Jpred> limit = getLimit(preset.getName());\r
97                 ConfiguredExecutable<Jpred> confClust = init(sequences);\r
98                 confClust.addParameters(preset.getOptions());\r
99                 return WSUtil.align(sequences, confClust, log, "presetAlign", limit);\r
100         }\r
101 \r
102         @Override\r
103         public String customAlign(List<FastaSequence> sequences,\r
104                         List<Option<Jpred>> options) throws JobSubmissionException,\r
105                         WrongParameterException {\r
106                 WSUtil.validateFastaInput(sequences);\r
107                 ConfiguredExecutable<Jpred> confClust = init(sequences);\r
108                 List<String> params = WSUtil.getCommands(options, Jpred.KEY_VALUE_SEPARATOR);\r
109                 confClust.addParameters(params);\r
110                 log.info("Setting parameters: " + params);\r
111                 return WSUtil.align(sequences, confClust, log, "customAlign", getLimit(""));\r
112         }\r
113 \r
114         @Override\r
115         public RunnerConfig<Jpred> getRunnerOptions() {\r
116                 return jpredOptions;\r
117         }\r
118 \r
119         @SuppressWarnings("unchecked")\r
120         @Override\r
121         public Alignment getResult(String jobId) throws ResultNotAvailableException {\r
122                 WSUtil.validateJobId(jobId);\r
123                 AsyncExecutor asyncEngine = Configurator.getAsyncEngine(jobId);\r
124                 ConfiguredExecutable<Jpred> jpred = (ConfiguredExecutable<Jpred>) asyncEngine.getResults(jobId);\r
125                 return jpred.getResults();\r
126         }\r
127 \r
128         @Override\r
129         public Limit<Jpred> getLimit(String presetName) {\r
130                 if (limitMan == null) {\r
131                         // No limit is configured\r
132                         return null;\r
133                 }\r
134                 Limit<Jpred> limit = limitMan.getLimitByName(presetName);\r
135                 return limit;\r
136         }\r
137 \r
138         @Override\r
139         public LimitsManager<Jpred> getLimits() {\r
140                 return limitMan;\r
141         }\r
142 \r
143         @Override\r
144         public boolean cancelJob(String jobId) {\r
145                 WSUtil.validateJobId(jobId);\r
146                 boolean result = WSUtil.cancelJob(jobId);\r
147                 return result;\r
148         }\r
149 \r
150         @Override\r
151         public JobStatus getJobStatus(String jobId) {\r
152                 WSUtil.validateJobId(jobId);\r
153                 JobStatus status = WSUtil.getJobStatus(jobId);\r
154                 return status;\r
155         }\r
156 \r
157         @Override\r
158         public PresetManager<Jpred> getPresets() {\r
159                 return jpredPresets;\r
160         }\r
161 \r
162         @Override\r
163         public ChunkHolder pullExecStatistics(String jobId, long position) {\r
164                 WSUtil.validateJobId(jobId);\r
165                 String file = Configurator.getWorkDirectory(jobId) + File.separator + Jpred.getStatFile();\r
166                 ChunkHolder cholder = WSUtil.pullFile(file, position);\r
167                 return cholder;\r
168         }\r
169 \r
170 }\r