// Emacs: -*- C++ -*-

//
//	Copyright 1993, Center for Computer Vision and Visualization,
//	University of Florida.  All rights reserved.
//


//
// $Log: nbh-reduction,v $
// Revision 1.3  1994/07/25  17:26:39  thoth
// Name sanitization
//
// Revision 1.2  1994/05/08  19:43:05  thoth
// New automatic neighborhood decomposition capability.
//
// Revision 1.1  1994/01/31  16:50:30  thoth
// Initial revision
//

#if 0
#ifndef zero_extend_scan_IBTYPE
#define zero_extend_scan_IBTYPE

static void zero_extend_vec_scan(const IA_SetStructure &ss,
				 const IBTYPE **src,
				 IBTYPE **dest,
				 IBTYPE zero)
{
    for (unsigned i=0; i<ss.nintervals(); i++) {
	IA_ss_interval	temp(ss.retrieve_interval(i));
	if (temp.substructure == IA_SetStructure::FIRST_ONLY) {
	    (*src) += temp.count;
	} else if (temp.substructure == IA_SetStructure::BOTH) {
	    for (unsigned j=0; j<temp.count; j++) {
		*((*dest)++) = *((*src)++);
	    }
	} else if (temp.substructure == IA_SetStructure::SECOND_ONLY) {
	    for (unsigned j=0; j<temp.count; j++)
		*((*dest)++) = zero;
	} else {
	    for (unsigned j=0; j<temp.count; j++)
		zero_extend_vec_scan(temp.substructure, src, dest, zero);
	}
    }
}

static void zero_extend_iter_scan(const IA_SetStructure &ss,
				  IA_IVIter<IA_Point<int>,IBTYPE> *srciter,
				  IBTYPE **dest,
				  IBTYPE zero)
{
    for (unsigned i=0; i<ss.nintervals(); i++) {
	IA_ss_interval	temp(ss.retrieve_interval(i));
	if (temp.substructure == IA_SetStructure::FIRST_ONLY) {
	    IBTYPE	blah;
	    for (unsigned j=0; j<temp.count; j++) 
		(*srciter)(blah);
	} else if (temp.substructure == IA_SetStructure::BOTH) {
	    IBTYPE	blah;
	    for (unsigned j=0; j<temp.count; j++) {
		(*srciter)(blah);
		*((*dest)++) = blah;
	    }
	} else if (temp.substructure == IA_SetStructure::SECOND_ONLY) {
	    for (unsigned j=0; j<temp.count; j++)
		*((*dest)++) = zero;
	} else {
	    for (unsigned j=0; j<temp.count; j++)
		zero_extend_iter_scan(temp.substructure, srciter, dest, zero);
	}
    }
}

static void zero_extend(const IA_CoreImage<IA_Point<int>,IBTYPE> &srcimg,
			const IA_Set<IA_Point<int> > &dest_ps, IBTYPE *dest_data,
			IBTYPE zero)
{
    IA_SetStructure	ss;
    intersect_with_dualstruct(srcimg.domain(), dest_ps, &ss);

    if (srcimg.type() == IA_VectorI<IA_Point<int>,IBTYPE>::s_type()) {
	IBTYPE	*srcdata = ((IA_VectorI<IA_Point<int>,IBTYPE>*)srcimg.bip)->vec;
	zero_extend_vec_scan(ss, &srcdata, &dest_data, zero);
    } else {
	IA_IVIter<IA_Point<int>,IBTYPE>	iter(srcimg);
	zero_extend_iter_scan(ss, &iter, &dest_data, zero);
    }
}

#endif
#endif

//
// backward neighborhood reductions
//

#ifdef NEED_SPARE
static void
backw_CONV_inv_core(IA_Point<int> src_infimum,
		    IA_Point<int> src_width,
		    const RTYPE *src_data, // length is prod(src_width)
		    const IA_Set<IA_Point<int> > &nbh,
		    IA_Set<IA_Point<int> > dest_ps,
		    RTYPE *const dest_data)
{
    const int	dimen = src_width.dim();

    int		nbh_sz = nbh.card();
    int	*const nbh_offsets = new int[nbh_sz];

    {
	int	*o_scan = nbh_offsets;
	IA_PSIter<IA_Point<int> >	iter(nbh);
	IA_Point<int>	ip;
	while (iter(ip)) {
	    *o_scan = ip[0];
	    for (unsigned i=1; i<dimen; i++) {
		*o_scan *= src_width[i];
		*o_scan += ip[i];
	    }
	    o_scan++;
	}
    }

    IA_PSIter<IA_Point<int> >	iter(dest_ps);
    IA_Point<int>	ip;
    //RTYPE *const	dest_data = new RTYPE[dest_ps.card()];
    RTYPE *	valp = dest_data;
    while (iter(ip)) {
	int	offset= ip[0] - src_infimum[0];
	unsigned i;
	for (i=1; i<dimen; i++) {
	    offset *= src_width[i];
	    offset += ip[i] - src_infimum[i];
	}
	const RTYPE *const base = src_data + offset;

	// _IVAL_=(base[nbh_offsets[i]])
	// _IRESULT_=(*valp)

	// INITIALIZE ;

	for (i=0; i<nbh_sz; i++) {
	    ACCUMULATE ;
	}
	//{
	//    RESULT ;
	//}
	valp++;
    }
    delete[] nbh_offsets;

    //return dest_data;
}
#else
static void
backw_CONV_inv_core(IA_Point<int> src_infimum,
		    IA_Point<int> src_width,
		    const RTYPE *src_data, // length is prod(src_width)
		    const IA_Set<IA_Point<int> > &nbh,
		    IA_Set<IA_Point<int> > dest_ps,
		    RTYPE *const dest_data);
