/*
 * Copyright (c) 2005-2010, Haiku, Inc.
 * Distributed under the terms of the MIT license.
 *
 * Author:
 *		DarkWyrm <darkwyrm@gmail.com>
 */
#include "ResourceData.h"
#include "ResFields.h"
#include <stdlib.h>

ResourceData::ResourceData(void)
  :	fType(0),
  	fTypeString("Invalid"),
  	fID(-1),
  	fIDString("Invalid"),
  	fName(""),
  	fData(NULL),
  	fLength(0),
  	fAttr(false)
{
}


ResourceData::ResourceData(const type_code &code, const int32 &id,
							const char *name, char *data,
							const size_t &length)
  :	fType(code),
	fID(id),
	fName(name),
	fData(data),
	fLength(length),
  	fAttr(false)
{
	fIDString = "";
	fIDString << fID;
	fTypeString = MakeTypeString(code);
}


ResourceData::ResourceData(const ResourceData &data)
{
	*this = data;
}


ResourceData::~ResourceData(void)
{
	free(fData);
}


ResourceData &
ResourceData::operator=(const ResourceData &data)
{
	fType = data.fType;
	fTypeString = data.fTypeString;
	fID = data.fID;
	fIDString = data.fIDString;
	fName = data.fName;
	fAttr = data.fAttr;
	SetData(data.fData, data.fLength);
	
	return *this;
}

bool
ResourceData::SetFromResource(const int32 &index, BResources &res)
{
	char *name;
	if (!res.GetResourceInfo(index, (type_code*)&fType, &fID,
							(const char **)&name, &fLength)) {
		*this = ResourceData();
		return false;
	}
	fName = name;
	fTypeString = MakeTypeString(fType);
	fIDString = "";
	fIDString << fID;
	fAttr = false;
	char *data = (char *)res.LoadResource(fType, fID, &fLength);
	SetData(data, fLength);
	
	return true;
}


bool
ResourceData::SetFromAttribute(const char *name, BNode &node)
{
	attr_info info;
	if (node.GetAttrInfo(name, &info) != B_OK) {
		*this = ResourceData();
		return false;
	}
	
	fType = info.type;
	fID = -1;
	fIDString = "(attr)";
	fName = name;
	fLength = info.size;
	fAttr = true;

	fTypeString = MakeTypeString(fType);
	
	fData = (char *)malloc(fLength);
	if (fData) {
		ssize_t size = node.ReadAttr(name, info.type, 0, (void*)fData, fLength);
		if (size >= 0) {
			fLength = (size_t) size;
			return true;
		}
	}
	
	*this = ResourceData();
	return false;
}


void
ResourceData::SetTo(const type_code &code, const int32 &id,
					const char *name, char *data, const size_t &length)
{
	fType = code;
	fTypeString = MakeTypeString(code);
	fID = id;
	fIDString = "";
	fIDString << fID;
	fName = name;
	SetData(data, length);
	
}


void
ResourceData::SetType(const type_code &code)
{
	fType = code;
	fTypeString = MakeTypeString(code);
}


void
ResourceData::SetID(const int32 &id)
{
	fID = id;
	fIDString = "";
	fIDString << fID;
}


void
ResourceData::SetData(const char *data, const size_t &size)
{
	free(fData);
	
	fLength = size;
	
	if (size > 0) {
		fData = (char *)malloc(size);
		memcpy(fData, data, size);
	}
	else
		fData = NULL;
}