Exemplo n.º 1
0
    def run(self, dataRef):
        """Make input catalogs."""
        skyMap = dataRef.get("deepCoadd_skyMap", immediate=True)
        tractId = dataRef.dataId["tract"]
        tract = skyMap[tractId]
        extPatch = tract.getNumPatches()
        nPatch = extPatch.getX() * extPatch.getY()
        nFakes = self.config.rhoFakes * nPatch
        ra_vert, dec_vert = zip(*tract.getVertexList())
        ra_vert = sorted(ra_vert)
        dec_vert = sorted(dec_vert)
        ra0 = ra_vert[0].asDegrees()
        ra1 = ra_vert[-1].asDegrees()
        dec0 = dec_vert[0].asDegrees()
        dec1 = dec_vert[-1].asDegrees()

        """
        Added by Song Huang 2016-09-02
        Tracts in the Deep and Wide layers are defined to have overlaps
        between the two adjacent Tracts: 1 arcmin ~ 0.01667 deg
        """
        if self.config.innerTract:
            # ra0 += 0.0167
            # ra1 -= 0.0167
            # dec0 += 0.0167
            # dec1 -= 0.0167
            ra0 += 0.09
            ra1 -= 0.09
            dec0 += 0.09
            dec1 -= 0.09

        radUse = self.config.rad
        raArr, decArr = np.array(zip(*makeRaDecCat.getRandomRaDec(nFakes, ra0, ra1, dec0, dec1, rad=radUse)))
        """
        Added by Song Huang 2016-09-01
        Filter the random RA, DEC using two filters
        """
        if (self.config.acpMask != "") or (self.config.rejMask != ""):
            try:
                from shapely import wkb
                from shapely.geometry import Point
                from shapely.prepared import prep

                # Filter through the accept mask
                acpUse = self.config.acpMask
                if (acpUse != "") and os.path.isfile(acpUse):
                    print "## Filter through : %s" % acpUse
                    acpWkb = open(acpUse, "r")
                    acpRegs = wkb.loads(acpWkb.read().decode("hex"))
                    acpPrep = prep(acpRegs)
                    acpWkb.close()
                    inside = np.asarray(map(lambda x, y: acpPrep.contains(Point(x, y)), raArr, decArr))
                else:
                    inside = np.isfinite(raArr)

                # Filter through the reject mask
                rejUse = self.config.rejMask
                if (rejUse != "") and os.path.isfile(rejUse):
                    print "## Filter through : %s" % rejUse
                    rejWkb = open(rejUse, "r")
                    rejRegs = wkb.loads(rejWkb.read().decode("hex"))
                    rejPrep = prep(rejRegs)
                    rejWkb.close()
                    masked = np.asarray(map(lambda x, y: rejPrep.contains(Point(x, y)), raArr, decArr))
                else:
                    masked = np.isnan(raArr)

                useful = np.asarray(map(lambda x, y: x and (not y), inside, masked))
                ra, dec = raArr[useful], decArr[useful]

                print "## %d out of %d objects left" % (len(ra), len(raArr))

            except ImportError:
                warnings.warn("Can not import Shapely, no filter performed!")
                ra, dec = raArr, decArr
        else:
            ra, dec = raArr, decArr

        # Number of fakes that will be added
        nFakes = len(ra)
        # Create an empty astropy.Table object
        outTab = astropy.table.Table()
        # Add columns for Ra, Dec
        outTab.add_column(astropy.table.Column(name="RA", data=ra))
        outTab.add_column(astropy.table.Column(name="Dec", data=dec))
        if self.config.inputCat is not None:
            galData = astropy.table.Table().read(self.config.inputCat)
            randInd = np.random.choice(range(len(galData)), size=nFakes)
            mergedData = galData[randInd]

            for colname in mergedData.columns:
                colAdd = astropy.table.Column(name=colname, data=mergedData[colname])
                outTab.add_column(colAdd)

            # Replace ID with a unique integer (using index)
            if ("ID" in outTab.colnames) and (self.config.uniqueID):
                outTab.rename_column("ID", "modelID")
                outTab.add_column(astropy.table.Column(name="ID", data=np.arange(len(outTab))))
            elif "ID" not in outTab.colnames:
                outTab.add_column(astropy.table.Column(name="ID", data=np.arange(len(outTab))))

            # Generate multiBand catalog at the same time
            magList = [col for col in galData.colnames if "mag_" in col]
            if len(magList) >= 1:
                print "Find magnitude in %d band(s)" % len(magList)
                for mag in magList:
                    try:
                        outTab.remove_column("mag")
                    except KeyError:
                        pass
                    outTab.add_column(astropy.table.Column(name="mag", data=mergedData[mag]))
                    filt = mag.split("_")[1].upper()
                    outFits = os.path.join(self.config.outDir, "src_%d_radec_%s.fits" % (tractId, filt))
                    outTab.write(outFits, overwrite=True)
            else:
                outTab.write(os.path.join(self.config.outDir, "src_%d_radec.fits" % tractId), overwrite=True)
        else:
            outTab.write(os.path.join(self.config.outDir, "src_%d_radec_only.fits" % tractId), overwrite=True)
