private void GenerateInitObjects(GeneratorExecutionContext context, CreateOnInitSyntaxReceiver createOnInitSyntaxReceiver)
        {
            //Get the CreateOnInit attribute, and make sure it exits. The only assembly that will have this is the main TC one
            INamedTypeSymbol onInitAttribute = context.Compilation.GetTypeByMetadataName("Team_Capture.CreateOnInit");

            if (onInitAttribute == null)
            {
                return;
            }

            //TODO: Check if class inherits from mono behaviour
            INamedTypeSymbol monoBehaviourSymbol = context.Compilation.GetTypeByMetadataName("UnityEngine.MonoBehaviour");

            if (monoBehaviourSymbol == null)
            {
                return;
            }

            SemanticModel model = null;

            foreach (ClassDeclarationSyntax classDeclaration in createOnInitSyntaxReceiver.Classes)
            {
                //Get model
                if (model == null || model.SyntaxTree != classDeclaration.SyntaxTree)
                {
                    model = context.Compilation.GetSemanticModel(classDeclaration.SyntaxTree);
                }

                ISymbol classSymbol = ModelExtensions.GetDeclaredSymbol(model, classDeclaration);

                //Make sure the class has the CreateOnInit attribute
                AttributeData attributeData = classSymbol?.GetAttributes().FirstOrDefault(x =>
                                                                                          SymbolEqualityComparer.Default.Equals(x.AttributeClass, onInitAttribute));
                if (attributeData == null)
                {
                    continue;
                }

                //Check if the class is partial or not
                bool isPartial = classDeclaration.Modifiers
                                 .Any(m => m.IsKind(SyntaxKind.PartialKeyword));
                if (!isPartial)
                {
                    context.ReportDiagnostic(Diagnostic.Create(DiagnosticHelper.NotPartialDiagnostic, Location.None));
                    continue;
                }

                string visibility = classDeclaration.Modifiers.GetIfModiferIncludesPublicOrInternal();
                string className  = classSymbol.Name;

                //TODO: We should check that the argument exists and has no arguments
                //Get what method to call on init
                TypedConstant callOnInitData = attributeData.GetArgument(CallOnInitArgumentName);
                string        callOnInit     = callOnInitData.GetValueOrDefault <string>(null);

                //Get the object's name
                TypedConstant objectNameOverride = attributeData.GetArgument(ObjectNameOverrideName);
                string        objectName         = objectNameOverride.GetValueOrDefault(className);

                //Get what our load type is
                TypedConstant             loadTypeData = attributeData.GetArgument(LoadTypeName);
                RuntimeInitializeLoadType loadType     = loadTypeData.GetValueOrDefault(RuntimeInitializeLoadType.BeforeSceneLoad);

                //Generate code
                string code = Template.Parse(CodeTemplates.CreateOnInitTemplate).Render(new
                {
                    Namespace    = classSymbol.ContainingNamespace.ToString(),
                    Visiblity    = visibility,
                    Classname    = className,
                    Loadtype     = loadType.ToString(),
                    Objectname   = objectName,
                    Methodtocall = callOnInit == null ? string.Empty : $"{callOnInit}();"
                });

                context.AddSource($"{className}.gen", code);
            }
        }