# F Sharp Programming/Discriminated Unions

F# : Discriminated Unions |

**Discriminated unions**, also called **tagged unions**, represent a finite, well-defined set of choices. Discriminated unions are often the tool of choice for building up more complicated data structures including linked lists and a wide range of trees.

## Creating Discriminated Unions

[edit | edit source]Discriminated unions are defined using the following syntax:

```
type unionName =
| Case1
| Case2 of datatype
| ...
```

By convention, union names start with a lowercase character, and union cases are written in PascalCase.

## Union Basics: an On/Off switch

[edit | edit source]Let's say we have a light switch. For most of us, a light switch has two possible states: the light switch can be ON, or it can be OFF. We can use an F# union to model our light switch's state as follows:

```
type switchstate =
| On
| Off
```

We've defined a union called `switchstate`

which has two cases, `On`

and `Off`

. You can create and use instances of `switchstate`

as follows:

```
type switchstate =
| On
| Off
let x = On (* creates an instance of switchstate *)
let y = Off (* creates another instance of switchstate *)
let main() =
printfn "x: %A" x
printfn "y: %A" y
main()
```

This program has the following types:

```
type switchstate = On | Off
val x : switchstate
val y : switchstate
```

It outputs the following:

x: On y: Off

Notice that we create an instance of `switchstate`

simply by using the name of its cases; this is because, in a literal sense, the cases of a union are constructors. As you may have guessed, since we use the same syntax for constructing objects as for matching them, we can pattern match on unions in the following way:

```
type switchstate =
| On
| Off
let toggle = function (* pattern matching input *)
| On -> Off
| Off -> On
let main() =
let x = On
let y = Off
let z = toggle y
printfn "x: %A" x
printfn "y: %A" y
printfn "z: %A" z
printfn "toggle z: %A" (toggle z)
main()
```

The function `toggle`

has the type `val toggle : switchstate -> switchstate`

.

This program has the following output:

x: On y: Off z: On toggle z: Off

## Holding Data In Unions: a dimmer switch

[edit | edit source]The example above is kept deliberately simple. In fact, in many ways, the discriminated union defined above doesn't appear much different from an enum value. However, let's say we wanted to change our light switch model into a model of a dimmer switch, or in other words a light switch that allows users to adjust a lightbulb's power output from 0% to 100% power.

We can modify our union above to accommodate three possible states: On, Off, and an adjustable value between 0 and 100:

```
type switchstate =
| On
| Off
| Adjustable of float
```

We've added a new case, `Adjustable of float`

. This case is fundamentally the same as the others, except it takes a single `float`

value in its constructor:

```
open System
type switchstate =
| On
| Off
| Adjustable of float
let toggle = function
| On -> Off
| Off -> On
| Adjustable(brightness) ->
(* Matches any switchstate of type Adjustable. Binds
the value passed into the constructor to the variable
'brightness'. Toggles dimness around the halfway point. *)
let pivot = 0.5
if brightness <= pivot then
Adjustable(brightness + pivot)
else
Adjustable(brightness - pivot)
let main() =
let x = On
let y = Off
let z = Adjustable(0.25) (* takes a float in constructor *)
printfn "x: %A" x
printfn "y: %A" y
printfn "z: %A" z
printfn "toggle z: %A" (toggle z)
Console.ReadKey(true) |> ignore
main()
```

This program outputs:

x: On y: Off z: Adjustable 0.25 toggle z: Adjustable 0.75

## Creating Trees

[edit | edit source]Discriminated unions can easily model a wide variety of trees and hierarchical data structures.

For example, let's consider the following binary tree:

Each node of our tree contains exactly two branches, and each branch can either be an integer or another tree. We can represent this tree as follows:

```
type tree =
| Leaf of int
| Node of tree * tree
```

We can create an instance of the tree above using the following code:

```
open System
type tree =
| Leaf of int
| Node of tree * tree
let simpleTree =
Node(
Leaf 1,
Node(
Leaf 2,
Node(
Node(
Leaf 4,
Leaf 5
),
Leaf 3
)
)
)
let main() =
printfn "%A" simpleTree
Console.ReadKey(true) |> ignore
main()
```

This program outputs the following:

Node (Leaf 1,Node (Leaf 2,Node (Node (Leaf 4,Leaf 5),Leaf 3)))

Very often, we want to recursively enumerate through all of the nodes in a tree structure. For example, if we wanted to count the total number of Leaf nodes in our tree, we can use:

