X-Git-Url: http://git.mutantstargoat.com/user/nuclear/?p=laserbrain_demo;a=blobdiff_plain;f=src%2Fdatamap.cc;h=dcbae788b0cc2fe3d7b0ea6f8b0154a701ceab86;hp=e466082ec0949500bbe5b47c1b8732cf21be396a;hb=6ef619c6d92c698728576a4ec1c798a0f716d9a4;hpb=b30241a8a51be904b22459a1d0cc3322e0a505d9

diff --git a/src/datamap.cc b/src/datamap.cc
index e466082..dcbae78 100644
--- a/src/datamap.cc
+++ b/src/datamap.cc
@@ -1,9 +1,10 @@
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
+#include <algorithm>
 #include "datamap.h"
 
-#ifdef WIN32
+#if defined(WIN32) || defined(__WIN32__)
 #include <malloc.h>
 #else
 #include <alloca.h>
@@ -11,6 +12,11 @@
 
 static char *clean_line(char *s);
 
+DataMap::DataMap()
+{
+	strip_paths = false;
+}
+
 void DataMap::clear()
 {
 	dmap.clear();
@@ -22,6 +28,11 @@ void DataMap::set_path(const char *path)
 	root = std::string(path);
 }
 
+void DataMap::set_strip(bool s)
+{
+	strip_paths = s;
+}
+
 bool DataMap::load_map(const char *fname)
 {
 	std::string path = root.empty() ? fname : root + std::string("/") + fname;
@@ -91,6 +102,13 @@ int DataMap::lookup(const char *in, char *buf, int bsz) const
 {
 	std::string res;
 
+	if(strip_paths) {
+		const char *ptr = strrchr(in, '/');
+		if(ptr) {
+			in = ptr + 1;
+		}
+	}
+
 	char *inbuf = (char*)alloca(strlen(in) + 1);
 	strcpy(inbuf, in);
 	in = clean_line(inbuf);