Allow preambles to omit the tikzit-specific preview stuff

The editable preambles include a bunch of things that *have* to be there
for previews to work, which makes figuring out what you can edit hard.

We now add a \documentclass automagically if it is missing, and if
\begin{document} is missing, we assume the preview-specific stuff should
be added.

OSX changes are untested.

2 files changed, 40 insertions, 10 deletions

diff --git a/tikzit/src/common/Preambles.h b/tikzit/src/common/Preambles.h
index 95081c4..2fb084a 100644
--- a/tikzit/src/common/Preambles.h
+++ b/tikzit/src/common/Preambles.h
@@ -25,6 +25,8 @@
 #import <Foundation/Foundation.h>
 #import "StyleManager.h"
 
+@class Graph;
+
 @interface Preambles : NSObject {
 	NSMutableDictionary *preambleDict;
 	NSString *selectedPreambleName;
@@ -63,6 +65,9 @@
 - (NSString*)defaultPreambleName;
 - (NSString*)currentPostamble;
 
+- (NSString*)buildDocumentForTikz:(NSString*)tikz;
+- (NSString*)buildDocumentForGraph:(Graph*)g;
+
 @end
 
 // vi:ft=objc:noet:ts=4:sts=4:sw=4
diff --git a/tikzit/src/common/Preambles.m b/tikzit/src/common/Preambles.m
index d6d18e9..651bbcd 100644
--- a/tikzit/src/common/Preambles.m
+++ b/tikzit/src/common/Preambles.m
@@ -24,14 +24,13 @@
 #import "Preambles.h"
 #import "NodeStyle.h"
 #import "EdgeStyle.h"
+#import "Graph.h"
 
-static NSString *PREAMBLE_HEAD =
-@"\\documentclass{article}\n"
+static NSString *DEF_PREAMBLE_START =
 @"\\usepackage[svgnames]{xcolor}\n"
 @"\\usepackage{tikz}\n"
 @"\\usetikzlibrary{decorations.markings}\n"
 @"\\usetikzlibrary{shapes.geometric}\n"
-@"\\pagestyle{empty}\n"
 @"\n"
 @"\\pgfdeclarelayer{edgelayer}\n"
 @"\\pgfdeclarelayer{nodelayer}\n"
@@ -41,6 +40,7 @@ static NSString *PREAMBLE_HEAD =
 
 static NSString *PREAMBLE_TAIL =
 @"\n"
+@"\\pagestyle{empty}\n"
 @"\\usepackage[graphics,tightpage,active]{preview}\n"
 @"\\PreviewEnvironment{tikzpicture}\n"
 @"\\newlength{\\imagewidth}\n"
@@ -71,6 +71,13 @@ static NSString *POSTAMBLE =
 	return self;
 }
 
+- (void)dealloc {
+	[selectedPreambleName release];
+	[styles release];
+	[styleManager release];
+	[super dealloc];
+}
+
 - (NSString*)preambleForName:(NSString*)name {
 	if ([name isEqualToString:@"default"])
 		return [self defaultPreamble];
@@ -152,8 +159,8 @@ static NSString *POSTAMBLE =
 }
 
 - (NSString*)defaultPreamble {
-	return [NSString stringWithFormat:@"%@%@%@",
-			PREAMBLE_HEAD, [self styleDefinitions], PREAMBLE_TAIL];
+	return [NSString stringWithFormat:@"%@%@",
+			DEF_PREAMBLE_START, [self styleDefinitions]];
 }
 
 - (BOOL)selectedPreambleIsDefault {
@@ -255,11 +262,29 @@ static NSString *POSTAMBLE =
 	return YES;
 }
 
-- (void)dealloc {
-	[selectedPreambleName release];
-	[styles release];
-	[styleManager release];
-	[super dealloc];
+- (NSString*)buildDocumentForTikz:(NSString*)tikz
+{
+	NSString *preamble = [self currentPreamble];
+	NSString *doc_head = @"";
+	if (![preamble hasPrefix:@"\\documentclass"]) {
+		doc_head = @"\\documentclass{article}\n";
+	}
+	NSString *preamble_suffix = @"";
+	if ([preamble rangeOfString:@"\\begin{document}"
+						options:NSBackwardsSearch].length == 0) {
+		preamble_suffix = PREAMBLE_TAIL;
+	}
+    return [NSString stringWithFormat:@"%@%@%@%@%@",
+			 doc_head,
+             [self currentPreamble],
+			 preamble_suffix,
+             tikz,
+             POSTAMBLE];
+}
+
+- (NSString*)buildDocumentForGraph:(Graph*)g
+{
+	return [self buildDocumentForTikz:[g tikz]];
 }
 
 @end