2 * Jalview - A Sequence Alignment Editor and Viewer ($$Version-Rel$$)
3 * Copyright (C) $$Year-Rel$$ The Jalview Authors
5 * This file is part of Jalview.
7 * Jalview is free software: you can redistribute it and/or
8 * modify it under the terms of the GNU General Public License
9 * as published by the Free Software Foundation, either version 3
10 * of the License, or (at your option) any later version.
12 * Jalview is distributed in the hope that it will be useful, but
13 * WITHOUT ANY WARRANTY; without even the implied warranty
14 * of MERCHANTABILITY or FITNESS FOR A PARTICULAR
15 * PURPOSE. See the GNU General Public License for more details.
17 * You should have received a copy of the GNU General Public License
18 * along with Jalview. If not, see <http://www.gnu.org/licenses/>.
19 * The Jalview Authors are detailed in the 'AUTHORS' file.
22 package jalview.fts.service.pdb;
24 import jalview.fts.api.FTSDataColumnI;
25 import jalview.fts.api.FTSRestClientI;
26 import jalview.fts.core.FTSRestRequest;
27 import jalview.fts.core.FTSRestResponse;
28 import jalview.fts.core.GFTSPanel;
29 import jalview.gui.Help;
30 import jalview.gui.Help.HelpId;
31 import jalview.gui.SequenceFetcher;
32 import jalview.util.MessageManager;
34 import java.util.HashMap;
35 import java.util.HashSet;
38 import javax.help.HelpSetException;
40 @SuppressWarnings("serial")
41 public class PDBFTSPanel extends GFTSPanel
43 private static String defaultFTSFrameTitle = MessageManager
44 .getString("label.pdb_sequence_fetcher");
46 private static Map<String, Integer> tempUserPrefs = new HashMap<>();
48 private static final String PDB_FTS_CACHE_KEY = "CACHE.PDB_FTS";
50 private static final String PDB_AUTOSEARCH = "FTS.PDB.AUTOSEARCH";
52 public PDBFTSPanel(SequenceFetcher fetcher)
55 pageLimit = PDBFTSRestClient.getInstance().getDefaultResponsePageSize();
56 this.seqFetcher = fetcher;
57 this.progressIndicator = (fetcher == null) ? null
58 : fetcher.getProgressIndicator();
62 public void searchAction(boolean isFreshSearch)
64 mainFrame.requestFocusInWindow();
75 boolean allowEmptySequence = false;
76 if (getTypedText().length() > 0)
78 setSearchInProgress(true);
79 long startTime = System.currentTimeMillis();
81 String searchTarget = ((FTSDataColumnI) cmb_searchTarget
82 .getSelectedItem()).getCode();
83 wantedFields = PDBFTSRestClient.getInstance()
84 .getAllDefaultDisplayedFTSDataColumns();
85 String searchTerm = decodeSearchTerm(getTypedText(),
88 FTSRestRequest request = new FTSRestRequest();
89 request.setAllowEmptySeq(allowEmptySequence);
90 request.setResponseSize(100);
91 request.setFieldToSearchBy("(" + searchTarget + ":");
92 request.setSearchTerm(searchTerm + ")");
93 request.setOffSet(offSet);
94 request.setWantedFields(wantedFields);
95 FTSRestClientI pdbRestClient = PDBFTSRestClient.getInstance();
96 FTSRestResponse resultList;
99 resultList = pdbRestClient.executeRequest(request);
100 } catch (Exception e)
102 setErrorMessage(e.getMessage());
104 setSearchInProgress(false);
108 if (resultList.getSearchSummary() != null
109 && resultList.getSearchSummary().size() > 0)
111 getResultTable().setModel(FTSRestResponse.getTableModel(request,
112 resultList.getSearchSummary()));
113 FTSRestResponse.configureTableColumn(getResultTable(),
114 wantedFields, tempUserPrefs);
115 getResultTable().setVisible(true);
118 long endTime = System.currentTimeMillis();
119 totalResultSetCount = resultList.getNumberOfItemsFound();
120 resultSetCount = resultList.getSearchSummary() == null ? 0
121 : resultList.getSearchSummary().size();
122 String result = (resultSetCount > 0)
123 ? MessageManager.getString("label.results")
124 : MessageManager.getString("label.result");
126 if (isPaginationEnabled() && resultSetCount > 0)
128 updateSearchFrameTitle(defaultFTSFrameTitle + " - " + result
130 + totalNumberformatter.format((Number) (offSet + 1))
132 + totalNumberformatter
133 .format((Number) (offSet + resultSetCount))
135 + totalNumberformatter
136 .format((Number) totalResultSetCount)
137 + " " + " (" + (endTime - startTime) + " milli secs)");
141 updateSearchFrameTitle(defaultFTSFrameTitle + " - "
142 + resultSetCount + " " + result + " ("
143 + (endTime - startTime) + " milli secs)");
146 setSearchInProgress(false);
147 refreshPaginatorState();
148 updateSummaryTableSelections();
150 txt_search.updateCache();
155 public static String decodeSearchTerm(String enteredText,
158 String foundSearchTerms = enteredText;
159 StringBuilder foundSearchTermsBuilder = new StringBuilder();
160 if (enteredText.contains(";"))
162 String[] searchTerms = enteredText.split(";");
163 for (String searchTerm : searchTerms)
165 if (searchTerm.contains(":"))
167 foundSearchTermsBuilder.append(targetField).append(":")
168 .append(searchTerm.split(":")[0]).append(" OR ");
172 foundSearchTermsBuilder.append(targetField).append(":")
173 .append(searchTerm).append(" OR ");
176 int endIndex = foundSearchTermsBuilder.lastIndexOf(" OR ");
177 foundSearchTerms = foundSearchTermsBuilder.toString();
178 if (foundSearchTerms.contains(" OR "))
180 foundSearchTerms = foundSearchTerms
181 .substring(targetField.length() + 1, endIndex);
184 else if (enteredText.contains(":"))
186 foundSearchTerms = foundSearchTerms.split(":")[0];
188 return foundSearchTerms;
192 public void okAction()
194 // mainFrame.dispose();
195 disableActionButtons();
196 StringBuilder selectedIds = new StringBuilder();
197 HashSet<String> selectedIdsSet = new HashSet<>();
198 int primaryKeyColIndex = 0;
201 primaryKeyColIndex = getFTSRestClient()
202 .getPrimaryKeyColumIndex(wantedFields, false);
203 } catch (Exception e)
207 int[] selectedRows = getResultTable().getSelectedRows();
208 String searchTerm = getTypedText();
209 for (int summaryRow : selectedRows)
211 String idStr = getResultTable()
212 .getValueAt(summaryRow, primaryKeyColIndex).toString();
213 selectedIdsSet.add(getPDBIdwithSpecifiedChain(idStr, searchTerm));
216 for (String idStr : paginatorCart)
218 selectedIdsSet.add(getPDBIdwithSpecifiedChain(idStr, searchTerm));
221 for (String selectedId : selectedIdsSet)
223 selectedIds.append(selectedId).append(";");
226 String ids = selectedIds.toString();
227 // System.out.println(">>>>>>>>>>>>>>>> selected Ids: " + ids);
228 seqFetcher.getTextArea().setText(ids);
229 Thread worker = new Thread(seqFetcher);
231 delayAndEnableActionButtons();
234 public static String getPDBIdwithSpecifiedChain(String pdbId,
237 String pdbIdWithChainCode = "";
238 if (searchTerm.contains(";"))
240 String[] foundTerms = searchTerm.split(";");
241 for (String foundTerm : foundTerms)
243 if (foundTerm.contains(pdbId))
245 pdbIdWithChainCode = foundTerm;
249 else if (searchTerm.contains(pdbId))
251 pdbIdWithChainCode = searchTerm;
255 pdbIdWithChainCode = pdbId;
257 return pdbIdWithChainCode;
261 public FTSRestClientI getFTSRestClient()
263 return PDBFTSRestClient.getInstance();
267 public String getFTSFrameTitle()
269 return defaultFTSFrameTitle;
273 public boolean isPaginationEnabled()
279 public Map<String, Integer> getTempUserPrefs()
281 return tempUserPrefs;
285 public String getCacheKey()
287 return PDB_FTS_CACHE_KEY;
291 public String getAutosearchPreference()
293 return PDB_AUTOSEARCH;
297 protected void showHelp()
301 Help.showHelpWindow(HelpId.PdbFts);
302 } catch (HelpSetException e1)
304 e1.printStackTrace();