#endif

static void
backw_CONV_inv_core(IA_Point<int> src_infimum,
		    IA_Point<int> src_width,
		    const IBTYPE *src_data, // length is prod(src_width)
		    const IA_Set<IA_Point<int> > &nbh,
		    IA_Set<IA_Point<int> > dest_ps,
		    RTYPE *const dest_data)
{
    const int	dimen = src_width.dim();

    int		nbh_sz = nbh.card();
    int	*const nbh_offsets = new int[nbh_sz];

    {
	int	*o_scan = nbh_offsets;
	IA_PSIter<IA_Point<int> >	iter(nbh);
	IA_Point<int>	ip;
	while (iter(ip)) {
	    *o_scan = ip[0];
	    for (unsigned i=1; i<dimen; i++) {
		*o_scan *= src_width[i];
		*o_scan += ip[i];
	    }
	    o_scan++;
	}
    }

    if (dest_ps.boxy()) {

	IA_Point<int>	inf_ = dest_ps.inf();
	IA_Point<int>	sup_ = dest_ps.sup();
	IA_Point<int>	dest_width = sup_-inf_+1;

	int	offset = inf_ [0] - src_infimum[0];
	int	i;
	for (i=1; i<dimen; i++) {
	    offset *= src_width[i];
	    offset += inf_[i] - src_infimum[i];
	}

	IA_Point<int>	bases = extend_to_point(0, inf_.dim());
	i = dimen;
	bases[i-1] = 1;
	for (i--; i>0; i--) {
	    bases[i-1] = bases[i]*src_width[i];
	}

	IA_Point<int>	ip = extend_to_point(0, inf_.dim());

	RTYPE *	valp = dest_data;

	do {
	    const IBTYPE *const base = src_data + offset;

	    // _IVAL_=(base[nbh_offsets[i]])
	    // _IRESULT_=(*valp)

	    for (i=0; i<nbh_sz; i++) {
		ACCUMULATE ;
	    }
	    valp++;

	    // now we go to next point
	    for (i=dimen-1; i>=0; i--) {
		ip[i]++;
		offset+=bases[i];
		if (ip[i]<dest_width[i])
		    break;
		ip[i] = 0;
		offset -= dest_width[i]*bases[i];
	    }
	    // no more code here, might corrupt i
	} while (i>=0);

    } else {

	IA_PSIter<IA_Point<int> >	iter(dest_ps);
	IA_Point<int>	ip;
	//RTYPE *const	dest_data = new RTYPE[dest_ps.card()];
	RTYPE *	valp = dest_data;
	while (iter(ip)) {
	    int	offset= ip[0] - src_infimum[0];
	    unsigned i;
	    for (i=1; i<dimen; i++) {
		offset *= src_width[i];
		offset += ip[i] - src_infimum[i];
	    }
	    const IBTYPE *const base = src_data + offset;

	    // _IVAL_=(base[nbh_offsets[i]])
	    // _IRESULT_=(*valp)

	    // INITIALIZE ;

	    for (i=0; i<nbh_sz; i++) {
		ACCUMULATE ;
	    }
	    //{
	    //    RESULT ;
	    //}
	    valp++;
	}

    }				// end if boxy()

    delete[] nbh_offsets;

    //return dest_data;
}

