* Copyright 2004-2009, Haiku. All rights reserved.
* Distributed under the terms of the MIT license.
*
* Authors:
* Jérôme Duval
*/
#include "MethodReplicant.h"
#include <new>
#include <string.h>
#include <Alert.h>
#include <AppDefs.h>
#include <Debug.h>
#include <Dragger.h>
#include <Bitmap.h>
#include <MenuItem.h>
#include <Message.h>
#include <Messenger.h>
#include <PopUpMenu.h>
#include "remote_icon.h"
#include "InputServerTypes.h"
#ifdef DEBUG
# define CALLED() PRINT(("CALLED %s \n", __PRETTY_FUNCTION__));
#else
# define CALLED()
#endif
MethodReplicant::MethodReplicant(const char* signature)
:
BView(BRect(0, 0, 15, 15), REPLICANT_CTL_NAME, B_FOLLOW_ALL, B_WILL_DRAW),
fMenu("", false, false)
{
fSegments = new BBitmap(BRect(0, 0, kRemoteWidth - 1, kRemoteHeight - 1),
kRemoteColorSpace);
fSegments->SetBits(kRemoteBits, kRemoteWidth * kRemoteHeight, 0,
kRemoteColorSpace);
ASSERT(signature != NULL);
fSignature = strdup(signature);
fMenu.SetFont(be_plain_font);
fMenu.SetRadioMode(true);
}
MethodReplicant::MethodReplicant(BMessage* message)
:
BView(message),
fMenu("", false, false)
{
fSegments = new BBitmap(BRect(0, 0, kRemoteWidth - 1, kRemoteHeight - 1),
kRemoteColorSpace);
fSegments->SetBits(kRemoteBits, kRemoteWidth * kRemoteHeight, 0,
kRemoteColorSpace);
const char* signature = NULL;
message->FindString("add_on", &signature);
ASSERT(signature != NULL);
fSignature = strdup(signature);
fMenu.SetFont(be_plain_font);
fMenu.SetRadioMode(true);
}
MethodReplicant::~MethodReplicant()
{
delete fSegments;
free(fSignature);
}
MethodReplicant*
MethodReplicant::Instantiate(BMessage* data)
{
CALLED();
if (!validate_instantiation(data, REPLICANT_CTL_NAME))
return NULL;
return new(std::nothrow) MethodReplicant(data);
}
status_t
MethodReplicant::Archive(BMessage* data, bool deep) const
{
BView::Archive(data, deep);
data->AddString("add_on", fSignature);
return B_NO_ERROR;
}
void
MethodReplicant::AttachedToWindow()
{
CALLED();
SetViewColor(Parent()->ViewColor());
BMessenger messenger(this);
BMessage msg(IS_METHOD_REGISTER);
msg.AddMessenger("address", messenger);
BMessenger inputMessenger(fSignature);
if (inputMessenger.SendMessage(&msg) != B_OK)
printf("error when contacting input_server\n");
}
void
MethodReplicant::MessageReceived(BMessage* message)
{
PRINT(("%s what:%c%c%c%c\n", __PRETTY_FUNCTION__, (char)(message->what >> 24),
(char)(message->what >> 16), (char)(message->what >> 8), (char)message->what));
PRINT_OBJECT(*message);
switch (message->what) {
case B_ABOUT_REQUESTED:
{
BAlert* alert = new BAlert("About Method Replicant",
"Method Replicant (Replicant)\n"
" Brought to you by Jérôme DUVAL.\n\n"
"Haiku, 2004-2009", "OK");
alert->SetFlags(alert->Flags() | B_CLOSE_ON_ESCAPE);
alert->Go();
break;
}
case IS_UPDATE_NAME:
UpdateMethodName(message);
break;
case IS_UPDATE_ICON:
UpdateMethodIcon(message);
break;
case IS_UPDATE_MENU:
UpdateMethodMenu(message);
break;
case IS_UPDATE_METHOD:
UpdateMethod(message);
break;
case IS_ADD_METHOD:
AddMethod(message);
break;
case IS_REMOVE_METHOD:
RemoveMethod(message);
break;
default:
BView::MessageReceived(message);
break;
}
}
void
MethodReplicant::Draw(BRect rect)
{
BView::Draw(rect);
SetDrawingMode(B_OP_OVER);
DrawBitmap(fSegments);
}
void
MethodReplicant::MouseDown(BPoint point)
{
CALLED();
uint32 mouseButtons;
BPoint where;
GetMouse(&where, &mouseButtons, true);
where = ConvertToScreen(point);
fMenu.SetTargetForItems(this);
BMenuItem* item = fMenu.Go(where, true, true,
BRect(where - BPoint(4, 4), where + BPoint(4, 4)));
if (dynamic_cast<MethodMenuItem*>(item) != NULL) {
BMessage msg(IS_SET_METHOD);
msg.AddInt32("cookie", ((MethodMenuItem*)item)->Cookie());
BMessenger messenger(fSignature);
messenger.SendMessage(&msg);
}
}
void
MethodReplicant::MouseUp(BPoint point)
{
}
void
MethodReplicant::UpdateMethod(BMessage* message)
{
CALLED();
int32 cookie;
if (message->FindInt32("cookie", &cookie) != B_OK) {
fprintf(stderr, "can't find cookie in message\n");
return;
}
MethodMenuItem* item = FindItemByCookie(cookie);
if (item == NULL) {
fprintf(stderr, "can't find item with cookie %" B_PRIx32 "\n", cookie);
return;
}
item->SetMarked(true);
fSegments->SetBits(item->Icon(), kRemoteWidth * kRemoteHeight, 0,
kRemoteColorSpace);
Invalidate();
}
void
MethodReplicant::UpdateMethodIcon(BMessage* message)
{
CALLED();
int32 cookie;
if (message->FindInt32("cookie", &cookie) != B_OK) {
fprintf(stderr, "can't find cookie in message\n");
return;
}
const uchar* data;
ssize_t numBytes;
if (message->FindData("icon", B_ANY_TYPE, (const void**)&data, &numBytes)
!= B_OK) {
fprintf(stderr, "can't find icon in message\n");
return;
}
MethodMenuItem* item = FindItemByCookie(cookie);
if (item == NULL) {
fprintf(stderr, "can't find item with cookie %" B_PRIx32 "\n", cookie);
return;
}
item->SetIcon(data);
}
void
MethodReplicant::UpdateMethodMenu(BMessage* message)
{
CALLED();
int32 cookie;
if (message->FindInt32("cookie", &cookie) != B_OK) {
fprintf(stderr, "can't find cookie in message\n");
return;
}
BMessage msg;
if (message->FindMessage("menu", &msg) != B_OK) {
fprintf(stderr, "can't find menu in message\n");
return;
}
PRINT_OBJECT(msg);
BMessenger messenger;
if (message->FindMessenger("target", &messenger) != B_OK) {
fprintf(stderr, "can't find target in message\n");
return;
}
BMenu* menu = (BMenu*)BMenu::Instantiate(&msg);
if (menu == NULL) {
PRINT(("can't instantiate menu\n"));
} else
menu->SetTargetForItems(messenger);
MethodMenuItem* item = FindItemByCookie(cookie);
if (item == NULL) {
fprintf(stderr, "can't find item with cookie %" B_PRIx32 "\n", cookie);
return;
}
int32 index = fMenu.IndexOf(item);
MethodMenuItem* item2 = NULL;
if (menu) {
item2 = new MethodMenuItem(cookie, item->Label(), item->Icon(),
menu, messenger);
} else
item2 = new MethodMenuItem(cookie, item->Label(), item->Icon());
item = (MethodMenuItem*)fMenu.RemoveItem(index);
fMenu.AddItem(item2, index);
item2->SetMarked(item->IsMarked());
delete item;
}
void
MethodReplicant::UpdateMethodName(BMessage* message)
{
CALLED();
int32 cookie;
if (message->FindInt32("cookie", &cookie) != B_OK) {
fprintf(stderr, "can't find cookie in message\n");
return;
}
const char* name;
if (message->FindString("name", &name) != B_OK) {
fprintf(stderr, "can't find name in message\n");
return;
}
MethodMenuItem* item = FindItemByCookie(cookie);
if (item == NULL) {
fprintf(stderr, "can't find item with cookie %" B_PRIx32 "\n", cookie);
return;
}
item->SetName(name);
}
MethodMenuItem*
MethodReplicant::FindItemByCookie(int32 cookie)
{
for (int32 i = 0; i < fMenu.CountItems(); i++) {
MethodMenuItem* item = (MethodMenuItem*)fMenu.ItemAt(i);
PRINT(("cookie : 0x%" B_PRIx32 "\n", item->Cookie()));
if (item->Cookie() == cookie)
return item;
}
return NULL;
}
void
MethodReplicant::AddMethod(BMessage* message)
{
CALLED();
int32 cookie;
if (message->FindInt32("cookie", &cookie) != B_OK) {
fprintf(stderr, "can't find cookie in message\n");
return;
}
const char* name;
if (message->FindString("name", &name) != B_OK) {
fprintf(stderr, "can't find name in message\n");
return;
}
const uchar* icon;
ssize_t numBytes;
if (message->FindData("icon", B_ANY_TYPE, (const void**)&icon, &numBytes)
!= B_OK) {
fprintf(stderr, "can't find icon in message\n");
return;
}
BMessage menuMsg;
if (message->FindMessage("menu", &menuMsg) != B_OK) {
fprintf(stderr, "can't find menu in message\n");
return;
}
PRINT_OBJECT(menuMsg);
BMessenger messenger;
if (message->FindMessenger("target", &messenger) != B_OK) {
fprintf(stderr, "can't find target in message\n");
return;
}
BMenu* menu = static_cast<BMenu*>(BMenu::Instantiate(&menuMsg));
if (menu == NULL) {
PRINT(("can't instantiate menu\n"));
} else
menu->SetTargetForItems(messenger);
MethodMenuItem* item = FindItemByCookie(cookie);
if (item != NULL) {
fprintf(stderr, "item with cookie %" B_PRIx32 " already exists\n", cookie);
return;
}
if (menu != NULL) {
item = new MethodMenuItem(cookie, name, icon, menu, messenger);
} else
item = new MethodMenuItem(cookie, name, icon);
fMenu.AddItem(item);
item->SetTarget(this);
if (fMenu.CountItems() == 1)
item->SetMarked(true);
}
void
MethodReplicant::RemoveMethod(BMessage* message)
{
CALLED();
int32 cookie;
if (message->FindInt32("cookie", &cookie) != B_OK) {
fprintf(stderr, "can't find cookie in message\n");
return;
}
MethodMenuItem* item = FindItemByCookie(cookie);
if (item == NULL) {
fprintf(stderr, "can't find item with cookie %" B_PRIx32 "\n", cookie);
return;
}
fMenu.RemoveItem(item);
delete item;
}
