1 files changed, 72 insertions, 0 deletions
diff --git a/apps/pci/optimizers/solver/ml_model.py b/apps/pci/optimizers/solver/ml_model.py
new file mode 100644
index 0000000..c239be8
--- /dev/null
+++ b/apps/pci/optimizers/solver/ml_model.py
@@ -0,0 +1,72 @@
+# -------------------------------------------------------------------------
+#   Copyright (C) 2020 Wipro Limited.
+#
+#   Licensed under the Apache License, Version 2.0 (the "License");
+#   you may not use this file except in compliance with the License.
+#   You may obtain a copy of the License at
+#
+#       http://www.apache.org/licenses/LICENSE-2.0
+#
+#   Unless required by applicable law or agreed to in writing, software
+#   distributed under the License is distributed on an "AS IS" BASIS,
+#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#   See the License for the specific language governing permissions and
+#   limitations under the License.
+#
+# -------------------------------------------------------------------------
+#
+
+import json
+
+from apps.pci.optimizers.solver.pci_utils import get_id
+from osdf.adapters.dcae import des
+from osdf.adapters.dcae.des import DESException
+from osdf.config.base import osdf_config
+from osdf.logging.osdf_logging import error_log
+
+
+class MlModel(object):
+    def __init__(self):
+        self.config = osdf_config.core['PCI']
+
+    def get_additional_inputs(self, dzn_data, network_cell_info):
+        """Add/update additional info to the existing models.
+
+        The method returns nothing. Instead, it modifies the dzn_data
+        :params: dzn_data: map with data for the optimization
+        """
+        self.compute_ml_model(dzn_data, network_cell_info)
+
+    def compute_ml_model(self, dzn_data, network_cell_info):
+        average_ho_threshold = self.config['ML']['average_ho_threshold']
+        latest_ho_threshold = self.config['ML']['latest_ho_threshold']
+
+        fixed_cells = set()
+        for cell in network_cell_info['cell_list']:
+            cell_id = cell['cell_id']
+            average_ho, latest_ho = self.get_ho_details(cell['cell_id'])
+            if average_ho > average_ho_threshold or latest_ho > latest_ho_threshold:
+                fixed_cells.add(get_id(network_cell_info, cell_id))
+
+        fixed_cells.update(dzn_data.get('PCI_UNCHANGEABLE_CELLS', []))
+        dzn_data['PCI_UNCHANGEABLE_CELLS'] = fixed_cells
+
+    def get_ho_details(self, cell_id):
+        service_id = self.config['DES']['service_id']
+        request_data = self.config['DES']['filter']
+        request_data['cell_id'] = cell_id
+        try:
+            result = des.extract_data(service_id, json.dumps(request_data))
+        except DESException as e:
+            error_log.error("Error while calling DES {}".format(e))
+            return 0, 0
+
+        if not result:
+            return 0, 0
+
+        ho_list = []
+        for pm_data in result:
+            ho = pm_data['overallHoAtt']
+            ho_list.append(ho)
+
+        return sum(ho_list) / len(ho_list), ho_list[0]