epicf/ParticleInteractionModel.py at master · dumbman/epicf · 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
import sys

class ParticleInteractionModel():

    def __init__(self):
        self.noninteracting = self.binary = self.pic = False
        self.particle_interaction_model = None


    @classmethod
    def init_from_config(cls, conf):
        new_obj = cls()
        new_obj.check_correctness_of_related_config_fields(conf)
        new_obj.get_values_from_config(conf)
        ParticleInteractionModel.mark_partintmodel_sec_as_used(conf)
        return new_obj


    @staticmethod
    def mark_partintmodel_sec_as_used(conf):
        # For now simply mark sections as 'used' instead of removing them.
        conf["ParticleInteractionModel"]["used"] = "True"


    def check_correctness_of_related_config_fields(self, conf):
        conf_part = conf["ParticleInteractionModel"]
        model = conf_part["particle_interaction_model"]
        # 'PIC' or 'noninteracting' or 'binary'
        if model != "noninteracting" and model != "binary" and model != "PIC":
            print("Error: wrong value of 'particle_interaction_model': {}".format(model))
            print("Allowed values : 'noninteracting', 'binary', 'PIC'")
            print("Aborting")
            sys.exit(-1)


    def get_values_from_config(self, conf):
        conf_part = conf["ParticleInteractionModel"]
        self.particle_interaction_model = conf_part["particle_interaction_model"]
        if self.particle_interaction_model == "noninteracting":
            self.noninteracting = True
        elif self.particle_interaction_model == "binary":
            self.binary = True
        elif self.particle_interaction_model == "PIC":
            self.pic = True


    @classmethod
    def init_from_h5(cls, h5group):
        new_obj = cls()
        new_obj.particle_interaction_model = h5group.attrs["particle_interaction_model"]
        if new_obj.particle_interaction_model == "noninteracting":
            new_obj.noninteracting = True
        elif new_obj.particle_interaction_model == "binary":
            new_obj.binary = True
        elif new_obj.particle_interaction_model == "PIC":
            new_obj.pic = True
        return new_obj


    def __str__(self):
        return "particle_interaction_model = {}".format(self.particle_interaction_model)


    def print(self):
        print("### ParticleInteractionModel:")
        print(self)
        print("self.noninteracting = {}".format(self.noninteracting))
        print("self.binary = {}".format(self.binary))
        print("self.pic = {}".format(self.pic))


    def write_to_file(self, h5file):
        groupname = "/ParticleInteractionModel"
        h5group = h5file.create_group(groupname)
        h5group.attrs["particle_interaction_model"] = self.particle_interaction_model