/*
 * Java ATK Wrapper for GNOME
 * Copyright (C) 2015 Magdalen Berns <m.berns@thismagpie.com>
 *
 * 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.GNOME.Accessibility;

import javax.accessibility.*;
import java.lang.ref.WeakReference;

public class AtkTableCell {

    WeakReference<AccessibleContext> _ac;
    WeakReference<AccessibleTable> _acc_pt;
    public int row, rowSpan, column, columnSpan;

    public AtkTableCell (AccessibleContext ac) {
        this._ac = new WeakReference<AccessibleContext>(ac);
        Accessible parent = ac.getAccessibleParent();
        _acc_pt = null;
        row = -1;
        rowSpan = -1;
        column = -1;
        columnSpan = -1;
        if (parent == null){
            return;
        }
        AccessibleContext pc = parent.getAccessibleContext();
        if(pc == null){
            return;
        }
        AccessibleTable pt = pc.getAccessibleTable();
        if(pt == null){
            return;
        }
        _acc_pt = new WeakReference<AccessibleTable>(pt);
        int index = ac.getAccessibleIndexInParent();
        if (!(pt instanceof AccessibleExtendedTable)){
            return;
        }
        AccessibleExtendedTable aet = (AccessibleExtendedTable) pt;
        row = aet.getAccessibleRow(index);
        column = aet.getAccessibleColumn(index);
        rowSpan = pt.getAccessibleRowExtentAt(row,column);
        columnSpan = pt.getAccessibleColumnExtentAt(row,column);
    }

    public static AtkTableCell createAtkTableCell(AccessibleContext ac){
        return AtkUtil.invokeInSwing ( () -> { return new AtkTableCell(ac); }, null);
    }

    /**
    * getTable
    * @return: Reference to the accessible of the containing table as an
    *          AccessibleTable instance.
    */
    public AccessibleTable getTable() {
        if (_acc_pt == null)
            return null;
        return _acc_pt.get();
    }

    public AccessibleContext[] getAccessibleColumnHeader(){
        if (_acc_pt == null)
            return null;
        return AtkUtil.invokeInSwing( () -> {
            AccessibleTable iteration = _acc_pt.get().getAccessibleColumnHeader();
            if (iteration != null){
                int length = iteration.getAccessibleColumnCount();
                AccessibleContext[] result = new AccessibleContext[length];
                for (int i = 0; i < length; i++){
                    result[i] = iteration.getAccessibleAt(0,i).getAccessibleContext();
                }
                return result;
            }
            return null;
        }, null);
    }

    public AccessibleContext[] getAccessibleRowHeader(){
        if (_acc_pt == null)
            return null;
        return AtkUtil.invokeInSwing( () -> {
            AccessibleTable iteration = _acc_pt.get().getAccessibleRowHeader();
            if (iteration != null){
                int length = iteration.getAccessibleRowCount();
                AccessibleContext[] result = new AccessibleContext[length];
                for (int i = 0; i < length; i++){
                    result[i] = iteration.getAccessibleAt(i,0).getAccessibleContext();
                }
                return result;
            }
            return null;
        }, null);
    }

}