persist.h 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293
  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_queues.h>
  30. #include <nvbio/basic/types.h>
  31. #include <thrust/device_vector.h>
  32. #include <thrust/sort.h>
  33. #include <string>
  34. namespace nvbio {
  35. namespace bowtie2 {
  36. namespace cuda {
  37. struct SeedHit;
  38. struct HitQueue;
  39. struct SeedHitDequeArray;
  40. // clear the persisting files
  41. //
  42. void persist_clear(const std::string& file_name);
  43. // persist a set of hits
  44. //
  45. void persist_hits(
  46. const std::string& file_name,
  47. const char* name,
  48. const uint32 anchor,
  49. const uint32 count,
  50. const SeedHitDequeArray& hit_deques);
  51. // persist a set of reads
  52. //
  53. void persist_reads(
  54. const std::string& file_name,
  55. const char* name,
  56. const uint32 anchor,
  57. const uint32 count,
  58. const thrust::device_vector<uint32>::iterator iterator);
  59. // persist a set of selected hits
  60. //
  61. void persist_selection(
  62. const std::string& file_name,
  63. const char* name,
  64. const uint32 anchor,
  65. const uint32 read_count,
  66. const packed_read* read_infos_dptr,
  67. const uint32 n_multi,
  68. const uint32 hits_queue_size,
  69. const ReadHitsIndex& hits_index,
  70. const HitQueues& hits_queue);
  71. // persist a set of scores
  72. //
  73. void persist_scores(
  74. const std::string& file_name,
  75. const char* name,
  76. const uint32 anchor,
  77. const uint32 read_count,
  78. const uint32 n_multi,
  79. const uint32 hits_queue_size,
  80. const ScoringQueues& scoring_queues);
  81. } // namespace cuda
  82. } // namespace bowtie2
  83. } // namespace nvbio