%%
%% %CopyrightBegin%
%% 
%% Copyright Ericsson AB 1998-2009. All Rights Reserved.
%% 
%% The contents of this file are subject to the Erlang Public License,
%% Version 1.1, (the "License"); you may not use this file except in
%% compliance with the License. You should have received a copy of the
%% Erlang Public License along with this software. If not, it can be
%% retrieved online at http://www.erlang.org/.
%% 
%% Software distributed under the License is distributed on an "AS IS"
%% basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See
%% the License for the specific language governing rights and limitations
%% under the License.
%% 
%% %CopyrightEnd%
%%
%%
%%-----------------------------------------------------------------
%% File: icenum.erl
%% 
%%
%%-----------------------------------------------------------------
%%
%% Code generation for enum's.
%%-----------------------------------------------------------------
-module(icenum).

-import(ic_codegen, [emit/2, emit/3, emit/4, emit_c_enc_rpt/4, emit_c_dec_rpt/4]).

-include("icforms.hrl").
-include("ic.hrl").

%%-----------------------------------------------------------------
%% External exports
%%-----------------------------------------------------------------
-export([enum_gen/4]).

%%-----------------------------------------------------------------
%% Internal exports
%%-----------------------------------------------------------------
-export([]).

enum_gen(G, N, X, c) when is_record(X, enum) ->
    emit_c_enum(G, N, X);
enum_gen(_G, _N, _X, _L) ->
    ok.


emit_c_enum(G, N, X) ->
    case ic_genobj:is_hrlfile_open(G) of
	true ->
	    EnumName = [ic_forms:get_id2(X) | N],

	    case ic_pragma:is_local(G,EnumName) of
		true ->

		    Fd = ic_genobj:hrlfiled(G),
		    EnumNameStr = ic_util:to_undersc(EnumName),
		    ic_code:insert_typedef(G, EnumNameStr, {enum, EnumNameStr}),
		    {tk_enum,_,_,EList} = ic_forms:get_tk(X),
		    emit(Fd, "\n#ifndef __~s__\n",[ic_util:to_uppercase(EnumNameStr)]),	
		    emit(Fd, "#define __~s__\n",[ic_util:to_uppercase(EnumNameStr)]),
		    ic_codegen:mcomment_light(Fd,
					      [io_lib:format("Enum definition: ~s",
							     [EnumNameStr])],
					      c),
		    emit(Fd, "typedef CORBA_enum {", []),
		    emit_c_enum_values(G, N, Fd, EList),
		    emit(Fd, "} ~s ;\n\n", [EnumNameStr]),
		    create_c_enum_file(G, N, EnumNameStr, EList),
		    emit(Fd, "\n#endif\n\n");

		false ->   %% Do not generate included types att all.
		    ok
	    end;

	false ->
	    ok
    end.


emit_c_enum_values(_G, N, Fd, [E]) ->
    emit(Fd, "~s", [ic_util:to_undersc([E| N])]);
emit_c_enum_values(G, N, Fd, [E |Es]) ->
    emit(Fd, "~s, ", [ic_util:to_undersc([E| N])]),
    emit_c_enum_values(G, N, Fd, Es).


open_c_coding_file(G, Name) ->
    SName = string:concat(ic_util:mk_oe_name(G, "code_"), Name),
    FName =  
        ic_file:join(ic_options:get_opt(G, stubdir),ic_file:add_dot_c(SName)),
    case file:open(FName, [write]) of
        {ok, Fd} ->
            {Fd, SName};
        Other ->
            exit(Other)
    end.


create_c_enum_file(G, N, Name, Elist) ->

    {Fd , SName} = open_c_coding_file(G, Name),
    HFd = ic_genobj:hrlfiled(G), %% Write on stubfile header
    HrlFName = filename:basename(ic_genobj:include_file(G)),
    ic_codegen:emit_stub_head(G, Fd, SName, c),
    emit(Fd, "#include \"~s\"\n\n",[HrlFName]),

    %%  %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
    %%  Fd = ic_genobj:stubfiled(G), %% Write on stubfile
    %%  HFd = ic_genobj:hrlfiled(G), %% Write on stubfile header
    %%  HrlFName = filename:basename(ic_genobj:include_file(G)),
    %%  emit(Fd, "#include \"~s\"\n\n",[HrlFName]),
    %%  %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

    emit(Fd, "char* ~s[~p] = {\n", [ic_util:mk_oe_name(G, Name),
				    length(Elist)]),
    emit_c_enum_array_values(Fd, Elist),
    emit(Fd, "};\n\n",[]),
    emit_sizecount(G, N, Fd, HFd, Name, Elist),
    emit_encode(G, N, Fd, HFd, Name, Elist),
    emit_decode(G, N, Fd, HFd, Name, Elist),
    file:close(Fd).

