summaryrefslogtreecommitdiff
blob: d5f321db2131538b5e07dbc9c9a32df9a0ffc0c9 (plain)
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
107
108
109
#include <stdlib.h>

#include "../core/char.h"
#include "../core/index.h"

#include "../error/error.h"

#include "../sequence/sequence.h"

#include "knowledge.h"

static int weighted_random_pick
(
   const struct JH_knowledge_sequence_data sd [const restrict static 1],
   JH_index result [const restrict static 1]
)
{
   JH_index accumulator, random_number;

   accumulator = 0;

   if (sd->occurrences == 0)
   {
      return -1;
   }

   random_number = JH_index_random_up_to(sd->occurrences);
   /*@ ensures (0 <= random_number <= weights_sum); @*/

   *result = 0;

   for (;;)
   {
      accumulator += sd->targets[*result].occurrences;

      if (accumulator < random_number)
      {
         *result += 1;
      }
      else
      {
         *result = sd->targets[*result].id;

         return 0;
      }
   }
}

int JH_knowledge_random_tws_target
(
   const struct JH_knowledge k [const static 1],
   JH_index target [const restrict static 1],
   const JH_index word_id,
   const JH_index sequence_id
)
{
   JH_index s_index;

   if
   (
      JH_knowledge_find_markov_sequence
      (
         sequence_id,
         &(k->words[word_id].tws),
         &s_index
      ) < 0
   )
   {
      return -1;
   }

   return
      weighted_random_pick
      (
         &(k->words[word_id].tws.sequences_ref[s_index]),
         target
      );
}

int JH_knowledge_random_swt_target
(
   const struct JH_knowledge k [const static 1],
   const JH_index sequence_id,
   const JH_index word_id,
   JH_index target [const restrict static 1]
)
{
   JH_index s_index;

   if
   (
      JH_knowledge_find_markov_sequence
      (
         sequence_id,
         &(k->words[word_id].swt),
         &s_index
      ) < 0
   )
   {
      return -1;
   }

   return
      weighted_random_pick
      (
         &(k->words[word_id].swt.sequences_ref[s_index]),
         target
      );
}