source: trunk/greenstone3-extensions/vishnu/src/ckindexer/tree.c@ 8189

#include <sys/stat.h>
#include <sys/types.h>
#include <dirent.h>
#include <fcntl.h>
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <assert.h>
#include <ctype.h>

#include "index_cw.h"

void walkTree(FILE *fp, WordTreeFreqPtr wordFreqTree)
{
    assert(fp);
    assert(wordFreqTree);
    if(wordFreqTree->less)
        walkTree(fp, wordFreqTree->less);
    { 
        fprintf(fp,"%s %d\n", wordFreqTree->word, wordFreqTree->count);
    }
    if(wordFreqTree->more)
        walkTree(fp, wordFreqTree->more);
}

/*
  Print the entire list of possibly interesting
  words for all the documents
*/
void printTreeToFile(char *workDir, long docNum, WordTreeFreqPtr wordFreqTree)
{

    FILE *fp;
    char wdsNameBuff[200];

    assert(workDir);

    if(docNum == 1 || docNum % SUBDIRSIZE == 0)
                createSubDir(workDir,docNum / SUBDIRSIZE,"dir");

    sprintf(wdsNameBuff,"%s/dir%ld/doc%ld.txt",workDir,docNum / SUBDIRSIZE,docNum);
    fp=fopen(wdsNameBuff,"w");
    if(fp==NULL)
    {
        fprintf(stderr,"MG_GET: Could not create %s\n",wdsNameBuff);
        exit (1);
    }
    if(wordFreqTree)walkTree(fp, wordFreqTree);
    else fprintf(stderr,"printTreeToFile with null tree\n");
    fclose(fp);
}

void createSubDir(char *workDir, long dirNum, char *pref)
{
    char newName[256];
    sprintf(newName,"%s/%s%ld",workDir,pref,dirNum);
    mkdir(newName,0777);
}     

void createDir(char *workDir, char *name)
{
    char newName[256];
    sprintf(newName,"%s/%s",workDir,name);
    mkdir(newName,0777);
}     


/*
    create and initialise a tree node
*/
int createTreeFreqNode(char *wordBuff, 
                             WordTreeFreqPtr *theTree)
{
//fprintf(stderr,"createTreeFreqNode %s\n",wordBuff);
    *theTree=(WordTreeFreqPtr)malloc(sizeof(WordTreeFreq));
    assert(*theTree);
    (*theTree)->less=NULL;
    (*theTree)->more=NULL;
    (*theTree)->count=1;
    (*theTree)->word = malloc(strlen(wordBuff)+2);
    assert((*theTree)->word);
    strcpy((*theTree)->word,wordBuff);
    return 0;
}

/*
     Put a word in the tree pointed to by theTree from the document docNum
*/

int addWordToWds(char *wordBuff, WordTreeFreqPtr *theTree)
{
    int cmp;

//fprintf(stderr,"addWordToWds %s\n",wordBuff);
    if (*theTree==NULL)
        return createTreeFreqNode(wordBuff,theTree);
    else
    {
        cmp=strcmp(wordBuff,(*theTree)->word);
        if (cmp==0)
        {
            (*theTree)->count++;
                        return 0;
        }
        else
        if (cmp < 0)
            return addWordToWds(wordBuff, &(*theTree)->less);
        else
            return addWordToWds(wordBuff, &(*theTree)->more);
    }
}

/*
  Free the tree of WordTreeFreq structures 
*/
void vapeFreqs( WordTreeFreqPtr theTree)
{
    if(theTree==NULL) return;
    vapeFreqs(theTree->less);
    vapeFreqs(theTree->more);
    free(theTree->word);
    free(theTree);
}