traceback.h 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220
  1. /*
  2. * nvbio
  3. * Copyright (c) 2011-2014, NVIDIA CORPORATION. All rights reserved.
  4. *
  5. * Redistribution and use in source and binary forms, with or without
  6. * modification, are permitted provided that the following conditions are met:
  7. * * Redistributions of source code must retain the above copyright
  8. * notice, this list of conditions and the following disclaimer.
  9. * * Redistributions in binary form must reproduce the above copyright
  10. * notice, this list of conditions and the following disclaimer in the
  11. * documentation and/or other materials provided with the distribution.
  12. * * Neither the name of the NVIDIA CORPORATION nor the
  13. * names of its contributors may be used to endorse or promote products
  14. * derived from this software without specific prior written permission.
  15. *
  16. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
  17. * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  18. * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  19. * DISCLAIMED. IN NO EVENT SHALL NVIDIA CORPORATION BE LIABLE FOR ANY
  20. * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  21. * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  22. * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  23. * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  24. * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  25. * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  26. */
  27. #pragma once
  28. #include <nvBowtie/bowtie2/cuda/defs.h>
  29. #include <nvBowtie/bowtie2/cuda/scoring.h>
  30. namespace nvbio {
  31. namespace io {
  32. struct Alignment;
  33. struct BestAlignments;
  34. } // namespace io
  35. } // namespace nvbio
  36. namespace nvbio {
  37. namespace bowtie2 {
  38. namespace cuda {
  39. struct ParamsPOD;
  40. template <typename ScoringScheme> struct TracebackPipelineState;
  41. template <typename ScoringScheme> struct AllMappingPipelineState;
  42. ///@addtogroup nvBowtie
  43. ///@{
  44. ///@addtogroup Traceback
  45. ///@{
  46. ///
  47. /// execute a batch of banded-alignment traceback calculations
  48. ///
  49. void banded_traceback_best(
  50. const uint32 aln_idx,
  51. const uint32 count,
  52. const uint32* idx,
  53. io::Alignment* best_data,
  54. const uint32 best_stride,
  55. const uint32 band_len,
  56. const TracebackPipelineState<EditDistanceScoringScheme>& pipeline,
  57. const ParamsPOD& params);
  58. ///
  59. /// execute a batch of banded-alignment traceback calculations
  60. ///
  61. void banded_traceback_best(
  62. const uint32 aln_idx,
  63. const uint32 count,
  64. const uint32* idx,
  65. io::Alignment* best_data,
  66. const uint32 best_stride,
  67. const uint32 band_len,
  68. const TracebackPipelineState<SmithWatermanScoringScheme<> >& pipeline,
  69. const ParamsPOD& params);
  70. ///
  71. /// execute a batch of opposite alignment traceback calculations
  72. ///
  73. void opposite_traceback_best(
  74. const uint32 aln_idx,
  75. const uint32 count,
  76. const uint32* idx,
  77. io::Alignment* best_data,
  78. const uint32 best_stride,
  79. const TracebackPipelineState<EditDistanceScoringScheme>& pipeline,
  80. const ParamsPOD& params);
  81. ///
  82. /// execute a batch of opposite alignment traceback calculations
  83. ///
  84. void opposite_traceback_best(
  85. const uint32 aln_idx,
  86. const uint32 count,
  87. const uint32* idx,
  88. io::Alignment* best_data,
  89. const uint32 best_stride,
  90. const TracebackPipelineState<SmithWatermanScoringScheme<> >& pipeline,
  91. const ParamsPOD& params);
  92. ///
  93. /// execute a batch of banded-alignment traceback calculations
  94. ///
  95. void banded_traceback_all(
  96. const uint32 count,
  97. const uint32* idx,
  98. const uint32 buffer_offset,
  99. const uint32 buffer_size,
  100. io::Alignment* alignments,
  101. const uint32 band_len,
  102. const AllMappingPipelineState<EditDistanceScoringScheme>& pipeline,
  103. const ParamsPOD& params);
  104. ///
  105. /// execute a batch of banded-alignment traceback calculations
  106. ///
  107. void banded_traceback_all(
  108. const uint32 count,
  109. const uint32* idx,
  110. const uint32 buffer_offset,
  111. const uint32 buffer_size,
  112. io::Alignment* alignments,
  113. const uint32 band_len,
  114. const AllMappingPipelineState<SmithWatermanScoringScheme<> >& pipeline,
  115. const ParamsPOD& params);
  116. ///
  117. /// finish a batch of alignment calculations
  118. ///
  119. void finish_alignment_best(
  120. const uint32 aln_idx,
  121. const uint32 count,
  122. const uint32* idx,
  123. io::Alignment* best_data,
  124. const uint32 best_stride,
  125. const uint32 band_len,
  126. const TracebackPipelineState<EditDistanceScoringScheme>& pipeline,
  127. const SmithWatermanScoringScheme<> scoring_scheme,
  128. const ParamsPOD& params);
  129. ///
  130. /// finish a batch of alignment calculations
  131. ///
  132. void finish_alignment_best(
  133. const uint32 aln_idx,
  134. const uint32 count,
  135. const uint32* idx,
  136. io::Alignment* best_data,
  137. const uint32 best_stride,
  138. const uint32 band_len,
  139. const TracebackPipelineState<SmithWatermanScoringScheme<> >& pipeline,
  140. const SmithWatermanScoringScheme<> scoring_scheme,
  141. const ParamsPOD& params);
  142. ///
  143. /// finish a batch of opposite alignment calculations
  144. ///
  145. void finish_opposite_alignment_best(
  146. const uint32 aln_idx,
  147. const uint32 count,
  148. const uint32* idx,
  149. io::Alignment* best_data,
  150. const uint32 best_stride,
  151. const uint32 band_len,
  152. const TracebackPipelineState<EditDistanceScoringScheme>& pipeline,
  153. const SmithWatermanScoringScheme<> scoring_scheme,
  154. const ParamsPOD& params);
  155. ///
  156. /// finish a batch of opposite alignment calculations
  157. ///
  158. void finish_opposite_alignment_best(
  159. const uint32 aln_idx,
  160. const uint32 count,
  161. const uint32* idx,
  162. io::Alignment* best_data,
  163. const uint32 best_stride,
  164. const uint32 band_len,
  165. const TracebackPipelineState<SmithWatermanScoringScheme<> >& pipeline,
  166. const SmithWatermanScoringScheme<> scoring_scheme,
  167. const ParamsPOD& params);
  168. ///
  169. /// finish a batch of alignment calculations, all-mapping
  170. ///
  171. void finish_alignment_all(
  172. const uint32 count,
  173. const uint32* idx,
  174. const uint32 buffer_offset,
  175. const uint32 buffer_size,
  176. io::Alignment* alignments,
  177. const uint32 band_len,
  178. const AllMappingPipelineState<EditDistanceScoringScheme>& pipeline,
  179. const SmithWatermanScoringScheme<> scoring_scheme,
  180. const ParamsPOD& params);
  181. ///
  182. /// finish a batch of alignment calculations, all-mapping
  183. ///
  184. void finish_alignment_all(
  185. const uint32 count,
  186. const uint32* idx,
  187. const uint32 buffer_offset,
  188. const uint32 buffer_size,
  189. io::Alignment* alignments,
  190. const uint32 band_len,
  191. const AllMappingPipelineState<SmithWatermanScoringScheme<> >& pipeline,
  192. const SmithWatermanScoringScheme<> scoring_scheme,
  193. const ParamsPOD& params);
  194. ///@} // group Traceback
  195. ///@} // group nvBowtie
  196. } // namespace cuda
  197. } // namespace bowtie2
  198. } // namespace nvbio