diff --git a/main.cc b/main.cc index 34fe7ef..4be8173 100644 --- a/main.cc +++ b/main.cc @@ -304,6 +304,129 @@ bool CGCComponentBar_( return false; } +bool HueSaturationPad( + char const* const name , + float* const hue , + float* const saturation , + const float size = 180.f ) noexcept +{ + using namespace ImGui; + + // Check/set bounding box + const auto wSize{ ImMin( size , CalcItemWidth( ) ) }; + auto* const win( GetCurrentWindow( ) ); + const ImVec2 cPos( win->DC.CursorPos ); + const ImRect bb{ cPos , cPos + ImVec2( wSize , wSize ) }; + auto& style( GetStyle( ) ); + auto id{ win->GetID( name ) }; + ItemSize( bb , style.FramePadding.y ); + if ( !ItemAdd( bb , id ) ) { + return false; + } + + const ImVec2 wCenter{ cPos + ImVec2( wSize * .5f , wSize * .5f ) }; + const float wThickness = wSize * 0.08f; + const float wOuterRadius = wSize * 0.50f; + const float wInnerRadius = wOuterRadius - wThickness; + + auto* const ctx( GetCurrentContext( ) ); + const auto mPos{ ctx->IO.MousePos }; + const auto rmPos{ mPos - wCenter }; + const auto mcSqDist{ rmPos.x * rmPos.x + rmPos.y * rmPos.y }; + const bool hovered{ mcSqDist <= wSize * wSize * .25f }; + const bool tabFocus{ FocusableItemRegister( win , id ) }; + if ( tabFocus || ( hovered && ctx->IO.MouseClicked[ 0 ] ) ) { + SetActiveID( id , win ); + FocusWindow( win ); + } + + const bool active{ ctx->ActiveId == id }; + float nHue{ *hue } , nSat{ *saturation }; + if ( active ) { + if ( ctx->IO.MouseDown[ 0 ] ) { + nHue = ( 1.f + atan2f( -rmPos.y , -rmPos.x ) / IM_PI ) * .5f; + nSat = ImSaturate( sqrtf( mcSqDist ) / wInnerRadius ); + } else { + ClearActiveID( ); + } + } else if ( hovered && ctx->IO.KeyCtrl && ctx->IO.MouseWheel != 0.f ) { + const auto change{ ctx->IO.MouseWheel * .005f }; + if ( ctx->IO.KeyShift ) { + nSat = ImSaturate( nSat + change ); + } else { + nHue = fmodf( nHue + change + 1.f , 1.f ); + } + } + + //- DRAW --------------------------------------------------------------- + + auto* const dl( GetWindowDrawList( ) ); + const float aeps = 1.5f / wOuterRadius; // Half a pixel arc length in radians (2pi cancels out). + const int segmentPerArc = ImMax( 4 , (int) wOuterRadius / 12 ); + const ImU32 hue_colors[] = { + IM_COL32( 255 , 0 , 0 , 255 ) , + IM_COL32( 255 , 255 , 0 , 255 ) , + IM_COL32( 0 , 255 , 0 , 255 ) , + IM_COL32( 0 , 255 , 255 , 255 ) , + IM_COL32( 0 , 0 , 255 , 255 ) , + IM_COL32( 255 , 0 , 255 , 255 ) , + IM_COL32( 255 , 0 , 0 , 255 ) + }; + const auto avgRadius( ( wInnerRadius + wOuterRadius ) * .5f ); + + // Pad background and cross + const auto bgColor{ GetColorU32( ( hovered || active ) + ? ImVec4( .4f , .4f , .4f , 1 ) + : ImVec4( .25f , .25f , .25f , 1 ) ) }; + dl->AddCircleFilled( wCenter , avgRadius , bgColor ); + dl->AddLine( wCenter - ImVec2( 0 , avgRadius ) , + wCenter + ImVec2( 0 , avgRadius ) , + GetColorU32( ImVec4( .1 , .1 , .1 , 1 ) ) ); + dl->AddLine( wCenter - ImVec2( avgRadius , 0 ) , + wCenter + ImVec2( avgRadius , 0 ) , + GetColorU32( ImVec4( .1 , .1 , .1 , 1 ) ) ); + // Color wheel + for ( int n = 0 ; n < 6 ; n ++ ) { + const float a0 = ( n ) / 6.f * 2.f * IM_PI - aeps; + const float a1 = ( n + 1.f ) / 6.f * 2.f * IM_PI + aeps; + const int vStart = dl->_VtxCurrentIdx; + dl->PathArcTo( wCenter , avgRadius , a0 , a1 , segmentPerArc ); + dl->PathStroke( IM_COL32_WHITE , false, wThickness); + + // Paint colors over existing vertices + const ImVec2 gradientP0( + wCenter.x + cosf( a0 ) * wInnerRadius , + wCenter.y + sinf( a0 ) * wInnerRadius ); + const ImVec2 gradientP1( + wCenter.x + cosf( a1 ) * wInnerRadius , + wCenter.y + sinf( a1 ) * wInnerRadius ); + ShadeVertsLinearColorGradientKeepAlpha( + dl->_VtxWritePtr - ( dl->_VtxCurrentIdx - vStart ), + dl->_VtxWritePtr , + gradientP0 , gradientP1 , + hue_colors[ n ] , hue_colors[ n + 1 ] ); + } + // Cursor from hue and saturation + const float cAngle{ nHue * 2 * IM_PI }; + const float cDist{ nSat * wInnerRadius }; + ImVec4 cColor{ 0 , 0 , 0 , 1 }; + ColorConvertHSVtoRGB( *hue , *saturation , 1 , cColor.x , cColor.y , cColor.z ); + dl->AddCircleFilled( wCenter + ImVec2( cos( cAngle ) * cDist , sin( cAngle ) * cDist ) , + ( wOuterRadius - wInnerRadius ) * .5f , + GetColorU32( cColor ) ); + dl->AddCircle( wCenter + ImVec2( cos( cAngle ) * cDist , sin( cAngle ) * cDist ) , + ( wOuterRadius - wInnerRadius ) * .5f , + GetColorU32( ImVec4( 1 , 1 , 1 , 1 ) ) ); + + // Update values + if ( *hue != nHue || *saturation != nSat ) { + *hue = nHue; + *saturation = nSat; + return true; + } + return false; +} + bool ColorGradingControls( char const* const name , float* const red , @@ -320,7 +443,7 @@ bool ColorGradingControls( ImGuiWindow* const window{ GetCurrentWindow() }; ImGuiStorage* const storage{ window->DC.StateStorage }; - const bool wheelMode{ storage->GetBool( window->GetID( name ) ) }; + const bool wheelMode{ storage->GetBool( window->GetID( name ) , true ) }; // Mode selection bool modeChanged{ false }; @@ -334,6 +457,16 @@ bool ColorGradingControls( bool changed; if ( wheelMode ^ modeChanged ) { changed = false; + const float scRed { ImSaturate( ( *red - base ) / ( unit * 2 ) ) } , + scGreen{ ImSaturate( ( *green - base ) / ( unit * 2 ) ) } , + scBlue { ImSaturate( ( *blue - base ) / ( unit * 2 ) ) }; + float H , S , V; + ColorConvertRGBtoHSV( scRed , scGreen , scBlue , H , S , V ); + + // FIXME test + static float fh = 0.f , fs = 0.f; + HueSaturationPad( "##wheel" , &fh , &fs , 180.f ); + #warning implement the fuck } else { PushItemWidth( -1 );