clust_io.cpp

高斯混合模型的C语言实现

/*
* All questions regarding the software should be addressed to
* 
*       Prof. Charles A. Bouman
*       Purdue University
*       School of Electrical and Computer Engineering
*       1285 Electrical Engineering Building
*       West Lafayette, IN 47907-1285
*       USA
*       [phone]
*       [phone] (fax)
*       email:  [e-mail]
*       http://www.ece.purdue.edu/~bouman
* 
* Copyright (c) 1995 The Board of Trustees of Purdue University.
*
* Permission to use, copy, modify, and distribute this software and its
* documentation for any purpose, without fee, and without written agreement is
* hereby granted, provided that the above copyright notice and the following
* two paragraphs appear in all copies of this software.
*
* IN NO EVENT SHALL PURDUE UNIVERSITY BE LIABLE TO ANY PARTY FOR DIRECT,
* INDIRECT, SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES ARISING OUT OF THE
* USE OF THIS SOFTWARE AND ITS DOCUMENTATION, EVEN IF PURDUE UNIVERSITY HAS
* BEEN ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*
* PURDUE UNIVERSITY SPECIFICALLY DISCLAIMS ANY WARRANTIES, INCLUDING, BUT NOT
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
* PARTICULAR PURPOSE.  THE SOFTWARE PROVIDED HEREUNDER IS ON AN "AS IS" BASIS,
* AND PURDUE UNIVERSITY HAS NO OBLIGATION TO PROVIDE MAINTENANCE, SUPPORT,
* UPDATES, ENHANCEMENTS, OR MODIFICATIONS.
*/
#include "stdafx.h"


#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, struct SigSet *S);
static void get_title(FILE *fd, struct SigSet *S);
static void get_class(FILE *fd, struct SigSet *S);
static void get_classnum(FILE *fd, struct SigSet::ClassSig *C);
static void get_classtype(FILE *fd, struct SigSet::ClassSig *C);
static void get_classtitle(FILE *fd, struct SigSet::ClassSig *C);
static void get_subclass(FILE *fd, struct SigSet *S, struct SigSet::ClassSig *C);
static void get_subclass_pi(FILE *fd, struct SigSet::ClassSig::SubSig *Sp);
static void get_subclass_means(FILE *fd, struct SigSet::ClassSig::SubSig *Sp, int nbands);
static void get_subclass_covar(FILE *fd, struct SigSet::ClassSig::SubSig *Sp, int nbands);
static void G_strip (register char *buf);




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

  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((struct ClassSig *)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, "  covar inv:\n");
      for (b1 = 0; b1 < S->nbands; b1++) 
	  {
		fprintf (fd, "   ");
		for (b2 = 0; b2 < S->nbands; b2++)
			fprintf (fd, " %g", Sp->Rinv[b1][b2]);
		fprintf (fd, "\n");
      }

      fprintf (fd, "  const:\n");
	  fprintf(fd, "    %g\n", Sp->cnst);


      fprintf (fd, " endsubclass:\n");
    }
    fprintf (fd, "endclass:\n");

	//Write Probability
	fprintf(fd, "\nProbability of subclasses for each sample: \n\n");
	k = 0;
	for (i = 0; i < Cp->ClassData.npixels; i++)
	{
		fprintf(fd, "No. %5d: ", k++);
		for(j = 0; j < S->nbands; j++)
		{
			fprintf(fd, "%6.3f ", Cp->ClassData.x[i][j]);
		}
		for (j = 0; j < S->ClassSig->nsubclasses; j++)
		{
			fprintf(fd, "%6.3f ", Cp->ClassData.p[i][j]);
		}
		fprintf(fd, "%6d ", Cp->ClassData.ml_label[i]);
		fprintf(fd, "\n");
	}
	fprintf(fd, "\nEND Probability of subclasses\n");
  }

}


void 
I_ReadSigSet (FILE *fd, struct 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, struct SigSet *S)
{
  fscanf (fd, "%d", &S->nbands);
}

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

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

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

  printf ("get_class\n");
  C = (struct SigSet::ClassSig *)(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, struct SigSet::ClassSig *C)
{
  fscanf (fd, "%ld", &C->classnum);
}

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

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

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

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

  Sp = (struct SigSet::ClassSig::SubSig *)I_NewSubSig(S,(struct ClassSig *)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, struct SigSet::ClassSig::SubSig *Sp)
{
  fscanf (fd, "%lf", &Sp->pi);
}


static void
get_subclass_means(FILE *fd, struct SigSet::ClassSig::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, struct SigSet::ClassSig::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;
  }
}