static void
backw_decomp_CONV_inv(IA_Point<int> src_infimum,
		      IA_Point<int> src_width,
		      const IBTYPE	*src_data,
		      IA_Set<IA_Point<int> >	invnbh,
		      IA_Set<IA_Point<int> >	dest_ps,
		      RTYPE *dest_data)
{
    const int	sz = dest_ps.card();

    int	i;

    IA_PointSetDecomposition	decomp(invnbh);
    for (i=0; i<decomp.nfragments(); i++) {
	const IA_decomp_frag	&frag = decomp.get_frag(i);
	if (frag.plate.card()==1) {
	    backw_CONV_inv_core(src_infimum, src_width, src_data,
				       frag.bar, dest_ps, dest_data);
	} else {
	    IA_Point<int>	inf_ = dest_ps.inf() + frag.bar.inf();
	    IA_Point<int>	sup_ = dest_ps.sup() + frag.bar.sup();
	    IA_Set<IA_Point<int> >	intermed_ps(inf_, sup_);
	    const int sz = intermed_ps.card();
	    RTYPE	*intermed = new RTYPE[sz];
	    for (int j=0; j<sz; j++)
		intermed[j] = ZERO;
	    backw_decomp_CONV_inv
		(src_infimum, src_width, src_data,
		 frag.plate, intermed_ps, intermed);
	    backw_CONV_inv_core(inf_, sup_ - inf_ + 1, intermed,
				frag.bar, dest_ps, dest_data);
	    delete[] intermed;
	}
    }

}

static IA_Image<IA_Point<int>,RTYPE>
backw_CONV_inv(const IA_Image<IA_Point<int>,IBTYPE> &img,
	 const IA_Set<IA_Point<int> > &invnbh,
	 IA_Set<IA_Point<int> > dest_ps)
{
    IA_Point<int>	inf_ = dest_ps.inf()+invnbh.inf();
    IA_Point<int>	sup_ = dest_ps.sup()+invnbh.sup();
    IA_Set<IA_Point<int> >	src_ps = IA_Set<IA_Point<int> >(inf_,sup_);

    IBTYPE	*const src_data = new IBTYPE[src_ps.card()];

    zero_extend(img, src_ps, (IBTYPE*)src_data, IBTYPE(ZERO));

#if 1
    const int sz = dest_ps.card();
    RTYPE	*const dest_data = new RTYPE[sz];
    for (int i=0; i<sz; i++)
	dest_data[i] = ZERO;
    if (IA_Neighborhood<IA_Point<int>, IA_Point<int> >::auto_decomposition)
	backw_decomp_CONV_inv (inf_, sup_-inf_ + 1, src_data,
			       invnbh, dest_ps, dest_data);
    else
	backw_CONV_inv_core(inf_, sup_-inf_ + 1, src_data,
			       invnbh, dest_ps, dest_data);
    IA_Image<IA_Point<int>,RTYPE> rval (dest_ps, dest_data, dest_ps.card(), 1);
#else
    IA_Image<IA_Point<int>,RTYPE>	rval =
	backw_CONV_inv_core(inf_, sup_-inf_ + 1, src_data, invnbh, dest_ps);
#endif

    delete[] src_data;

    return rval;
}

IA_Image<IA_Point<int>,RTYPE>
CONV(const IA_CoreImage<IA_Point<int>,IBTYPE> &img_,
     const IA_Neighborhood<IA_Point<int>,IA_Point<int> > &nbh,
     IA_Set<IA_Point<int> > dest_ps)
{
    IA_Image<IA_Point<int>,IBTYPE>	img(img_);

    if (nbh.type() == IA_InvariantNbh<IA_Point<int>,IA_Point<int> >::s_type()) {
	return backw_CONV_inv
	    (img,
	     ((IA_InvariantNbh<IA_Point<int>,IA_Point<int> >*) nbh.bnbh)->value,
	     dest_ps);
    } else {
	RTYPE *const	dest_data = new RTYPE[dest_ps.card()];
	RTYPE	*valp = dest_data;
	IA_PSIter<IA_Point<int> >	dest_iter(dest_ps);
	IA_Point<int>	base_ip;
	while (dest_iter(base_ip)) {
	    IA_Set<IA_Point<int> >	ps = nbh(base_ip);

	    IA_PSIter<IA_Point<int> >	nbh_iter(ps);
	    IA_Point<int>	nbh_ip;

	    // _IVAL_=img(nbh_ip)
	    // _IRESULT_=(*valp)

	    INITIALIZE ;

	    while ( nbh_iter(nbh_ip) ) {
		if (! img.domain().contains(nbh_ip))
		    continue;
		ACCUMULATEVAR ;
	    }
	    {
		RESULT ;
	    }
	    valp++;
	}
	return IA_Image<IA_Point<int>,RTYPE>(dest_ps, dest_data,
					   dest_ps.card(), 1);
    }
}

//
// foward neighborhood reductions
//

