checksums.h 2.9 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182
  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/seed_hit_deque_array.h>
  30. #include <nvbio/basic/types.h>
  31. #include <crc/crc.h>
  32. #include <thrust/device_vector.h>
  33. #include <thrust/sort.h>
  34. namespace nvbio {
  35. namespace bowtie2 {
  36. namespace cuda {
  37. struct SeedHit;
  38. // compute a checksum of a device sequence
  39. //
  40. template <typename Iterator>
  41. uint64 device_checksum(
  42. Iterator begin,
  43. Iterator end)
  44. {
  45. typedef typename thrust::iterator_traits<Iterator>::value_type value_type;
  46. const uint32 n = uint32( end - begin );
  47. // copy values to a temporary vector
  48. thrust::device_vector<value_type> debug_copy_dvec( n );
  49. thrust::copy(
  50. begin,
  51. end,
  52. debug_copy_dvec.begin() );
  53. // sort temporary vector
  54. thrust::sort( debug_copy_dvec.begin(), debug_copy_dvec.end() );
  55. // copy to the host
  56. thrust::host_vector<value_type> debug_copy_hvec( debug_copy_dvec );
  57. // compute a crc
  58. const char* ptr = (const char*)thrust::raw_pointer_cast( &debug_copy_hvec.front() );
  59. return crcCalc( ptr, sizeof(value_type)*n );
  60. }
  61. // compute a checksum for the returned SA ranges
  62. //
  63. void hits_checksum(
  64. const uint32 n_reads,
  65. SeedHitDequeArray& hit_deques,
  66. uint64& crc,
  67. uint64& sum);
  68. } // namespace cuda
  69. } // namespace bowtie2
  70. } // namespace nvbio