#include "muofuncs.h"
#include "PHIODataNode.h"
#include "dMuiRoadsOutv1.h"
#include "dMuoTrkRoadOutv1.h"
#include "dMuoTracksOutv2.h"
#include "dMuiPseudoTriggerOutv1.h"

#include "mMuoFillTrackOut.hh"
#include "mMuoFillTrkRoadOut.hh"
#include "mMuiFillRoadOut.hh"
#include <iostream>

mMuoFillTrackOut* mMuoFillTrackOutM;
mMuoFillTrkRoadOut* mMuoFillTrkRoadOutM;
mMuiFillRoadOut* mMuiFillRoadOutM;

int init_muo(PHCompositeNode *udstNode)
{

  dMuiRoadsOutv1* dMuiRoadsO = new dMuiRoadsOutv1();

  PHIODataNode<PHObject>* dMuiRoadsOutNode =
    new PHIODataNode<PHObject>(dMuiRoadsO,"dMuiRoadsOut","PHObject");
  udstNode->addNode(dMuiRoadsOutNode);

  dMuoTrkRoadOutv1* dMuoTrkRoadO = new dMuoTrkRoadOutv1();

  PHIODataNode<PHObject>* dMuoTrkRoadOutNode =
    new PHIODataNode<PHObject>(dMuoTrkRoadO,"dMuoTrkRoadOut","PHObject");
  udstNode->addNode(dMuoTrkRoadOutNode);

  dMuoTracksOutv2* dMuoTracksO = new dMuoTracksOutv2();

  PHIODataNode<PHObject>* dMuoTracksOutNode =
    new PHIODataNode<PHObject>(dMuoTracksO,"dMuoTracksOut","PHObject");
  udstNode->addNode(dMuoTracksOutNode);

  mMuoFillTrackOutM = new mMuoFillTrackOut;
  mMuoFillTrkRoadOutM = new mMuoFillTrkRoadOut;
  mMuiFillRoadOutM = new mMuiFillRoadOut;  

  //Copy the dMuiPseudoTrigger Table
  dMuiPseudoTriggerOutv1* muiTrigger = new dMuiPseudoTriggerOutv1;

  PHIODataNode<PHObject>* muiTriggerNode =
    new PHIODataNode<PHObject>(muiTrigger,"dMuiPseudoTriggerOut","PHObject");
  udstNode->addNode(muiTriggerNode);
  
  return 1;
}

int process_muo(PHCompositeNode *topNode)
{
  PHNodeIterator iter(topNode);
  PHCompositeNode* dstNode = static_cast<PHCompositeNode*>(iter.findFirst("PHCompositeNode", "DST"));
  if(dstNode==0)
  {
    cout<<PHWHERE<<" DST node not found.\n";
    return 0;
  }
  PHCompositeNode* udstNode = static_cast<PHCompositeNode*>(iter.findFirst("PHCompositeNode", "UDST"));
  if(udstNode==0)
  {
    cout<<PHWHERE<<" UDST node not found.\n";
    return 0;
  }

  //Let's Find the dMuiRoadsOut if available we'll copy from here.
  PHIODataNode<dMuiRoadsOut>::iterator iDSTRoads(dstNode);
  if(iDSTRoads.find("dMuiRoadsOut")==0)
  {
    //No dMuiRoads so we'll assume we have a dMuiRoadsWrapper
    //cout<<"Filling dMuiRoadsOut from Wrapper\n";
    mMuiFillRoadOutM->event(topNode);
  }else
  {
    PHIODataNode<dMuiRoadsOut>::iterator iUDSTRoads(udstNode);
    if(iUDSTRoads.find("dMuiRoadsOut")==0)
    {
      PHMESSAGE("dMuiRoadsOut not found on the UDST node.");
    }else
    {
      //cout<<"Filling dMuiRoadsOut from Out Class\n";
      (*iUDSTRoads).Reset();
      (*iUDSTRoads).FillFromClass((*iDSTRoads));
    }
  }
  
  //Let's Find the dMuoTrkRoadOut if available we'll copy from here.
  PHIODataNode<dMuoTrkRoadOut>::iterator iDSTTrkRd(dstNode);
  if(iDSTTrkRd.find("dMuoTrkRoadOut")==0)
  {
    //No dMuoTrkRoadOut so we'll assume we have a dMuoTrackRoadRelWrapper
    //cout<<"Filling dMuoTrkRoadOut from Wrapper\n";
    mMuoFillTrkRoadOutM->event(topNode);
  }else
  {
    PHIODataNode<dMuoTrkRoadOut>::iterator iUDSTTrkRd(udstNode);
    if(iUDSTTrkRd.find("dMuoTrkRoadOut")==0)
    {
      PHMESSAGE("dMuoTrkRoadOut not found on the UDST node.");
    }else
    {
      //cout<<"Filling dMuoTrkRoadOut from Out Class\n";
      (*iUDSTTrkRd).Reset();
      (*iUDSTTrkRd).FillFromClass((*iDSTTrkRd));
    }
  }
  //Let's Find the dMuoTracksOut if available we'll copy from here.
  PHIODataNode<dMuoTracksOut>::iterator iDSTTrack(dstNode);
  if(iDSTTrack.find("dMuoTracksOut")==0)
  {
    //No dMuoTracksOut so we'll assume we have a dMuoTracksWrapper
    //cout<<"Filling dMuoTracksOut from Wrapper\n";
    mMuoFillTrackOutM->event(topNode);
  }else
  {
    PHIODataNode<dMuoTracksOut>::iterator iUDSTTrack(udstNode);
    if(iUDSTTrack.find("dMuoTracksOut")==0)
    {
      PHMESSAGE("dMuoTracksOut not found on the UDST node.");
    }else
    {
      //cout<<"Filling dMuoTracksOut from Out Class\n";
      (*iUDSTTrack).Reset();
      (*iUDSTTrack).FillFromClass((*iDSTTrack));
    }
  }
  
  //Let's find the dMuiPseudoTriggerOut
  //Let's find the DST
  {
    PHIODataNode<dMuiPseudoTriggerOut>::iterator dstTrig(dstNode);
    if(dstTrig.find("dMuiPseudoTriggerOut")==0)
    {cout<<PHWHERE<<" dMuiPseudoTriggerOut not found on the DST node.\n";
    }else
    {
      //Now lets find the udst node
      PHCompositeNode* udstNode = static_cast<PHCompositeNode*>(iter.findFirst("PHCompositeNode", "UDST"));
      if(udstNode==0)
      { cout<<PHWHERE<<" UDST node not found.\n";
      }else
      {
        PHIODataNode<dMuiPseudoTriggerOut>::iterator udstTrig(udstNode);
        if(udstTrig.find("dMuiPseudoTriggerOut")==0)
        {cout<<PHWHERE<<" dMuiPseudoTriggerOut not found on the UDST node.\n";
        }else
        {
          //Now that we have found both nodes let's copy
          (*udstTrig).set_SingleShallow((*dstTrig).SingleShallow());
          (*udstTrig).set_DoubleShallow((*dstTrig).DoubleShallow());
          (*udstTrig).set_SingleDeep((*dstTrig).SingleDeep());
          (*udstTrig).set_DeepShallow((*dstTrig).DeepShallow());
          (*udstTrig).set_DoubleDeep((*dstTrig).DoubleDeep());
          (*udstTrig).set_MapType((*dstTrig).MapType());
        }
      }
    }

  }

  return 1;
}