Changeset 115 for cpp/frams/_demos/loader_test.cpp

Timestamp:

01/26/14 06:01:27 (11 years ago)

Author:

sz

Message:

loader_test can now extract raw genotype from a Framsticks file (*.gen/*.expt), which makes it useful in cooperation with other tools, for example:
loader_test walking.gen "Basic Quadruped" | theater -g -

File:

• cpp/frams/_demos/loader_test.cpp (modified) (3 diffs)

cpp/frams/_demos/loader_test.cpp

@@ -14 +14 @@
 int main(int argc,char*argv[])
 {
-puts("This example shows how to load genotypes from the standard Framsticks genotype file");
 if (argc<2)
         {
-        puts("(Please give the file name as command line argument)");
+        fprintf(stderr,"Arguments: filename [optional: genotype name or index (1-based)]\n"
+             "If a genotype is indicated (by providing the optional genotype identifier), the program will output the raw genotype, suitable for Framsticks Theater's genotype viewer mode. If the second optional argument is not given, the genotype names from the file will be listed.\n"
+             "Example: loader_test walking.gen \"Basic Quadruped\" | theater -g -"
+                );
         return 1;
         }
@@ -23 +25 @@
 long count=0,totalsize=0;
 MiniGenotypeLoader loader(argv[1]);
+const char* selected=(argc<3)?NULL:argv[2];
+int selected_index=(selected&&isdigit(selected[0]))?atol(selected):0;
 // using char* constructor (passing the file name to open)
 MiniGenotype *loaded;
@@ -30 +34 @@
         count++;
         totalsize+=loaded->genotype.len();
-        printf("%d. %s\t(%d characters)\n",count,(const char*)loaded->name,loaded->genotype.len());
+        if (selected)
+                {
+                if (selected_index)
+                        {
+                        if (selected_index!=count)
+                                continue;
+                        }
+                else
+                        {
+                        if (strcmp((const char*)loaded->name,selected))
+                                continue;
+                        }
+                puts((const char*)loaded->genotype);
+                return 0;
+                }
+        fprintf(stderr,"%d. %s\t(%d characters)\n",count,(const char*)loaded->name,loaded->genotype.len());
         }
 // the loop repeats until loaded==NULL, which could be beacause of error
 if (loader.getStatus()==MiniGenotypeLoader::OnError)
-        printf("Error: %s",(const char*)loader.getError());
+        fprintf(stderr,"Error: %s",(const char*)loader.getError());
 // (otherwise it was the end of the file)
-
-printf("\ntotal: %d items, %d characters\n",count,totalsize);
-return 0;
+if (selected)
+        {
+        fprintf(stderr,"genotype %s not found in %s\n",selected,argv[1]);
+        return 2;
+        }
+else
+        {
+        fprintf(stderr,"\ntotal: %d items, %d characters\n",count,totalsize);
+        return 0;
+        }
 }