Mercurial > hg > orthanc-stone
view Samples/Shared/RadiographyEditorApp.cpp @ 885:56e4e9281076 am-dev
sharing code between Qt/SDL BasiceScene sample
author | Alain Mazy <alain@mazy.be> |
---|---|
date | Tue, 09 Jul 2019 16:44:25 +0200 |
parents | 80829436ce0c |
children | 6e79e8c9021c |
line wrap: on
line source
/** * Stone of Orthanc * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics * Department, University Hospital of Liege, Belgium * Copyright (C) 2017-2019 Osimis S.A., Belgium * * This program is free software: you can redistribute it and/or * modify it under the terms of the GNU Affero General Public License * as published by the Free Software Foundation, either version 3 of * the License, or (at your option) any later version. * * This program is distributed in the hope that it will be useful, but * WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Affero General Public License for more details. * * You should have received a copy of the GNU Affero General Public License * along with this program. If not, see <http://www.gnu.org/licenses/>. **/ #include "RadiographyEditorApp.h" #include "../../Applications/Sdl/SdlOpenGLWindow.h" #include "../../Framework/Scene2D/CairoCompositor.h" #include "../../Framework/Scene2D/OpenGLCompositor.h" #include "../../Framework/Scene2D/ColorTextureSceneLayer.h" #include "../../Framework/Scene2D/PanSceneTracker.h" #include "../../Framework/Scene2D/RotateSceneTracker.h" #include "../../Framework/Scene2D/Scene2D.h" #include "../../Framework/Scene2D/ZoomSceneTracker.h" #include "../../Framework/Scene2DViewport/CreateAngleMeasureTracker.h" #include "../../Framework/Scene2DViewport/CreateLineMeasureTracker.h" #include "../../Framework/Scene2DViewport/UndoStack.h" #include "../../Framework/StoneInitialization.h" // From Orthanc framework #include <Core/Logging.h> #include <Core/OrthancException.h> #include <Core/Images/Image.h> #include <Core/Images/ImageProcessing.h> #include <Core/Images/PngWriter.h> #include <boost/ref.hpp> #include <boost/make_shared.hpp> #include <SDL.h> #include <stdio.h> namespace OrthancStone { const char* MeasureToolToString(size_t i) { static const char* descs[] = { "GuiTool_Rotate", "GuiTool_Pan", "GuiTool_Zoom", "GuiTool_LineMeasure", "GuiTool_CircleMeasure", "GuiTool_AngleMeasure", "GuiTool_EllipseMeasure", "GuiTool_LAST" }; if (i >= GuiTool_LAST) { throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError, "Wrong tool index"); } return descs[i]; } boost::shared_ptr<Scene2D> RadiographyEditorApp::GetScene() { return controller_->GetScene(); } boost::shared_ptr<const Scene2D> RadiographyEditorApp::GetScene() const { return controller_->GetScene(); } void RadiographyEditorApp::SelectNextTool() { currentTool_ = static_cast<GuiTool>(currentTool_ + 1); if (currentTool_ == GuiTool_LAST) currentTool_ = static_cast<GuiTool>(0);; printf("Current tool is now: %s\n", MeasureToolToString(currentTool_)); } void RadiographyEditorApp::DisplayInfoText() { // do not try to use stuff too early! if (compositor_.get() == NULL) return; std::stringstream msg; for (std::map<std::string, std::string>::const_iterator kv = infoTextMap_.begin(); kv != infoTextMap_.end(); ++kv) { msg << kv->first << " : " << kv->second << std::endl; } std::string msgS = msg.str(); TextSceneLayer* layerP = NULL; if (GetScene()->HasLayer(FIXED_INFOTEXT_LAYER_ZINDEX)) { TextSceneLayer& layer = dynamic_cast<TextSceneLayer&>( GetScene()->GetLayer(FIXED_INFOTEXT_LAYER_ZINDEX)); layerP = &layer; } else { std::auto_ptr<TextSceneLayer> layer(new TextSceneLayer); layerP = layer.get(); layer->SetColor(0, 255, 0); layer->SetFontIndex(1); layer->SetBorder(20); layer->SetAnchor(BitmapAnchor_TopLeft); //layer->SetPosition(0,0); GetScene()->SetLayer(FIXED_INFOTEXT_LAYER_ZINDEX, layer.release()); } // position the fixed info text in the upper right corner layerP->SetText(msgS.c_str()); double cX = compositor_->GetWidth() * (-0.5); double cY = compositor_->GetHeight() * (-0.5); GetScene()->GetCanvasToSceneTransform().Apply(cX,cY); layerP->SetPosition(cX, cY); } void RadiographyEditorApp::DisplayFloatingCtrlInfoText(const PointerEvent& e) { ScenePoint2D p = e.GetMainPosition().Apply(GetScene()->GetCanvasToSceneTransform()); char buf[128]; sprintf(buf, "S:(%0.02f,%0.02f) C:(%0.02f,%0.02f)", p.GetX(), p.GetY(), e.GetMainPosition().GetX(), e.GetMainPosition().GetY()); if (GetScene()->HasLayer(FLOATING_INFOTEXT_LAYER_ZINDEX)) { TextSceneLayer& layer = dynamic_cast<TextSceneLayer&>(GetScene()->GetLayer(FLOATING_INFOTEXT_LAYER_ZINDEX)); layer.SetText(buf); layer.SetPosition(p.GetX(), p.GetY()); } else { std::auto_ptr<TextSceneLayer> layer(new TextSceneLayer); layer->SetColor(0, 255, 0); layer->SetText(buf); layer->SetBorder(20); layer->SetAnchor(BitmapAnchor_BottomCenter); layer->SetPosition(p.GetX(), p.GetY()); GetScene()->SetLayer(FLOATING_INFOTEXT_LAYER_ZINDEX, layer.release()); } } void RadiographyEditorApp::HideInfoText() { GetScene()->DeleteLayer(FLOATING_INFOTEXT_LAYER_ZINDEX); } ScenePoint2D RadiographyEditorApp::GetRandomPointInScene() const { unsigned int w = compositor_->GetWidth(); LOG(TRACE) << "compositor_->GetWidth() = " << compositor_->GetWidth(); unsigned int h = compositor_->GetHeight(); LOG(TRACE) << "compositor_->GetHeight() = " << compositor_->GetHeight(); if ((w >= RAND_MAX) || (h >= RAND_MAX)) LOG(WARNING) << "Canvas is too big : tools will not be randomly placed"; int x = rand() % w; int y = rand() % h; LOG(TRACE) << "random x = " << x << "random y = " << y; ScenePoint2D p = compositor_->GetPixelCenterCoordinates(x, y); LOG(TRACE) << "--> p.GetX() = " << p.GetX() << " p.GetY() = " << p.GetY(); ScenePoint2D r = p.Apply(GetScene()->GetCanvasToSceneTransform()); LOG(TRACE) << "--> r.GetX() = " << r.GetX() << " r.GetY() = " << r.GetY(); return r; } void RadiographyEditorApp::CreateRandomMeasureTool() { static bool srandCalled = false; if (!srandCalled) { srand(42); srandCalled = true; } int i = rand() % 2; LOG(TRACE) << "random i = " << i; switch (i) { case 0: // line measure { boost::shared_ptr<CreateLineMeasureCommand> cmd = boost::make_shared<CreateLineMeasureCommand>( boost::ref(IObserver::GetBroker()), controller_, GetRandomPointInScene()); cmd->SetEnd(GetRandomPointInScene()); controller_->PushCommand(cmd); } break; case 1: // angle measure { boost::shared_ptr<CreateAngleMeasureCommand> cmd = boost::make_shared<CreateAngleMeasureCommand>( boost::ref(IObserver::GetBroker()), controller_, GetRandomPointInScene()); cmd->SetCenter(GetRandomPointInScene()); cmd->SetSide2End(GetRandomPointInScene()); controller_->PushCommand(cmd); } break; } } void RadiographyEditorApp::OnMouseMove(int x, int y, OrthancStone::KeyboardModifiers modifiers) { DisplayInfoText(); if (activeTracker_.get() == NULL && (modifiers & OrthancStone::KeyboardModifiers_Alt)) { // The "left-ctrl" key is down, while no tracker is present // Let's display the info text PointerEvent e; e.AddPosition(compositor_->GetPixelCenterCoordinates(x, y)); DisplayFloatingCtrlInfoText(e); } else { HideInfoText(); //LOG(TRACE) << "(event.type == SDL_MOUSEMOTION)"; if (activeTracker_.get() != NULL) { //LOG(TRACE) << "(activeTracker_.get() != NULL)"; PointerEvent e; e.AddPosition(compositor_->GetPixelCenterCoordinates(x, y)); //LOG(TRACE) << "event.button.x = " << event.button.x << " " << // "event.button.y = " << event.button.y; LOG(TRACE) << "activeTracker_->PointerMove(e); " << e.GetMainPosition().GetX() << " " << e.GetMainPosition().GetY(); activeTracker_->PointerMove(e); if (!activeTracker_->IsAlive()) activeTracker_.reset(); } } } void RadiographyEditorApp::OnKeyPressed(char keyChar, OrthancStone::KeyboardModifiers modifiers) { DisplayInfoText(); switch (keyChar) { case '\033': // escape { if (activeTracker_) { activeTracker_->Cancel(); if (!activeTracker_->IsAlive()) activeTracker_.reset(); } };break; case 't': { if (!activeTracker_) SelectNextTool(); else { LOG(WARNING) << "You cannot change the active tool when an interaction" " is taking place"; } };break; case 'm': CreateRandomMeasureTool(); break; case 's': controller_->FitContent(compositor_->GetWidth(), compositor_->GetHeight()); break; case 'z': LOG(TRACE) << "z has been pressed. modifier = " << modifiers; if (modifiers & OrthancStone::KeyboardModifiers_Control) { if (controller_->CanUndo()) { LOG(TRACE) << "Undoing..."; controller_->Undo(); } else { LOG(WARNING) << "Nothing to undo!!!"; } } break; case 'y': LOG(TRACE) << "y has been pressed. modifier = " << modifiers; if (modifiers & OrthancStone::KeyboardModifiers_Control) { if (controller_->CanRedo()) { LOG(TRACE) << "Redoing..."; controller_->Redo(); } else { LOG(WARNING) << "Nothing to redo!!!"; } } break; case 'c': TakeScreenshot( "screenshot.png", compositor_->GetWidth(), compositor_->GetHeight()); break; } } void RadiographyEditorApp::OnMouseDown(int x, int y, OrthancStone::KeyboardModifiers modifiers, OrthancStone::MouseButton button) { DisplayInfoText(); PointerEvent e; e.AddPosition(compositor_->GetPixelCenterCoordinates(x, y)); // TODO: set modifiers in e if (activeTracker_) { activeTracker_->PointerDown(e); if (!activeTracker_->IsAlive()) activeTracker_.reset(); } else { // we ATTEMPT to create a tracker if need be activeTracker_ = CreateSuitableTracker(button, e); } } void RadiographyEditorApp::OnMouseUp(int x, int y, OrthancStone::KeyboardModifiers modifiers, OrthancStone::MouseButton button) { DisplayInfoText(); if (activeTracker_) { PointerEvent e; e.AddPosition(compositor_->GetPixelCenterCoordinates(x, y)); // TODO: set modifiers in e activeTracker_->PointerUp(e); if (!activeTracker_->IsAlive()) activeTracker_.reset(); } } void RadiographyEditorApp::HandleApplicationEvent( const SDL_Event & event) { DisplayInfoText(); if (event.type == SDL_MOUSEMOTION) { int scancodeCount = 0; const uint8_t* keyboardState = SDL_GetKeyboardState(&scancodeCount); if (activeTracker_.get() == NULL && SDL_SCANCODE_LALT < scancodeCount && keyboardState[SDL_SCANCODE_LALT]) { // The "left-ctrl" key is down, while no tracker is present // Let's display the info text PointerEvent e; e.AddPosition(compositor_->GetPixelCenterCoordinates( event.button.x, event.button.y)); // TODO: set modifiers in e DisplayFloatingCtrlInfoText(e); } else { HideInfoText(); //LOG(TRACE) << "(event.type == SDL_MOUSEMOTION)"; if (activeTracker_.get() != NULL) { //LOG(TRACE) << "(activeTracker_.get() != NULL)"; PointerEvent e; e.AddPosition(compositor_->GetPixelCenterCoordinates( event.button.x, event.button.y)); // TODO: set modifiers in e //LOG(TRACE) << "event.button.x = " << event.button.x << " " << // "event.button.y = " << event.button.y; LOG(TRACE) << "activeTracker_->PointerMove(e); " << e.GetMainPosition().GetX() << " " << e.GetMainPosition().GetY(); activeTracker_->PointerMove(e); if (!activeTracker_->IsAlive()) activeTracker_.reset(); } } } else if (event.type == SDL_MOUSEBUTTONUP) { if (activeTracker_) { PointerEvent e; e.AddPosition(compositor_->GetPixelCenterCoordinates(event.button.x, event.button.y)); // TODO: set modifiers in e activeTracker_->PointerUp(e); if (!activeTracker_->IsAlive()) activeTracker_.reset(); } } else if (event.type == SDL_MOUSEBUTTONDOWN) { PointerEvent e; e.AddPosition(compositor_->GetPixelCenterCoordinates( event.button.x, event.button.y)); // TODO: set modifiers in e if (activeTracker_) { activeTracker_->PointerDown(e); if (!activeTracker_->IsAlive()) activeTracker_.reset(); } else { // we ATTEMPT to create a tracker if need be // activeTracker_ = CreateSuitableTracker(event, e); } } else if (event.type == SDL_KEYDOWN && event.key.repeat == 0 /* Ignore key bounce */) { switch (event.key.keysym.sym) { case SDLK_ESCAPE: if (activeTracker_) { activeTracker_->Cancel(); if (!activeTracker_->IsAlive()) activeTracker_.reset(); } break; case SDLK_t: if (!activeTracker_) SelectNextTool(); else { LOG(WARNING) << "You cannot change the active tool when an interaction" " is taking place"; } break; case SDLK_m: CreateRandomMeasureTool(); break; case SDLK_s: controller_->FitContent(compositor_->GetWidth(), compositor_->GetHeight()); break; case SDLK_z: LOG(TRACE) << "SDLK_z has been pressed. event.key.keysym.mod == " << event.key.keysym.mod; if (event.key.keysym.mod & KMOD_CTRL) { if (controller_->CanUndo()) { LOG(TRACE) << "Undoing..."; controller_->Undo(); } else { LOG(WARNING) << "Nothing to undo!!!"; } } break; case SDLK_y: LOG(TRACE) << "SDLK_y has been pressed. event.key.keysym.mod == " << event.key.keysym.mod; if (event.key.keysym.mod & KMOD_CTRL) { if (controller_->CanRedo()) { LOG(TRACE) << "Redoing..."; controller_->Redo(); } else { LOG(WARNING) << "Nothing to redo!!!"; } } break; case SDLK_c: TakeScreenshot( "screenshot.png", compositor_->GetWidth(), compositor_->GetHeight()); break; default: break; } } } void RadiographyEditorApp::OnSceneTransformChanged( const ViewportController::SceneTransformChanged& message) { DisplayInfoText(); } boost::shared_ptr<IFlexiblePointerTracker> RadiographyEditorApp::CreateSuitableTracker( OrthancStone::MouseButton button, const PointerEvent & e) { using namespace Orthanc; switch (button) { case OrthancStone::MouseButton_Middle: return boost::shared_ptr<IFlexiblePointerTracker>(new PanSceneTracker (controller_, e)); case OrthancStone::MouseButton_Right: return boost::shared_ptr<IFlexiblePointerTracker>(new ZoomSceneTracker (controller_, e, compositor_->GetHeight())); case OrthancStone::MouseButton_Left: { //LOG(TRACE) << "CreateSuitableTracker: case SDL_BUTTON_LEFT:"; // TODO: we need to iterate on the set of measuring tool and perform // a hit test to check if a tracker needs to be created for edition. // Otherwise, depending upon the active tool, we might want to create // a "measuring tool creation" tracker // TODO: if there are conflicts, we should prefer a tracker that // pertains to the type of measuring tool currently selected (TBD?) boost::shared_ptr<IFlexiblePointerTracker> hitTestTracker = TrackerHitTest(e); if (hitTestTracker != NULL) { //LOG(TRACE) << "hitTestTracker != NULL"; return hitTestTracker; } else { switch (currentTool_) { case GuiTool_Rotate: //LOG(TRACE) << "Creating RotateSceneTracker"; return boost::shared_ptr<IFlexiblePointerTracker>(new RotateSceneTracker( controller_, e)); case GuiTool_Pan: return boost::shared_ptr<IFlexiblePointerTracker>(new PanSceneTracker( controller_, e)); case GuiTool_Zoom: return boost::shared_ptr<IFlexiblePointerTracker>(new ZoomSceneTracker( controller_, e, compositor_->GetHeight())); //case GuiTool_AngleMeasure: // return new AngleMeasureTracker(GetScene(), e); //case GuiTool_CircleMeasure: // return new CircleMeasureTracker(GetScene(), e); //case GuiTool_EllipseMeasure: // return new EllipseMeasureTracker(GetScene(), e); case GuiTool_LineMeasure: return boost::shared_ptr<IFlexiblePointerTracker>(new CreateLineMeasureTracker( IObserver::GetBroker(), controller_, e)); case GuiTool_AngleMeasure: return boost::shared_ptr<IFlexiblePointerTracker>(new CreateAngleMeasureTracker( IObserver::GetBroker(), controller_, e)); case GuiTool_CircleMeasure: LOG(ERROR) << "Not implemented yet!"; return boost::shared_ptr<IFlexiblePointerTracker>(); case GuiTool_EllipseMeasure: LOG(ERROR) << "Not implemented yet!"; return boost::shared_ptr<IFlexiblePointerTracker>(); default: throw OrthancException(ErrorCode_InternalError, "Wrong tool!"); } } } default: return boost::shared_ptr<IFlexiblePointerTracker>(); } } RadiographyEditorApp::RadiographyEditorApp(OrthancStone::IOracle& oracle, IObservable& oracleObservable, ICompositorFactory* compositorFactory) : IObserver(oracleObservable.GetBroker()), oracle_(oracle), compositorFactory_(compositorFactory), currentTool_(GuiTool_Rotate) { boost::shared_ptr<UndoStack> undoStack(new UndoStack); controller_ = boost::shared_ptr<ViewportController>(new ViewportController(undoStack, IObserver::GetBroker())); controller_->RegisterObserverCallback( new Callable<RadiographyEditorApp, ViewportController::SceneTransformChanged> (*this, &RadiographyEditorApp::OnSceneTransformChanged)); TEXTURE_2x2_1_ZINDEX = 1; TEXTURE_1x1_ZINDEX = 2; TEXTURE_2x2_2_ZINDEX = 3; LINESET_1_ZINDEX = 4; LINESET_2_ZINDEX = 5; FLOATING_INFOTEXT_LAYER_ZINDEX = 6; FIXED_INFOTEXT_LAYER_ZINDEX = 7; } void RadiographyEditorApp::PrepareScene() { // Texture of 2x2 size { Orthanc::Image i(Orthanc::PixelFormat_RGB24, 2, 2, false); uint8_t* p = reinterpret_cast<uint8_t*>(i.GetRow(0)); p[0] = 255; p[1] = 0; p[2] = 0; p[3] = 0; p[4] = 255; p[5] = 0; p = reinterpret_cast<uint8_t*>(i.GetRow(1)); p[0] = 0; p[1] = 0; p[2] = 255; p[3] = 255; p[4] = 0; p[5] = 0; GetScene()->SetLayer(TEXTURE_2x2_1_ZINDEX, new ColorTextureSceneLayer(i)); std::auto_ptr<ColorTextureSceneLayer> l(new ColorTextureSceneLayer(i)); l->SetOrigin(-3, 2); l->SetPixelSpacing(1.5, 1); l->SetAngle(20.0 / 180.0 * M_PI); GetScene()->SetLayer(TEXTURE_2x2_2_ZINDEX, l.release()); } // Texture of 1x1 size { Orthanc::Image i(Orthanc::PixelFormat_RGB24, 1, 1, false); uint8_t* p = reinterpret_cast<uint8_t*>(i.GetRow(0)); p[0] = 255; p[1] = 0; p[2] = 0; std::auto_ptr<ColorTextureSceneLayer> l(new ColorTextureSceneLayer(i)); l->SetOrigin(-2, 1); l->SetAngle(20.0 / 180.0 * M_PI); GetScene()->SetLayer(TEXTURE_1x1_ZINDEX, l.release()); } // Some lines { std::auto_ptr<PolylineSceneLayer> layer(new PolylineSceneLayer); layer->SetThickness(1); PolylineSceneLayer::Chain chain; chain.push_back(ScenePoint2D(0 - 0.5, 0 - 0.5)); chain.push_back(ScenePoint2D(0 - 0.5, 2 - 0.5)); chain.push_back(ScenePoint2D(2 - 0.5, 2 - 0.5)); chain.push_back(ScenePoint2D(2 - 0.5, 0 - 0.5)); layer->AddChain(chain, true, 255, 0, 0); chain.clear(); chain.push_back(ScenePoint2D(-5, -5)); chain.push_back(ScenePoint2D(5, -5)); chain.push_back(ScenePoint2D(5, 5)); chain.push_back(ScenePoint2D(-5, 5)); layer->AddChain(chain, true, 0, 255, 0); double dy = 1.01; chain.clear(); chain.push_back(ScenePoint2D(-4, -4)); chain.push_back(ScenePoint2D(4, -4 + dy)); chain.push_back(ScenePoint2D(-4, -4 + 2.0 * dy)); chain.push_back(ScenePoint2D(4, 2)); layer->AddChain(chain, false, 0, 0, 255); GetScene()->SetLayer(LINESET_1_ZINDEX, layer.release()); } // Some text { std::auto_ptr<TextSceneLayer> layer(new TextSceneLayer); layer->SetText("Hello"); GetScene()->SetLayer(LINESET_2_ZINDEX, layer.release()); } } void RadiographyEditorApp::DisableTracker() { if (activeTracker_) { activeTracker_->Cancel(); activeTracker_.reset(); } } void RadiographyEditorApp::TakeScreenshot(const std::string& target, unsigned int canvasWidth, unsigned int canvasHeight) { CairoCompositor compositor(*GetScene(), canvasWidth, canvasHeight); compositor.SetFont(0, Orthanc::EmbeddedResources::UBUNTU_FONT, FONT_SIZE_0, Orthanc::Encoding_Latin1); compositor.Refresh(); Orthanc::ImageAccessor canvas; compositor.GetCanvas().GetReadOnlyAccessor(canvas); Orthanc::Image png(Orthanc::PixelFormat_RGB24, canvas.GetWidth(), canvas.GetHeight(), false); Orthanc::ImageProcessing::Convert(png, canvas); Orthanc::PngWriter writer; writer.WriteToFile(target, png); } boost::shared_ptr<IFlexiblePointerTracker> RadiographyEditorApp::TrackerHitTest(const PointerEvent & e) { // std::vector<boost::shared_ptr<MeasureTool>> measureTools_; return boost::shared_ptr<IFlexiblePointerTracker>(); } void RadiographyEditorApp::FitContent(unsigned int width, unsigned int height) { controller_->FitContent(width, height); } void RadiographyEditorApp::UpdateSize() { if (dynamic_cast<OpenGLCompositor*>(compositor_.get()) != NULL) { dynamic_cast<OpenGLCompositor*>(compositor_.get())->UpdateSize(); } } void RadiographyEditorApp::Refresh() { compositor_.reset(compositorFactory_->GetCompositor(*GetScene())); compositor_->Refresh(); // the following is paramount because the compositor holds a reference // to the scene and we do not want this reference to become dangling // TODO ???? compositor_.reset(NULL); } void RadiographyEditorApp::SetInfoDisplayMessage( std::string key, std::string value) { if (value == "") infoTextMap_.erase(key); else infoTextMap_[key] = value; DisplayInfoText(); } }