DicctionaryProyect/graph.h at master · adrian-guevara/DicctionaryProyect · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
#ifndef __GRAPH_H__
#define __GRAPH_H__

#include <stdlib.h>
#include "parsing.h"
#include "node.h"
#include "ubication.h"
#include "probability.h"

int color = 0;

Node *addWordsToGraph(Node **graph, Word *words);
Node *addWordToGraph(Node **graph, Word *word);
char *getPrediction(Node *graph, char *phrase);
void addGraphFileToGraph(Node **graph, FILE *graphFile);
void addLinksToGraph(Node *node, Word *words, Node **graph);
void addTextFileToGraph(Node **graph, FILE *textFile);
void saveNodes(Node *nodes, FILE *graphFile);
void saveLinks(Link *links, FILE *graphFile);

char *getPrediction(Node *graph, char *phrase){
  Word *words = getWordsFromPhrase(phrase);
  graph = searchNode(graph, words->word);
  if(graph==NULL)
    phrase = NULL;
  else{
    Link *link = getMostLikelyLink(graph->links);
    if(link!=NULL)
      phrase = link->node->word;
    else
      phrase = NULL;
  }
  deleteWords(words);
  return phrase;
}
void addGraphFileToGraph(Node **graph, FILE *graphFile){
  if(graphFile==NULL)
   return;
  while(!feof(graphFile)){
    if(Word *words = getWordsFromGraphFile(graphFile)){
      addLinksToGraph(addWordToGraph(graph, words), words->next, graph);
      deleteWords(words);
    }
  }
  color++;
  fclose(graphFile);
  assignUbications(*graph);
  updateConditionalProbabilities(*graph);
}
void addLinksToGraph(Node *node, Word *words, Node **graph){
  for(; words!=NULL; words = words->next){
    Link *link = searchOrInsertLink(node, searchOrInsertNode(graph, words->word));
    if(link!=NULL)
      link->count += words->count;
  }
}
Node *addWordToGraph(Node **graph, Word *word){
  Node *node = searchOrInsertNode(graph, word->word);
  if(node!=NULL){
    node->count += word->count;
    node->color = color;
  }
  return node;
}
void addTextFileToGraph(Node **graph, FILE *textFile){
  if(textFile==NULL)
   return;
  while(!feof(textFile)){
    if(Word *words = getWordsFromTextFile(textFile)){
      addWordsToGraph(graph, words);
      deleteWords(words);
    }
  }
  color++;
  fclose(textFile);
  assignUbications(*graph);
  updateConditionalProbabilities(*graph);
}
Node *addWordsToGraph(Node **graph, Word *words){
  if(words!=NULL){
    Node *node = addWordToGraph(graph, words);
    Node *nextNode = addWordsToGraph(graph, words->next);
    if(Link *link = searchOrInsertLink(node, nextNode))
      link->count++;
    return node;
  }
  return NULL;
}
void saveNodes(Node *nodes, FILE *graphFile){
  if(graphFile==NULL)
   return;
  for(; nodes!=NULL; nodes = nodes->next){
    fprintf(graphFile, "%s ", nodes->word);
    fprintf(graphFile, "%d ", nodes->count);
    saveLinks(nodes->links, graphFile);
  }
  fclose(graphFile);
}
void saveLinks(Link *links, FILE *graphFile){
  for(; links!=NULL; links = links->next){
    fprintf(graphFile, "%s ", links->node->word);
    fprintf(graphFile, "%d ", links->count);
  }
  fputc('\n', graphFile);
}
#endif