maskeros.c

Go to the documentation of this file.

/*****************************************************************************/
#include "tpcclibConfig.h"
/*****************************************************************************/
#include <stdio.h>
#include <stdlib.h>
#include <math.h>
#include <string.h>
#include <unistd.h>
#include <time.h>
/*****************************************************************************/
#include "libtpcmisc.h"
#include "libtpcimgio.h"
#include "libtpcimgp.h"
/*****************************************************************************/
 
/*****************************************************************************/
static char *info[] = {
  "Erode mask image in ECAT 6.3 or 7.x, NIfTI-1, or Analyze 7.5 format.",
  "Original mask image file is overwritten if name for new file is not given.",
  " ",
  "Usage: @P [Options] maskfile [newmaskfile]",
  " ",
  "Options:",
  " -struct=<<cube>|<rcube>|<star>>",
  "     Select the structuring element for erosion; either 3x3x3 cube (default),",
  "     rounded cube (3x3x3 cube without corners), or star (cube on its corner,",
  "     consisting of 7 voxels).",
  " -stdoptions", // List standard options like --help, -v, etc
  " ",
  "See also: maskdila, imgmask, imgthrs, imgmax, imgbox",
  " ",
  "Keywords: image, mask, erosion, threshold",
  0};
/*****************************************************************************/
 
/*****************************************************************************/
/* Turn on the globbing of the command line, since it is disabled by default in
   mingw-w64 (_dowildcard=0); in MinGW32 define _CRT_glob instead, if necessary;
   In Unix&Linux wildcard command line processing is enabled by default. */
/*
#undef _CRT_glob
#define _CRT_glob -1
*/
int _dowildcard = -1;
/*****************************************************************************/
 
/*****************************************************************************/
int main(int argc, char **argv)
{
  int     ai, help=0, version=0, verbose=1;
  int     ret;
  char    maskfile[FILENAME_MAX], outfile[FILENAME_MAX];
  int     structuring_element=1; // 1=cube, 2=rcube, 3=star
  char   *cptr=NULL;
  unsigned int mn;
 
 
  /*
   *  Get arguments
   */
  if(argc==1) {tpcPrintUsage(argv[0], info, stderr); return(1);}
  maskfile[0]=outfile[0]=(char)0;
  /* Options */
  for(ai=1; ai<argc; ai++) if(*argv[ai]=='-') { /* options */
    cptr=argv[ai]+1; if(*cptr=='-') cptr++; if(cptr==NULL) continue;
    if(tpcProcessStdOptions(argv[ai], &help, &version, &verbose)==0) continue;
    if(strncasecmp(cptr, "STRUCT=", 7)==0) {
      cptr+=7;
      if(strncasecmp(cptr, "CUBE", 1)==0) {structuring_element=1; continue;}
      if(strncasecmp(cptr, "RCUBE", 1)==0) {structuring_element=2; continue;}
      if(strncasecmp(cptr, "STAR", 1)==0) {structuring_element=3; continue;}
    }
    fprintf(stderr, "Error: invalid option '%s'.\n", argv[ai]); 
    return(1);
  } else break;
  
  /* Print help or version? */
  if(help==2) {tpcHtmlUsage(argv[0], info, ""); return(0);}
  if(help) {tpcPrintUsage(argv[0], info, stdout); return(0);}
  if(version) {tpcPrintBuild(argv[0], stdout); return(0);}
 
  /* Process other arguments, starting from the first non-option */
  if(ai<argc) {strlcpy(maskfile, argv[ai], FILENAME_MAX); ai++;}
  if(ai<argc) {strlcpy(outfile, argv[ai], FILENAME_MAX); ai++;}
  if(ai<argc) {fprintf(stderr, "Error: too many arguments.\n"); return(1);}
 
  /* Did we get all the information that we need? */
  if(!maskfile[0]) {
    fprintf(stderr, "Error: missing command-line argument; use option --help\n");
    return(1);
  }
 
 
  /* In verbose mode print options */
  if(verbose>1) {
    printf("maskfile := %s\n", maskfile);
    if(outfile[0]) printf("outfile := %s\n", outfile);
    printf("structuring_element := %d\n", structuring_element);
    fflush(stdout);
  }
 
  /* If file name for output was not given, the overwrite input */
  if(!outfile[0]) strcpy(outfile, maskfile);
 
 
  /*
   *  Read mask
   */
  IMG mask; imgInit(&mask);
 
  if(verbose>0) {printf("reading %s\n", maskfile); fflush(stdout);}
  ret=imgRead(maskfile, &mask);
  if(ret) {
    fprintf(stderr, "Error: %s\n", mask.statmsg);
    if(verbose>1) printf("ret := %d\n", ret);
    return(2);
  }
  if(mask.dimt>1) {
    fprintf(stderr, "Error: mask cannot be dynamic image.\n");
    imgEmpty(&mask); return(2);
  }
  /* Check whether mask has any voxels to begin with */
  mn=imgMaskCount(&mask);
  if(mn==0) {
    fprintf(stderr, "Warning: initial mask contains no positive voxels.\n");
    imgEmpty(&mask);
    return(0);
  }
  if(verbose>1) {
    printf("initial_nr_of_positive_voxels := %u\n", mn);
    fflush(stdout);
  }
 
 
  /*
   *  Make the structuring element (dimensions must be odd)
   */
  IMG selem; imgInit(&selem);
  ret=imgStructuringElement(&selem, structuring_element, verbose-1);
  if(ret!=0) {
    fprintf(stderr, "Error: cannot make structuring element.\n");
    imgEmpty(&mask); imgEmpty(&selem);
    return(5);
  }
 
 
  /*
   *  Erode the mask
   */
  if(verbose>0) {printf("eroding\n"); fflush(stdout);}
  ret=imgMaskErode(&mask, &selem);
  if(ret<0) {
    fprintf(stderr, "Error: cannot erode the mask.\n");
    if(verbose>1) printf("ret := %d\n", -ret);
    imgEmpty(&mask); imgEmpty(&selem);
    return(8);
  }
  if(ret==0) {
    fprintf(stdout, "No voxels eroded.\n");
    imgEmpty(&mask); imgEmpty(&selem);
    return(0);
  }
  if(verbose>0) {
    fprintf(stdout, "%d mask voxel(s) eroded.\n", ret);
    fflush(stdout);
  }
  /* Check whether mask has any voxels left */
  mn=imgMaskCount(&mask);
  if(mn==0) {
    fprintf(stderr, "Warning: all voxels eroded; empty mask not saved.\n");
    imgEmpty(&mask); imgEmpty(&selem);
    return(0);
  }
  if(verbose>1) {
    printf("nr_of_positive_voxels := %u\n", mn);
  }
  imgEmpty(&selem);
 
 
  /*
   *  Save the modified mask
   */
  if(verbose>2) printf("writing mask\n");
  ret=imgWrite(outfile, &mask);
  if(ret) {
    fprintf(stderr, "Error: %s\n", mask.statmsg);
    imgEmpty(&mask); return(11);
  }
  imgEmpty(&mask);
  if(verbose>0) printf("eroded mask written in %s.\n\n", outfile);
 
  return(0);
}
/*****************************************************************************/
 
/*****************************************************************************/