X-Git-Url: https://git.draconx.ca/gitweb/upkg.git/blobdiff_plain/2225a1b8432cfaf640b9b69c1dc961322eda8f54..37a77c2825545435cfb3093eae1995efd35c20ba:/src/engine/music.c

diff --git a/src/engine/music.c b/src/engine/music.c
index 87a549b..f64513e 100644
--- a/src/engine/music.c
+++ b/src/engine/music.c
@@ -1,9 +1,30 @@
+/*
+ *  upkg: tool for manipulating Unreal Tournament packages.
+ *  Copyright (C) 2009 Nick Bowler
+ *
+ *  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 2 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, write to the Free Software
+ *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ */
+
 #include <stdio.h>
 #include <string.h>
 #include <glib-object.h>
 
 #include "exportable.h"
+#include "loadable.h"
 #include "uobject.h"
+#include "music-module.h"
 #include "music.h"
 #include "upkg.h"
 
@@ -12,40 +33,81 @@
 
 struct music_priv {
 	struct upkg_file *f;
+	struct music_mod *mod;
+	unsigned loaded;
 };
 
 static void exportable_init(UObjectExportable *);
+static void loadable_init(UObjectLoadable *);
 
 G_DEFINE_DYNAMIC_TYPE_EXTENDED(EngineMusic, engine_music, U_OBJECT_TYPE, 0,
 	G_IMPLEMENT_INTERFACE(U_OBJECT_TYPE_EXPORTABLE, exportable_init)
+	G_IMPLEMENT_INTERFACE(U_OBJECT_TYPE_LOADABLE, loadable_init)
 );
 
-static int export(GObject *o, FILE *f)
+static int load(GObject *o)
 {
 	struct music_priv *priv = MUSIC_GET_PRIV(o);
-	unsigned char buf[1024];
 
-	if (!priv->f || upkg_export_seek(priv->f, 0, SEEK_SET) != 0)
-		return -1;
+	if (!priv->loaded) {
+		g_return_val_if_fail(priv->f != NULL, -1);
 
-	while (!priv->f->eof) {
-		size_t rc = upkg_export_read(priv->f, buf, sizeof buf);
-		if (rc == 0) {
-			if (priv->f->eof) break;
+		if (upkg_export_seek(priv->f, 0, SEEK_SET) != 0) {
 			return -1;
 		}
 
-		if (fwrite(buf, 1, rc, f) != rc)
+		priv->mod = music_mod_open(priv->f);
+		if (!priv->mod) {
 			return -1;
+		}
 	}
 
+	priv->loaded++;
 	return 0;
 }
 
+static void unload(GObject *o)
+{
+	struct music_priv *priv = MUSIC_GET_PRIV(o);
+	g_return_if_fail(priv->loaded > 0);
+
+	if (--priv->loaded == 0) {
+		music_mod_close(priv->mod);
+	}
+}
+
+static int export(GObject *o, FILE *f)
+{
+	struct music_priv *priv = MUSIC_GET_PRIV(o);
+	int rc;
+
+	if (load(o) != 0)
+		return -1;
+
+	rc = music_mod_dump(priv->mod, f);
+
+	unload(o);
+
+	return rc;
+}
+
 static int export_name(GObject *o, char *buf, size_t n)
 {
 	struct music_priv *priv = MUSIC_GET_PRIV(o);
-	return snprintf(buf, n, "%s", priv->f ? priv->f->name : "");
+	const char *type;
+	int rc;
+
+	if (load(o) != 0) {
+		if (n > 0) *buf = 0;
+		return 0;
+	}
+
+	type = music_mod_type(priv->mod);
+	rc = snprintf(buf, n, "%s.%s", priv->f->name, type);
+
+	unload(o);
+
+	return rc;
 }
 
 static void exportable_init(UObjectExportable *e)
@@ -54,10 +116,15 @@ static void exportable_init(UObjectExportable *e)
 	e->export_name = export_name;
 }
 
+static void loadable_init(UObjectLoadable *l)
+{
+	l->load   = load;
+	l->unload = unload;
+}
+
 static int deserialize(UObject *o, struct upkg_file *f)
 {
 	struct music_priv *priv = MUSIC_GET_PRIV(o);
-	EngineMusic *m = ENGINE_MUSIC(o);
 	size_t rc, pos = 0, buflen;
 	unsigned char buf[32];
 	long size;
@@ -86,7 +153,6 @@ static int deserialize(UObject *o, struct upkg_file *f)
 	f->base += pos;
 	f->len   = size;
 	upkg_export_seek(f, 0, SEEK_SET);
-
 	priv->f = f;
 
 	return 0;
@@ -99,17 +165,33 @@ void music_register(GTypeModule *m)
 
 static void engine_music_init(EngineMusic *m)
 {
+	struct music_priv *priv = MUSIC_GET_PRIV(m);
+	*priv = (struct music_priv){0};
+}
+
+static void engine_music_finalize(GObject *o)
+{
+	struct music_priv *priv = MUSIC_GET_PRIV(o);
+
+	if (priv->loaded >= 1) {
+		priv->loaded = 1;
+		unload(o);
+	}
 }
 
 static void engine_music_class_init(EngineMusicClass *class)
 {
 	UObjectClass *uo = U_OBJECT_CLASS(class);
-
+	GObjectClass *go = G_OBJECT_CLASS(class);
 	g_type_class_add_private(class, sizeof (struct music_priv));
 
 	uo->deserialize = deserialize;
+	go->finalize    = engine_music_finalize;
+
+	music_mod_init();
 }
 
 static void engine_music_class_finalize(EngineMusicClass *class)
 {
+	music_mod_exit();
 }