inspectorJ -- JavaTM Profiler

sf project site browse source checkout source

src/inspectorj/model/classtablemodel.cpp

Go to the documentation of this file.

/***************************************************************************
 *   inspectorJ - java profiler                                            *
 *   Copyright (C) 2007 by  James May
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 *   This program is distributed in the hope that it will be useful,       *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program; if not, write to the                         *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/

#include "inspectorj/model/classtablemodel.h"
#include <iostream>

namespace inspectorj {
namespace model {

using namespace inspectorj::jdwp;
using inspectorj::toolset::ToolSetUtils;

bool javaClassSort(JavaClass* lhs, JavaClass* rhs)
{
    return *lhs < *rhs;
}

ClassTableModel::ClassTableModel(QVector<JavaClass*> &classes, SessionProfile **prof)
    : QAbstractTableModel(),
      totalClassesLoaded(0), 
      filteredClassesLoaded(0),
      profile(prof),
      classes(classes)
{

}

ClassTableModel::~ClassTableModel()
{
}

int ClassTableModel::rowCount(const QModelIndex &parent) const
{
    return classes.size();
}

int ClassTableModel::columnCount(const QModelIndex &parent) const
{
    return 1;
}

QVariant ClassTableModel::data(const QModelIndex &index, int role) const
{
    if (!index.isValid())
        return QVariant();
    
    if (role == Qt::TextAlignmentRole) {
        return int(Qt::AlignLeft | Qt::AlignVCenter);
    } else if (role == Qt::DisplayRole) {
        return classes[index.row()]->getClassSignature();
    }
    return QVariant();    
}

QVariant ClassTableModel::headerData(int section, Qt::Orientation orientation,
                    int role) const
{
    if(orientation == Qt::Horizontal) {
        if (role == Qt::TextAlignmentRole) {
            return int(Qt::AlignLeft | Qt::AlignVCenter);
        } else if (role == Qt::DisplayRole) {
            switch( section ){
                case 0 : return QVariant(" Class References "); break;
            }   
        } 
        return QVariant();        
    }
    if(orientation == Qt::Vertical) {
        if (role == Qt::DisplayRole) {
            return QVariant("");  
        } 
        return QVariant();        
    }
    return QVariant();
}

void ClassTableModel::initModel(JDWPPacket& packet)
{
    filteredClassesLoaded = 0;
    regexFilterList.clear();
    if (profile) {
        QStringList filters = (*profile)->getClassFilters();
        QStringListIterator filterIter(filters);
        while(filterIter.hasNext()) {
            QRegExp regexp;
            ToolSetUtils::createClassFilter(regexp, filterIter.next());
            regexFilterList << regexp;
        }
    }
    classes.clear();
    int count;
    packet >> count;
    totalClassesLoaded = count;
    
    while (count > 0) {

        jbyte refTypeTag;
        int status;
        ReferenceTypeID refTypeId;
        QString classSignature;
        
        packet >> refTypeTag;
        packet >> refTypeId;
        packet >> classSignature;
        packet >> status;
        
        QList<QRegExp>::iterator iter;
        iter = regexFilterList.begin();
        if (iter == regexFilterList.end()) {            
            // no filters evidently so add all classes
            
            JavaClass *javaClass = new JavaClass();
            
            javaClass->setRefTypeTag(refTypeTag);
            javaClass->setRefTypeId(refTypeId);
            javaClass->setJNIClassSignature(classSignature);
            javaClass->setClassSignature(ToolSetUtils::normalizeClassSignature(classSignature));
            javaClass->setStatus(status);
            
            classes.append(javaClass);
            filteredClassesLoaded++;
        }
        else {
            while (iter != regexFilterList.end()) {
                QRegExp regexp;
                regexp = *iter;

                if (classSignature.contains(regexp)) {
                    JavaClass *javaClass = new JavaClass();
                    
                    javaClass->setRefTypeTag(refTypeTag);
                    javaClass->setRefTypeId(refTypeId);
                    javaClass->setJNIClassSignature(classSignature);
                    javaClass->setClassSignature(ToolSetUtils::normalizeClassSignature(classSignature));
                    javaClass->setStatus(status);                        
                    
                    classes.append(javaClass);                                                
                    // keep track of how many classes match our filters
                    filteredClassesLoaded++;
                }
                ++iter;
            }
        }
        count--;
    } 
    qSort(classes.begin(), classes.end(), javaClassSort);   
}

int ClassTableModel::getTotalClassesLoaded()
{
    return totalClassesLoaded;
}

int ClassTableModel::getFilteredClassesLoaded()
{
    return filteredClassesLoaded;
           
}

const JavaClass* ClassTableModel::getClassAtIndex(const QModelIndex& index)
{
    return classes.at(index.row());
}

} // end namespace inspectorj
} // end namespace model

---