static IA_Image<IA_Point<int>,RTYPE>
forw_CONV_inv_core(IA_Point<int> src_infimum,
	      IA_Point<int> src_width,
	      const IBTYPE *src_data, // length is prod(src_width)
	      const IA_Set<IA_Point<int> > &nbh,
	      IA_Set<IA_Point<int> > dest_ps)
{
    const int	dimen = src_width.dim();

    int		nbh_sz = nbh.card();
    int	*const nbh_offsets = new int[nbh_sz];

    {
	int	*o_scan = nbh_offsets;
	IA_PSIter<IA_Point<int> >	iter(nbh);
	IA_Point<int>	ip;
	while (iter(ip)) {
	    *o_scan = -ip[0];
	    for (unsigned i=1; i<dimen; i++) {
		*o_scan *= src_width[i];
		*o_scan -= ip[i];
	    }
	    o_scan++;
	}
    }

    IA_PSIter<IA_Point<int> >	iter(dest_ps);
    IA_Point<int>	ip;
    RTYPE *const	dest_data = new RTYPE[dest_ps.card()];
    RTYPE *	valp = dest_data;
    while (iter(ip)) {
	int	offset= ip[0] - src_infimum[0];
	unsigned i;
	for (i=1; i<dimen; i++) {
	    offset *= src_width[i];
	    offset += ip[i] - src_infimum[i];
	}
	const IBTYPE *const base = src_data + offset;

	// _IVAL_=(base[nbh_offsets[i]])
	// _IRESULT_=(*valp)

	*valp = ZERO;
	//INITIALIZE ;

	for (i=0; i<nbh_sz; i++) {
	    ACCUMULATE ;
	}
	//{
	//    RESULT ;
	//}
	valp++;
    }
    delete[] nbh_offsets;

    return IA_Image<IA_Point<int>,RTYPE>(dest_ps, dest_data, dest_ps.card(), 1);
}

static
IA_Image<IA_Point<int>,RTYPE>
forw_CONV_inv(const IA_Image<IA_Point<int>,IBTYPE> &img,
	const IA_Set<IA_Point<int> > &invnbh,
	IA_Set<IA_Point<int> > dest_ps)
{
    IA_Point<int>	inf_ = dest_ps.inf()-invnbh.sup();
    IA_Point<int>	sup_ = dest_ps.sup()-invnbh.inf();
    IA_Set<IA_Point<int> >	src_ps = IA_Set<IA_Point<int> >(inf_,sup_);

    IBTYPE	*const src_data = new IBTYPE[src_ps.card()];

    zero_extend(img, src_ps, (IBTYPE*)src_data, IBTYPE(ZERO));
    IA_Image<IA_Point<int>,RTYPE>	rval =
	forw_CONV_inv_core(inf_, sup_-inf_ + 1, src_data, invnbh, dest_ps);

    delete[] src_data;

    return rval;
}

IA_Image<IA_Point<int>,RTYPE>
CONV(const IA_Neighborhood<IA_Point<int>,IA_Point<int> > &nbh,
     const IA_CoreImage<IA_Point<int>,IBTYPE> &img_,
     IA_Set<IA_Point<int> > dest_ps)
{
    IA_Image<IA_Point<int>,IBTYPE>	img(img_);

    if (nbh.type() == IA_InvariantNbh<IA_Point<int>,IA_Point<int> >::s_type()) {
	return forw_CONV_inv
	    (img,
	     ((IA_InvariantNbh<IA_Point<int>,IA_Point<int> >*)nbh.bnbh)->value,
	     dest_ps);
    } else {
	unsigned	size = dest_ps.card();
	RTYPE *const	dest_data = new RTYPE[size];

	for (int i=0; i<size; i++) {
	    // _IRESULT_=dest_data[i]
	    FORW_INITIALIZE ;
	}

	IA_PSIter<IA_Point<int> >	src_iter(img.domain());
	IA_Point<int>	base_ip;
	while (src_iter(base_ip)) {
	    IA_Set<IA_Point<int> >	ps = nbh(base_ip);

	    IA_PSIter<IA_Point<int> >	nbh_iter(ps);
	    IA_Point<int>	nbh_ip;
	    IBTYPE	ival = img(base_ip);

	    // _IVAL_=ival
	    // _IRESULT_=dest_data[offset]

	    while ( nbh_iter(nbh_ip) ) {
		if (! dest_ps.contains(nbh_ip))
		    continue;
		unsigned	offset = dest_ps.index(nbh_ip);
//		cout << "nbh(" << base_ip << ") =  " << tv << " and nbh()("
//		     << nbh_ip <<") = " << nbh_val << endl;
		FORW_ACCUMULATEVAR ;
	    }
	}
	return IA_Image<IA_Point<int>,RTYPE>(dest_ps, dest_data, size, 1);
    }
}
