/*
 * Copyright 2008-2009 Haiku Inc. All rights reserved.
 * Distributed under the terms of the MIT License.
 *
 * Authors:
 *              Pieter Panman
 */


#include "DevicePCI.h"

#include <sstream>
#include <stdlib.h>

#include <Catalog.h>

#undef B_TRANSLATION_CONTEXT
#define B_TRANSLATION_CONTEXT "DevicePCI"

extern "C" {
#include "dm_wrapper.h"
#include "pcihdr.h"
#include "pci-utils.h"
}


DevicePCI::DevicePCI(Device* parent)
        :
        Device(parent),
        fClassBaseId(0),
        fClassSubId(0),
        fClassApiId(0),
        fVendorId(0),
        fDeviceId(0),
        fSubsystemVendorId(0),
        fSubSystemId(0)
{
}


DevicePCI::~DevicePCI()
{
}


BString ToHex(uint16 num)
{
        std::stringstream ss;
        ss.flags(std::ios::hex | std::ios::showbase);
        ss << num;
        return BString(ss.str().c_str());
}


void
DevicePCI::InitFromAttributes()
{
        // Process the attributes
        fClassBaseId = atoi(fAttributeMap[B_DEVICE_TYPE].String());
        fClassSubId = atoi(fAttributeMap[B_DEVICE_SUB_TYPE].String());
        fClassApiId = atoi(fAttributeMap[B_DEVICE_INTERFACE].String());
        fVendorId = atoi(fAttributeMap[B_DEVICE_VENDOR_ID].String());
        fDeviceId = atoi(fAttributeMap[B_DEVICE_ID].String());

        // Looks better in Hex, so rewrite
        fAttributeMap[B_DEVICE_TYPE] = ToHex(fClassBaseId);
        fAttributeMap[B_DEVICE_SUB_TYPE] = ToHex(fClassSubId);
        fAttributeMap[B_DEVICE_INTERFACE] = ToHex(fClassApiId);
        fAttributeMap[B_DEVICE_VENDOR_ID] = ToHex(fVendorId);
        fAttributeMap[B_DEVICE_ID] = ToHex(fDeviceId);

        // Fetch ClassInfo      
        char classInfo[128];
        get_class_info(fClassBaseId, fClassSubId, fClassApiId, classInfo,
                sizeof(classInfo));
        
        // Fetch ManufacturerName
        BString ManufacturerName;
        const char *venShort;
        const char *venFull;
        get_vendor_info(fVendorId, &venShort, &venFull);
        if (!venShort && !venFull) {
                ManufacturerName << B_TRANSLATE("unknown");
        } else if (venShort && venFull) {
                ManufacturerName << venFull << "(" << venShort << ")";
        } else {
                ManufacturerName << (venShort ? venShort : venFull);
        }
        
        // Fetch DeviceName
        BString DeviceName;
        const char *devShort;
        const char *devFull;
        get_device_info(fVendorId, fDeviceId, fSubsystemVendorId, fSubSystemId,
                &devShort, &devFull);
        if (!devShort && !devFull) {
                DeviceName << B_TRANSLATE("unknown");
        } else if (devShort && devFull) {
                DeviceName << devFull << "(" << devShort << ")";
        } else {
                DeviceName << (devShort ? devShort : devFull);
        }
        
        SetAttribute(B_TRANSLATE("Device name"), DeviceName);
        SetAttribute(B_TRANSLATE("Manufacturer"), ManufacturerName);
#if 0
        // These are a source of confusion for users, leading them to think there
        // is no driver for their device. Until we can display something useful,
        // let's not show the lines at all.
        SetAttribute(B_TRANSLATE("Driver used"), B_TRANSLATE("Not implemented"));
        SetAttribute(B_TRANSLATE("Device paths"), B_TRANSLATE("Not implemented"));
#endif
        SetAttribute(B_TRANSLATE("Class info"), classInfo);
        fCategory = (Category)fClassBaseId;
        BString outlineName;
        outlineName << ManufacturerName << " " << DeviceName;
        SetText(outlineName.String());
}