Exemplo n.º 2
0
    def run(self, dataRef):
        """Make input catalogs."""
        skyMap = dataRef.get('deepCoadd_skyMap', immediate=True)
        tractId = dataRef.dataId['tract']
        tract = skyMap[tractId]
        extPatch = tract.getNumPatches()
        nPatch = extPatch.getX() * extPatch.getY()
        nFakes = self.config.rhoFakes * nPatch
        ra_vert, dec_vert = zip(*tract.getVertexList())
        ra_vert = sorted(ra_vert)
        dec_vert = sorted(dec_vert)
        ra0 = ra_vert[0].asDegrees()
        ra1 = ra_vert[-1].asDegrees()
        dec0 = dec_vert[0].asDegrees()
        dec1 = dec_vert[-1].asDegrees()
        """
        Added by Song Huang 2016-09-02
        Tracts in the Deep and Wide layers are defined to have overlaps
        between the two adjacent Tracts: 1 arcmin ~ 0.01667 deg
        """
        if self.config.innerTract:
            ra0 += 0.0167
            ra1 -= 0.0167
            dec0 += 0.0167
            dec1 -= 0.0167

        radUse = self.config.rad
        raArr, decArr = np.array(
            zip(*makeRaDecCat.getRandomRaDec(
                nFakes, ra0, ra1, dec0, dec1, rad=radUse)))
        """
        Added by Song Huang 2016-09-01
        Filter the random RA, DEC using two filters
        """
        if (self.config.acpMask != '') or (self.config.rejMask != ''):
            try:
                from shapely import wkb
                from shapely.geometry import Point
                from shapely.prepared import prep

                # Filter through the accept mask
                acpUse = self.config.acpMask
                if (acpUse != '') and os.path.isfile(acpUse):
                    print "## Filter through : %s" % acpUse
                    acpWkb = open(acpUse, 'r')
                    acpRegs = wkb.loads(acpWkb.read().decode('hex'))
                    acpPrep = prep(acpRegs)
                    acpWkb.close()
                    inside = np.asarray(
                        map(lambda x, y: acpPrep.contains(Point(x, y)), raArr,
                            decArr))
                else:
                    inside = np.isfinite(raArr)

                # Filter through the reject mask
                rejUse = self.config.rejMask
                if (rejUse != '') and os.path.isfile(rejUse):
                    print "## Filter through : %s" % rejUse
                    rejWkb = open(rejUse, 'r')
                    rejRegs = wkb.loads(rejWkb.read().decode('hex'))
                    rejPrep = prep(rejRegs)
                    rejWkb.close()
                    masked = np.asarray(
                        map(lambda x, y: rejPrep.contains(Point(x, y)), raArr,
                            decArr))
                else:
                    masked = np.isnan(raArr)

                useful = np.asarray(
                    map(lambda x, y: x and (not y), inside, masked))
                ra, dec = raArr[useful], decArr[useful]

                print "## %d out of %d objects left" % (len(ra), len(raArr))

            except ImportError:
                warnings.warn('Can not import Shapely, no filter performed!')
                ra, dec = raArr, decArr
        else:
            ra, dec = raArr, decArr

        # Number of fakes that will be added
        nFakes = len(ra)
        # Create an empty astropy.Table object
        outTab = astropy.table.Table()
        # Add columns for Ra, Dec
        outTab.add_column(astropy.table.Column(name="RA", data=ra))
        outTab.add_column(astropy.table.Column(name="Dec", data=dec))
        if self.config.inputCat is not None:
            galData = astropy.table.Table().read(self.config.inputCat)
            randInd = np.random.choice(range(len(galData)), size=nFakes)
            mergedData = galData[randInd]

            for colname in mergedData.columns:
                colAdd = astropy.table.Column(name=colname,
                                              data=mergedData[colname])
                outTab.add_column(colAdd)
            # Replace ID with a unique integer (using index)
            if ('ID' in outTab.colnames) and (self.config.uniqueID):
                outTab.rename_column('ID', 'modelID')
                outTab.add_column(
                    astropy.table.Column(name="ID",
                                         data=np.arange(len(outTab))))

            # Generate multiBand catalog at the same time
            magList = [col for col in galData.colnames if 'mag_' in col]
            if len(magList) >= 1:
                print "Find magnitude in %d band(s)" % len(magList)
                for mag in magList:
                    try:
                        outTab.remove_column('mag')
                    except KeyError:
                        pass
                    outTab.add_column(
                        astropy.table.Column(name='mag', data=mergedData[mag]))
                    filt = mag.split('_')[1].upper()
                    outFits = os.path.join(
                        self.config.outDir,
                        'src_%d_radec_%s.fits' % (tractId, filt))
                    outTab.write(outFits, overwrite=True)
            else:
                outTab.write(os.path.join(self.config.outDir,
                                          'src_%d_radec.fits' % tractId),
                             overwrite=True)
        else:
            outTab.write(os.path.join(self.config.outDir,
                                      'src_%d_radec_only.fits' % tractId),
                         overwrite=True)
