# Copyright 2013, Big Switch Networks, Inc.
#
# LoxiGen is licensed under the Eclipse Public License, version 1.0 (EPL), with
# the following special exception:
#
# LOXI Exception
#
# As a special exception to the terms of the EPL, you may distribute libraries
# generated by LoxiGen (LoxiGen Libraries) under the terms of your choice, provided
# that copyright and licensing notices generated by LoxiGen are not altered or removed
# from the LoxiGen Libraries and the notice provided below is (i) included in
# the LoxiGen Libraries, if distributed in source code form and (ii) included in any
# documentation for the LoxiGen Libraries, if distributed in binary form.
#
# Notice: "Copyright 2013, Big Switch Networks, Inc. This library was generated by the LoxiGen Compiler."
#
# You may not use this file except in compliance with the EPL or LOXI Exception. You may obtain
# a copy of the EPL at:
#
# http://www.eclipse.org/legal/epl-v10.html
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# EPL for the specific language governing permissions and limitations
# under the EPL.

#
# Miscellaneous type information
#
# Define the map between sub-class types and wire values.  In each
# case, an array indexed by wire version gives a hash from identifier
# to wire value.
#

import of_g
import sys
from generic_utils import *
import oxm
import loxi_utils.loxi_utils as loxi_utils

invalid_type = "invalid_type"
invalid_value = "0xeeee"  # Note, as a string

################################################################
#
# Define type data for inheritance classes:
#   instructions, actions, queue properties and OXM
#
# Messages are not in this group; they're treated specially for now
#
# These are indexed by wire protocol number
#
################################################################

instruction_types = {
    # version 1.0
    of_g.VERSION_1_0:dict(),

    # version 1.1
    of_g.VERSION_1_1:dict(
        goto_table = 1,
        write_metadata = 2,
        write_actions = 3,
        apply_actions = 4,
        clear_actions = 5,
        experimenter = 0xffff
        ),

    # version 1.2
    of_g.VERSION_1_2:dict(
        goto_table = 1,
        write_metadata = 2,
        write_actions = 3,
        apply_actions = 4,
        clear_actions = 5,
        experimenter = 0xffff
        ),

    # version 1.3
    of_g.VERSION_1_3:dict(
        goto_table = 1,
        write_metadata = 2,
        write_actions = 3,
        apply_actions = 4,
        clear_actions = 5,
        meter = 6,
        experimenter = 0xffff
        )
    }

of_1_3_action_types = dict(
    output       = 0,
    copy_ttl_out = 11,
    copy_ttl_in  = 12,
    set_mpls_ttl = 15,
    dec_mpls_ttl = 16,
    push_vlan    = 17,
    pop_vlan     = 18,
    push_mpls    = 19,
    pop_mpls     = 20,
    set_queue    = 21,
    group        = 22,
    set_nw_ttl   = 23,
    dec_nw_ttl   = 24,
    set_field    = 25,
    push_pbb     = 26,
    pop_pbb      = 27,
    experimenter = 0xffff,
    bsn_mirror = 0xffff,
    bsn_set_tunnel_dst = 0xffff,
    nicira_dec_ttl = 0xffff
    )

