How do I pass an enumerated function argument in Julia?

Question

I'm a C and Matlab programmer moving to Julia, and I'm trying to understand how function options should look when passed in.

Sometimes, a function provides different functionality based on an argument passed in with a limited number of different options. In C, it could look something like this:

enum Options{
    OPTION_1,
    OPTION_2
};

// Other arguments omitted
void foo(..., enum Options option){
    switch(option){
    case OPTION_1:
        // Do something
    break;
    case OPTION_2:
        // Do something else
    break;
    }
}

In Julia, I am not sure how to structure this. My first attempt used a hierarchy of abstract types, and then a function that accepts a singleton type to make the decision. See the following code:

abstract Options
abstract Option_1 <: Options
abstract Option_2 <: Options

function foo{T<:Options}(..., ::Type{T})
    if isa(Option_1, Type{T}
        //Do something
    elseif isa(Option_2, Type{T})
        //Do something else
    end
end

However, this seems like a very strange way to approach the problem; creating types just to control function input feels awfully strange.

Also, to clarify, I don't think that this is a solution solvable by general parametric methods in Julia. I am looking for the user to be able to specify a flag (such as run version 1 or version 2), not have different functionality based on input type.

Thanks for the help!

Jeremy Wall Jeremy Wall · Accepted Answer · 2014-03-14T04:23:04

I think parametric methods are actually exactly what you are looking for.

abstract Option_1
abstract Option_2

foo{T<:Options_1}(...) = do_something()
foo{T<:Options_2}(...) = do_something_else()

if there is common code between the two implementations then factor it out into another function and use it in both. Julia doesn't have enums but it does have ways to accomplish the same thing and parametric methods are a perfectly valid way to do so.

How do I pass an enumerated function argument in Julia?

2 Answers