contrib/objcwrapper: move `model` to an attribute of `CodeGenerator`

[nit.git] / contrib / objcwrapper / src / objc_generator.nit

diff --git a/contrib/objcwrapper/src/objc_generator.nit b/contrib/objcwrapper/src/objc_generator.nit
index db273cb..6ca50a2 100644 (file)
--- a/contrib/objcwrapper/src/objc_generator.nit
+++ b/contrib/objcwrapper/src/objc_generator.nit
@@ -63,10 +63,17 @@ redef class Sys
        end
 end
 
+# Wrapper generator
 class CodeGenerator
+
+       # `ObjcModel` to wrap
+       var model: ObjcModel
+
        # Generate Nit code to wrap `classes`
-       fun generate(classes: Array[ObjcClass])
+       fun generate
        do
+               var classes = model.classes
+
                # Open specified path or stdin
                var file
                var path = opt_output.value