[not really working] Started work on OStream

5 files changed, 118 insertions, 10 deletions

diff --git a/Source/Library/Common/OStream.proto.cpp b/Source/Library/Common/OStream.proto.cpp
new file mode 100644
index 0000000..bbf5dc3
--- /dev/null
+++ b/Source/Library/Common/OStream.proto.cpp
@@ -0,0 +1,60 @@
+#include <OStream.proto.h>
+
+OStream::OStream() {
+	m_buffer = NULL;
+	m_last = NULL;
+}
+
+OStream::OStream(const OStream& other) {
+	m_buffer = NULL;
+	m_last = NULL;
+}
+
+void OStream::put(const String &s) {
+	waitLock();
+	if (m_buffer == NULL or m_last == NULL) {
+		m_buffer = m_last = new SimpleList<String>(s);
+	} else {
+		m_last = m_last->addAtEnd(s);
+	}
+	unlock();
+}
+
+void OStream::flush() {
+	if (m_buffer == NULL or m_last == NULL) return;
+
+	waitLock();
+	//Calculate buffer size
+	int size = 0;
+	SimpleList<String>* iter = m_buffer;
+	for (; iter != 0; iter = iter->next()) {
+		size += iter->v().size();
+	}
+	//Get stuff
+	String buff(WChar(" "), size);
+	iter = m_buffer;
+	for (int i = 0, pos = 0; i < size; i++, pos++) {
+		if (pos >= iter->v().size()) {
+			iter = iter->next();
+			pos = 0;
+		}
+		buff[i] = iter->v()[pos];
+	}
+	//Write it
+	write(buff);
+
+	delete m_buffer;
+	m_buffer = NULL;
+	m_last = NULL;
+
+	unlock();
+}
+
+OStream& OStream::operator<< (ostream_modifiers_e m) {
+	if (m == FLUSH) {
+		flush();
+	} else if (m == ENDL) {
+		put("\n");
+	}
+	return *this;
+}
diff --git a/Source/Library/Common/OStream.proto.h b/Source/Library/Common/OStream.proto.h
new file mode 100644
index 0000000..2fc68e1
--- /dev/null
+++ b/Source/Library/Common/OStream.proto.h
@@ -0,0 +1,38 @@
+#ifndef DEF_ISTREAM_PROTO_H
+#define DEF_ISTREAM_PROTO_H
+
+#include <String.class.h>
+#include <SimpleList.class.h>
+#include <Mutex.class.h>
+
+enum ostream_modifiers_e {
+	FLUSH,
+	ENDL
+};
+
+class OStream : private Mutex {
+	private:
+	SimpleList<String> *m_buffer;
+	SimpleList<String> *m_last;
+	void operator =(OStream& other);
+
+	protected:
+	virtual void write(const String &s) = 0;
+	
+	public:
+	OStream();
+	OStream(const OStream& other);
+	virtual ~OStream() { flush(); }
+
+	void put(const String& s);
+	void flush();
+	
+	//Formatting functions
+	OStream& operator << (const String& s) { put(s); if (s.contains("\n")) flush(); return *this; }
+	OStream& operator << (s64int i) { put(String::number(i)); return *this; }
+	OStream& operator << (s32int i) { put(String::number(i)); return *this; }
+	OStream& operator << (u32int i) { put(String::hex(i)); return *this; }
+	OStream& operator << (ostream_modifiers_e m);
+};
+
+#endif
diff --git a/Source/Library/Common/SimpleList.class.h b/Source/Library/Common/SimpleList.class.h
index 8fcd834..2bbbace 100644
--- a/Source/Library/Common/SimpleList.class.h
+++ b/Source/Library/Common/SimpleList.class.h
@@ -25,6 +25,19 @@ class SimpleList {
 		return new SimpleList<T>(value, this);
 	}
 
+	SimpleList<T>* addAtEnd(const T& value) {
+		if (this == 0) {
+			return new SimpleList<T>(value);
+		} else {
+			if (m_next == 0) {
+				m_next = new SimpleList<T>(value);
+				return m_next;
+			} else {
+				return m_next->addAtEnd(value);
+			}
+		}
+	}
+
 	SimpleList<T>* next() {
 		return m_next;
 	}
diff --git a/Source/Library/Makefile b/Source/Library/Makefile
index b3040d4..ed65ac4 100644
--- a/Source/Library/Makefile
+++ b/Source/Library/Makefile
@@ -19,6 +19,7 @@ Objects = Common/WChar.class.uo \
 		  Common/TextFile.class.uo \
 		  Common/ByteArray.class.uo \
 		  Common/Rand.ns.uo \
+		  Common/OStream.proto.uo \
 		  Common/cppsupport.wtf.uo \
 		  Userland/App/ShellApp.proto.uo \
 		  Userland/Syscall/Syscall.wtf.uo \
diff --git a/Source/Library/Userland/Binding/VirtualTerminal.class.h b/Source/Library/Userland/Binding/VirtualTerminal.class.h
index 70c6b23..df44124 100644
--- a/Source/Library/Userland/Binding/VirtualTerminal.class.h
+++ b/Source/Library/Userland/Binding/VirtualTerminal.class.h
@@ -6,10 +6,10 @@
 #include <VirtualTerminal.iface.h>
 #include <Kbd.iface.h>
 
-#include <String.class.h>
+#include <OStream.proto.h>
 #include <WChar.class.h>
 
-class VirtualTerminal : public RessourceCaller {
+class VirtualTerminal : public RessourceCaller, public OStream {
 	public:
 	static VirtualTerminal getIn() {
 		u32int id = RessourceCaller::sCall(VTIF_OBJTYPE, VTIF_SGETPRINVT);
@@ -21,13 +21,13 @@ class VirtualTerminal : public RessourceCaller {
 	}
 	VirtualTerminal(u32int id) : RessourceCaller(id, VTIF_OBJTYPE) {}
 
-	void writeHex(u32int number) {
+	/*void writeHex(u32int number) {
 		doCall(VTIF_WRITEHEX, number);
 	}
 	void writeDec(s64int number) {
 		doCall(VTIF_WRITEDEC, (number >> 32), number);
-	}
-	void write(String s) {
+	}*/
+	void write(const String &s) {
 		doCall(VTIF_WRITE, (u32int)&s);
 	}
 	keypress_t getKeypress(bool show = true, bool block = true) {
@@ -35,6 +35,7 @@ class VirtualTerminal : public RessourceCaller {
 		return *ptr;
 	}
 	String readLine(bool show = true) {
+		flush();
 		return String::unserialize(doCall(VTIF_READLINE, (show ? 1 : 0)));
 	}
 	void setColor(u8int fg, u8int bg = 0xFF) {
@@ -64,11 +65,6 @@ class VirtualTerminal : public RessourceCaller {
 	void put(u8int line, u8int col, WChar c) {
 		moveCursor(line, col); put(c);
 	}
-	
-	inline VirtualTerminal& operator<<(const String& s) { write(s); return *this; }
-	inline VirtualTerminal& operator<<(s32int i) { writeDec(i); return *this; }
-	inline VirtualTerminal& operator<<(s64int i) { writeDec(i); return *this; }
-	inline VirtualTerminal& operator<<(u32int i) { writeHex(i); return *this; }
 };
 
 #endif