SisMaker
/
genProto
1
0
複製 0
程式碼 問題 0 合併請求 0 Projects 0 版本發佈 3 Wiki 活動
瀏覽代碼

ft: lua序列化代码生成

master
SisMaker 1 年之前
父節點
ccbd54f3b8
當前提交
1aa59c2374
共有 2 個檔案被更改，包括 437 行新增 和 365 行删除
分割檢視
Diff Options
Show Stats Download Patch File Download Diff File
  1. +67
    -365
      src/writeLua/gLuaGen.erl
  2. +370
    -0
      src/writeLua/gLuaGen1.erl

+ 67
- 365
src/writeLua/gLuaGen.erl 查看文件

@ -1,370 +1,72 @@
-module(gLuaGen).
-export([
genLua/4
genLua/4
]).
-export([start/0]).
-export([test/0]).
%%
%% API Functions
%%
start() ->
StructList = protocol_def:get_struct_def(),
TypeList = gen_common:get_type(),
FileStr = make_protocal(StructList, "", TypeList),
file:write_file("NetPacket.lua", FileStr),
EnumStr = make_protocal_enum(protocol_def:get_enum_def(), ""),
ConstStr = make_protocal_const(protocol_def:get_version()),
file:write_file("NetEnumDef.lua", EnumStr ++ "\n" ++ ConstStr),
MsgTypeStr = make_protocal_msg_type(StructList, "", 1),
file:write_file("NetMsgType.lua", MsgTypeStr).
make_protocal([], StructStr, _TypeList) ->
StructStr;
make_protocal([Struct | Structs], StructStr, TypeList) ->
[StructName | NewStructList] = Struct,
VariableStr = make_protocal_Variable(NewStructList, "", TypeList),
EncodeStr = make_protocal_encode(NewStructList, "", TypeList),
DecodeStr = make_protocal_decode(NewStructList, "", TypeList),
BuildStr = make_protocal_build(StructName),
CreateStr = make_protocal_create(StructName),
StructStr1 = StructStr ++ "\nfunction " ++ atom_to_list(StructName) ++ " ()" ++
"\n" ++
" local tb = {}\n" ++
make_getmsgid(StructName) ++
%% " public int getMsgID()\n" ++
%% " {\n" ++
%% " return NetMsgType.msg_" ++ atom_to_list(StructName) ++ ";\n" ++
%% " }\n" ++
VariableStr ++ EncodeStr ++ DecodeStr ++ BuildStr ++ CreateStr ++ "\nend\n",
make_protocal(Structs, StructStr1, TypeList).
%%%%%%%%%%%%%%%%%%%%%%%%%%%%生成变量%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
make_protocal_Variable([], VariableStr, _TypeList) ->
VariableStr;
make_protocal_Variable([{Type, Name} | Vars], VariableStr, TypeList) ->
VariableStr1 =
case lists:keyfind(Type, 1, TypeList) of
{_Type1, base_type} ->
VariableStr ++ " tb." ++ get_csharp_Var(Name) ++ " = 0\n";
{_Type1, string_type} ->
VariableStr ++ " tb." ++ get_csharp_Var(Name) ++ " = \"\"\n";
{_Type, enum_type} ->
VariableStr ++ " tb." ++ get_csharp_Var(Name) ++ " = 0\n";
_ ->
VariableStr ++ " tb." ++ get_csharp_Var(Name) ++ " = {}\n"
end,
make_protocal_Variable(Vars, VariableStr1, TypeList);
make_protocal_Variable([{Type, Name, DefaultValue} | Vars], VariableStr, TypeList)
when Type =/= array ->
VariableStr1 =
case lists:keyfind(Type, 1, TypeList) of
{_Type1, base_type} ->
VariableStr ++ " tb." ++ get_csharp_Var(Name) ++ " = " ++ get_default_value(Type, DefaultValue) ++ "\n";
{_Type1, string_type} ->
VariableStr ++ " tb." ++ get_csharp_Var(Name) ++ " = " ++ get_default_value(Type, DefaultValue) ++ ";\n";
{_Type, enum_type} ->
VariableStr ++ " tb." ++ get_csharp_Var(Name) ++ " = 0\n";
_ ->
VariableStr ++ " tb." ++ get_csharp_Var(Name) ++ " = {}\n"
end,
make_protocal_Variable(Vars, VariableStr1, TypeList);
make_protocal_Variable([{array, Type, Name} | Vars], VariableStr, TypeList) ->
VariableStr1 =
case lists:keyfind(Type, 1, TypeList) of
{_Type1, base_type} ->
VariableStr ++ " tb." ++ get_csharp_Var(Name) ++ " = {}\n";
{_Type1, string_type} ->
VariableStr ++ " tb." ++ get_csharp_Var(Name) ++ " = {}\n";
{_Type, enum_type} ->
VariableStr ++ " tb." ++ atom_to_list(Name) ++ " = {}\n";
_ ->
VariableStr ++ " tb." ++ get_csharp_Var(Name) ++ " = {}\n"
end,
make_protocal_Variable(Vars, VariableStr1, TypeList).
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
make_protocal_build(StructName) ->
"\n tb.build = function(byteArray)\n" ++
" byteArray.write_uint16(NetMsgType[\"msg_" ++ atom_to_list(StructName) ++ "\"])\n" ++
" return tb.encode(byteArray)\n" ++
" end\n".
make_protocal_create(_StructName)->
"\n return tb\n".
%%%%%%%%%%%%%%%%%%%%%%%%%%%%生成encode%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%% byteArray.write_uint16(account.size());
%% for(int i=0; i<account.size();i++)
%% {
%% byteArray.write_string(account[i]);
%% }
make_protocal_encode([], EncodeStr, _TypeList) ->
"\n tb.encode = function(byteArray)\n"
++ EncodeStr ++ " return byteArray\n"
++ " end\n\n";
make_protocal_encode([Struct | Vars], EncodeStr, TypeList) ->
EncodeStr1 = EncodeStr ++ make_protocal_encode_item(Struct, TypeList),
make_protocal_encode(Vars, EncodeStr1, TypeList).
%% 生成decode中的数组
make_protocal_encode_item({Type, Name}, TypeList) ->
case lists:keyfind(Type, 1, TypeList) of
{Type, base_type} ->
" \tbyteArray.write_"++ atom_to_list(Type) ++"(tb." ++ get_csharp_Var(Name) ++ ")\n";
{Type, string_type} ->
" \tbyteArray.write_"++ atom_to_list(Type) ++"(tb." ++ get_csharp_Var(Name) ++ ")\n";
{_Type, enum_type} ->
make_protocal_encode_item({int, Name}, TypeList);
_ ->
" tb." ++ get_csharp_Var(Name) ++ ".encode(byteArray);\n"
%%" " ++ get_csharp_Var(Name) ++ ".encode(byteArray)\n\n"
end;
make_protocal_encode_item({Type, Name, _DefaultValue}, TypeList)
when Type /= array->
make_protocal_encode_item({Type, Name}, TypeList);
make_protocal_encode_item({array, Type, Name}, TypeList) ->
case lists:keyfind(Type, 1, TypeList) of
{Type1, base_type} ->
make_protocal_encode_item_base_type(Name, Type1);
{Type1, string_type} ->
make_protocal_encode_item_base_type(Name, Type1);
{_Type, enum_type} ->
make_protocal_encode_item({array, int, Name}, TypeList);
_ ->
NameStr = get_csharp_Var(Name),
" byteArray.write_uint16(#(tb." ++ NameStr ++ "))\n" ++
" for k, v in pairs(tb." ++ NameStr ++ ") do\n" ++
" byteArray = v.encode(byteArray)\n" ++
" end\n"
end.
make_protocal_encode_item_base_type(Name, Type) ->
NameStr = get_csharp_Var(Name),
TypeStr = get_csharp_type(Type),
" byteArray.write_uint16(#(tb." ++ NameStr ++ "))\n" ++
" for k, v in pairs (tb." ++ NameStr ++ ") do\n" ++
" byteArray.write_" ++ TypeStr ++ "(v)\n" ++
" end\n".
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%生成decode%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
make_protocal_decode([], DecodeStr, _TypeList) ->
" tb.decode = function(byteArray)\n" ++ DecodeStr ++ " end\n";
make_protocal_decode([Struct | Vars], DecodeStr, TypeList) ->
DecodeStr1 = DecodeStr ++ make_protocal_decode_item(Struct, TypeList),
make_protocal_decode(Vars, DecodeStr1, TypeList).
make_protocal_decode_item({Type, Name}, TypeList) ->
case lists:keyfind(Type, 1, TypeList) of
{Type1, base_type} ->
" tb." ++ get_csharp_Var(Name) ++ " = byteArray.read_"++ atom_to_list(Type1) ++"();\n";
{Type1, string_type} ->
" tb." ++ get_csharp_Var(Name) ++ " = byteArray.read_"++ atom_to_list(Type1) ++"();\n";
{_Type, enum_type} ->
make_protocal_decode_item({int, Name}, TypeList);
_ ->
TypeString = get_csharp_type(Type),
" tb." ++ get_csharp_Var(Name) ++ " = " ++ TypeString ++ "();\n" ++
" tb." ++ get_csharp_Var(Name) ++ ".decode(byteArray);\n"
%%" " ++ get_csharp_Var(Name) ++ ".decode(byteArray);\n"
end;
make_protocal_decode_item({Type, Name, _DefaultValue}, TypeList)
when Type /= array->
make_protocal_decode_item({Type, Name}, TypeList);
make_protocal_decode_item({array, Type, Name}, TypeList) ->
case lists:keyfind(Type, 1, TypeList) of
{Type1, base_type} ->
make_protocal_encode_item_decode_type(Name, Type1);
{Type1, string_type} ->
make_protocal_encode_item_decode_type(Name, Type1);
{_Type, enum_type} ->
make_protocal_decode_item({array, int, Name}, TypeList);
_ ->
TypeString = get_csharp_type(Type),
NameStr = get_csharp_Var(Name),
CountString = "countOf" ++ NameStr,
_ArrayString = "arrayOf" ++ NameStr,
" local " ++ CountString ++ " = byteArray.read_uint16()\n" ++
" tb."++NameStr++" = {}\n" ++
" for i = 1, " ++ CountString ++ " do\n" ++
" local temp = " ++ TypeString ++ "()\n" ++
" temp.decode(byteArray)\n" ++
" table.insert(tb."++NameStr++", temp)\n" ++
" end\n"
end.
make_protocal_encode_item_decode_type(Name, Type) ->
NameStr = get_csharp_Var(Name),
TypeStr = atom_to_list(Type),
CountString = "countOf" ++ NameStr,
" local " ++ CountString ++ " = byteArray.read_uint16()\n" ++
" tb."++NameStr++" = {}\n" ++
" for i = 1, " ++ CountString ++ " do\n" ++
" table.insert(tb." ++ NameStr ++ ", byteArray.read_" ++ TypeStr ++ "())\n" ++
" end\n".
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%生成网络类型%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
make_protocal_msg_type([], MsgTypeStr, _Index) ->
"NetMsgType = \n" ++
"{\n" ++
MsgTypeStr ++
"}";
make_protocal_msg_type([Struct|Structs], MsgTypeStr, Index) ->
[StructName | _NewStructList] = Struct,
case Structs == [] of
true ->
make_protocal_msg_type(Structs, MsgTypeStr ++ " [\"msg_" ++ atom_to_list(StructName) ++ "\"] = "++integer_to_list(Index)++"\n", Index+1);
false ->
make_protocal_msg_type(Structs, MsgTypeStr ++ " [\"msg_" ++ atom_to_list(StructName) ++ "\"] = "++integer_to_list(Index)++",\n", Index+1)
end.
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%生成枚举类型%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
make_protocal_const(Ver) ->
"function get_proto_version()
return "++integer_to_list(Ver)++";
end\n".
make_protocal_enum([], EnumStr) ->
EnumStr;
make_protocal_enum([Enum | Enums], EnumStr) ->
{EnumName , EnumItems} = Enum,
EnumStr1 = EnumStr ++
string:to_lower(atom_to_list(EnumName)) ++ " = \n{\n" ++
make_protocal_enum_item(EnumItems, 1, "") ++ "}\n",
make_protocal_enum(Enums, EnumStr1).
make_protocal_enum_item([EnumItem | []], Index, EnumItemStr) ->
EnumItemStr ++ " [\"" ++ atom_to_list(EnumItem) ++ "\"] = "++integer_to_list(Index)++"\n";
make_protocal_enum_item([EnumItem | EnumItems], Index, EnumItemStr) ->
EnumItemStr1 = EnumItemStr ++ " [\"" ++ atom_to_list(EnumItem) ++ "\"]= "++integer_to_list(Index)++",\n",
make_protocal_enum_item(EnumItems, Index + 1, EnumItemStr1).
get_csharp_Var(Type)->
TypeString = atom_to_list(Type),
case lists:keyfind(TypeString, 1, get_keyword_mapping()) of
{TypeString, CSharpTypeString}->
CSharpTypeString;
_ ->
TypeString
end.
get_csharp_type(Type) ->
TypeString = atom_to_list(Type),
case lists:keyfind(TypeString, 1, get_cshapr_type_mapping()) of
{TypeString, CSharpTypeString}->
CSharpTypeString;
_ ->
TypeString
end.
get_cshapr_type_mapping()->
[
{"uint", "UInt"},
{"int16", "Int16"},
{"uint16", "UInt16"},
{"int64", "Int64"},
{"uint64", "uint64"},
{"uchar", "char"}
].
get_keyword_mapping() ->
[
{"lock", "Lock"},
{"params", "Params"}
].
make_getmsgid(StructName)->
" tb.getMsgID = function()\n" ++
" " ++
" return NetMsgType[\"msg_" ++ atom_to_list(StructName) ++ "\"]\n" ++
" end\n".
get_default_value(Type, DefaultValue)
when (Type == int orelse
Type == uint orelse
Type == uint64 orelse
Type == uint16 orelse
Type == int64 orelse
Type == short orelse
Type == int16)
andalso is_integer(DefaultValue)->
integer_to_list(DefaultValue);
get_default_value(Type, DefaultValue)
when (Type == float orelse
Type == long orelse
Type == double orelse
Type == uint16 orelse
Type == int64 orelse
Type == int16)
andalso is_float(DefaultValue)->
float_to_list(DefaultValue);
get_default_value(char, DefaultValue)
when is_list(DefaultValue)->
"'" ++ DefaultValue ++ "'";
get_default_value(string, DefaultValue)
when is_atom(DefaultValue)->
get_default_value(string, atom_to_list(DefaultValue));
get_default_value(string, DefaultValue)
when is_list(DefaultValue)->
"\"" ++ DefaultValue ++ "\"".
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%测试函数%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
test() ->
%% test_make_protocal_encode_item(),
%% test_make_protocal_decode_item(),
%% test_make_protocal_encode(),
%% test_make_protocal_decode().
start().
%% test_make_protocal_encode_item() ->
%% "\t\tbyteArray.write_uint16(result.size());\n\t\tfor(int i=0; i<result.size();i++)\n\t{\n\t\t\tresult[i].encode(byteArray);\n\t}\n"
%% = make_protocal_encode_item({array, notify_login_result, result}, gen_common:get_type()),
%% "\t\tbyteArray.write_uint16(account.size());\n\t\tfor(int i=0; i<account.size();i++)\n\t{\n\t\t\tbyteArray.write_string(account[i]);\n\t}\n"
%% = make_protocal_encode_item({array, string, account}, gen_common:get_type()),
%% "\t\tbyteArray.write_uint16(account.size());\n\t\tfor(int i=0; i<account.size();i++)\n\t{\n\t\t\tbyteArray.write_int(account[i]);\n\t}\n"
%% = make_protocal_encode_item({array, login_result, account}, gen_common:get_type()),
%% "\t\tbyteArray.write_string(account);\n" = make_protocal_encode_item({string, account}, gen_common:get_type()),
%% "\t\tbyteArray.write_int(account);\n" = make_protocal_encode_item({login_result, account}, gen_common:get_type()),
%% "\t\taccount.encode(byteArray);\n" = make_protocal_encode_item({notify_login_result, account}, gen_common:get_type()).
%%
%% test_make_protocal_decode_item() ->
%% "uint16 size = byteArray.read_uint16();\naccount.reserve(size);\nfor(int i=0; i<size;i++)\n{\n\taccount.push_back(byteArray.read_string());\n}\n"
%% = make_protocal_decode_item({array, string, account}, gen_common:get_type()),
%% "uint16 size = byteArray.read_uint16();\nresult.resize(size);\nfor(int i=0; i<size;i++)\n{\n\tresult[i].decode(byteArray);\n}\n"
%% = make_protocal_decode_item({array, notify_login_result, result}, gen_common:get_type()),
%% "uint16 size = byteArray.read_uint16();\naccount.reserve(size);\nfor(int i=0; i<size;i++)\n{\n\taccount.push_back(byteArray.read_int());\n}\n"
%% = make_protocal_decode_item({array, login_result, account}, gen_common:get_type()),
%% "account = byteArray.read_string();\n" = make_protocal_decode_item({string, account}, gen_common:get_type()),
%% "account = byteArray.read_int();\n" = make_protocal_decode_item({login_result, account}, gen_common:get_type()),
%% "account.decode(byteArray);\n" = make_protocal_decode_item({notify_login_result, account}, gen_common:get_type()).
%%
%% test_make_protocal_encode() ->
%% Result = make_protocal_encode([{array, string, account},
%% {string, pwd},
%% {player_data, data},
%% {array, notify_login_result, result}], "", gen_common:get_type()),
%% io:format("~p~n", [Result]).
%%
%% test_make_protocal_decode() ->
%% Result = make_protocal_decode([{array, string, account},
%% {string, pwd},
%% {player_data, data},
%% {array, notify_login_result, result}], "", gen_common:get_type()),
%% io:format("~p~n", [Result]).
%%
%% test_make_protocal_Variable() ->
%% Result = make_protocal_Variable([{array, string, account},
%% {string, pwd},
%% {player_data, data},
%% {array, notify_login_result, result}], "", gen_common:get_type()),
%% io:format("~p~n", [Result]).
spellClassHead(MsgName, MsgId) ->
<<"\tpublic class ", MsgName/binary, " : ProtocolCore.ISerialize, ProtocolCore.IDeserialize<", MsgName/binary, ">\n\t{\n\t\tpublic const int ProtoId = ", (integer_to_binary(MsgId))/binary, ";\n">>.
spellClassMember(FieldList) ->
<<<<(gCsField:builtRecStr(OneTypeName))/binary>> || OneTypeName <- FieldList>>.
spellCalssDSTem() ->
<<"\n\t\tpublic byte[] Serialize()\n\t\t{\n\t\t\tusing var memoryStream = new MemoryStream();
\t\t\tusing var binaryWriter = new BinaryWriter(memoryStream);
\t\t\tSerialize(binaryWriter);
\t\t\treturn memoryStream.ToArray();
\t\t}
\t\tpublic void Deserialize(byte[] data)
\t\t{
\t\t\tusing var memoryStream = new MemoryStream(data);
\t\t\tusing var binaryReader = new BinaryReader(memoryStream);
\t\t\tDeserialize(binaryReader);
\t\t}\n">>.
spellCalssSerialize(FieldList) ->
FunHead = <<"\t\tpublic void Serialize(BinaryWriter binaryWriter)\n\t\t{\n">>,
FunBody = <<<<"\t\t\tbinaryWriter.WriteValue(", NameStr/binary, ");\n">> || {_TypeStr, NameStr} <- FieldList>>,
<<FunHead/binary, FunBody/binary, "\t\t}\n">>.
spellCalssDeserialize(FieldList) ->
FunHead = <<"\t\tpublic void Deserialize(BinaryReader binaryReader)\n\t\t{\n">>,
FunBody = <<<<"\t\t\tbinaryReader.ReadValue(out ", NameStr/binary, ");\n">> || {_TypeStr, NameStr} <- FieldList>>,
<<FunHead/binary, FunBody/binary, "\t\t}\n">>.
spellClassEnd() ->
<<"\t}\n">>.
genLua(SortedSProtoList, _SortedErrList, _HrlDir, CSDir) ->
FunSpell =
fun({MsgName, MsgId, FieldList}, ClassBinAcc) ->
H = spellClassHead(MsgName, MsgId),
M = spellClassMember(FieldList),
DS = spellCalssDSTem(),
S = spellCalssSerialize(FieldList),
D = spellCalssDeserialize(FieldList),
E = spellClassEnd(),
<<ClassBinAcc/binary, H/binary, M/binary, DS/binary, S/binary, D/binary, E/binary>>
end,
LastClassBinAcc = lists:foldl(FunSpell, <<>>, SortedSProtoList),
%% todo error code
%% ErrCodeStr = spellErrCodeHrl(SortedErrList, <<>>),
CSEndStr = protoEnd(),
OutputHrlStr = <<LastClassBinAcc/binary, CSEndStr/binary>>,
CSFilename = do_write_cs(CSDir, protoMsg, OutputHrlStr),
io:format("protoConvert erl dir : ~s ~n", [CSDir]),
io:format("protoConvert to erl file ~s succ.~n", [CSFilename]),
ok.
do_write_cs(OutDir, Mod, BinStr) ->
Filename = filename:join([OutDir, atom_to_list(Mod) ++ ".cs"]),
case file:write_file(Filename, BinStr) of
ok ->
ok;
_Ret ->
io:format("write to cs file error:~p ~n", [_Ret])
end,
Filename.

