/*=========================================================================

 medInria

 Copyright (c) INRIA 2013 - 2018. All rights reserved.
 See LICENSE.txt for details.
 
  This software is distributed WITHOUT ANY WARRANTY; without even
  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  PURPOSE.

=========================================================================*/

#include <medAnnIntSeedPointHelper.h>

#include <medSeedPointAnnotationData.h>

#include <medAnnotationInteractor.h>

#include <medVtkSeedPointHandle.h>

#include <medAbstractImageView.h>
#include <medVtkViewBackend.h>
#include <vtkImageView2D.h>

#include <vtkCommand.h>
#include <vtkHandleWidget.h>
#include <vtkProperty.h>
#include <vtkProperty2D.h>
#include <vtkRenderer.h>
#include <vtkRenderWindow.h>
#include <vtkSeedWidget.h>
#include <vtkSeedRepresentation.h>
#include <vtkSphereHandleRepresentation.h>
#include <vtkSmartPointer.h>

class vtkCommandWithDisable : public vtkCommand 
{
public:
    bool isEnabled() const { return m_disabledCount == 0; }
    void enable()   { --m_disabledCount; }
    void disable()  { ++m_disabledCount; }
protected:
    vtkCommandWithDisable() : vtkCommand(), m_disabledCount(0) {}
    unsigned int m_disabledCount;

};

class vtkScopedCommandDisabler {
public:
    vtkScopedCommandDisabler( vtkCommandWithDisable * cmd ) : m_cmd(cmd) 
    {
        m_cmd->disable();
    }

    ~vtkScopedCommandDisabler()
    {
        m_cmd->enable();
    }
private:
    vtkCommandWithDisable * m_cmd;
};

// This callback is responsible for setting the seed label.
class vtkSeedCallback : public vtkCommandWithDisable
{
public:
    static vtkSeedCallback *New() 
    { return new vtkSeedCallback; }
    virtual void Execute(vtkObject*obj, unsigned long event, void *calldata)
    {
        if ( !this->isEnabled() ) 
            return;

        if (event == vtkCommand::PlacePointEvent)
        {
        }
        if (event == vtkCommand::EndInteractionEvent)
        {
            vtkSeedWidget * spW = vtkSeedWidget::SafeDownCast(obj);
            if (spW)
            {
                helper->refreshFromWidget(spW);
            }
        }
    }
    medAnnIntSeedPointHelper * helper;
protected:
    vtkSeedCallback() : vtkCommandWithDisable() {}
};


class medAnnIntSeedPointHelperPrivate {
public:
    medAnnIntSeedPointHelper::ActorMap actors;

    vtkSmartPointer<vtkSeedCallback> spCbk;
};

medAnnIntSeedPointHelper::medAnnIntSeedPointHelper(medAnnotationInteractor * annInt)
    : msegAnnIntHelper(annInt) ,
    d(new medAnnIntSeedPointHelperPrivate)
{
    d->spCbk = vtkSmartPointer<vtkSeedCallback>::New();
}

medAnnIntSeedPointHelper::~medAnnIntSeedPointHelper()
{
    delete d;
    d = nullptr;
}

bool medAnnIntSeedPointHelper::addAnnotation( medAnnotationData* annData )
{
    medSeedPointAnnotationData * spad = qobject_cast<medSeedPointAnnotationData*>(annData);
    if ( !spad ) 
        return false;

    medAbstractImageView * view = this->getView();
    medVtkViewBackend* backend = static_cast<medVtkViewBackend*>(view->backend());

    //double zero[2] = {0,0};
    typedef vtkSmartPointer<medVtkSeedPointHandle> RepSmartPointer;
    RepSmartPointer handleRep = RepSmartPointer::New();
    // Blue, close to selected color in table widget.
    handleRep->GetSelectedProperty()->SetColor(0x33/255., 0x99/255., 0xff/255.);

    vtkSmartPointer<vtkSeedWidget> seedActor2d = vtkSmartPointer<vtkSeedWidget>::New();
    seedActor2d->SetCurrentRenderer(backend->view2D->GetRenderer());
    seedActor2d->SetInteractor(backend->view2D->GetRenderer()->GetRenderWindow()->GetInteractor());
    seedActor2d->SetEnabled(1);
    seedActor2d->GetSeedRepresentation()->SetHandleRepresentation( handleRep );
    seedActor2d->CreateDefaultRepresentation();
    seedActor2d->CompleteInteraction();

   // view->renderer2d()->AddActor(seedActor2d->GetRepresentation());

    vtkSmartPointer<vtkSeedWidget> seedActor3d = vtkSmartPointer<vtkSeedWidget>::New();
    seedActor3d->SetCurrentRenderer(backend->view3D->GetRenderer());
    seedActor3d->SetInteractor(backend->view3D->GetRenderer()->GetRenderWindow()->GetInteractor());
    seedActor3d->SetEnabled(1);
    seedActor3d->GetSeedRepresentation()->SetHandleRepresentation( handleRep );
    seedActor3d->CreateDefaultRepresentation();
    seedActor3d->CompleteInteraction();
    //view->renderer3d()->AddActor(seedActor3d->GetRepresentation());

    ActorInfo & actorInfo(this->getActorMap()[spad]);
    actorInfo.actor2d = seedActor2d;
    actorInfo.actor3d = seedActor3d;

    this->annotationModified(annData);

    d->spCbk->helper = this;
//    widget->AddObserver(vtkCommand::PlacePointEvent,scbk);
    seedActor2d->AddObserver(vtkCommand::EndInteractionEvent,d->spCbk);
    seedActor3d->AddObserver(vtkCommand::EndInteractionEvent,d->spCbk);


    backend->view2D->Render();
    backend->view3D->Render();

    return true;
}