# Indexed by OF version
action_types = {
    # version 1.0
    of_g.VERSION_1_0:dict(
        output = 0,
        set_vlan_vid = 1,
        set_vlan_pcp = 2,
        strip_vlan = 3,
        set_dl_src = 4,
        set_dl_dst = 5,
        set_nw_src = 6,
        set_nw_dst = 7,
        set_nw_tos = 8,
        set_tp_src = 9,
        set_tp_dst = 10,
        enqueue = 11,
        experimenter = 0xffff,
        bsn_mirror = 0xffff,
        bsn_set_tunnel_dst = 0xffff,
        nicira_dec_ttl = 0xffff
        ),

    # version 1.1
    of_g.VERSION_1_1:dict(
        output = 0,
        set_vlan_vid = 1,
        set_vlan_pcp = 2,
        set_dl_src = 3,
        set_dl_dst = 4,
        set_nw_src = 5,
        set_nw_dst = 6,
        set_nw_tos = 7,
        set_nw_ecn = 8,
        set_tp_src = 9,
        set_tp_dst = 10,
        copy_ttl_out = 11,
        copy_ttl_in = 12,
        set_mpls_label = 13,
        set_mpls_tc = 14,
        set_mpls_ttl = 15,
        dec_mpls_ttl = 16,
        push_vlan = 17,
        pop_vlan = 18,
        push_mpls = 19,
        pop_mpls = 20,
        set_queue = 21,
        group = 22,
        set_nw_ttl = 23,
        dec_nw_ttl = 24,
        experimenter = 0xffff,
        bsn_mirror = 0xffff,
        bsn_set_tunnel_dst = 0xffff,
        nicira_dec_ttl = 0xffff
        ),

    # version 1.2
    of_g.VERSION_1_2:dict(
        output       = 0,
        copy_ttl_out = 11,
        copy_ttl_in  = 12,
        set_mpls_ttl = 15,
        dec_mpls_ttl = 16,
        push_vlan    = 17,
        pop_vlan     = 18,
        push_mpls    = 19,
        pop_mpls     = 20,
        set_queue    = 21,
        group        = 22,
        set_nw_ttl   = 23,
        dec_nw_ttl   = 24,
        set_field    = 25,
        experimenter = 0xffff,
        bsn_mirror = 0xffff,
        bsn_set_tunnel_dst = 0xffff,
        nicira_dec_ttl = 0xffff
        ),

    # version 1.3
    of_g.VERSION_1_3:of_1_3_action_types

    }

action_id_types = {
    # version 1.0
    of_g.VERSION_1_0:dict(),
    of_g.VERSION_1_1:dict(),
    of_g.VERSION_1_2:dict(),
    of_g.VERSION_1_3:of_1_3_action_types
    }

queue_prop_types = {
    # version 1.0
    of_g.VERSION_1_0:dict(
        min_rate      = 1,
        # experimenter  = 0xffff
        ),
    # version 1.1
    of_g.VERSION_1_1:dict(
        min_rate      = 1,
        #  experimenter  = 0xffff
        ),
    # version 1.2
    of_g.VERSION_1_2:dict(
        min_rate      = 1,
        max_rate      = 2,
        experimenter  = 0xffff
        ),
    # version 1.3
    of_g.VERSION_1_3:dict(
        min_rate      = 1,
        max_rate      = 2,
        experimenter  = 0xffff
        )
    }

oxm_types = {
    # version 1.0
    of_g.VERSION_1_0:dict(),

    # version 1.1
    of_g.VERSION_1_1:dict(),

    # version 1.2
    of_g.VERSION_1_2:oxm.oxm_wire_type,

    # version 1.3
    of_g.VERSION_1_3:oxm.oxm_wire_type  # FIXME needs update for 1.3?
    }

hello_elem_types = {
    # version 1.0
    of_g.VERSION_1_0:dict(),

    # version 1.1
    of_g.VERSION_1_1:dict(),

    # version 1.2
    of_g.VERSION_1_2:dict(),

    # version 1.3
    of_g.VERSION_1_3:dict(
        versionbitmap = 1
        )
    }

table_feature_prop_types = {
    # version 1.0
    of_g.VERSION_1_0:dict(),

    # version 1.1
    of_g.VERSION_1_1:dict(),

    # version 1.2
    of_g.VERSION_1_2:dict(),

    # version 1.3
    of_g.VERSION_1_3:dict(
        instructions           = 0,
        instructions_miss      = 1,
        next_tables            = 2,
        next_tables_miss       = 3,
        write_actions          = 4,
        write_actions_miss     = 5,
        apply_actions          = 6,
        apply_actions_miss     = 7,
        match                  = 8,
        wildcards              = 10,
        write_setfield         = 12,
        write_setfield_miss    = 13,
        apply_setfield         = 14,
        apply_setfield_miss    = 15,
#        experimenter           = 0xFFFE,
#        experimenter_miss      = 0xFFFF,
        experimenter            = 0xFFFF,  # Wrong: should be experimenter_miss
        )
    }