Exemplo n.º 3
0
    def run(self, dataRef):
        print dataRef.dataId
        skyMap = dataRef.get('deepCoadd_skyMap', immediate=True)
        tractId = dataRef.dataId['tract']
        tract = skyMap[tractId]
        extPatch = tract.getNumPatches()
        nPatch = extPatch.getX() * extPatch.getY()
        nFakes = self.config.rhoFakes * nPatch
        ra_vert, dec_vert = zip(*tract.getVertexList())
        ra_vert = sorted(ra_vert)
        dec_vert = sorted(dec_vert)
        ra0 = ra_vert[0].asDegrees()
        ra1 = ra_vert[-1].asDegrees()
        dec0 = dec_vert[0].asDegrees()
        dec1 = dec_vert[-1].asDegrees()
        """
        Added by Song Huang 2016-09-02
        Tracts in the Deep and Wide layers are defined to have overlaps of 1 arcmin
        between the two adjacent Tracts: 1 arcmin ~ 0.01667 deg
        """
        if self.config.innerTract:
            ra0 += 0.0167
            ra1 -= 0.0167
            dec0 += 0.0167
            dec1 -= 0.0167
        raArr, decArr = np.array(zip(*makeRaDecCat.getRandomRaDec(nFakes, ra0, ra1,
                                                                  dec0, dec1,
                                                                  rad=self.config.rad)))
        """
        Added by Song Huang 2016-09-01
        Filter the random RA, DEC using two filters
        """
        if self.config.acpMask is not None or self.config.rejMask is not None:
            try:
                from shapely import wkb
                from shapely.geometry import Point
            except ImportError:
                raise Exception('Please install the Shapely library before using this function')

            if os.path.isfile(self.config.acpMask):
                print "## Filter through : %s" % self.config.acpMask
                acpWkb = open(self.config.acpMask, 'r')
                acpRegs = wkb.loads(acpWkb.read().decode('hex'))
                acpWkb.close()
                inside = np.asarray(map(lambda x, y: acpRegs.contains(Point(x, y)),
                                    raArr, decArr))
            else:
                inside = np.isfinite(raArr)

            if os.path.isfile(self.config.rejMask):
                print "## Filter through : %s" % self.config.rejMask
                rejWkb = open(self.config.rejMask, 'r')
                rejRegs = wkb.loads(rejWkb.read().decode('hex'))
                rejWkb.close()
                masked = np.asarray(map(lambda x, y: rejRegs.contains(Point(x, y)),
                                    raArr, decArr))
            else:
                masked = np.isnan(raArr)

            useful = np.asarray(map(lambda x, y: x and (not y), inside, masked))
            ra, dec = raArr[useful], decArr[useful]

            print "## %d out of %d objects left" % (len(ra), len(raArr))
            nFakes = len(ra)
        else:
            ra, dec = raArr, decArr

        outTab = astropy.table.Table()
        outTab.add_column(astropy.table.Column(name="RA", data=ra))
        outTab.add_column(astropy.table.Column(name="Dec", data=dec))
        if self.config.inputCat is not None:
            galData = astropy.table.Table().read(self.config.inputCat)
            randInd = np.random.choice(range(len(galData)), size=nFakes)
            mergedData = galData[randInd]

            for colname in mergedData.columns:
                outTab.add_column(astropy.table.Column(name=colname,
                                                   data=mergedData[colname]))

            """ Modified by SH; to generate multiBand catalog at the same time"""
            magList = [col for col in galData.colnames if 'mag_' in col]
            if len(magList) >= 1:
                print "Find magnitude in %d band(s)"%len(magList)
                for mag in magList:
                    try:
                        outTab.remove_column('mag')
                    except KeyError:
                        pass
                    outTab.add_column(astropy.table.Column(name='mag',
                                                        data=mergedData[mag]))
                    outFits = os.path.join(self.config.outDir,
                            'src_%d_radec_%s.fits'%(tractId, mag.split('_')[1].upper()))
                    print outFits
                    outTab.write(outFits, overwrite=True)
            else:
                outTab.write(os.path.join(self.config.outDir,
                                          'src_%d_radec.fits'%tractId),
                             overwrite=True)
        else:
            outTab.write(os.path.join(self.config.outDir,
                                      'src_%d_radec_only.fits'%tractId),
                         overwrite=True)