void medAnnIntSeedPointHelper::removeAnnotation( medAnnotationData * annData )
{
    medSeedPointAnnotationData * spad = qobject_cast<medSeedPointAnnotationData*>(annData);

    medAbstractImageView * view = this->getView();
    medVtkViewBackend* backend = static_cast<medVtkViewBackend*>(view->backend());

    ActorInfo & actorInfo(this->getActorMap()[spad]);

    backend->view2D->GetRenderer()->RemoveActor(actorInfo.actor2d->GetRepresentation());
    backend->view2D->GetRenderer()->RemoveActor(actorInfo.actor3d->GetRepresentation());

    d->actors.erase(spad);

    backend->view2D->Render();
    backend->view3D->Render();
}

void medAnnIntSeedPointHelper::annotationModified( medAnnotationData* annData )
{
    medSeedPointAnnotationData * spad = qobject_cast<medSeedPointAnnotationData*>(annData);

    vtkScopedCommandDisabler disabler( d->spCbk );

    ActorInfo & actorInfo(this->getActorMap()[spad]);
    vtkSeedWidget * seedActor2d =  vtkSeedWidget::SafeDownCast(actorInfo.actor2d);
    vtkSeedWidget * seedActor3d =  vtkSeedWidget::SafeDownCast(actorInfo.actor3d);

    const int numSeeds = spad->getNumberOfSeeds();

    // Delete excess seeds, and add new seeds.
    vtkSeedRepresentation * rep2d = seedActor2d->GetSeedRepresentation();
    while ( rep2d->GetNumberOfSeeds() > numSeeds ) {
        seedActor2d->DeleteSeed(rep2d->GetNumberOfSeeds() - 1);
    }
    while ( rep2d->GetNumberOfSeeds() < numSeeds ) {
        int iSeed = rep2d->GetNumberOfSeeds();
        seedActor2d->CreateNewHandle()->EnabledOn();
        seedActor2d->GetCurrentRenderer()->AddActor( rep2d->GetHandleRepresentation(iSeed) );
    }

    vtkSeedRepresentation * rep3d = seedActor3d->GetSeedRepresentation();
    while ( rep3d->GetNumberOfSeeds() > numSeeds ) {
        seedActor3d->DeleteSeed(rep3d->GetNumberOfSeeds() - 1);
    }
    while ( rep3d->GetNumberOfSeeds() < numSeeds ) {
        int iSeed = rep3d->GetNumberOfSeeds();
        seedActor3d->CreateNewHandle()->EnabledOn();
        seedActor3d->GetCurrentRenderer()->AddActor( rep3d->GetHandleRepresentation(iSeed) );
    }

    const int selectedSeed = spad->selectedSeed();
//    for ( int iSeed(0); iSeed<numSeeds; ++iSeed) {
//        medAbstractViewCoordinates::qtVector3dToDouble(spad->centerWorld(iSeed), p);
//        rep2d->GetHandleRepresentation(iSeed)->SetWorldPosition(p);
//        rep3d->GetHandleRepresentation(iSeed)->SetWorldPosition(p);

//        int isHighlighted = ( iSeed == selectedSeed ) ? 1 : 0;
//        rep2d->GetHandleRepresentation(iSeed)->Highlight(isHighlighted);
//        rep3d->GetHandleRepresentation(iSeed)->Highlight(isHighlighted);
//    }

    if ( selectedSeed >= 0 ) {
        seedActor2d->RestartInteraction();
        seedActor3d->RestartInteraction();
    } else {
        seedActor2d->CompleteInteraction();
        seedActor3d->CompleteInteraction();
    }

    medAbstractImageView * view = this->getView();
    medVtkViewBackend* backend = static_cast<medVtkViewBackend*>(view->backend());

    //this->getmseg()->currentView()->Modified();

    backend->view2D->Render();
    backend->view3D->Render();

    return;
}

void medAnnIntSeedPointHelper::refreshFromWidget( vtkSeedWidget * spW )
{
    ActorMap::iterator it;
    if ( ! this->findActorMapForWidget(spW, it) ){
        return;
    }

    medSeedPointAnnotationData * spData = qobject_cast<medSeedPointAnnotationData*>(it->first);
    if ( !spData ) 
        return;

    vtkSeedRepresentation * rep = spW->GetSeedRepresentation();
    const int numSeeds = rep->GetNumberOfSeeds();

    if ( numSeeds < spData->getNumberOfSeeds() ) {
        spData->setNumberOfSeeds(numSeeds);
    }

    for ( int iSeed(0); iSeed<numSeeds; ++iSeed) {
        double pos[3];
        spW->GetSeedRepresentation()->GetSeedWorldPosition(iSeed,pos);
//        spData->setCenterWorld(iSeed, medAbstractViewCoordinates::doubleToQtVector3d(pos));
    }
}

medAnnIntSeedPointHelper::ActorMap & medAnnIntSeedPointHelper::getActorMap()
{
    return d->actors;
}

bool medAnnIntSeedPointHelper::findActorMapForWidget(vtkAbstractWidget * w, ActorMap::iterator & it)
{
    for( it = d->actors.begin(); it != d->actors.end(); ++it ) {
        if ( ( it->second.actor2d == w ) ||
            ( it->second.actor3d == w ) ) 
        {
            return true;
        }
    }
    return false;
}