Minor refactor on optbase and its derivatives

ams/opt/constraint.py

@@ -64,6 +64,12 @@ def __init__(self,
         self.dual = None
         self.code = None
 
+    def get_idx(self):
+        raise NotImplementedError
+
+    def get_all_idxes(self):
+        raise NotImplementedError
+
     @ensure_symbols
     def parse(self):
         """

ams/opt/exprcalc.py

@@ -35,23 +35,11 @@ def __init__(self,
                  model: Optional[str] = None,
                  src: Optional[str] = None,
                  ):
-        OptzBase.__init__(self, name=name, info=info, unit=unit)
+        OptzBase.__init__(self, name=name, info=info, unit=unit, model=model)
         self.optz = None
         self.e_str = e_str
         self.code = None
-        self.model = model
-        self.owner = None
         self.src = src
-        self.is_group = False
-
-    def get_idx(self):
-        if self.is_group:
-            return self.owner.get_idx()
-        elif self.owner is None:
-            logger.info(f'ExpressionCalc <{self.name}> has no owner.')
-            return None
-        else:
-            return self.owner.idx.v
 
     @ensure_symbols
     def parse(self):

ams/opt/expression.py

@@ -57,26 +57,14 @@ def __init__(self,
                  vtype: Optional[str] = float,
                  horizon: Optional[str] = None,
                  ):
-        OptzBase.__init__(self, name=name, info=info, unit=unit)
+        OptzBase.__init__(self, name=name, info=info, unit=unit, model=model)
         self.tex_name = tex_name
         self.e_str = e_str
         self.optz = None
         self.code = None
-        self.model = model
-        self.owner = None
         self.src = src
-        self.is_group = False
         self.horizon = horizon
 
-    def get_idx(self):
-        if self.is_group:
-            return self.owner.get_idx()
-        elif self.owner is None:
-            logger.info(f'ExpressionCalc <{self.name}> has no owner.')
-            return None
-        else:
-            return self.owner.idx.v
-
     @ensure_symbols
     def parse(self):
         """

ams/opt/optbase.py

@@ -88,6 +88,7 @@ def __init__(self,
                  name: Optional[str] = None,
                  info: Optional[str] = None,
                  unit: Optional[str] = None,
+                 model: Optional[str] = None,
                  ):
         self.om = None
         self.name = name
@@ -97,6 +98,9 @@ def __init__(self,
         self.rtn = None
         self.optz = None  # corresponding optimization element
         self.code = None
+        self.model = model  # indicate if this element belongs to a model or group
+        self.owner = None  # instance of the owner model or group
+        self.is_group = False
 
     @ensure_symbols
     def parse(self):
@@ -153,3 +157,12 @@ def size(self):
 
     def __repr__(self):
         return f'{self.__class__.__name__}: {self.name}'
+
+    def get_idx(self):
+        if self.is_group:
+            return self.owner.get_idx()
+        elif self.owner is None:
+            logger.info(f'{self.class_name} <{self.name}> has no owner.')
+            return None
+        else:
+            return self.owner.idx.v

ams/opt/var.py

@@ -115,11 +115,8 @@ def __init__(self,
         self.tex_name = tex_name if tex_name else name
         # variable internal index inside a model (assigned in run time)
         self.id = None
-        OptzBase.__init__(self, name=name, info=info, unit=unit)
+        OptzBase.__init__(self, name=name, info=info, unit=unit, model=model)
         self.src = src
-        self.is_group = False
-        self.model = model  # indicate if this variable is a group variable
-        self.owner = None  # instance of the owner model or group
         self.v0 = v0
         self.horizon = horizon
         self._shape = shape
@@ -166,15 +163,6 @@ def v(self, value):
         else:
             self.optz.value = value
 
-    def get_idx(self):
-        if self.is_group:
-            return self.owner.get_idx()
-        elif self.owner is None:
-            logger.info(f'Variable <{self.name}> has no owner.')
-            return None
-        else:
-            return self.owner.idx.v
-
     @ensure_symbols
     def parse(self):
         """