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

生物技术

开发平台:

C/C++

  1. /*
  2.  * ===========================================================================
  3.  * PRODUCTION $Log: conservation_colorer.hpp,v $
  4.  * PRODUCTION Revision 1000.1  2004/04/12 17:32:11  gouriano
  5.  * PRODUCTION PRODUCTION: UPGRADED [CATCHUP_003] Dev-tree R1.15
  6.  * PRODUCTION
  7.  * ===========================================================================
  8.  */
  9. /*  $Id: conservation_colorer.hpp,v 1000.1 2004/04/12 17:32:11 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:  Paul Thiessen
  35. *
  36. * File Description:
  37. *      Classes to color alignment blocks by sequence conservation
  38. *
  39. * ===========================================================================
  40. */
  41. #ifndef CN3D_CONSERVATION_COLORER__HPP
  42. #define CN3D_CONSERVATION_COLORER__HPP
  43. #include <corelib/ncbistl.hpp>
  44. #include <map>
  45. #include <vector>
  46. #include "vector_math.hpp"
  47. #include "cn3d_colors.hpp"
  48. BEGIN_SCOPE(Cn3D)
  49. class UngappedAlignedBlock;
  50. class BlockMultipleAlignment;
  51. class ConservationColorer
  52. {
  53. private:
  54.     const BlockMultipleAlignment *alignment;
  55.     typedef std::map < const UngappedAlignedBlock *, std::vector < int > > BlockMap;
  56.     BlockMap blocks;
  57.     typedef std::map < char, int > ColumnProfile;
  58.     typedef std::vector < ColumnProfile > AlignmentProfile;
  59.     AlignmentProfile alignmentProfile;
  60.     int GetProfileIndex(const UngappedAlignedBlock *block, int blockColumn) const
  61.         { return blocks.find(block)->second[blockColumn]; }
  62.     void GetProfileIndexAndResidue(const UngappedAlignedBlock *block, int blockColumn, int row,
  63.         int *profileIndex, char *residue) const;
  64.     int nColumns;
  65.     bool basicColorsCurrent, fitColorsCurrent;
  66.     void CalculateBasicConservationColors(void);
  67.     void CalculateFitConservationColors(void);
  68.     std::vector < bool > identities;
  69.     typedef std::vector < Vector > ColumnColors;
  70.     ColumnColors varietyColors, weightedVarietyColors, informationContentColors;
  71.     typedef std::map < char, Vector > ResidueColors;
  72.     typedef std::vector < ResidueColors > FitColors;
  73.     FitColors fitColors;
  74.     typedef std::map < const UngappedAlignedBlock * , std::vector < Vector > >  BlockFitColors;
  75.     BlockFitColors blockFitColors, blockZFitColors, blockRowFitColors;
  76. public:
  77.     ConservationColorer(const BlockMultipleAlignment *parent);
  78.     // add an aligned block to the profile
  79.     void AddBlock(const UngappedAlignedBlock *block);
  80.     // frees memory used by color storage (but keeps blocks around)
  81.     void FreeColors(void);
  82.     // clears everything, including alignment blocks
  83.     void Clear(void);
  84.     // color accessors
  85.     const Vector *GetIdentityColor(const UngappedAlignedBlock *block, int blockColumn)
  86.     {
  87.         CalculateBasicConservationColors();
  88.         return GlobalColors()->Get(Colors::eConservationMap,
  89.             (identities[GetProfileIndex(block, blockColumn)] ? Colors::nConservationMap - 1 : 0));
  90.     }
  91.     const Vector *GetVarietyColor(const UngappedAlignedBlock *block, int blockColumn)
  92.     {
  93.         CalculateBasicConservationColors();
  94.         return &(varietyColors[GetProfileIndex(block, blockColumn)]);
  95.     }
  96.     const Vector *GetWeightedVarietyColor(const UngappedAlignedBlock *block, int blockColumn)
  97.     {
  98.         CalculateBasicConservationColors();
  99.         return &(weightedVarietyColors[GetProfileIndex(block, blockColumn)]);
  100.     }
  101.     const Vector *GetInformationContentColor(const UngappedAlignedBlock *block, int blockColumn)
  102.     {
  103.         CalculateBasicConservationColors();
  104.         return &(informationContentColors[GetProfileIndex(block, blockColumn)]);
  105.     }
  106.     const Vector *GetFitColor(const UngappedAlignedBlock *block, int blockColumn, int row)
  107.     {
  108.         int profileIndex;
  109.         char residue;
  110.         CalculateFitConservationColors();
  111.         GetProfileIndexAndResidue(block, blockColumn, row, &profileIndex, &residue);
  112.         return &(fitColors[profileIndex].find(residue)->second);
  113.     }
  114.     const Vector *GetBlockFitColor(const UngappedAlignedBlock *block, int row)
  115.     {
  116.         CalculateFitConservationColors();
  117.         return &(blockFitColors.find(block)->second[row]);
  118.     }
  119.     const Vector *GetBlockZFitColor(const UngappedAlignedBlock *block, int row)
  120.     {
  121.         CalculateFitConservationColors();
  122.         return &(blockZFitColors.find(block)->second[row]);
  123.     }
  124.     const Vector *GetBlockRowFitColor(const UngappedAlignedBlock *block, int row)
  125.     {
  126.         CalculateFitConservationColors();
  127.         return &(blockRowFitColors.find(block)->second[row]);
  128.     }
  129. };
  130. END_SCOPE(Cn3D)
  131. #endif // CN3D_CONSERVATION_COLORER__HPP
  132. /*
  133. * ---------------------------------------------------------------------------
  134. * $Log: conservation_colorer.hpp,v $
  135. * Revision 1000.1  2004/04/12 17:32:11  gouriano
  136. * PRODUCTION: UPGRADED [CATCHUP_003] Dev-tree R1.15
  137. *
  138. * Revision 1.15  2004/02/19 17:04:54  thiessen
  139. * remove cn3d/ from include paths; add pragma to disable annoying msvc warning
  140. *
  141. * Revision 1.14  2003/02/13 18:31:39  thiessen
  142. * separate basic from fit color calculation
  143. *
  144. * Revision 1.13  2003/02/06 16:39:53  thiessen
  145. * add block row fit coloring
  146. *
  147. * Revision 1.12  2003/02/03 19:20:03  thiessen
  148. * format changes: move CVS Log to bottom of file, remove std:: from .cpp files, and use new diagnostic macros
  149. *
  150. * Revision 1.11  2003/01/30 14:00:23  thiessen
  151. * add Block Z Fit coloring
  152. *
  153. * Revision 1.10  2003/01/28 21:07:56  thiessen
  154. * add block fit coloring algorithm; tweak row dragging; fix style bug
  155. *
  156. * Revision 1.9  2001/08/24 00:40:57  thiessen
  157. * tweak conservation colors and opengl font handling
  158. *
  159. * Revision 1.8  2001/06/04 14:33:54  thiessen
  160. * add proximity sort; highlight sequence on browser launch
  161. *
  162. * Revision 1.7  2001/06/02 17:22:58  thiessen
  163. * fixes for GCC
  164. *
  165. * Revision 1.6  2001/03/22 00:32:36  thiessen
  166. * initial threading working (PSSM only); free color storage in undo stack
  167. *
  168. * Revision 1.5  2001/02/13 20:31:45  thiessen
  169. * add information content coloring
  170. *
  171. * Revision 1.4  2000/10/16 14:25:20  thiessen
  172. * working alignment fit coloring
  173. *
  174. * Revision 1.3  2000/10/05 18:34:35  thiessen
  175. * first working editing operation
  176. *
  177. * Revision 1.2  2000/10/04 17:40:45  thiessen
  178. * rearrange STL #includes
  179. *
  180. * Revision 1.1  2000/09/20 22:24:57  thiessen
  181. * working conservation coloring; split and center unaligned justification
  182. *
  183. */