meter_band_types = {
    # version 1.0
    of_g.VERSION_1_0:dict(),

    # version 1.1
    of_g.VERSION_1_1:dict(),

    # version 1.2
    of_g.VERSION_1_2:dict(),

    # version 1.3
    of_g.VERSION_1_3:dict(
        drop                   = 1,
        dscp_remark            = 2,
        experimenter           = 0xFFFF,
        )
    }

# All inheritance data for non-messages
inheritance_data = dict(
    of_instruction = instruction_types,
    of_action = action_types,
    of_action_id = action_id_types,
    of_oxm = oxm_types,
    of_queue_prop = queue_prop_types,
    of_hello_elem = hello_elem_types,
    of_table_feature_prop = table_feature_prop_types,
    of_meter_band = meter_band_types
    )

################################################################
# Now generate the maps from parent to list of subclasses
################################################################

# # These lists have entries which are a fixed type, no inheritance
# fixed_lists = [
#     "of_list_bucket",
#     "of_list_bucket_counter",
#     "of_list_flow_stats_entry",
#     "of_list_group_desc_stats_entry",
#     "of_list_group_stats_entry",
#     "of_list_packet_queue",
#     "of_list_port_desc",
#     "of_list_port_stats_entry",
#     "of_list_queue_stats_entry",
#     "of_list_table_stats_entry"
#     ]

# for cls in fixed_lists:
#     base_type = list_to_entry_type(cls)
#     of_g.inheritance_map[base_type] = [base_type]

inheritance_map = dict()
for parent, versioned in inheritance_data.items():
    inheritance_map[parent] = set()
    for ver, subclasses in versioned.items():
        for subcls in subclasses:
            inheritance_map[parent].add(subcls)

def class_is_virtual(cls):
    """
    Returns True if cls is a virtual class
    """
    if cls in inheritance_map:
        return True
    if cls.find("header") > 0:
        return True
    if loxi_utils.class_is_list(cls):
        return True
    return False

################################################################
#
# These are message types
#
################################################################

