1 files changed, 13 insertions, 2 deletions

diff --git a/fits.cpp b/fits.cpp
index 682886a..d8676bb 100644
--- a/fits.cpp
+++ b/fits.cpp
@@ -11,9 +11,11 @@ int main(int argc, char* argv[]) {
   Real iniVar = 0.0;
   long unsigned maxSteps = 1e12;
 
+  bool cosine = false;
+
   int opt;
 
-  while ((opt = getopt(argc, argv, "M:s:S:B:i:N:")) != -1) {
+  while ((opt = getopt(argc, argv, "M:s:S:B:i:N:c:")) != -1) {
     switch (opt) {
     case 'N':
       Nend = (unsigned)atof(optarg);
@@ -33,6 +35,9 @@ int main(int argc, char* argv[]) {
     case 'i':
       iniVar = atof(optarg);
       break;
+    case 'c':
+      cosine = true;
+      break;
     default:
       exit(1);
     }
@@ -40,7 +45,13 @@ int main(int argc, char* argv[]) {
 
   Rng r;
 
-  Data data = generateData([](Real x) {return std::sin(2 * M_PI * x);}, M, σ, r);
+  Data data;
+
+  if (cosine) {
+    data = generateData([](Real x) {return std::cos(2 * M_PI * x);}, M, σ, r);
+  } else {
+    data = generateData([](Real x) {return std::sin(2 * M_PI * x);}, M, σ, r);
+  }
 
   std::cout << std::setprecision(15);