1 files changed, 33 insertions, 0 deletions
diff --git a/aocharbutton.cpp b/aocharbutton.cpp
new file mode 100644
index 0000000..1c0b4f7
--- /dev/null
+++ b/aocharbutton.cpp
@@ -0,0 +1,33 @@
+#include "aocharbutton.h"
+
+#include "path_functions.h"
+#include "file_functions.h"
+
+#include <QFile>
+
+AOCharButton::AOCharButton(QWidget *parent)
+{
+  m_parent = parent;
+
+  this->resize(60, 60);
+}
+
+void AOCharButton::set_image(QString p_character)
+{
+  QString image_path = get_character_path(p_character) + "char_icon.png";
+  QString legacy_path = get_demothings_path() + p_character.toLower() + "_char_icon.png";
+
+  if (file_exists(image_path))
+    this->setStyleSheet("border-image:url(\"" + image_path + "\")");
+  else if (file_exists(legacy_path))
+  {
+    this->setStyleSheet("border-image:url(\"" + legacy_path + "\")");
+    //ninja optimization
+    QFile::copy(legacy_path, image_path);
+  }
+  else
+  {
+    this->setStyleSheet("border-image:url()");
+    this->setText(p_character);
+  }
+}