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

生物技术

开发平台:

C/C++

  1. /*
  2.  * ===========================================================================
  3.  * PRODUCTION $Log: test_getalignments.cpp,v $
  4.  * PRODUCTION Revision 1000.0  2004/06/01 21:22:07  gouriano
  5.  * PRODUCTION PRODUCTION: IMPORTED [GCC34_MSVC7] Dev-tree R1.3
  6.  * PRODUCTION
  7.  * ===========================================================================
  8.  */
  9. /*  $Id: test_getalignments.cpp,v 1000.0 2004/06/01 21:22:07 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.  * Authors:  Mike DiCuccio
  35.  *
  36.  * File Description:
  37.  *    Test application for plugins
  38.  */
  39. #include <ncbi_pch.hpp>
  40. #include <corelib/ncbiapp.hpp>
  41. #include <corelib/ncbiargs.hpp>
  42. #include <corelib/ncbienv.hpp>
  43. #include <corelib/ncbireg.hpp>
  44. #include <gui/objutils/utils.hpp>
  45. #include <gui/objutils/mate_pair.hpp>
  46. #include <gui/objutils/alignment.hpp>
  47. #include <gui/objutils/pw_alignment.hpp>
  48. #include <gui/objutils/label.hpp>
  49. #include <objmgr/object_manager.hpp>
  50. #include <objmgr/scope.hpp>
  51. #include <objtools/data_loaders/genbank/gbloader.hpp>
  52. #include <objects/seqalign/Seq_align.hpp>
  53. #include <serial/objostr.hpp>
  54. #include <serial/serial.hpp>
  55. USING_NCBI_SCOPE;
  56. USING_SCOPE(objects);
  57. class CTestGetAlignments : public CNcbiApplication
  58. {
  59. public:
  60.     virtual void Init(void);
  61.     virtual int  Run(void);
  62.     virtual void Exit(void);
  63. };
  64. void CTestGetAlignments::Init(void)
  65. {
  66.     // Create command-line argument descriptions class
  67.     auto_ptr<CArgDescriptions> arg_desc(new CArgDescriptions);
  68.     // Specify USAGE context
  69.     arg_desc->SetUsageContext(GetArguments().GetProgramBasename(),
  70.                               "Test: CSeqUtils::GetAlignments()");
  71.     arg_desc->AddKey("acc", "Accession",
  72.                      "Accession to test",
  73.                      CArgDescriptions::eString);
  74.     // Setup arg.descriptions for this application
  75.     SetupArgDescriptions(arg_desc.release());
  76. }
  77. int CTestGetAlignments::Run(void)
  78. {
  79.     // Get arguments
  80.     CArgs args = GetArgs();
  81.     string id_str = args["acc"].AsString();
  82.     CSeq_id id(id_str);
  83.     if ( id.Which() == CSeq_id::e_not_set) {
  84.         LOG_POST(Fatal << "don't understand accession: " << id_str);
  85.     }
  86.     CRef<CObjectManager> obj_mgr(new CObjectManager());
  87.     obj_mgr->RegisterDataLoader(*new CGBDataLoader(),
  88.                                 CObjectManager::eDefault);
  89.     CRef<CScope> scope(new CScope(*obj_mgr));
  90.     scope->AddDefaults();
  91.     CBioseq_Handle handle = scope->GetBioseqHandle(id);
  92.     if ( !handle ) {
  93.         LOG_POST(Fatal << "failed to retrieve sequence for: " << id_str);
  94.     }
  95.     CLayoutEngine::TObjects objs;
  96.     CSeqUtils::GetAlignments(handle,
  97.                              TSeqRange(0, handle.GetBioseqLength()),
  98.                              objs);
  99.     cout << "got " << objs.size() << " alignments:" << endl;
  100.     auto_ptr<CObjectOStream> os(CObjectOStream::Open(eSerial_AsnText, cout));
  101.     ITERATE (CLayoutEngine::TObjects, iter, objs) {
  102.         const CLayoutObject& obj = **iter;
  103.         if (dynamic_cast<const CLayoutPWAlign*>(&obj)) {
  104.             const CLayoutPWAlign& pwalign =
  105.                 dynamic_cast<const CLayoutPWAlign&>(obj);
  106.             const CSeq_align& align = pwalign.GetAlignment();
  107.             string label;
  108.             CLabel::GetLabel(align, &label, CLabel::eDefault, scope);
  109.             cout << string(72, '-') << endl;
  110.             cout << "pairwise alignment: " << label << endl;
  111.             *os << align;
  112.         } else if (dynamic_cast<const CLayoutMatePair*>(&obj)) {
  113.             const CLayoutMatePair& mpalign =
  114.                 dynamic_cast<const CLayoutMatePair&>(obj);
  115.             const CLayoutMatePair::TAlignList& aligns =
  116.                 mpalign.GetSeqAligns();
  117.             cout << string(72, '-') << endl;
  118.             cout << "mate pair alignment: " << aligns.size()
  119.                 << " alignments" << endl;
  120.             cout << "mate pair library: " << mpalign.GetLibraryId() << endl;
  121.             cout << "mate pair error code: " << mpalign.GetError() << endl;
  122.             ITERATE (CLayoutMatePair::TAlignList, iter, aligns) {
  123.                 *os << (*iter)->GetAlignment();
  124.             }
  125.         } else if (dynamic_cast<const CLayoutAlign*>(&obj)) {
  126.             const CLayoutAlign& aobj =
  127.                 dynamic_cast<const CLayoutAlign&>(obj);
  128.             const CSeq_align& align = aobj.GetAlignment();
  129.             string label;
  130.             CLabel::GetLabel(align, &label, CLabel::eDefault, scope);
  131.             cout << string(72, '-') << endl;
  132.             cout << "pairwise alignment: " << label << endl;
  133.             *os << align;
  134.         } else {
  135.             LOG_POST(Error << "unhandled alignment object");
  136.         }
  137.     }
  138.     return 0;
  139. }
  140. /////////////////////////////////////////////////////////////////////////////
  141. //  Cleanup
  142. void CTestGetAlignments::Exit(void)
  143. {
  144.     SetDiagStream(0);
  145. }
  146. /////////////////////////////////////////////////////////////////////////////
  147. //  MAIN
  148. int main(int argc, const char* argv[])
  149. {
  150.     // Execute main application function
  151.     return CTestGetAlignments().AppMain(argc, argv, 0, eDS_Default, 0);
  152. }
  153. /*
  154.  * ===========================================================================
  155.  * $Log: test_getalignments.cpp,v $
  156.  * Revision 1000.0  2004/06/01 21:22:07  gouriano
  157.  * PRODUCTION: IMPORTED [GCC34_MSVC7] Dev-tree R1.3
  158.  *
  159.  * Revision 1.3  2004/05/21 22:27:44  gorelenk
  160.  * Added PCH ncbi_pch.hpp
  161.  *
  162.  * Revision 1.2  2004/05/10 18:23:53  dicuccio
  163.  * Dump pairwise alignments.  Added tracing of mate pair specifics
  164.  *
  165.  * Revision 1.1  2004/05/09 23:59:55  dicuccio
  166.  * Standardized names of demo and test applications.  Cleaned up LIB
  167.  * specifications to eliminate unneeded libraries after splitting libgui_utils
  168.  * into libgui_utils and libgui_objutils
  169.  *
  170.  * ===========================================================================
  171.  */