/*

 * $Id$

 *

 * Copyright (C) 2003-2009 JNode.org

 *

 * This library is free software; you can redistribute it and/or modify it

 * under the terms of the GNU Lesser General Public License as published

 * by the Free Software Foundation; either version 2.1 of the License, or

 * (at your option) any later version.

 *

 * This library 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 Lesser General Public 

 * License for more details.

 *

 * You should have received a copy of the GNU Lesser General Public License

 * along with this library; If not, write to the Free Software Foundation, Inc., 

 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

 */

package org.jnode.vm.memmgr.generational;

import org.jnode.annotation.MagicPermission;

import org.jnode.vm.ObjectVisitor;

import org.jnode.vm.VmMagic;

import org.jnode.vm.classmgr.ObjectFlags;

import org.vmmagic.pragma.Uninterruptible;

/**

 * @author Loic Rouchon (loic.rouchon@insa-lyon.fr)

 */

@MagicPermission

final class GenGCSweepVisitor extends ObjectVisitor implements ObjectFlags, Uninterruptible {

    private GenGCYellowObjectManager yellowObjectManager;

    private long markedObjectsSize = 0L;

    public GenGCSweepVisitor(GenGCYellowObjectManager yellowObjectManager) {

        this.yellowObjectManager = yellowObjectManager;

    }

    /**

     * Mark every visited object white.

     * 

     * @param object

     * @return boolean

     */

    @Override

    public final boolean visit(Object object) {

        if (VmMagic.getObjectColor(object) == GC_YELLOW) {

            yellowObjectManager.deleteObject(object);

            markedObjectsSize += VmDefaultHeap.getObjectSize(object);

        }

        return true;

    }

    /**

     * @param currentHeap

     *            The currentHeap to set.

     */

    public final void setCurrentHeap(VmDefaultHeap currentHeap) {

        yellowObjectManager.setCurrentHeap(currentHeap);

    }

    public void resetMarkedObjectsSize() {

        markedObjectsSize = 0L;

    }

    public long getMarkedObjectsSize() {

        return markedObjectsSize;

    }

}