Esempio n. 1
0
 public Segment(JBIG2StreamDecoder streamDecoder)
 {
     this.decoder      = streamDecoder;
     huffmanDecoder    = decoder.getHuffmanDecoder();
     arithmeticDecoder = decoder.getArithmeticDecoder();
     mmrDecoder        = decoder.getMMRDecoder();
 }
        public override void readSegment()
        {
            if (JBIG2StreamDecoder.debug)
            {
                Console.WriteLine("==== Reading Text Region ====");
            }

            base.readSegment();

            /** read text region Segment flags */
            readTextRegionFlags();

            short[] buff = new short[4];
            decoder.readbyte(buff);
            long noOfSymbolInstances = BinaryOperation.getInt32(buff);

            if (JBIG2StreamDecoder.debug)
            {
                Console.WriteLine("noOfSymbolInstances = " + noOfSymbolInstances);
            }

            int noOfReferredToSegments = segmentHeader.getReferredToSegmentCount();

            int[] referredToSegments = segmentHeader.getReferredToSegments();

            //List codeTables = new ArrayList();
            List <Segment> segmentsReferenced = new List <Segment>();
            long           noOfSymbols        = 0;

            if (JBIG2StreamDecoder.debug)
            {
                Console.WriteLine("noOfReferredToSegments = " + noOfReferredToSegments);
            }
            int i = 0;

            for (i = 0; i < noOfReferredToSegments; i++)
            {
                Segment seg  = decoder.findSegment(referredToSegments[i]);
                int     type = seg.getSegmentHeader().getSegmentType();

                if (type == Segment.SYMBOL_DICTIONARY)
                {
                    segmentsReferenced.Add(seg);
                    noOfSymbols += ((SymbolDictionarySegment)seg).getNoOfExportedSymbols();
                }
                else if (type == Segment.TABLES)
                {
                    //codeTables.add(seg);
                }
            }

            long symbolCodeLength = 0;
            int  count            = 1;

            while (count < noOfSymbols)
            {
                symbolCodeLength++;
                count <<= 1;
            }

            int currentSymbol = 0;

            JBIG2Bitmap[] symbols = new JBIG2Bitmap[noOfSymbols];
            foreach (Segment it in segmentsReferenced)
            {
                if (it.getSegmentHeader().getSegmentType() == Segment.SYMBOL_DICTIONARY)
                {
                    JBIG2Bitmap[] bitmaps = ((SymbolDictionarySegment)it).getBitmaps();
                    for (int j = 0; j < bitmaps.Length; j++)
                    {
                        symbols[currentSymbol] = bitmaps[j];
                        currentSymbol++;
                    }
                }
            }

            long[,] huffmanFSTable = null;
            long[,] huffmanDSTable = null;
            long[,] huffmanDTTable = null;

            long[,] huffmanRDWTable = null;
            long[,] huffmanRDHTable = null;

            long[,] huffmanRDXTable   = null;
            long[,] huffmanRDYTable   = null;
            long[,] huffmanRSizeTable = null;

            bool sbHuffman = textRegionFlags.getFlagValue(TextRegionFlags.SB_HUFF) != 0;

            i = 0;
            if (sbHuffman)
            {
                int sbHuffFS = textRegionHuffmanFlags.getFlagValue(TextRegionHuffmanFlags.SB_HUFF_FS);
                if (sbHuffFS == 0)
                {
                    huffmanFSTable = HuffmanDecoder.huffmanTableF;
                }
                else if (sbHuffFS == 1)
                {
                    huffmanFSTable = HuffmanDecoder.huffmanTableG;
                }
                else
                {
                }

                int sbHuffDS = textRegionHuffmanFlags.getFlagValue(TextRegionHuffmanFlags.SB_HUFF_DS);
                if (sbHuffDS == 0)
                {
                    huffmanDSTable = HuffmanDecoder.huffmanTableH;
                }
                else if (sbHuffDS == 1)
                {
                    huffmanDSTable = HuffmanDecoder.huffmanTableI;
                }
                else if (sbHuffDS == 2)
                {
                    huffmanDSTable = HuffmanDecoder.huffmanTableJ;
                }
                else
                {
                }

                int sbHuffDT = textRegionHuffmanFlags.getFlagValue(TextRegionHuffmanFlags.SB_HUFF_DT);
                if (sbHuffDT == 0)
                {
                    huffmanDTTable = HuffmanDecoder.huffmanTableK;
                }
                else if (sbHuffDT == 1)
                {
                    huffmanDTTable = HuffmanDecoder.huffmanTableL;
                }
                else if (sbHuffDT == 2)
                {
                    huffmanDTTable = HuffmanDecoder.huffmanTableM;
                }
                else
                {
                }

                int sbHuffRDW = textRegionHuffmanFlags.getFlagValue(TextRegionHuffmanFlags.SB_HUFF_RDW);
                if (sbHuffRDW == 0)
                {
                    huffmanRDWTable = HuffmanDecoder.huffmanTableN;
                }
                else if (sbHuffRDW == 1)
                {
                    huffmanRDWTable = HuffmanDecoder.huffmanTableO;
                }
                else
                {
                }

                int sbHuffRDH = textRegionHuffmanFlags.getFlagValue(TextRegionHuffmanFlags.SB_HUFF_RDH);
                if (sbHuffRDH == 0)
                {
                    huffmanRDHTable = HuffmanDecoder.huffmanTableN;
                }
                else if (sbHuffRDH == 1)
                {
                    huffmanRDHTable = HuffmanDecoder.huffmanTableO;
                }
                else
                {
                }

                int sbHuffRDX = textRegionHuffmanFlags.getFlagValue(TextRegionHuffmanFlags.SB_HUFF_RDX);
                if (sbHuffRDX == 0)
                {
                    huffmanRDXTable = HuffmanDecoder.huffmanTableN;
                }
                else if (sbHuffRDX == 1)
                {
                    huffmanRDXTable = HuffmanDecoder.huffmanTableO;
                }
                else
                {
                }

                int sbHuffRDY = textRegionHuffmanFlags.getFlagValue(TextRegionHuffmanFlags.SB_HUFF_RDY);
                if (sbHuffRDY == 0)
                {
                    huffmanRDYTable = HuffmanDecoder.huffmanTableN;
                }
                else if (sbHuffRDY == 1)
                {
                    huffmanRDYTable = HuffmanDecoder.huffmanTableO;
                }
                else
                {
                }

                int sbHuffRSize = textRegionHuffmanFlags.getFlagValue(TextRegionHuffmanFlags.SB_HUFF_RSIZE);
                if (sbHuffRSize == 0)
                {
                    huffmanRSizeTable = HuffmanDecoder.huffmanTableA;
                }
                else
                {
                }
            }

            long[][] runLengthTable  = new long[36][];
            long[][] symbolCodeTable = new long[noOfSymbols + 1][];
            if (sbHuffman)
            {
                decoder.consumeRemainingBits();

                for (i = 0; i < 32; i++)
                {
                    runLengthTable[i] = new long[] { i, decoder.readBits(4), 0, 0 };
                }

                runLengthTable[32] = new long[] { 0x103, decoder.readBits(4), 2, 0 };

                runLengthTable[33] = new long[] { 0x203, decoder.readBits(4), 3, 0 };

                runLengthTable[34] = new long[] { 0x20b, decoder.readBits(4), 7, 0 };

                runLengthTable[35] = new long[] { 0, 0, HuffmanDecoder.jbig2HuffmanEOT };

                runLengthTable = HuffmanDecoder.buildTable(runLengthTable, 35);

                for (i = 0; i < noOfSymbols; i++)
                {
                    symbolCodeTable[i] = new long[] { i, 0, 0, 0 };
                }

                i = 0;
                while (i < noOfSymbols)
                {
                    long j = huffmanDecoder.decodeInt(runLengthTable).intResult();
                    if (j > 0x200)
                    {
                        for (j -= 0x200; j != 0 && i < noOfSymbols; j--)
                        {
                            symbolCodeTable[i++][1] = 0;
                        }
                    }
                    else if (j > 0x100)
                    {
                        for (j -= 0x100; j != 0 && i < noOfSymbols; j--)
                        {
                            symbolCodeTable[i][1] = symbolCodeTable[i - 1][1];
                            i++;
                        }
                    }
                    else
                    {
                        symbolCodeTable[i++][1] = j;
                    }
                }

                symbolCodeTable[noOfSymbols][1] = 0;
                symbolCodeTable[noOfSymbols][2] = HuffmanDecoder.jbig2HuffmanEOT;
                symbolCodeTable = HuffmanDecoder.buildTable(symbolCodeTable, (int)noOfSymbols);

                decoder.consumeRemainingBits();
            }
            else
            {
                symbolCodeTable = null;
                arithmeticDecoder.resetIntStats((int)symbolCodeLength);
                arithmeticDecoder.start();
            }

            bool symbolRefine        = textRegionFlags.getFlagValue(TextRegionFlags.SB_REFINE) != 0;
            long logStrips           = textRegionFlags.getFlagValue(TextRegionFlags.LOG_SB_STRIPES);
            int  defaultPixel        = textRegionFlags.getFlagValue(TextRegionFlags.SB_DEF_PIXEL);
            int  combinationOperator = textRegionFlags.getFlagValue(TextRegionFlags.SB_COMB_OP);
            bool transposed          = textRegionFlags.getFlagValue(TextRegionFlags.TRANSPOSED) != 0;
            int  referenceCorner     = textRegionFlags.getFlagValue(TextRegionFlags.REF_CORNER);
            int  sOffset             = textRegionFlags.getFlagValue(TextRegionFlags.SB_DS_OFFSET);
            int  template            = textRegionFlags.getFlagValue(TextRegionFlags.SB_R_TEMPLATE);

            if (symbolRefine)
            {
                arithmeticDecoder.resetRefinementStats(template, null);
            }

            JBIG2Bitmap bitmap = new JBIG2Bitmap(regionBitmapWidth, regionBitmapHeight, arithmeticDecoder, huffmanDecoder, mmrDecoder);

            bitmap.readTextRegion2(sbHuffman, symbolRefine, noOfSymbolInstances, logStrips, noOfSymbols, symbolCodeTable, symbolCodeLength, symbols, defaultPixel, combinationOperator, transposed, referenceCorner, sOffset, huffmanFSTable, huffmanDSTable, huffmanDTTable, huffmanRDWTable, huffmanRDHTable, huffmanRDXTable, huffmanRDYTable, huffmanRSizeTable, template, symbolRegionAdaptiveTemplateX, symbolRegionAdaptiveTemplateY, decoder);

            if (inlineImage)
            {
                PageInformationSegment pageSegment = decoder.findPageSegement(segmentHeader.getPageAssociation());
                JBIG2Bitmap            pageBitmap  = pageSegment.getPageBitmap();

                if (JBIG2StreamDecoder.debug)
                {
                    Console.WriteLine(pageBitmap + " " + bitmap);
                }

                int externalCombinationOperator = regionFlags.getFlagValue(RegionFlags.EXTERNAL_COMBINATION_OPERATOR);
                pageBitmap.combine(bitmap, regionBitmapXLocation, regionBitmapYLocation, externalCombinationOperator);
            }
            else
            {
                bitmap.setBitmapNumber(getSegmentHeader().getSegmentNumber());
                decoder.appendBitmap(bitmap);
            }

            decoder.consumeRemainingBits();
        }
        public byte[] decodeJBIG2(byte[] data, ImageFormat format = ImageFormat.TIFF, int NewWidth = 0, int NewHeight = 0)
        {
            reader = new Big2StreamReader(data);
            resetDecoder();
            bool validFile = checkHeader();

            if (JBIG2StreamDecoder.debug)
            {
                Console.WriteLine("validFile = " + validFile);
            }
            if (!validFile)
            {
                /**
                 * Assume this is a stream from a PDF so there is no file header,
                 * end of page segments, or end of file segments. Organisation must
                 * be sequential, and the number of pages is assumed to be 1.
                 */
                noOfPagesKnown           = true;
                randomAccessOrganisation = false;
                noOfPages = 1;
                /** check to see if there is any global data to be read */
                if (globalData != null)
                {
                    /** set the reader to read from the global data */
                    reader = new Big2StreamReader(globalData);

                    huffmanDecoder    = new HuffmanDecoder(reader);
                    mmrDecoder        = new MMRDecoder(reader);
                    arithmeticDecoder = new ArithmeticDecoder(reader);

                    /** read in the global data segments */
                    readSegments();

                    /** set the reader back to the main data */
                    reader = new Big2StreamReader(data);
                }
                else
                {
                    /**
                     * There's no global data, so move the file pointer back to the
                     * start of the stream
                     */
                    reader.movePointer(-8);
                }
            }
            else
            {
                /**
                 * We have the file header, so assume it is a valid stand-alone
                 * file.
                 */

                if (JBIG2StreamDecoder.debug)
                {
                    Console.WriteLine("==== File Header ====");
                }

                setFileHeaderFlags();

                if (JBIG2StreamDecoder.debug)
                {
                    Console.WriteLine("randomAccessOrganisation = " + randomAccessOrganisation);
                    Console.WriteLine("noOfPagesKnown = " + noOfPagesKnown);
                }

                if (noOfPagesKnown)
                {
                    noOfPages = getNoOfPages();

                    if (JBIG2StreamDecoder.debug)
                    {
                        Console.WriteLine("noOfPages = " + noOfPages);
                    }
                }
            }

            huffmanDecoder    = new HuffmanDecoder(reader);
            mmrDecoder        = new MMRDecoder(reader);
            arithmeticDecoder = new ArithmeticDecoder(reader);

            /** read in the main segment data */
            readSegments();

            //Create Image
            var rawimage   = findPageSegement(1).getPageBitmap();
            int width      = (int)rawimage.getWidth();
            int height     = (int)rawimage.getHeight();
            var dataStream = rawimage.getData(true);

            var newarray = new byte[dataStream.Length];

            Array.Copy(dataStream, newarray, dataStream.Length);
            int stride = (width * 1 + 7) / 8;

            var bitmap = new WriteableBitmap(width, height, 96, 96, System.Windows.Media.PixelFormats.BlackWhite, null);

            bitmap.WritePixels(new System.Windows.Int32Rect(0, 0, width, height), newarray, stride, 0);

            MemoryStream stream3 = new MemoryStream();

            if (format == ImageFormat.TIFF)
            {
                var encoder = new TiffBitmapEncoder();
                encoder.Frames.Add(BitmapFrame.Create(bitmap));
                encoder.Save(stream3);
            }
            else if (format == ImageFormat.JPEG)
            {
                var encoder = new JpegBitmapEncoder();
                encoder.Frames.Add(BitmapFrame.Create(bitmap));
                encoder.Save(stream3);
            }

            if (NewWidth != 0 && NewHeight != 0)
            {
                var newbitmap = ResizeHelpers.ScaleImage(stream3.ToArray(), NewWidth, NewHeight);
                return(newbitmap);
            }

            return(stream3.ToArray());
        }