message_types = {
    # version 1.0
    of_g.VERSION_1_0:dict(
        hello                   = 0,
        error_msg               = 1,
        echo_request            = 2,
        echo_reply              = 3,
        experimenter            = 4,
        features_request        = 5,
        features_reply          = 6,
        get_config_request      = 7,
        get_config_reply        = 8,
        set_config              = 9,
        packet_in               = 10,
        flow_removed            = 11,
        port_status             = 12,
        packet_out              = 13,
        flow_mod                = 14,
        port_mod                = 15,
        stats_request           = 16,
        stats_reply             = 17,
        barrier_request         = 18,
        barrier_reply           = 19,
        queue_get_config_request = 20,
        queue_get_config_reply  = 21,
        table_mod               = 22    # Unofficial 1.0 extension
        ),

    # version 1.1
    of_g.VERSION_1_1:dict(
        hello                   = 0,
        error_msg               = 1,
        echo_request            = 2,
        echo_reply              = 3,
        experimenter            = 4,
        features_request        = 5,
        features_reply          = 6,
        get_config_request      = 7,
        get_config_reply        = 8,
        set_config              = 9,
        packet_in               = 10,
        flow_removed            = 11,
        port_status             = 12,
        packet_out              = 13,
        flow_mod                = 14,
        group_mod               = 15,
        port_mod                = 16,
        table_mod               = 17,
        stats_request           = 18,
        stats_reply             = 19,
        barrier_request         = 20,
        barrier_reply           = 21,
        queue_get_config_request = 22,
        queue_get_config_reply  = 23
        ),

    # version 1.2
    of_g.VERSION_1_2:dict(
        hello                   = 0,
        error_msg               = 1,
        echo_request            = 2,
        echo_reply              = 3,
        experimenter            = 4,
        features_request        = 5,
        features_reply          = 6,
        get_config_request      = 7,
        get_config_reply        = 8,
        set_config              = 9,
        packet_in               = 10,
        flow_removed            = 11,
        port_status             = 12,
        packet_out              = 13,
        flow_mod                = 14,
        group_mod               = 15,
        port_mod                = 16,
        table_mod               = 17,
        stats_request           = 18,
        stats_reply             = 19,
        barrier_request         = 20,
        barrier_reply           = 21,
        queue_get_config_request = 22,
        queue_get_config_reply   = 23,
        role_request            = 24,
        role_reply              = 25,
        ),

    # version 1.3
    of_g.VERSION_1_3:dict(
        hello                   = 0,
        error_msg               = 1,
        echo_request            = 2,
        echo_reply              = 3,
        experimenter            = 4,
        features_request        = 5,
        features_reply          = 6,
        get_config_request      = 7,
        get_config_reply        = 8,
        set_config              = 9,
        packet_in               = 10,
        flow_removed            = 11,
        port_status             = 12,
        packet_out              = 13,
        flow_mod                = 14,
        group_mod               = 15,
        port_mod                = 16,
        table_mod               = 17,
        stats_request           = 18,  # FIXME Multipart
        stats_reply             = 19,
        barrier_request         = 20,
        barrier_reply           = 21,
        queue_get_config_request = 22,
        queue_get_config_reply   = 23,
        role_request            = 24,
        role_reply              = 25,
        async_get_request       = 26,
        async_get_reply         = 27,
        async_set               = 28,
        meter_mod               = 29
        )
    }

################################################################
#
# These are other objects that have a notion of type but are
# not (yet) promoted to objects with inheritance
#
################################################################

stats_types = {
    # version 1.0
    of_g.VERSION_1_0:dict(
        desc = 0,
        flow = 1,
        aggregate = 2,
        table = 3,
        port = 4,
        queue = 5,
        experimenter = 0xffff
        ),

    # version 1.1
    of_g.VERSION_1_1:dict(
        desc = 0,
        flow = 1,
        aggregate = 2,
        table = 3,
        port = 4,
        queue = 5,
        group = 6,
        group_desc = 7,
        experimenter = 0xffff
        ),

    # version 1.2
        of_g.VERSION_1_2:dict(
        desc = 0,
        flow = 1,
        aggregate = 2,
        table = 3,
        port = 4,
        queue = 5,
        group = 6,
        group_desc = 7,
        group_features = 8,
        experimenter = 0xffff
        ),

    # version 1.3
        of_g.VERSION_1_3:dict(
        desc = 0,
        flow = 1,
        aggregate = 2,
        table = 3,
        port = 4,
        queue = 5,
        group = 6,
        group_desc = 7,
        group_features = 8,
        meter = 9,
        meter_config = 10,
        meter_features = 11,
        table_features = 12,
        port_desc = 13,
        experimenter = 0xffff
        )
    }

common_flow_mod_types = dict(
    add = 0,
    modify = 1,
    modify_strict = 2,
    delete = 3,
    delete_strict = 4
    )

flow_mod_types = {
    # version 1.0
    of_g.VERSION_1_0:common_flow_mod_types,
    of_g.VERSION_1_1:common_flow_mod_types,
    of_g.VERSION_1_2:common_flow_mod_types,
    of_g.VERSION_1_3:common_flow_mod_types
    }