+ 370
- 0
src/writeLua/gLuaGen1.erl 查看文件

@ -0,0 +1,370 @@
-module(gLuaGen1).
-export([
genLua/4
]).
-export([start/0]).
-export([test/0]).
%%
%% API Functions
%%
start() ->
StructList = protocol_def:get_struct_def(),
TypeList = gen_common:get_type(),
FileStr = make_protocal(StructList, "", TypeList),
file:write_file("NetPacket.lua", FileStr),
EnumStr = make_protocal_enum(protocol_def:get_enum_def(), ""),
ConstStr = make_protocal_const(protocol_def:get_version()),
file:write_file("NetEnumDef.lua", EnumStr ++ "\n" ++ ConstStr),
MsgTypeStr = make_protocal_msg_type(StructList, "", 1),
file:write_file("NetMsgType.lua", MsgTypeStr).
make_protocal([], StructStr, _TypeList) ->
StructStr;
make_protocal([Struct | Structs], StructStr, TypeList) ->
[StructName | NewStructList] = Struct,
VariableStr = make_protocal_Variable(NewStructList, "", TypeList),
EncodeStr = make_protocal_encode(NewStructList, "", TypeList),
DecodeStr = make_protocal_decode(NewStructList, "", TypeList),
BuildStr = make_protocal_build(StructName),
CreateStr = make_protocal_create(StructName),
StructStr1 = StructStr ++ "\nfunction " ++ atom_to_list(StructName) ++ " ()" ++
"\n" ++
" local tb = {}\n" ++
make_getmsgid(StructName) ++
%% " public int getMsgID()\n" ++
%% " {\n" ++
%% " return NetMsgType.msg_" ++ atom_to_list(StructName) ++ ";\n" ++
%% " }\n" ++
VariableStr ++ EncodeStr ++ DecodeStr ++ BuildStr ++ CreateStr ++ "\nend\n",
make_protocal(Structs, StructStr1, TypeList).
%%%%%%%%%%%%%%%%%%%%%%%%%%%%生成变量%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
make_protocal_Variable([], VariableStr, _TypeList) ->
VariableStr;
make_protocal_Variable([{Type, Name} | Vars], VariableStr, TypeList) ->
VariableStr1 =
case lists:keyfind(Type, 1, TypeList) of
{_Type1, base_type} ->
VariableStr ++ " tb." ++ get_csharp_Var(Name) ++ " = 0\n";
{_Type1, string_type} ->
VariableStr ++ " tb." ++ get_csharp_Var(Name) ++ " = \"\"\n";
{_Type, enum_type} ->
VariableStr ++ " tb." ++ get_csharp_Var(Name) ++ " = 0\n";
_ ->
VariableStr ++ " tb." ++ get_csharp_Var(Name) ++ " = {}\n"
end,
make_protocal_Variable(Vars, VariableStr1, TypeList);
make_protocal_Variable([{Type, Name, DefaultValue} | Vars], VariableStr, TypeList)
when Type =/= array ->
VariableStr1 =
case lists:keyfind(Type, 1, TypeList) of
{_Type1, base_type} ->
VariableStr ++ " tb." ++ get_csharp_Var(Name) ++ " = " ++ get_default_value(Type, DefaultValue) ++ "\n";
{_Type1, string_type} ->
VariableStr ++ " tb." ++ get_csharp_Var(Name) ++ " = " ++ get_default_value(Type, DefaultValue) ++ ";\n";
{_Type, enum_type} ->
VariableStr ++ " tb." ++ get_csharp_Var(Name) ++ " = 0\n";
_ ->
VariableStr ++ " tb." ++ get_csharp_Var(Name) ++ " = {}\n"
end,
make_protocal_Variable(Vars, VariableStr1, TypeList);
make_protocal_Variable([{array, Type, Name} | Vars], VariableStr, TypeList) ->
VariableStr1 =
case lists:keyfind(Type, 1, TypeList) of
{_Type1, base_type} ->
VariableStr ++ " tb." ++ get_csharp_Var(Name) ++ " = {}\n";
{_Type1, string_type} ->
VariableStr ++ " tb." ++ get_csharp_Var(Name) ++ " = {}\n";
{_Type, enum_type} ->
VariableStr ++ " tb." ++ atom_to_list(Name) ++ " = {}\n";
_ ->
VariableStr ++ " tb." ++ get_csharp_Var(Name) ++ " = {}\n"
end,
make_protocal_Variable(Vars, VariableStr1, TypeList).
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
make_protocal_build(StructName) ->
"\n tb.build = function(byteArray)\n" ++
" byteArray.write_uint16(NetMsgType[\"msg_" ++ atom_to_list(StructName) ++ "\"])\n" ++
" return tb.encode(byteArray)\n" ++
" end\n".
make_protocal_create(_StructName)->
"\n return tb\n".
%%%%%%%%%%%%%%%%%%%%%%%%%%%%生成encode%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%% byteArray.write_uint16(account.size());
%% for(int i=0; i<account.size();i++)
%% {
%% byteArray.write_string(account[i]);
%% }
make_protocal_encode([], EncodeStr, _TypeList) ->
"\n tb.encode = function(byteArray)\n"
++ EncodeStr ++ " return byteArray\n"
++ " end\n\n";
make_protocal_encode([Struct | Vars], EncodeStr, TypeList) ->
EncodeStr1 = EncodeStr ++ make_protocal_encode_item(Struct, TypeList),
make_protocal_encode(Vars, EncodeStr1, TypeList).
%% 生成decode中的数组
make_protocal_encode_item({Type, Name}, TypeList) ->
case lists:keyfind(Type, 1, TypeList) of
{Type, base_type} ->
" \tbyteArray.write_"++ atom_to_list(Type) ++"(tb." ++ get_csharp_Var(Name) ++ ")\n";
{Type, string_type} ->
" \tbyteArray.write_"++ atom_to_list(Type) ++"(tb." ++ get_csharp_Var(Name) ++ ")\n";
{_Type, enum_type} ->
make_protocal_encode_item({int, Name}, TypeList);
_ ->
" tb." ++ get_csharp_Var(Name) ++ ".encode(byteArray);\n"
%%" " ++ get_csharp_Var(Name) ++ ".encode(byteArray)\n\n"
end;
make_protocal_encode_item({Type, Name, _DefaultValue}, TypeList)
when Type /= array->
make_protocal_encode_item({Type, Name}, TypeList);
make_protocal_encode_item({array, Type, Name}, TypeList) ->
case lists:keyfind(Type, 1, TypeList) of
{Type1, base_type} ->
make_protocal_encode_item_base_type(Name, Type1);
{Type1, string_type} ->
make_protocal_encode_item_base_type(Name, Type1);
{_Type, enum_type} ->
make_protocal_encode_item({array, int, Name}, TypeList);
_ ->
NameStr = get_csharp_Var(Name),
" byteArray.write_uint16(#(tb." ++ NameStr ++ "))\n" ++
" for k, v in pairs(tb." ++ NameStr ++ ") do\n" ++
" byteArray = v.encode(byteArray)\n" ++
" end\n"
end.
make_protocal_encode_item_base_type(Name, Type) ->
NameStr = get_csharp_Var(Name),
TypeStr = get_csharp_type(Type),
" byteArray.write_uint16(#(tb." ++ NameStr ++ "))\n" ++
" for k, v in pairs (tb." ++ NameStr ++ ") do\n" ++
" byteArray.write_" ++ TypeStr ++ "(v)\n" ++
" end\n".
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%生成decode%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
make_protocal_decode([], DecodeStr, _TypeList) ->
" tb.decode = function(byteArray)\n" ++ DecodeStr ++ " end\n";
make_protocal_decode([Struct | Vars], DecodeStr, TypeList) ->
DecodeStr1 = DecodeStr ++ make_protocal_decode_item(Struct, TypeList),
make_protocal_decode(Vars, DecodeStr1, TypeList).
make_protocal_decode_item({Type, Name}, TypeList) ->
case lists:keyfind(Type, 1, TypeList) of
{Type1, base_type} ->
" tb." ++ get_csharp_Var(Name) ++ " = byteArray.read_"++ atom_to_list(Type1) ++"();\n";
{Type1, string_type} ->
" tb." ++ get_csharp_Var(Name) ++ " = byteArray.read_"++ atom_to_list(Type1) ++"();\n";
{_Type, enum_type} ->
make_protocal_decode_item({int, Name}, TypeList);
_ ->
TypeString = get_csharp_type(Type),
" tb." ++ get_csharp_Var(Name) ++ " = " ++ TypeString ++ "();\n" ++
" tb." ++ get_csharp_Var(Name) ++ ".decode(byteArray);\n"
%%" " ++ get_csharp_Var(Name) ++ ".decode(byteArray);\n"
end;
make_protocal_decode_item({Type, Name, _DefaultValue}, TypeList)
when Type /= array->
make_protocal_decode_item({Type, Name}, TypeList);
make_protocal_decode_item({array, Type, Name}, TypeList) ->
case lists:keyfind(Type, 1, TypeList) of
{Type1, base_type} ->
make_protocal_encode_item_decode_type(Name, Type1);
{Type1, string_type} ->
make_protocal_encode_item_decode_type(Name, Type1);
{_Type, enum_type} ->
make_protocal_decode_item({array, int, Name}, TypeList);
_ ->
TypeString = get_csharp_type(Type),
NameStr = get_csharp_Var(Name),
CountString = "countOf" ++ NameStr,
_ArrayString = "arrayOf" ++ NameStr,
" local " ++ CountString ++ " = byteArray.read_uint16()\n" ++
" tb."++NameStr++" = {}\n" ++
" for i = 1, " ++ CountString ++ " do\n" ++
" local temp = " ++ TypeString ++ "()\n" ++
" temp.decode(byteArray)\n" ++
" table.insert(tb."++NameStr++", temp)\n" ++
" end\n"
end.
make_protocal_encode_item_decode_type(Name, Type) ->
NameStr = get_csharp_Var(Name),
TypeStr = atom_to_list(Type),
CountString = "countOf" ++ NameStr,
" local " ++ CountString ++ " = byteArray.read_uint16()\n" ++
" tb."++NameStr++" = {}\n" ++
" for i = 1, " ++ CountString ++ " do\n" ++
" table.insert(tb." ++ NameStr ++ ", byteArray.read_" ++ TypeStr ++ "())\n" ++
" end\n".
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%生成网络类型%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
make_protocal_msg_type([], MsgTypeStr, _Index) ->
"NetMsgType = \n" ++
"{\n" ++
MsgTypeStr ++
"}";
make_protocal_msg_type([Struct|Structs], MsgTypeStr, Index) ->
[StructName | _NewStructList] = Struct,
case Structs == [] of
true ->
make_protocal_msg_type(Structs, MsgTypeStr ++ " [\"msg_" ++ atom_to_list(StructName) ++ "\"] = "++integer_to_list(Index)++"\n", Index+1);
false ->
make_protocal_msg_type(Structs, MsgTypeStr ++ " [\"msg_" ++ atom_to_list(StructName) ++ "\"] = "++integer_to_list(Index)++",\n", Index+1)
end.
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%生成枚举类型%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
make_protocal_const(Ver) ->
"function get_proto_version()
return "++integer_to_list(Ver)++";
end\n".
make_protocal_enum([], EnumStr) ->
EnumStr;
make_protocal_enum([Enum | Enums], EnumStr) ->
{EnumName , EnumItems} = Enum,
EnumStr1 = EnumStr ++
string:to_lower(atom_to_list(EnumName)) ++ " = \n{\n" ++
make_protocal_enum_item(EnumItems, 1, "") ++ "}\n",
make_protocal_enum(Enums, EnumStr1).
make_protocal_enum_item([EnumItem | []], Index, EnumItemStr) ->
EnumItemStr ++ " [\"" ++ atom_to_list(EnumItem) ++ "\"] = "++integer_to_list(Index)++"\n";
make_protocal_enum_item([EnumItem | EnumItems], Index, EnumItemStr) ->
EnumItemStr1 = EnumItemStr ++ " [\"" ++ atom_to_list(EnumItem) ++ "\"]= "++integer_to_list(Index)++",\n",
make_protocal_enum_item(EnumItems, Index + 1, EnumItemStr1).
get_csharp_Var(Type)->
TypeString = atom_to_list(Type),
case lists:keyfind(TypeString, 1, get_keyword_mapping()) of
{TypeString, CSharpTypeString}->
CSharpTypeString;
_ ->
TypeString
end.
get_csharp_type(Type) ->
TypeString = atom_to_list(Type),
case lists:keyfind(TypeString, 1, get_cshapr_type_mapping()) of
{TypeString, CSharpTypeString}->
CSharpTypeString;
_ ->
TypeString
end.
get_cshapr_type_mapping()->
[
{"uint", "UInt"},
{"int16", "Int16"},
{"uint16", "UInt16"},
{"int64", "Int64"},
{"uint64", "uint64"},
{"uchar", "char"}
].
get_keyword_mapping() ->
[
{"lock", "Lock"},
{"params", "Params"}
].
make_getmsgid(StructName)->
" tb.getMsgID = function()\n" ++
" " ++
" return NetMsgType[\"msg_" ++ atom_to_list(StructName) ++ "\"]\n" ++
" end\n".
get_default_value(Type, DefaultValue)
when (Type == int orelse
Type == uint orelse
Type == uint64 orelse
Type == uint16 orelse
Type == int64 orelse
Type == short orelse
Type == int16)
andalso is_integer(DefaultValue)->
integer_to_list(DefaultValue);
get_default_value(Type, DefaultValue)
when (Type == float orelse
Type == long orelse
Type == double orelse
Type == uint16 orelse
Type == int64 orelse
Type == int16)
andalso is_float(DefaultValue)->
float_to_list(DefaultValue);
get_default_value(char, DefaultValue)
when is_list(DefaultValue)->
"'" ++ DefaultValue ++ "'";
get_default_value(string, DefaultValue)
when is_atom(DefaultValue)->
get_default_value(string, atom_to_list(DefaultValue));
get_default_value(string, DefaultValue)
when is_list(DefaultValue)->
"\"" ++ DefaultValue ++ "\"".
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%测试函数%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
test() ->
%% test_make_protocal_encode_item(),
%% test_make_protocal_decode_item(),
%% test_make_protocal_encode(),
%% test_make_protocal_decode().
start().
%% test_make_protocal_encode_item() ->
%% "\t\tbyteArray.write_uint16(result.size());\n\t\tfor(int i=0; i<result.size();i++)\n\t{\n\t\t\tresult[i].encode(byteArray);\n\t}\n"
%% = make_protocal_encode_item({array, notify_login_result, result}, gen_common:get_type()),
%% "\t\tbyteArray.write_uint16(account.size());\n\t\tfor(int i=0; i<account.size();i++)\n\t{\n\t\t\tbyteArray.write_string(account[i]);\n\t}\n"
%% = make_protocal_encode_item({array, string, account}, gen_common:get_type()),
%% "\t\tbyteArray.write_uint16(account.size());\n\t\tfor(int i=0; i<account.size();i++)\n\t{\n\t\t\tbyteArray.write_int(account[i]);\n\t}\n"
%% = make_protocal_encode_item({array, login_result, account}, gen_common:get_type()),
%% "\t\tbyteArray.write_string(account);\n" = make_protocal_encode_item({string, account}, gen_common:get_type()),
%% "\t\tbyteArray.write_int(account);\n" = make_protocal_encode_item({login_result, account}, gen_common:get_type()),
%% "\t\taccount.encode(byteArray);\n" = make_protocal_encode_item({notify_login_result, account}, gen_common:get_type()).
%%
%% test_make_protocal_decode_item() ->
%% "uint16 size = byteArray.read_uint16();\naccount.reserve(size);\nfor(int i=0; i<size;i++)\n{\n\taccount.push_back(byteArray.read_string());\n}\n"
%% = make_protocal_decode_item({array, string, account}, gen_common:get_type()),
%% "uint16 size = byteArray.read_uint16();\nresult.resize(size);\nfor(int i=0; i<size;i++)\n{\n\tresult[i].decode(byteArray);\n}\n"
%% = make_protocal_decode_item({array, notify_login_result, result}, gen_common:get_type()),
%% "uint16 size = byteArray.read_uint16();\naccount.reserve(size);\nfor(int i=0; i<size;i++)\n{\n\taccount.push_back(byteArray.read_int());\n}\n"
%% = make_protocal_decode_item({array, login_result, account}, gen_common:get_type()),
%% "account = byteArray.read_string();\n" = make_protocal_decode_item({string, account}, gen_common:get_type()),
%% "account = byteArray.read_int();\n" = make_protocal_decode_item({login_result, account}, gen_common:get_type()),
%% "account.decode(byteArray);\n" = make_protocal_decode_item({notify_login_result, account}, gen_common:get_type()).
%%
%% test_make_protocal_encode() ->
%% Result = make_protocal_encode([{array, string, account},
%% {string, pwd},
%% {player_data, data},
%% {array, notify_login_result, result}], "", gen_common:get_type()),
%% io:format("~p~n", [Result]).
%%
%% test_make_protocal_decode() ->
%% Result = make_protocal_decode([{array, string, account},
%% {string, pwd},
%% {player_data, data},
%% {array, notify_login_result, result}], "", gen_common:get_type()),
%% io:format("~p~n", [Result]).
%%
%% test_make_protocal_Variable() ->
%% Result = make_protocal_Variable([{array, string, account},
%% {string, pwd},
%% {player_data, data},
%% {array, notify_login_result, result}], "", gen_common:get_type()),
%% io:format("~p~n", [Result]).

Write Preview
Loading…
取消
儲存