package com.ximple.io.dgn7;

import com.vividsolutions.jts.geom.CoordinateList;
import com.vividsolutions.jts.geom.GeometryFactory;
import com.vividsolutions.jts.geom.LineString;
import com.vividsolutions.jts.geom.MultiLineString;
import org.apache.commons.io.FileUtils;
import org.apache.log4j.Logger;
import org.geotools.TestData;
import org.testng.Assert;
import org.testng.annotations.BeforeTest;
import org.testng.annotations.Test;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.RandomAccessFile;
import java.nio.channels.FileChannel;

/**
 * Created by IntelliJ IDEA.
 * User: Ulysses
 * Date: 2010/4/21
 */
public class ElementFactoryTest {
    private final static Logger logger = Logger.getLogger(Dgn7fileReaderTest.class);

    // private final static String testFilePath = "test-data\\testHV.dgn";
    private final static String testFilePath = "testHV.dgn";

    private static GeometryFactory factory = new GeometryFactory();

    private FileInputStream _fs;
    private LineStringElement _testLineString = null;
    private ComplexChainElement _testComplexChain = null;

    @BeforeTest
    public void setUp() throws IOException, Dgn7fileException {
        File dataFile = TestData.file(this, testFilePath);
        if (!dataFile.exists()) {
            return;
        }

        _fs = new FileInputStream(dataFile);

        fetchTestElement(_fs);
        _fs.close();
    }

    private void fetchTestElement(FileInputStream fs) throws Dgn7fileException, IOException {
        FileChannel fc = _fs.getChannel();
        Dgn7fileReader reader = new Dgn7fileReader(fc, new Lock());
        int count = 0;
        Element lastComplex = null;
        while (reader.hasNext()) {
            Element.FileRecord record = reader.nextElement();
            if (record.element() != null) {
                Element element = (Element) record.element();
                ElementType type = element.getElementType();
                boolean completed = false;
                if ((!type.isComplexElement()) && (!element.isComponentElement())) {
                    if (lastComplex != null) {
                        // @todo add process in here
                        if (!processCompleteElement(lastComplex)) break;
                        lastComplex = null;
                    }

                    // @todo add process in here
                    if (!processCompleteElement(element)) break;
                } else if (element.isComponentElement()) {
                    if (lastComplex != null) {
                        ((ComplexElement) lastComplex).add(element);
                    } else {
                        logger.warn("wong." + element.toString());
                        Assert.fail("Component Element cannot found parent.");
                    }
                } else if (type.isComplexElement()) {
                    if (lastComplex != null) {
                        // @todo add process in here
                        if (!processCompleteElement(lastComplex)) break;
                    }
                    lastComplex = element;
                }
            }
            count++;
        }
    }

    private boolean processCompleteElement(Element element) throws IOException {
        if (element instanceof LineStringElement) {
            _testLineString = (LineStringElement) element;
        }

        if (element instanceof ComplexChainElement) {
            _testComplexChain = (ComplexChainElement) element;
        }

        if ((_testLineString != null) && (_testComplexChain != null)) return false;
        return true;
    }

    @Test
    public void testCreateLineString() throws IOException, Dgn7fileException {
        Assert.assertNotNull(_testLineString);
        LineStringElement originElement = _testLineString;

        LineString geometry = (LineString) originElement.toGeometry(factory);
        Element cloneElement = ElementFactory.createLineString(geometry);
        Assert.assertTrue(cloneElement instanceof LineStringElement);
        cloneElement.setLevelIndex(originElement.getLevelIndex());
        cloneElement.setColorIndex(originElement.getColorIndex());
        cloneElement.setWeight(originElement.getWeight());
        cloneElement.setLineStyle(originElement.getLineStyle());
        cloneElement.setRange(geometry.getEnvelopeInternal());

        int len = originElement.raw.length;
        int lenClone = cloneElement.raw.length;
        // Assert.assertEquals(lenClone, len - originElement.getUserAttributeDataOffset() + 1);
        if (originElement.hasUserAttributeData()) {
            Assert.assertEquals(lenClone, originElement.getUserAttributeDataOffset());
        }
        System.out.println("origin size=(" + len + ")-:- Clone size=(" + lenClone + ")");
        int headerSize = ElementFactory.MINIMAL_ELEMLENGTH / 2;
        for (int i = 0; i <= headerSize; i++) {
            if (originElement.raw[i] != cloneElement.raw[i]) {
                System.out.print("different index=" + i + ":");
                System.out.println("origin[" + Integer.toHexString(originElement.raw[i])
                        + "]-clone:[" + Integer.toHexString(cloneElement.raw[i]) + "]");
            }
        }
    }

    @Test
    public void testCreateComplexChain() throws IOException, Dgn7fileException {
        Assert.assertNotNull(_testComplexChain);
        ComplexChainElement originElement = _testComplexChain;

        System.out.print("len=" + originElement.raw.length);
        for (Element elm : originElement) {
            System.out.print(":" + elm.raw.length);
        }
        System.out.println();

        MultiLineString geometries = (MultiLineString) originElement.toGeometry(factory);
        LineString geometry = factory.createLineString(geometries.getCoordinates());

        ComplexChainElement cloneElement = (ComplexChainElement) ElementFactory.createComplexChain(geometry);
        Assert.assertTrue(cloneElement instanceof ComplexChainElement);
        cloneElement.setLevelIndex(originElement.getLevelIndex());
        cloneElement.setColorIndex(originElement.getColorIndex());
        cloneElement.setWeight(originElement.getWeight());
        cloneElement.setLineStyle(originElement.getLineStyle());
        cloneElement.setRange(geometry.getEnvelopeInternal());

        int len = originElement.raw.length;
        int lenClone = cloneElement.raw.length;
        // Assert.assertEquals(lenClone, len - originElement.getUserAttributeDataOffset() + 1);
        System.out.print("clonelen=" + cloneElement.raw.length);
        for (Element elm : cloneElement) {
            System.out.print(":" + elm.raw.length);
        }
        System.out.println();

        if (originElement.hasUserAttributeData()) {
            Assert.assertEquals(lenClone, originElement.getUserAttributeDataOffset() + 4);
        }
        System.out.println("origin size=(" + len + ")-:- Clone size=(" + lenClone + ")");
        int headerSize = ElementFactory.MINIMAL_ELEMLENGTH / 2 + 6;
        for (int i = 0; i < headerSize; i++) {
            if (originElement.raw[i] != cloneElement.raw[i]) {
                System.out.print("different index=" + i + ":");
                System.out.println("origin[" + Integer.toHexString(originElement.raw[i])
                        + "]-clone:[" + Integer.toHexString(cloneElement.raw[i]) + "]");
            }
        }
    }
}