GitModel.h

Go to the documentation of this file.

// This may look like C code, but it's really -*- C++ -*-
/*
 * Copyright (C) 2008 Emweb bv, Herent, Belgium.
 *
 * See the LICENSE file for terms of use.
 */
#ifndef GIT_MODEL_H_
#define GIT_MODEL_H_

#include <Wt/WAbstractItemModel>

#include "Git.h"


class GitModel : public Wt::WAbstractItemModel
{
public:
  static const int ContentsRole = Wt::UserRole;
  static const int FilePathRole = Wt::UserRole + 1;

  GitModel(Wt::WObject *parent = 0);

  void setRepositoryPath(const std::string& repositoryPath);

  void loadRevision(const std::string& revName);

  virtual Wt::WModelIndex parent(const Wt::WModelIndex& index) const;

  virtual int columnCount(const Wt::WModelIndex& parent = Wt::WModelIndex())
    const;

  virtual int rowCount(const Wt::WModelIndex& parent = Wt::WModelIndex()) const;

  virtual Wt::WModelIndex
  index(int row, int column, const Wt::WModelIndex& parent = Wt::WModelIndex())
    const;

  virtual boost::any
  data(const Wt::WModelIndex& index, int role = Wt::DisplayRole) const;
  
  virtual boost::any
  headerData(int section, Wt::Orientation orientation = Wt::Horizontal,
             int role = Wt::DisplayRole) const;

  using WAbstractItemModel::data;

private:
  Git git_;

  struct ChildIndex {
    int parentId;
    int index;

    ChildIndex(int aParent, int anIndex)
      : parentId(aParent), index(anIndex) { }

    bool operator< (const ChildIndex& other) const {
      if (parentId < other.parentId)
        return true;
      else if (parentId > other.parentId)
        return false;
      else return index < other.index;
    }
  };

  class Tree {
  public:
    Tree(int parentId, int index, const Git::ObjectId& object,
         int rowCount)
      : index_(parentId, index),
        treeObject_(object),
        rowCount_(rowCount)
    { }

    int parentId() const { return index_.parentId; }

    int index() const { return index_.index; }

    const Git::ObjectId& treeObject() const { return treeObject_; }

    int rowCount() const { return rowCount_; }

  private:
    ChildIndex    index_;
    Git::ObjectId treeObject_;
    int           rowCount_;
  };

  typedef std::map<ChildIndex, int> ChildPointerMap;

  mutable std::vector<Tree> treeData_;

  mutable ChildPointerMap   childPointer_;

  int getTreeId(int parentId, int childIndex) const;

  Git::Object getObject(const Wt::WModelIndex& index) const;
};

#endif // GIT_MODEL_H_