//<<<<<< INCLUDES                                                       >>>>>>

#include "Utilities/Configuration/interface/Architecture.h"
#include "Visualisation/MuonVis/interface/VisMuBarRpcSimHitsTwig.h"
#include "Visualisation/MuonVis/interface/VisMuBarRpcMyHitsTwig.h"
#include "Visualisation/MuonVis/interface/VisMuBarSegmentTwig.h"
#include "Visualisation/MuonVis/interface/VisMuBarSimHitsTwig.h"
#include "Visualisation/MuonVis/interface/VisMuDataProxy.h"
#include "Visualisation/MuonVis/interface/VisMuEndRecHitsTwig.h"
#include "Visualisation/MuonVis/interface/VisMuEndRpcSimHitsTwig.h"
#include "Visualisation/MuonVis/interface/VisMuEndSegmentTwig.h"
#include "Visualisation/MuonVis/interface/VisMuEndSimHitsTwig.h"
#include "Visualisation/MuonVis/interface/VisMuRecTracksTwig.h"
#include "Visualisation/MuonVis/interface/VisMuSimTracksTwig.h"
#include "Visualisation/MuonVis/interface/VisMuTwig.h"
#include "Ig_Modules/IgStudio/interface/IgDocumentData.h"
#include "Ig_Framework/IgObjectBrowser/interface/IgRepSet.h"
#include <qapplication.h>

//<<<<<< PRIVATE DEFINES                                                >>>>>>
//<<<<<< PRIVATE CONSTANTS                                              >>>>>>
//<<<<<< PRIVATE TYPES                                                  >>>>>>
//<<<<<< PRIVATE VARIABLE DEFINITIONS                                   >>>>>>
//<<<<<< PUBLIC VARIABLE DEFINITIONS                                    >>>>>>
//<<<<<< CLASS STRUCTURE INITIALIZATION                                 >>>>>>

IG_DEFINE_STATE_ELEMENT (VisMuDataProxy, "Data/COBRA/Event/Muon");

//<<<<<< PRIVATE FUNCTION DEFINITIONS                                   >>>>>>
//<<<<<< PUBLIC FUNCTION DEFINITIONS                                    >>>>>>
//<<<<<< MEMBER FUNCTION DEFINITIONS                                    >>>>>>

VisMuDataProxy::VisMuDataProxy (IgState *state)
    : VisDataProxy (state, s_key, EVENT_THREAD,
		    create_callback (this, &VisMuDataProxy::init))
{
}

void
VisMuDataProxy::init (void) 
{
    qApp->lock ();
    
    // FIXME: Do something here if Event is not found: create
    // a parent Twig?
    IgTwig *eventTwig = IgDocumentData::get (state ())->find ("Event");
    IgTwig *muonTwig = new VisMuTwig (state (), eventTwig);
    IgTwig *muonBarrel = new IgSimpleTwig (muonTwig, "Barrel", true, false, true);
    IgTwig *muonEndcap = new IgSimpleTwig (muonTwig, "Endcap", true, false, true);

    new VisMuBarSimHitsTwig (state (), muonBarrel);
    new VisMuBarRpcSimHitsTwig (state (), muonBarrel);
    new VisMuBarRpcMyHitsTwig (state (), muonBarrel);
    new VisMuBarSegmentTwig (state (), muonBarrel);

    new VisMuEndSimHitsTwig (state (), muonEndcap);
    new VisMuEndRpcSimHitsTwig (state (), muonEndcap);
    new VisMuEndRecHitsTwig (state (), muonEndcap);
    new VisMuEndSegmentTwig (state (), muonEndcap);

    new VisMuSimTracksTwig (state (), muonTwig);
    new VisMuRecTracksTwig (state (), muonTwig);
    IgRepSet::update (eventTwig, IgTwig::STRUCTURE_MASK);

    qApp->unlock (false);
}