use teia::complex;
use teia::complex::{BoundaryFacesPositions, Complex};
use teia::indexed_vec::IndexedVec;
use teia::simplex;
use teia::simplex::Simplex;
use teia::traits::*;
use teia::z2vector::Z2VectorVec;

fn main() {
    println!("# All-in-one complex example");
    let mut comp: Complex<IndexedVec<_>, _> = Complex::new();

    comp.push(simplex![0]).unwrap();
    comp.push(simplex![1]).unwrap();
    comp.push(simplex![2]).unwrap();
    comp.push(simplex![0, 1]).unwrap();
    comp.push(simplex![0, 2]).unwrap();
    comp.push(simplex![1, 2]).unwrap();
    comp.push(simplex![0, 1, 2]).unwrap();
    comp.push(simplex![0, 1, 2, 3]).unwrap();

    println!("");

    println!("## Debug print");
    println!("{:?}", comp);

    println!("");

    println!("## Iterator");
    for t in comp.basis.iter() {
        println!("{:?}", t);
    }

    println!("");

    println!("## Range(3..6)");
    for t in comp.basis.range(3..6) {
        println!("{:?}", t);
    }

    println!("");

    println!("## `complex::compute_boundary`");
    for index in comp.basis.index_range() {
        print!("    index = {}, range = {:?}", index, 0..index);
        let res: Option<Z2VectorVec> =
            complex::compute_boundary(comp.basis.range(0..index), &comp.basis[index]);
        println!("  -> {:?}", res);
    }

    println!("");

    println!("## The iterator `BoundaryFacesPositions` ");
    for index in comp.basis.index_range() {
        print!("    index = {}, range = {:?}", index, 0..index);
        let iter = BoundaryFacesPositions::new(comp.basis.range(0..index), &comp.basis[index]);
        let res: Option<Z2VectorVec> = iter.collect();
        println!("  -> {:?}", res);
    }

    println!("");

    println!("## `BoundaryFacesPositions` for range(3..6) and simplex![0,1,2]");
    {
        let simp = simplex![0, 1, 2];
        let range = comp.basis.range(3..6);
        let iter = BoundaryFacesPositions::new(range, &simp);
        for item in iter {
            println!("    {:?}", item);
        }
    }

    println!("");

    println!("## Complex::boundaries()");
    for chain in comp.boundaries::<Z2VectorVec>() {
        println!("{:?}", chain);
    }

    println!("");

    println!("# Separated complex example");

    let mut comp0 = Complex::<IndexedVec<_>, Simplex>::new();
    comp0.push(simplex![0]).unwrap();
    comp0.push(simplex![1]).unwrap();
    comp0.push(simplex![2]).unwrap();
    comp0.push(simplex![3]).unwrap();

    let mut comp1 = Complex::<IndexedVec<_>, Simplex>::with_prev(&comp0);
    comp1.push(simplex![0, 1]).unwrap();
    comp1.push(simplex![0, 2]).unwrap();
    comp1.push(simplex![0, 3]).unwrap();
    comp1.push(simplex![1, 2]).unwrap();
    comp1.push(simplex![1, 3]).unwrap();
    comp1.push(simplex![2, 3]).unwrap();

    let mut comp2 = Complex::<IndexedVec<_>, Simplex>::with_prev(&comp1);
    comp2.push(simplex![0, 1, 2]).unwrap();
    comp2.push(simplex![0, 1, 3]).unwrap();
    comp2.push(simplex![0, 2, 3]).unwrap();
    comp2.push(simplex![1, 2, 3]).unwrap();

    let mut comp3 = Complex::<IndexedVec<_>, Simplex>::with_prev(&comp2);
    comp3.push(simplex![0, 1, 2, 3]).unwrap();

    println!("");
    println!("## Complex 0");
    for simp in comp0.basis.iter() {
        println!("{:?}", simp);
    }

    println!("");
    println!("## Complex 1");
    for simp in comp1.basis.iter() {
        println!("{:?}", simp);
    }

    println!("");
    println!("## Complex 2");
    for simp in comp2.basis.iter() {
        println!("{:?}", simp);
    }

    println!("");
    println!("## Complex 3");
    for simp in comp3.basis.iter() {
        println!("{:?}", simp);
    }

    println!("");
    println!("## Complex 1's boundaries from complex 0");
    for chain in comp1.boundaries_from::<Z2VectorVec, _>(&comp0) {
        println!("{:?}", chain);
    }

    println!("");
    println!("## Complex 2's boundaries from complex 1");
    for chain in comp2.boundaries_from::<Z2VectorVec, _>(&comp1) {
        println!("{:?}", chain);
    }

    println!("");
    println!("## Complex 3's boundaries from complex 2");
    for chain in comp3.boundaries_from::<Z2VectorVec, _>(&comp2) {
        println!("{:?}", chain);
    }
}