Exemplo n.º 4
0
    def run(self, dataRef):
        print dataRef.dataId
        skyMap = dataRef.get('deepCoadd_skyMap', immediate=True)
        tractId = dataRef.dataId['tract']
        tract = skyMap[tractId]
        extPatch = tract.getNumPatches()
        nPatch = extPatch.getX() * extPatch.getY()
        nFakes = self.config.rhoFakes * nPatch
        ra_vert, dec_vert = zip(*tract.getVertexList())
        ra_vert = sorted(ra_vert)
        dec_vert = sorted(dec_vert)
        ra0 = ra_vert[0].asDegrees()
        ra1 = ra_vert[-1].asDegrees()
        dec0 = dec_vert[0].asDegrees()
        dec1 = dec_vert[-1].asDegrees()
        """
        Added by Song Huang 2016-09-02
        Tracts in the Deep and Wide layers are defined to have overlaps of 1 arcmin
        between the two adjacent Tracts: 1 arcmin ~ 0.01667 deg
        """
        if self.config.innerTract:
            ra0 += 0.0167
            ra1 -= 0.0167
            dec0 += 0.0167
            dec1 -= 0.0167
        raArr, decArr = np.array(
            zip(*makeRaDecCat.getRandomRaDec(
                nFakes, ra0, ra1, dec0, dec1, rad=self.config.rad)))
        """
        Added by Song Huang 2016-09-01
        Filter the random RA, DEC using two filters
        """
        if self.config.acpMask is not None or self.config.rejMask is not None:
            try:
                from shapely import wkb
                from shapely.geometry import Point
            except ImportError:
                raise Exception(
                    'Please install the Shapely library before using this function'
                )

            if os.path.isfile(self.config.acpMask):
                print "## Filter through : %s" % self.config.acpMask
                acpWkb = open(self.config.acpMask, 'r')
                acpRegs = wkb.loads(acpWkb.read().decode('hex'))
                acpWkb.close()
                inside = np.asarray(
                    map(lambda x, y: acpRegs.contains(Point(x, y)), raArr,
                        decArr))
            else:
                inside = np.isfinite(raArr)

            if os.path.isfile(self.config.rejMask):
                print "## Filter through : %s" % self.config.rejMask
                rejWkb = open(self.config.rejMask, 'r')
                rejRegs = wkb.loads(rejWkb.read().decode('hex'))
                rejWkb.close()
                masked = np.asarray(
                    map(lambda x, y: rejRegs.contains(Point(x, y)), raArr,
                        decArr))
            else:
                masked = np.isnan(raArr)

            useful = np.asarray(map(lambda x, y: x and (not y), inside,
                                    masked))
            ra, dec = raArr[useful], decArr[useful]

            print "## %d out of %d objects left" % (len(ra), len(raArr))
            nFakes = len(ra)
        else:
            ra, dec = raArr, decArr

        outTab = astropy.table.Table()
        outTab.add_column(astropy.table.Column(name="RA", data=ra))
        outTab.add_column(astropy.table.Column(name="Dec", data=dec))
        if self.config.inputCat is not None:
            galData = astropy.table.Table().read(self.config.inputCat)
            randInd = np.random.choice(range(len(galData)), size=nFakes)
            mergedData = galData[randInd]

            for colname in mergedData.columns:
                outTab.add_column(
                    astropy.table.Column(name=colname,
                                         data=mergedData[colname]))
            """ Modified by SH; to generate multiBand catalog at the same time"""
            magList = [col for col in galData.colnames if 'mag_' in col]
            if len(magList) >= 1:
                print "Find magnitude in %d band(s)" % len(magList)
                for mag in magList:
                    try:
                        outTab.remove_column('mag')
                    except KeyError:
                        pass
                    outTab.add_column(
                        astropy.table.Column(name='mag', data=mergedData[mag]))
                    outFits = os.path.join(
                        self.config.outDir, 'src_%d_radec_%s.fits' %
                        (tractId, mag.split('_')[1].upper()))
                    print outFits
                    outTab.write(outFits, overwrite=True)
            else:
                outTab.write(os.path.join(self.config.outDir,
                                          'src_%d_radec.fits' % tractId),
                             overwrite=True)
        else:
            outTab.write(os.path.join(self.config.outDir,
                                      'src_%d_radec_only.fits' % tractId),
                         overwrite=True)