/*                                                                              
 * PDFedit - free program for PDF document manipulation.                        
 * Copyright (C) 2006, 2007  PDFedit team:      Michal Hocko, 
 *                                              Miroslav Jahoda,       
 *                                              Jozef Misutka, 
 *                                              Martin Petricek                                             
 *
 * Project is hosted on http://sourceforge.net/projects/pdfedit                                                                      
 */ 
/** @file
 TreeItemPage - class holding CPage in tree, descendant of TreeItemAbstract
 @author Martin Petricek
*/

#include "treeitempage.h"
#include "qspage.h"
#include "treedata.h"
#include "treeitem.h"
#include "treeitemannotation.h"
#include "treeitemcontentstream.h"
#include "treeitemdict.h"
#include "treeitempageobserver.h"
#include "util.h"
#include <cannotation.h>
#include <cobject.h>
#include <cpage.h>
#include <cpdf.h>
#include <qobject.h>

namespace gui {

using namespace std;
using namespace pdfobjects;

/** constructor of TreeItemPage - create root item from given object
 @param _data TreeData containing necessary information about tree in which this item will be inserted
 @param parent Q_ListView in which to put item
 @param _page CPage Object contained in this item
 @param name Name of this item - will be shown in treeview
 @param after Item after which this one will be inserted
 */
TreeItemPage::TreeItemPage(TreeData *_data,boost::shared_ptr<CPage> _page,Q_ListView *parent,const QString name/*=QString::null*/,Q_ListViewItem *after/*=NULL*/):TreeItemAbstract(name,_data,parent,after) {
 init(_page,name);
}

/** constructor of TreeItemPage - create child item from given object
 @param _data TreeData containing necessary information about tree in which this item will be inserted
 @param parent Q_ListViewItem which is parent of this object
 @param _page CPage Object contained in this item
 @param name Name of this item - will be shown in treeview
 @param after Item after which this one will be inserted
 */
TreeItemPage::TreeItemPage(TreeData *_data,boost::shared_ptr<CPage> _page,Q_ListViewItem *parent,const QString name/*=QString::null*/,Q_ListViewItem *after/*=NULL*/):TreeItemAbstract(name,_data,parent,after) {
 init(_page,name);
}

/** Initialize item from given CPage
 @param page Object used to initialize this item
 @param name Name of this item - will be shown in treeview
 */
void TreeItemPage::init(boost::shared_ptr<CPage> page,const QString &name) {
 obj=page;
 // object name
 if (name.isNull()) {
  setText(0,QObject::tr("<no name>"));
 } else {
  setText(0,name);
 }
 // object type
 setText(1,QObject::tr("Page"));
 setDragEnabled(true);//Drag drop enabled for this item
 reload(false);//get childs
 initObserver();
}

/** Sets observer for this item */
void TreeItemPage::initObserver() {
 guiPrintDbg(debug::DBG_DBG,"Set Observer");
 observer=boost::shared_ptr<TreeItemPageObserver>(new TreeItemPageObserver(this));
 obj->registerObserver(observer);
}

/** Unsets observer for this item */
void TreeItemPage::uninitObserver() {
 observer->deactivate();
 obj->unregisterObserver(observer);
 observer.reset();
 guiPrintDbg(debug::DBG_DBG,"UnSet Observer");
}

/**
 return CPage stored inside this item
 @return stored object (CPage)
*/
boost::shared_ptr<CPage> TreeItemPage::getObject() {
 return obj;
}

/** 
 This method is needed for "deep reload" to work
 Try to replace page inside this treeitem with a new one.
 @param newPage new page to replace the old one
 @return true if page replaced, false on error.
*/
bool TreeItemPage::setObject(boost::shared_ptr<CPage> newPage) {
 obj=newPage;
 return true;
}

/** default destructor */
TreeItemPage::~TreeItemPage() {
 uninitObserver();
}

//See TreeItemAbstract for description of this virtual method
TreeItemAbstract* TreeItemPage::createChild(const QString &name,ChildType typ,Q_ListViewItem *after/*=NULL*/) {
 if (typ==0) { //Return page dictionary
  return TreeItem::create(data,this,obj->getDictionary(),QObject::tr("Dictionary"),after);
 }
 if (typ==1) {
  size_t streamNumber=name.mid(1).toUInt();
  if (streamNumber>=streams.size()) {
   //Invalid or old data -> try to re-get list of streams from CPage
   obj->getContentStreams(streams);
  }
  assert(streamNumber<streams.size());  //Still Invalid data ?
  QString num=QString::number(streamNumber);
  return new TreeItemContentStream(data,this,streams[streamNumber],QObject::tr("Stream")+" "+num,after,QString("Stream")+num);
 }
 if (typ==2) {
  size_t annotNumber=name.mid(1).toUInt();
  if (annotNumber>=anots.size()) {
   //Invalid or old data -> try to re-get list of s from CPage
   obj->getAllAnnotations(anots);
  }
  assert(annotNumber<anots.size());  //Still Invalid data ?
  QString num=QString::number(annotNumber);
  return new TreeItemAnnotation(data,this,anots[annotNumber],obj,QObject::tr("Annotation")+" "+num,after,QString("Annotation")+num);
 }
 assert(0);
 return NULL;
}

//See TreeItemAbstract for description of this virtual method
bool TreeItemPage::deepReload(__attribute__((unused)) const QString &childName,Q_ListViewItem *oldItem) {
 TreeItemDict *itc=dynamic_cast<TreeItemDict*>(oldItem);
 if (itc) { //Is a page dictionary
  //If replaced, return success, otherwise failure
  guiPrintDbg(debug::DBG_DBG,"Replacing page dictionary: is_same = " << (obj->getDictionary().get()==itc->getObject().get()));
  return itc->setObject(obj->getDictionary());
 }
 //TODO: support stream, annotation ?
 //Anything else=not supported
 return false;
}


//See TreeItemAbstract for description of this virtual method
bool TreeItemPage::validChild(const QString &name,Q_ListViewItem *oldChild) {
 TreeItemDict *itp=dynamic_cast<TreeItemDict*>(oldChild);
 if (itp) { //Is a page dictionary
  return obj->getDictionary().get()==itp->getObject().get();
 }
 TreeItemContentStream *its=dynamic_cast<TreeItemContentStream*>(oldChild);
 if (its && name[0]=='s') { //Is a content stream
  size_t streamNumber=name.mid(1).toUInt();
  assert(streamNumber<streams.size());  //These should be already weed out ... 
  return streams[streamNumber].get()==its->getObject().get();
 }
 TreeItemAnnotation *ita=dynamic_cast<TreeItemAnnotation*>(oldChild);
 if (ita && name[0]=='a') { //Is an annotation
  size_t annotNumber=name.mid(1).toUInt();
  assert(annotNumber<anots.size());  //These should be already weed out ... 
  return anots[annotNumber].get()==ita->getObject().get();
 }
 //Something else
 return false;
}


//See TreeItemAbstract for description of this virtual method
ChildType TreeItemPage::getChildType(const QString &name) {
 if (name=="Dict") { //Return page dictionary
  return 0;
 } else { //It's a stream
  char c=name[0].latin1();
  if (c=='s') return 1;//Stream
  if (c=='a') return 2;//Annot
  assert(0);
  return -1;
 }
}

//See TreeItemAbstract for description of this virtual method
QStringList TreeItemPage::getChildNames() {
 QStringList childs;
 if (data->showODict()) childs+="Dict";
 if (data->showStream()) {
  obj->getContentStreams(streams);
  for (size_t i=0;i<streams.size();i++) {
   childs+=QString("s")+QString::number(i);
  }
 }
 if (data->showAnnot()) {
  obj->getAllAnnotations(anots);
  for (size_t i=0;i<anots.size();i++) {
   childs+=QString("a")+QString::number(i);
  }
 }
 return childs;
}

//See TreeItemAbstract for description of this virtual method
bool TreeItemPage::haveChild() {
 if (data->showODict()) return true;
 if (data->showStream()) return true;
 if (data->showAnnot()) return true;
 return false;
}

//See TreeItemAbstract for description of this virtual method
void TreeItemPage::reloadSelf() {
 //Reload list of streams. Might be needed later
 obj->getContentStreams(streams);
 obj->getAllAnnotations(anots);
}

//See TreeItemAbstract for description of this virtual method
QSCObject* TreeItemPage::getQSObject() {
 return new QSPage(obj,data->base());
}

//See TreeItemAbstract for description of this virtual method
QSCObject* TreeItemPage::getQSObject(BaseCore *_base) {
 //CPage uses shared pointer, so it can be copied safely into another document
 return new QSPage(obj,_base);
}

//See TreeItemAbstract for description of this virtual method
void TreeItemPage::remove() {
 CPdf * pdf=obj->getDictionary()->getPdf();
 if (!pdf) {
  //The page is obviously not in any pdf, so it can't be removed
  return;
 }
 try {
  //Remove the page from document
  size_t pPos=pdf->getPagePosition(obj);
  pdf->removePage(pPos);
  //Page is now removed
 } catch (...) {
  //Some exception, like page was not found in PDF
 }
}

} // namespace gui