# These do not translate to objects (yet)
error_types = {
    # version 1.0
    of_g.VERSION_1_0:dict(
        hello_failed        = 0,
        bad_request         = 1,
        bad_action          = 2,
        flow_mod_failed     = 3,
        port_mod_failed     = 4,
        queue_op_failed     = 5
        ),

    # version 1.1
    of_g.VERSION_1_1:dict(
        hello_failed         = 0,
        bad_request          = 1,
        bad_action           = 2,
        bad_instruction      = 3,
        bad_match            = 4,
        flow_mod_failed      = 5,
        group_mod_failed     = 6,
        port_mod_failed      = 7,
        table_mod_failed     = 8,
        queue_op_failed      = 9,
        switch_config_failed = 10
        ),

    # version 1.2
    of_g.VERSION_1_2:dict(
        hello_failed         = 0,
        bad_request          = 1,
        bad_action           = 2,
        bad_instruction      = 3,
        bad_match            = 4,
        flow_mod_failed      = 5,
        group_mod_failed     = 6,
        port_mod_failed      = 7,
        table_mod_failed     = 8,
        queue_op_failed      = 9,
        switch_config_failed = 10,
        role_request_failed  = 11,
        experimenter = 0xffff
        ),

    # version 1.3
    of_g.VERSION_1_3:dict(
        hello_failed         = 0,
        bad_request          = 1,
        bad_action           = 2,
        bad_instruction      = 3,
        bad_match            = 4,
        flow_mod_failed      = 5,
        group_mod_failed     = 6,
        port_mod_failed      = 7,
        table_mod_failed     = 8,
        queue_op_failed      = 9,
        switch_config_failed = 10,
        role_request_failed  = 11,
        meter_mod_failed     = 12,
        table_features_failed= 13,
        experimenter = 0xffff
        )
    }

##
# These are the objects whose length is specified by an external
# reference, specifically another data member in the class.
# 
#external_length_spec = {
#    ("of_packet_out", "actions", OF_VERSION_1_0) : "actions_len",
#    ("of_packet_out", "actions", OF_VERSION_1_1) : "actions_len",
#    ("of_packet_out", "actions", OF_VERSION_1_2) : "actions_len",
#    ("of_packet_out", "actions", OF_VERSION_1_3) : "actions_len"
#}


################################################################
#
# type_val is the primary data structure that maps an 
# (class_name, version) pair to the wire data type value
#
################################################################

type_val = dict()

for version, classes in message_types.items():
    for cls in classes:
        name = "of_" + cls
        type_val[(name, version)] = classes[cls]

for parent, versioned in inheritance_data.items():
    for version, subclasses in versioned.items():
        for subcls, value in subclasses.items():
            name = parent + "_" + subcls
            type_val[(name, version)] = value

# Special case OF-1.2 match type
type_val[("of_match_v3", of_g.VERSION_1_2)] = 0x8000
type_val[("of_match_v3", of_g.VERSION_1_3)] = 0x8000

# Utility function
def dict_to_array(d, m_val, def_val=-1):
    """
    Given a dictionary, d, with each value a small integer,
    produce an array indexed by the integer whose value is the key.
    @param d The dictionary
    @param m_val Ignore values greater than m_val
    @param def_val The default value (for indices not in range of d)
    """

    # Get the max value in range for hash
    max_val = 0
    for key in d:
        if (d[key] > max_val) and (d[key] < m_val):
            max_val = d[key]
    ar = []
    for x in range(0, max_val + 1):
        ar.append(def_val)
    for key in d:
        if (d[key] < m_val):
            ar[d[key]] = key
    return ar

def type_array_len(version_indexed, max_val):
    """
    Given versioned information about a type, calculate how long
    the unified array should be.

    @param version_indexed A dict indexed by version. Each value is a 
    dict indexed by a name and whose value is an integer
    @param max_val Ignore values greater than this for length calcs
    """
    # First, find the max length of all arrays
    arr_len = 0
    for version, val_dict in version_indexed.items():
        ar = dict_to_array(val_dict, max_val, invalid_type)
        if arr_len < len(ar):
            arr_len = len(ar)
    return arr_len

# FIXME:  Need to move update for multipart messages

