--- a/CustomCodeGeneratorOrRefactoring.st	Wed Sep 17 23:01:25 2014 +0200
+++ b/CustomCodeGeneratorOrRefactoring.st	Sat Sep 20 19:05:41 2014 +0200
@@ -3,7 +3,7 @@
 Object subclass:#CustomCodeGeneratorOrRefactoring
 	instanceVariableNames:'compositeChangeCollector compositeChangeNesting userPreferences
 		confirmChanges generateComments dialog changeManager codeBuilder
-		refactoryBuilder'
+		refactoryBuilder formatter'
 	classVariableNames:''
 	poolDictionaries:''
 	category:'Interface-Refactoring-Custom'
@@ -189,6 +189,20 @@
     "Created: / 11-05-2014 / 00:27:49 / Jakub Nesveda <nesvejak@fit.cvut.cz>"
 !
 
+formatter
+
+    ^ formatter
+
+    "Created: / 19-09-2014 / 22:18:33 / Jakub Nesveda <nesvejak@fit.cvut.cz>"
+!
+
+formatter: aSourceCodeFormatter
+
+    formatter := aSourceCodeFormatter
+
+    "Created: / 19-09-2014 / 22:18:50 / Jakub Nesveda <nesvejak@fit.cvut.cz>"
+!
+
 refactoryBuilder
 
     ^ refactoryBuilder
@@ -203,6 +217,18 @@
     "Modified (format): / 23-08-2014 / 00:14:33 / Jakub Nesveda <nesvejak@fit.cvut.cz>"
 !
 
+sourceCodeGenerator    
+    "Returns initialized source code generator"
+    | sourceCodeGenerator |
+
+    sourceCodeGenerator := CustomSourceCodeGenerator new.
+    sourceCodeGenerator formatter: formatter.
+    ^ sourceCodeGenerator.
+
+    "Created: / 19-09-2014 / 20:56:22 / Jakub Nesveda <nesvejak@fit.cvut.cz>"
+    "Modified: / 19-09-2014 / 22:11:02 / Jakub Nesveda <nesvejak@fit.cvut.cz>"
+!
+
 userPreferences
 
     ^ userPreferences
@@ -390,13 +416,14 @@
     userPreferences := UserPreferences current.
     generateComments := userPreferences generateComments.
 
+    self setUpFormatter.
     self setUpCodeBuilder.
     self setUpRefactoryBuilder.
     self setUpDialog.
     self setUpChangeManager
 
     "Created: / 17-03-2014 / 22:27:32 / Jakub Nesveda <nesvejak@fit.cvut.cz>"
-    "Modified: / 17-09-2014 / 22:43:54 / Jakub Nesveda <nesvejak@fit.cvut.cz>"
+    "Modified: / 18-09-2014 / 23:12:42 / Jakub Nesveda <nesvejak@fit.cvut.cz>"
 !
 
 setUpChangeManager
@@ -421,6 +448,13 @@
     "Created: / 09-06-2014 / 22:57:08 / Jakub Nesveda <nesvejak@fit.cvut.cz>"
 !
 
+setUpFormatter
+
+    formatter := CustomRBLocalSourceCodeFormatter new
+
+    "Created: / 18-09-2014 / 23:12:42 / Jakub Nesveda <nesvejak@fit.cvut.cz>"
+!
+
 setUpRefactoryBuilder
 
     refactoryBuilder := CustomRefactoryBuilder new.