clust_io.cpp

混合高斯模型的C++程序

#include "clust_defs.h"
#include "clust_io.h"
#include "clust_util.h"


#define eq(a,b) strcmp(a,b)==0


static int gettag(FILE *fd, char *tag);
static void get_nbands(FILE *fd, SigSet *S);
static void get_title(FILE *fd, SigSet *S);
static void get_class(FILE *fd, SigSet *S);
static void get_classnum(FILE *fd, ClassSig *C);
static void get_classtype(FILE *fd, ClassSig *C);
static void get_classtitle(FILE *fd, ClassSig *C);
static void get_subclass(FILE *fd, SigSet *S, ClassSig *C);
static void get_subclass_pi(FILE *fd, SubSig *Sp);
static void get_subclass_means(FILE *fd, SubSig *Sp, int nbands);
static void get_subclass_covar(FILE *fd, SubSig *Sp, int nbands);
static void G_strip (register char *buf);




void
I_WriteSigSet(FILE *fd, SigSet *S)
{
  ClassSig *Cp;
  SubSig *Sp;
  int i,j, b1, b2;

  fprintf (fd, "title: %s\n",I_GetSigTitle(S));
  fprintf (fd, "nbands: %d\n", S->nbands);
  for (i=0; i < S->nclasses; i++)
  {
    Cp = &S->classSig[i];
    if (!Cp->used) continue;
    fprintf (fd, "class:\n");
    fprintf (fd, " classnum: %ld\n", Cp->classnum);
    fprintf (fd, " classtitle: %s\n",I_GetClassTitle(Cp));
    fprintf (fd, " classtype: %d\n", Cp->type);

    for (j = 0; j < Cp->nsubclasses; j++) {
      Sp = &Cp->subSig[j];
      fprintf (fd, " subclass:\n");
      fprintf (fd, "  pi: %g\n", Sp->pi);
      fprintf (fd, "  means:");
      for (b1 = 0; b1 < S->nbands; b1++)
		fprintf (fd, " %g", Sp->means[b1]);
      fprintf (fd, "\n");
      fprintf (fd, "  covar:\n");
      for (b1 = 0; b1 < S->nbands; b1++) {
	fprintf (fd, "   ");
	for (b2 = 0; b2 < S->nbands; b2++)
	  fprintf (fd, " %g", Sp->R[b1][b2]);
	fprintf (fd, "\n");
      }
      fprintf (fd, " endsubclass:\n");
    }
    fprintf (fd, "endclass:\n");
  }
}


void 
I_ReadSigSet (FILE *fd, SigSet *S)
{
  char tag[256];

  I_InitSigSet (S);

  while (gettag(fd, tag)) {
    if (eq (tag, "title:")) get_title(fd, S);
    if (eq (tag, "nbands:")) get_nbands(fd, S);
    if (eq (tag, "class:")) get_class (fd, S);
  }
}

static int 
gettag(FILE *fd, char *tag)
{
  if(fscanf(fd, "%s", tag) != 1) return 0;
  G_strip (tag);
  return 1;
}

static void
get_nbands(FILE *fd, SigSet *S)
{
  fscanf (fd, "%d", &S->nbands);
}

static void
get_title(FILE *fd, SigSet *S)
{
  char title[1024];

  *title = 0;
  fscanf (fd, "%[\n]", title);
  I_SetSigTitle(S, title);
}

static void
get_class(FILE *fd, SigSet *S)
{
  char tag[1024];
  ClassSig *C;

  //printf ("get_class\n");
  C = I_NewClassSig(S);
  while (gettag(fd, tag)) {
    if (eq(tag, "endclass:"))
      break;
    if (eq(tag, "classnum:"))
      get_classnum(fd, C);
    if (eq(tag, "classtype:"))
      get_classtype(fd, C);
    if (eq(tag, "classtitle:"))
      get_classtitle(fd, C);
    if (eq(tag, "subclass:"))
      get_subclass(fd, S, C);
  }
}

static void
get_classnum(FILE *fd, ClassSig *C)
{
  fscanf (fd, "%ld", &C->classnum);
}

static void
get_classtype(FILE *fd, ClassSig *C)
{
  fscanf (fd, "%d", &C->type);
}

static void
get_classtitle(FILE *fd, ClassSig *C)
{
  char title[1024];

  *title = 0;
  fscanf (fd, "%[\n]", title);
  I_SetClassTitle (C, title);
}

static void
get_subclass(FILE *fd, SigSet *S, ClassSig *C)
{
  SubSig *Sp;
  char tag[1024];

  Sp = I_NewSubSig(S,C);

  while (gettag(fd, tag)) {
    if (eq(tag, "endsubclass:"))
	    break;
    if (eq(tag, "pi:"))
	    get_subclass_pi(fd, Sp);
    if (eq(tag, "means:"))
	    get_subclass_means (fd, Sp, S->nbands);
    if (eq(tag, "covar:"))
	    get_subclass_covar (fd, Sp, S->nbands);
  }
}


static void
get_subclass_pi(FILE *fd, SubSig *Sp)
{
  fscanf (fd, "%lf", &Sp->pi);
}


static void
get_subclass_means(FILE *fd, SubSig *Sp, int nbands)
{
  int i;

  for (i = 0; i < nbands; i++)
    fscanf (fd, "%lf", &Sp->means[i]);
}


static void
get_subclass_covar(FILE *fd, SubSig *Sp, int nbands)
{
  int i,j;

  for (i = 0; i < nbands; i++)
  for (j = 0; j < nbands; j++)
    fscanf (fd, "%lf", &Sp->R[i][j]);
}


static void
G_strip (register char *buf)
{
  register char *a, *b;

  /* remove leading white space */
  for (a = b = buf; *a == ' ' || *a == '\t'; a++)
    ;
  if (a != b)
    while(*a) {
      *b = *a;
      b++;
      a++;
    }
/*
    while (*b++ = *a++)
      ;
*/
  /* remove trailing white space */
  for (a = buf; *a; a++)
	    ;
  if (a != buf) {
    for (a--; *a == ' ' || *a == '\t'; a--)
      ;
    a++;
    *a = 0;
  }
}