1 files changed, 18 insertions, 6 deletions
diff --git a/StarsEx/DataSource.cpp b/StarsEx/DataSource.cpp
index f352e05..06c7658 100644
--- a/StarsEx/DataSource.cpp
+++ b/StarsEx/DataSource.cpp
@@ -15,9 +15,12 @@
 #include "Text.h"
 #include "Utils.h"
 
+using Group = DataSource::Group;
 
-DataSource::DataSource() :
-    m_prefix {""}
+
+DataSource::DataSource(Group group) :
+    m_prefix {""},
+    m_group {group}
 {
 }
 
@@ -27,6 +30,13 @@ DataSource::~DataSource()
 }
 
 
+Group
+DataSource::GetGroup() const
+{
+    return m_group;
+}
+
+
 void
 DataSource::SetPrefix(const char* prefix)
 {
@@ -37,13 +47,14 @@ DataSource::SetPrefix(const char* prefix)
 }
 
 
-ArchiveDataSource::ArchiveDataSource(const char* name) :
-    ArchiveDataSource(new DataArchive(name))
+ArchiveDataSource::ArchiveDataSource(const char* name, Group group) :
+    ArchiveDataSource(new DataArchive(name), group)
 {
 }
 
 
-ArchiveDataSource::ArchiveDataSource(DataArchive* archive) :
+ArchiveDataSource::ArchiveDataSource(DataArchive* archive, Group group) :
+    DataSource(group),
     m_archive {archive}
 {
 }
@@ -94,7 +105,8 @@ ArchiveDataSource::Load(const char* name, std::uint8_t*& buf, bool null_terminat
 }
 
 
-FileSystemDataSource::FileSystemDataSource(const char* path) :
+FileSystemDataSource::FileSystemDataSource(const char* path, Group group) :
+    DataSource(group),
     m_path {path}
 {
 }