* Copyright 2010 Wim van der Meer <WPJvanderMeer@gmail.com>
* All rights reserved. Distributed under the terms of the MIT license.
*
* Authors:
* Wim van der Meer
*/
#include "ScreenshotApp.h"
#include <stdlib.h>
#include <Bitmap.h>
#include <Catalog.h>
#include <Locale.h>
#include <Roster.h>
#include <Screen.h>
#include <WindowPrivate.h>
#include "ScreenshotWindow.h"
#include "SelectAreaView.h"
#include "Utility.h"
ScreenshotApp::ScreenshotApp()
:
BApplication("application/x-vnd.haiku-screenshot"),
fScreenshotWindow(NULL),
fUtility(new Utility),
fSilent(false),
fClipboard(false),
fLaunchWithAreaSelect(false)
{
}
ScreenshotApp::~ScreenshotApp()
{
delete fUtility;
}
void
ScreenshotApp::MessageReceived(BMessage* message)
{
status_t status = B_OK;
switch (message->what) {
case SS_UTILITY_DATA:
{
BMessage bitmap;
status = message->FindMessage("wholeScreen", &bitmap);
if (status != B_OK)
break;
fUtility->wholeScreen = new BBitmap(&bitmap);
status = message->FindMessage("cursorBitmap", &bitmap);
if (status != B_OK)
break;
fUtility->cursorBitmap = new BBitmap(&bitmap);
status = message->FindMessage("cursorAreaBitmap", &bitmap);
if (status != B_OK)
break;
fUtility->cursorAreaBitmap = new BBitmap(&bitmap);
status = message->FindPoint("cursorPosition",
&fUtility->cursorPosition);
if (status != B_OK)
break;
status = message->FindRect("activeWindowFrame",
&fUtility->activeWindowFrame);
if (status != B_OK)
break;
status = message->FindRect("tabFrame", &fUtility->tabFrame);
if (status != B_OK)
break;
status = message->FindFloat("borderSize", &fUtility->borderSize);
if (status != B_OK)
break;
status = message->FindBool("selectArea", &fLaunchWithAreaSelect);
if (status != B_OK)
break;
break;
}
case SS_SELECT_AREA_FRAME:
{
BRect frame;
status_t status = message->FindRect("selectAreaFrame", &frame);
if (status != B_OK)
break;
if (!frame.IsValid() && fLaunchWithAreaSelect)
be_app->PostMessage(B_QUIT_REQUESTED);
else if (fScreenshotWindow == NULL) {
fScreenshotWindow = new ScreenshotWindow(*fUtility, fSilent, fClipboard);
fLaunchWithAreaSelect = false;
}
fScreenshotWindow->SetSelectedArea(frame);
break;
}
case SS_LAUNCH_AREA_SELECTOR:
{
BWindow* selectAreaWindow = new BWindow(BScreen().Frame(),
"Area Window", kWindowScreenWindow,
B_ASYNCHRONOUS_CONTROLS|B_NOT_RESIZABLE|B_NOT_CLOSABLE|B_NOT_ZOOMABLE);
selectAreaWindow->AddChild(new SelectAreaView(fUtility->wholeScreen));
selectAreaWindow->Show();
break;
}
default:
BApplication::MessageReceived(message);
break;
}
if (status != B_OK)
be_app->PostMessage(B_QUIT_REQUESTED);
}
void
ScreenshotApp::ArgvReceived(int32 argc, char** argv)
{
for (int32 i = 0; i < argc; i++) {
if (strcmp(argv[i], "-s") == 0
|| strcmp(argv[i], "--silent") == 0)
fSilent = true;
else if (strcmp(argv[i], "-c") == 0
|| strcmp(argv[i], "--clipboard") == 0)
fClipboard = true;
}
}
void
ScreenshotApp::ReadyToRun()
{
if (fLaunchWithAreaSelect)
be_app->PostMessage(new BMessage(SS_LAUNCH_AREA_SELECTOR));
else
fScreenshotWindow = new ScreenshotWindow(*fUtility, fSilent, fClipboard);
}
int
main()
{
ScreenshotApp app;
return app.Run();
}
