added support for synthesizing random distance maps to expand to (for

Author: brf2

mris_expand/mris_expand.cpp

@@ -36,6 +36,7 @@
 #include "const.h"
 #include "timer.h"
 #include "version.h"
+#include "randomfields.h"
 
 int main(int argc, char *argv[]) ;
 static int get_option(int argc, char *argv[]) ;
@@ -47,7 +48,12 @@ static int use_thickness = 0 ;
 static int nsurfaces = 1 ;
 static const char *thickness_name = "thickness" ;
 static const char *pial_name = "pial" ;
-static const char *tmap_fname = NULL ;
+static char *tmap_fname = NULL ;
+static char *tmap_write_fname = NULL ;
+static float tmap_std = 0.0 ;
+static float tmap_min = .25 ;
+static float tmap_max = .75 ;
+static int tmap_avgs = 0 ;
 static int nbrs = 2 ;
 
 static char *orig_name = NULL ;
@@ -173,12 +179,50 @@ main(int argc, char *argv[])
     parms.mri_dtrans = MRIread(tmap_fname);
     if (parms.mri_dtrans == NULL)
       ErrorExit(ERROR_NOFILE,
-                "%s: could not tmap vertex percentages from %s\n", Progname, tmap_fname) ;
+                "%s: could not read tmap vertex percentages from %s\n", Progname, tmap_fname) ;
     if (parms.mri_dtrans->width != mris->nvertices)
       ErrorExit(ERROR_NOFILE,
-                "%s: could not tmap width %d != mris->nvertices %d in %s\n", 
+                "%s: tmap width %d != mris->nvertices %d in %s\n", 
 		Progname, parms.mri_dtrans->width, mris->nvertices, tmap_fname) ;
-    printf("setting vno distance to %f\n", MRIgetVoxVal(parms.mri_dtrans, 0, 0, 0, 0));
+  }
+  else if (!FZERO(tmap_std))    // create a random map of distances
+  {
+    RFS    *rfs = RFspecInit(0, NULL);
+    int    vno ;
+
+    parms.mri_dtrans = MRIalloc(mris->nvertices, 1, 1, MRI_FLOAT) ;
+    if (parms.mri_dtrans == NULL)
+      ErrorExit(ERROR_NOFILE,
+                "%s: could not allocate tmap vertex percentages from %s\n", Progname, tmap_fname) ;
+
+    printf("creating random tmap distances\n") ;
+    rfs->name = strcpyalloc("gaussian");
+    rfs->params[0] = mm_out ;         // mean is halfway through the ribbon
+    rfs->params[1] = tmap_std;     // std
+
+    for (vno = 0 ; vno < mris->nvertices ; vno++)
+      mris->vertices[vno].val = RFdrawVal(rfs);
+
+    MRISaverageVals(mris, tmap_avgs) ;
+    for (vno = 0 ; vno < mris->nvertices ; vno++)
+    {
+      float dist = mris->vertices[vno].val ;
+
+      dist = MIN(MAX(dist, tmap_min), tmap_max) ;
+      if (mris->vertices[vno].ripflag)
+	dist = 0 ;
+
+      if (vno == Gdiag_no)
+	printf("vno %d: val %f, ripflag %d\n", vno, dist, mris->vertices[vno].ripflag) ;
+
+      mris->vertices[vno].val = dist ;
+      MRIsetVoxVal(parms.mri_dtrans, vno, 0, 0, 0, dist) ;
+    }
+    if (tmap_write_fname != NULL)
+    {
+      printf("writing random tmap to %s\n", tmap_write_fname) ;
+      MRISwriteValues(mris, tmap_write_fname) ;
+    }
   }
   else
     parms.mri_dtrans = NULL ;
@@ -329,6 +373,12 @@ get_option(int argc, char *argv[])
     nargs = 1 ;
     fprintf(stderr, "l_surf_repulse = %2.3f\n", parms.l_surf_repulse) ;
   }
+  else if (!stricmp(option, "wd"))
+  {
+    tmap_write_fname = argv[2] ;
+    nargs = 1 ;
+    fprintf(stderr, "writing random tmap to %s\n", tmap_write_fname) ;
+  }
   else if (!stricmp(option, "pial"))
   {
     pial_name = argv[2] ;
@@ -338,9 +388,22 @@ get_option(int argc, char *argv[])
   else if (!stricmp(option, "tmap"))
   {
     use_thickness = -1 ;
-    tmap_fname = argv[2] ;
-    printf("reading thickness target percent map from %s\n", tmap_fname) ;
-    nargs = 1 ;
+    if (!stricmp(argv[2], "random"))
+    {
+      tmap_std = atof(argv[3]);
+      tmap_min = atof(argv[4]);
+      tmap_max = atof(argv[5]);
+      tmap_avgs = atoi(argv[6]);
+      printf("creating random tmap in [%2.2f, %2.2f] with std %2.2f and %d averages\n",
+	     tmap_min, tmap_max, tmap_std, tmap_avgs) ;
+      nargs = 5 ;
+    }
+    else
+    {
+      tmap_fname = argv[2] ;
+      printf("reading thickness target percent map from %s\n", tmap_fname) ;
+      nargs = 1 ;
+    }
   }
   else switch (toupper(*option))
     {
@@ -414,5 +477,10 @@ usage_exit(int code)
   printf("  Example: mris_expand -thickness lh.white 0.5 lh.graymid\n");
   printf("  Example: mris_expand -label labelfile lh.white 0.5 lh.graymid\n");
   printf("  Example: mris_expand -tmap thickness_pct_target.mgz lh.white 0.5 lh.graymid\n");
+  printf("     use a prespecified map of percent thickness to compute the target locations for expansion\n");
+  printf("  Example: mris_expand -tmap random 2 .25 .75 100 -wd tmap.mgz lh.white 0.5 lh.graymid\n");
+  printf("     creates a random target distance map with gaussian sampling (mean=.5, std=2) and cropping to .25/.75\n");
+  printf("     and spatial averaging 100 times (cropping is average averaging). The map will be written to \n");
+  printf("     file tmap.mgz\n");
   exit(code) ;
 }