options.h 3.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485
  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. using namespace nvbio;
  30. struct runtime_options
  31. {
  32. // genome file name or shared memory handle name
  33. const char *genome_file_name;
  34. // input reads file name
  35. const char *input_file_name;
  36. // output alignment file name
  37. const char *output_file_name;
  38. // whether to allow using mmap() to load the genome
  39. bool genome_use_mmap;
  40. // input read batch size
  41. uint64 batch_size;
  42. // MEM search options
  43. uint32 min_intv; // min and max interval sizes for MEM search
  44. uint32 max_intv;
  45. uint32 min_seed_len; // minimum read span, MEMs that span less than this many bps will be dropped
  46. uint32 mems_batch; // number of MEMs to locate at once
  47. uint32 w;
  48. uint32 max_chain_gap;
  49. uint32 split_len;
  50. uint32 split_width;
  51. float mask_level;
  52. float chain_drop_ratio;
  53. NVBIO_HOST_DEVICE
  54. runtime_options()
  55. {
  56. genome_file_name = NULL;
  57. input_file_name = NULL;
  58. output_file_name = NULL;
  59. // default options
  60. genome_use_mmap = true;
  61. batch_size = 256 * 1024;
  62. min_intv = 1;
  63. max_intv = 10000;
  64. min_seed_len = 19;
  65. mems_batch = 16 * 1024 * 1024;
  66. split_len = 28;
  67. split_width = 10;
  68. mask_level = 0.5f;
  69. chain_drop_ratio = 0.5f;
  70. w = 100;
  71. max_chain_gap = 10000;
  72. };
  73. };
  74. extern struct runtime_options command_line_options;
  75. void parse_command_line(int argc, char **argv);