Terrain

  1# coding: utf-8
  2from Element import Element
  3import sys
  4import os
  5import shutil
  6
  7
  8class TERRAIN_BRDF_TYPE:
  9    LAMBERTIAN = "Lambertian"
 10    LAND_ALBEDO_MAP = "Land Albedo Map"
 11    SOILSPECT = "Soilspect"
 12    RPV = "RPV"
 13    ART = "ART"
 14
 15
 16class TERRAIN_TYPE:
 17    PLANE = "PLANE"
 18    RASTER = "RASTER"
 19    MESH = "MESH"
 20
 21
 22class Terrain(Element):
 23
 24    def __init__(self):
 25        super().__init__()
 26        # terrain
 27        self.terr_BRDF_Type = TERRAIN_BRDF_TYPE.LAMBERTIAN
 28        self.optical_scale = 1
 29        self.extent_height = 100
 30        self.extent_width = 100
 31        self.terr_file = ""
 32        self.terrain_type = TERRAIN_TYPE.PLANE
 33        self.temperature = "T300"
 34
 35        # conditional properties
 36        self.optical = "dark_soil_mollisol"
 37        self.landalbedo = ""
 38        self.soilSpectParams = dict()
 39        self.rpvParams = dict()
 40        self.artParams = dict()
 41
 42    def set_extent_width(self, width):
 43        self.extent_width = width
 44
 45    def set_extent_height(self, height):
 46        self.extent_height = height
 47
 48    def get_extent_width(self):
 49        return self.extent_width
 50
 51    def get_extent_height(self):
 52        return self.extent_height
 53
 54    def set_terrain_type(self, terrain_type):
 55        self.terrain_type = terrain_type
 56
 57    def set_terrain_file(self, terrain_file):
 58        filename = os.path.basename(terrain_file)
 59        dist_dem_file = os.path.join(self.get_sim().get_sim_dir(), "Parameters", filename)
 60        if dist_dem_file != terrain_file:
 61            shutil.copy(terrain_file, dist_dem_file)
 62            hdr_file = os.path.join(terrain_file+".hdr")
 63            if os.path.exists(hdr_file):
 64                shutil.copy(hdr_file, os.path.join(dist_dem_file+".hdr"))
 65        self.terr_file = filename
 66
 67    def set_terr_brdf_type(self, brdf_type):
 68        self.terr_BRDF_Type = brdf_type
 69
 70    def get_terr_brdf_type(self):
 71        return self.terr_BRDF_Type
 72
 73    def get_optical(self):
 74        if self.terr_BRDF_Type != TERRAIN_BRDF_TYPE.LAMBERTIAN:
 75            print("Terrain Type: ", self.terr_BRDF_Type, "does not have property \"optical\"")
 76            sys.exit()
 77        return self.optical
 78
 79    def set_optical(self, optical_name):
 80        if self.terr_BRDF_Type != TERRAIN_BRDF_TYPE.LAMBERTIAN:
 81            print("Terrain Type: ", self.terr_BRDF_Type, "does not have property \"optical\"")
 82            sys.exit()
 83        self.optical = optical_name
 84
 85    def get_landalbedo_file(self):
 86        if self.terr_BRDF_Type != TERRAIN_BRDF_TYPE.LAND_ALBEDO_MAP:
 87            print("Terrain Type: ", self.terr_BRDF_Type, "does not have property \"Land Albedo Map\"")
 88            sys.exit()
 89        return self.landalbedo
 90
 91    def set_landalbedo_file(self, landalbedo_file):  # 需要复制文件
 92        if self.terr_BRDF_Type != TERRAIN_BRDF_TYPE.LAND_ALBEDO_MAP:
 93            print("Terrain Type: ", self.terr_BRDF_Type, "does not have property \"Land Albedo Map\"")
 94            sys.exit()
 95        filename = os.path.basename(landalbedo_file)
 96        dist_albedomap = os.path.join(self.get_sim().get_sim_dir(), "Parameters", filename)
 97        if dist_albedomap != landalbedo_file:
 98            shutil.copy(landalbedo_file, dist_albedomap)
 99        landalbedo_file_header = landalbedo_file + ".hdr"