stats_reply_list = [
    "of_aggregate_stats_reply",
    "of_desc_stats_reply",
    "of_experimenter_stats_reply",
    "of_flow_stats_reply",
    "of_group_stats_reply",
    "of_group_desc_stats_reply",
    "of_group_features_stats_reply",
    "of_meter_stats_reply",
    "of_meter_config_stats_reply",
    "of_meter_features_stats_reply",
    "of_port_stats_reply",
    "of_port_desc_stats_reply",
    "of_queue_stats_reply",
    "of_table_stats_reply",
    "of_table_features_stats_reply"
]

stats_request_list = [
    "of_aggregate_stats_request",
    "of_desc_stats_request",
    "of_experimenter_stats_request",
    "of_flow_stats_request",
    "of_group_stats_request",
    "of_group_desc_stats_request",
    "of_group_features_stats_request",
    "of_meter_stats_request",
    "of_meter_config_stats_request",
    "of_meter_features_stats_request",
    "of_port_stats_request",
    "of_port_desc_stats_request",
    "of_queue_stats_request",
    "of_table_stats_request",
    "of_table_features_stats_request"
]

flow_mod_list = [
    "of_flow_add",
    "of_flow_modify",
    "of_flow_modify_strict",
    "of_flow_delete",
    "of_flow_delete_strict"
]

def sub_class_map(base_type, version):
    """
    Returns an iterable object giving the instance nameys and subclass types
    for the base_type, version values
    """
    rv = []
    if base_type not in inheritance_map:
        return rv

    for instance in inheritance_map[base_type]:
        subcls = loxi_utils.instance_to_class(instance, base_type)
        if not loxi_utils.class_in_version(subcls, version):
            continue
        rv.append((instance, subcls))

    return rv

################################################################
#
# Extension related data and functions
#
################################################################

# Per OF Version, per experimenter, map exp msg type (subtype) to object IDs
# @fixme Generate defines for OF_<exp>_SUBTYPE_<msg> for the values below?
extension_message_subtype = {
    # version 1.0
    of_g.VERSION_1_0:dict(  # Version 1.0 extensions
        bsn = {   # BSN extensions; indexed by class name, value is subtype
            "of_bsn_set_ip_mask"             : 0,
            "of_bsn_get_ip_mask_request"     : 1,
            "of_bsn_get_ip_mask_reply"       : 2,
            "of_bsn_set_mirroring"           : 3,
            "of_bsn_get_mirroring_request"   : 4,
            "of_bsn_get_mirroring_reply"     : 5,
            "of_bsn_shell_command"           : 6,
            "of_bsn_shell_output"            : 7,
            "of_bsn_shell_status"            : 8,
            "of_bsn_get_interfaces_request"  : 9,
            "of_bsn_get_interfaces_reply"    : 10,
            },
        nicira = {   # Nicira extensions, value is subtype
            "of_nicira_controller_role_request"      : 10,
            "of_nicira_controller_role_reply"        : 11,
            },
        ),
    of_g.VERSION_1_1:dict(  # Version 1.0 extensions
        bsn = {   # BSN extensions; indexed by class name, value is subtype
            "of_bsn_set_mirroring"           : 3,
            "of_bsn_get_mirroring_request"   : 4,
            "of_bsn_get_mirroring_reply"     : 5,
            "of_bsn_get_interfaces_request"  : 9,
            "of_bsn_get_interfaces_reply"    : 10,
            },
        ),
    of_g.VERSION_1_2:dict(  # Version 1.0 extensions
        bsn = {   # BSN extensions; indexed by class name, value is subtype
            "of_bsn_set_mirroring"           : 3,
            "of_bsn_get_mirroring_request"   : 4,
            "of_bsn_get_mirroring_reply"     : 5,
            "of_bsn_get_interfaces_request"  : 9,
            "of_bsn_get_interfaces_reply"    : 10,
            },
        ),
    of_g.VERSION_1_3:dict(  # Version 1.0 extensions
        bsn = {   # BSN extensions; indexed by class name, value is subtype
            "of_bsn_set_mirroring"           : 3,
            "of_bsn_get_mirroring_request"   : 4,
            "of_bsn_get_mirroring_reply"     : 5,
            "of_bsn_get_interfaces_request"  : 9,
            "of_bsn_get_interfaces_reply"    : 10,
            },
        ),
}

