COBOL data invariants

Author: kevinlano

ASTBasicTerm.java

@@ -2669,7 +2669,25 @@ public Vector normaliseAntlr()
     return alts; 
   } 
 
-  public Vector cobolDataDefinitions(java.util.Map context)
+  public int cobolDataWidth()
+  { if ("integerLiteral".equals(tag))
+    { return value.length(); }
+
+    if ("X".equals(value) || "9".equals(value) || 
+        ",".equals(value) || "A".equals(value) ||
+        "B".equals(value) || "0".equals(value) ||
+        "/".equals(value) || "+".equals(value) ||
+        "-".equals(value) || "$".equals(value) ||
+        "£".equals(value) || ".".equals(value) ||
+        "Z".equals(value) || "*".equals(value))
+    { return 1; } 
+
+    // S, P, V do not add to length. 
+ 
+    return 0; 
+  } 
+
+  public Vector cobolDataDefinitions(java.util.Map context, Vector invs)
   { // Each immediately higher level item becomes an attribute
     // of container. If composite, it also becomes a class
     Vector res = new Vector(); 

ASTCompositeTerm.java

@@ -1,6 +1,9 @@
 import java.util.Vector; 
 import java.io.*; 
 
+import javax.swing.*;
+
+
 /******************************
 * Copyright (c) 2003--2022 Kevin Lano
 * This program and the accompanying materials are made available under the
@@ -40134,7 +40137,43 @@ else if ("*".equals(suf))
     return alternatives; 
   } 
 
-  public Vector cobolDataDefinitions(java.util.Map context)
+  public int cobolDataWidth()
+  { if ("dataPictureClause".equals(tag))
+    { // (PIC | PICTURE) IS? pictureString
+
+      int sze = terms.size(); 
+      ASTTerm ptrm = (ASTTerm) terms.get(sze-1); 
+      return ptrm.cobolDataWidth(); 
+    } 
+
+    if ("pictureString".equals(tag))
+    { int res = 0; 
+      for (int i = 0; i < terms.size(); i++) 
+      { ASTTerm tt = (ASTTerm) terms.get(i); 
+        int wdth = tt.cobolDataWidth(); 
+        res = res + wdth; 
+      } 
+      return res; 
+    } 
+
+    if ("pictureCardinality".equals(tag))
+    { ASTTerm val = (ASTTerm) terms.get(1); 
+      String lit = val.literalForm(); 
+      try { 
+        int crd = Integer.parseInt(lit); 
+        return crd-1; 
+      } catch (Exception _ex) { return 0; } 
+    } 
+
+    if ("pictureChars".equals(tag))
+    { ASTTerm val = (ASTTerm) terms.get(0); 
+      return val.cobolDataWidth(); 
+    }   
+
+    return 0; 
+  } 
+ 
+  public Vector cobolDataDefinitions(java.util.Map context, Vector invs)
   { // Each non-filler item at level > 01 becomes an attribute
     // of a relevant container. If composite (no picture), 
     // it also becomes a class & container in turn.
@@ -40144,7 +40183,7 @@ public Vector cobolDataDefinitions(java.util.Map context)
     if ("compilationUnit".equals(tag))
     { for (int i = 0; i < terms.size(); i++) 
       { ASTTerm tt = (ASTTerm) terms.get(i); 
-        Vector ttres = tt.cobolDataDefinitions(context); 
+        Vector ttres = tt.cobolDataDefinitions(context, invs); 
         res.addAll(ttres); 
       } 
       return res; 
@@ -40153,18 +40192,37 @@ public Vector cobolDataDefinitions(java.util.Map context)
     if ("programUnit".equals(tag))
     { for (int i = 0; i < terms.size(); i++) 
       { ASTTerm tt = (ASTTerm) terms.get(i); 
-        Vector ttres = tt.cobolDataDefinitions(context); 
+        Vector ttres = tt.cobolDataDefinitions(context, invs); 
         res.addAll(ttres); 
       } 
       return res; 
     } 
 
+    if ("identificationDivision".equals(tag))
+    { // (IDENTIFICATION | ID) DIVISION . programIdParagraph identificationDivisionBody*
+
+      for (int i = 0; i < terms.size(); i++) 
+      { ASTTerm trm = (ASTTerm) terms.get(i); 
+        if ("programIdParagraph".equals(trm.getTag()))
+        { trm.cobolDataDefinitions(context,invs); } 
+      } 
+      return res; 
+    } 
+
+    if ("programIdParagraph".equals(tag))
+    { // PROGRAM-ID . programName _* 
+      ASTTerm nme = (ASTTerm) terms.get(2); 
+      String pname = nme.literalForm() + "_Class"; 
+      context.put("programName", pname);
+      return res; 
+    } 
+
     if ("dataDivision".equals(tag))
     { // DATA DIVISION . dataDivisionSection*
 
       for (int i = 3; i < terms.size(); i++) 
       { ASTTerm tt = (ASTTerm) terms.get(i); 
-        Vector ttres = tt.cobolDataDefinitions(context); 
+        Vector ttres = tt.cobolDataDefinitions(context, invs); 
         res.addAll(ttres); 
       } 
       return res; 
@@ -40175,7 +40233,7 @@ public Vector cobolDataDefinitions(java.util.Map context)
       // workingStorageSection | linkageSection | ...
 
       ASTTerm tt = (ASTTerm) terms.get(0); 
-      Vector ttres = tt.cobolDataDefinitions(context); 
+      Vector ttres = tt.cobolDataDefinitions(context, invs); 
       return ttres; 
     } 
 
@@ -40184,7 +40242,7 @@ public Vector cobolDataDefinitions(java.util.Map context)
 
       for (int i = 3; i < terms.size(); i++) 
       { ASTTerm tt = (ASTTerm) terms.get(i); 
-        Vector ttres = tt.cobolDataDefinitions(context); 
+        Vector ttres = tt.cobolDataDefinitions(context, invs); 
         res.addAll(ttres); 
       } 
       return res; 
@@ -40197,7 +40255,7 @@ public Vector cobolDataDefinitions(java.util.Map context)
       // dataDescriptionEntryExecSql
 
       ASTTerm tt = (ASTTerm) terms.get(0); 
-      Vector ttres = tt.cobolDataDefinitions(context); 
+      Vector ttres = tt.cobolDataDefinitions(context, invs); 
       return ttres; 
     } 
 
@@ -40238,7 +40296,18 @@ public Vector cobolDataDefinitions(java.util.Map context)
       Entity container = (Entity) context.get("container"); 
 
       if (ASTTerm.hasTag(terms,"dataPictureClause"))
-      { // It is a basic item, not an entity
+      { // It is a basic data item, not an entity
+
+        ASTTerm pictureClause = 
+          ASTTerm.getTermByTag(terms,"dataPictureClause"); 
+        int wdth = 0; 
+        if (pictureClause != null) 
+        { wdth = pictureClause.cobolDataWidth(); } 
+        // JOptionPane.showMessageDialog(null, "Width of " + fieldName + " is " + wdth, 
+        //                  "", 
+        //                  JOptionPane.INFORMATION_MESSAGE);  
+ 
+
         if (container == null) // no container, so top-level attribute
         { if ("FILLER".equals(fieldName)) { } 
           else 
@@ -40253,8 +40322,47 @@ public Vector cobolDataDefinitions(java.util.Map context)
         else // basic attribute of some container 
         { int contLevel = container.levelNumber; 
 
-          Integer previousLevel = (Integer) context.get("previousLevel");
+          Integer startPosition = (Integer) context.get("startPosition"); 
+          if (startPosition != null) 
+          { int startPos = startPosition.intValue(); 
+            int endPos = startPos + wdth - 1; 
+            context.put("startPosition", endPos + 1);
+ 
+            String cname = container.getName(); 
+            String ownername = 
+               cname.substring(0,cname.length()-6); 
+            String progname = 
+               (String) context.get("programName"); 
+            JOptionPane.showMessageDialog(null, progname + ":: " + fieldName + 
+               " = " + ownername + 
+               ".subrange(" + startPos + "," + endPos + ")", 
+                          "", 
+                          JOptionPane.INFORMATION_MESSAGE);
+            BasicExpression owner = 
+              BasicExpression.newAttributeBasicExpression(
+                                   ownername, 
+                                   new Type("String", null)); 
+            BasicExpression attr = 
+              BasicExpression.newAttributeBasicExpression(
+                                   fieldName, 
+                                   new Type("String", null));
+            Vector spars = new Vector(); 
+            spars.add(new BasicExpression(startPos)); 
+            spars.add(new BasicExpression(endPos));  
+            BasicExpression substr = 
+              BasicExpression.newFunctionBasicExpression(
+                "subrange", owner, spars); 
+            BinaryExpression inv = 
+              new BinaryExpression("=", attr, substr);
+            Constraint cons = 
+              Constraint.getConstraint(inv); 
+            cons.ownerName = progname;  
+            invs.add(cons);   
+          } 
+          else 
+          { context.put("startPosition", 1 + wdth); } 
 
+          Integer previousLevel = (Integer) context.get("previousLevel");
           int prevLevel = previousLevel.intValue(); 
 
           if (levelNumber >= prevLevel) 
@@ -40300,12 +40408,20 @@ else if (levelNumber < prevLevel)
           Integer previousLevel = (Integer) context.get("previousLevel");
 
           int prevLevel = -1; 
-		  if (previousLevel != null) 
-		  { prevLevel = previousLevel.intValue(); }     
+          if (previousLevel != null) 
+          { prevLevel = previousLevel.intValue(); }     
 
           res.add(newent);
           if (container == null) // top-level
-          { res.add(att); } 
+          { String pname = 
+              (String) context.get("programName");
+            if (pname != null)  
+            { Entity prog = new Entity(pname);
+              prog.levelNumber = -1;  
+              newent.container = prog;
+            }  
+            res.add(att); 
+          } 
           else if (levelNumber >= prevLevel)  
           { newent.container = container; 
             container.addAttribute(att); 
@@ -40320,7 +40436,8 @@ else if (levelNumber >= prevLevel)
           } 
 
           context.put("container", newent); 
-          context.put("previousLevel", new Integer(levelNumber)); 
+          context.put("previousLevel", new Integer(levelNumber));
+          context.put("startPosition", new Integer(1));  
         } 
       }  
     } 

ASTSymbolTerm.java

@@ -842,7 +842,23 @@ public Vector normaliseAntlr()
   public String antlrElement2cstl(Vector rulerefs, Vector conds)
   { return symbol; } 
 
-  public Vector cobolDataDefinitions(java.util.Map context)
+  public int cobolDataWidth()
+  { 
+    if ("X".equals(symbol) || "9".equals(symbol) || 
+        ",".equals(symbol) || "A".equals(symbol) ||
+        "B".equals(symbol) || "0".equals(symbol) ||
+        "/".equals(symbol) || "+".equals(symbol) ||
+        "-".equals(symbol) || "$".equals(symbol) ||
+        "£".equals(symbol) || ".".equals(symbol) ||
+        "Z".equals(symbol) || "*".equals(symbol))
+    { return 1; } 
+
+    // S, P, V do not add to length. 
+ 
+    return 0; 
+  } 
+
+  public Vector cobolDataDefinitions(java.util.Map context, Vector invs)
   { Vector res = new Vector(); 
     return res; 
   } 

ASTTerm.java

@@ -1092,6 +1092,16 @@ public static boolean hasTag(Vector terms, String tag)
     return res; 
   } 
 
+  public static ASTTerm getTermByTag(Vector terms, String tag)
+  { ASTTerm res = null; 
+    for (int i = 0; i < terms.size(); i++) 
+    { ASTTerm tt = (ASTTerm) terms.get(i); 
+      if (tt.hasTag(tag))
+      { return tt; } 
+    } 
+    return res; 
+  } 
+
   public abstract boolean hasTag(String tagx); 
 
   public abstract boolean hasSingleTerm(); 
@@ -3653,8 +3663,9 @@ public static Vector randomCompositeASTTermsForTag(
 
   public abstract Vector normaliseAntlr();
 
-  public abstract Vector cobolDataDefinitions(java.util.Map context); 
+  public abstract Vector cobolDataDefinitions(java.util.Map context, Vector invs); 
 
+  public abstract int cobolDataWidth(); 
 
   public static void main(String[] args) 
   { // ASTBasicTerm t = new ASTBasicTerm("OclBasicExpression", "true");