100        dist_albedomap_header = dist_albedomap+".hdr"
101        if os.path.exists(landalbedo_file_header) and landalbedo_file_header != dist_albedomap_header:
102            shutil.copy(landalbedo_file_header, dist_albedomap+".hdr")
103        self.landalbedo = filename
104
105    def init_terrain_from_json(self, json_object):
106        self.terr_BRDF_Type = json_object["scene"]["terrain"]["terrBRDFType"]
107        self.optical_scale = json_object["scene"]["terrain"]["optical_scale"]
108        self.extent_height = json_object["scene"]["terrain"]["extent_height"]
109        self.extent_width = json_object["scene"]["terrain"]["extent_width"]
110        self.terr_file = json_object["scene"]["terrain"]["terr_file"]
111        self.terrain_type = json_object["scene"]["terrain"]["terrain_type"]
112
113        if self.terr_BRDF_Type == "Lambertian":
114            if "optical" in json_object["scene"]["terrain"]:
115                self.optical = json_object["scene"]["terrain"]["optical"]
116        elif self.terr_BRDF_Type == "Land Albedo Map":
117            if "landalbedo" in json_object["scene"]["terrain"]:
118                self.landalbedo = json_object["scene"]["terrain"]["landalbedo"]
119        elif self.terr_BRDF_Type == "Soilspect":
120            if "soilSpectParams" in json_object["scene"]["terrain"]:
121                if "c3" in json_object["scene"]["terrain"]["soilSpectParams"]:
122                    self.soilSpectParams["c3"] = json_object["scene"]["terrain"]["soilSpectParams"]["c3"]
123                if "c4" in json_object["scene"]["terrain"]["soilSpectParams"]:
124                    self.soilSpectParams["c4"] = json_object["scene"]["terrain"]["soilSpectParams"]["c4"]
125                if "h1" in json_object["scene"]["terrain"]["soilSpectParams"]:
126                    self.soilSpectParams["h1"] = json_object["scene"]["terrain"]["soilSpectParams"]["h1"]
127                if "h2" in json_object["scene"]["terrain"]["soilSpectParams"]:
128                    self.soilSpectParams["h2"] = json_object["scene"]["terrain"]["soilSpectParams"]["h2"]
129                if "c1" in json_object["scene"]["terrain"]["soilSpectParams"]:
130                    self.soilSpectParams["c1"] = json_object["scene"]["terrain"]["soilSpectParams"]["c1"]
131                if "albedo" in json_object["scene"]["terrain"]["soilSpectParams"]:
132                    self.soilSpectParams["albedo"] = json_object["scene"]["terrain"]["soilSpectParams"]["albedo"]
133                if "c2" in json_object["scene"]["terrain"]["soilSpectParams"]:
134                    self.soilSpectParams["c2"] = json_object["scene"]["terrain"]["soilSpectParams"]["c2"]
135        elif self.terr_BRDF_Type == "RPV":
136            if "rpvParams" in json_object["scene"]["terrain"]:
137                if "rho0" in json_object["scene"]["terrain"]["rpvParams"]:
138                    self.rpvParams["rho0"] = json_object["scene"]["terrain"]["rpvParams"]["rho0"]
139                if "k" in json_object["scene"]["terrain"]["rpvParams"]:
140                    self.rpvParams["k"] = json_object["scene"]["terrain"]["rpvParams"]["k"]
141                if "THETA" in json_object["scene"]["terrain"]["rpvParams"]:
142                    self.rpvParams["THETA"] = json_object["scene"]["terrain"]["rpvParams"]["THETA"]
143                if "rhoc" in json_object["scene"]["terrain"]["rpvParams"]:
144                    self.rpvParams["rhoc"] = json_object["scene"]["terrain"]["rpvParams"]["rhoc"]
145        elif self.terr_BRDF_Type == "ART":
146            if "artParams" in json_object["scene"]["terrain"]:
147                if "snow_particle_size" in json_object["scene"]["terrain"]["artParams"]:
148                    self.artParams["snow_particle_size"] = json_object["scene"]["terrain"]["artParams"]["snow_particle_size"]
149                if "snow_pollution_content" in json_object["scene"]["terrain"]["artParams"]:
150                    self.artParams["snow_pollution_content"] = json_object["scene"]["terrain"]["artParams"]["snow_pollution_content"]
151
152        return self
153
154    def to_json_object(self):
155        json_obj = ""
156        if not self.get_sim().get_scene().get_sensor().thermal_radiation:
157            if self.terr_BRDF_Type == "Lambertian":
158                json_obj = {"terrBRDFType": self.terr_BRDF_Type,
159                            "optical_scale": self.optical_scale,
160                            "extent_height": self.extent_height,
161                            "terr_file": self.terr_file,
162                            "terrain_type": self.terrain_type,
163                            "optical": self.optical,
164                            "extent_width": self.extent_width}
165            elif self.terr_BRDF_Type == "Land Albedo Map":
166                json_obj = {"terrBRDFType": self.terr_BRDF_Type,
167                            "optical_scale": self.optical_scale,
168                            "extent_height": self.extent_height,
169                            "terr_file": self.terr_file,
170                            "terrain_type": self.terrain_type,
171                            "landalbedo": self.landalbedo,
172                            "extent_width": self.extent_width}
173            elif self.terr_BRDF_Type == "Soilspect":
174                json_obj = {"terrBRDFType": self.terr_BRDF_Type,
175                            "optical_scale": self.optical_scale,
176                            "extent_height": self.extent_height,
177                            "terr_file": self.terr_file,
178                            "terrain_type": self.terrain_type,
179                            "soilSpectParams": self.soilSpectParams,
180                            "extent_width": self.extent_width}
181            elif self.terr_BRDF_Type == "RPV":
182                json_obj = {"terrBRDFType": self.terr_BRDF_Type,
183                            "optical_scale": self.optical_scale,
184                            "extent_height": self.extent_height,
185                            "terr_file": self.terr_file,
186                            "terrain_type": self.terrain_type,
187                            "rpvParams": self.rpvParams,
188                            "extent_width": self.extent_width}
189            elif self.terr_BRDF_Type == "ART":
190                json_obj = {"terrBRDFType": self.terr_BRDF_Type,
191                            "optical_scale": self.optical_scale,
192                            "extent_height": self.extent_height,
193                            "terr_file": self.terr_file,
194                            "terrain_type": self.terrain_type,
195                            "artParams": self.artParams,
196                            "extent_width": self.extent_width}
197        else:
198            json_obj = {"terrBRDFType": self.terr_BRDF_Type,
199                        "optical_scale": self.optical_scale,
200                        "extent_height": self.extent_height,
201                        "terr_file": self.terr_file,
202                        "temperature": self.temperature,
203                        "terrain_type": self.terrain_type,
204                        "optical": self.optical,
205                        "extent_width": self.extent_width}
206        return json_obj