func.h 2.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576
  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 <nvbio/basic/types.h>
  29. namespace nvbio {
  30. namespace bowtie2 {
  31. namespace cuda {
  32. ///@addtogroup nvBowtie
  33. ///@{
  34. struct SimpleFunc
  35. {
  36. enum Type { LinearFunc = 0, LogFunc = 1, SqrtFunc = 2 };
  37. NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
  38. SimpleFunc(const Type _type = LinearFunc, const float _k = 0.0f, const float _m = 1.0f) : type(_type), k(_k), m(_m) {}
  39. NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
  40. int32 operator() (const int32 x) const
  41. {
  42. return int32( k + m * (type == LogFunc ? logf(float(x)) :
  43. type == SqrtFunc ? sqrtf(float(x)) :
  44. float(x)) );
  45. }
  46. const char* type_string() const
  47. {
  48. return type == LogFunc ? "log" :
  49. type == SqrtFunc ? "sqrt" :
  50. "linear";
  51. }
  52. const char* type_symbol() const
  53. {
  54. return type == LogFunc ? "G" :
  55. type == SqrtFunc ? "S" :
  56. "L";
  57. }
  58. Type type;
  59. float k;
  60. float m;
  61. };
  62. ///@} // group nvBowtie
  63. } // namespace cuda
  64. } // namespace bowtie2
  65. } // namespace nvbio