emit_c_enum_array_values(Fd, [E]) ->
    emit(Fd, "  ~p\n", [E]);
emit_c_enum_array_values(Fd, [E |Es]) ->
    emit(Fd, "  ~p,\n", [E]),
    emit_c_enum_array_values(Fd, Es).


emit_sizecount(G, _N, Fd, HFd, Name, _Elist) ->

    emit(HFd, "int ~s~s(CORBA_Environment *oe_env, int*, int*);\n",
	 [ic_util:mk_oe_name(G, "sizecalc_"), Name]),

    emit(Fd, "int ~s~s(CORBA_Environment *oe_env, int* oe_size_count_index, int* oe_size)\n"
	 "{\n",
	 [ic_util:mk_oe_name(G, "sizecalc_"), Name]),
    emit(Fd, "  int oe_error_code = 0;\n\n",[]),

    AlignName = lists:concat(["*oe_size + sizeof(",Name,")"]),
    emit(Fd, "  *oe_size = ~s;\n\n",[ic_util:mk_align(AlignName)]),

    emit(Fd, "  if ((oe_error_code = ei_decode_atom(oe_env->_inbuf, oe_size_count_index, 0)) < 0) {\n"),
    emit_c_enc_rpt(Fd, "    ", "ei_decode_atom", []),
    emit(Fd, "    return oe_error_code;\n   }\n"),
    emit(Fd, "  return 0;\n\n",[]),
    emit(Fd, "}\n\n",[]).


emit_encode(G, _N, Fd, HFd, Name, _Elist) ->

    emit(HFd, "int ~s~s(CORBA_Environment *oe_env, ~s);\n",
	 [ic_util:mk_oe_name(G, "encode_"), Name, Name]),

    emit(Fd, "int ~s~s(CORBA_Environment *oe_env, ~s oe_rec) {\n",
	 [ic_util:mk_oe_name(G, "encode_"), Name, Name]),
    emit(Fd, "  int oe_error_code = 0;\n\n",[]),

    emit(Fd, "  if ((oe_error_code = oe_ei_encode_atom(oe_env, ~s[oe_rec])) < 0) {\n", 
	 [ic_util:mk_oe_name(G, Name)]),
    emit_c_enc_rpt(Fd, "    ", "oe_ei_encode_atom", []),
    emit(Fd, "    return oe_error_code;\n  }\n"),

    emit(Fd, "  return 0;\n\n",[]),
    emit(Fd, "}\n\n",[]).

emit_decode(G, _N, Fd, HFd, Name, Elist) ->

    emit(HFd, "int ~s~s(CORBA_Environment *oe_env, char *, int*, ~s *);\n",
	 [ic_util:mk_oe_name(G, "decode_"), Name, Name]),

    emit(Fd, "int ~s~s(CORBA_Environment *oe_env, char *oe_first, int* oe_outindex, "
	 "~s *oe_out) {\n\n",
	 [ic_util:mk_oe_name(G, "decode_"), Name, Name]),
    emit(Fd, "  int oe_error_code = 0;\n",[]),
    emit(Fd, "  int oe_i;\n",[]),
    emit(Fd, "  char oe_atom[256];\n\n",[]),

    AlignName = lists:concat(["*oe_outindex + sizeof(",Name,")"]),
    emit(Fd, "  *oe_outindex = ~s;\n\n",[ic_util:mk_align(AlignName)]),

    emit(Fd, "  if ((oe_error_code = ei_decode_atom(oe_env->_inbuf, &oe_env->_iin, oe_atom)) < 0) {\n"),
    emit_c_enc_rpt(Fd, "    ", "ei_decode_atom", []),
    emit(Fd, "    return oe_error_code;\n  }\n"),

    Len = length(Elist),
    emit(Fd, "  for(oe_i = 0; oe_i < ~p && strcmp(oe_atom, ~s[oe_i]); oe_i++);\n",
	 [Len, ic_util:mk_oe_name(G, Name)]),
    emit(Fd, "    *oe_out = oe_i;\n\n", []),

    emit(Fd, "  if (oe_i == ~p) {\n",[Len]),
    emit_c_enc_rpt(Fd, "    ", "decode atom failure", []),
    emit(Fd, "    return oe_error_code;\n  }\n"),

    emit(Fd, "  return 0;\n",[]),
    emit(Fd, "}\n\n",[]).