```
open System
type tree =
| Leaf of int
| Node of tree * tree
let simpleTree =
Node (Leaf 1, Node (Leaf 2, Node (Node (Leaf 4, Leaf 5), Leaf 3)))
let rec countLeaves = function
| Leaf(_) -> 1
| Node(tree1, tree2) ->
(countLeaves tree1) + (countLeaves tree2)
let main() =
printfn "countLeaves simpleTree: %i" (countLeaves simpleTree)
Console.ReadKey(true) |> ignore
main()
```

This program outputs:

countLeaves simpleTree: 5

## Generalizing Unions For All Datatypes

[edit | edit source]Note that our binary tree above only operates on integers. It is possible to construct unions that are generalized to operate on all possible data types. We can modify the definition of our union to the following:

```
type 'a tree =
| Leaf of 'a
| Node of 'a tree * 'a tree
(* The syntax above is "OCaml" style. It's common to see
unions defined using the ".NET" style as follows which
surrounds the type parameter with <'s and >'s after the
type name:
type tree<'a> =
| Leaf of 'a
| Node of tree<'a> * tree<'a>
*)
```

We can use the union defined above to define a binary tree of any data type:

```
open System
type 'a tree =
| Leaf of 'a
| Node of 'a tree * 'a tree
let firstTree =
Node(
Leaf 1,
Node(
Leaf 2,
Node(
Node(
Leaf 4,
Leaf 5
),
Leaf 3
)
)
)
let secondTree =
Node(
Node(
Node(
Leaf "Red",
Leaf "Orange"
),
Node(
Leaf "Yellow",
Leaf "Green"
)
),
Node(
Leaf "Blue",
Leaf "Violet"
)
)
let prettyPrint tree =
let rec loop depth tree =
let spacer = new String(' ', depth)
match tree with
| Leaf(value) ->
printfn "%s |- %A" spacer value
| Node(tree1, tree2) ->
printfn "%s |" spacer
loop (depth + 1) tree1
loop (depth + 1) tree2
loop 0 tree
let main() =
printfn "firstTree:"
prettyPrint firstTree
printfn "secondTree:"
prettyPrint secondTree
Console.ReadKey(true) |> ignore
main()
```

The functions above have the following types:

```
type 'a tree =
| Leaf of 'a
| Node of 'a tree * 'a tree
val firstTree : int tree
val secondTree : string tree
val prettyPrint : 'a tree -> unit
```

This program outputs:

firstTree: | |- 1 | |- 2 | | |- 4 |- 5 |- 3 secondTree: | | | |- "Red" |- "Orange" | |- "Yellow" |- "Green" | |- "Blue" |- "Violet"

## Examples

[edit | edit source]### Built-in Union Types

[edit | edit source]F# has several built-in types derived from discriminated unions, some of which have already been introduced in this tutorial. These types include:

```
type 'a list =
| Cons of 'a * 'a list
| Nil
type 'a option =
| Some of 'a
| None
```

### Propositional Logic

[edit | edit source]The ML family of languages, which includes F# and its parent language OCaml, were originally designed for the development of automated theorem provers. Union types allow F# programmers to represent propositional logic remarkably concisely. To keep things simple, let's limit our propositions to four possible cases:

```
type proposition =
| True
| Not of proposition
| And of proposition * proposition
| Or of proposition * proposition
```

Let's say we had a series of propositions and wanted to determine whether they evaluate to true or false. We can easily write an eval function by recursively enumerating through a propositional statement as follows:

```
let rec eval = function
| True -> true
| Not(prop) -> not (eval(prop))
| And(prop1, prop2) -> eval(prop1) && eval(prop2)
| Or(prop1, prop2) -> eval(prop1) || eval(prop2)
```

The `eval`

function has the type `val eval : proposition -> bool`

.

Here is a full program using the eval function:

```
open System
type proposition =
| True
| Not of proposition
| And of proposition * proposition
| Or of proposition * proposition
let prop1 =
(* ~t || ~~t *)
Or(
Not True,
Not (Not True)
)
let prop2 =
(* ~(t && ~t) || ( (t || t) || ~t) *)
Or(
Not(
And(
True,
Not True
)
),
Or(
Or(
True,
True
),
Not True
)
)
let prop3 =
(* ~~~~~~~t *)
Not(Not(Not(Not(Not(Not(Not True))))))
let rec eval = function
| True -> true
| Not(prop) -> not (eval(prop))
| And(prop1, prop2) -> eval(prop1) && eval(prop2)
| Or(prop1, prop2) -> eval(prop1) || eval(prop2)
let main() =
let testProp name prop = printfn "%s: %b" name (eval prop)
testProp "prop1" prop1
testProp "prop2" prop2
testProp "prop3" prop3
Console.ReadKey(true) |> ignore
main()
```

This program outputs the following:

prop1: true prop2: true prop3: false

## Additional Reading

[edit | edit source]Theorem Proving Examples (OCaml)