entrez2_client.cpp
上传用户:yhdzpy8989
上传日期:2007-06-13
资源大小:13604k
文件大小:8k
源码类别:

生物技术

开发平台:

C/C++

  1. /*
  2.  * ===========================================================================
  3.  * PRODUCTION $Log: entrez2_client.cpp,v $
  4.  * PRODUCTION Revision 1000.2  2004/06/01 19:32:08  gouriano
  5.  * PRODUCTION PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.9
  6.  * PRODUCTION
  7.  * ===========================================================================
  8.  */
  9. /* $Id: entrez2_client.cpp,v 1000.2 2004/06/01 19:32:08 gouriano Exp $
  10.  * ===========================================================================
  11.  *
  12.  *                            PUBLIC DOMAIN NOTICE
  13.  *               National Center for Biotechnology Information
  14.  *
  15.  *  This software/database is a "United States Government Work" under the
  16.  *  terms of the United States Copyright Act.  It was written as part of
  17.  *  the author's official duties as a United States Government employee and
  18.  *  thus cannot be copyrighted.  This software/database is freely available
  19.  *  to the public for use. The National Library of Medicine and the U.S.
  20.  *  Government have not placed any restriction on its use or reproduction.
  21.  *
  22.  *  Although all reasonable efforts have been taken to ensure the accuracy
  23.  *  and reliability of the software and data, the NLM and the U.S.
  24.  *  Government do not and cannot warrant the performance or results that
  25.  *  may be obtained by using this software or data. The NLM and the U.S.
  26.  *  Government disclaim all warranties, express or implied, including
  27.  *  warranties of performance, merchantability or fitness for any particular
  28.  *  purpose.
  29.  *
  30.  *  Please cite the author in any work or product based on this material.
  31.  *
  32.  * ===========================================================================
  33.  *
  34.  * Author:  Josh Cherry
  35.  *
  36.  * File Description:
  37.  *   Entrez2 network client
  38.  *
  39.  * Remark:
  40.  *   This code was originally generated by application DATATOOL
  41.  *   using specifications from the data definition file
  42.  *   'entrez2.asn'.
  43.  */
  44. // standard includes
  45. // generated includes
  46. #include <ncbi_pch.hpp>
  47. #include <objects/entrez2/entrez2_client.hpp>
  48. #include <objects/entrez2/Entrez2_db_id.hpp>
  49. #include <objects/entrez2/Entrez2_link_id.hpp>
  50. #include <objects/entrez2/Entrez2_db_id.hpp>
  51. #include <objects/entrez2/Entrez2_id.hpp>
  52. #include <objects/entrez2/Entrez2_get_links.hpp>
  53. #include <objects/entrez2/Entrez2_id_list.hpp>
  54. #include <objects/entrez2/Entrez2_boolean_element.hpp>
  55. #include <objects/entrez2/Entrez2_boolean_exp.hpp>
  56. #include <objects/entrez2/Entrez2_eval_boolean.hpp>
  57. #include <objects/entrez2/Entrez2_boolean_reply.hpp>
  58. // generated classes
  59. BEGIN_NCBI_SCOPE
  60. BEGIN_objects_SCOPE // namespace ncbi::objects::
  61. // destructor
  62. CEntrez2Client::~CEntrez2Client(void)
  63. {
  64. }
  65. /// A simplified interface for getting neighbors (links)
  66. /// This form just yields a vector of UIDs
  67. void CEntrez2Client::GetNeighbors(int query_uid, const string& db,
  68.                                   const string& link_type,
  69.                                   vector<int>& neighbor_uids)
  70. {
  71.     vector<int> uids;
  72.     uids.push_back(query_uid);
  73.     GetNeighbors(uids, db, link_type, neighbor_uids);
  74. }
  75. /// This form just yields a vector of UIDs
  76. void CEntrez2Client::GetNeighbors(const vector<int>& query_uids,
  77.                                   const string& db,
  78.                                   const string& link_type,
  79.                                   vector<int>& neighbor_uids)
  80. {
  81.     // first retrieve the link_set
  82.     CRef<CEntrez2_link_set> link_set;
  83.     link_set = GetNeighbors(query_uids, db, link_type);
  84.     
  85.     // then extract the UIDs
  86.     CEntrez2_id_list::TConstUidIterator it
  87.         = link_set->GetIds().GetConstUidIterator();
  88.     for ( ;  !it.AtEnd();  ++it) {
  89.         neighbor_uids.push_back(*it);
  90.     }
  91. }
  92. /// This form returns the entire CEntrez2_link_set object,
  93. /// which includes scores.
  94. CRef<CEntrez2_link_set> CEntrez2Client::GetNeighbors(int query_uid,
  95.                                                      const string& db,
  96.                                                      const string& link_type)
  97. {
  98.     vector<int> uids;
  99.     uids.push_back(query_uid);
  100.     return GetNeighbors(uids, db, link_type);
  101. }
  102. /// This form returns the entire CEntrez2_link_set object,
  103. /// which includes scores.
  104. CRef<CEntrez2_link_set>
  105. CEntrez2Client::GetNeighbors(const vector<int>& query_uids,
  106.                              const string& db,
  107.                              const string& link_type)
  108. {
  109.     CEntrez2_id_list uids;
  110.     uids.SetDb() = CEntrez2_db_id(db);
  111.     uids.AssignUids(query_uids);
  112.     
  113.     CEntrez2_get_links gl;
  114.     gl.SetUids(uids);
  115.     gl.SetLinktype().Set(db + "_" + link_type);
  116.     CRef<CEntrez2_link_set> link_set = AskGet_links(gl);
  117.     return link_set;
  118. }
  119. /// Retrieve counts of the various types of neighbors available
  120. CRef<CEntrez2_link_count_list>
  121. CEntrez2Client::GetNeighborCounts(int query_uid,
  122.                                   const string& db)
  123. {
  124.     CEntrez2_id uid;
  125.     uid.SetDb() = CEntrez2_db_id(db);
  126.     uid.SetUid(query_uid);
  127.     return AskGet_link_counts(uid);
  128. }
  129. /// Query a db with a string, returning uids as integers
  130. void CEntrez2Client::Query(const string& query, const string& db,
  131.                            vector<int>& result_uids)
  132. {
  133.     CRef<CEntrez2_boolean_element> bel(new CEntrez2_boolean_element);
  134.     bel->SetStr(query);
  135.     CEntrez2_boolean_exp bexp;
  136.     bexp.SetDb().Set(db);
  137.     bexp.SetExp().push_back(bel);
  138.     CEntrez2_eval_boolean req;
  139.     req.SetReturn_UIDs(true);
  140.     req.SetQuery(bexp);
  141.     CRef<CEntrez2_boolean_reply> reply = AskEval_boolean(req);
  142.     
  143.     // now extract the UIDs
  144.     if (!reply->GetUids().CanGetUids ()) {
  145.         // this happens when no matches were found
  146.         return;
  147.     }
  148.     for (CEntrez2_id_list::TConstUidIterator it
  149.              = reply->GetUids().GetConstUidIterator();  
  150.          !it.AtEnd();  ++it) {
  151.         result_uids.push_back(*it);
  152.     }
  153. }
  154. /// Given some uids, a database, and an entrez query string,
  155. /// determine which of these uids match the query string
  156. void CEntrez2Client::FilterIds(const vector<int>& query_uids, const string& db,
  157.                                const string& query_string,
  158.                                vector<int>& result_uids)
  159. {
  160.     if (query_uids.empty()) {
  161.         return;
  162.     }
  163.     string uids;
  164.     ITERATE (vector<int>, uid, query_uids) {
  165.         if ( !uids.empty() ) {
  166.             uids += " OR ";
  167.         }
  168.         uids += NStr::IntToString(*uid) + "[UID]";
  169.     }
  170.     string whole_query = "(" + query_string + ") AND (" + uids + ")";
  171.     Query(whole_query, db, result_uids);
  172. }
  173. END_objects_SCOPE // namespace ncbi::objects::
  174. END_NCBI_SCOPE
  175. /*
  176. * ===========================================================================
  177. *
  178. * $Log: entrez2_client.cpp,v $
  179. * Revision 1000.2  2004/06/01 19:32:08  gouriano
  180. * PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.9
  181. *
  182. * Revision 1.9  2004/05/19 17:20:02  gorelenk
  183. * Added include of PCH - ncbi_pch.hpp
  184. *
  185. * Revision 1.8  2004/03/11 17:29:07  dicuccio
  186. * Added APIs for retrieving neighbors for multiple UIDs.  Changed how link type
  187. * is interpreted - both link and db are databases
  188. *
  189. * Revision 1.7  2004/01/20 05:36:56  jcherry
  190. * Added missing '&' on argument to FilterIds (pass by reference)
  191. *
  192. * Revision 1.6  2003/10/21 13:48:49  grichenk
  193. * Redesigned type aliases in serialization library.
  194. * Fixed the code (removed CRef-s, added explicit
  195. * initializers etc.)
  196. *
  197. * Revision 1.5  2003/10/17 16:44:01  jcherry
  198. * Fixed behavior when Query finds nothing
  199. *
  200. * Revision 1.4  2003/10/16 20:10:23  jcherry
  201. * Added some simplified interfaces for querying
  202. *
  203. * Revision 1.3  2003/10/08 19:56:44  jcherry
  204. * OK, we're back
  205. *
  206. * Revision 1.1  2003/10/08 19:18:01  jcherry
  207. * Added a simplified interface for getting neighbors
  208. *
  209. *
  210. * ===========================================================================
  211. */
  212. /* Original file checksum: lines: 64, chars: 1896, CRC32: cd6a8df4 */