Add IndicatorSet

Author: odow

README.md

@@ -231,6 +231,7 @@ Here is a summary of the sets defined by MathOptFormat.
 | `"PositiveSemidefiniteConeSquare"` | The cone of symmetric positive semidefinite matrices, with side length `side_dimension`. The entries of the matrix are given column by column (or equivalently, row by row). The matrix is both constrained to be symmetric and to be positive semidefinite. That is, if the functions in entries `(i, j)` and `(j, i)` are different, then a constraint will be added to make sure that the entries are equal. | {"head": "PositiveSemidefiniteConeSquare", "side_dimension": 2} |
 | `"PowerCone"` | [x, y, z] ∈ {R³: x^{exponent} y^{1-exponent} ≥ \|z\|; x, y ≥ 0} | {"head": "PowerCone", "exponent": 2.0} |
 | `"DualPowerCone"` | [u, v, w] ∈ {R³: (u / exponent)^{exponent} (v / (1-exponent))^{1-exponent} ≥ \|w\|; u, v ≥ 0} | {"head": "DualPowerCone", "exponent": 2.0} |
+| `"IndicatorSet"` | If `activate_on=one`: (y, x) ∈ {0,1}×Rᴺ: y = 0 ⟹ x ∈ S, otherwise when `activate_on=zero`: (y, x) ∈ {0,1}×Rᴺ: y = 1 ⟹ x ∈ S. | {"head": "IndicatorSet", "set": {"head": "LessThan", "upper": 2.0}, "activate_on": "one"} |
 
 ### Nonlinear functions
 

mof.schema.json

@@ -749,7 +749,26 @@
  "type": "number"
  }
  }
+ }, {
+ "description": "If `activate_on=one`: (y, x) ∈ {0,1}×Rᴺ: y = 0 ⟹ x ∈ S, otherwise when `activate_on=zero`: (y, x) ∈ {0,1}×Rᴺ: y = 1 ⟹ x ∈ S.",
+ "examples": ["{\"head\": \"IndicatorSet\", \"set\": {\"head\": \"LessThan\", \"upper\": 2.0}, \"activate_on\": \"one\"}"],
+ "required": ["set", "activate_on"],
+ "properties": {
+ "head": {
+ "const": "IndicatorSet"
+ },
+ "set": {
+ "oneOf": [{
+ "$ref": "#/definitions/scalar_sets"
+ }, {
+ "$ref": "#/definitions/vector_sets"
+ }]
+ },
+ "activate_on": {
+ "enum": ["one", "zero"]
+ }
+ }
  }]
  }
  }
-}
+}

mof/mof.go

@@ -13,31 +13,36 @@ import (
 //go:generate go run generate-static-schema.go ../mof.schema.json
 
 func main() {
-switch os.Args[1] {
-case "validate":
-if len(os.Args) != 3 {
-fmt.Println("Invalid arguments to `mof validate`")
+if len(os.Args) < 2 {
+fmt.Println("Invalid arguments to `mof`. Here is the help:\n")
+PrintHelp()
+} else {
+switch os.Args[1] {
+case "validate":
+if len(os.Args) != 3 {
+fmt.Println("Invalid arguments to `mof validate`")
+PrintHelp()
+}
+filename := os.Args[2]
+if err := ValidateFile(filename); err != nil {
+fmt.Printf("%s is not a valid MOF file\nThe error is:\n", filename)
+fmt.Println(err)
+} else {
+fmt.Printf("Success! %s conforms to the MathOptFormat schema", filename)
+}
+case "summarize":
+summary, err := SummarizeSchema()
+if err != nil {
+fmt.Println(err)
+} else {
+fmt.Println(summary)
+}
+case "help":
+PrintHelp()
+default:
+fmt.Println("Invalid arguments to `mof`.")
 PrintHelp()
 }
-filename := os.Args[2]
-if err := ValidateFile(filename); err != nil {
-fmt.Printf("%s is not a valid MOF file\nThe error is:\n", filename)
-fmt.Println(err)
-} else {
-fmt.Printf("Success! %s conforms to the MathOptFormat schema", filename)
-}
-case "summarize":
-summary, err := SummarizeSchema()
-if err != nil {
-fmt.Println(err)
-} else {
-fmt.Println(summary)
-}
-case "help":
-PrintHelp()
-default:
-fmt.Println("Invalid arguments to `mof`.")
-PrintHelp()
 }
 }