@ -12,60 +12,6 @@ namespace SixLabors.ImageSharp.Formats.Jpeg.Common.Decoder
/// </summary>
/// </summary>
[StructLayout(LayoutKind.Sequential)] [StructLayout(LayoutKind.Sequential)]
internal unsafe struct JpegBlockPostProcessor internal unsafe struct JpegBlockPostProcessor
{
/// <summary>
/// The <see cref="ComputationData"/>
/// </summary>
private ComputationData data ;
/// <summary>
/// Pointers to elements of <see cref="data"/>
/// </summary>
private DataPointers pointers ;
private Size subSamplingDivisors ;
/// <summary>
/// Initialize the <see cref="JpegBlockPostProcessor"/> instance on the stack.
/// </summary>
public static void Init ( JpegBlockPostProcessor * postProcessor , IRawJpegData decoder , IJpegComponent component )
{
postProcessor - > data = ComputationData . Create ( ) ;
postProcessor - > pointers = new DataPointers ( & postProcessor - > data ) ;
int qtIndex = component . QuantizationTableIndex ;
postProcessor - > data . DequantiazationTable = ZigZag . CreateDequantizationTable ( ref decoder . QuantizationTables [ qtIndex ] ) ;
postProcessor - > subSamplingDivisors = component . SubSamplingDivisors ;
}
public void ProcessBlockColorsInto (
ref Block8x8 sourceBlock ,
BufferArea < float > destArea )
{
sourceBlock . CopyToFloatBlock ( ref this . data . SourceBlock ) ;
Block8x8F * b = this . pointers . SourceBlock ;
// Dequantize:
b - > MultiplyInplace ( ref this . data . DequantiazationTable ) ;
FastFloatingPointDCT . TransformIDCT ( ref * b , ref this . data . WorkspaceBlock1 , ref this . data . WorkspaceBlock2 ) ;
this . data . WorkspaceBlock1 . NormalizeColorsInplace ( ) ;
// To conform better to libjpeg we actually NEED TO loose precision here.
// This is because they store blocks as Int16 between all the operations.
// Unfortunately, we need to emulate this to be "more accurate" :(
this . data . WorkspaceBlock1 . RoundInplace ( ) ;
this . data . WorkspaceBlock1 . CopyTo ( destArea , this . subSamplingDivisors . Width , this . subSamplingDivisors . Height ) ;
}
/// <summary>
/// Holds the "large" data blocks needed for computations.
/// </summary>
[StructLayout(LayoutKind.Sequential)]
public struct ComputationData
{ {
/// <summary>
/// <summary>
/// Source block
/// Source block
@ -87,65 +33,44 @@ namespace SixLabors.ImageSharp.Formats.Jpeg.Common.Decoder
/// </summary>
/// </summary>
public Block8x8F DequantiazationTable ; public Block8x8F DequantiazationTable ;
/// <summary>
private Size subSamplingDivisors ;
/// The jpeg unzig data
/// </summary>
public ZigZag Unzig ;
/// <summary>
/// <summary>
/// Creates and initializes a new <see cref="ComputationData"/> instance
/// Initialize the <see cref="JpegBlockPostProcessor"/> instance on the stack.
/// </summary>
/// </summary>
/// <returns>The <see cref="ComputationData"/></returns>
public static void Init ( JpegBlockPostProcessor * postProcessor , IRawJpegData decoder , IJpegComponent component )
public static ComputationData Create ( )
{ {
var data = default ( ComputationData ) ; int qtIndex = component . QuantizationTableIndex ;
data . Unzig = ZigZag . CreateUnzigTable ( ) ; postProcessor - > DequantiazationTable = ZigZag . CreateDequantizationTable ( ref decoder . QuantizationTables [ qtIndex ] ) ;
return data ; postProcessor - > subSamplingDivisors = component . SubSamplingDivisors ;
}
} }
/// <summary>
public void ProcessBlockColorsInto (
/// Contains pointers to the memory regions of <see cref="ComputationData"/> so they can be easily passed around to pointer based utility methods of <see cref="Block8x8F"/>
ref Block8x8 sourceBlock ,
/// </summary>
BufferArea < float > destArea )
public struct DataPointers
{ {
/// <summary>
ref Block8x8F b = ref this . SourceBlock ;
/// Pointer to <see cref="ComputationData.SourceBlock"/>
sourceBlock . CopyToFloatBlock ( ref b ) ;
/// </summary>
public Block8x8F * SourceBlock ;
/// <summary>
// Dequantize:
/// Pointer to <see cref="ComputationData.WorkspaceBlock1"/>
b . MultiplyInplace ( ref this . DequantiazationTable ) ;
/// </summary>
public Block8x8F * WorkspaceBlock1 ;
/// <summary>
FastFloatingPointDCT . TransformIDCT ( ref b , ref this . WorkspaceBlock1 , ref this . WorkspaceBlock2 ) ;
/// Pointer to <see cref="ComputationData.WorkspaceBlock2"/>
/// </summary>
public Block8x8F * WorkspaceBlock2 ;
/// <summary>
// To conform better to libjpeg we actually NEED TO loose precision here.
/// Pointer to <see cref="ComputationData.DequantiazationTable"/>
// This is because they store blocks as Int16 between all the operations.
/// </summary>
// To be "more accurate", we need to emulate this by rounding!
public Block8x8F * DequantiazationTable ; if ( SimdUtils . IsAvx2CompatibleArchitecture )
{
/// <summary>
this . WorkspaceBlock1 . NormalizeColorsAndRoundInplaceAvx2 ( ) ;
/// Pointer to <see cref="ComputationData.Unzig"/> as int*
}
/// </summary>
else
public int * Unzig ;
/// <summary>
/// Initializes a new instance of the <see cref="DataPointers" /> struct.
/// </summary>
/// <param name="dataPtr">Pointer to <see cref="ComputationData"/></param>
internal DataPointers ( ComputationData * dataPtr )
{ {
this . SourceBlock = & dataPtr - > SourceBlock ; this . WorkspaceBlock1 . NormalizeColorsInplace ( ) ;
this . WorkspaceBlock1 = & dataPtr - > WorkspaceBlock1 ; this . WorkspaceBlock1 . RoundInplace ( ) ;
this . WorkspaceBlock2 = & dataPtr - > WorkspaceBlock2 ;
this . DequantiazationTable = & dataPtr - > DequantiazationTable ;
this . Unzig = dataPtr - > Unzig . Data ;
} }
this . WorkspaceBlock1 . CopyTo ( destArea , this . subSamplingDivisors . Width , this . subSamplingDivisors . Height ) ;
} }
} }
} }
Anton Firszov
9 years ago