public override void OnInspectorGUI()
        {
            base.OnInspectorGUI();

            if (GUILayout.Button("Create proto file"))
            {
                UniBufHelper.GenerateProtoFile(GetTypeModel(), _config.ProtoFilePath);
            }

            if (GUILayout.Button("Compile model"))
            {
                UniBufHelper.CompileModel(GetTypeModel());
            }
        }
示例#2
0
        public static RuntimeTypeModel CreateTypeModel(IEnumerable <Type> types)
        {
            RuntimeTypeModel typeModel = TypeModel.Create();

            foreach (var t in types)
            {
                var contract = t.GetCustomAttributes(typeof(ProtoContractAttribute), false);
                if (contract.Length > 0)
                {
                    MetaType metaType = typeModel.Add(t, true);

                    // support ISerializationCallbackReceiver
                    if (typeof(ISerializationCallbackReceiver).IsAssignableFrom(t))
                    {
                        MethodInfo beforeSerializeMethod  = t.GetMethod("OnBeforeSerialize");
                        MethodInfo afterDeserializeMethod = t.GetMethod("OnAfterDeserialize");

                        metaType.SetCallbacks(beforeSerializeMethod, null, null, afterDeserializeMethod);
                    }
                }
            }

            UniBufConfig   config  = UniBufHelper.GetConfig();
            HashSet <Type> hashSet = new HashSet <Type>();

            foreach (MetaType metaType in typeModel.GetTypes().Cast <MetaType>().ToArray())
            {
                hashSet.Add(metaType.Type);
                OverrideMetaType(metaType, config.TypeOverride);
                foreach (ValueMember valueMember in metaType.GetFields())
                {
                    if (valueMember.MemberType.IsPrimitive || valueMember.MemberType == typeof(string))
                    {
                        continue;
                    }
                    if (!hashSet.Contains(valueMember.MemberType))
                    {
                        MetaType memberType = typeModel.Add(valueMember.MemberType, false);
                        hashSet.Add(memberType.Type);
                        OverrideMetaType(memberType, config.TypeOverride);
                    }
                }
            }

            return(typeModel);
        }
        public void OnPreprocessBuild(BuildReport report)
        {
            UniBufConfig config = UniBufHelper.GetConfig();
            bool         isTest =
                (report.summary.options & BuildOptions.IncludeTestAssemblies) == BuildOptions.IncludeTestAssemblies;

            var typeModel = TypeModelFactory.CreateTypeModel(UnityAssembliesUtils.GetAllRuntimeTypes(isTest));

            if (config.AutoCompileModel)
            {
                UniBufHelper.CompileModel(typeModel);
            }

            if (config.AutoGenerateProtoFile)
            {
                UniBufHelper.GenerateProtoFile(typeModel, config.ProtoFilePath);
            }
        }