# Set to empty dict if no extension actions defined
# Per OF Version, per experimenter, map actions to subtype
extension_action_subtype = {
    # version 1.0
    of_g.VERSION_1_0:dict(  # Version 1.0 extensions
        bsn = {   # of_action_bsn_
            "of_action_bsn_mirror"           : 1,
            "of_action_bsn_set_tunnel_dst"   : 2,
            },
        nicira = {   # of_action_nicira_
            "of_action_nicira_dec_ttl"       : 18,
            }
        ),
    of_g.VERSION_1_1:dict(  # Version 1.0 extensions
        bsn = {   # of_action_bsn_
            "of_action_bsn_mirror"           : 1,
            "of_action_bsn_set_tunnel_dst"   : 2,
            },
        nicira = {   # of_action_nicira_
            "of_action_nicira_dec_ttl"       : 18,
            }
        ),
    of_g.VERSION_1_2:dict(  # Version 1.0 extensions
        bsn = {   # of_action_bsn_
            "of_action_bsn_mirror"           : 1,
            "of_action_bsn_set_tunnel_dst"   : 2,
            },
        nicira = {   # of_action_nicira_
            "of_action_nicira_dec_ttl"       : 18,
            }
        ),
    of_g.VERSION_1_3:dict(  # Version 1.0 extensions
        bsn = {   # of_action_bsn_
            "of_action_bsn_mirror"           : 1,
            "of_action_bsn_set_tunnel_dst"   : 2,
            },
        nicira = {   # of_action_nicira_
            "of_action_nicira_dec_ttl"       : 18,
            }
        ),
}

# Set to empty dict if no extension actions defined
# Per OF Version, per experimenter, map actions to subtype
extension_action_id_subtype = {
    # version 1.0
    of_g.VERSION_1_0:dict(),
    of_g.VERSION_1_1:dict(),
    of_g.VERSION_1_2:dict(),
    of_g.VERSION_1_3:dict(  # Version 1.3 extensions
        bsn = {   # of_action_bsn_
            "of_action_id_bsn_mirror"           : 1,
            "of_action_id_bsn_set_tunnel_dst"   : 2,
            },
        nicira = {   # of_action_nicira_
            "of_action_id_nicira_dec_ttl"       : 18,
            }
        ),
}

# Set to empty dict if no extension instructions defined
extension_instruction_subtype = {}

# Set to empty dict if no extension instructions defined
extension_queue_prop_subtype = {}

# Set to empty dict if no extension instructions defined
extension_table_feature_prop_subtype = {}

extension_objects = [
    extension_message_subtype,
    extension_action_subtype,
    extension_action_id_subtype,
    extension_instruction_subtype,
    extension_queue_prop_subtype,
    extension_table_feature_prop_subtype
]

################################################################
# These are extension type generic (for messages, actions...)
################################################################

def extension_to_experimenter_name(cls):
    """
    Return the name of the experimenter if class is an
    extension, else None

    This is brute force; we search all extension data for a match
    """
    
    for ext_obj in extension_objects:
        for version, exp_list in ext_obj.items():
            for exp_name, classes in exp_list.items():
                if cls in classes:
                    return exp_name
    return None

def extension_to_experimenter_id(cls):
    """
    Return the ID of the experimenter if class is an
    extension, else None
    """
    exp_name = extension_to_experimenter_name(cls)
    if exp_name:
        return of_g.experimenter_name_to_id[exp_name]
    return None

def extension_to_experimenter_macro_name(cls):
    """
    Return the "macro name" of the ID of the experimenter if class is an
    extension, else None
    """
    exp_name = extension_to_experimenter_name(cls)
    if exp_name:
        return "OF_EXPERIMENTER_ID_" + exp_name.upper()
    return None

