Initial commit

include/basic_color_printer.tpp

@@ -0,0 +1,71 @@
+/**
+	roflcat
+	Copyright (C) 2019 rexy712
+
+	This program is free software: you can redistribute it and/or modify
+	it under the terms of the GNU General Public License as published by
+	the Free Software Foundation, either version 3 of the License, or
+	(at your option) any later version.
+
+	This program is distributed in the hope that it will be useful,
+	but WITHOUT ANY WARRANTY; without even the implied warranty of
+	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+	GNU General Public License for more details.
+
+	You should have received a copy of the GNU General Public License
+	along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#ifndef BASIC_COLOR_PRINTER_TPP
+#define BASIC_COLOR_PRINTER_TPP
+
+#include "basic_color_printer.hpp"
+
+#ifndef _XOPEN_SOURCE
+#define _XOPEN_SOURCE //for wcwidth
+#endif
+#include <cwchar> //wcwidth
+
+#include <cstdio> //wprintf, putwchar
+
+template<int I>
+constexpr basic_color_printer<I>::basic_color_printer(const cmd_args& args)noexcept:
+	color_printer_base<basic_color_printer<I>>(args){}
+template<int I>
+basic_color_printer<I>& basic_color_printer<I>::_print(const wchar_t s)noexcept{
+	if(s == L'\n'){
+		inc_line();
+	}else{
+		int width = wcwidth(s);
+		if(width > 0)
+			inc_col(wcwidth(s));
+		else
+			inc_col(1);
+	}
+	long new_index = static_cast<int>(this->m_freq*m_col + this->m_spread*m_line) % color_lookup_len;
+	if(new_index < 0)
+		new_index += color_lookup_len;
+	if(new_index != m_index){
+		m_index = new_index;
+		wprintf(L"\033[38;5;%sm", color_lookup_table[m_index]);
+	}
+	putwchar(s);
+	return *this;
+}
+template<int I>
+constexpr basic_color_printer<I>& basic_color_printer<I>::_reset(void)noexcept{
+	color_printer_base<basic_color_printer<I>>::_reset();
+	m_index = -1; //force color update on next print
+	return *this;
+}
+template<int I>
+constexpr void basic_color_printer<I>::inc_col(int i){
+	m_col += i;
+}
+template<int I>
+constexpr void basic_color_printer<I>::inc_line(void){
+	++m_line;
+	m_col = 0;
+}
+
+#endif