add some error checking

1 files changed, 9 insertions, 4 deletions

diff --git a/src/c/quikr_train.c b/src/c/quikr_train.c
index f19a554..1585629 100644
--- a/src/c/quikr_train.c
+++ b/src/c/quikr_train.c
@@ -22,7 +22,7 @@ int main(int argc, char **argv) {
 
 
   int c;
-  int kmer = 0;
+  int kmer = 6;
 
   char *fasta_file = NULL;
   char *output_file = NULL;
@@ -88,15 +88,17 @@ int main(int argc, char **argv) {
     exit(EXIT_FAILURE);
   }
 
-  if(kmer == 0)
-    kmer = 6;
-
   if(verbose) {
     printf("kmer size: %d\n", kmer);
     printf("fasta file: %s\n", fasta_file);
     printf("output file: %s\n", output_file);
   }
 
+  if(access (fasta_file, F_OK) == -1) {
+    fprintf(stderr, "Error: could not find %s\n", fasta_file);
+    exit(EXIT_FAILURE);
+  }
+
   if(strcmp(&output_file[strlen(output_file) - 3], ".gz") != 0) {
     char *temp = malloc(sizeof(strlen(output_file) + 4));
     sprintf(temp, "%s.gz", output_file);
@@ -107,6 +109,9 @@ int main(int argc, char **argv) {
   // 4 ^ Kmer gives us the width, or the number of permutations of ACTG with kmer length
   int width = pow(4, kmer);
   int sequences = count_sequences(fasta_file);
+  if(sequences == 0) {
+    fprintf(stderr, "Error: %s contains 0 fasta sequences\n", fasta_file);
+  }
 
   if(verbose)
     printf("sequences: %d\nwidth: %d\n", sequences, width);