def extension_to_subtype(cls, version):
    """
    Generic across all extension objects, return subtype identifier
    """
    for ext_obj in extension_objects:
        for version, exp_list in ext_obj.items():
            for exp_name, classes in exp_list.items():
                if cls in classes:
                    return classes[cls]

def class_is_extension(cls, version):
    """
    Return True if class, version is recognized as an extension
    of any type (message, action....)

    Accepts of_g.OF_VERSION_ANY
    """

    for ext_obj in extension_objects:
        if cls_is_ext_obj(cls, version, ext_obj):
            return True

    return False

# Internal
def cls_is_ext_obj(cls, version, ext_obj):
    """
    @brief Return True if cls in an extension of type ext_obj
    @param cls The class to check
    @param version The version to check
    @param ext_obj The extension object dictionary (messages, actions...)

    Accepts of_g.VERSION_ANY
    """

    # Call with each version if "any" is passed
    if version == of_g.VERSION_ANY:
        for v in of_g.of_version_range:
            if cls_is_ext_obj(cls, v, ext_obj):
                return True
    else:   # Version specified
        if version in ext_obj:
            for exp, subtype_vals in ext_obj[version].items():
                if cls in subtype_vals:
                    return True

    return False
    
################################################################
# These are extension message specific
################################################################

def message_is_extension(cls, version):
    """
    Return True if cls, version is recognized as an  extension
    This is brute force, searching records for a match
    """
    return cls_is_ext_obj(cls, version, extension_message_subtype)

def extension_message_to_subtype(cls, version):
    """
    Return the subtype of the experimenter message if the class is an
    extension, else None
    """
    if version in extension_message_subtype:
        for exp, classes in extension_message_subtype[version].items():
            for ext_class, subtype in classes.items():
                if cls == ext_class:
                    return subtype
    return None

################################################################
# These are extension action specific
################################################################

def action_is_extension(cls, version):
    """
    Return True if cls, version is recognized as an action extension
    This is brute force, searching records for a match
    """
    return cls_is_ext_obj(cls, version, extension_action_subtype)

def extension_action_to_subtype(cls, version):
    """
    Return the ID of the action subtype (for its experimenteer)
    if class is an action extension, else None
    """
    if version in extension_action_subtype:
        for exp, classes in extension_action_subtype[version].items():
            if cls in classes:
                return classes[cls]

    return None

################################################################
# These are extension action specific
################################################################

def action_id_is_extension(cls, version):
    """
    Return True if cls, version is recognized as an action ID extension
    This is brute force, searching records for a match
    """
    if version not in [of_g.VERSION_1_3]: # Action IDs only 1.3
        return False
    return cls_is_ext_obj(cls, version, extension_action_id_subtype)

def extension_action_id_to_subtype(cls, version):
    """
    Return the ID of the action ID subtype (for its experimenteer)
    if class is an action ID extension, else None
    """
    if version in extension_action_id_subtype:
        for exp, classes in extension_action_id_subtype[version].items():
            if cls in classes:
                return classes[cls]

    return None

################################################################
# These are extension instruction specific
################################################################

def instruction_is_extension(cls, version):
    """
    Return True if cls, version is recognized as an instruction extension
    This is brute force, searching records for a match
    """
    return cls_is_ext_obj(cls, version, extension_instruction_subtype)

################################################################
# These are extension queue_prop specific
################################################################

def queue_prop_is_extension(cls, version):
    """
    Return True if cls, version is recognized as an instruction extension
    This is brute force, searching records for a match
    """
    return cls_is_ext_obj(cls, version, extension_queue_prop_subtype)

################################################################
# These are extension table_feature_prop specific
################################################################

def table_feature_prop_is_extension(cls, version):
    """
    Return True if cls, version is recognized as an instruction extension
    This is brute force, searching records for a match
    """
    return cls_is_ext_obj(cls, version,
                          extension_table_feature_prop_subtype)