code/communication.py at master · theSquaredError/code · 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

'''
Provides mappings from:
     vocabulary -> QS pair
     QS pair    -> vocabulary
     location   -> QS pair
'''
import random
import string
import torch
import constants
# from graph_world import World


class Communication:
    def __init__(self,n_locations) -> None:
        self.locations = n_locations
        self.octant_vocab = torch.rand(constants.QUADRANT_SIZE)
        self.segment_vocab = torch.rand(constants.SEGMENT_SIZE)
        self.qs_pairs =[]
        self.qs_map = {}
        self.qs_vocab = {}
        '''
        for location in (self.locations):
            octant, segment = World.quadrant_circle_pair(location)
            self.qs_pairs.append([octant,segment])
            self.qs_map[location] = [octant, segment]

        for qs_pair in self.qs_pairs:
            q_word = self.octant_vocab[qs_pair[0]]
            s_word = self.segment_vocab[qs_pair[1]]

            self.qs_vocab[qs_pair] = [q_word, s_word]
        '''
    @staticmethod
    def generate_vocabulary(n_quadrant,n_segment):
        # Signal space, concept
        vocabularies_map = []
        vocabularies = []
        for i in range(n_quadrant):
            res = ''.join(random.choices(string.ascii_lowercase, k=3))
            if res in vocabularies:
                i-=1
                continue
            vocabularies.append(res)
            vocabularies_map.append([i+1,res])

        for i in range(n_segment):
            res = ''.join(random.choices(string.ascii_lowercase, k=3))
            if res in vocabularies:
                i -= 1
                continue
            vocabularies.append(res)
            vocabularies_map.append([i+1+100, res])
        return vocabularies_map
    def find_vocab(self, qs_pair):
        try:
            vocab = self.qs_vocab(qs_pair)
            return vocab
        except:
            return -1

    def find_QSPair(self,location):
        try:
            qs_pair = self.qs_map[location]
            return qs_pair
        except:
            return -1


'''
if __name__=='__main__':
    vocab_map = Communication.generate_vocabulary(3,3)
    print(vocab_map)
'''