Home | History | Annotate | Download | only in chromium 
      1 /*
      2  * Copyright (C) 2010 Google Inc. All rights reserved.
      3  *
      4  * Redistribution and use in source and binary forms, with or without
      5  * modification, are permitted provided that the following conditions are
      6  * met:
      7  *
      8  *     * Redistributions of source code must retain the above copyright
      9  * notice, this list of conditions and the following disclaimer.
     10  *     * Redistributions in binary form must reproduce the above
     11  * copyright notice, this list of conditions and the following disclaimer
     12  * in the documentation and/or other materials provided with the
     13  * distribution.
     14  *     * Neither the name of Google Inc. nor the names of its
     15  * contributors may be used to endorse or promote products derived from
     16  * this software without specific prior written permission.
     17  *
     18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
     19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
     20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
     21  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
     22  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
     23  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
     24  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
     25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
     26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
     28  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     29  */
     30 
     31 #import <AppKit/AppKit.h>
     32 #include <signal.h>
     33 #include <stdio.h>
     34 #include <stdlib.h>
     35 
     36 // This is a simple helper app that changes the color sync profile to the
     37 // generic profile and back when done.  This program is managed by the layout
     38 // test script, so it can do the job for multiple DumpRenderTree while they are
     39 // running layout tests.
     40 
     41 static CMProfileRef userColorProfile = 0;
     42 
     43 static void saveCurrentColorProfile()
     44 {
     45     CGDirectDisplayID displayID = CGMainDisplayID();
     46     CMProfileRef previousProfile;
     47     CMError error = CMGetProfileByAVID((UInt32)displayID, &previousProfile);
     48     if (error) {
     49         NSLog(@"failed to get the current color profile, pixmaps won't match. "
     50               @"Error: %d", (int)error);
     51     } else {
     52         userColorProfile = previousProfile;
     53     }
     54 }
     55 
     56 static void installLayoutTestColorProfile()
     57 {
     58     // To make sure we get consistent colors (not dependent on the Main display),
     59     // we force the generic rgb color profile.  This cases a change the user can
     60     // see.
     61 
     62     CGDirectDisplayID displayID = CGMainDisplayID();
     63     NSColorSpace* genericSpace = [NSColorSpace genericRGBColorSpace];
     64     CMProfileRef genericProfile = (CMProfileRef)[genericSpace colorSyncProfile];
     65     CMError error = CMSetProfileByAVID((UInt32)displayID, genericProfile);
     66     if (error) {
     67         NSLog(@"failed install the generic color profile, pixmaps won't match. "
     68               @"Error: %d", (int)error);
     69     }
     70 }
     71 
     72 static void restoreUserColorProfile(void)
     73 {
     74     if (!userColorProfile)
     75         return;
     76     CGDirectDisplayID displayID = CGMainDisplayID();
     77     CMError error = CMSetProfileByAVID((UInt32)displayID, userColorProfile);
     78     CMCloseProfile(userColorProfile);
     79     if (error) {
     80         NSLog(@"Failed to restore color profile, use System Preferences -> "
     81               @"Displays -> Color to reset. Error: %d", (int)error);
     82     }
     83     userColorProfile = 0;
     84 }
     85 
     86 static void simpleSignalHandler(int sig)
     87 {
     88     // Try to restore the color profile and try to go down cleanly
     89     restoreUserColorProfile();
     90     exit(128 + sig);
     91 }
     92 
     93 int main(int argc, char* argv[])
     94 {
     95     NSAutoreleasePool* pool = [[NSAutoreleasePool alloc] init];
     96 
     97     // Hooks the ways we might get told to clean up...
     98     signal(SIGINT, simpleSignalHandler);
     99     signal(SIGHUP, simpleSignalHandler);
    100     signal(SIGTERM, simpleSignalHandler);
    101 
    102     // Save off the current profile, and then install the layout test profile.
    103     saveCurrentColorProfile();
    104     installLayoutTestColorProfile();
    105 
    106     // Let the script know we're ready
    107     printf("ready\n");
    108     fflush(stdout);
    109 
    110     // Wait for any key (or signal)
    111     getchar();
    112 
    113     // Restore the profile
    114     restoreUserColorProfile();
    115 
    116     [pool release];
    117     return 0;
    118 }
    119